00001 #include "util/ersatz_progress.hh"
00002
00003 #include <algorithm>
00004 #include <ostream>
00005 #include <limits>
00006 #include <string>
00007
00008 namespace util {
00009
00010 namespace { const unsigned char kWidth = 100; }
00011
00012 const char kProgressBanner[] = "----5---10---15---20---25---30---35---40---45---50---55---60---65---70---75---80---85---90---95--100\n";
00013
00014 ErsatzProgress::ErsatzProgress() : current_(0), next_(std::numeric_limits<uint64_t>::max()), complete_(next_), out_(NULL) {}
00015
00016 ErsatzProgress::~ErsatzProgress() {
00017 if (out_) Finished();
00018 }
00019
00020 ErsatzProgress::ErsatzProgress(uint64_t complete, std::ostream *to, const std::string &message)
00021 : current_(0), next_(complete / kWidth), complete_(complete), stones_written_(0), out_(to) {
00022 if (!out_) {
00023 next_ = std::numeric_limits<uint64_t>::max();
00024 return;
00025 }
00026 if (!message.empty()) *out_ << message << '\n';
00027 *out_ << kProgressBanner;
00028 }
00029
00030 void ErsatzProgress::Milestone() {
00031 if (!out_) { current_ = 0; return; }
00032 if (!complete_) return;
00033 unsigned char stone = std::min(static_cast<uint64_t>(kWidth), (current_ * kWidth) / complete_);
00034
00035 for (; stones_written_ < stone; ++stones_written_) {
00036 (*out_) << '*';
00037 }
00038 if (stone == kWidth) {
00039 (*out_) << std::endl;
00040 next_ = std::numeric_limits<uint64_t>::max();
00041 out_ = NULL;
00042 } else {
00043 next_ = std::max(next_, ((stone + 1) * complete_ + kWidth - 1) / kWidth);
00044 }
00045 }
00046
00047 }