Magellan Linux

Contents of /trunk/mkinitrd-magellan/busybox/miscutils/fbsplash.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 816 - (show annotations) (download)
Fri Apr 24 18:33:46 2009 UTC (15 years, 1 month ago) by niro
File MIME type: text/plain
File size: 10140 byte(s)
-updated to busybox-1.13.4
1 /* vi: set sw=4 ts=4: */
2 /*
3 * Copyright (C) 2008 Michele Sanges <michele.sanges@otomelara.it>,
4 * <michele.sanges@gmail.it>
5 *
6 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
7 *
8 * Usage:
9 * - use kernel option 'vga=xxx' or otherwise enable framebuffer device.
10 * - put somewhere fbsplash.cfg file and an image in .ppm format.
11 * - run applet: $ setsid fbsplash [params] &
12 * -c: hide cursor
13 * -d /dev/fbN: framebuffer device (if not /dev/fb0)
14 * -s path_to_image_file (can be "-" for stdin)
15 * -i path_to_cfg_file
16 * -f path_to_fifo (can be "-" for stdin)
17 * - if you want to run it only in presence of a kernel parameter
18 * (for example fbsplash=on), use:
19 * grep -q "fbsplash=on" </proc/cmdline && setsid fbsplash [params]
20 * - commands for fifo:
21 * "NN" (ASCII decimal number) - percentage to show on progress bar.
22 * "exit" (or just close fifo) - well you guessed it.
23 */
24
25 #include "libbb.h"
26 #include <linux/fb.h>
27
28 /* If you want logging messages on /tmp/fbsplash.log... */
29 #define DEBUG 0
30
31 #define BYTES_PER_PIXEL 2
32
33 typedef unsigned short DATA;
34
35 struct globals {
36 #if DEBUG
37 bool bdebug_messages; // enable/disable logging
38 FILE *logfile_fd; // log file
39 #endif
40 unsigned char *addr; // pointer to framebuffer memory
41 unsigned ns[7]; // n-parameters
42 const char *image_filename;
43 struct fb_var_screeninfo scr_var;
44 struct fb_fix_screeninfo scr_fix;
45 };
46 #define G (*ptr_to_globals)
47 #define INIT_G() do { \
48 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
49 } while (0)
50
51 #define nbar_width ns[0] // progress bar width
52 #define nbar_height ns[1] // progress bar height
53 #define nbar_posx ns[2] // progress bar horizontal position
54 #define nbar_posy ns[3] // progress bar vertical position
55 #define nbar_colr ns[4] // progress bar color red component
56 #define nbar_colg ns[5] // progress bar color green component
57 #define nbar_colb ns[6] // progress bar color blue component
58
59 #if DEBUG
60 #define DEBUG_MESSAGE(strMessage, args...) \
61 if (G.bdebug_messages) { \
62 fprintf(G.logfile_fd, "[%s][%s] - %s\n", \
63 __FILE__, __FUNCTION__, strMessage); \
64 }
65 #else
66 #define DEBUG_MESSAGE(...) ((void)0)
67 #endif
68
69
70 /**
71 * Open and initialize the framebuffer device
72 * \param *strfb_device pointer to framebuffer device
73 */
74 static void fb_open(const char *strfb_device)
75 {
76 int fbfd = xopen(strfb_device, O_RDWR);
77
78 // framebuffer properties
79 xioctl(fbfd, FBIOGET_VSCREENINFO, &G.scr_var);
80 xioctl(fbfd, FBIOGET_FSCREENINFO, &G.scr_fix);
81
82 if (G.scr_var.bits_per_pixel != 16)
83 bb_error_msg_and_die("only 16 bpp is supported");
84
85 // map the device in memory
86 G.addr = mmap(NULL,
87 G.scr_var.xres * G.scr_var.yres
88 * BYTES_PER_PIXEL /*(G.scr_var.bits_per_pixel / 8)*/ ,
89 PROT_WRITE, MAP_SHARED, fbfd, 0);
90 if (G.addr == MAP_FAILED)
91 bb_perror_msg_and_die("can't mmap %s", strfb_device);
92 close(fbfd);
93 }
94
95
96 /**
97 * Draw hollow rectangle on framebuffer
98 */
99 static void fb_drawrectangle(void)
100 {
101 int cnt;
102 DATA thispix;
103 DATA *ptr1, *ptr2;
104 unsigned char nred = G.nbar_colr/2;
105 unsigned char ngreen = G.nbar_colg/2;
106 unsigned char nblue = G.nbar_colb/2;
107
108 nred >>= 3; // 5-bit red
109 ngreen >>= 2; // 6-bit green
110 nblue >>= 3; // 5-bit blue
111 thispix = nblue + (ngreen << 5) + (nred << (5+6));
112
113 // horizontal lines
114 ptr1 = (DATA*)(G.addr + (G.nbar_posy * G.scr_var.xres + G.nbar_posx) * BYTES_PER_PIXEL);
115 ptr2 = (DATA*)(G.addr + ((G.nbar_posy + G.nbar_height - 1) * G.scr_var.xres + G.nbar_posx) * BYTES_PER_PIXEL);
116 cnt = G.nbar_width - 1;
117 do {
118 *ptr1++ = thispix;
119 *ptr2++ = thispix;
120 } while (--cnt >= 0);
121
122 // vertical lines
123 ptr1 = (DATA*)(G.addr + (G.nbar_posy * G.scr_var.xres + G.nbar_posx) * BYTES_PER_PIXEL);
124 ptr2 = (DATA*)(G.addr + (G.nbar_posy * G.scr_var.xres + G.nbar_posx + G.nbar_width - 1) * BYTES_PER_PIXEL);
125 cnt = G.nbar_posy + G.nbar_height - 1 - G.nbar_posy;
126 do {
127 *ptr1 = thispix; ptr1 += G.scr_var.xres;
128 *ptr2 = thispix; ptr2 += G.scr_var.xres;
129 } while (--cnt >= 0);
130 }
131
132
133 /**
134 * Draw filled rectangle on framebuffer
135 * \param nx1pos,ny1pos upper left position
136 * \param nx2pos,ny2pos down right position
137 * \param nred,ngreen,nblue rgb color
138 */
139 static void fb_drawfullrectangle(int nx1pos, int ny1pos, int nx2pos, int ny2pos,
140 unsigned char nred, unsigned char ngreen, unsigned char nblue)
141 {
142 int cnt1, cnt2, nypos;
143 DATA thispix;
144 DATA *ptr;
145
146 nred >>= 3; // 5-bit red
147 ngreen >>= 2; // 6-bit green
148 nblue >>= 3; // 5-bit blue
149 thispix = nblue + (ngreen << 5) + (nred << (5+6));
150
151 cnt1 = ny2pos - ny1pos;
152 nypos = ny1pos;
153 do {
154 ptr = (DATA*)(G.addr + (nypos * G.scr_var.xres + nx1pos) * BYTES_PER_PIXEL);
155 cnt2 = nx2pos - nx1pos;
156 do {
157 *ptr++ = thispix;
158 } while (--cnt2 >= 0);
159
160 nypos++;
161 } while (--cnt1 >= 0);
162 }
163
164
165 /**
166 * Draw a progress bar on framebuffer
167 * \param percent percentage of loading
168 */
169 static void fb_drawprogressbar(unsigned percent)
170 {
171 int i, left_x, top_y, width, height;
172
173 // outer box
174 left_x = G.nbar_posx;
175 top_y = G.nbar_posy;
176 width = G.nbar_width - 1;
177 height = G.nbar_height - 1;
178 if ((height | width) < 0)
179 return;
180 // NB: "width" of 1 actually makes rect with width of 2!
181 fb_drawrectangle();
182
183 // inner "empty" rectangle
184 left_x++;
185 top_y++;
186 width -= 2;
187 height -= 2;
188 if ((height | width) < 0)
189 return;
190 fb_drawfullrectangle(
191 left_x, top_y,
192 left_x + width, top_y + height,
193 G.nbar_colr, G.nbar_colg, G.nbar_colb);
194
195 if (percent > 0) {
196 // actual progress bar
197 width = width * percent / 100;
198 i = height;
199 if (height == 0)
200 height++; // divide by 0 is bad
201 while (i >= 0) {
202 // draw one-line thick "rectangle"
203 // top line will have gray lvl 200, bottom one 100
204 unsigned gray_level = 100 + i*100/height;
205 fb_drawfullrectangle(
206 left_x, top_y, left_x + width, top_y,
207 gray_level, gray_level, gray_level);
208 top_y++;
209 i--;
210 }
211 }
212 }
213
214
215 /**
216 * Draw image from PPM file
217 */
218 static void fb_drawimage(void)
219 {
220 char head[256];
221 char s[80];
222 FILE *theme_file;
223 unsigned char *pixline;
224 unsigned i, j, width, height, line_size;
225
226 memset(head, 0, sizeof(head));
227 theme_file = xfopen_stdin(G.image_filename);
228
229 // parse ppm header
230 while (1) {
231 if (fgets(s, sizeof(s), theme_file) == NULL)
232 bb_error_msg_and_die("bad PPM file '%s'", G.image_filename);
233
234 if (s[0] == '#')
235 continue;
236
237 if (strlen(head) + strlen(s) >= sizeof(head))
238 bb_error_msg_and_die("bad PPM file '%s'", G.image_filename);
239
240 strcat(head, s);
241 if (head[0] != 'P' || head[1] != '6')
242 bb_error_msg_and_die("bad PPM file '%s'", G.image_filename);
243
244 // width, height, max_color_val
245 if (sscanf(head, "P6 %u %u %u", &width, &height, &i) == 3)
246 break;
247 // TODO: i must be <= 255!
248 }
249
250 line_size = width*3;
251 if (width > G.scr_var.xres)
252 width = G.scr_var.xres;
253 if (height > G.scr_var.yres)
254 height = G.scr_var.yres;
255
256 pixline = xmalloc(line_size);
257 for (j = 0; j < height; j++) {
258 unsigned char *pixel = pixline;
259 DATA *src = (DATA *)(G.addr + j * G.scr_fix.line_length);
260
261 if (fread(pixline, 1, line_size, theme_file) != line_size)
262 bb_error_msg_and_die("bad PPM file '%s'", G.image_filename);
263 for (i = 0; i < width; i++) {
264 unsigned thispix;
265 thispix = (((unsigned)pixel[0] << 8) & 0xf800)
266 | (((unsigned)pixel[1] << 3) & 0x07e0)
267 | (((unsigned)pixel[2] >> 3));
268 *src++ = thispix;
269 pixel += 3;
270 }
271 }
272 free(pixline);
273 fclose(theme_file);
274 }
275
276
277 /**
278 * Parse configuration file
279 * \param *cfg_filename name of the configuration file
280 */
281 static void init(const char *cfg_filename)
282 {
283 static const char const param_names[] ALIGN1 =
284 "BAR_WIDTH\0" "BAR_HEIGHT\0"
285 "BAR_LEFT\0" "BAR_TOP\0"
286 "BAR_R\0" "BAR_G\0" "BAR_B\0"
287 #if DEBUG
288 "DEBUG\0"
289 #endif
290 ;
291 char *token[2];
292 parser_t *parser = config_open2(cfg_filename, xfopen_stdin);
293 while (config_read(parser, token, 2, 2, "#=",
294 (PARSE_NORMAL | PARSE_MIN_DIE) & ~(PARSE_TRIM | PARSE_COLLAPSE))) {
295 unsigned val = xatoi_u(token[1]);
296 int i = index_in_strings(param_names, token[0]);
297 if (i < 0)
298 bb_error_msg_and_die("syntax error: '%s'", token[0]);
299 if (i >= 0 && i < 7)
300 G.ns[i] = val;
301 #if DEBUG
302 if (i == 7) {
303 G.bdebug_messages = val;
304 if (G.bdebug_messages)
305 G.logfile_fd = xfopen_for_write("/tmp/fbsplash.log");
306 }
307 #endif
308 }
309 config_close(parser);
310 }
311
312
313 int fbsplash_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
314 int fbsplash_main(int argc UNUSED_PARAM, char **argv)
315 {
316 const char *fb_device, *cfg_filename, *fifo_filename;
317 FILE *fp = fp; // for compiler
318 char *num_buf;
319 unsigned num;
320 bool bCursorOff;
321
322 INIT_G();
323
324 // parse command line options
325 fb_device = "/dev/fb0";
326 cfg_filename = NULL;
327 fifo_filename = NULL;
328 bCursorOff = 1 & getopt32(argv, "cs:d:i:f:",
329 &G.image_filename, &fb_device, &cfg_filename, &fifo_filename);
330
331 // parse configuration file
332 if (cfg_filename)
333 init(cfg_filename);
334
335 // We must have -s IMG
336 if (!G.image_filename)
337 bb_show_usage();
338
339 fb_open(fb_device);
340
341 if (fifo_filename && bCursorOff) {
342 // hide cursor (BEFORE any fb ops)
343 full_write(STDOUT_FILENO, "\x1b" "[?25l", 6);
344 }
345
346 fb_drawimage();
347
348 if (!fifo_filename)
349 return EXIT_SUCCESS;
350
351 fp = xfopen_stdin(fifo_filename);
352 if (fp != stdin) {
353 // For named pipes, we want to support this:
354 // mkfifo cmd_pipe
355 // fbsplash -f cmd_pipe .... &
356 // ...
357 // echo 33 >cmd_pipe
358 // ...
359 // echo 66 >cmd_pipe
360 // This means that we don't want fbsplash to get EOF
361 // when last writer closes input end.
362 // The simplest way is to open fifo for writing too
363 // and become an additional writer :)
364 open(fifo_filename, O_WRONLY); // errors are ignored
365 }
366
367 fb_drawprogressbar(0);
368 // Block on read, waiting for some input.
369 // Use of <stdio.h> style I/O allows to correctly
370 // handle a case when we have many buffered lines
371 // already in the pipe
372 while ((num_buf = xmalloc_fgetline(fp)) != NULL) {
373 if (strncmp(num_buf, "exit", 4) == 0) {
374 DEBUG_MESSAGE("exit");
375 break;
376 }
377 num = atoi(num_buf);
378 if (isdigit(num_buf[0]) && (num <= 100)) {
379 #if DEBUG
380 char strVal[10];
381 sprintf(strVal, "%d", num);
382 DEBUG_MESSAGE(strVal);
383 #endif
384 fb_drawprogressbar(num);
385 }
386 free(num_buf);
387 }
388
389 if (bCursorOff) // restore cursor
390 full_write(STDOUT_FILENO, "\x1b" "[?25h", 6);
391
392 return EXIT_SUCCESS;
393 }