#include <stdio.h>
#include <string.h>
#include <grass/gis.h>
#include <grass/colors.h>
#include <grass/raster.h>
#include <grass/display.h>
#include <grass/symbol.h>
#include <grass/glocale.h>

#include "options.h"
#include "local_proto.h"

#define CHUNK 128

static int coors_allocated = 0;
static double *xarray;
static double *yarray;

static float xincr;
static float yincr;

static double rotation; /* degrees counter-clockwise from east */

static RGBA_Color last_color;

static double t, b, l, r;

static double cur_x, cur_y;

int set_graph_stuff(void)
{
    D_get_dst(&t, &b, &l, &r);

    if (mapunits) {
        xincr = (r - l) / 100.;
        if (xincr < 0.0)
            xincr = -xincr; /* mod: shapiro 13 jun 1991 */
        yincr = (b - t) / 100.;
        if (yincr < 0.0)
            yincr = -yincr; /* mod: shapiro 13 jun 1991 */
    }
    else
        xincr = yincr = 1;

    rotation = 0.0; /* init */

    return 0;
}

int set_text_size(void)
{
    if (hsize >= 0. && vsize >= 0. && hsize <= 100. && vsize <= 100.) {
        D_text_size(hsize * xincr, vsize * yincr);
        G_debug(3, "text size initialized to [%.1f,%.1f]", hsize * xincr,
                vsize * yincr);
    }
    return (0);
}

int do_draw(const char *str)
{
    float xper, yper;

    if (2 != sscanf(str, "%*s %f %f", &xper, &yper)) {
        G_warning(_("Problem parsing coordinates [%s]"), str);
        return (-1);
    }

    D_line_abs(cur_x, cur_y, xper, yper);
    cur_x = xper;
    cur_y = yper;

    return (0);
}

int do_move(const char *str)
{
    float xper, yper;

    if (2 != sscanf(str, "%*s %f %f", &xper, &yper)) {
        G_warning(_("Problem parsing coordinates [%s]"), str);
        return (-1);
    }

    D_pos_abs(xper, yper);
    cur_x = xper;
    cur_y = yper;

    return (0);
}

int do_color(const char *str)
{
    char in_color[64];
    int R, G, B, color = 0;

    if (1 != sscanf(str, "%*s %s", in_color)) {
        G_warning(_("Unable to read color"));
        return (-1);
    }

    /* Parse and select color */
    color = G_str_to_color(in_color, &R, &G, &B);
    if (color == 0) {
        G_warning(_("[%s]: No such color"), in_color);
        /* store for backup */
        last_color.a = RGBA_COLOR_NONE;
        return (-1);
    }
    if (color == 1) {
        D_RGB_color(R, G, B);
        /* store for backup */
        set_last_color(R, G, B, RGBA_COLOR_OPAQUE);
    }
    if (color == 2) { /* color == 'none' */
        R = D_translate_color(DEFAULT_BG_COLOR);
        D_use_color(R);
        /* store for backup */
        set_last_color(0, 0, 0, RGBA_COLOR_NONE);
    }
    return (0);
}

int do_linewidth(const char *str)
{
    double width;

    if (1 != sscanf(str, "%*s %lf", &width)) {
        G_warning(_("Problem parsing command [%s]"), str);
        return (-1);
    }

    D_line_width(width);
    G_debug(3, "line width set to %.1f", width);

    return (0);
}

int do_poly(char *buff, FILE *infile)
{
    int num;
    char origcmd[64];
    float xper, yper;
    int to_return;

    sscanf(buff, "%s", origcmd);

    num = 0;

    for (;;) {
        if ((to_return = G_getl2(buff, 128, infile)) != 1)
            break;

        if (2 != sscanf(buff, "%f %f", &xper, &yper)) {

            if ('#' == buff[0]) {
                G_debug(3, " skipping comment line [%s]", buff);
                continue;
            }

            G_debug(3, "coordinate pair not found. ending polygon. [%s]", buff);
            break;
        }

        check_alloc(num + 1);

        xarray[num] = xper;
        yarray[num] = yper;

        num++;
    }

    if (num) {
        /* this check is here so you can use the "polyline" command
           to make an unfilled polygon */
        if (!strcmp(origcmd, "polygon"))
            D_polygon_abs(xarray, yarray, num);
        else
            D_polyline_abs(xarray, yarray, num);
    }

    return (to_return);
}

int do_size(const char *str)
{
    float xper, yper;
    int ret;

    ret = sscanf(str, "%*s %f %f", &xper, &yper);

    if (ret != 2 && ret != 1) {
        G_warning(_("Problem parsing command [%s]"), str);
        return (-1);
    }

    /* if only one size is given assume same value in both axes */
    if (ret == 1)
        yper = xper;

    if (xper < 0. || yper < 0. || xper > 100. || yper > 100.)
        return (-1);

    D_text_size(xper * xincr, yper * yincr);
    G_debug(3, "text size set to [%.1f,%.1f]", xper * xincr, yper * yincr);

    return (0);
}

int do_rotate(const char *str)
{
    if (1 != sscanf(str, "%*s %lf", &rotation)) {
        G_warning(_("Problem parsing command [%s]"), str);
        return (-1);
    }

    D_text_rotation(rotation);
    G_debug(3, "rotation set to %.1f degrees", rotation);

    return (0);
}

int do_text(const char *str)
{
    const char *ptr = str;

    /* skip to beginning of actual text */
    for (; *ptr != ' '; ptr++)
        ;
    for (; *ptr == ' '; ptr++)
        ;
    D_text(ptr);

    return 0;
}

int check_alloc(int num)
{
    int to_alloc;

    if (num < coors_allocated)
        return 0;

    to_alloc = num + CHUNK;

    xarray = G_realloc(xarray, to_alloc * sizeof(double));
    yarray = G_realloc(yarray, to_alloc * sizeof(double));

    coors_allocated = to_alloc;

    return 0;
}

int do_icon(const char *str)
{
    double xper, yper;
    char type;
    double size;
    double ix, iy;

    if (4 != sscanf(str, "%*s %c %lf %lf %lf", &type, &size, &xper, &yper)) {
        G_warning(_("Problem parsing command [%s]"), str);
        return (-1);
    }

    ix = xper;
    iy = yper;
    size *= yincr;

    D_begin();

    switch (type & 0x7F) {
    case 'o':
        D_move_abs(ix - size, iy - size);
        D_cont_abs(ix - size, iy + size);
        D_cont_abs(ix + size, iy + size);
        D_cont_abs(ix + size, iy - size);
        D_cont_abs(ix - size, iy - size);
        break;
    case 'x':
        D_move_abs(ix - size, iy - size);
        D_cont_abs(ix + size, iy + size);
        D_move_abs(ix - size, iy + size);
        D_cont_abs(ix + size, iy - size);
        break;
    case '+':
    default:
        D_move_abs(ix, iy - size);
        D_cont_abs(ix, iy + size);
        D_move_abs(ix - size, iy);
        D_cont_abs(ix + size, iy);
        break;
    }

    D_end();
    D_stroke();

    return (0);
}

int do_symbol(const char *str)
{
    double xper, yper;
    double size;
    double ix, iy;
    char *symb_name;
    SYMBOL *Symb;
    char *line_color_str, *fill_color_str;
    RGBA_Color *line_color, *fill_color;
    int R, G, B, ret;

    line_color = G_malloc(sizeof(RGBA_Color));
    fill_color = G_malloc(sizeof(RGBA_Color));

    symb_name =
        G_malloc(strlen(str) + 1); /* well, it won't be any bigger than this */
    line_color_str = G_malloc(strlen(str) + 1);
    fill_color_str = G_malloc(strlen(str) + 1);

    G_debug(3, "do_symbol() [%s]", str);

    /* set default colors so colors are optional */
    strcpy(line_color_str, DEFAULT_FG_COLOR);
    strcpy(fill_color_str, "grey");

    if (sscanf(str, "%*s %s %lf %lf %lf %s %s", symb_name, &size, &xper, &yper,
               line_color_str, fill_color_str) < 4) {
        G_warning(_("Problem parsing command [%s]"), str);
        return (-1);
    }

    ix = xper;
    iy = yper;
    size *= yincr;

    /* parse line color */
    ret = G_str_to_color(line_color_str, &R, &G, &B);
    line_color->r = (unsigned char)R;
    line_color->g = (unsigned char)G;
    line_color->b = (unsigned char)B;

    if (ret == 1) {
        /* here alpha is only used as an on/off switch, otherwise unused by the
         * display drivers */
        line_color->a = RGBA_COLOR_OPAQUE;
    }
    else if (ret == 2)
        line_color->a = RGBA_COLOR_NONE;
    else {
        G_warning(_("[%s]: No such color"), line_color_str);
        return (-1);
    }

    /* parse fill color */
    ret = G_str_to_color(fill_color_str, &R, &G, &B);
    fill_color->r = (unsigned char)R;
    fill_color->g = (unsigned char)G;
    fill_color->b = (unsigned char)B;

    if (ret == 1)
        fill_color->a = RGBA_COLOR_OPAQUE;
    else if (ret == 2)
        fill_color->a = RGBA_COLOR_NONE;
    else {
        G_warning(_("[%s]: No such color"), fill_color_str);
        return (-1);
    }

    Symb = S_read(symb_name);

    if (Symb == NULL) {
        G_warning(_("Cannot read symbol, cannot display points"));
        return (-1);
    }
    else
        S_stroke(Symb, size, rotation, 0);

    D_symbol(Symb, ix, iy, line_color, fill_color);

    /* restore previous d.graph draw color */
    if (last_color.a == RGBA_COLOR_OPAQUE)
        D_RGB_color(last_color.r, last_color.g, last_color.b);
    else if (last_color.a == RGBA_COLOR_NONE)
        D_use_color(D_parse_color(DEFAULT_BG_COLOR, 0));
    else /* unset or bad */
        D_RGB_color(line_color->r, line_color->g, line_color->b);

    G_free(symb_name);
    G_free(line_color_str);
    G_free(fill_color_str);
    G_free(line_color);
    G_free(fill_color);

    return (0);
}

/* RGBA are 0-255; alpha is only used as an on/off switch. maybe test a<127<a ?
 */
void set_last_color(int R, int G, int B, int alpha)
{
    if (alpha == RGBA_COLOR_OPAQUE) {
        last_color.r = (unsigned char)R;
        last_color.g = (unsigned char)G;
        last_color.b = (unsigned char)B;
        last_color.a = RGBA_COLOR_OPAQUE;
    }
    else if (alpha == RGBA_COLOR_NONE) {
        last_color.a = RGBA_COLOR_NONE;
    }
    else
        last_color.a = RGBA_COLOR_NONE;
}