xref: /netbsd-src/external/lgpl3/gmp/dist/tests/spinner.c (revision 1daf83e636cd998f45e5597a8f995a540e2d5b4a)
1 /* A stupid little spinning wheel designed to make it look like useful work
2    is being done.
3 
4 Copyright 1999-2001 Free Software Foundation, Inc.
5 
6 This file is part of the GNU MP Library test suite.
7 
8 The GNU MP Library test suite is free software; you can redistribute it
9 and/or modify it under the terms of the GNU General Public License as
10 published by the Free Software Foundation; either version 3 of the License,
11 or (at your option) any later version.
12 
13 The GNU MP Library test suite is distributed in the hope that it will be
14 useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
16 Public License for more details.
17 
18 You should have received a copy of the GNU General Public License along with
19 the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */
20 
21 #include "config.h"
22 
23 #include <signal.h>
24 #include <stdio.h>
25 #include <stdlib.h>
26 #if HAVE_UNISTD_H
27 #include <unistd.h>     /* for isatty */
28 #endif
29 
30 #include "gmp-impl.h"
31 
32 #include "tests.h"
33 
34 
35 /* "alarm" is not available on mingw32, and the SIGALRM constant is not
36    defined.  Don't bother with a spinner in this case.  */
37 #if ! HAVE_ALARM || ! defined (SIGALRM)
38 #define alarm(n)          abort()
39 #define signal(sig,func)  SIG_ERR
40 #endif
41 
42 
43 /* An application can update this to get a count printed with the spinner.
44    If left at 0, no count is printed. */
45 
46 unsigned long  spinner_count = 0;
47 
48 
49 int  spinner_wanted = -1;  /* -1 uninitialized, 1 wanted, 0 not */
50 int  spinner_tick = 1;     /* 1 ready to print, 0 not */
51 
52 
53 /*ARGSUSED*/
54 RETSIGTYPE
spinner_signal(int signum)55 spinner_signal (int signum)
56 {
57   spinner_tick = 1;
58 
59   if (signal (SIGALRM, spinner_signal) == SIG_ERR)
60     {
61       printf ("spinner_signal(): Oops, cannot reinstall SIGALRM\n");
62       abort ();
63     }
64   alarm (1);
65 }
66 
67 
68 /* Initialize the spinner.
69 
70    This is done the first time spinner() is called, so an application
71    doesn't need to call this directly.
72 
73    The spinner is only wanted if the output is a tty.  */
74 
75 #define SPINNER_WANTED_INIT() \
76   if (spinner_wanted < 0) spinner_init ()
77 
78 void
spinner_init(void)79 spinner_init (void)
80 {
81   spinner_wanted = isatty (STDOUT_FILENO);
82   if (spinner_wanted == -1)
83     abort ();
84 
85   if (!spinner_wanted)
86     return;
87 
88   if (signal (SIGALRM, spinner_signal) == SIG_ERR)
89     {
90       printf ("(no spinner)\r");
91       spinner_tick = 0;
92       return;
93     }
94   alarm (1);
95 
96   /* unbuffered output so the spinner will show up */
97   setbuf (stdout, NULL);
98 }
99 
100 
101 void
spinner(void)102 spinner (void)
103 {
104   static const char  data[] = { '|', '/', '-', '\\' };
105   static int         pos = 0;
106 
107   char  buf[128];
108 
109   SPINNER_WANTED_INIT ();
110 
111   if (spinner_tick)
112     {
113       buf[0] = data[pos];
114       pos = (pos + 1) % numberof (data);
115       spinner_tick = 0;
116 
117       if (spinner_count != 0)
118 	{
119 	  sprintf (buf+1, " %lu\r", spinner_count);
120 	}
121       else
122 	{
123 	  buf[1] = '\r';
124 	  buf[2] = '\0';
125 	}
126       fputs (buf, stdout);
127     }
128 }
129