#include <stdio.h>
#include <errno.h>
#include <stdint.h>
#include <string.h>
#include <stdlib.h>

#include "coord_query.h"
#include "timing.h"

int coord_query_loop(int argc, char** argv, mk_index_fn mk_index, free_index_fn free_index, lookup_fn lookup) {
    if (argc != 2) {
        fprintf(stderr, "Usage: %s FILE\n", argv[0]);
        exit(1);
    }

    uint64_t start, runtime;
    int n;

    start = microseconds();
    struct record *rs = read_records(argv[1], &n);
    runtime = microseconds()-start;

    if (rs) {
        printf("Reading records: %dms\n", (int)runtime/1000);

        start = microseconds();
        void *index = mk_index(rs, n);
        runtime = microseconds()-start;
        printf("Building index: %dms\n", (int)runtime/1000);

        char *line = NULL;
        size_t line_len;

        uint64_t runtime_sum = 0;
        while (getline(&line, &line_len, stdin) != -1) {
            double lon, lat;
            sscanf(line, "%lf %lf", &lon, &lat);

            start = microseconds();
            const struct record *r = lookup(index, lon, lat);
            runtime = microseconds()-start;

            if (r) {
                printf("(%f,%f): %s (%f,%f)\n", lon, lat, r->name, r->lon, r->lat);
            } else {
                printf("(%f,%f): not found\n", lon, lat);
            }

            printf("Query time: %dus\n", (int)runtime);
            runtime_sum += runtime;
        }

        printf("Total query runtime: %dus\n", (int)runtime_sum);

        free(line);
        free_index(index);
        free_records(rs, n);
        return 0;
    } else {
        fprintf(stderr, "Failed to read input from %s (errno: %s)\n",
                        argv[1], strerror(errno));
        return 1;
    }
}