| Austin Schuh | dace2a6 | 2020-08-18 10:56:48 -0700 | [diff] [blame] | 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 |
| 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 |
| 79 | spinner_init (void) |
| 80 | { |
| 81 | spinner_wanted = isatty (fileno (stdout)); |
| 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 |
| 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 | } |