/home/mhx/git/github/libmoost/include/moost/progress/policy/spinner.hpp

Go to the documentation of this file.

/* vim:set ts=3 sw=3 sts=3 et: */
#ifndef MOOST_UTILS_PROGRESS_POLICY_SPINNER_HPP__
#define MOOST_UTILS_PROGRESS_POLICY_SPINNER_HPP__

#include <vector>
#include <iostream>
#include <iomanip>

#include <boost/date_time/posix_time/posix_time_types.hpp>
#include <boost/thread/thread_time.hpp>

namespace moost { namespace progress { namespace policy {

   class spinner
   {
   public:
      typedef size_t counter_type;

      spinner(size_t msecs = 500, std::ostream & out = std::cout)
         : counter_(0), msecs_(msecs), out_(out), ptime_(boost::get_system_time())
      {
         frames_.push_back('|');
         frames_.push_back('/');
         frames_.push_back('-');
         frames_.push_back('\\');

         frame_ = frames_.begin();
      }

      counter_type  operator+=(counter_type const incr)
      {
         boost::posix_time::ptime now = boost::get_system_time();

         if(now >= ptime_)
         {
            animate();
            ptime_ = now + boost::posix_time::milliseconds(msecs_);
         }

         return counter_ += incr;
      }

      counter_type operator ++ ()
      {
         return *this += 1;
      }

      counter_type operator ++ (int)
      {
         counter_type const tmp = counter_;
         ++*this;
         return tmp;
      }

      counter_type count() const
      {
         return counter_;
      }

      counter_type expected_count() const
      {
         return counter_;
      }

   private:
      void animate()
      {
         out_ << *frame_++ << '\r' << std::flush;
         if(frame_ == frames_.end())
         {
            frame_ = frames_.begin();
         }
      }

      std::vector<char> frames_;
      std::vector<char>::const_iterator frame_;
      counter_type counter_;
      size_t const msecs_; // the min number of milliseconds before we'll update spinner
      std::ostream & out_;
      boost::posix_time::ptime ptime_;
   };

}}}

#endif // MOOST_UTILS_PROGRESS_POLICY_SPINNER_HPP__