File size: 1,848 Bytes
1ce325b |
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 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
/* Progress bar suitable for chains of workers */
#ifndef UTIL_STREAM_MULTI_PROGRESS_H
#define UTIL_STREAM_MULTI_PROGRESS_H
#include <boost/thread/mutex.hpp>
#include <cstddef>
#include <stdint.h>
namespace util { namespace stream {
class WorkerProgress;
class MultiProgress {
public:
static const unsigned char kWidth = 100;
MultiProgress();
~MultiProgress();
// Turns on showing (requires SetTarget too).
void Activate();
void SetTarget(uint64_t complete);
WorkerProgress Add();
void Finished();
private:
friend class WorkerProgress;
void Milestone(WorkerProgress &worker);
bool active_;
uint64_t complete_;
boost::mutex mutex_;
// \0 at the end.
char display_[kWidth + 1];
std::size_t character_handout_;
MultiProgress(const MultiProgress &);
MultiProgress &operator=(const MultiProgress &);
};
class WorkerProgress {
public:
// Default contrutor must be initialized with operator= later.
WorkerProgress() : parent_(NULL) {}
// Not threadsafe for the same worker by default.
WorkerProgress &operator++() {
if (++current_ >= next_) {
parent_->Milestone(*this);
}
return *this;
}
WorkerProgress &operator+=(uint64_t amount) {
current_ += amount;
if (current_ >= next_) {
parent_->Milestone(*this);
}
return *this;
}
private:
friend class MultiProgress;
WorkerProgress(uint64_t next, MultiProgress &parent, char character)
: current_(0), next_(next), parent_(&parent), stone_(0), character_(character) {}
uint64_t current_, next_;
MultiProgress *parent_;
// Previous milestone reached.
unsigned char stone_;
// Character to display in bar.
char character_;
};
}} // namespaces
#endif // UTIL_STREAM_MULTI_PROGRESS_H
|