+/****************************************************************************
+ * Copyright 2018-2019,2020 Thomas E. Dickey *
+ * Copyright 1998-2014,2017 Free Software Foundation, Inc. *
+ * *
+ * 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. *
+ ****************************************************************************/
/*
- * $Id: rain.c,v 1.15 2000/09/02 18:41:22 tom Exp $
+ * $Id: rain.c,v 1.51 2020/02/02 23:34:34 tom Exp $
*/
#include <test.priv.h>
+#include <popup_msg.h>
-#include <term.h> /* for tparm() */
+/* rain 11/3/1980 EPS/CITHEP */
-#include <signal.h>
+#ifdef USE_PTHREADS
+#include <pthread.h>
+#endif
-/* rain 11/3/1980 EPS/CITHEP */
+WANT_USE_WINDOW();
+
+#define MAX_THREADS 10
+#define MAX_DROP 5
+
+struct DATA;
+
+typedef void (*DrawPart) (struct DATA *);
+
+typedef struct DATA {
+ int y, x;
+#ifdef USE_PTHREADS
+ DrawPart func;
+ int state;
+#endif
+} DATA;
+
+#ifdef USE_PTHREADS
+pthread_cond_t cond_next_drop;
+pthread_mutex_t mutex_next_drop;
+static int used_threads;
+
+typedef struct {
+ pthread_t myself;
+ long counter;
+} STATS;
+
+static STATS drop_threads[MAX_THREADS];
+#endif
+
+#if HAVE_USE_WINDOW
+static int
+safe_wgetch(WINDOW *w, void *data GCC_UNUSED)
+{
+ return wgetch(w);
+}
+#endif
+
+static void
+onsig(int n GCC_UNUSED)
+{
+ stop_curses();
+ ExitProgram(EXIT_FAILURE);
+}
+
+static double
+ranf(void)
+{
+ long r = (rand() & 077777);
+ return ((double) r / 32768.);
+}
+
+static int
+random_x(void)
+{
+ return (int) (((double) (COLS - 4) * ranf()) + 2);
+}
-static float ranf(void);
-static void onsig(int sig);
+static int
+random_y(void)
+{
+ return (int) (((double) (LINES - 4) * ranf()) + 2);
+}
-static int next_j(int j)
+static int
+next_j(int j)
{
- if (j==0) j=4; else --j;
+ if (j == 0)
+ j = MAX_DROP - 1;
+ else
+ --j;
if (has_colors()) {
- int z = (int)(3*ranf());
- chtype color = COLOR_PAIR(z);
- if (z)
- color |= A_BOLD;
- attrset(color);
+ int z = (int) (3 * ranf());
+ (void) attrset(AttrArg(COLOR_PAIR(z), (z ? A_BOLD : A_NORMAL)));
}
return j;
}
+static void
+part1(DATA * drop)
+{
+ MvAddCh(drop->y, drop->x, '.');
+}
+
+static void
+part2(DATA * drop)
+{
+ MvAddCh(drop->y, drop->x, 'o');
+}
+
+static void
+part3(DATA * drop)
+{
+ MvAddCh(drop->y, drop->x, 'O');
+}
+
+static void
+part4(DATA * drop)
+{
+ MvAddCh(drop->y - 1, drop->x, '-');
+ MvAddStr(drop->y, drop->x - 1, "|.|");
+ MvAddCh(drop->y + 1, drop->x, '-');
+}
+
+static void
+part5(DATA * drop)
+{
+ MvAddCh(drop->y - 2, drop->x, '-');
+ MvAddStr(drop->y - 1, drop->x - 1, "/ \\");
+ MvAddStr(drop->y, drop->x - 2, "| O |");
+ MvAddStr(drop->y + 1, drop->x - 1, "\\ /");
+ MvAddCh(drop->y + 2, drop->x, '-');
+}
+
+static void
+part6(DATA * drop)
+{
+ MvAddCh(drop->y - 2, drop->x, ' ');
+ MvAddStr(drop->y - 1, drop->x - 1, " ");
+ MvAddStr(drop->y, drop->x - 2, " ");
+ MvAddStr(drop->y + 1, drop->x - 1, " ");
+ MvAddCh(drop->y + 2, drop->x, ' ');
+}
+
+#ifdef USE_PTHREADS
+static void
+napsome(void)
+{
+ napms(60);
+}
+
+/*
+ * This runs inside the use_window() mutex.
+ */
+static int
+really_draw(WINDOW *win, void *arg)
+{
+ DATA *data = (DATA *) arg;
+
+ (void) win;
+ next_j(data->state);
+ data->func(data);
+ refresh();
+ return OK;
+}
+
+static void
+draw_part(void (*func) (DATA *), int state, DATA * data)
+{
+ data->func = func;
+ data->state = state;
+ use_window(stdscr, really_draw, (void *) data);
+ napsome();
+}
+
+/*
+ * Tell the threads that one of them can start work on a new raindrop.
+ * They may all be busy if we're sending requests too rapidly.
+ */
+static int
+put_next_drop(void)
+{
+ pthread_cond_signal(&cond_next_drop);
+ pthread_mutex_unlock(&mutex_next_drop);
+
+ return 0;
+}
+
+/*
+ * Wait until we're assigned the task of drawing a new raindrop.
+ */
+static int
+get_next_drop(void)
+{
+ pthread_mutex_lock(&mutex_next_drop);
+ pthread_cond_wait(&cond_next_drop, &mutex_next_drop);
+
+ return TRUE;
+}
+
+static void *
+draw_drop(void *arg)
+{
+ DATA mydata;
+ int mystats;
+
+ /*
+ * Find myself in the list of threads so we can count the number of loops.
+ */
+ for (mystats = 0; mystats < MAX_THREADS; ++mystats) {
+#if defined(_WIN32) && !defined(__WINPTHREADS_VERSION)
+ if (drop_threads[mystats].myself.p == pthread_self().p)
+#else
+ if (drop_threads[mystats].myself == pthread_self())
+#endif
+ break;
+ }
+
+ do {
+ if (mystats < MAX_THREADS)
+ drop_threads[mystats].counter++;
+
+ /*
+ * Make a copy of caller's data. We're cheating for the cases after
+ * the first loop since we still have a pointer into the main thread
+ * to the data which it uses for setting up this thread (but it has
+ * been modified to use different coordinates).
+ */
+ mydata = *(DATA *) arg;
+
+ draw_part(part1, 0, &mydata);
+ draw_part(part2, 1, &mydata);
+ draw_part(part3, 2, &mydata);
+ draw_part(part4, 3, &mydata);
+ draw_part(part5, 4, &mydata);
+ draw_part(part6, 0, &mydata);
+ } while (get_next_drop());
+
+ return NULL;
+}
+
+/*
+ * The description of pthread_create() is misleading, since it implies that
+ * threads will exit cleanly after their function returns.
+ *
+ * Since they do not (and the number of threads is limited by system
+ * resources), make a limited number of threads, and signal any that are
+ * waiting when we want a thread past that limit.
+ */
+static int
+start_drop(DATA * data)
+{
+ int rc;
+
+ if (!used_threads) {
+ /* mutex and condition for signalling thread */
+ pthread_mutex_init(&mutex_next_drop, NULL);
+ pthread_cond_init(&cond_next_drop, NULL);
+ }
+
+ if (used_threads < MAX_THREADS) {
+ rc = pthread_create(&(drop_threads[used_threads].myself),
+ NULL,
+ draw_drop,
+ data);
+ ++used_threads;
+ } else {
+ rc = put_next_drop();
+ }
+ return rc;
+}
+#endif
+
+static int
+get_input(void)
+{
+ return USING_WINDOW1(stdscr, wgetch, safe_wgetch);
+}
+
+static void
+usage(void)
+{
+ static const char *msg[] =
+ {
+ "Usage: rain [options]"
+ ,""
+ ,"Options:"
+#if HAVE_USE_DEFAULT_COLORS
+ ," -d invoke use_default_colors"
+#endif
+ };
+ size_t n;
+
+ for (n = 0; n < SIZEOF(msg); n++)
+ fprintf(stderr, "%s\n", msg[n]);
+
+ ExitProgram(EXIT_FAILURE);
+}
+
int
-main(
- int argc GCC_UNUSED,
- char *argv[] GCC_UNUSED)
+main(int argc, char *argv[])
{
-int x, y, j;
-static int xpos[5], ypos[5];
-float r;
-float c;
+ static const char *help[] =
+ {
+ "Commands:",
+ " q/Q exit the program",
+ " s do single-step",
+ " <space> undo single-step",
+ "",
+ 0
+ };
+
+ bool done = FALSE;
+ DATA drop;
+#ifndef USE_PTHREADS
+ DATA last[MAX_DROP];
+#endif
+ int j = 0;
+ int ch;
+#if HAVE_USE_DEFAULT_COLORS
+ bool d_option = FALSE;
+#endif
- for (j=SIGHUP;j<=SIGTERM;j++)
- if (signal(j,SIG_IGN)!=SIG_IGN) signal(j,onsig);
+ while ((ch = getopt(argc, argv, "d")) != -1) {
+ switch (ch) {
+#if HAVE_USE_DEFAULT_COLORS
+ case 'd':
+ d_option = TRUE;
+ break;
+#endif
+ default:
+ usage();
+ /* NOTREACHED */
+ }
+ }
+ if (optind < argc)
+ usage();
+
+ setlocale(LC_ALL, "");
- initscr();
+ InitAndCatch(initscr(), onsig);
if (has_colors()) {
int bg = COLOR_BLACK;
start_color();
#if HAVE_USE_DEFAULT_COLORS
- if (use_default_colors() == OK)
- bg = -1;
+ if (d_option && (use_default_colors() == OK))
+ bg = -1;
#endif
- init_pair(1, COLOR_BLUE, bg);
- init_pair(2, COLOR_CYAN, bg);
+ init_pair(1, COLOR_BLUE, (short) bg);
+ init_pair(2, COLOR_CYAN, (short) bg);
}
nl();
noecho();
curs_set(0);
timeout(0);
- r = (float)(LINES - 4);
- c = (float)(COLS - 4);
- for (j=5;--j>=0;) {
- xpos[j]=(int)(c* ranf())+2;
- ypos[j]=(int)(r* ranf())+2;
+#ifndef USE_PTHREADS
+ for (j = MAX_DROP; --j >= 0;) {
+ last[j].x = random_x();
+ last[j].y = random_y();
}
+ j = 0;
+#endif
- for (j=0;;) {
- x=(int)(c*ranf())+2;
- y=(int)(r*ranf())+2;
+ while (!done) {
+ drop.x = random_x();
+ drop.y = random_y();
- mvaddch(y,x, '.');
+#ifdef USE_PTHREADS
+ if (start_drop(&drop) != 0) {
+ beep();
+ }
+#else
+ /*
+ * The non-threaded code draws parts of each drop on each loop.
+ */
+ part1(&drop);
- mvaddch(ypos[j], xpos[j], 'o');
+ part2(&last[j]);
j = next_j(j);
- mvaddch(ypos[j], xpos[j], 'O');
+ part3(&last[j]);
j = next_j(j);
- mvaddch( ypos[j]-1, xpos[j], '-');
- mvaddstr(ypos[j], xpos[j]-1, "|.|");
- mvaddch( ypos[j]+1, xpos[j], '-');
+ part4(&last[j]);
j = next_j(j);
- mvaddch( ypos[j]-2, xpos[j], '-');
- mvaddstr(ypos[j]-1, xpos[j]-1, "/ \\");
- mvaddstr(ypos[j], xpos[j]-2, "| O |");
- mvaddstr(ypos[j]+1, xpos[j]-1, "\\ /");
- mvaddch( ypos[j]+2, xpos[j], '-');
+ part5(&last[j]);
j = next_j(j);
- mvaddch( ypos[j]-2, xpos[j], ' ');
- mvaddstr(ypos[j]-1, xpos[j]-1, " ");
- mvaddstr(ypos[j], xpos[j]-2, " ");
- mvaddstr(ypos[j]+1, xpos[j]-1, " ");
- mvaddch( ypos[j]+2, xpos[j], ' ');
-
- xpos[j] = x; ypos[j] = y;
-
- switch(getch())
- {
- case('q'):
- case('Q'):
- curs_set(1);
- endwin();
- return(EXIT_SUCCESS);
+ part6(&last[j]);
+
+ last[j] = drop;
+#endif
+
+ switch (get_input()) {
+ case ('q'):
+ case ('Q'):
+ done = TRUE;
+ break;
case 's':
nodelay(stdscr, FALSE);
break;
nodelay(stdscr, TRUE);
break;
#ifdef KEY_RESIZE
- case(KEY_RESIZE):
- r = (float)(LINES - 4);
- c = (float)(COLS - 4);
+ case (KEY_RESIZE):
break;
#endif
+ case HELP_KEY_1:
+ popup_msg(stdscr, help);
+ break;
+ case ERR:
+ break;
+ default:
+ beep();
}
napms(50);
}
-}
-
-static void
-onsig(int n GCC_UNUSED)
-{
- curs_set(1);
- endwin();
- exit(EXIT_FAILURE);
-}
-
-static float
-ranf(void)
-{
- long r = (rand() & 077777);
- return ((float) r / 32768.);
+ stop_curses();
+#ifdef USE_PTHREADS
+ printf("Counts per thread:\n");
+ for (j = 0; j < MAX_THREADS; ++j)
+ printf(" %d:%ld\n", j, drop_threads[j].counter);
+#endif
+ ExitProgram(EXIT_SUCCESS);
}
projects / ncurses.git / blobdiff