color_rule.c

Go to the documentation of this file.

#include <grass/gis.h>

#define LIMIT(x) if (x < 0) x = 0; else if (x > 255) x = 255;

static int add_color_rule(const void *, int, int, int,
                          const void *, int, int, int,
                          struct _Color_Info_ *, int,
                          DCELL *, DCELL *, RASTER_MAP_TYPE);


int
G_add_d_raster_color_rule(const DCELL * val1, int r1, int g1, int b1,
                          const DCELL * val2, int r2, int g2, int b2,
                          struct Colors *colors)
{
    add_color_rule(val1, r1, g1, b1, val2, r2, g2, b2, &colors->fixed,
                   colors->version, &colors->cmin, &colors->cmax, DCELL_TYPE);
    return 1;
}


int
G_add_f_raster_color_rule(const FCELL * cat1, int r1, int g1, int b1,
                          const FCELL * cat2, int r2, int g2, int b2,
                          struct Colors *colors)
{
    add_color_rule(cat1, r1, g1, b1, cat2, r2, g2, b2, &colors->fixed,
                   colors->version, &colors->cmin, &colors->cmax, FCELL_TYPE);
    return 1;
}


int
G_add_c_raster_color_rule(const CELL * cat1, int r1, int g1, int b1,
                          const CELL * cat2, int r2, int g2, int b2,
                          struct Colors *colors)
{
    add_color_rule(cat1, r1, g1, b1, cat2, r2, g2, b2, &colors->fixed,
                   colors->version, &colors->cmin, &colors->cmax, CELL_TYPE);
    return 1;
}


int
G_add_raster_color_rule(const void *val1, int r1, int g1, int b1,
                        const void *val2, int r2, int g2, int b2,
                        struct Colors *colors, RASTER_MAP_TYPE data_type)
{
    add_color_rule(val1, r1, g1, b1, val2, r2, g2, b2, &colors->fixed,
                   colors->version, &colors->cmin, &colors->cmax, data_type);
    return 1;
}


int
G_add_color_rule(CELL cat1, int r1, int g1, int b1, CELL cat2, int r2, int g2,
                 int b2, struct Colors *colors)
{
    add_color_rule((void *)&cat1, r1, g1, b1, (void *)&cat2, r2, g2, b2,
                   &colors->fixed, colors->version, &colors->cmin,
                   &colors->cmax, CELL_TYPE);
    return 1;
}

int
G_add_modular_d_raster_color_rule(const DCELL * val1, int r1, int g1, int b1,
                                  const DCELL * val2, int r2, int g2, int b2,
                                  struct Colors *colors)
{
    DCELL min, max;

    if (colors->version < 0)
        return -1;              /* can't use this on 3.0 colors */
    min = colors->cmin;
    max = colors->cmax;
    add_color_rule(val1, r1, g1, b1, val2, r2, g2, b2, &colors->modular, 0,
                   &colors->cmin, &colors->cmax, DCELL_TYPE);
    colors->cmin = min;         /* don't reset these */
    colors->cmax = max;

    return 1;
}

int
G_add_modular_f_raster_color_rule(const FCELL * val1, int r1, int g1, int b1,
                                  const FCELL * val2, int r2, int g2, int b2,
                                  struct Colors *colors)
{
    DCELL min, max;

    if (colors->version < 0)
        return -1;              /* can;t use this on 3.0 colors */
    min = colors->cmin;
    max = colors->cmax;
    add_color_rule(val1, r1, g1, b1, val2, r2, g2, b2, &colors->modular, 0,
                   &colors->cmin, &colors->cmax, FCELL_TYPE);
    colors->cmin = min;         /* don't reset these */
    colors->cmax = max;

    return 1;
}

int
G_add_modular_c_raster_color_rule(const CELL * val1, int r1, int g1, int b1,
                                  const CELL * val2, int r2, int g2, int b2,
                                  struct Colors *colors)
{
    return G_add_modular_color_rule(*val1, r1, g1, b1, *val2, r2, g2, b2,
                                    colors);
}

int
G_add_modular_raster_color_rule(const void *val1, int r1, int g1, int b1,
                                const void *val2, int r2, int g2, int b2,
                                struct Colors *colors,
                                RASTER_MAP_TYPE data_type)
{
    CELL min, max;

    if (colors->version < 0)
        return -1;              /* can't use this on 3.0 colors */
    min = colors->cmin;
    max = colors->cmax;
    add_color_rule(val1, r1, g1, b1, val2, r2, g2, b2, &colors->modular, 0,
                   &colors->cmin, &colors->cmax, data_type);
    colors->cmin = min;         /* don't reset these */
    colors->cmax = max;

    return 1;
}

int
G_add_modular_color_rule(CELL cat1, int r1, int g1, int b1, CELL cat2, int r2,
                         int g2, int b2, struct Colors *colors)
{
    CELL min, max;

    if (colors->version < 0)
        return -1;              /* can;t use this on 3.0 colors */
    min = colors->cmin;
    max = colors->cmax;
    add_color_rule((void *)&cat1, r1, g1, b1, (void *)&cat2, r2, g2, b2,
                   &colors->modular, 0, &colors->cmin, &colors->cmax,
                   CELL_TYPE);
    colors->cmin = min;         /* don't reset these */
    colors->cmax = max;

    return 1;
}

static int add_color_rule(const void *pt1, int r1, int g1, int b1,
                          const void *pt2, int r2, int g2, int b2,
                          struct _Color_Info_ *cp, int version, DCELL * cmin,
                          DCELL * cmax, RASTER_MAP_TYPE data_type)
{
    struct _Color_Rule_ *rule, *next;
    unsigned char red, grn, blu;
    DCELL min, max, val1, val2;
    CELL cat;

    val1 = G_get_raster_value_d(pt1, data_type);
    val2 = G_get_raster_value_d(pt2, data_type);
    /* allocate a low:high rule */
    rule = (struct _Color_Rule_ *)G_malloc(sizeof(*rule));
    rule->next = rule->prev = NULL;

    /* make sure colors are in the range [0,255] */
    LIMIT(r1);
    LIMIT(g1);
    LIMIT(b1);
    LIMIT(r2);
    LIMIT(g2);
    LIMIT(b2);

    /* val1==val2, use average color */
    /* otherwise make sure low < high */
    if (val1 == val2) {
        rule->low.value = rule->high.value = val1;
        rule->low.red = rule->high.red = (r1 + r2) / 2;
        rule->low.grn = rule->high.grn = (g1 + g2) / 2;
        rule->low.blu = rule->high.blu = (b1 + b2) / 2;
    }
    else if (val1 < val2) {
        rule->low.value = val1;
        rule->low.red = r1;
        rule->low.grn = g1;
        rule->low.blu = b1;

        rule->high.value = val2;
        rule->high.red = r2;
        rule->high.grn = g2;
        rule->high.blu = b2;
    }
    else {
        rule->low.value = val2;
        rule->low.red = r2;
        rule->low.grn = g2;
        rule->low.blu = b2;

        rule->high.value = val1;
        rule->high.red = r1;
        rule->high.grn = g1;
        rule->high.blu = b1;
    }

    /* keep track of the overall min and max, excluding null */
    if (G_is_d_null_value(&(rule->low.value)))
        return 0;
    if (G_is_d_null_value(&(rule->high.value)))
        return 0;
    min = rule->low.value;
    max = rule->high.value;
    if (min <= max) {
        if (cp->min > cp->max) {
            cp->min = min;
            cp->max = max;
        }
        else {
            if (cp->min > min)
                cp->min = min;
            if (cp->max < max)
                cp->max = max;
        }
    }
    if (*cmin > *cmax) {
        *cmin = cp->min;
        *cmax = cp->max;
    }
    else {
        if (*cmin > cp->min)
            *cmin = cp->min;
        if (*cmax < cp->max)
            *cmax = cp->max;
    }

    /* If version is old style (i.e., pre 4.0),
     *     interpolate this rule from min to max
     *     and insert each cat into the lookup table.
     *     Then free the rule.
     * Otherwise, free the lookup table, if active.
     *     G_organize_colors() will regenerate it
     *     Link this rule into the list of rules
     */

    if (version < 0) {
        for (cat = (CELL) min; cat <= (CELL) max; cat++) {
            G__interpolate_color_rule((DCELL) cat, &red, &grn, &blu, rule);
            G__insert_color_into_lookup(cat, (int)red, (int)grn, (int)blu,
                                        cp);
        }
        G_free(rule);
    }
    else {
        if (cp->rules)
            cp->rules->prev = rule;
        rule->next = cp->rules;
        cp->rules = rule;

        /* prune the rules:
         * remove all rules that are contained by this rule 
         */
        min = rule->low.value;  /* mod 4.1 */
        max = rule->high.value; /* mod 4.1 */
        cp->n_rules++;
        for (rule = rule->next; rule; rule = next) {
            next = rule->next;  /* has to be done here, not in for stmt */
            if (min <= rule->low.value && max >= rule->high.value) {
                if ((rule->prev->next = next))  /* remove from the list */
                    next->prev = rule->prev;
                G_free(rule);
                cp->n_rules--;
            }
        }

        /* free lookup array, if allocated */
        G__color_free_lookup(cp);
        G__color_free_fp_lookup(cp);
    }

    return 0;
}