#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <errno.h>
#include "pnglite.h"

#include <cammy/sram.h>

static uint32_t bayer_matrix[256] = {
    /*
     * Black gamut
     */
    0x00000000, 0x00000000, 0x00000000, 0x00000000, 0x00000000,

    /*
     * Black to dark gray gradient
     */
    0x00018000, 0x00018000, 0x00018000, 0x00018000, 0x00018000,
    0x00018020, 0x00018020, 0x00018020, 0x00018020, 0x00018020,
    0x000180a0, 0x000180a0, 0x000180a0, 0x000180a0, 0x000180a0,
    0x0001a0a0, 0x0001a0a0, 0x0001a0a0, 0x0001a0a0, 0x0001a0a0,
    0x0001a4a0, 0x0001a4a0, 0x0001a4a0, 0x0001a4a0, 0x0001a4a0, 0x0001a4a0,
    0x0001a4a1, 0x0001a4a1, 0x0001a4a1, 0x0001a4a1, 0x0001a4a1,
    0x0001a4a5, 0x0001a4a5, 0x0001a4a5, 0x0001a4a5, 0x0001a4a5,
    0x0001a5a5, 0x0001a5a5, 0x0001a5a5, 0x0001a5a5, 0x0001a5a5,
    0x0001ada5, 0x0001ada5, 0x0001ada5, 0x0001ada5, 0x0001ada5, 0x0001ada5,
    0x0001ada7, 0x0001ada7, 0x0001ada7, 0x0001ada7, 0x0001ada7,
    0x0001adaf, 0x0001adaf, 0x0001adaf, 0x0001adaf, 0x0001adaf,
    0x0001afaf, 0x0001afaf, 0x0001afaf, 0x0001afaf, 0x0001afaf,
    0x0001df5f, 0x0001df5f, 0x0001df5f, 0x0001df5f, 0x0001df5f, 0x0001df5f,
    0x0001df7f, 0x0001df7f, 0x0001df7f, 0x0001df7f, 0x0001df7f,
    0x0001dfff, 0x0001dfff, 0x0001dfff, 0x0001dfff, 0x0001dfff,
    0x0001ffff, 0x0001ffff, 0x0001ffff, 0x0001ffff, 0x0001ffff,

    /*
     * Dark gray to light gray gradient
     */
    0x01028000, 0x01028000, 0x01028000, 0x01028000, 0x01028000, 0x01028000,
    0x01028020, 0x01028020, 0x01028020, 0x01028020, 0x01028020,
    0x010280a0, 0x010280a0, 0x010280a0, 0x010280a0, 0x010280a0,
    0x0102a0a0, 0x0102a0a0, 0x0102a0a0, 0x0102a0a0, 0x0102a0a0,
    0x0102a4a0, 0x0102a4a0, 0x0102a4a0, 0x0102a4a0, 0x0102a4a0, 0x0102a4a0,
    0x0102a4a1, 0x0102a4a1, 0x0102a4a1, 0x0102a4a1, 0x0102a4a1,
    0x0102a4a5, 0x0102a4a5, 0x0102a4a5, 0x0102a4a5, 0x0102a4a5,
    0x0102a5a5, 0x0102a5a5, 0x0102a5a5, 0x0102a5a5, 0x0102a5a5,
    0x0102ada5, 0x0102ada5, 0x0102ada5, 0x0102ada5, 0x0102ada5, 0x0102ada5,
    0x0102ada7, 0x0102ada7, 0x0102ada7, 0x0102ada7, 0x0102ada7,
    0x0102adaf, 0x0102adaf, 0x0102adaf, 0x0102adaf, 0x0102adaf,
    0x0102afaf, 0x0102afaf, 0x0102afaf, 0x0102afaf, 0x0102afaf,
    0x0102df5f, 0x0102df5f, 0x0102df5f, 0x0102df5f, 0x0102df5f, 0x0102df5f,
    0x0102df7f, 0x0102df7f, 0x0102df7f, 0x0102df7f, 0x0102df7f,
    0x0102dfff, 0x0102dfff, 0x0102dfff, 0x0102dfff, 0x0102dfff,
    0x0102ffff, 0x0102ffff, 0x0102ffff, 0x0102ffff, 0x0102ffff,

    /* Light gray to white gradient */
    0x02038000, 0x02038000, 0x02038000, 0x02038000, 0x02038000, 0x02038000,
    0x02038020, 0x02038020, 0x02038020, 0x02038020, 0x02038020,
    0x020380a0, 0x020380a0, 0x020380a0, 0x020380a0, 0x020380a0,
    0x0203a0a0, 0x0203a0a0, 0x0203a0a0, 0x0203a0a0, 0x0203a0a0,
    0x0203a4a0, 0x0203a4a0, 0x0203a4a0, 0x0203a4a0, 0x0203a4a0, 0x0203a4a0,
    0x0203a4a1, 0x0203a4a1, 0x0203a4a1, 0x0203a4a1, 0x0203a4a1,
    0x0203a4a5, 0x0203a4a5, 0x0203a4a5, 0x0203a4a5, 0x0203a4a5,
    0x0203a5a5, 0x0203a5a5, 0x0203a5a5, 0x0203a5a5, 0x0203a5a5,
    0x0203ada5, 0x0203ada5, 0x0203ada5, 0x0203ada5, 0x0203ada5, 0x0203ada5,
    0x0203ada7, 0x0203ada7, 0x0203ada7, 0x0203ada7, 0x0203ada7,
    0x0203adaf, 0x0203adaf, 0x0203adaf, 0x0203adaf, 0x0203adaf,
    0x0203afaf, 0x0203afaf, 0x0203afaf, 0x0203afaf, 0x0203afaf,
    0x0203df5f, 0x0203df5f, 0x0203df5f, 0x0203df5f, 0x0203df5f, 0x0203df5f,
    0x0203df7f, 0x0203df7f, 0x0203df7f, 0x0203df7f, 0x0203df7f,
    0x0203dfff, 0x0203dfff, 0x0203dfff, 0x0203dfff, 0x0203dfff,
    0x0203ffff, 0x0203ffff, 0x0203ffff, 0x0203ffff, 0x0203ffff,
};

static void usage(int argc, char **argv, const char *message, ...) {
    if (message) {
        va_list args;

        va_start(args, message);
        vfprintf(stderr, message, args);
        va_end(args);
         fprintf(stderr, "\n");
    }

    fprintf(stderr, "usage: %s file.sav frame1..30 photo.png\n", argv[0]);

    exit(1);
}

static inline uint8_t *buf_malloc(png_t *png) {
    return malloc(4 * png->width * png->height);
}

static void image_copy(cammy_sram_frame *frame, uint8_t *buf) {
    size_t x, y;

    memset(&frame->thumb, '\xff', sizeof(cammy_sram_thumb));
    memset(&frame->photo, '\x00', sizeof(cammy_sram_photo));

    for (y=0; y<CAMMY_SRAM_PHOTO_HEIGHT; y++) {
        for (x=0; x<CAMMY_SRAM_PHOTO_WIDTH; x++) {
            cammy_sram_tile *tile = &frame->photo.tiles[y>>3][x>>3];

            int tile_x = x & 7,
                tile_y = y & 7;

            uint8_t r = buf[4*CAMMY_SRAM_PHOTO_WIDTH*y+4*x],
                    g = buf[4*CAMMY_SRAM_PHOTO_WIDTH*y+4*x+1],
                    b = buf[4*CAMMY_SRAM_PHOTO_WIDTH*y+4*x+2];

            uint8_t gray = (uint8_t)
                ((0.2126 * (float)r)
               + (0.7152 * (float)g)
               + (0.0722 * (float)b));

            uint32_t slot = bayer_matrix[gray];
            uint32_t from = (slot & 0x03000000) >> 24;
            uint32_t to   = (slot & 0x00030000) >> 16;
            uint8_t  value;

            if (slot & (0x8000 >> ((y & 3) << 2) >> (x & 3))) {
                value = to   ^ 0x03;
            } else {
                value = from ^ 0x03;
            }

            tile->data[2*tile_y]   |=  (value & 0x01)       << (tile_x ^ 7);
            tile->data[2*tile_y+1] |= ((value & 0x02) >> 1) << (tile_x ^ 7);
        }
    }
}

int main(int argc, char **argv) {
    cammy_sram *sram;
    png_t *png;
    int frame = 0;
    uint8_t *buf;

    int error;

    if (argc < 2) {
        usage(argc, argv, "No save file provided");
    } else if (argc < 3) {
        usage(argc, argv, "No frame number provided");
    } else if (argc < 4) {
        usage(argc, argv, "No photo provided");
    }

    frame = atoi(argv[2]);

    if (frame < 1 || frame > CAMMY_SRAM_FRAME_COUNT) {
        usage(argc, argv, "Invalid frame number");
    }

    if ((sram = cammy_sram_open(argv[1])) == NULL) {
        fprintf(stderr, "%s: %s: %s: %s\n",
            argv[0], "cammy_sram_open()", argv[1], strerror(errno));

        goto error_sram_open;
    }

    png_init(malloc, free);

    if ((png = malloc(sizeof(*png))) == NULL) {
        goto error_malloc_png;
    }

    if ((error = png_open_file_read(png, argv[3])) < 0) {
        fprintf(stderr, "%s: %s: %s: %s\n",
            argv[0], "png_open_file_read()", argv[3], png_error_string(error));

        goto error_png_open_file_read;
    }

    if (png->width  != CAMMY_SRAM_PHOTO_WIDTH
     || png->height != CAMMY_SRAM_PHOTO_HEIGHT) {
        fprintf(stderr, "%s: %s: %s\n",
            argv[0], argv[3], "Invalid image dimensions");

        goto error_invalid_dimensions;
    }

    if ((buf = buf_malloc(png)) == NULL) {
        fprintf(stderr, "%s: %s: %s\n",
            argv[0], "malloc()", strerror(errno));

        goto error_buf_malloc;
    }

    if ((error = png_get_data(png, buf)) < 0) {
        fprintf(stderr, "%s: %s: %s: %s\n",
            argv[0], "png_get_data()", argv[3], png_error_string(error));

        goto error_png_get_data;
    }

    image_copy(&sram->data->frames[frame-1], buf);

    free(buf);

    png_close_file(png);

    free(png);

    cammy_sram_close(sram);

    return 0;

error_png_get_data:
    free(buf);

error_buf_malloc:
error_invalid_dimensions:
    png_close_file(png);

error_png_open_file_read:
    free(png);

error_malloc_png:
    cammy_sram_close(sram);

error_sram_open:
    return 1;
}