1 
2 /*
3  *  Convert a logo in ASCII PNM format to C source suitable for inclusion in
4  *  the Linux kernel
5  *
6  *  (C) Copyright 2001-2003 by Geert Uytterhoeven <geert@linux-m68k.org>
7  *
8  *  --------------------------------------------------------------------------
9  *
10  *  This file is subject to the terms and conditions of the GNU General Public
11  *  License. See the file COPYING in the main directory of the Linux
12  *  distribution for more details.
13  */
14 
15 #include <ctype.h>
16 #include <errno.h>
17 #include <stdarg.h>
18 #include <stdio.h>
19 #include <stdlib.h>
20 #include <string.h>
21 #include <unistd.h>
22 
23 
24 static const char *programname;
25 static const char *filename;
26 static const char *logoname = "linux_logo";
27 static const char *outputname;
28 static FILE *out;
29 
30 
31 #define LINUX_LOGO_MONO		1	/* monochrome black/white */
32 #define LINUX_LOGO_VGA16	2	/* 16 colors VGA text palette */
33 #define LINUX_LOGO_CLUT224	3	/* 224 colors */
34 #define LINUX_LOGO_GRAY256	4	/* 256 levels grayscale */
35 
36 static const char *logo_types[LINUX_LOGO_GRAY256+1] = {
37     [LINUX_LOGO_MONO] = "LINUX_LOGO_MONO",
38     [LINUX_LOGO_VGA16] = "LINUX_LOGO_VGA16",
39     [LINUX_LOGO_CLUT224] = "LINUX_LOGO_CLUT224",
40     [LINUX_LOGO_GRAY256] = "LINUX_LOGO_GRAY256"
41 };
42 
43 #define MAX_LINUX_LOGO_COLORS	224
44 
45 struct color {
46     unsigned char red;
47     unsigned char green;
48     unsigned char blue;
49 };
50 
51 static const struct color clut_vga16[16] = {
52     { 0x00, 0x00, 0x00 },
53     { 0x00, 0x00, 0xaa },
54     { 0x00, 0xaa, 0x00 },
55     { 0x00, 0xaa, 0xaa },
56     { 0xaa, 0x00, 0x00 },
57     { 0xaa, 0x00, 0xaa },
58     { 0xaa, 0x55, 0x00 },
59     { 0xaa, 0xaa, 0xaa },
60     { 0x55, 0x55, 0x55 },
61     { 0x55, 0x55, 0xff },
62     { 0x55, 0xff, 0x55 },
63     { 0x55, 0xff, 0xff },
64     { 0xff, 0x55, 0x55 },
65     { 0xff, 0x55, 0xff },
66     { 0xff, 0xff, 0x55 },
67     { 0xff, 0xff, 0xff },
68 };
69 
70 
71 static int logo_type = LINUX_LOGO_CLUT224;
72 static unsigned int logo_width;
73 static unsigned int logo_height;
74 static struct color **logo_data;
75 static struct color logo_clut[MAX_LINUX_LOGO_COLORS];
76 static unsigned int logo_clutsize;
77 static int is_plain_pbm = 0;
78 
79 static void die(const char *fmt, ...)
80     __attribute__ ((noreturn)) __attribute ((format (printf, 1, 2)));
81 static void usage(void) __attribute ((noreturn));
82 
83 
get_number(FILE * fp)84 static unsigned int get_number(FILE *fp)
85 {
86     int c, val;
87 
88     /* Skip leading whitespace */
89     do {
90 	c = fgetc(fp);
91 	if (c == EOF)
92 	    die("%s: end of file\n", filename);
93 	if (c == '#') {
94 	    /* Ignore comments 'till end of line */
95 	    do {
96 		c = fgetc(fp);
97 		if (c == EOF)
98 		    die("%s: end of file\n", filename);
99 	    } while (c != '\n');
100 	}
101     } while (isspace(c));
102 
103     /* Parse decimal number */
104     val = 0;
105     while (isdigit(c)) {
106 	val = 10*val+c-'0';
107 	/* some PBM are 'broken'; GiMP for example exports a PBM without space
108 	 * between the digits. This is Ok cause we know a PBM can only have a '1'
109 	 * or a '0' for the digit. */
110 	if (is_plain_pbm)
111 		break;
112 	c = fgetc(fp);
113 	if (c == EOF)
114 	    die("%s: end of file\n", filename);
115     }
116     return val;
117 }
118 
get_number255(FILE * fp,unsigned int maxval)119 static unsigned int get_number255(FILE *fp, unsigned int maxval)
120 {
121     unsigned int val = get_number(fp);
122     return (255*val+maxval/2)/maxval;
123 }
124 
read_image(void)125 static void read_image(void)
126 {
127     FILE *fp;
128     unsigned int i, j;
129     int magic;
130     unsigned int maxval;
131 
132     /* open image file */
133     fp = fopen(filename, "r");
134     if (!fp)
135 	die("Cannot open file %s: %s\n", filename, strerror(errno));
136 
137     /* check file type and read file header */
138     magic = fgetc(fp);
139     if (magic != 'P')
140 	die("%s is not a PNM file\n", filename);
141     magic = fgetc(fp);
142     switch (magic) {
143 	case '1':
144 	case '2':
145 	case '3':
146 	    /* Plain PBM/PGM/PPM */
147 	    break;
148 
149 	case '4':
150 	case '5':
151 	case '6':
152 	    /* Binary PBM/PGM/PPM */
153 	    die("%s: Binary PNM is not supported\n"
154 		"Use pnmnoraw(1) to convert it to ASCII PNM\n", filename);
155 
156 	default:
157 	    die("%s is not a PNM file\n", filename);
158     }
159     logo_width = get_number(fp);
160     logo_height = get_number(fp);
161 
162     /* allocate image data */
163     logo_data = (struct color **)malloc(logo_height*sizeof(struct color *));
164     if (!logo_data)
165 	die("%s\n", strerror(errno));
166     for (i = 0; i < logo_height; i++) {
167 	logo_data[i] = malloc(logo_width*sizeof(struct color));
168 	if (!logo_data[i])
169 	    die("%s\n", strerror(errno));
170     }
171 
172     /* read image data */
173     switch (magic) {
174 	case '1':
175 	    /* Plain PBM */
176 	    is_plain_pbm = 1;
177 	    for (i = 0; i < logo_height; i++)
178 		for (j = 0; j < logo_width; j++)
179 		    logo_data[i][j].red = logo_data[i][j].green =
180 			logo_data[i][j].blue = 255*(1-get_number(fp));
181 	    break;
182 
183 	case '2':
184 	    /* Plain PGM */
185 	    maxval = get_number(fp);
186 	    for (i = 0; i < logo_height; i++)
187 		for (j = 0; j < logo_width; j++)
188 		    logo_data[i][j].red = logo_data[i][j].green =
189 			logo_data[i][j].blue = get_number255(fp, maxval);
190 	    break;
191 
192 	case '3':
193 	    /* Plain PPM */
194 	    maxval = get_number(fp);
195 	    for (i = 0; i < logo_height; i++)
196 		for (j = 0; j < logo_width; j++) {
197 		    logo_data[i][j].red = get_number255(fp, maxval);
198 		    logo_data[i][j].green = get_number255(fp, maxval);
199 		    logo_data[i][j].blue = get_number255(fp, maxval);
200 		}
201 	    break;
202     }
203 
204     /* close file */
205     fclose(fp);
206 }
207 
is_black(struct color c)208 static inline int is_black(struct color c)
209 {
210     return c.red == 0 && c.green == 0 && c.blue == 0;
211 }
212 
is_white(struct color c)213 static inline int is_white(struct color c)
214 {
215     return c.red == 255 && c.green == 255 && c.blue == 255;
216 }
217 
is_gray(struct color c)218 static inline int is_gray(struct color c)
219 {
220     return c.red == c.green && c.red == c.blue;
221 }
222 
is_equal(struct color c1,struct color c2)223 static inline int is_equal(struct color c1, struct color c2)
224 {
225     return c1.red == c2.red && c1.green == c2.green && c1.blue == c2.blue;
226 }
227 
write_header(void)228 static void write_header(void)
229 {
230     /* open logo file */
231     if (outputname) {
232 	out = fopen(outputname, "w");
233 	if (!out)
234 	    die("Cannot create file %s: %s\n", outputname, strerror(errno));
235     } else {
236 	out = stdout;
237     }
238 
239     fputs("/*\n", out);
240     fputs(" *  DO NOT EDIT THIS FILE!\n", out);
241     fputs(" *\n", out);
242     fprintf(out, " *  It was automatically generated from %s\n", filename);
243     fputs(" *\n", out);
244     fprintf(out, " *  Linux logo %s\n", logoname);
245     fputs(" */\n\n", out);
246     fputs("#include <linux/linux_logo.h>\n\n", out);
247     fprintf(out, "static unsigned char %s_data[] __initdata = {\n",
248 	    logoname);
249 }
250 
write_footer(void)251 static void write_footer(void)
252 {
253     fputs("\n};\n\n", out);
254     fprintf(out, "const struct linux_logo %s __initconst = {\n", logoname);
255     fprintf(out, "\t.type\t\t= %s,\n", logo_types[logo_type]);
256     fprintf(out, "\t.width\t\t= %d,\n", logo_width);
257     fprintf(out, "\t.height\t\t= %d,\n", logo_height);
258     if (logo_type == LINUX_LOGO_CLUT224) {
259 	fprintf(out, "\t.clutsize\t= %d,\n", logo_clutsize);
260 	fprintf(out, "\t.clut\t\t= %s_clut,\n", logoname);
261     }
262     fprintf(out, "\t.data\t\t= %s_data\n", logoname);
263     fputs("};\n\n", out);
264 
265     /* close logo file */
266     if (outputname)
267 	fclose(out);
268 }
269 
270 static int write_hex_cnt;
271 
write_hex(unsigned char byte)272 static void write_hex(unsigned char byte)
273 {
274     if (write_hex_cnt % 12)
275 	fprintf(out, ", 0x%02x", byte);
276     else if (write_hex_cnt)
277 	fprintf(out, ",\n\t0x%02x", byte);
278     else
279 	fprintf(out, "\t0x%02x", byte);
280     write_hex_cnt++;
281 }
282 
write_logo_mono(void)283 static void write_logo_mono(void)
284 {
285     unsigned int i, j;
286     unsigned char val, bit;
287 
288     /* validate image */
289     for (i = 0; i < logo_height; i++)
290 	for (j = 0; j < logo_width; j++)
291 	    if (!is_black(logo_data[i][j]) && !is_white(logo_data[i][j]))
292 		die("Image must be monochrome\n");
293 
294     /* write file header */
295     write_header();
296 
297     /* write logo data */
298     for (i = 0; i < logo_height; i++) {
299 	for (j = 0; j < logo_width;) {
300 	    for (val = 0, bit = 0x80; bit && j < logo_width; j++, bit >>= 1)
301 		if (logo_data[i][j].red)
302 		    val |= bit;
303 	    write_hex(val);
304 	}
305     }
306 
307     /* write logo structure and file footer */
308     write_footer();
309 }
310 
write_logo_vga16(void)311 static void write_logo_vga16(void)
312 {
313     unsigned int i, j, k;
314     unsigned char val;
315 
316     /* validate image */
317     for (i = 0; i < logo_height; i++)
318 	for (j = 0; j < logo_width; j++) {
319 	    for (k = 0; k < 16; k++)
320 		if (is_equal(logo_data[i][j], clut_vga16[k]))
321 		    break;
322 	    if (k == 16)
323 		die("Image must use the 16 console colors only\n"
324 		    "Use ppmquant(1) -map clut_vga16.ppm to reduce the number "
325 		    "of colors\n");
326 	}
327 
328     /* write file header */
329     write_header();
330 
331     /* write logo data */
332     for (i = 0; i < logo_height; i++)
333 	for (j = 0; j < logo_width; j++) {
334 	    for (k = 0; k < 16; k++)
335 		if (is_equal(logo_data[i][j], clut_vga16[k]))
336 		    break;
337 	    val = k<<4;
338 	    if (++j < logo_width) {
339 		for (k = 0; k < 16; k++)
340 		    if (is_equal(logo_data[i][j], clut_vga16[k]))
341 			break;
342 		val |= k;
343 	    }
344 	    write_hex(val);
345 	}
346 
347     /* write logo structure and file footer */
348     write_footer();
349 }
350 
write_logo_clut224(void)351 static void write_logo_clut224(void)
352 {
353     unsigned int i, j, k;
354 
355     /* validate image */
356     for (i = 0; i < logo_height; i++)
357 	for (j = 0; j < logo_width; j++) {
358 	    for (k = 0; k < logo_clutsize; k++)
359 		if (is_equal(logo_data[i][j], logo_clut[k]))
360 		    break;
361 	    if (k == logo_clutsize) {
362 		if (logo_clutsize == MAX_LINUX_LOGO_COLORS)
363 		    die("Image has more than %d colors\n"
364 			"Use ppmquant(1) to reduce the number of colors\n",
365 			MAX_LINUX_LOGO_COLORS);
366 		logo_clut[logo_clutsize++] = logo_data[i][j];
367 	    }
368 	}
369 
370     /* write file header */
371     write_header();
372 
373     /* write logo data */
374     for (i = 0; i < logo_height; i++)
375 	for (j = 0; j < logo_width; j++) {
376 	    for (k = 0; k < logo_clutsize; k++)
377 		if (is_equal(logo_data[i][j], logo_clut[k]))
378 		    break;
379 	    write_hex(k+32);
380 	}
381     fputs("\n};\n\n", out);
382 
383     /* write logo clut */
384     fprintf(out, "static unsigned char %s_clut[] __initdata = {\n",
385 	    logoname);
386     write_hex_cnt = 0;
387     for (i = 0; i < logo_clutsize; i++) {
388 	write_hex(logo_clut[i].red);
389 	write_hex(logo_clut[i].green);
390 	write_hex(logo_clut[i].blue);
391     }
392 
393     /* write logo structure and file footer */
394     write_footer();
395 }
396 
write_logo_gray256(void)397 static void write_logo_gray256(void)
398 {
399     unsigned int i, j;
400 
401     /* validate image */
402     for (i = 0; i < logo_height; i++)
403 	for (j = 0; j < logo_width; j++)
404 	    if (!is_gray(logo_data[i][j]))
405 		die("Image must be grayscale\n");
406 
407     /* write file header */
408     write_header();
409 
410     /* write logo data */
411     for (i = 0; i < logo_height; i++)
412 	for (j = 0; j < logo_width; j++)
413 	    write_hex(logo_data[i][j].red);
414 
415     /* write logo structure and file footer */
416     write_footer();
417 }
418 
die(const char * fmt,...)419 static void die(const char *fmt, ...)
420 {
421     va_list ap;
422 
423     va_start(ap, fmt);
424     vfprintf(stderr, fmt, ap);
425     va_end(ap);
426 
427     exit(1);
428 }
429 
usage(void)430 static void usage(void)
431 {
432     die("\n"
433 	"Usage: %s [options] <filename>\n"
434 	"\n"
435 	"Valid options:\n"
436 	"    -h          : display this usage information\n"
437 	"    -n <name>   : specify logo name (default: linux_logo)\n"
438 	"    -o <output> : output to file <output> instead of stdout\n"
439 	"    -t <type>   : specify logo type, one of\n"
440 	"                      mono    : monochrome black/white\n"
441 	"                      vga16   : 16 colors VGA text palette\n"
442 	"                      clut224 : 224 colors (default)\n"
443 	"                      gray256 : 256 levels grayscale\n"
444 	"\n", programname);
445 }
446 
main(int argc,char * argv[])447 int main(int argc, char *argv[])
448 {
449     int opt;
450 
451     programname = argv[0];
452 
453     opterr = 0;
454     while (1) {
455 	opt = getopt(argc, argv, "hn:o:t:");
456 	if (opt == -1)
457 	    break;
458 
459 	switch (opt) {
460 	    case 'h':
461 		usage();
462 		break;
463 
464 	    case 'n':
465 		logoname = optarg;
466 		break;
467 
468 	    case 'o':
469 		outputname = optarg;
470 		break;
471 
472 	    case 't':
473 		if (!strcmp(optarg, "mono"))
474 		    logo_type = LINUX_LOGO_MONO;
475 		else if (!strcmp(optarg, "vga16"))
476 		    logo_type = LINUX_LOGO_VGA16;
477 		else if (!strcmp(optarg, "clut224"))
478 		    logo_type = LINUX_LOGO_CLUT224;
479 		else if (!strcmp(optarg, "gray256"))
480 		    logo_type = LINUX_LOGO_GRAY256;
481 		else
482 		    usage();
483 		break;
484 
485 	    default:
486 		usage();
487 		break;
488 	}
489     }
490     if (optind != argc-1)
491 	usage();
492 
493     filename = argv[optind];
494 
495     read_image();
496     switch (logo_type) {
497 	case LINUX_LOGO_MONO:
498 	    write_logo_mono();
499 	    break;
500 
501 	case LINUX_LOGO_VGA16:
502 	    write_logo_vga16();
503 	    break;
504 
505 	case LINUX_LOGO_CLUT224:
506 	    write_logo_clut224();
507 	    break;
508 
509 	case LINUX_LOGO_GRAY256:
510 	    write_logo_gray256();
511 	    break;
512     }
513     exit(0);
514 }
515