/*
# This file is part of the Astrometry.net suite.
# Licensed under a 3-clause BSD style license - see LICENSE
*/

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>

#include "starutil.h"
#include "mathutil.h"
#include "boilerplate.h"
#include "bl.h"
#include "solvedfile.h"

const char* OPTIONS = "hum:SM:jwps";

void printHelp(char* progname) {
	BOILERPLATE_HELP_HEADER(stderr);
	fprintf(stderr, "\nUsage: %s <solved-file> ...\n"
			"    [-s]: just summary info, no field numbers\n"
			"    [-p]: print percent solved\n"
			"    [-u]: print UNsolved fields\n"
			"    [-j]: just the field numbers, no headers, etc.\n"
			"    [-m <max-field>]: for unsolved mode, max field number.\n"
			"    [-w]: format for the wiki.\n"
			"    [-M <variable-name>]: format for Matlab.\n"
			"\n", progname);
}


int main(int argc, char** args) {
    int argchar;
	char* progname = args[0];
	char** inputfiles = NULL;
	int ninputfiles = 0;
	int i;
	anbool unsolved = FALSE;
	int maxfield = 0;
	anbool wiki = FALSE;
	char* matlab = NULL;
	anbool justnums = FALSE;
	anbool percent = FALSE;
	anbool printinfo = FALSE;
	anbool printnums = FALSE;
	anbool summary = FALSE;
	int ncounted = 0;
	int ntotal = 0;

    while ((argchar = getopt (argc, args, OPTIONS)) != -1) {
		switch (argchar) {
		case 's':
			summary = TRUE;
			break;
		case 'p':
			percent = TRUE;
			break;
		case 'j':
			justnums = TRUE;
			break;
		case 'M':
			matlab = optarg;
			break;
		case 'w':
			wiki = TRUE;
			break;
		case 'u':
			unsolved = TRUE;
			break;
		case 'm':
			maxfield = atoi(optarg);
			break;
		case 'h':
		default:
			printHelp(progname);
			exit(-1);
		}
	}
	if (optind < argc) {
		ninputfiles = argc - optind;
		inputfiles = args + optind;
	} else {
		printHelp(progname);
		exit(-1);
	}

	printinfo = (!matlab && !justnums);
	printnums = !summary;

	if (matlab)
		printf("%s=[", matlab);

	for (i=0; i<ninputfiles; i++) {
		int j;
		il* list;

		if (printinfo)
			printf("File %s\n", inputfiles[i]);
		if (wiki)
			printf("|| %i || ", i+1);

		if (unsolved)
			list = solvedfile_getall(inputfiles[i], 1, maxfield, 0);
		else
			list = solvedfile_getall_solved(inputfiles[i], 1, maxfield, 0);

		if (percent) {
			int nt = solvedfile_getsize(inputfiles[i]);
			int nc = il_size(list);
			printf("%s: %i/%i (%f %%) %ssolved\n", inputfiles[i], nc, nt, (100.0 * nc / (double)nt), unsolved ? "un":"");
			ncounted += nc;
			ntotal += nt;
		}

		if (!list) {
			fprintf(stderr, "Failed to get list of fields.\n");
			exit(-1);
		}
		if (printnums) {
			for (j=0; j<il_size(list); j++)
				printf("%i ", il_get(list, j));
		}
		il_free(list);

		if (wiki)
			printf(" ||");

		printf("\n");
	}
	if (matlab)
		printf("];\n");

	if (percent) {
		printf("Total: %i/%i (%f %%) %ssolved\n", ncounted, ntotal, (100.0 * ncounted / (double)ntotal), unsolved ? "un":"");
	}

	return 0;
}