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. */
draw_background_line(int line,int start,int npixels)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 */
draw_background(void)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  */
tile_image(int width,int height)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 
read_png_file(FILE * fp)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 
jpeg_sig_cmp(uint8_t * bytes,int len)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 
read_jpeg_file(FILE * fp,uint8_t * header,int len)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 */
vesacon_default_background(void)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 */
vesacon_set_background(unsigned int rgb)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 
lss16_sig_cmp(const void * header,int len)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 
read_lss16_file(FILE * fp,const void * header,int header_len)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 
vesacon_load_background(const char * filename)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 
__vesacon_init_background(void)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