/* ----------------------------------------------------------------------- *
 *
 *   Copyright 2006-2008 H. Peter Anvin - All Rights Reserved
 *
 *   Permission is hereby granted, free of charge, to any person
 *   obtaining a copy of this software and associated documentation
 *   files (the "Software"), to deal in the Software without
 *   restriction, including without limitation the rights to use,
 *   copy, modify, merge, publish, distribute, sublicense, and/or
 *   sell copies of the Software, and to permit persons to whom
 *   the Software is furnished to do so, subject to the following
 *   conditions:
 *
 *   The above copyright notice and this permission notice shall
 *   be included in all copies or substantial portions of the Software.
 *
 *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 *   EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 *   OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 *   NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 *   HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 *   WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 *   FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 *   OTHER DEALINGS IN THE SOFTWARE.
 *
 * ----------------------------------------------------------------------- */

#include <stdio.h>
#include <png.h>
#include <tinyjpeg.h>
#include <com32.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <minmax.h>
#include <stdbool.h>
#include <ilog2.h>
#include <syslinux/loadfile.h>
#include <string.h>
#include "vesa.h"
#include "video.h"

/*** FIX: This really should be alpha-blended with color index 0 ***/

/* For best performance, "start" should be a multiple of 4, to assure
   aligned dwords. */
static void draw_background_line(int line, int start, int npixels)
{
    uint32_t *bgptr = &__vesacon_background[line*__vesa_info.mi.h_res+start];
    unsigned int bytes_per_pixel = __vesacon_bytes_per_pixel;
    size_t fbptr = line * __vesa_info.mi.logical_scan + start*bytes_per_pixel;

    __vesacon_copy_to_screen(fbptr, bgptr, npixels);
}

/* This draws the border, then redraws the text area */
static void draw_background(void)
{
    int i;
    const int bottom_border = VIDEO_BORDER +
	(TEXT_PIXEL_ROWS % __vesacon_font_height);
    const int right_border = VIDEO_BORDER + (TEXT_PIXEL_COLS % FONT_WIDTH);

    for (i = 0; i < VIDEO_BORDER; i++)
	draw_background_line(i, 0, __vesa_info.mi.h_res);

    for (i = VIDEO_BORDER; i < __vesa_info.mi.v_res - bottom_border; i++) {
	draw_background_line(i, 0, VIDEO_BORDER);
	draw_background_line(i, __vesa_info.mi.h_res - right_border,
			     right_border);
    }

    for (i = __vesa_info.mi.v_res - bottom_border;
	 i < __vesa_info.mi.v_res; i++)
	draw_background_line(i, 0, __vesa_info.mi.h_res);

    __vesacon_redraw_text();
}

/*
 * Tile an image in the UL corner across the entire screen
 */
static void tile_image(int width, int height)
{
    int xsize = __vesa_info.mi.h_res;
    int ysize = __vesa_info.mi.v_res;
    int x, y, yr;
    int xl, yl;
    uint32_t *sp, *dp, *drp, *dtp;

    drp = __vesacon_background;
    for (y = 0 ; y < ysize ; y += height) {
	yl = min(height, ysize-y);
	dtp = drp;
	for (x = 0 ; x < xsize ; x += width) {
	    xl = min(width, xsize-x);
	    if (x || y) {
		sp = __vesacon_background;
		dp = dtp;
		for (yr = 0 ; yr < yl ; yr++) {
		    memcpy(dp, sp, xl*sizeof(uint32_t));
		    dp += xsize;
		    sp += xsize;
		}
	    }
	    dtp += xl;
	}
	drp += xsize*height;
    }
}

static int read_png_file(FILE * fp)
{
    png_structp png_ptr = NULL;
    png_infop info_ptr = NULL;
    int color_type, bit_depth;
    png_uint_32 height, width;
#if 0
    png_color_16p image_background;
    static const png_color_16 my_background = { 0, 0, 0, 0, 0 };
#endif
    png_bytep row_pointers[__vesa_info.mi.v_res], rp;
    int i;
    int rv = -1;

    png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
    info_ptr = png_create_info_struct(png_ptr);

    if (!png_ptr || !info_ptr || setjmp(png_jmpbuf(png_ptr)))
	goto err;

    png_init_io(png_ptr, fp);
    png_set_sig_bytes(png_ptr, 8);

    png_set_user_limits(png_ptr, __vesa_info.mi.h_res, __vesa_info.mi.v_res);

    png_read_info(png_ptr, info_ptr);

    /* Set the appropriate set of transformations.  We need to end up
       with 32-bit BGRA format, no more, no less. */

    png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type,
		 NULL, NULL, NULL);
    
    /* Expand to RGB first... */
    if (color_type & PNG_COLOR_MASK_PALETTE)
	png_set_palette_to_rgb(png_ptr);
    else if (!(color_type & PNG_COLOR_MASK_COLOR))
	png_set_gray_to_rgb(png_ptr);

    /* Add alpha channel, if need be */
    if (!(color_type & PNG_COLOR_MASK_ALPHA)) {
	if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))
	    png_set_tRNS_to_alpha(png_ptr);
	else
	    png_set_add_alpha(png_ptr, ~0, PNG_FILLER_AFTER);
    }

    /* Adjust the byte order, if necessary */
    png_set_bgr(png_ptr);

    /* Make sure we end up with 8-bit data */
    bit_depth = png_get_bit_depth(png_ptr, info_ptr);
    if (bit_depth == 16)
	png_set_strip_16(png_ptr);
    else if (bit_depth < 8)
	png_set_packing(png_ptr);

#if 0
    if (png_get_bKGD(png_ptr, info_ptr, &image_background))
	png_set_background(png_ptr, image_background,
			   PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
    else
	png_set_background(png_ptr, &my_background,
			   PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);
#endif

    /* Whew!  Now we should get the stuff we want... */
    rp = (png_bytep)__vesacon_background;
    for (i = 0; i < (int)height; i++) {
	row_pointers[i] = rp;
	rp += __vesa_info.mi.h_res << 2;
    }

    png_read_image(png_ptr, row_pointers);

    tile_image(width, height);

    rv = 0;

err:
    if (png_ptr)
	png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp) NULL);
    return rv;
}

static int jpeg_sig_cmp(uint8_t * bytes, int len)
{
    (void)len;

    return (bytes[0] == 0xff && bytes[1] == 0xd8) ? 0 : -1;
}

static int read_jpeg_file(FILE * fp, uint8_t * header, int len)
{
    struct jdec_private *jdec = NULL;
    void *jpeg_file = NULL;
    size_t length_of_file;
    unsigned int width, height;
    int rv = -1;
    unsigned char *components[1];
    unsigned int bytes_per_row[1];

    rv = floadfile(fp, &jpeg_file, &length_of_file, header, len);
    if (rv)
	goto err;

    jdec = tinyjpeg_init();
    if (!jdec)
	goto err;

    if (tinyjpeg_parse_header(jdec, jpeg_file, length_of_file) < 0)
	goto err;

    tinyjpeg_get_size(jdec, &width, &height);
    if (width > __vesa_info.mi.h_res || height > __vesa_info.mi.v_res)
	goto err;

    components[0] = (void *)__vesacon_background;
    tinyjpeg_set_components(jdec, components, 1);
    bytes_per_row[0] = __vesa_info.mi.h_res << 2;
    tinyjpeg_set_bytes_per_row(jdec, bytes_per_row, 1);

    tinyjpeg_decode(jdec, TINYJPEG_FMT_BGRA32);
    tile_image(width, height);

    rv = 0;

err:
    /* Don't use tinyjpeg_free() here, since we didn't allow tinyjpeg
       to allocate the frame buffer */
    if (jdec)
	free(jdec);

    if (jpeg_file)
	free(jpeg_file);

    return rv;
}

/* Simple grey Gaussian hole, enough to look interesting */
int vesacon_default_background(void)
{
    int x, y, dx, dy, dy2;
    int z;
    unsigned int shft;
    uint8_t *bgptr = (uint8_t *)__vesacon_background;
    uint8_t k;

    if (__vesacon_pixel_format == PXF_NONE)
	return 0;		/* Not in graphics mode */

    z = max(__vesa_info.mi.v_res, __vesa_info.mi.h_res) >> 1;
    z = ((z*z) >> 11) - 1;
    shft = ilog2(z) + 1;

    for (y = 0, dy = -(__vesa_info.mi.v_res >> 1);
	 y < __vesa_info.mi.v_res; y++, dy++) {
	dy2 = dy * dy;
	for (x = 0, dx = -(__vesa_info.mi.h_res >> 1);
	     x < __vesa_info.mi.h_res; x++, dx++) {
	    k = __vesacon_linear_to_srgb[500 + ((dx*dx + dy2) >> shft)];
	    bgptr[0] = k;	/* Blue */
	    bgptr[1] = k;	/* Green */
	    bgptr[2] = k;	/* Red */
	    bgptr += 4;		/* Dummy alpha */
	}
    }

    draw_background();
    return 0;
}

/* Set the background to a single flat color */
int vesacon_set_background(unsigned int rgb)
{
    void *bgptr = __vesacon_background;
    unsigned int count = __vesa_info.mi.h_res * __vesa_info.mi.v_res;

    if (__vesacon_pixel_format == PXF_NONE)
	return 0;		/* Not in graphics mode */

    asm volatile ("rep; stosl":"+D" (bgptr), "+c"(count)
		  :"a"(rgb)
		  :"memory");

    draw_background();
    return 0;
}

struct lss16_header {
    uint32_t magic;
    uint16_t xsize;
    uint16_t ysize;
};

#define LSS16_MAGIC 0x1413f33d

static inline int lss16_sig_cmp(const void *header, int len)
{
    const struct lss16_header *h = header;

    if (len != 8)
	return 1;

    return !(h->magic == LSS16_MAGIC &&
	     h->xsize <= __vesa_info.mi.h_res &&
	     h->ysize <= __vesa_info.mi.v_res);
}

static int read_lss16_file(FILE * fp, const void *header, int header_len)
{
    const struct lss16_header *h = header;
    uint32_t colors[16], color;
    bool has_nybble;
    uint8_t byte;
    int count;
    int nybble, prev;
    enum state {
	st_start,
	st_c0,
	st_c1,
	st_c2,
    } state;
    int i, x, y;
    uint32_t *bgptr = __vesacon_background;

    /* Assume the header, 8 bytes, has already been loaded. */
    if (header_len != 8)
	return -1;

    for (i = 0; i < 16; i++) {
	uint8_t rgb[3];
	if (fread(rgb, 1, 3, fp) != 3)
	    return -1;

	colors[i] = (((rgb[0] & 63) * 255 / 63) << 16) +
	    (((rgb[1] & 63) * 255 / 63) << 8) +
	    ((rgb[2] & 63) * 255 / 63);
    }

    /* By spec, the state machine is per row */
    for (y = 0; y < h->ysize; y++) {
	state = st_start;
	has_nybble = false;
	color = colors[prev = 0];	/* By specification */
	count = 0;

	x = 0;
	while (x < h->xsize) {
	    if (!has_nybble) {
		if (fread(&byte, 1, 1, fp) != 1)
		    return -1;
		nybble = byte & 0xf;
		has_nybble = true;
	    } else {
		nybble = byte >> 4;
		has_nybble = false;
	    }

	    switch (state) {
	    case st_start:
		if (nybble != prev) {
		    *bgptr++ = color = colors[prev = nybble];
		    x++;
		} else {
		    state = st_c0;
		}
		break;

	    case st_c0:
		if (nybble == 0) {
		    state = st_c1;
		} else {
		    count = nybble;
		    goto do_run;
		}
		break;

	    case st_c1:
		count = nybble + 16;
		state = st_c2;
		break;

	    case st_c2:
		count += nybble << 4;
		goto do_run;

do_run:
		count = min(count, h->xsize - x);
		x += count;
		asm volatile ("rep; stosl":"+D" (bgptr),
			      "+c"(count):"a"(color));
		state = st_start;
		break;
	    }
	}

	/* Zero-fill rest of row */
	i = __vesa_info.mi.h_res - x;
	asm volatile ("rep; stosl":"+D" (bgptr), "+c"(i):"a"(0):"memory");
    }

    /* Zero-fill rest of screen */
    i = (__vesa_info.mi.v_res - y) * __vesa_info.mi.h_res;
    asm volatile ("rep; stosl":"+D" (bgptr), "+c"(i):"a"(0):"memory");

    return 0;
}

int vesacon_load_background(const char *filename)
{
    FILE *fp = NULL;
    uint8_t header[8];
    int rv = 1;

    if (__vesacon_pixel_format == PXF_NONE)
	return 0;		/* Not in graphics mode */

    fp = fopen(filename, "r");

    if (!fp)
	goto err;

    if (fread(header, 1, 8, fp) != 8)
	goto err;

    if (!png_sig_cmp(header, 0, 8)) {
	rv = read_png_file(fp);
    } else if (!jpeg_sig_cmp(header, 8)) {
	rv = read_jpeg_file(fp, header, 8);
    } else if (!lss16_sig_cmp(header, 8)) {
	rv = read_lss16_file(fp, header, 8);
    }

    /* This actually displays the stuff */
    draw_background();

err:
    if (fp)
	fclose(fp);

    return rv;
}

int __vesacon_init_background(void)
{
    /* __vesacon_background was cleared by calloc() */

    /* The VESA BIOS has already cleared the screen */
    return 0;
}