/*
 * r3.univar
 *
 *  Calculates univariate statistics from the non-null 3d cells of a raster3d
 *  map
 *
 *   Copyright (C) 2004-2007 by the GRASS Development Team
 *   Author(s): Soeren Gebbert
 *              Based on r.univar from Hamish Bowman, University of Otago, New
 *              Zealand and Martin Landa heapsort code from
 *              http://de.wikipedia.org/wiki/Heapsort Zonal stats: Markus Metz
 *
 *      This program is free software under the GNU General Public
 *      License (>=v2). Read the file COPYING that comes with GRASS
 *      for details.
 *
 */

#include <string.h>
#include "globals.h"
#include "grass/raster3d.h"

param_type param;
zone_type zone_info;

/* ************************************************************************* */
/* Set up the arguments we are expecting ********************************** */
/* ************************************************************************* */
void set_params(void)
{
    param.inputfile = G_define_standard_option(G_OPT_R3_MAP);

    param.zonefile = G_define_standard_option(G_OPT_R3_INPUT);
    param.zonefile->key = "zones";
    param.zonefile->required = NO;
    param.zonefile->description =
        _("3D Raster map used for zoning, must be of type CELL");

    param.output_file = G_define_standard_option(G_OPT_F_OUTPUT);
    param.output_file->required = NO;
    param.output_file->description =
        _("Name for output file (if omitted or \"-\" output to stdout)");

    param.percentile = G_define_option();
    param.percentile->key = "percentile";
    param.percentile->type = TYPE_DOUBLE;
    param.percentile->required = NO;
    param.percentile->multiple = YES;
    param.percentile->options = "0-100";
    param.percentile->answer = "90";
    param.percentile->description =
        _("Percentile to calculate (requires extended statistics flag)");

    param.separator = G_define_standard_option(G_OPT_F_SEP);

    param.shell_style = G_define_flag();
    param.shell_style->key = 'g';
    param.shell_style->description = _("Print the stats in shell script style");

    param.extended = G_define_flag();
    param.extended->key = 'e';
    param.extended->description = _("Calculate extended statistics");

    param.table = G_define_flag();
    param.table->key = 't';
    param.table->description =
        _("Table output format instead of standard output format");

    return;
}

/* *************************************************************** */
/* **** the main functions for r3.univar ************************* */
/* *************************************************************** */
int main(int argc, char *argv[])
{
    FCELL val_f; /* for misc use */
    DCELL val_d; /* for misc use */
    int map_type, zmap_type;
    univar_stat *stats;

    char *infile, *zonemap;
    void *map, *zmap = NULL;
    RASTER3D_Region region;
    unsigned int i;
    unsigned int rows, cols, depths;
    unsigned int x, y, z;
    double dmin, dmax;
    int zone, n_zones /* , use_zone = 0 */;
    const char *mapset, *name;

    struct GModule *module;

    G_gisinit(argv[0]);

    module = G_define_module();
    G_add_keyword(_("raster3d"));
    G_add_keyword(_("statistics"));
    G_add_keyword(_("univariate statistics"));

    module->label = _("Calculates univariate statistics from the non-null "
                      "cells of a 3D raster map.");
    module->description =
        _("Statistics include number of cells counted, minimum and maximum cell"
          " values, range, arithmetic mean, population variance, standard "
          "deviation,"
          " coefficient of variation, and sum.");

    /* Define the different options */
    set_params();

    if (G_parser(argc, argv))
        exit(EXIT_FAILURE);

    /* Set the defaults */
    Rast3d_init_defaults();

    /* get the current region */
    Rast3d_get_window(&region);

    cols = region.cols;
    rows = region.rows;
    depths = region.depths;

    name = param.output_file->answer;
    if (name != NULL && strcmp(name, "-") != 0) {
        if (NULL == freopen(name, "w", stdout)) {
            G_fatal_error(_("Unable to open file <%s> for writing"), name);
        }
    }

    /* table field separator */
    zone_info.sep = G_option_to_separator(param.separator);

    dmin = NAN;
    dmax = NAN;
    zone_info.min = 0;
    zone_info.max = 0;
    zone_info.n_zones = 0;

    /* open 3D zoning raster with default region */
    if ((zonemap = param.zonefile->answer) != NULL) {
        if (NULL == (mapset = G_find_raster3d(zonemap, "")))
            Rast3d_fatal_error(_("3D raster map <%s> not found"), zonemap);

        zmap = Rast3d_open_cell_old(zonemap, G_find_raster3d(zonemap, ""),
                                    &region, RASTER3D_TILE_SAME_AS_FILE,
                                    RASTER3D_USE_CACHE_DEFAULT);

        if (zmap == NULL)
            Rast3d_fatal_error(_("Unable to open 3D raster map <%s>"), zonemap);

        zmap_type = Rast3d_tile_type_map(zmap);

        if (Rast3d_read_cats(zonemap, mapset, &(zone_info.cats)))
            G_warning("No category support for zoning raster");

        Rast3d_range_init(zmap);
        Rast3d_range_load(zmap);
        Rast3d_range_min_max(zmap, &dmin, &dmax);

        /* properly round dmin and dmax */
        if (dmin < 0)
            zone_info.min = dmin - 0.5;
        else
            zone_info.min = dmin + 0.5;
        if (dmax < 0)
            zone_info.max = dmax - 0.5;
        else
            zone_info.max = dmax + 0.5;

        G_debug(1, "min: %d, max: %d", zone_info.min, zone_info.max);
        zone_info.n_zones = zone_info.max - zone_info.min + 1;

        /* use_zone = 1; */
    }

    /* Open 3D input raster with default region */
    infile = param.inputfile->answer;

    if (NULL == G_find_raster3d(infile, ""))
        Rast3d_fatal_error(_("3D raster map <%s> not found"), infile);

    map = Rast3d_open_cell_old(infile, G_find_raster3d(infile, ""), &region,
                               RASTER3D_TILE_SAME_AS_FILE,
                               RASTER3D_USE_CACHE_DEFAULT);

    if (map == NULL)
        Rast3d_fatal_error(_("Unable to open 3D raster map <%s>"), infile);

    map_type = Rast3d_tile_type_map(map);

    i = 0;
    while (param.percentile->answers[i])
        i++;

    n_zones = zone_info.n_zones;

    if (n_zones == 0)
        n_zones = 1;

    stats = create_univar_stat_struct(map_type, i);
    for (i = 0; i < (unsigned int)n_zones; i++) {
        unsigned int j;

        for (j = 0; j < stats[i].n_perc; j++) {
            sscanf(param.percentile->answers[j], "%lf", &(stats[i].perc[j]));
        }
    }

    for (z = 0; z < depths; z++) { /* From the bottom to the top */
        if (!(param.shell_style->answer))
            G_percent(z, depths - 1, 2);
        for (y = 0; y < rows; y++) {
            for (x = 0; x < cols; x++) {
                zone = 0;
                if (zone_info.n_zones) {
                    if (zmap_type == FCELL_TYPE) {
                        Rast3d_get_value(zmap, x, y, z, &val_f, FCELL_TYPE);
                        if (Rast3d_is_null_value_num(&val_f, FCELL_TYPE))
                            continue;
                        if (val_f < 0)
                            zone = val_f - 0.5;
                        else
                            zone = val_f + 0.5;
                    }
                    else if (zmap_type == DCELL_TYPE) {
                        Rast3d_get_value(zmap, x, y, z, &val_d, DCELL_TYPE);
                        if (Rast3d_is_null_value_num(&val_d, DCELL_TYPE))
                            continue;
                        if (val_d < 0)
                            zone = val_d - 0.5;
                        else
                            zone = val_d + 0.5;
                    }
                    zone -= zone_info.min;
                }
                if (map_type == FCELL_TYPE) {
                    Rast3d_get_value(map, x, y, z, &val_f, map_type);
                    if (!Rast3d_is_null_value_num(&val_f, map_type)) {
                        if (param.extended->answer) {
                            if (stats[zone].n >= stats[zone].n_alloc) {
                                size_t msize;

                                stats[zone].n_alloc += 1000;
                                msize = stats[zone].n_alloc * sizeof(FCELL);
                                stats[zone].fcell_array = (FCELL *)G_realloc(
                                    (void *)stats[zone].fcell_array, msize);
                            }

                            stats[zone].fcell_array[stats[zone].n] = val_f;
                        }

                        stats[zone].sum += val_f;
                        stats[zone].sumsq += (val_f * val_f);
                        stats[zone].sum_abs += fabs(val_f);

                        if (stats[zone].first) {
                            stats[zone].max = val_f;
                            stats[zone].min = val_f;
                            stats[zone].first = FALSE;
                        }
                        else {
                            if (val_f > stats[zone].max)
                                stats[zone].max = val_f;
                            if (val_f < stats[zone].min)
                                stats[zone].min = val_f;
                        }
                        stats[zone].n++;
                    }
                    stats[zone].size++;
                }
                else if (map_type == DCELL_TYPE) {
                    Rast3d_get_value(map, x, y, z, &val_d, map_type);
                    if (!Rast3d_is_null_value_num(&val_d, map_type)) {
                        if (param.extended->answer) {
                            if (stats[zone].n >= stats[zone].n_alloc) {
                                size_t msize;

                                stats[zone].n_alloc += 1000;
                                msize = stats[zone].n_alloc * sizeof(DCELL);
                                stats[zone].dcell_array = (DCELL *)G_realloc(
                                    (void *)stats[zone].dcell_array, msize);
                            }

                            stats[zone].dcell_array[stats[zone].n] = val_d;
                        }

                        stats[zone].sum += val_d;
                        stats[zone].sumsq += val_d * val_d;
                        stats[zone].sum_abs += fabs(val_d);

                        if (stats[zone].first) {
                            stats[zone].max = val_d;
                            stats[zone].min = val_d;
                            stats[zone].first = FALSE;
                        }
                        else {
                            if (val_d > stats[zone].max)
                                stats[zone].max = val_d;
                            if (val_d < stats[zone].min)
                                stats[zone].min = val_d;
                        }
                        stats[zone].n++;
                    }
                    stats[zone].size++;
                }
            }
        }
    }

    /* close maps */
    Rast3d_close(map);
    if (zone_info.n_zones)
        Rast3d_close(zmap);

    /* create the output */
    if (param.table->answer)
        print_stats_table(stats);
    else
        print_stats(stats);

    /* release memory */
    free_univar_stat_struct(stats);

    exit(EXIT_SUCCESS);
}
