printsolved.c

/*
 # 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;
}