Implement arithmetic operators for issue #12

moshiba · Jan 18, 2020 · 31b542d · 31b542d
1 parent 8141542
commit 31b542d
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 2 deletions.
diff --git a/pbar/pbar.cpp b/pbar/pbar.cpp
@@ -137,8 +137,7 @@ std::string ProgressBar::format_meter() {
             << aesc::render::reset;
 
     // Inject running-bar
-    int processed =
-        bar_width * this->n / this->total;
+    int processed = bar_width * this->n / this->total;
     int remaining = bar_width - processed;
 
     if (bar_width > 0) {
@@ -266,4 +265,26 @@ void ProgressBar::reset() {
     this->display();
 }
 
+ProgressBar& ProgressBar::operator+=(const int n) {
+    this->update(n);
+    return *this;
+}
+
+ProgressBar& ProgressBar::operator-=(const int n) {
+    this->update(-n);
+    return *this;
+}
+
+// prefix
+ProgressBar& ProgressBar::operator++() {
+    this->update(1);
+    return *this;
+}
+
+// prefix
+ProgressBar& ProgressBar::operator--() {
+    this->update(-1);
+    return *this;
+}
+
 }  // namespace pbar
diff --git a/pbar/pbar.hpp b/pbar/pbar.hpp
@@ -98,6 +98,12 @@ class ProgressBar {
     void update(const int n = 1);
     void close();
     void reset();
+
+    ProgressBar& operator+=(const int n);
+    ProgressBar& operator-=(const int n);
+    // prefix
+    ProgressBar& operator++();
+    ProgressBar& operator--();
 };
 
 }  // namespace pbar