From http://www.jwz.org/xscreensaver/xscreensaver-5.37.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
44 # define BRIGHT_COLORS
45 # define release_sierpinski 0
46 # define sierpinski_handle_event 0
47 # include "xlockmore.h"         /* in xscreensaver distribution */
48 #else /* STANDALONE */
49 # include "xlock.h"             /* in xlockmore distribution */
50 #endif /* STANDALONE */
51
52 #ifdef MODE_sierpinski
53
54 ENTRYPOINT ModeSpecOpt sierpinski_opts =
55 {0, (XrmOptionDescRec *) NULL, 0, (argtype *) NULL, (OptionStruct *) NULL};
56
57 #ifdef USE_MODULES
58 ModStruct   sierpinski_description =
59 {"sierpinski", "init_sierpinski", "draw_sierpinski", (char *) NULL,
60  "refresh_sierpinski", "init_sierpinski", (char *) NULL, &sierpinski_opts,
61  400000, 2000, 100, 1, 64, 1.0, "",
62  "Shows Sierpinski's triangle", 0, NULL};
63
64 #endif
65
66 #define MAXCORNERS 4
67
68 typedef struct {
69         int         width, height;
70         int         time;
71         int         px, py;
72         int         total_npoints;
73         int         corners;
74         int         npoints[MAXCORNERS];
75         unsigned long colors[MAXCORNERS];
76         XPoint     *pointBuffer[MAXCORNERS];
77         XPoint      vertex[MAXCORNERS];
78 } sierpinskistruct;
79
80 static sierpinskistruct *tris = (sierpinskistruct *) NULL;
81
82 static void
83 startover(ModeInfo * mi)
84 {
85         int         j;
86         sierpinskistruct *sp = &tris[MI_SCREEN(mi)];
87
88         if (MI_NPIXELS(mi) > 2) {
89                 if (sp->corners == 3) {
90                         sp->colors[0] = (NRAND(MI_NPIXELS(mi)));
91                         sp->colors[1] = (sp->colors[0] + MI_NPIXELS(mi) / 7 +
92                          NRAND(2 * MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
93                         sp->colors[2] = (sp->colors[0] + 4 * MI_NPIXELS(mi) / 7 +
94                          NRAND(2 * MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
95                 } else if (sp->corners == 4) {
96                         sp->colors[0] = (NRAND(MI_NPIXELS(mi)));
97                         sp->colors[1] = (sp->colors[0] + MI_NPIXELS(mi) / 7 +
98                              NRAND(MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
99                         sp->colors[2] = (sp->colors[0] + 3 * MI_NPIXELS(mi) / 7 +
100                              NRAND(MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
101                         sp->colors[3] = (sp->colors[0] + 5 * MI_NPIXELS(mi) / 7 +
102                              NRAND(MI_NPIXELS(mi) / 7 + 1)) % MI_NPIXELS(mi);
103                 } else {
104                         (void) fprintf(stderr, "colors not set for %d corners\n", sp->corners);
105                 }
106         }
107         for (j = 0; j < sp->corners; j++) {
108                 sp->vertex[j].x = NRAND(sp->width);
109                 sp->vertex[j].y = NRAND(sp->height);
110         }
111         sp->px = NRAND(sp->width);
112         sp->py = NRAND(sp->height);
113         sp->time = 0;
114
115         MI_CLEARWINDOW(mi);
116 }
117
118 static void
119 free_sierpinski(ModeInfo * mi)
120 {
121         sierpinskistruct *sp = &tris[MI_SCREEN(mi)];
122         int corner;
123
124         for (corner = 0; corner < MAXCORNERS; corner++)
125                 if (sp->pointBuffer[corner] != NULL) {
126                         (void) free((void *) sp->pointBuffer[corner]);
127                         sp->pointBuffer[corner] = (XPoint *) NULL;
128                 }
129 }
130
131 ENTRYPOINT void
132 init_sierpinski(ModeInfo * mi)
133 {
134         int         i;
135         sierpinskistruct *sp;
136
137         MI_INIT (mi, tris, free_sierpinski);
138         sp = &tris[MI_SCREEN(mi)];
139
140         sp->width = MI_WIDTH(mi);
141         sp->height = MI_HEIGHT(mi);
142
143         sp->total_npoints = MI_COUNT(mi);
144         if (sp->total_npoints < 1)
145                 sp->total_npoints = 1;
146         sp->corners = MI_SIZE(mi);
147         if (sp->corners < 3 || sp->corners > 4) {
148                 sp->corners = (int) (LRAND() & 1) + 3;
149         }
150         for (i = 0; i < sp->corners; i++) {
151                 if (!sp->pointBuffer[i])
152                         if ((sp->pointBuffer[i] = (XPoint *) malloc(sp->total_npoints *
153                                         sizeof (XPoint))) == NULL) {
154                                 free_sierpinski(mi);
155                                 return;
156                         }
157         }
158         startover(mi);
159 }
160
161 ENTRYPOINT void
162 draw_sierpinski(ModeInfo * mi)
163 {
164         Display    *display = MI_DISPLAY(mi);
165         GC          gc = MI_GC(mi);
166         XPoint     *xp[MAXCORNERS];
167         int         i, v;
168         sierpinskistruct *sp;
169
170         if (tris == NULL)
171                 return;
172         sp = &tris[MI_SCREEN(mi)];
173         if (sp->pointBuffer[0] == NULL)
174                 return;
175
176         MI_IS_DRAWN(mi) = True;
177         if (MI_NPIXELS(mi) <= 2)
178                 XSetForeground(display, gc, MI_WHITE_PIXEL(mi));
179         for (i = 0; i < sp->corners; i++)
180                 xp[i] = sp->pointBuffer[i];
181         for (i = 0; i < sp->total_npoints; i++) {
182                 v = NRAND(sp->corners);
183                 sp->px = (sp->px + sp->vertex[v].x) / 2;
184                 sp->py = (sp->py + sp->vertex[v].y) / 2;
185                 xp[v]->x = sp->px;
186                 xp[v]->y = sp->py;
187                 xp[v]++;
188                 sp->npoints[v]++;
189         }
190         for (i = 0; i < sp->corners; i++) {
191                 if (MI_NPIXELS(mi) > 2)
192                         XSetForeground(display, gc, MI_PIXEL(mi, sp->colors[i]));
193                 XDrawPoints(display, MI_WINDOW(mi), gc, sp->pointBuffer[i], sp->npoints[i],
194                             CoordModeOrigin);
195                 sp->npoints[i] = 0;
196         }
197         if (++sp->time >= MI_CYCLES(mi))
198                 startover(mi);
199 }
200
201 ENTRYPOINT void
202 reshape_sierpinski(ModeInfo * mi, int width, int height)
203 {
204   XClearWindow (MI_DISPLAY (mi), MI_WINDOW(mi));
205   init_sierpinski (mi);
206 }
207
208 ENTRYPOINT void
209 refresh_sierpinski(ModeInfo * mi)
210 {
211         MI_CLEARWINDOW(mi);
212 }
213
214 XSCREENSAVER_MODULE ("Sierpinski", sierpinski)
215
216 #endif /* MODE_sierpinski */