git.hungrycats.org Git - xscreensaver/blob - hacks/worm.c

   1
   2 /* -*- Mode: C; tab-width: 4 -*- */
   3 /* worm --- draw wiggly worms */
   4
   5 #if 0
   6 static const char sccsid[] = "@(#)worm.c        4.04 97/07/28 xlockmore";
   7 #endif
   8
   9 /*-
  10  * Copyright (c) 1991 by Patrick J. Naughton.
  11  *
  12  * Permission to use, copy, modify, and distribute this software and its
  13  * documentation for any purpose and without fee is hereby granted,
  14  * provided that the above copyright notice appear in all copies and that
  15  * both that copyright notice and this permission notice appear in
  16  * supporting documentation.
  17  *
  18  * This file is provided AS IS with no warranties of any kind.  The author
  19  * shall have no liability with respect to the infringement of copyrights,
  20  * trade secrets or any patents by this file or any part thereof.  In no
  21  * event will the author be liable for any lost revenue or profits or
  22  * other special, indirect and consequential damages.
  23  *
  24  * Revision History:
  25  * 10-May-97: Compatible with xscreensaver
  26  * 03-Sep-96: fixed bug in allocation of space for worms, added 3d support
  27  *            Henrik Theiling <theiling@coli.uni-sb.de>
  28  * 27-Sep-95: put back malloc
  29  * 23-Sep-93: got rid of "rint". (David Bagley)
  30  * 27-Sep-91: got rid of all malloc calls since there were no calls to free().
  31  * 25-Sep-91: Integrated into X11R5 contrib xlock.
  32  *
  33  * Adapted from a concept in the Dec 87 issue of Scientific American p. 142.
  34  *
  35  * SunView version: Brad Taylor <brad@sun.com>
  36  * X11 version: Dave Lemke <lemke@ncd.com>
  37  * xlock version: Boris Putanec <bp@cs.brown.edu>
  38  */
  39
  40 #ifdef STANDALONE
  41 # define DEFAULTS       "*delay:  17000 \n"     \
  42                                         "*count:  -20 \n"               \
  43                                         "*cycles:  10 \n"               \
  44                                         "*size:   -3 \n"                \
  45                                         "*ncolors: 150 \n"              \
  46                                         "*use3d:   False \n"    \
  47                                         "*delta3d: 1.5 \n"              \
  48                                         "*right3d: red \n"              \
  49                                         "*left3d:  blue \n"             \
  50                                         "*both3d:  magenta \n"  \
  51                                         "*none3d:  black \n" \
  52                                         "*fpsSolid:  true \n" \
  53
  54 # define SMOOTH_COLORS
  55 # define release_worm 0
  56 # define reshape_worm 0
  57 # define worm_handle_event 0
  58 # include "xlockmore.h"         /* in xscreensaver distribution */
  59 #else /* STANDALONE */
  60 # include "xlock.h"             /* in xlockmore distribution */
  61 #endif /* STANDALONE */
  62
  63 ENTRYPOINT ModeSpecOpt worm_opts =
  64 {0, NULL, 0, NULL, NULL};
  65
  66 #define MINSIZE 1
  67
  68 #define SEGMENTS  36
  69 #define MINWORMS 1
  70
  71 #define MAXZ      750
  72 #define MINZ      100
  73 #define SCREENZ   200
  74 #define GETZDIFF(z) (MI_DELTA3D(mi)*20.0*(1.0-(SCREENZ)/((float)(z)+MINZ)))
  75 #define IRINT(x) ((int)(((x)>0.0)?(x)+0.5:(x)-0.5))
  76
  77 /* How many segments to draw per cycle when redrawing */
  78 #define REDRAWSTEP 3
  79
  80 typedef struct {
  81         XPoint     *circ;
  82         int        *diffcirc;
  83         int         dir, dir2;
  84         int         tail;
  85         int         x, y, z;
  86         int         redrawing, redrawpos;
  87 } wormstuff;
  88
  89 typedef struct {
  90         int         xsize, ysize, zsize;
  91         int         wormlength;
  92         int         nc;
  93         int         nw;
  94         int         circsize;
  95         wormstuff  *worm;
  96         XRectangle *rects;      /* [NUMCOLORS * batchcount/NUMCOLORS+1] */
  97         int         maxsize;
  98         int        *size;
  99         unsigned int chromo;
 100 } wormstruct;
 101
 102 static float sintab[SEGMENTS];
 103 static float costab[SEGMENTS];
 104 static int  init_table = 0;
 105
 106 static wormstruct *worms = NULL;
 107
 108 static void
 109 worm_doit(ModeInfo * mi, int which, unsigned long color)
 110 {
 111         Display    *display = MI_DISPLAY(mi);
 112         Window      window = MI_WINDOW(mi);
 113         GC          gc = MI_GC(mi);
 114         wormstruct *wp = &worms[MI_SCREEN(mi)];
 115         wormstuff  *ws = &wp->worm[which];
 116         int         x, y, z;
 117         int         diff;
 118
 119         ws->tail++;
 120         if (ws->tail == wp->wormlength)
 121                 ws->tail = 0;
 122
 123         x = ws->circ[ws->tail].x;
 124         y = ws->circ[ws->tail].y;
 125
 126         if (MI_WIN_IS_USE3D(mi)) {
 127                 diff = ws->diffcirc[ws->tail];
 128                 if (MI_WIN_IS_INSTALL(mi)) {
 129                         XSetForeground(display, gc, MI_NONE_COLOR(mi));
 130                         XFillRectangle(display, window, gc, x - diff, y,
 131                                        wp->circsize, wp->circsize);
 132                         XFillRectangle(display, window, gc, x + diff, y,
 133                                        wp->circsize, wp->circsize);
 134                 } else {
 135                         XClearArea(display, window, x - diff, y,
 136                                    wp->circsize, wp->circsize, False);
 137                         XClearArea(display, window, x + diff, y,
 138                                    wp->circsize, wp->circsize, False);
 139                 }
 140         } else
 141                 XClearArea(display, window, x, y, wp->circsize, wp->circsize, False);
 142
 143         if (LRAND() & 1)
 144                 ws->dir = (ws->dir + 1) % SEGMENTS;
 145         else
 146                 ws->dir = (ws->dir + SEGMENTS - 1) % SEGMENTS;
 147
 148         x = (ws->x + IRINT((float) wp->circsize * costab[ws->dir]) +
 149              wp->xsize) % wp->xsize;
 150         y = (ws->y + IRINT((float) wp->circsize * sintab[ws->dir]) +
 151              wp->ysize) % wp->ysize;
 152
 153         ws->circ[ws->tail].x = x;
 154         ws->circ[ws->tail].y = y;
 155         ws->x = x;
 156         ws->y = y;
 157
 158         if (MI_WIN_IS_USE3D(mi)) {
 159                 if (LRAND() & 1)
 160                         ws->dir2 = (ws->dir2 + 1) % SEGMENTS;
 161                 else
 162                         ws->dir2 = (ws->dir2 + SEGMENTS - 1) % SEGMENTS;
 163                 /* for the z-axis the wrap-around looks bad, so worms should just turn around. */
 164                 z = (int) (ws->z + wp->circsize * sintab[ws->dir2]);
 165                 if (z < 0 || z >= wp->zsize)
 166                         z = (int) (ws->z - wp->circsize * sintab[ws->dir2]);
 167
 168                 diff = (int) (GETZDIFF(z) + 0.5);       /* ROUND */
 169                 ws->diffcirc[ws->tail] = diff;
 170
 171                 ws->z = z;
 172
 173                 /* right eye */
 174                 color = 0;
 175                 wp->rects[color * wp->maxsize + wp->size[color]].x = x + diff;
 176                 wp->rects[color * wp->maxsize + wp->size[color]].y = y;
 177                 wp->size[color]++;
 178
 179                 /* left eye */
 180                 color = 1;
 181                 wp->rects[color * wp->maxsize + wp->size[color]].x = x - diff;
 182                 wp->rects[color * wp->maxsize + wp->size[color]].y = y;
 183                 wp->size[color]++;
 184
 185 #if 0
 186                 if (ws->redrawing) {    /* Too hard for now */
 187                         int         j;
 188
 189                         for (j = 0; j < REDRAWSTEP; j++) {
 190                                 int         k = (ws->tail - ws->redrawpos + wp->wormlength)
 191                                 % wp->wormlength;
 192
 193                                 color = 0;
 194                                 wp->rects[color * wp->maxsize + wp->size[color]].x =
 195                                         ws->circ[k].x + ws->diffcirc[k];
 196                                 wp->rects[color * wp->maxsize + wp->size[color]].y =
 197                                         ws->circ[k].y;
 198                                 wp->size[color]++;
 199
 200                                 color = 1;
 201                                 wp->rects[color * wp->maxsize + wp->size[color]].x =
 202                                         ws->circ[k].x - ws->diffcirc[k];
 203                                 wp->rects[color * wp->maxsize + wp->size[color]].y =
 204                                         ws->circ[k].y;
 205                                 wp->size[color]++;
 206
 207                                 if (++(ws->redrawpos) >= wp->wormlength) {
 208                                         ws->redrawing = 0;
 209                                         break;
 210                                 }
 211                         }
 212                 }
 213 #endif
 214
 215         } else {
 216
 217                 wp->rects[color * wp->maxsize + wp->size[color]].x = x;
 218                 wp->rects[color * wp->maxsize + wp->size[color]].y = y;
 219                 wp->size[color]++;
 220                 if (ws->redrawing) {
 221                         int         j;
 222
 223                         ws->redrawpos++;
 224                         /* Compensates for the changed ws->tail
 225                            since the last callback. */
 226
 227                         for (j = 0; j < REDRAWSTEP; j++) {
 228                                 int         k = (ws->tail - ws->redrawpos + wp->wormlength)
 229                                 % wp->wormlength;
 230
 231                                 wp->rects[color * wp->maxsize + wp->size[color]].x = ws->circ[k].x;
 232                                 wp->rects[color * wp->maxsize + wp->size[color]].y = ws->circ[k].y;
 233                                 wp->size[color]++;
 234
 235                                 if (++(ws->redrawpos) >= wp->wormlength) {
 236                                         ws->redrawing = 0;
 237                                         break;
 238                                 }
 239                         }
 240                 }
 241         }
 242 }
 243
 244 ENTRYPOINT void
 245 free_worm(ModeInfo * mi)
 246 {
 247         wormstruct *wp;
 248         int         wn;
 249
 250         if(!worms)
 251                 return;
 252         wp = &worms[MI_SCREEN(mi)];
 253
 254         if (wp->worm) {
 255                 for (wn = 0; wn < wp->nw; wn++) {
 256                         if (wp->worm[wn].circ)
 257                                 (void) free((void *) wp->worm[wn].circ);
 258                         if (wp->worm[wn].diffcirc)
 259                                 (void) free((void *) wp->worm[wn].diffcirc);
 260                 }
 261                 (void) free((void *) wp->worm);
 262                 wp->worm = NULL;
 263         }
 264         if (wp->rects) {
 265                 (void) free((void *) wp->rects);
 266                 wp->rects = NULL;
 267         }
 268         if (wp->size) {
 269                 (void) free((void *) wp->size);
 270                 wp->size = NULL;
 271         }
 272 }
 273
 274 ENTRYPOINT void
 275 init_worm (ModeInfo * mi)
 276 {
 277         wormstruct *wp;
 278         int         size = MI_SIZE(mi);
 279         int         i, j;
 280
 281         MI_INIT (mi, worms);
 282         wp = &worms[MI_SCREEN(mi)];
 283         if (MI_NPIXELS(mi) <= 2 || MI_WIN_IS_USE3D(mi))
 284                 wp->nc = 2;
 285         else
 286                 wp->nc = MI_NPIXELS(mi);
 287         if (wp->nc > NUMCOLORS)
 288                 wp->nc = NUMCOLORS;
 289
 290         free_worm(mi);
 291         wp->nw = MI_BATCHCOUNT(mi);
 292         if (wp->nw < -MINWORMS)
 293                 wp->nw = NRAND(-wp->nw - MINWORMS + 1) + MINWORMS;
 294         else if (wp->nw < MINWORMS)
 295                 wp->nw = MINWORMS;
 296         if (!wp->worm)
 297                 wp->worm = (wormstuff *) malloc(wp->nw * sizeof (wormstuff));
 298
 299         if (!wp->size)
 300                 wp->size = (int *) malloc(NUMCOLORS * sizeof (int));
 301
 302         wp->maxsize = (REDRAWSTEP + 1) * wp->nw;        /*  / wp->nc + 1; */
 303         if (!wp->rects)
 304                 wp->rects =
 305                         (XRectangle *) malloc(wp->maxsize * NUMCOLORS * sizeof (XRectangle));
 306
 307
 308         if (!init_table) {
 309                 init_table = 1;
 310                 for (i = 0; i < SEGMENTS; i++) {
 311                         sintab[i] = SINF(i * 2.0 * M_PI / SEGMENTS);
 312                         costab[i] = COSF(i * 2.0 * M_PI / SEGMENTS);
 313                 }
 314         }
 315         wp->xsize = MI_WIN_WIDTH(mi);
 316         wp->ysize = MI_WIN_HEIGHT(mi);
 317         wp->zsize = MAXZ - MINZ + 1;
 318         if (MI_NPIXELS(mi) > 2)
 319                 wp->chromo = NRAND(MI_NPIXELS(mi));
 320
 321         if (size < -MINSIZE)
 322                 wp->circsize = NRAND(-size - MINSIZE + 1) + MINSIZE;
 323         else if (size < MINSIZE)
 324                 wp->circsize = MINSIZE;
 325         else
 326                 wp->circsize = size;
 327
 328         for (i = 0; i < wp->nc; i++) {
 329                 for (j = 0; j < wp->maxsize; j++) {
 330                         wp->rects[i * wp->maxsize + j].width = wp->circsize;
 331                         wp->rects[i * wp->maxsize + j].height = wp->circsize;
 332
 333                 }
 334         }
 335         (void) memset((char *) wp->size, 0, wp->nc * sizeof (int));
 336
 337         wp->wormlength = (int) sqrt(wp->xsize + wp->ysize) *
 338                 MI_CYCLES(mi) / 8;      /* Fudge this to something reasonable */
 339         for (i = 0; i < wp->nw; i++) {
 340                 wp->worm[i].circ = (XPoint *) malloc(wp->wormlength * sizeof (XPoint));
 341                 wp->worm[i].diffcirc = (int *) malloc(wp->wormlength * sizeof (int));
 342
 343                 for (j = 0; j < wp->wormlength; j++) {
 344                         wp->worm[i].circ[j].x = wp->xsize / 2;
 345                         wp->worm[i].circ[j].y = wp->ysize / 2;
 346                         if (MI_WIN_IS_USE3D(mi))
 347                                 wp->worm[i].diffcirc[j] = 0;
 348                 }
 349                 wp->worm[i].dir = NRAND(SEGMENTS);
 350                 wp->worm[i].dir2 = NRAND(SEGMENTS);
 351                 wp->worm[i].tail = 0;
 352                 wp->worm[i].x = wp->xsize / 2;
 353                 wp->worm[i].y = wp->ysize / 2;
 354                 wp->worm[i].z = SCREENZ - MINZ;
 355                 wp->worm[i].redrawing = 0;
 356         }
 357
 358         if (MI_WIN_IS_INSTALL(mi) && MI_WIN_IS_USE3D(mi)) {
 359                 XSetForeground(MI_DISPLAY(mi), MI_GC(mi), MI_NONE_COLOR(mi));
 360                 XFillRectangle(MI_DISPLAY(mi), MI_WINDOW(mi), MI_GC(mi),
 361                                0, 0, wp->xsize, wp->ysize);
 362         } else
 363                 XClearWindow(MI_DISPLAY(mi), MI_WINDOW(mi));
 364 }
 365
 366 ENTRYPOINT void
 367 draw_worm (ModeInfo * mi)
 368 {
 369         Display    *display = MI_DISPLAY(mi);
 370         Window      window = MI_WINDOW(mi);
 371         GC          gc = MI_GC(mi);
 372         wormstruct *wp = &worms[MI_SCREEN(mi)];
 373         unsigned long wcolor;
 374         int         i;
 375
 376         (void) memset((char *) wp->size, 0, wp->nc * sizeof (int));
 377
 378         for (i = 0; i < wp->nw; i++) {
 379                 if (MI_NPIXELS(mi) > 2) {
 380                         wcolor = (i + wp->chromo) % wp->nc;
 381
 382                         worm_doit(mi, i, wcolor);
 383                 } else
 384                         worm_doit(mi, i, (unsigned long) 0);
 385         }
 386
 387         if (MI_WIN_IS_USE3D(mi)) {
 388                 if (MI_WIN_IS_INSTALL(mi))
 389                         XSetFunction(display, gc, GXor);
 390                 XSetForeground(display, gc, MI_RIGHT_COLOR(mi));
 391                 XFillRectangles(display, window, gc, &(wp->rects[0]), wp->size[0]);
 392
 393                 XSetForeground(display, gc, MI_LEFT_COLOR(mi));
 394                 XFillRectangles(display, window, gc, &(wp->rects[wp->maxsize]), wp->size[1]);
 395                 if (MI_WIN_IS_INSTALL(mi))
 396                         XSetFunction(display, gc, GXcopy);
 397         } else if (MI_NPIXELS(mi) > 2) {
 398                 for (i = 0; i < wp->nc; i++) {
 399                         XSetForeground(display, gc, MI_PIXEL(mi, i));
 400                         XFillRectangles(display, window, gc, &(wp->rects[i * wp->maxsize]), wp->size[i]);
 401                 }
 402         } else {
 403                 XSetForeground(display, gc, MI_WIN_WHITE_PIXEL(mi));
 404                 XFillRectangles(display, window, gc,
 405                                 &(wp->rects[0]), wp->size[0]);
 406         }
 407
 408         if (++wp->chromo == (unsigned long) wp->nc)
 409                 wp->chromo = 0;
 410 }
 411
 412 #ifndef STANDALONE
 413 ENTRYPOINT void
 414 refresh_worm (ModeInfo * mi)
 415 {
 416         if (MI_WIN_IS_USE3D(mi))
 417                 /* The 3D code does drawing&clearing by XORing.  We do not
 418                    want to go to too much trouble here to make it redraw
 419                    correctly. */
 420                 XClearWindow(MI_DISPLAY(mi), MI_WINDOW(mi));
 421         else if (worms != NULL) {
 422                 wormstruct *wp = &worms[MI_SCREEN(mi)];
 423                 int         i;
 424
 425                 for (i = 0; i < wp->nw; i++) {
 426                         wp->worm[i].redrawing = 1;
 427                         wp->worm[i].redrawpos = 0;
 428                 }
 429         }
 430 }
 431 #endif
 432
 433 XSCREENSAVER_MODULE ("Worm", worm)