varia.c

Go to the documentation of this file.

/*
 Copyright (C) 2002 M. Marques, A. Castro, A. Rubio, G. Bertsch
 
 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2, or (at your option)
 any later version.
 
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
 
 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 02110-1301, USA.
 
*/
 
#include <config.h>
 
#include <assert.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
 
#ifdef _POSIX_VERSION
#include <sys/ioctl.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/utsname.h>
#include <termios.h>
#endif
 
#include "varia.h"
 
void sysname(char **c) {
#ifdef _POSIX_VERSION
  struct utsname name;
  uname(&name);
  *c = (char *)malloc(sizeof(name.nodename) + sizeof(name.sysname) + 4);
  strcpy(*c, name.nodename);
  strcat(*c, " (");
  strcat(*c, name.sysname);
  strcat(*c, ")");
#else
  *c = (char *)malloc(8);
  strcpy(*c, "unknown");
#endif
}
 
static int foreground_proc(void) {
#ifdef _POSIX_VERSION
  static pid_t pgrp = -1;
  int ctty_pgrp;
 
  if (pgrp == -1)
    pgrp = getpgrp();
 
#ifdef _POSIX_VERSION
  return ((ctty_pgrp = tcgetpgrp(STDOUT_FILENO)) != -1 && ctty_pgrp == pgrp);
#endif
 
#else
  return 0;
#endif
}
 
int getttywidth(void) {
#ifdef _POSIX_VERSION
  struct winsize winsize;
 
  if (ioctl(fileno(stdout), TIOCGWINSZ, &winsize) != -1)
    return (winsize.ws_col ? winsize.ws_col : 80);
  else
#endif
    return (80);
}
 
void foreground_print_progress(const int actual, const int max, const struct timeval *start) {
 
    char buf[512], fmt[64];
 
    int ratio = 100 * actual / max;
    if (ratio < 0) { ratio = 0; }
    if (ratio > 100) {
        fprintf(stderr,
                "Internal warning: progress_bar called with actual %i > max %i\n",
                actual, max);
    }
 
    sprintf(buf, "%d", max);
    int i = strlen(buf);
    if (i < 3)
        i = 3;
    sprintf(fmt, "\r[%%%dd/%%%dd]", i, i);
    sprintf(buf, fmt, actual, max);
    sprintf(buf + strlen(buf), " %3d%%", ratio);
 
    const int barlength = getttywidth() - strlen(buf) - 16;
    if (barlength > 0) {
        i = barlength * ratio / 100;
        sprintf(buf + strlen(buf), "|%.*s%*s|", i,
                "*******************************************************"
                "*******************************************************"
                "*******************************************************"
                "*******************************************************"
                "*******************************************************"
                "*******************************************************"
                "*******************************************************",
                barlength - i, "");
    }
 
    /* time information now */
    struct timeval now;
    (void) gettimeofday(&now, (struct timezone *) 0);
    const double elapsed = (double)(now.tv_sec - start->tv_sec);
 
    if (elapsed <= 0.0 || actual <= 0) {
        sprintf(buf + strlen(buf), "     --:-- ETA");
    } else {
        int remaining = (int) (max / (actual / elapsed) - elapsed);
        if (remaining < 0) { remaining = 0; }
        i = remaining / 3600;
        if (i)
            sprintf(buf + strlen(buf), "%4d:", i);
        else
            sprintf(buf + strlen(buf), "     ");
        i = remaining % 3600;
        sprintf(buf + strlen(buf), "%02d:%02d%s", i / 60, i % 60, " ETA");
    }
    printf("%s", buf);
}
 
void progress_bar(int actual, int max) {
    static struct timeval start;
    static int show_bar;
 
    assert(max > 0 && "progress_bar: max must be > 0");
 
    // Initialise on first call
    if (actual < 0) {
        show_bar = isatty(STDOUT_FILENO);
        if (!show_bar) { return; }
        (void) gettimeofday(&start, (struct timezone *) 0);
        actual = 0;
    }
 
    if (!show_bar) { return; }
 
    if (foreground_proc() == 1) {
        foreground_print_progress(actual, max, &start);
    }
 
    fflush(stdout);
}
 
#undef timersub