Progress.cpp

Go to the documentation of this file.

 
 /*
 * ProgressBar.cpp
 *
 *  Created on: 14.01.2011
 *      Author: Thomas Wiemann
 */
 
 
#include "lvr2/io/Progress.hpp"
 
#include <sstream>
#include <iostream>
 
using std::stringstream;
using std::cout;
using std::endl;
using std::flush;
 
namespace lvr2
{
 
ProgressCallbackPtr ProgressBar::m_progressCallback = 0;
ProgressTitleCallbackPtr ProgressBar::m_titleCallback = 0;
 
ProgressBar::ProgressBar(size_t max_val, string prefix)
{
        m_prefix = prefix;
        m_maxVal = max_val;
    m_currentVal = 0;
        m_percent = 0;
 
        if(m_titleCallback)
        {
                // Remove time brackets
                unsigned index;
                index = prefix.find_last_of("]");
                m_titleCallback(prefix.substr(index+1));
        }
}
 
ProgressBar::~ProgressBar()
{
 
}
 
void ProgressBar::setProgressCallback(ProgressCallbackPtr ptr)
{
        m_progressCallback = ptr;
}
 
void ProgressBar::setProgressTitleCallback(ProgressTitleCallbackPtr ptr)
{
        m_titleCallback = ptr;
}
 
void ProgressBar::operator++()
{
    boost::mutex::scoped_lock lock(m_mutex);
 
    m_currentVal++;
    short difference = (short)((float)m_currentVal/m_maxVal * 100 - m_percent);
    if (difference < 1)
    {
        return;
    }
 
    while (difference >= 1)
    {
        m_percent++;
        difference--;
        print_bar();
 
        if(m_progressCallback)
        {
                m_progressCallback(m_percent);
        }
    }
 
}
 
void ProgressBar::operator+=(size_t n)
{
    boost::mutex::scoped_lock lock(m_mutex);
 
    m_currentVal+= n;
    short difference = (short)((float)m_currentVal/m_maxVal * 100 - m_percent);
    if (difference < 1)
    {
        return;
    }
 
    while (difference >= 1)
    {
        m_percent++;
        difference--;
        print_bar();
 
        if(m_progressCallback)
        {
            m_progressCallback(m_percent);
        }
    }
 
}
 
void ProgressBar::print_bar()
{
        cout <<  "\r" << m_prefix << " " << m_percent << "%" << flush;
}
 
ProgressCounter::ProgressCounter(int stepVal, string prefix)
{
        m_prefix = prefix;
        m_stepVal = stepVal;
        m_currentVal = 0;
}
 
void ProgressCounter::operator++()
{
        boost::mutex::scoped_lock lock(m_mutex);
        m_currentVal++;
        if(m_currentVal % m_stepVal == 0)
        {
                print_progress();
        }
}
 
void ProgressCounter::print_progress()
{
        cout << "\r" << m_prefix << " " << m_currentVal << flush;
}
 
PacmanProgressCallbackPtr PacmanProgressBar::m_progressCallback = 0;
PacmanProgressTitleCallbackPtr PacmanProgressBar::m_titleCallback = 0;
 
PacmanProgressBar::PacmanProgressBar(size_t max_val, string prefix, size_t bar_length)
:
        m_prefix(prefix)
        ,m_bar_length(bar_length)
{
        m_maxVal = max_val;
    m_currentVal = 0;
        m_percent = 0;
 
        if(m_titleCallback)
        {
                // Remove time brackets
                unsigned index;
                index = prefix.find_last_of("]");
                m_titleCallback(prefix.substr(index+1));
        }
}
 
PacmanProgressBar::~PacmanProgressBar()
{
 
}
 
void PacmanProgressBar::setProgressCallback(ProgressCallbackPtr ptr)
{
        m_progressCallback = ptr;
}
 
void PacmanProgressBar::setProgressTitleCallback(ProgressTitleCallbackPtr ptr)
{
        m_titleCallback = ptr;
}
 
void PacmanProgressBar::operator++()
{
    boost::mutex::scoped_lock lock(m_mutex);
 
    m_currentVal++;
    short difference = (short)((float)m_currentVal/m_maxVal * 100 - m_percent);
 
        if (difference < 1)
    {
        return;
    }
 
    while (difference >= 1)
    {
        m_percent++;
        difference--;
        print_bar();
 
        if(m_progressCallback)
        {
                m_progressCallback(m_percent);
        }
    }
 
}
 
void PacmanProgressBar::print_bar()
{
        int char_idx = static_cast<float>(m_percent)/100.0 * (m_bar_length);
 
        cout <<  "\r" << m_prefix << " " << m_percent << "%" << " | ";
 
        for(size_t i=0; i< char_idx; i++)
        {
                cout << " ";
        }
 
        if(char_idx % 2 == 0)
        {
                cout << "ᗧ";
        }else{
                cout << "O";
        }
 
        for(size_t i=char_idx; i < m_bar_length; i++)
        {
                if(i%2 == 0)
                {
                        cout << " ";
                } else {
                        cout << "•";
                }
        }
 
    cout << " | " << flush;
}
 
} // namespace lvr