#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <grass/gis.h>
#include <grass/glocale.h>
#include <grass/spawn.h>
#include "global.h"

static int die(void);

int get_stats(void)
{
    char buf[1024];
    int i, nl, ns;
    FILE *fd;
    char **tokens;

    if (stats_flag == EVERYTHING)
	stats_file = G_tempfile();

    if (stats_flag != REPORT_ONLY) {
	char tmp[50];
	int n_argv = 50;
	const char **argv = G_calloc(n_argv, sizeof(*argv));
	int argc = 0;

	argv[argc++] = "r.stats";
	argv[argc++] = "-acr";

	/* if (!masking) argv[argc++] = "-m"; */
	if (G_verbose() == G_verbose_min())
	    argv[argc++] = "--quiet";

	if (no_nulls)
	    argv[argc++] = "-n";

	if (no_nulls_all)
	    argv[argc++] = "-N";

	if (as_int)
	    argv[argc++] = "-i";

	if (cat_ranges)
	    argv[argc++] = "-C";
	else if (nsteps != 255) {
	    sprintf(tmp, "nsteps=%d", nsteps);
	    argv[argc++] = tmp;
	}

	argv[argc++] = "separator=:";

	argv[argc++] = SF_REDIRECT_FILE;
	argv[argc++] = SF_STDOUT;
	argv[argc++] = SF_MODE_OUT;
	argv[argc++] = stats_file;

        if (do_sort == SORT_ASC)
            argv[argc++] = "sort=asc";
        else if (do_sort == SORT_DESC)
            argv[argc++] = "sort=desc";
        
	for (i = 0; i < nlayers; i++) {
	    char *name = G_fully_qualified_name(layers[i].name, layers[i].mapset);
	    char *buf = G_malloc(6 + strlen(name) + 1);

	    sprintf(buf, "input=%s", name);
	    G_free(name);

	    if (argc + 1 >= n_argv) {
		n_argv += 50;
		argv = G_realloc(argv, n_argv * sizeof(*argv));
	    }

	    argv[argc++] = buf;
	}

	argv[argc++] = NULL;

	if (G_vspawn_ex(argv[0], argv) != 0) {
	    remove(stats_file);
	    G_fatal_error("error running r.stats");
	}
    }

    if (stats_flag == STATS_ONLY)
	return 0;

    fd = fopen(stats_file, "r");

    if (fd == NULL) {
	if (stats_flag == EVERYTHING)
	    unlink(stats_file);
	G_fatal_error(_("Unable to open result file <%s>"), stats_file);
    }

    while (G_getl(buf, sizeof buf, fd)) {
	tokens = G_tokenize(buf, ":");
	i = 0;
	ns = nstats++;
	Gstats = (GSTATS *) G_realloc(Gstats, nstats * sizeof(GSTATS));
	Gstats[ns].cats = (CELL *) G_calloc(nlayers, sizeof(long));
	for (nl = 0; nl < nlayers; nl++) {
	    if (sscanf(tokens[i], "%d", &Gstats[ns].cats[nl]) != 1) {
		if (tokens[i][0] == '*')
		    Rast_set_c_null_value(&Gstats[ns].cats[nl], 1);
		else
		    die();
	    }
	    i++;
	}
	if (sscanf(tokens[i++], "%lf", &Gstats[ns].area) != 1)
	    die();
	if (sscanf(tokens[i++], "%ld", &Gstats[ns].count) != 1)
	    die();
	G_free_tokens(tokens);
    }
    fclose(fd);
    if (stats_flag == EVERYTHING)
	unlink(stats_file);

    return 0;
}

static int die(void)
{
    if (stats_flag == EVERYTHING)
	unlink(stats_file);
    G_fatal_error(_("Problem reading r.stats output"));
}