PBar.hpp

Go to the documentation of this file.

#ifndef PROGRESS_BAR_PBAR_HPP
#define PROGRESS_BAR_PBAR_HPP
 
#include <iostream>
#include <stdio.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <cstring>
#include <sstream>
#include <iomanip>
#include <array>
 
#include "PBarStyle.hpp"

class ProgressBar
{
   public:

   ProgressBar(std::string style = "DEFAULT", const std::string& customText = "", 
               const std::string& color = "");
   ProgressBar(const std::string& customLeftBorder, const char customCompleteSymbol, 
               const char customNextCompleteSymbol, const char customNotCompleteSymbol,
               const std::string& customRightBorder, const std::string& color, 
               const std::string& customText = "");
   void SetColor(const std::string& color);
   void SetStyle(std::string style, const std::string& color);
   void SetCustomStyle(const std::string& customLeftBorder, const char customCompleteSymbol, 
                       const char customNextCompleteSymbol, const char customNotCompleteSymbol,
                       const std::string& customRightBorder, const std::string& color);
   void SetWidth(const short customWidth);
   void SetText(const std::string& customText);
   void Print(const double progress);
   void Finish();
   void Clear();
   void RePrint();
   void Reset();
   void SetBarStep(const double step);
   void SetPrecision(const short precision);

   template <typename... T>
   void HandleOutput(T... args)
   {
      Clear();
 
      ((std::cout << args << " "), ...);
      std::cout << std::endl;
 
      if (printedProgress > 1.) return;
      RePrint();
   }

   virtual ~ProgressBar();
 
   protected:

   short utf8_strlen (const std::string& str);
   short GetTerminalWidth();
   std::string DtoStr (const double val, const short precision = 2);

   const short EMPTY_SPACE_WIDTH = 4; 

   std::string barText;

   std::string barColor;
   std::string leftBorder;
   std::string rightBorder;
   std::string completeSymbol;
   std::string nextCompleteSymbol;
   std::string notCompleteSymbol;

   short fullWidth;
   short barBodyWidth;

   double barStep = 0.01;
   double printedProgress = -0.01;
   short barPrecision = 0;

   struct winsize terminalWindow;
};
 
#endif /* PROGRESS_BAR__PBAR_HPP */