ProgressTime.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "ProgressTime.h"
#include "string_utils.h"
#include "string_color.h"
 
using namespace std;
 
 
ProgressTime::ProgressTime(){
        initialisationProgressTime(0, '=', '>', ' ');
}
 
 
ProgressTime::ProgressTime(size_t nbSteps, char chPlein, char chMiddle, char chEnd){
        initialisationProgressTime(nbSteps, chPlein, chMiddle, chEnd);
}
 
 
ProgressTime::ProgressTime(const ProgressTime & other){
        p_buffer = NULL;
        copyProgressTime(other);
}
 
ProgressTime::~ProgressTime(){
        if(p_buffer != NULL){
                delete [] p_buffer;
        }
}
 
void ProgressTime::start(){
        p_startTime = clock();
        p_iCurrent = 0;
        p_oldAvancement = 0;
        p_avancement = 0;
        p_secondesRestantes = 0;
        p_oldTempsRestant = 0;
}
 
void ProgressTime::print(){
        ++p_iCurrent;
        p_avancement = (p_iCurrent*p_sizeBarr)/p_nbSteps;
        p_tempsRestant = (clock_t)(((clock() - p_startTime)*((clock_t)(p_nbSteps - p_iCurrent)))/((clock_t)p_iCurrent));
        if(p_oldAvancement != p_avancement || p_oldTempsRestant != p_tempsRestant){
                p_pourcent = (p_iCurrent*100)/p_nbSteps;
                //on efface la ligne de la console où il y a la barre de chargement
                clear();
                cerr << "\r" << p_buffer << "\r\r\r\r";
                //on affiche la barre de chargement
                feelbuffer(p_avancement);
                p_secondesRestantes = p_tempsRestant/CLOCKS_PER_SEC;
                cerr << "[" << p_buffer << "|" << p_pourcent << "%] " << p_secondesRestantes << "s";
                //on met old_taille à jour
                p_oldAvancement = p_avancement;
                p_oldTempsRestant = p_tempsRestant;
        }
}
 
void ProgressTime::finish(){
        clear();
        cerr << "\r" << p_buffer << "\r\r\r\r";
        feelbuffer(p_sizeBarr);
        cerr << "[" << p_buffer << "|" << p_pourcent << "%] " << p_secondesRestantes << "s";
        cerr << endl;
}
 
 
void ProgressTime::setNbSteps(size_t nbSteps){
        p_nbSteps = nbSteps;
}
 
 
void ProgressTime::setChPlein(char chPlein){
        p_chPlein = chPlein;
}
 
 
void ProgressTime::setChMiddle(char chMiddle){
        p_chMiddle = chMiddle;
}
 
 
void ProgressTime::setChEnd(char chEnd){
        p_chEnd = chEnd;
}
 
 
ProgressTime & ProgressTime::operator = (const ProgressTime & other){
        copyProgressTime(other);
        return *this;
}
 
 
void ProgressTime::copyProgressTime(const ProgressTime & other){
        if(p_buffer != NULL){
                delete [] p_buffer;
        }
        p_buffer = new char[other.p_sizeBarr + 1];                    //on pense au caractère \0
        p_startTime = other.p_startTime;
        p_tempsRestant = other.p_tempsRestant;
        p_nbSteps = other.p_nbSteps;
        p_iCurrent = other.p_iCurrent;
        p_avancement = other.p_avancement;
        p_oldAvancement = other.p_oldAvancement;
        p_secondesRestantes = other.p_secondesRestantes;
        p_oldTempsRestant = other.p_oldTempsRestant;
        p_consoleWidth = other.p_consoleWidth;
        p_pourcent = other.p_pourcent;
        p_sizeBarr = other.p_sizeBarr;
        p_chPlein = other.p_chPlein;
        p_chMiddle = other.p_chMiddle;
        p_chEnd = other.p_chEnd;
        
        p_buffer[p_sizeBarr] = '\0';
}
 
 
void ProgressTime::initialisationProgressTime(size_t nbSteps, char chPlein, char chMiddle, char chEnd){
        p_nbSteps = nbSteps;
        p_chPlein = chPlein;
        p_chMiddle = chMiddle;
        p_chEnd = chEnd;
        start();
        p_consoleWidth = getNbColTerminal();
        p_sizeBarr = (3u*p_consoleWidth)/4u; //on prend 3/4 de la taille de la console
        p_buffer = new char[p_sizeBarr + 1u];
}
 
 
void ProgressTime::feelbuffer(unsigned int taille){
        if(taille <= p_sizeBarr){
                for(unsigned int i(0); i < taille; ++i){
                        p_buffer[i] = p_chPlein;
                }
                p_buffer[taille] = p_chMiddle;
        }
        for(unsigned int i(taille + 1); i <= p_sizeBarr; ++i){
                p_buffer[i] = p_chEnd;
        }
        p_buffer[p_sizeBarr] = '\0';
}
 
void ProgressTime::clear(){
        for(unsigned int i(0); i < p_sizeBarr; ++i){
                p_buffer[i] = '\r';
        }
        p_buffer[p_sizeBarr] = '\0';
}