From http://www.jwz.org/xscreensaver/xscreensaver-5.38.tar.gz
[xscreensaver] / hacks / sierpinski.c
1 /* -*- Mode: C; tab-width: 4 -*- */
2 /* sierpinski --- Sierpinski's triangle fractal */
3
4 #if 0
5 static const char sccsid[] = "@(#)sierpinski.c  5.00 2000/11/01 xlockmore";
6 #endif
7
8 /*-
9  * Copyright (c) 1996 by Desmond Daignault
10  *
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.
16  *
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.
22  *
23  * Dots initially appear where they "should not".  Later they get
24  * "focused".  This is correct behavior.
25  *
26  * Revision History:
27  * 01-Nov-2000: Allocation checks
28  * 18-Sep-1997: 3D version Antti Kuntsi <kuntsi@iki.fi>.
29  * 20-May-1997: Changed the name tri to sierpinski for more compatiblity
30  * 10-May-1997: Jamie Zawinski <jwz@jwz.org> compatible with xscreensaver
31  * 05-Sep-1996: Desmond Daignault Datatimes Incorporated
32  *            <tekdd@dtol.datatimes.com> .
33  */
34
35 #ifdef STANDALONE
36 # define MODE_sierpinski
37 # define DEFAULTS       "*delay: 400000 \n" \
38                                         "*count: 2000 \n" \
39                                         "*cycles: 100 \n" \
40                                         "*ncolors: 64 \n" \
41                                         "*fpsSolid: true \n" \
42                                         "*ignoreRotation: True \n" \
43                                     "*lowrez: True \n" \
44
45 # define BRIGHT_COLORS
46 # define release_sierpinski 0
47 # define reshape_sierpinski 0
48 # define sierpinski_handle_event 0
49 # include "xlockmore.h"         /* in xscreensaver distribution */
50 #else /* STANDALONE */
51 # include "xlock.h"             /* in xlockmore distribution */
52 #endif /* STANDALONE */
53
54 #ifdef MODE_sierpinski
55
56 ENTRYPOINT ModeSpecOpt sierpinski_opts =
57 {0, (XrmOptionDescRec *) NULL, 0, (argtype *) NULL, (OptionStruct *) NULL};
58
59 #ifdef USE_MODULES
60 ModStruct   sierpinski_description =
61 {"sierpinski", "init_sierpinski", "draw_sierpinski", (char *) NULL,
62  "refresh_sierpinski", "init_sierpinski", "free_sierpinski", &sierpinski_opts,
63  400000, 2000, 100, 1, 64, 1.0, "",
64  "Shows Sierpinski's triangle", 0, NULL};
65
66 #endif
67
68 #define MAXCORNERS 4
69
70 typedef struct {
71         int         width, height;
72         int         time;
73         int         px, py;
74         int         total_npoints;
75         int         corners;
76         int         npoints[MAXCORNERS];
77         unsigned long colors[MAXCORNERS];
78         XPoint     *pointBuffer[MAXCORNERS];
79         XPoint      vertex[MAXCORNERS];
80 } sierpinskistruct;
81
82 static sierpinskistruct *tris = (sierpinskistruct *) NULL;
83
84 static void
85 startover(ModeInfo * mi)
86 {
87         int         j;
88         sierpinskistruct *sp = &tris[MI_SCREEN(mi)];
89
90         if (MI_NPIXELS(mi) > 2) {
91                 if (sp->corners == 3) {
92                         sp->colors[0] = (NRAND(MI_NPIXELS(mi)));
93                         sp->colors[1] = (sp->colors[0] + MI_NPIXELS(mi) / 7 +
94                          NRAND(2 * MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
95                         sp->colors[2] = (sp->colors[0] + 4 * MI_NPIXELS(mi) / 7 +
96                          NRAND(2 * MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
97                 } else if (sp->corners == 4) {
98                         sp->colors[0] = (NRAND(MI_NPIXELS(mi)));
99                         sp->colors[1] = (sp->colors[0] + MI_NPIXELS(mi) / 7 +
100                              NRAND(MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
101                         sp->colors[2] = (sp->colors[0] + 3 * MI_NPIXELS(mi) / 7 +
102                              NRAND(MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
103                         sp->colors[3] = (sp->colors[0] + 5 * MI_NPIXELS(mi) / 7 +
104                              NRAND(MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
105                 } else {
106                         (void) fprintf(stderr, "colors not set for %d corners\n", sp->corners);
107                 }
108         }
109         for (j = 0; j < sp->corners; j++) {
110                 sp->vertex[j].x = NRAND(sp->width);
111                 sp->vertex[j].y = NRAND(sp->height);
112         }
113         sp->px = NRAND(sp->width);
114         sp->py = NRAND(sp->height);
115         sp->time = 0;
116
117         MI_CLEARWINDOW(mi);
118 }
119
120 ENTRYPOINT void
121 free_sierpinski(ModeInfo * mi)
122 {
123         sierpinskistruct *sp = &tris[MI_SCREEN(mi)];
124         int corner;
125
126         for (corner = 0; corner < MAXCORNERS; corner++)
127                 if (sp->pointBuffer[corner] != NULL) {
128                         (void) free((void *) sp->pointBuffer[corner]);
129                         sp->pointBuffer[corner] = (XPoint *) NULL;
130                 }
131 }
132
133 ENTRYPOINT void
134 init_sierpinski(ModeInfo * mi)
135 {
136         int         i;
137         sierpinskistruct *sp;
138
139         MI_INIT (mi, tris);
140         sp = &tris[MI_SCREEN(mi)];
141
142         sp->width = MI_WIDTH(mi);
143         sp->height = MI_HEIGHT(mi);
144
145         sp->total_npoints = MI_COUNT(mi);
146         if (sp->total_npoints < 1)
147                 sp->total_npoints = 1;
148         sp->corners = MI_SIZE(mi);
149         if (sp->corners < 3 || sp->corners > 4) {
150                 sp->corners = (int) (LRAND() & 1) + 3;
151         }
152         for (i = 0; i < sp->corners; i++) {
153                 if (!sp->pointBuffer[i])
154                         if ((sp->pointBuffer[i] = (XPoint *) malloc(sp->total_npoints *
155                                         sizeof (XPoint))) == NULL) {
156                                 free_sierpinski(mi);
157                                 return;
158                         }
159         }
160         startover(mi);
161 }
162
163 ENTRYPOINT void
164 draw_sierpinski(ModeInfo * mi)
165 {
166         Display    *display = MI_DISPLAY(mi);
167         GC          gc = MI_GC(mi);
168         XPoint     *xp[MAXCORNERS];
169         int         i, v;
170         sierpinskistruct *sp;
171
172         if (tris == NULL)
173                 return;
174         sp = &tris[MI_SCREEN(mi)];
175         if (sp->pointBuffer[0] == NULL)
176                 return;
177
178         MI_IS_DRAWN(mi) = True;
179         if (MI_NPIXELS(mi) <= 2)
180                 XSetForeground(display, gc, MI_WHITE_PIXEL(mi));
181         for (i = 0; i < sp->corners; i++)
182                 xp[i] = sp->pointBuffer[i];
183         for (i = 0; i < sp->total_npoints; i++) {
184                 v = NRAND(sp->corners);
185                 sp->px = (sp->px + sp->vertex[v].x) / 2;
186                 sp->py = (sp->py + sp->vertex[v].y) / 2;
187                 xp[v]->x = sp->px;
188                 xp[v]->y = sp->py;
189                 xp[v]++;
190                 sp->npoints[v]++;
191         }
192         for (i = 0; i < sp->corners; i++) {
193                 if (MI_NPIXELS(mi) > 2)
194                         XSetForeground(display, gc, MI_PIXEL(mi, sp->colors[i]));
195                 XDrawPoints(display, MI_WINDOW(mi), gc, sp->pointBuffer[i], sp->npoints[i],
196                             CoordModeOrigin);
197                 sp->npoints[i] = 0;
198         }
199         if (++sp->time >= MI_CYCLES(mi))
200                 startover(mi);
201 }
202
203 #ifndef STANDALONE
204 ENTRYPOINT void
205 refresh_sierpinski(ModeInfo * mi)
206 {
207         MI_CLEARWINDOW(mi);
208 }
209 #endif
210
211 XSCREENSAVER_MODULE ("Sierpinski", sierpinski)
212
213 #endif /* MODE_sierpinski */