1 /* -*- Mode: C; tab-width: 4 -*- */
2 /* lightning --- fractal lightning bolds */
5 static const char sccsid[] = "@(#)lightning.c 5.00 2000/11/01 xlockmore";
9 * Copyright (c) 1996 by Keith Romberg <kromberg@saxe.com>
11 * Permission to use, copy, modify, and distribute this software and its
12 * documentation for any purpose and without fee is hereby granted,
13 * provided that the above copyright notice appear in all copies and that
14 * both that copyright notice and this permission notice appear in
15 * supporting documentation.
17 * This file is provided AS IS with no warranties of any kind. The author
18 * shall have no liability with respect to the infringement of copyrights,
19 * trade secrets or any patents by this file or any part thereof. In no
20 * event will the author be liable for any lost revenue or profits or
21 * other special, indirect and consequential damages.
24 * 01-Nov-2000: Allocation checks
25 * 10-May-1997: Compatible with xscreensaver
26 * 14-Jul-1996: Cleaned up code.
27 * 27-Jun-1996: Written and submitted by Keith Romberg <kromberg@saxe.com>.
31 # define MODE_lightning
32 # define DEFAULTS "*delay: 10000 \n" \
35 # define BRIGHT_COLORS
36 # define free_lightning 0
37 # define release_lightning 0
38 # define reshape_lightning 0
39 # define lightning_handle_event 0
40 # include "xlockmore.h" /* in xscreensaver distribution */
41 #else /* STANDALONE */
42 # include "xlock.h" /* in xlockmore distribution */
43 #endif /* STANDALONE */
47 ENTRYPOINT ModeSpecOpt lightning_opts =
48 {0, (XrmOptionDescRec *) NULL, 0, (argtype *) NULL, (OptionStruct *) NULL};
51 ModStruct lightning_description =
52 {"lightning", "init_lightning", "draw_lightning", (char *) NULL,
53 "refresh_lightning", "init_lightning", (char *) NULL, &lightning_opts,
54 10000, 1, 1, 1, 64, 0.6, "",
55 "Shows Keith's fractal lightning bolts", 0, NULL};
60 #define BOLT_ITERATION 4
61 #define LONG_FORK_ITERATION 3
62 #define MEDIUM_FORK_ITERATION 2
63 #define SMALL_FORK_ITERATION 1
65 #define WIDTH_VARIATION 30
66 #define HEIGHT_VARIATION 15
68 #define DELAY_TIME_AMOUNT 15
69 #define MULTI_DELAY_TIME_BASE 5
71 #define MAX_WIGGLES 16
73 #define WIGGLE_AMOUNT 14
75 #define RANDOM_FORK_PROBILITY 4
77 #define FIRST_LEVEL_STRIKE 0
78 #define LEVEL_ONE_STRIKE 1
79 #define LEVEL_TWO_STRIKE 2
81 #define BOLT_VERTICIES ((1<<BOLT_ITERATION)-1)
82 /* BOLT_ITERATION = 4. 2^(BOLT_ITERATION) - 1 = 15 */
84 #define NUMBER_FORK_VERTICIES 9
86 #define FLASH_PROBILITY 20
87 #define MAX_FLASH_AMOUNT 2 /* half the total duration of the bolt */
90 XPoint ForkVerticies[NUMBER_FORK_VERTICIES];
96 XPoint middle[BOLT_VERTICIES];
104 int flash_begin, flash_stop;
110 Lightning bolts[BOLT_NUMBER];
111 int scr_width, scr_height;
120 static Storm *Helga = (Storm *) NULL;
122 /*------------------- function prototypes ----------------------------*/
124 static int distance(XPoint a, XPoint b);
126 static int setup_multi_strike(void);
127 static int flashing_strike(void);
128 static void flash_duration(int *start, int *end, int total_duration);
129 static void random_storm(Storm * st);
130 static void generate(XPoint A, XPoint B, int iter, XPoint * verts, int *vert_index);
131 static void create_fork(Fork * f, XPoint start, XPoint end, int level);
133 static void first_strike(Lightning bolt, ModeInfo * mi);
134 static void draw_bolt(Lightning * bolt, ModeInfo * mi);
135 static void draw_line(ModeInfo * mi, XPoint * p, int number, GC use, int x_offset);
136 static void level1_strike(Lightning bolt, ModeInfo * mi);
137 static void level2_strike(Lightning bolt, ModeInfo * mi);
139 static int storm_active(Storm * st);
140 static void update_bolt(Lightning * bolt, int time_now);
141 static void wiggle_bolt(Lightning * bolt);
142 static void wiggle_line(XPoint * p, int number, int wiggle_amount);
144 /*------------------------- functions ---------------------------------*/
147 setup_multi_strike(void)
149 int result, multi_prob;
151 multi_prob = NRAND(100);
155 else if ((multi_prob >= 51) && (multi_prob < 75))
157 else if ((multi_prob >= 76) && (multi_prob < 92))
160 result = BOLT_NUMBER; /* 4 */
165 /*-------------------------------------------------------------------------*/
168 flashing_strike(void)
170 int tmp = NRAND(FLASH_PROBILITY);
172 if (tmp <= FLASH_PROBILITY)
177 /*-------------------------------------------------------------------------*/
180 flash_duration(int *start, int *end, int total_duration)
184 mid = total_duration / MAX_FLASH_AMOUNT;
185 d = NRAND(total_duration / MAX_FLASH_AMOUNT) / 2;
190 /*-------------------------------------------------------------------------*/
193 random_storm(Storm * st)
198 for (i = 0; i < st->multi_strike; i++) {
199 st->bolts[i].end1.x = NRAND(st->scr_width);
200 st->bolts[i].end1.y = 0;
201 st->bolts[i].end2.x = NRAND(st->scr_width);
202 st->bolts[i].end2.y = st->scr_height;
203 st->bolts[i].wiggle_number = WIGGLE_BASE + NRAND(MAX_WIGGLES);
204 if ((st->bolts[i].flash = flashing_strike()))
205 flash_duration(&(st->bolts[i].flash_begin), &(st->bolts[i].flash_stop),
206 st->bolts[i].wiggle_number);
208 st->bolts[i].flash_begin = st->bolts[i].flash_stop = 0;
209 st->bolts[i].wiggle_amount = WIGGLE_AMOUNT;
211 st->bolts[i].delay_time = NRAND(DELAY_TIME_AMOUNT);
213 st->bolts[i].delay_time = NRAND(DELAY_TIME_AMOUNT) +
214 (MULTI_DELAY_TIME_BASE * i);
215 st->bolts[i].strike_level = FIRST_LEVEL_STRIKE;
217 generate(st->bolts[i].end1, st->bolts[i].end2, BOLT_ITERATION,
218 st->bolts[i].middle, &tmp);
219 st->bolts[i].fork_number = 0;
220 st->bolts[i].visible = 0;
221 for (j = 0; j < BOLT_VERTICIES; j++) {
222 if (st->bolts[i].fork_number >= 2)
224 if (NRAND(100) < RANDOM_FORK_PROBILITY) {
225 p.x = NRAND(st->scr_width);
226 p.y = st->scr_height;
227 st->bolts[i].forks_start[st->bolts[i].fork_number] = j;
228 create_fork(&(st->bolts[i].branch[st->bolts[i].fork_number]),
229 st->bolts[i].middle[j], p, j);
230 st->bolts[i].fork_number++;
237 generate(XPoint A, XPoint B, int iter, XPoint * verts, int *vert_index)
241 mid.x = (A.x + B.x) / 2 + NRAND(WIDTH_VARIATION) - WIDTH_VARIATION / 2;
242 mid.y = (A.y + B.y) / 2 + NRAND(HEIGHT_VARIATION) - HEIGHT_VARIATION / 2;
245 verts[*vert_index].x = mid.x;
246 verts[*vert_index].y = mid.y;
250 generate(A, mid, iter - 1, verts, vert_index);
251 generate(mid, B, iter - 1, verts, vert_index);
254 /*------------------------------------------------------------------------*/
257 create_fork(Fork * f, XPoint start, XPoint end, int level)
261 f->ForkVerticies[0].x = start.x;
262 f->ForkVerticies[0].y = start.y;
265 generate(start, end, LONG_FORK_ITERATION, f->ForkVerticies, &tmp);
267 } else if ((level > 6) && (level <= 11)) {
268 generate(start, end, MEDIUM_FORK_ITERATION, f->ForkVerticies, &tmp);
271 if (distance(start, end) > 100) {
272 generate(start, end, MEDIUM_FORK_ITERATION, f->ForkVerticies, &tmp);
275 generate(start, end, SMALL_FORK_ITERATION, f->ForkVerticies, &tmp);
280 f->ForkVerticies[f->num_used - 1].x = end.x;
281 f->ForkVerticies[f->num_used - 1].y = end.y;
284 /*------------------------------------------------------------------------*/
287 update_bolt(Lightning * bolt, int time_now)
290 if ((bolt->wiggle_amount == 0) && (bolt->wiggle_number > 2))
291 bolt->wiggle_number = 0;
292 if (((time_now % 3) == 0))
293 bolt->wiggle_amount++;
295 if (((time_now >= bolt->delay_time) && (time_now < bolt->flash_begin)) ||
296 (time_now > bolt->flash_stop))
301 if (time_now == bolt->delay_time)
302 bolt->strike_level = FIRST_LEVEL_STRIKE;
303 else if (time_now == (bolt->delay_time + 1))
304 bolt->strike_level = LEVEL_ONE_STRIKE;
305 else if ((time_now > (bolt->delay_time + 1)) &&
306 (time_now <= (bolt->delay_time + bolt->flash_begin - 2)))
307 bolt->strike_level = LEVEL_TWO_STRIKE;
308 else if (time_now == (bolt->delay_time + bolt->flash_begin - 1))
309 bolt->strike_level = LEVEL_ONE_STRIKE;
310 else if (time_now == (bolt->delay_time + bolt->flash_stop + 1))
311 bolt->strike_level = LEVEL_ONE_STRIKE;
313 bolt->strike_level = LEVEL_TWO_STRIKE;
316 /*------------------------------------------------------------------------*/
319 draw_bolt(Lightning * bolt, ModeInfo * mi)
322 if (bolt->strike_level == FIRST_LEVEL_STRIKE)
323 first_strike(*bolt, mi);
324 else if (bolt->strike_level == LEVEL_ONE_STRIKE)
325 level1_strike(*bolt, mi);
327 level2_strike(*bolt, mi);
331 /*------------------------------------------------------------------------*/
334 first_strike(Lightning bolt, ModeInfo * mi)
336 Display *display = MI_DISPLAY(mi);
337 Window window = MI_WINDOW(mi);
341 XSetForeground(display, gc, MI_WHITE_PIXEL(mi));
342 XDrawLine(display, window, gc,
343 bolt.end1.x, bolt.end1.y, bolt.middle[0].x, bolt.middle[0].y);
344 draw_line(mi, bolt.middle, BOLT_VERTICIES, gc, 0);
345 XDrawLine(display, window, gc,
346 bolt.middle[BOLT_VERTICIES - 1].x, bolt.middle[BOLT_VERTICIES - 1].y,
347 bolt.end2.x, bolt.end2.y);
349 for (i = 0; i < bolt.fork_number; i++)
350 draw_line(mi, bolt.branch[i].ForkVerticies, bolt.branch[i].num_used,
354 /*------------------------------------------------------------------------*/
357 draw_line(ModeInfo * mi, XPoint * points, int number, GC to_use, int offset)
361 for (i = 0; i < number - 1; i++) {
362 if (points[i].y <= points[i + 1].y)
363 XDrawLine(MI_DISPLAY(mi), MI_WINDOW(mi), to_use, points[i].x + offset,
364 points[i].y, points[i + 1].x + offset, points[i + 1].y);
366 if (points[i].x < points[i + 1].x)
367 XDrawLine(MI_DISPLAY(mi), MI_WINDOW(mi), to_use, points[i].x +
368 offset, points[i].y + offset, points[i + 1].x + offset,
369 points[i + 1].y + offset);
371 XDrawLine(MI_DISPLAY(mi), MI_WINDOW(mi), to_use, points[i].x -
372 offset, points[i].y + offset, points[i + 1].x - offset,
373 points[i + 1].y + offset);
378 /*------------------------------------------------------------------------*/
381 level1_strike(Lightning bolt, ModeInfo * mi)
383 Display *display = MI_DISPLAY(mi);
384 Window window = MI_WINDOW(mi);
385 Storm *st = &Helga[MI_SCREEN(mi)];
389 if (MI_NPIXELS(mi) > 2) /* color */
390 XSetForeground(display, gc, MI_PIXEL(mi, st->color));
392 XSetForeground(display, gc, MI_WHITE_PIXEL(mi));
393 XDrawLine(display, window, gc,
394 bolt.end1.x - 1, bolt.end1.y, bolt.middle[0].x - 1, bolt.middle[0].y);
395 draw_line(mi, bolt.middle, BOLT_VERTICIES, gc, -1);
396 XDrawLine(display, window, gc,
397 bolt.middle[BOLT_VERTICIES - 1].x - 1,
398 bolt.middle[BOLT_VERTICIES - 1].y, bolt.end2.x - 1, bolt.end2.y);
399 XDrawLine(display, window, gc,
400 bolt.end1.x + 1, bolt.end1.y, bolt.middle[0].x + 1, bolt.middle[0].y);
401 draw_line(mi, bolt.middle, BOLT_VERTICIES, gc, 1);
402 XDrawLine(display, window, gc,
403 bolt.middle[BOLT_VERTICIES - 1].x + 1,
404 bolt.middle[BOLT_VERTICIES - 1].y, bolt.end2.x + 1, bolt.end2.y);
406 for (i = 0; i < bolt.fork_number; i++) {
407 draw_line(mi, bolt.branch[i].ForkVerticies, bolt.branch[i].num_used,
409 draw_line(mi, bolt.branch[i].ForkVerticies, bolt.branch[i].num_used,
412 first_strike(bolt, mi);
415 /*------------------------------------------------------------------------*/
418 distance(XPoint a, XPoint b)
420 return ((int) sqrt((double) (a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y)));
423 /*------------------------------------------------------------------------*/
426 level2_strike(Lightning bolt, ModeInfo * mi)
428 Display *display = MI_DISPLAY(mi);
429 Window window = MI_WINDOW(mi);
430 Storm *st = &Helga[MI_SCREEN(mi)];
434 /* This was originally designed to be a little darker then the
435 level1 strike. This was changed to get it to work on
436 multiscreens and to add more color variety. I tried
437 stippling but it did not look good. */
438 if (MI_NPIXELS(mi) > 2)
439 XSetForeground(display, gc, MI_PIXEL(mi, st->color));
441 XSetForeground(display, gc, MI_WHITE_PIXEL(mi));
442 XDrawLine(display, window, gc,
443 bolt.end1.x - 2, bolt.end1.y, bolt.middle[0].x - 2, bolt.middle[0].y);
444 draw_line(mi, bolt.middle, BOLT_VERTICIES, gc, -2);
445 XDrawLine(display, window, gc,
446 bolt.middle[BOLT_VERTICIES - 1].x - 2,
447 bolt.middle[BOLT_VERTICIES - 1].y, bolt.end2.x - 2, bolt.end2.y);
449 XDrawLine(display, window, gc,
450 bolt.end1.x + 2, bolt.end1.y, bolt.middle[0].x + 2, bolt.middle[0].y);
451 draw_line(mi, bolt.middle, BOLT_VERTICIES, gc, 2);
452 XDrawLine(display, window, gc,
453 bolt.middle[BOLT_VERTICIES - 1].x + 2,
454 bolt.middle[BOLT_VERTICIES - 1].y, bolt.end2.x + 2, bolt.end2.y);
456 for (i = 0; i < bolt.fork_number; i++) {
457 draw_line(mi, bolt.branch[i].ForkVerticies, bolt.branch[i].num_used,
459 draw_line(mi, bolt.branch[i].ForkVerticies, bolt.branch[i].num_used,
462 level1_strike(bolt, mi);
465 /*------------------------------------------------------------------------*/
468 storm_active(Storm * st)
470 int i, atleast_1 = 0;
472 for (i = 0; i < st->multi_strike; i++)
473 if (st->bolts[i].wiggle_number > 0)
479 /*------------------------------------------------------------------------*/
482 wiggle_bolt(Lightning * bolt)
486 wiggle_line(bolt->middle, BOLT_VERTICIES, bolt->wiggle_amount);
487 bolt->end2.x += NRAND(bolt->wiggle_amount) - bolt->wiggle_amount / 2;
488 bolt->end2.y += NRAND(bolt->wiggle_amount) - bolt->wiggle_amount / 2;
490 for (i = 0; i < bolt->fork_number; i++) {
491 wiggle_line(bolt->branch[i].ForkVerticies, bolt->branch[i].num_used,
492 bolt->wiggle_amount);
493 bolt->branch[i].ForkVerticies[0].x = bolt->middle[bolt->forks_start[i]].x;
494 bolt->branch[i].ForkVerticies[0].y = bolt->middle[bolt->forks_start[i]].y;
497 if (bolt->wiggle_amount > 1)
498 bolt->wiggle_amount -= 1;
500 bolt->wiggle_amount = 0;
503 /*------------------------------------------------------------------------*/
506 wiggle_line(XPoint * p, int number, int amount)
510 for (i = 0; i < number; i++) {
511 p[i].x += NRAND(amount) - amount / 2;
512 p[i].y += NRAND(amount) - amount / 2;
516 /*------------------------------------------------------------------------*/
519 init_lightning (ModeInfo * mi)
524 st = &Helga[MI_SCREEN(mi)];
526 st->scr_width = MI_WIDTH(mi);
527 st->scr_height = MI_HEIGHT(mi);
529 st->multi_strike = setup_multi_strike();
534 /*------------------------------------------------------------------------*/
537 draw_lightning (ModeInfo * mi)
544 st = &Helga[MI_SCREEN(mi)];
545 MI_IS_DRAWN(mi) = True;
548 XClearWindow(MI_DISPLAY(mi), MI_WINDOW(mi));
550 st->color = NRAND(MI_NPIXELS(mi));
552 if (storm_active(st))
558 for (i = 0; i < st->multi_strike; i++) {
559 if (st->bolts[i].visible)
560 draw_bolt(&(st->bolts[i]), mi);
561 update_bolt(&(st->bolts[i]), st->draw_time);
568 if (++st->busyLoop > 6) {
574 XClearWindow(MI_DISPLAY(mi), MI_WINDOW(mi));
576 if (storm_active(st))
582 if (++st->busyLoop > 100) {
592 refresh_lightning(ModeInfo * mi)
594 /* Do nothing, it will refresh by itself */
598 XSCREENSAVER_MODULE ("Lightning", lightning)
601 #endif /* MODE_lightning */