pngdriver/Color_table.c

Go to the documentation of this file.

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

#include <grass/gis.h>
#include <grass/colors.h>
#include "pngdriver.h"

static int r_shift, g_shift, b_shift, a_shift;
static int Red[256], Grn[256], Blu[256];

static void set_color(int i, int red, int grn, int blu)
{
    png_palette[i][0] = red;
    png_palette[i][1] = grn;
    png_palette[i][2] = blu;
    png_palette[i][3] = 0;
}

static void init_colors_rgb(void)
{
    NCOLORS = 1 << 24;

    if (G_is_little_endian()) {
        b_shift = 0;
        g_shift = 8;
        r_shift = 16;
        a_shift = 24;
    }
    else {
        b_shift = 24;
        g_shift = 16;
        r_shift = 8;
        a_shift = 0;
    }
}

static void init_colors_indexed(void)
{
    int n_pixels;
    int r, g, b;
    int i;

    NCOLORS = 256;

    n_pixels = 0;

    if (has_alpha)
        /* transparent color should be the first!
         * Its RGB value doesn't matter since we fake RGB-to-index. */
        set_color(n_pixels++, 0, 0, 0);

    for (r = 0; r < 6; r++) {
        for (g = 0; g < 6; g++) {
            for (b = 0; b < 6; b++) {
                int red = r * 0xFF / 5;
                int grn = g * 0xFF / 5;
                int blu = b * 0xFF / 5;

                set_color(n_pixels++, red, grn, blu);
            }
        }
    }

    while (n_pixels < NCOLORS)
        set_color(n_pixels++, 0, 0, 0);

    for (i = 0; i < 256; i++) {
        int k = i * 6 / 256;

        Red[i] = k * 6 * 6;
        Grn[i] = k * 6;
        Blu[i] = k;
    }
}

void init_color_table(void)
{
    if (true_color)
        init_colors_rgb();
    else
        init_colors_indexed();
}

static int get_color_rgb(int r, int g, int b, int a)
{
    return (r << r_shift) + (g << g_shift) + (b << b_shift) + (a << a_shift);
}

static int get_color_indexed(int r, int g, int b, int a)
{
    if (has_alpha && a >= 128)
        return 0;

    return Red[r] + Grn[g] + Blu[b] + has_alpha;
}

static void get_pixel_rgb(unsigned int pixel, int *r, int *g, int *b, int *a)
{
    *r = (pixel >> r_shift) & 0xFF;
    *g = (pixel >> g_shift) & 0xFF;
    *b = (pixel >> b_shift) & 0xFF;
    *a = (pixel >> a_shift) & 0xFF;
}

static void get_pixel_indexed(unsigned int pixel, int *r, int *g, int *b,
                              int *a)
{
    *r = png_palette[pixel][0];
    *g = png_palette[pixel][1];
    *b = png_palette[pixel][2];
    *a = png_palette[pixel][3];
}


void get_pixel(unsigned int pixel, int *r, int *g, int *b, int *a)
{
    if (true_color)
        get_pixel_rgb(pixel, r, g, b, a);
    else
        get_pixel_indexed(pixel, r, g, b, a);
}

unsigned int get_color(int r, int g, int b, int a)
{
    return true_color ? get_color_rgb(r, g, b, a)
        : get_color_indexed(r, g, b, a);
}

int PNG_lookup_color(int r, int g, int b)
{
    return true_color ? ((r << 16) | (g << 8) | (b << 0))
        : Red[r] + Grn[g] + Blu[b] + has_alpha;
}