942 lines
15 KiB
C
942 lines
15 KiB
C
/****************************************************************************
|
|
* apps/examples/pdcurses/rain_main.c
|
|
* $Id: worm.c,v 1.16 2008/07/13 16:08:17 wmcbrine Exp $
|
|
*
|
|
* Copyright (C) 2017 Gregory Nutt. All rights reserved.
|
|
* Copyright (c) 2005 Free Software Foundation, Inc.
|
|
* Adapted by: Gregory Nutt <gnutt@nuttx.org>
|
|
*
|
|
* Adapted from the original public domain pdcurses by Gregory Nutt and
|
|
* released as part of NuttX under the origin ncurses license:
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
* copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, distribute with modifications, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included
|
|
* in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
* IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
|
* DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
|
* OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR
|
|
* THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*
|
|
* Except as contained in this notice, the name(s) of the above copyright
|
|
* holders shall not be used in advertising or otherwise to promote the
|
|
* sale, use or other dealings in this Software without prior written
|
|
* authorization.
|
|
*
|
|
****************************************************************************/
|
|
|
|
/*
|
|
* @@@ @@@ @@@@@@@@@@ @@@@@@@@@@@ @@@@@@@@@@@@
|
|
* @@@ @@@ @@@@@@@@@@@@ @@@@@@@@@@@@ @@@@@@@@@@@@@
|
|
* @@@ @@@ @@@@ @@@@ @@@@ @@@@ @@@ @@@@
|
|
* @@@ @@ @@@ @@@ @@@ @@@ @@@ @@@ @@@
|
|
* @@@ @@@@ @@@ @@@ @@@ @@@ @@@ @@@ @@@
|
|
* @@@@ @@@@ @@@@ @@@ @@@ @@@ @@@ @@@ @@@
|
|
* @@@@@@@@@@@@ @@@@ @@@@ @@@ @@@ @@@ @@@
|
|
* @@@@ @@@@ @@@@@@@@@@@@ @@@ @@@ @@@ @@@
|
|
* @@ @@ @@@@@@@@@@ @@@ @@@ @@@ @@@
|
|
*
|
|
* Eric P. Scott
|
|
* Caltech High Energy Physics
|
|
* October, 1980
|
|
*
|
|
* Color by Eric S. Raymond
|
|
* July, 1995
|
|
*
|
|
* Options:
|
|
* -f fill screen with copies of 'WORM' at start.
|
|
* -l <n> set worm length
|
|
* -n <n> set number of worms
|
|
* -t make worms leave droppings
|
|
*/
|
|
|
|
/****************************************************************************
|
|
* Included Files
|
|
****************************************************************************/
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
|
|
#include "graphics/curses.h"
|
|
|
|
/****************************************************************************
|
|
* Pre-processor Definitions
|
|
****************************************************************************/
|
|
|
|
#define FLAVORS 7
|
|
|
|
/****************************************************************************
|
|
* Private Types
|
|
****************************************************************************/
|
|
|
|
struct worm
|
|
{
|
|
int orientation, head;
|
|
short *xpos, *ypos;
|
|
};
|
|
|
|
struct options
|
|
{
|
|
int nopts;
|
|
int opts[3];
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Data
|
|
****************************************************************************/
|
|
|
|
static const chtype init_flavor[FLAVORS] =
|
|
{
|
|
'O', '*', '#', '$', '%', '0', '@'
|
|
};
|
|
|
|
static const short xinc[] =
|
|
{
|
|
1, 1, 1, 0, -1, -1, -1, 0
|
|
};
|
|
|
|
static const short yinc[] =
|
|
{
|
|
-1, 0, 1, 1, 1, 0, -1, -1
|
|
};
|
|
|
|
static chtype flavor[FLAVORS];
|
|
static struct worm worm[40];
|
|
static const char *field;
|
|
static int length = 16;
|
|
static int number = 3;
|
|
static chtype trail = ' ';
|
|
|
|
static const struct options normal[8] =
|
|
{
|
|
{
|
|
3,
|
|
{
|
|
7, 0, 1
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
0, 1, 2
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
1, 2, 3
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
2, 3, 4
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
3, 4, 5
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
4, 5, 6
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
5, 6, 7
|
|
}
|
|
},
|
|
{
|
|
3,
|
|
{
|
|
6, 7, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options upper[8] =
|
|
{
|
|
{
|
|
1,
|
|
{
|
|
1, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
1, 2, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
4, 5, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
5, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
1, 5, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options left[8] =
|
|
{
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
2, 3, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
3, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
3, 7, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
7, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
7, 0, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options right[8] =
|
|
{
|
|
{
|
|
1,
|
|
{
|
|
7, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
3, 7, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
3, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
3, 4, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
6, 7, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options lower[8] =
|
|
{
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
0, 1, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
1, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
1, 5, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
5, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
5, 6, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options upleft[8] =
|
|
{
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
3, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
1, 3, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
1, 0, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options upright[8] =
|
|
{
|
|
{
|
|
2,
|
|
{
|
|
3, 5, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
3, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
5, 0, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options lowleft[8] =
|
|
{
|
|
{
|
|
3,
|
|
{
|
|
7, 0, 1
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
1, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
1, 7, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
7, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
static const struct options lowright[8] =
|
|
{
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
7, 0, 0
|
|
}
|
|
},
|
|
{
|
|
2,
|
|
{
|
|
5, 7, 0
|
|
}
|
|
},
|
|
{
|
|
1,
|
|
{
|
|
5, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
},
|
|
{
|
|
0,
|
|
{
|
|
0, 0, 0
|
|
}
|
|
}
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Functions
|
|
****************************************************************************/
|
|
|
|
static void cleanup(void)
|
|
{
|
|
standend();
|
|
refresh();
|
|
curs_set(1);
|
|
endwin();
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
#ifdef CONFIG_BUILD_KERNEL
|
|
int main(int argc, FAR char *argv[])
|
|
#else
|
|
int worm_main(int argc, char *argv[])
|
|
#endif
|
|
{
|
|
const struct options *op;
|
|
struct worm *w;
|
|
short **ref;
|
|
short *ip;
|
|
int x;
|
|
int y;
|
|
int n;
|
|
int h;
|
|
int last;
|
|
int bottom;
|
|
int seed;
|
|
|
|
for (x = 1; x < argc; x++)
|
|
{
|
|
char *p = argv[x];
|
|
|
|
if (*p == '-')
|
|
{
|
|
p++;
|
|
}
|
|
|
|
switch (*p)
|
|
{
|
|
case 'f':
|
|
field = "WORM";
|
|
break;
|
|
|
|
case 'l':
|
|
if (++x == argc)
|
|
{
|
|
goto usage;
|
|
}
|
|
|
|
if ((length = atoi(argv[x])) < 2 || length > 1024)
|
|
{
|
|
fprintf(stderr, "%s: Invalid length\n", *argv);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
break;
|
|
|
|
case 'n':
|
|
if (++x == argc)
|
|
{
|
|
goto usage;
|
|
}
|
|
|
|
if ((number = atoi(argv[x])) < 1 || number > 40)
|
|
{
|
|
fprintf(stderr, "%s: Invalid number of worms\n", *argv);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
break;
|
|
|
|
case 't':
|
|
trail = '.';
|
|
break;
|
|
|
|
default:
|
|
usage:
|
|
fprintf(stderr, "usage: %s [-field] [-length #] "
|
|
"[-number #] [-trail]\n", *argv);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
memcpy(flavor, init_flavor, FLAVORS * sizeof(chtype));
|
|
|
|
traceon();
|
|
initscr();
|
|
seed = time((time_t *) 0);
|
|
srand(seed);
|
|
|
|
noecho();
|
|
cbreak();
|
|
nonl();
|
|
keypad(stdscr, true);
|
|
|
|
curs_set(0);
|
|
|
|
bottom = LINES - 1;
|
|
last = COLS - 1;
|
|
|
|
#ifdef A_COLOR
|
|
if (has_colors())
|
|
{
|
|
int bg = COLOR_BLACK;
|
|
start_color();
|
|
|
|
# if defined(NCURSES_VERSION) || (defined(PDC_BUILD) && PDC_BUILD > 3000)
|
|
if (use_default_colors() == OK)
|
|
{
|
|
bg = -1;
|
|
}
|
|
# endif
|
|
|
|
# define SET_COLOR(num, fg) \
|
|
init_pair(num + 1, fg, bg); \
|
|
flavor[num] |= COLOR_PAIR(num + 1) | A_BOLD
|
|
|
|
SET_COLOR(0, COLOR_GREEN);
|
|
SET_COLOR(1, COLOR_RED);
|
|
SET_COLOR(2, COLOR_CYAN);
|
|
SET_COLOR(3, COLOR_WHITE);
|
|
SET_COLOR(4, COLOR_MAGENTA);
|
|
SET_COLOR(5, COLOR_BLUE);
|
|
SET_COLOR(6, COLOR_YELLOW);
|
|
}
|
|
#endif
|
|
|
|
ref = malloc(sizeof(short *) * LINES);
|
|
|
|
for (y = 0; y < LINES; y++)
|
|
{
|
|
ref[y] = malloc(sizeof(short) * COLS);
|
|
|
|
for (x = 0; x < COLS; x++)
|
|
{
|
|
ref[y][x] = 0;
|
|
}
|
|
}
|
|
|
|
#ifdef BADCORNER
|
|
/* if addressing the lower right corner doesn't work in your curses */
|
|
|
|
ref[bottom][last] = 1;
|
|
#endif
|
|
|
|
for (n = number, w = &worm[0]; --n >= 0; w++)
|
|
{
|
|
w->orientation = w->head = 0;
|
|
|
|
if ((ip = malloc(sizeof(short) * (length + 1))) == NULL)
|
|
{
|
|
fprintf(stderr, "%s: out of memory\n", *argv);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
w->xpos = ip;
|
|
|
|
for (x = length; --x >= 0;)
|
|
{
|
|
*ip++ = -1;
|
|
}
|
|
|
|
if ((ip = malloc(sizeof(short) * (length + 1))) == NULL)
|
|
{
|
|
fprintf(stderr, "%s: out of memory\n", *argv);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
w->ypos = ip;
|
|
|
|
for (y = length; --y >= 0;)
|
|
{
|
|
*ip++ = -1;
|
|
}
|
|
}
|
|
|
|
if (field)
|
|
{
|
|
const char *p = field;
|
|
|
|
for (y = bottom; --y >= 0;)
|
|
{
|
|
for (x = COLS; --x >= 0;)
|
|
{
|
|
addch((chtype) (*p++));
|
|
|
|
if (!*p)
|
|
{
|
|
p = field;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
napms(12);
|
|
refresh();
|
|
nodelay(stdscr, true);
|
|
|
|
for (;;)
|
|
{
|
|
int ch;
|
|
|
|
if ((ch = getch()) > 0)
|
|
{
|
|
#ifdef KEY_RESIZE
|
|
if (ch == KEY_RESIZE)
|
|
{
|
|
resize_term(0, 0);
|
|
erase();
|
|
|
|
if (last != COLS - 1)
|
|
{
|
|
for (y = 0; y <= bottom; y++)
|
|
{
|
|
ref[y] = realloc(ref[y], sizeof(short) * COLS);
|
|
|
|
for (x = last + 1; x < COLS; x++)
|
|
{
|
|
ref[y][x] = 0;
|
|
}
|
|
}
|
|
|
|
last = COLS - 1;
|
|
}
|
|
|
|
if (bottom != LINES - 1)
|
|
{
|
|
for (y = LINES; y <= bottom; y++)
|
|
{
|
|
free(ref[y]);
|
|
}
|
|
|
|
ref = realloc(ref, sizeof(short *) * LINES);
|
|
|
|
for (y = bottom + 1; y < LINES; y++)
|
|
{
|
|
ref[y] = malloc(sizeof(short) * COLS);
|
|
|
|
for (x = 0; x < COLS; x++)
|
|
{
|
|
ref[y][x] = 0;
|
|
}
|
|
}
|
|
|
|
bottom = LINES - 1;
|
|
}
|
|
}
|
|
|
|
#endif /* KEY_RESIZE */
|
|
|
|
/* Make it simple to put this into single-step mode, or resume normal
|
|
* operation - T. Dickey
|
|
*/
|
|
|
|
if (ch == 'q')
|
|
{
|
|
cleanup();
|
|
return EXIT_SUCCESS;
|
|
}
|
|
else if (ch == 's')
|
|
{
|
|
nodelay(stdscr, false);
|
|
}
|
|
else if (ch == ' ')
|
|
{
|
|
nodelay(stdscr, true);
|
|
}
|
|
}
|
|
|
|
for (n = 0, w = &worm[0]; n < number; n++, w++)
|
|
{
|
|
if ((x = w->xpos[h = w->head]) < 0)
|
|
{
|
|
move(y = w->ypos[h] = bottom, x = w->xpos[h] = 0);
|
|
addch(flavor[n % FLAVORS]);
|
|
ref[y][x]++;
|
|
}
|
|
else
|
|
{
|
|
y = w->ypos[h];
|
|
}
|
|
|
|
if (x > last)
|
|
{
|
|
x = last;
|
|
}
|
|
|
|
if (y > bottom)
|
|
{
|
|
y = bottom;
|
|
}
|
|
|
|
|
|
if (++h == length)
|
|
{
|
|
h = 0;
|
|
}
|
|
|
|
if (w->xpos[w->head = h] >= 0)
|
|
{
|
|
int x1 = w->xpos[h];
|
|
int y1 = w->ypos[h];
|
|
|
|
if (y1 < LINES && x1 < COLS && --ref[y1][x1] == 0)
|
|
{
|
|
move(y1, x1);
|
|
addch(trail);
|
|
}
|
|
}
|
|
|
|
op = &(x == 0 ? (y == 0 ? upleft :
|
|
(y == bottom ? lowleft : left)) :
|
|
(x == last ? (y == 0 ? upright :
|
|
(y == bottom ? lowright : right)) :
|
|
(y == 0 ? upper :
|
|
(y == bottom ? lower : normal))))[w->orientation];
|
|
|
|
switch (op->nopts)
|
|
{
|
|
case 0:
|
|
cleanup();
|
|
return EXIT_SUCCESS;
|
|
|
|
case 1:
|
|
w->orientation = op->opts[0];
|
|
break;
|
|
|
|
default:
|
|
w->orientation = op->opts[rand() % op->nopts];
|
|
break;
|
|
}
|
|
|
|
move(y += yinc[w->orientation], x += xinc[w->orientation]);
|
|
|
|
if (y < 0)
|
|
{
|
|
y = 0;
|
|
}
|
|
|
|
addch(flavor[n % FLAVORS]);
|
|
ref[w->ypos[h] = y][w->xpos[h] = x]++;
|
|
}
|
|
|
|
napms(12);
|
|
refresh();
|
|
}
|
|
}
|