Home | History | Annotate | Download | only in vesa
      1 /* ----------------------------------------------------------------------- *
      2  *
      3  *   Copyright 2006-2008 H. Peter Anvin - All Rights Reserved
      4  *
      5  *   Permission is hereby granted, free of charge, to any person
      6  *   obtaining a copy of this software and associated documentation
      7  *   files (the "Software"), to deal in the Software without
      8  *   restriction, including without limitation the rights to use,
      9  *   copy, modify, merge, publish, distribute, sublicense, and/or
     10  *   sell copies of the Software, and to permit persons to whom
     11  *   the Software is furnished to do so, subject to the following
     12  *   conditions:
     13  *
     14  *   The above copyright notice and this permission notice shall
     15  *   be included in all copies or substantial portions of the Software.
     16  *
     17  *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
     18  *   EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
     19  *   OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
     20  *   NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
     21  *   HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
     22  *   WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
     23  *   FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
     24  *   OTHER DEALINGS IN THE SOFTWARE.
     25  *
     26  * ----------------------------------------------------------------------- */
     27 
     28 #include <stdio.h>
     29 #include <png.h>
     30 #include <tinyjpeg.h>
     31 #include <com32.h>
     32 #include <unistd.h>
     33 #include <stdlib.h>
     34 #include <sys/stat.h>
     35 #include <minmax.h>
     36 #include <stdbool.h>
     37 #include <ilog2.h>
     38 #include <syslinux/loadfile.h>
     39 #include "vesa.h"
     40 #include "video.h"
     41 
     42 /*** FIX: This really should be alpha-blended with color index 0 ***/
     43 
     44 /* For best performance, "start" should be a multiple of 4, to assure
     45    aligned dwords. */
     46 static void draw_background_line(int line, int start, int npixels)
     47 {
     48     uint32_t *bgptr = &__vesacon_background[line*__vesa_info.mi.h_res+start];
     49     unsigned int bytes_per_pixel = __vesacon_bytes_per_pixel;
     50     size_t fbptr = line * __vesa_info.mi.logical_scan + start*bytes_per_pixel;
     51 
     52     __vesacon_copy_to_screen(fbptr, bgptr, npixels);
     53 }
     54 
     55 /* This draws the border, then redraws the text area */
     56 static void draw_background(void)
     57 {
     58     int i;
     59     const int bottom_border = VIDEO_BORDER +
     60 	(TEXT_PIXEL_ROWS % __vesacon_font_height);
     61     const int right_border = VIDEO_BORDER + (TEXT_PIXEL_COLS % FONT_WIDTH);
     62 
     63     for (i = 0; i < VIDEO_BORDER; i++)
     64 	draw_background_line(i, 0, __vesa_info.mi.h_res);
     65 
     66     for (i = VIDEO_BORDER; i < __vesa_info.mi.v_res - bottom_border; i++) {
     67 	draw_background_line(i, 0, VIDEO_BORDER);
     68 	draw_background_line(i, __vesa_info.mi.h_res - right_border,
     69 			     right_border);
     70     }
     71 
     72     for (i = __vesa_info.mi.v_res - bottom_border;
     73 	 i < __vesa_info.mi.v_res; i++)
     74 	draw_background_line(i, 0, __vesa_info.mi.h_res);
     75 
     76     __vesacon_redraw_text();
     77 }
     78 
     79 /*
     80  * Tile an image in the UL corner across the entire screen
     81  */
     82 static void tile_image(int width, int height)
     83 {
     84     int xsize = __vesa_info.mi.h_res;
     85     int ysize = __vesa_info.mi.v_res;
     86     int x, y, yr;
     87     int xl, yl;
     88     uint32_t *sp, *dp, *drp, *dtp;
     89 
     90     drp = __vesacon_background;
     91     for (y = 0 ; y < ysize ; y += height) {
     92 	yl = min(height, ysize-y);
     93 	dtp = drp;
     94 	for (x = 0 ; x < xsize ; x += width) {
     95 	    xl = min(width, xsize-x);
     96 	    if (x || y) {
     97 		sp = __vesacon_background;
     98 		dp = dtp;
     99 		for (yr = 0 ; yr < yl ; yr++) {
    100 		    memcpy(dp, sp, xl*sizeof(uint32_t));
    101 		    dp += xsize;
    102 		    sp += xsize;
    103 		}
    104 	    }
    105 	    dtp += xl;
    106 	}
    107 	drp += xsize*height;
    108     }
    109 }
    110 
    111 static int read_png_file(FILE * fp)
    112 {
    113     png_structp png_ptr = NULL;
    114     png_infop info_ptr = NULL;
    115 #if 0
    116     png_color_16p image_background;
    117     static const png_color_16 my_background = { 0, 0, 0, 0, 0 };
    118 #endif
    119     png_bytep row_pointers[__vesa_info.mi.v_res], rp;
    120     int i;
    121     int rv = -1;
    122 
    123     png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
    124     info_ptr = png_create_info_struct(png_ptr);
    125 
    126     if (!png_ptr || !info_ptr || setjmp(png_jmpbuf(png_ptr)))
    127 	goto err;
    128 
    129     png_init_io(png_ptr, fp);
    130     png_set_sig_bytes(png_ptr, 8);
    131 
    132     png_set_user_limits(png_ptr, __vesa_info.mi.h_res, __vesa_info.mi.v_res);
    133 
    134     png_read_info(png_ptr, info_ptr);
    135 
    136     /* Set the appropriate set of transformations.  We need to end up
    137        with 32-bit BGRA format, no more, no less. */
    138 
    139     /* Expand to RGB first... */
    140     if (info_ptr->color_type & PNG_COLOR_MASK_PALETTE)
    141 	png_set_palette_to_rgb(png_ptr);
    142     else if (!(info_ptr->color_type & PNG_COLOR_MASK_COLOR))
    143 	png_set_gray_to_rgb(png_ptr);
    144 
    145     /* Add alpha channel, if need be */
    146     if (!(png_ptr->color_type & PNG_COLOR_MASK_ALPHA)) {
    147 	if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))
    148 	    png_set_tRNS_to_alpha(png_ptr);
    149 	else
    150 	    png_set_add_alpha(png_ptr, ~0, PNG_FILLER_AFTER);
    151     }
    152 
    153     /* Adjust the byte order, if necessary */
    154     png_set_bgr(png_ptr);
    155 
    156     /* Make sure we end up with 8-bit data */
    157     if (info_ptr->bit_depth == 16)
    158 	png_set_strip_16(png_ptr);
    159     else if (info_ptr->bit_depth < 8)
    160 	png_set_packing(png_ptr);
    161 
    162 #if 0
    163     if (png_get_bKGD(png_ptr, info_ptr, &image_background))
    164 	png_set_background(png_ptr, image_background,
    165 			   PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
    166     else
    167 	png_set_background(png_ptr, &my_background,
    168 			   PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);
    169 #endif
    170 
    171     /* Whew!  Now we should get the stuff we want... */
    172     rp = (png_bytep)__vesacon_background;
    173     for (i = 0; i < (int)info_ptr->height; i++) {
    174 	row_pointers[i] = rp;
    175 	rp += __vesa_info.mi.h_res << 2;
    176     }
    177 
    178     png_read_image(png_ptr, row_pointers);
    179 
    180     tile_image(info_ptr->width, info_ptr->height);
    181 
    182     rv = 0;
    183 
    184 err:
    185     if (png_ptr)
    186 	png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp) NULL);
    187     return rv;
    188 }
    189 
    190 static int jpeg_sig_cmp(uint8_t * bytes, int len)
    191 {
    192     (void)len;
    193 
    194     return (bytes[0] == 0xff && bytes[1] == 0xd8) ? 0 : -1;
    195 }
    196 
    197 static int read_jpeg_file(FILE * fp, uint8_t * header, int len)
    198 {
    199     struct jdec_private *jdec = NULL;
    200     void *jpeg_file = NULL;
    201     size_t length_of_file;
    202     unsigned int width, height;
    203     int rv = -1;
    204     unsigned char *components[1];
    205     unsigned int bytes_per_row[1];
    206 
    207     rv = floadfile(fp, &jpeg_file, &length_of_file, header, len);
    208     if (rv)
    209 	goto err;
    210 
    211     jdec = tinyjpeg_init();
    212     if (!jdec)
    213 	goto err;
    214 
    215     if (tinyjpeg_parse_header(jdec, jpeg_file, length_of_file) < 0)
    216 	goto err;
    217 
    218     tinyjpeg_get_size(jdec, &width, &height);
    219     if (width > __vesa_info.mi.h_res || height > __vesa_info.mi.v_res)
    220 	goto err;
    221 
    222     components[0] = (void *)__vesacon_background;
    223     tinyjpeg_set_components(jdec, components, 1);
    224     bytes_per_row[0] = __vesa_info.mi.h_res << 2;
    225     tinyjpeg_set_bytes_per_row(jdec, bytes_per_row, 1);
    226 
    227     tinyjpeg_decode(jdec, TINYJPEG_FMT_BGRA32);
    228     tile_image(width, height);
    229 
    230     rv = 0;
    231 
    232 err:
    233     /* Don't use tinyjpeg_free() here, since we didn't allow tinyjpeg
    234        to allocate the frame buffer */
    235     if (jdec)
    236 	free(jdec);
    237 
    238     if (jpeg_file)
    239 	free(jpeg_file);
    240 
    241     return rv;
    242 }
    243 
    244 /* Simple grey Gaussian hole, enough to look interesting */
    245 int vesacon_default_background(void)
    246 {
    247     int x, y, dx, dy, dy2;
    248     int z;
    249     unsigned int shft;
    250     uint8_t *bgptr = (uint8_t *)__vesacon_background;
    251     uint8_t k;
    252 
    253     if (__vesacon_pixel_format == PXF_NONE)
    254 	return 0;		/* Not in graphics mode */
    255 
    256     z = max(__vesa_info.mi.v_res, __vesa_info.mi.h_res) >> 1;
    257     z = ((z*z) >> 11) - 1;
    258     shft = ilog2(z) + 1;
    259 
    260     for (y = 0, dy = -(__vesa_info.mi.v_res >> 1);
    261 	 y < __vesa_info.mi.v_res; y++, dy++) {
    262 	dy2 = dy * dy;
    263 	for (x = 0, dx = -(__vesa_info.mi.h_res >> 1);
    264 	     x < __vesa_info.mi.h_res; x++, dx++) {
    265 	    k = __vesacon_linear_to_srgb[500 + ((dx*dx + dy2) >> shft)];
    266 	    bgptr[0] = k;	/* Blue */
    267 	    bgptr[1] = k;	/* Green */
    268 	    bgptr[2] = k;	/* Red */
    269 	    bgptr += 4;		/* Dummy alpha */
    270 	}
    271     }
    272 
    273     draw_background();
    274     return 0;
    275 }
    276 
    277 /* Set the background to a single flat color */
    278 int vesacon_set_background(unsigned int rgb)
    279 {
    280     void *bgptr = __vesacon_background;
    281     unsigned int count = __vesa_info.mi.h_res * __vesa_info.mi.v_res;
    282 
    283     if (__vesacon_pixel_format == PXF_NONE)
    284 	return 0;		/* Not in graphics mode */
    285 
    286     asm volatile ("rep; stosl":"+D" (bgptr), "+c"(count)
    287 		  :"a"(rgb)
    288 		  :"memory");
    289 
    290     draw_background();
    291     return 0;
    292 }
    293 
    294 struct lss16_header {
    295     uint32_t magic;
    296     uint16_t xsize;
    297     uint16_t ysize;
    298 };
    299 
    300 #define LSS16_MAGIC 0x1413f33d
    301 
    302 static inline int lss16_sig_cmp(const void *header, int len)
    303 {
    304     const struct lss16_header *h = header;
    305 
    306     if (len != 8)
    307 	return 1;
    308 
    309     return !(h->magic == LSS16_MAGIC &&
    310 	     h->xsize <= __vesa_info.mi.h_res &&
    311 	     h->ysize <= __vesa_info.mi.v_res);
    312 }
    313 
    314 static int read_lss16_file(FILE * fp, const void *header, int header_len)
    315 {
    316     const struct lss16_header *h = header;
    317     uint32_t colors[16], color;
    318     bool has_nybble;
    319     uint8_t byte;
    320     int count;
    321     int nybble, prev;
    322     enum state {
    323 	st_start,
    324 	st_c0,
    325 	st_c1,
    326 	st_c2,
    327     } state;
    328     int i, x, y;
    329     uint32_t *bgptr = __vesacon_background;
    330 
    331     /* Assume the header, 8 bytes, has already been loaded. */
    332     if (header_len != 8)
    333 	return -1;
    334 
    335     for (i = 0; i < 16; i++) {
    336 	uint8_t rgb[3];
    337 	if (fread(rgb, 1, 3, fp) != 3)
    338 	    return -1;
    339 
    340 	colors[i] = (((rgb[0] & 63) * 255 / 63) << 16) +
    341 	    (((rgb[1] & 63) * 255 / 63) << 8) +
    342 	    ((rgb[2] & 63) * 255 / 63);
    343     }
    344 
    345     /* By spec, the state machine is per row */
    346     for (y = 0; y < h->ysize; y++) {
    347 	state = st_start;
    348 	has_nybble = false;
    349 	color = colors[prev = 0];	/* By specification */
    350 	count = 0;
    351 
    352 	x = 0;
    353 	while (x < h->xsize) {
    354 	    if (!has_nybble) {
    355 		if (fread(&byte, 1, 1, fp) != 1)
    356 		    return -1;
    357 		nybble = byte & 0xf;
    358 		has_nybble = true;
    359 	    } else {
    360 		nybble = byte >> 4;
    361 		has_nybble = false;
    362 	    }
    363 
    364 	    switch (state) {
    365 	    case st_start:
    366 		if (nybble != prev) {
    367 		    *bgptr++ = color = colors[prev = nybble];
    368 		    x++;
    369 		} else {
    370 		    state = st_c0;
    371 		}
    372 		break;
    373 
    374 	    case st_c0:
    375 		if (nybble == 0) {
    376 		    state = st_c1;
    377 		} else {
    378 		    count = nybble;
    379 		    goto do_run;
    380 		}
    381 		break;
    382 
    383 	    case st_c1:
    384 		count = nybble + 16;
    385 		state = st_c2;
    386 		break;
    387 
    388 	    case st_c2:
    389 		count += nybble << 4;
    390 		goto do_run;
    391 
    392 do_run:
    393 		count = min(count, h->xsize - x);
    394 		x += count;
    395 		asm volatile ("rep; stosl":"+D" (bgptr),
    396 			      "+c"(count):"a"(color));
    397 		state = st_start;
    398 		break;
    399 	    }
    400 	}
    401 
    402 	/* Zero-fill rest of row */
    403 	i = __vesa_info.mi.h_res - x;
    404 	asm volatile ("rep; stosl":"+D" (bgptr), "+c"(i):"a"(0):"memory");
    405     }
    406 
    407     /* Zero-fill rest of screen */
    408     i = (__vesa_info.mi.v_res - y) * __vesa_info.mi.h_res;
    409     asm volatile ("rep; stosl":"+D" (bgptr), "+c"(i):"a"(0):"memory");
    410 
    411     return 0;
    412 }
    413 
    414 int vesacon_load_background(const char *filename)
    415 {
    416     FILE *fp = NULL;
    417     uint8_t header[8];
    418     int rv = 1;
    419 
    420     if (__vesacon_pixel_format == PXF_NONE)
    421 	return 0;		/* Not in graphics mode */
    422 
    423     fp = fopen(filename, "r");
    424 
    425     if (!fp)
    426 	goto err;
    427 
    428     if (fread(header, 1, 8, fp) != 8)
    429 	goto err;
    430 
    431     if (!png_sig_cmp(header, 0, 8)) {
    432 	rv = read_png_file(fp);
    433     } else if (!jpeg_sig_cmp(header, 8)) {
    434 	rv = read_jpeg_file(fp, header, 8);
    435     } else if (!lss16_sig_cmp(header, 8)) {
    436 	rv = read_lss16_file(fp, header, 8);
    437     }
    438 
    439     /* This actually displays the stuff */
    440     draw_background();
    441 
    442 err:
    443     if (fp)
    444 	fclose(fp);
    445 
    446     return rv;
    447 }
    448 
    449 int __vesacon_init_background(void)
    450 {
    451     /* __vesacon_background was cleared by calloc() */
    452 
    453     /* The VESA BIOS has already cleared the screen */
    454     return 0;
    455 }
    456