#define i_implement
#include <stc/cstr.h>

// Olympics multimap example

struct OlympicsData { int year; const char *city, *country, *date; } ol_data[] = {
    {2026, "Milan and Cortina d'Ampezzo", "Italy", "February 6-22"},
    {2022, "Beijing", "China", "February 4-20"},
    {2018, "PyeongChang", "South Korea", "February 9-25"},
    {2014, "Sochi", "Russia", "February 7-23"},
    {2010, "Vancouver", "Canada", "February 12-28"},
    {2006, "Torino", "Italy", "February 10-26"},
    {2002, "Salt Lake City", "United States", "February 8-24"},
    {1998, "Nagano", "Japan", "February 7-22"},
    {1994, "Lillehammer", "Norway", "February 12-27"},
    {1992, "Albertville", "France", "February 8-23"},
    {1988, "Calgary", "Canada", "February 13-28"},
    {1984, "Sarajevo", "Yugoslavia", "February 8-19"},
    {1980, "Lake Placid", "United States", "February 13-24"},
    {1976, "Innsbruck", "Austria", "February 4-15"},
    {1972, "Sapporo", "Japan", "February 3-13"},
    {1968, "Grenoble", "France", "February 6-18"},
    {1964, "Innsbruck", "Austria", "January 29-February 9"},
    {1960, "Squaw Valley", "United States", "February 18-28"},
    {1956, "Cortina d'Ampezzo", "Italy", "January 26 - February 5"},
    {1952, "Oslo", "Norway", "February 14 - 25"},
    {1948, "St. Moritz", "Switzerland", "January 30 - February 8"},
    {1944, "canceled", "canceled", "canceled"},
    {1940, "canceled", "canceled", "canceled"},
    {1936, "Garmisch-Partenkirchen", "Germany", "February 6 - 16"},
    {1932, "Lake Placid", "United States", "February 4 - 15"},
    {1928, "St. Moritz", "Switzerland", "February 11 - 19"},
    {1924, "Chamonix", "France", "January 25 - February 5"},
};

typedef struct { int year; cstr city, date; } OlympicLoc;

int OlympicLoc_cmp(const OlympicLoc* a, const OlympicLoc* b);
OlympicLoc OlympicLoc_clone(OlympicLoc loc);
void OlympicLoc_drop(OlympicLoc* self);

// Create a clist<OlympicLoc>, can be sorted by year.
#define i_keyclass OlympicLoc // binds _cmp, _clone and _drop.
#define i_use_cmp
#define i_tag OL
#include <stc/clist.h>

// Create a csmap<cstr, clist_OL> where key is country name
#define i_key_str           // binds cstr_equ, cstr_hash, cstr_clone, ++
#define i_valclass clist_OL // binds clist_OL_clone, clist_OL_drop
#define i_tag OL
#include <stc/csmap.h>

int OlympicLoc_cmp(const OlympicLoc* a, const OlympicLoc* b) {
    return a->year - b->year;
}

OlympicLoc OlympicLoc_clone(OlympicLoc loc) {
    loc.city = cstr_clone(loc.city);
    loc.date = cstr_clone(loc.date);
    return loc;
}

void OlympicLoc_drop(OlympicLoc* self) {
    cstr_drop(&self->city);
    cstr_drop(&self->date);
}


int main(void)
{
    // Define the multimap with destructor defered to when block is completed.
    csmap_OL multimap = {0};
    const clist_OL empty = clist_OL_init();

    for (size_t i = 0; i < c_arraylen(ol_data); ++i)
    {
        struct OlympicsData* d = &ol_data[i];
        OlympicLoc loc = {.year = d->year,
                            .city = cstr_from(d->city),
                            .date = cstr_from(d->date)};
        // Insert an empty list for each new country, and append the entry to the list.
        // If country already exist in map, its list is returned from the insert function.
        clist_OL* list = &csmap_OL_emplace(&multimap, d->country, empty).ref->second;
        clist_OL_push_back(list, loc);
    }

    // Sort locations by year for each country.
    c_foreach (country, csmap_OL, multimap)
        clist_OL_sort(&country.ref->second);

    // Print the multimap:
    c_foreach (country, csmap_OL, multimap)
    {
        // Loop the locations for a country sorted by year
        c_foreach (loc, clist_OL, country.ref->second)
            printf("%s: %d, %s, %s\n", cstr_str(&country.ref->first),
                                                    loc.ref->year,
                                        cstr_str(&loc.ref->city),
                                        cstr_str(&loc.ref->date));
    }
    csmap_OL_drop(&multimap);
}