tests/spinner.c

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