/*
 * Copyright 1997-1999 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Sun designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Sun in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

#include "awt.h"
#include "awt_image.h"

extern "C" {
#include "img_colors.h"
} // extern "C"

char *programname = "awt_makecube";

unsigned char cube[LOOKUPSIZE * LOOKUPSIZE * LOOKUPSIZE];

unsigned char reds[256], greens[256], blues[256], indices[256];
int num_colors;

PALETTEENTRY sysPal[256];

int sys2cmap[256];
int cmap2sys[256];
int error[256];

int cmapsize = 0;
int virtcubesize = 0;
int makecube_verbose = 0;

void printPalette(char *label, HPALETTE hPal);

void usage(char *errmsg)
{
    fprintf(stderr, "%s\n", errmsg);
    fprintf(stderr, "usage: %s [-cmapsize N] [-cubesize N]\n", programname);
    fprintf(stderr, "\t-cmapsize N   set the number of colors to allocate\n");
    fprintf(stderr, "\t              in the colormap (2 <= N <= 256)\n");
    fprintf(stderr, "\t-cubesize N   set the size of the cube of colors to\n");
    fprintf(stderr, "                scan as potential entries in the cmap\n");
    fprintf(stderr, "                (N must be a power of 2 and <= 32)\n");
    exit(1);
}

void setsyscolor(int index, int red, int green, int blue)
{
    if (index >= 0) {
        if (sysPal[index].peFlags != 0) {
            usage("Internal error: system palette conflict");
        }
    } else {
        for (int i = 0; i < 256; i++) {
            if (sysPal[i].peFlags != 0) {
                if (sysPal[i].peRed   == red &&
                    sysPal[i].peGreen == green &&
                    sysPal[i].peBlue  == blue)
                {
                    // Already there.  Ignore it.
                    return;
                }
            } else if (index < 0) {
                index = i;
            }
        }
        if (index < 0) {
            usage("Internal error: ran out of system palette entries");
        }
    }
    sysPal[index].peRed   = red;
    sysPal[index].peGreen = green;
    sysPal[index].peBlue  = blue;
    sysPal[index].peFlags = 1;
}

void addcmapcolor(int red, int green, int blue)
{
    for (int i = 0; i < num_colors; i++) {
        if (red == reds[i] && green == greens[i] && blue == blues[i]) {
            return;
        }
    }
    if (num_colors >= cmapsize) {
        usage("Internal error: more than cmapsize static colors defined");
    }
    reds[num_colors]   = red;
    greens[num_colors] = green;
    blues[num_colors]  = blue;
    num_colors++;
}

int main(int argc, char **argv)
{
    int i;

    programname = argv[0];

    for (i = 1; i < argc; i++) {
        if (strcmp(argv[i], "-cmapsize") == 0) {
            if (i++ >= argc) {
                usage("no argument to -cmapsize");
            }
            cmapsize = atoi(argv[i]);
            if (cmapsize <= 2 || cmapsize > 256) {
                usage("colormap size must be between 2 and 256");
            }
        } else if (strcmp(argv[1], "-cubesize") == 0) {
            if (i++ >= argc) {
                usage("no argument to -cubesize");
            }
            virtcubesize = atoi(argv[i]);
            if (virtcubesize == 0 ||
                (virtcubesize & (virtcubesize - 1)) != 0 ||
                virtcubesize > 32)
            {
                usage("cube size must by a power of 2 <= 32");
            }
        } else if (strcmp(argv[i], "-verbose") == 0) {
            makecube_verbose = 1;
        } else {
            usage("unknown argument");
        }
    }

    if (cmapsize == 0) {
        cmapsize = CMAPSIZE;
    }
    if (virtcubesize == 0) {
        virtcubesize = VIRTCUBESIZE;
    }

    if (0) {  // For testing
        HDC hDC = CreateDC("DISPLAY", NULL, NULL, NULL);
        HPALETTE hPal = CreateHalftonePalette(hDC);
        printPalette("Halftone palette for current display", hPal);
        printPalette("Stock DEFAULT_PALETTE", (HPALETTE)GetStockObject(DEFAULT_PALETTE));
        BITMAPINFOHEADER bmInfo = {
            sizeof(BITMAPINFOHEADER), 1, 1, 1, 8, BI_RGB, 0, 1000, 1000, 0, 0
            };
        HBITMAP hBitmap = CreateDIBitmap(hDC, &bmInfo,
                                         0, NULL, NULL, DIB_RGB_COLORS);
        HDC hMemDC = CreateCompatibleDC(hDC);
        SelectObject(hDC, hBitmap);
        hPal = CreateHalftonePalette(hMemDC);
        printPalette("Halftone palette for 8-bit DIBitmap", hPal);
        exit(0);
    }

    // Allocate Windows static system colors.
    {
        PALETTEENTRY palEntries[256];
        HPALETTE hPal = (HPALETTE)GetStockObject(DEFAULT_PALETTE);
        int n = GetPaletteEntries(hPal, 0, 256, palEntries);
        for (i = 0; i < n; i++) {
            addcmapcolor(palEntries[i].peRed,
                         palEntries[i].peGreen,
                         palEntries[i].peBlue);
            setsyscolor((i < n / 2) ? i : i + (256 - n),
                        palEntries[i].peRed,
                        palEntries[i].peGreen,
                        palEntries[i].peBlue);
        }
    }

    // Allocate java.awt.Color constant colors.
    addcmapcolor(  0,   0,   0);        // black
    addcmapcolor(255, 255, 255);        // white
    addcmapcolor(255,   0,   0);        // red
    addcmapcolor(  0, 255,   0);        // green
    addcmapcolor(  0,   0, 255);        // blue
    addcmapcolor(255, 255,   0);        // yellow
    addcmapcolor(255,   0, 255);        // magenta
    addcmapcolor(  0, 255, 255);        // cyan
    addcmapcolor(192, 192, 192);        // lightGray
    addcmapcolor(128, 128, 128);        // gray
    addcmapcolor( 64,  64,  64);        // darkGray
    addcmapcolor(255, 175, 175);        // pink
    addcmapcolor(255, 200,   0);        // orange

    img_makePalette(cmapsize, virtcubesize, LOOKUPSIZE,
                    50.0f, 250.0f,
                    num_colors, TRUE, reds, greens, blues, cube);

    if (makecube_verbose) {
        fprintf(stderr, "Calculated colormap:\n");
        for (i = 0; i < cmapsize; i++) {
            fprintf(stderr, "%3d:(%3d,%3d,%3d)   ",
                    i, reds[i], greens[i], blues[i]);
        }
        fprintf(stderr, "\n");
    }

    // Now simulate adding the halftone palette to the system
    // palette to get an idea of palette ordering.
    {
        int cubevals[6] = {0, 44, 86, 135, 192, 255};
        for (int b = 0; b < 6; b++) {
            for (int g = 0; g < 6; g++) {
                for (int r = 0; r < 6; r++) {
                    setsyscolor(-1, cubevals[r], cubevals[g], cubevals[b]);
                }
            }
        }
        int grayvals[26] = {  0,  17,  24,  30,  37,  44,  52,  60,
                             68,  77,  86,  95, 105, 114, 125, 135,
                            146, 157, 168, 180, 192, 204, 216, 229,
                            242, 255 };
        for (i = 0; i < 26; i++) {
            setsyscolor(-1, grayvals[i], grayvals[i], grayvals[i]);
        }
    }

    if (makecube_verbose) {
        fprintf(stderr, "System palette with simulated halftone palette:\n");
        for (i = 0; i < 256; i++) {
            fprintf(stderr, "%3d:(%3d,%3d,%3d)   ",
                    i, sysPal[i].peRed, sysPal[i].peGreen, sysPal[i].peBlue);
        }
    }

    if (makecube_verbose) {
        HDC hDC = CreateDC("DISPLAY", NULL, NULL, NULL);
        HPALETTE hPal = CreateHalftonePalette(hDC);
        SelectPalette(hDC, hPal, FALSE);
        RealizePalette(hDC);
        PALETTEENTRY palEntries[256];
        int n = GetSystemPaletteEntries(hDC, 0, 256, palEntries);
        fprintf(stderr,
                "realized halftone palette reads back %d entries\n", n);
        int broken = 0;
        for (i = 0; i < 256; i++) {
            char *msg1 = "";
            char *msg2 = "";
            if (palEntries[i].peRed != sysPal[i].peRed ||
                palEntries[i].peGreen != sysPal[i].peGreen ||
                palEntries[i].peBlue != sysPal[i].peBlue)
            {
                msg1 = "no sysPal match!";
                if (sysPal[i].peFlags == 0) {
                    msg2 = "(OK)";
                } else {
                    broken++;
                }
            } else if (sysPal[i].peFlags == 0) {
                msg1 = "no sysPal entry...";
            }
            fprintf(stderr,
                    "palEntries[%3d] = (%3d, %3d, %3d), flags = %d  %s %s\n",
                    i,
                    palEntries[i].peRed,
                    palEntries[i].peGreen,
                    palEntries[i].peBlue,
                    palEntries[i].peFlags, msg1, msg2);
        }
        fprintf(stderr, "%d broken entries\n", broken);
    }

#if 0
#define BIGERROR (255 * 255 * 255)

    for (i = 0; i < 256; i++) {
        sys2cmap[i] = -1;
        cmap2sys[i] = -1;
        error[i] = BIGERROR;
        // error[i] = -1 means cmap[i] is locked to cmap2sys[i]
        // error[i] >= 0 means cmap[i] may lock to cmap2sys[i] on this run
    }

    int nummapped;
    int totalmapped = 0;
    do {
        int maxerror = BIGERROR;
        for (i = 0; i < 256; i++) {
            if (sysPal[i].peFlags == 0 || sys2cmap[i] >= 0) {
                continue;
            }
            int red   = sysPal[i].peRed;
            int green = sysPal[i].peGreen;
            int blue  = sysPal[i].peBlue;
            int e = maxerror;
            int ix = -1;
            for (int j = 0; j < 256; j++) {
                if (error[j] < 0) {
                    continue;
                }
                int t = red - reds[j];
                int d = t * t;
                t = green - greens[j];
                d += t * t;
                t = blue - blues[j];
                d += t * t;
                if (d < e) {
                    e = d;
                    ix = j;
                }
            }
            if (ix >= 0) {
                if (e < error[ix]) {
                    if (cmap2sys[ix] >= 0) {
                        // To be fair we will not accept any matches
                        // looser than this former match that we just
                        // displaced with a better match.
                        if (maxerror > error[ix]) {
                            maxerror = error[ix];
                        }
                        sys2cmap[cmap2sys[ix]] = -1;
                    }
                    error[ix] = e;
                    sys2cmap[i] = ix;
                    cmap2sys[ix] = i;
                }
            }
        }
        nummapped = 0;
        for (i = 0; i < 256; i++) {
            if (error[i] >= 0) {
                if (error[i] >= maxerror) {
                    // Throw this one back to be fair to a displaced entry.
                    error[i] = BIGERROR;
                    sys2cmap[cmap2sys[i]] = -1;
                    cmap2sys[i] = -1;
                    continue;
                }
                error[i] = -1;
                nummapped++;
            }
        }
        totalmapped += nummapped;
        if (makecube_verbose) {
            fprintf(stderr, "%3d colors mapped (%3d total), maxerror = %d\n",
                    nummapped, totalmapped, maxerror);
        }
    } while (nummapped != 0);

    for (i = 0; i < 256; i++) {
        if (cmap2sys[i] < 0) {
            for (int j = 0; j < 256; j++) {
                if (sys2cmap[j] < 0) {
                    cmap2sys[i] = j;
                    sys2cmap[j] = i;
                    break;
                }
            }
            if (j == 256) {
                usage("Internal error: no unused system entry for cmap entry!\n");
            }
        }
    }
#else
    for (i = 0; i < 256; i++) {
        if (i < 10) {
            sys2cmap[i] = i;
            cmap2sys[i] = i;
        } else if (i < 20) {
            sys2cmap[256 - 20 + i] = i;
            cmap2sys[i] = 256 - 20 + i;
        } else {
            sys2cmap[i - 10] = i;
            cmap2sys[i] = i - 10;
        }
    }
#endif

    if (makecube_verbose) {
        fprintf(stderr, "cmap2sys mapping: \n");
        for (i = 0; i < 256; i++) {
            fprintf(stderr, "%4d", cmap2sys[i]);
            if (sys2cmap[cmap2sys[i]] != i) {
                usage("Internal error: bad system palette back pointer!\n");
            }
        }
        fprintf(stderr, "\n");
    }

    printf("unsigned char awt_reds[256] = {");
    for (i = 0; i < 256; i++) {
        if ((i & 0xf) == 0) printf("\n\t");
        printf("%3d,", reds[sys2cmap[i]]);
    }
    printf("\n};\n");
    printf("unsigned char awt_greens[256] = {");
    for (i = 0; i < 256; i++) {
        if ((i & 0xf) == 0) printf("\n\t");
        printf("%3d,", greens[sys2cmap[i]]);
    }
    printf("\n};\n");
    printf("unsigned char awt_blues[256] = {");
    for (i = 0; i < 256; i++) {
        if ((i & 0xf) == 0) printf("\n\t");
        printf("%3d,", blues[sys2cmap[i]]);
    }
    printf("\n};\n");
    fflush(stdout);
    return 0;
}

void printPalette(char *label, HPALETTE hPal)
{
    PALETTEENTRY palEntries[256];
    fprintf(stderr, "%s (0x%08x):\n", label, hPal);
    int n = GetPaletteEntries(hPal, 0, 256, palEntries);
    for (int i = 0; i < n; i++) {
        fprintf(stderr, "palEntries[%3d] = (%3d, %3d, %3d), flags = %d\n",
                i,
                palEntries[i].peRed,
                palEntries[i].peGreen,
                palEntries[i].peBlue,
                palEntries[i].peFlags);
    }
}

/* This helps eliminate any dependence on javai.dll at build time. */
int
jio_fprintf (FILE *handle, const char *format, ...)
{
    int len;

    va_list args;
    va_start(args, format);
    len = vfprintf(handle, format, args);
    va_end(args);

    return len;
}