generator/progress.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

#ifndef PROGRESS_H_A34EF856
#define PROGRESS_H_A34EF856

#include <string>

namespace verbly {
  namespace generator {

    class progress {
      private:
        std::string message;
        int total;
        int cur = 0;
        int lprint = 0;

      public:
        progress(std::string message, int total) : message(message), total(total)
        {
          std::cout << message << "   0%" << std::flush;
        }

        void update(int val)
        {
          if (val <= total)
          {
            cur = val;
          } else {
            cur = total;
          }

          int pp = cur * 100 / total;
          if (pp != lprint)
          {
            lprint = pp;

            std::cout << "\b\b\b\b" << std::right;
            std::cout.width(3);
            std::cout << pp << "%" << std::flush;
          }
        }

        void update()
        {
          update(cur+1);
        }

        ~progress()
        {
          std::cout << "\b\b\b\b100%" << std::endl;
        }
    };

  };
};

#endif /* end of include guard: PROGRESS_H_A34EF856 */