Line data    Source code

       1              : #include "crpropa/ProgressBar.h"
       2              : 
       3              : #include <cstdio>
       4              : #include <iostream>
       5              : 
       6              : namespace crpropa {
       7              : 
       8              : /// Initialize a ProgressBar with [steps] number of steps, updated at [updateSteps] intervalls
       9            5 : ProgressBar::ProgressBar(unsigned long steps, unsigned long updateSteps) :
      10            5 :                 _steps(steps), _currentCount(0), _maxbarLength(10), _updateSteps(
      11            5 :                                 updateSteps), _nextStep(1), _startTime(0) {
      12            5 :         if (_updateSteps > _steps)
      13            3 :                 _updateSteps = _steps;
      14            5 :         arrow.append(">");
      15            5 : }
      16              : 
      17            1 : void ProgressBar::start(const std::string &title) {
      18            1 :         _startTime = time(NULL);
      19            1 :         std::string s = ctime(&_startTime);
      20            1 :         s.erase(s.end() - 1, s.end());
      21            1 :         stringTmpl = "  Started ";
      22              :         stringTmpl.append(s);
      23            1 :         stringTmpl.append(" : [%-10s] %3i%%    %s: %02i:%02i:%02i %s\r");
      24              :         std::cout << title << std::endl;
      25              : 
      26            1 : }
      27              : /// update the progressbar
      28              : /// should be called steps times in a loop
      29          100 : void ProgressBar::update() {
      30          100 :         _currentCount++;
      31          100 :         if (_currentCount == _nextStep || _currentCount == _steps
      32            0 :                         || _currentCount == 1000) {
      33          100 :                                 _nextStep += long(_steps / float(_updateSteps));
      34          100 :                         setPosition(_currentCount);
      35              :                         }
      36          100 : }
      37              : 
      38          100 : void ProgressBar::setPosition(unsigned long position) {
      39          100 :         int percentage = int(100 * (position / float(_steps)));
      40          100 :         time_t currentTime = time(NULL);
      41          100 :         if (position < _steps) {
      42          108 :                 while (arrow.size() <= _maxbarLength * percentage / 100)
      43            9 :                         arrow.insert(0, "=");
      44           99 :                 float tElapsed = currentTime - _startTime;
      45           99 :                 float tToGo = (_steps - position) * tElapsed / position;
      46           99 :                 std::printf(stringTmpl.c_str(), arrow.c_str(), percentage, "Finish in",
      47           99 :                                 int(tToGo / 3600), (int(tToGo) % 3600) / 60,
      48           99 :                                 int(tToGo) % 60, "");
      49           99 :                 fflush(stdout);
      50              :         } else {
      51            1 :                 float tElapsed = currentTime - _startTime;
      52            1 :                 std::string s = " - Finished at ";
      53            1 :                 s.append(ctime(&currentTime));
      54              :                 char fs[255];
      55              :                 std::snprintf(fs, 255, "%c[%d;%dm Finished %c[%dm", 27, 1, 32, 27, 0);
      56            1 :                 std::printf(stringTmpl.c_str(), fs, 100, "Needed",
      57            1 :                                 int(tElapsed / 3600), (int(tElapsed) % 3600) / 60,
      58            1 :                                 int(tElapsed) % 60, s.c_str());
      59              :         }
      60          100 : }
      61              : 
      62              : 
      63              : /// Mark the progressbar with an error
      64            0 : void ProgressBar::setError() {
      65            0 :         time_t currentTime = time(NULL);
      66            0 :         _currentCount++;
      67            0 :         float tElapsed = currentTime - _startTime;
      68            0 :         std::string s = " - Finished at ";
      69            0 :         s.append(ctime(&currentTime));
      70              :         char fs[255];
      71              :         std::snprintf(fs, 255, "%c[%d;%dm  ERROR   %c[%dm", 27, 1, 31, 27, 0);
      72            0 :         std::printf(stringTmpl.c_str(), fs, _currentCount, "Needed",
      73            0 :                         int(tElapsed / 3600), (int(tElapsed) % 3600) / 60,
      74            0 :                         int(tElapsed) % 60, s.c_str());
      75            0 : }
      76              : 
      77              : } // namespace crpropa