/*********************************************************************
 *   Copyright 1993, UCAR/Unidata
 *   See netcdf/COPYRIGHT file for copying and redistribution conditions.
 *********************************************************************/

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

#include "hdf.h"

#ifdef H4_HAVE_NETCDF
#include <netcdf.h>
#else
#include <hdf4_netcdf.h>
#endif

#include "ncdump.h"
#include "dumplib.h"
#include "vardata.h"

/*
 * Function from ncdump.c. "Fixes" variable names to remove spaces and other
 * "illegal" characters.
 */
extern char *sanitize_string(char *str, bool fix_str);

static void annotate(struct ncvar *vp, struct fspec *fsp, long cor[], long iel);

#define VD_STREQ(a, b) (*(a) == *(b) && strcmp((a), (b)) == 0)

/*
 * Print a row of variable values.  Makes sure output lines aren't too long
 * by judiciously inserting newlines.
 */
/* vp - variable */
/* len - number of values to print */
/* fmt
 * printf format used for each value.  If
 * nc_type is NC_CHAR and this is NULL,
 * character arrays will be printed as strings
 * enclosed in quotes.
 */
/* more
 * true if more data will follow, so add
 * trailing comma
 */
/* lastrow
 * true if this is the last row for this
 * variable, so terminate with ";" instead of
 * ","
 */
/* vals pointer to block of values */
static void
pr_vals(struct ncvar *vp, long len, char *fmt, bool more, bool lastrow, void *vals)
{
    long iel;
    union {
        char   *cp;
        short  *sp;
        nclong *lp;
        float  *fp;
        double *dp;
    } gp;
    char         *sp;
    unsigned char uc;
    float         fill_float;
    double        fill_double;
    char          sout[100]; /* temporary string for each encoded output */

    fill_float  = FILL_FLOAT; /* static initialization hits ultrix cc bug */
    fill_double = FILL_DOUBLE;
    switch (vp->type) {
        case NC_BYTE:
            gp.cp = (char *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                (void)sprintf(sout, fmt, *gp.cp++);
                (void)strcat(sout, ", ");
                lput(sout);
            }
            (void)sprintf(sout, fmt, *gp.cp++);
            lput(sout);
            break;
        case NC_CHAR:
            gp.cp = (char *)vals;
            if (fmt == 0 || VD_STREQ(fmt, "%s") || VD_STREQ(fmt, "")) { /* as string */
                printf("\"");
                /* adjust len so trailing nulls don't get printed */
                sp = gp.cp + len;
                while (len > 0 && *--sp == '\0')
                    len--;
                for (iel = 0; iel < len; iel++)
                    switch (uc = *gp.cp++ & 0377) {
                        case '\b':
                            printf("\\b");
                            break;
                        case '\f':
                            printf("\\f");
                            break;
                        case '\n': /* generate linebreaks after new-lines */
                            printf("\\n\",\n    \"");
                            break;
                        case '\r':
                            printf("\\r");
                            break;
                        case '\t':
                            printf("\\t");
                            break;
                        case '\v':
                            printf("\\v");
                            break;
                        case '\\':
                            printf("\\\\");
                            break;
                        case '\'':
                            printf("\\\'");
                            break;
                        case '\"':
                            printf("\\\"");
                            break;
                        default:
                            if (isprint(uc))
                                printf("%c", uc);
                            else
                                printf("\\%.3o", uc);
                            break;
                    }
                printf("\"");
            }
            else { /* use format from C_format attribute */
                for (iel = 0; iel < len - 1; iel++) {
                    (void)sprintf(sout, fmt, *gp.cp++);
                    (void)strcat(sout, ", ");
                    lput(sout);
                }
                (void)sprintf(sout, fmt, *gp.cp++);
                lput(sout);
            }
            break;
        case NC_SHORT:
            gp.sp = (short *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                (void)sprintf(sout, fmt, *gp.sp++);
                (void)strcat(sout, ", ");
                lput(sout);
            }
            (void)sprintf(sout, fmt, *gp.sp++);
            lput(sout);
            break;
        case NC_LONG:
            gp.lp = (nclong *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                (void)sprintf(sout, fmt, *gp.lp++);
                (void)strcat(sout, ", ");
                lput(sout);
            }
            (void)sprintf(sout, fmt, *gp.lp++);
            lput(sout);
            break;
        case NC_FLOAT:
            gp.fp = (float *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                if (*gp.fp >= fill_float)
                    (void)sprintf(sout, "FloatInf, ");
                else {
                    (void)sprintf(sout, fmt, (double)*gp.fp);
                    (void)strcat(sout, ", ");
                }
                lput(sout);
                gp.fp++;
            }
            if (*gp.fp >= fill_float)
                (void)sprintf(sout, "FloatInf");
            else
                (void)sprintf(sout, fmt, (double)*gp.fp);
            lput(sout);
            gp.fp++;
            break;
        case NC_DOUBLE:
            gp.dp = (double *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                if (*gp.dp >= fill_double)
                    (void)sprintf(sout, "DoubleInf, ");
                else {
                    (void)sprintf(sout, fmt, *gp.dp);
                    (void)strcat(sout, ", ");
                }
                lput(sout);
                gp.dp++;
            }
            if (*gp.dp >= fill_double)
                (void)sprintf(sout, "DoubleInf");
            else
                (void)sprintf(sout, fmt, *gp.dp);
            lput(sout);
            gp.dp++;
            break;
        default:
            error("pr_vals: bad type");
    }
    if (more) {
        lput(", ");
    }
    else {
        if (lastrow) {
            lput(" ;");
            lput("\n");
        }
        else {
            lput(",\n");
            lput("  ");
        }
    }
}

/*
 * print last delimiter in each line before annotation (, or ;)
 */
static void
lastdelim(bool more, bool lastrow)
{
    if (more) {
        printf(", ");
    }
    else {
        if (lastrow) {
            printf(";");
        }
        else {
            printf(",");
        }
    }
}

/*
 * Annotates a value in data section with var name and indices in comment
 */
/* vp  - variable */
/* fsp - formatting specs */
/* cor - corner coordinates */
/* iel - which element in current row */
static void
annotate(struct ncvar *vp, struct fspec *fsp, long cor[], long iel)
{
    int vrank = vp->ndims;
    int id;

    /* print indices according to data_lang */
    printf("  // %s(", vp->name);
    switch (fsp->data_lang) {
        case LANG_C:
            /* C variable indices */
            for (id = 0; id < vrank - 1; id++)
                printf("%d,", (int)cor[id]);
            printf("%d", (int)(cor[id] + iel));
            break;
        case LANG_F:
            /* Fortran variable indices */
            printf("%d", (int)(cor[vrank - 1] + iel + 1));
            for (id = vrank - 2; id >= 0; id--) {
                printf(",%d", (int)(1 + cor[id]));
            }
            break;
        case LANG_NONE:
        default:
            /* Might want to make this an error */
            break;
    }
    printf(")\n    ");
}

/*
 * Print a number of commented variable values, where the comments for each
 * value identify the variable, and each dimension index.
 */
/* vp - variable */
/* len - number of values to print */
/* fmt
 * printf format used for each value.  If
 * nc_type is NC_CHAR and this is NULL,
 * character arrays will be printed as strings
 * enclosed in quotes.
 */
/* more
 * true if more data for this row will follow,
 * so add trailing comma
 */
/* lastrow
 * true if this is the last row for this
 * variable, so terminate with ";" instead of
 * ","
 */
/* vals - pointer to block of values */
/* fsp  - formatting specs */
/* cor  - corner coordinates */
static void
pr_cvals(struct ncvar *vp, long len, char *fmt, bool more, bool lastrow, void *vals, struct fspec *fsp,
         long cor[])
{
    long iel;
    union {
        char   *cp;
        short  *sp;
        nclong *lp;
        float  *fp;
        double *dp;
    } gp;
    char         *sp;
    unsigned char uc;
    float         fill_float;
    double        fill_double;

    fill_float  = FILL_FLOAT; /* static initialization hits ultrix cc bug */
    fill_double = FILL_DOUBLE;
    switch (vp->type) {
        case NC_BYTE:
            gp.cp = (char *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                printf(fmt, *gp.cp++);
                printf(", ");
                annotate(vp, fsp, cor, iel);
            }
            printf(fmt, *gp.cp++);
            lastdelim(more, lastrow);
            annotate(vp, fsp, cor, iel);
            break;
        case NC_CHAR:
            gp.cp = (char *)vals;
            if (fmt == 0 || VD_STREQ(fmt, "%s") || VD_STREQ(fmt, "")) { /* as string */
                printf("\"");
                /* adjust len so trailing nulls don't get printed */
                sp = gp.cp + len;
                while (len > 0 && *--sp == '\0')
                    len--;
                for (iel = 0; iel < len; iel++)
                    switch (uc = *gp.cp++ & 0377) {
                        case '\b':
                            printf("\\b");
                            break;
                        case '\f':
                            printf("\\f");
                            break;
                        case '\n': /* generate linebreaks after new-lines */
                            printf("\\n\",\n    \"");
                            break;
                        case '\r':
                            printf("\\r");
                            break;
                        case '\t':
                            printf("\\t");
                            break;
                        case '\v':
                            printf("\\v");
                            break;
                        case '\\':
                            printf("\\\\");
                            break;
                        case '\'':
                            printf("\\\'");
                            break;
                        case '\"':
                            printf("\\\"");
                            break;
                        default:
                            if (isprint(uc))
                                printf("%c", uc);
                            else
                                printf("\\%.3o", uc);
                            break;
                    }
                printf("\"");
                annotate(vp, fsp, cor, 0);
            }
            else { /* use format from C_format attribute */
                for (iel = 0; iel < len - 1; iel++) {
                    printf(fmt, *gp.cp++);
                    printf(", ");
                    annotate(vp, fsp, cor, iel);
                }
                printf(fmt, *gp.cp++);
                lastdelim(more, lastrow);
                annotate(vp, fsp, cor, iel);
            }
            break;
        case NC_SHORT:
            gp.sp = (short *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                printf(fmt, *gp.sp++);
                printf(", ");
                annotate(vp, fsp, cor, iel);
            }
            printf(fmt, *gp.sp++);
            lastdelim(more, lastrow);
            annotate(vp, fsp, cor, iel);
            break;
        case NC_LONG:
            gp.lp = (nclong *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                printf(fmt, *gp.lp++);
                printf(", ");
                annotate(vp, fsp, cor, iel);
            }
            printf(fmt, *gp.lp++);
            lastdelim(more, lastrow);
            annotate(vp, fsp, cor, iel);
            break;
        case NC_FLOAT:
            gp.fp = (float *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                if (*gp.fp >= fill_float)
                    printf("FloatInf");
                else
                    printf(fmt, (double)*gp.fp);
                printf(",");
                annotate(vp, fsp, cor, iel);
                gp.fp++;
            }
            if (*gp.fp >= fill_float)
                printf("FloatInf");
            else
                printf(fmt, (double)*gp.fp);
            lastdelim(more, lastrow);
            annotate(vp, fsp, cor, iel);
            gp.fp++;
            break;
        case NC_DOUBLE:
            gp.dp = (double *)vals;
            for (iel = 0; iel < len - 1; iel++) {
                if (*gp.dp >= fill_double)
                    printf("DoubleInf");
                else {
                    printf(fmt, *gp.dp);
                }
                printf(",");
                annotate(vp, fsp, cor, iel);
                gp.dp++;
            }
            if (*gp.dp >= fill_double)
                printf("DoubleInf");
            else
                printf(fmt, *gp.dp);
            lastdelim(more, lastrow);
            annotate(vp, fsp, cor, iel);
            gp.dp++;
            break;
        default:
            error("pr_vals: bad type");
    }
}

/*
 * Updates a vector of ints, odometer style.  Returns 0 if odometer
 * overflowed, else 1.
 */
/* dims  - The "odometer" limits for each dimension  */
/* ndims - Number of dimensions */
/* odom  - The "odometer" vector to be updated */
/* add   - A vector to "add" to odom on each update */
static int
upcorner(long *dims, int ndims, long *odom, long *add)
{
    int id;
    int ret = 1;

    for (id = ndims - 1; id > 0; id--) {
        odom[id] += add[id];
        if (odom[id] >= dims[id]) {
            odom[id - 1]++;
            odom[id] -= dims[id];
        }
    }
    odom[0] += add[0];
    if (odom[0] >= dims[0])
        ret = 0;
    return ret;
}

/* vp    - variable */
/* vdims - variable dimension sizes */
/* ncid  - netcdf id */
/* varid - variable id */
/* fsp   - formatting specs */
int
vardata(struct ncvar *vp, long vdims[], int ncid, int varid, struct fspec *fsp)
{
    long cor[H4_MAX_VAR_DIMS]; /* corner coordinates */
    long edg[H4_MAX_VAR_DIMS]; /* edges of hypercube */
    long add[H4_MAX_VAR_DIMS]; /* "odometer" increment to next "row"  */
#define VALBUFSIZ 8192
    double vals[VALBUFSIZ / sizeof(double)]; /* aligned buffer */
    int    gulp = VALBUFSIZ / nctypelen(vp->type);

    int   id;
    int   ir;
    long  nels;
    long  ncols;
    long  nrows;
    int   vrank = vp->ndims;
    char *fixed_var;
    int   ret = 0, err_code = 0;

    /* printf format used to print each value */
    const char *fmt = get_fmt(ncid, varid, vp->type);

    nels = 1;
    for (id = 0; id < vrank; id++) {
        cor[id] = 0;
        edg[id] = 1;
        nels *= vdims[id]; /* total number of values for variable */
    }

    fixed_var = sanitize_string(vp->name, fsp->fix_str);

    if (vrank <= 1) {
        printf("\n %s = ", fixed_var);
        set_indent(strlen(fixed_var) + 4);
    }
    else {
        printf("\n %s =\n  ", fixed_var);
        set_indent(2);
    }

    if (vrank < 1) {
        ncols = 1;
    }
    else {
        ncols          = vdims[vrank - 1]; /* size of "row" along last dimension */
        edg[vrank - 1] = vdims[vrank - 1];
        for (id = 0; id < vrank; id++)
            add[id] = 0;
        if (vrank > 1)
            add[vrank - 2] = 1;
    }
    nrows = nels / ncols; /* number of "rows" */

    for (ir = 0; ir < nrows; ir++) {
        /*
         * rather than just printing a whole row at once (which might exceed
         * the capacity of MSDOS platforms, for example), we break each row
         * into smaller chunks, if necessary.
         */
        long corsav = 0;
        long left   = ncols;
        bool lastrow;

        if (vrank > 0) {
            corsav = cor[vrank - 1];
            if (fsp->brief_data_cmnts != false && vrank > 1 &&
                left > 0) { /* print brief comment with indices range */
                printf("// %s(", fixed_var);
                switch (fsp->data_lang) {
                    case LANG_C:
                        /* print brief comment with C variable indices */
                        for (id = 0; id < vrank - 1; id++)
                            printf("%d,", (int)cor[id]);
                        if (vdims[vrank - 1] == 1)
                            printf("0");
                        else
                            printf(" 0-%d", (int)vdims[vrank - 1] - 1);
                        break;
                    case LANG_F:
                        /* print brief comment with Fortran variable indices */
                        if (vdims[vrank - 1] == 1)
                            printf("1");
                        else
                            printf("1-%d ", (int)vdims[vrank - 1]);
                        for (id = vrank - 2; id >= 0; id--) {
                            printf(",%d", (int)(1 + cor[id]));
                        }
                        break;
                    case LANG_NONE:
                    default:
                        /* Might want to make this an error */
                        break;
                }
                printf(")\n    ");
                set_indent(4);
            }
        }
        lastrow = (ir == nrows - 1) ? true : false;
        while (left > 0) {
            long toget = left < gulp ? left : gulp;
            if (vrank > 0)
                edg[vrank - 1] = toget;

            /* ncvarget was casted to (void), thus ncdump misinformed users
               that the reading succeeded even though the data was corrupted and
               reading in fact failed (HDFFR-1468.)  Now when ncvarget fails,
               break out of the while loop and set error code to indicate this
               failure. -BMR, 2015/01/19 */
            ret = ncvarget(ncid, varid, cor, edg, (void *)vals);
            if (ret == -1) {
                err_code = ERR_READFAIL; /* to be returned to caller to
                                            indicate that ncvarget fails */
                break;
            }

            if (fsp->full_data_cmnts)
                pr_cvals(vp, toget, fmt, left > toget, lastrow, (void *)vals, fsp, cor);
            else
                pr_vals(vp, toget, fmt, left > toget, lastrow, (void *)vals);
            left -= toget;
            if (vrank > 0)
                cor[vrank - 1] += toget;
        }
        /* No failure occurs */
        if (ret >= 0) {
            if (vrank > 0)
                cor[vrank - 1] = corsav;
            if (ir < nrows - 1)
                if (!upcorner(vdims, vp->ndims, cor, add))
                    error("vardata: odometer overflowed!");
            set_indent(2);
        }
    }

    free(fixed_var);
    return (err_code);
    /* Previously, it was "return 0;"  If this function is revised, this
       return statement may be changed appropriately. (HDFFR-1468) */
}