NICE_VisLearning/baselib/ProgressBarQt.cpp

/** 
 * @file ProgressBarQt.cpp
 * @brief Show a Progress-Bar with time estimation
 * @author Michael Koch
 * @date 25/03/2010

 */
//#ifdef NOVISUAL
//#include <vislearning/nice_nonvis.h>
//#else
#include <vislearning/nice.h>
#include <core/imagedisplay/QtFramework.h>
//#endif

#include <iostream>
#include <sys/time.h>

#include "vislearning/baselib/ProgressBarQt.h"


using namespace OBJREC;

using namespace std;

using namespace NICE;

ProgressBarQt::ProgressBarQt(const std::string & _name, bool _useGraphics)
	: useGraphics (_useGraphics)
{
	name = _name;
	step = 0;

	display_on = false;
	working = true;
	if (graphicIsAvailable())
	{
		if (qApp == NULL)
		{
			QtFramework::instance();
		}

		dialogwindow = new QWidget;

		progressdialog = new QProgressDialog("Process at work ...", "Cancel",
				0, 100);
		layout = new QGridLayout(dialogwindow, 1, 1);
		layout->addWidget(progressdialog, 0, 0);
		dialogwindow->setLayout(layout);
	}
	reset(_name);
}

ProgressBarQt::~ProgressBarQt()
{
}

void ProgressBarQt::timediff2str(char *text, long time)
{
	int seconds;
	int minutes;
	int hours;
	int milliseconds;

	milliseconds = time % 100;
	time /= 100;
	seconds = time % 60;
	time /= 60;
	minutes = time % 60;
	time /= 60;
	hours = time;

	if (hours != 0)
	{
		sprintf(text, "%dh %dm %d.%d s", hours, minutes, seconds, milliseconds);
	}
	else if (minutes != 0)
	{
		sprintf(text, "%dm %d.%d s", minutes, seconds, milliseconds);
	}
	else
	{
		sprintf(text, "%d.%d s", seconds, milliseconds);
	}
}

void ProgressBarQt::displayTimeStat(int posy, char *text, long time)
{
	// Text (char *str,int x0,int y0,int val,int exp,Image img);
	char disptext[200];
	char timetext[200];

	timediff2str(timetext, time);
	sprintf(disptext, "%s %s", text, timetext);

}

double ProgressBarQt::getCurrentTime()
{
	struct timeval curtime;

	gettimeofday(&curtime, NULL);

	return curtime.tv_sec * 100.0 + curtime.tv_usec / 10000.0;
}

void ProgressBarQt::show()
{
	if (!display_on)
	{
		display_on = true;
	}
	if (graphicIsAvailable())
	{
		if (qApp == NULL)
		{
			QtFramework::instance();
		}
		dialogwindow->show();
	}
}

void ProgressBarQt::hide()
{

	if (display_on)
	{
		//		Show ( OFF, display, name );
		display_on = false;
	}
	if (graphicIsAvailable())
	{
		dialogwindow->hide();
	}
}

void ProgressBarQt::stop()
{

	working = false;
}

void ProgressBarQt::reset(const std::string & _name)
{
	name = _name;
	step = 0;
	elapsed_time = 0;
	avg_time_step = 0;
	start_time = getCurrentTime();
}

void ProgressBarQt::update(int count)
{
	step++;

	double progress = step / (double) count;

	elapsed_time = getCurrentTime() - start_time;
	avg_time_step = elapsed_time / step;
	estimated_time = (count - step) * avg_time_step;

	size_t mod;
	if (avg_time_step > 50.0)
		mod = 1;
	else
		mod = (size_t) (50.0 / avg_time_step) + 1;

	if ((mod <= 1) || (step % mod == 0))
	{
		char percent_text[10];
		sprintf(percent_text, "%4.2f %%", step * 100.0 / count);

		displayTimeStat(0, "Elapsed Time : ", (long int) elapsed_time);
		displayTimeStat(1, "Estimated Time : ", (long int) estimated_time);
		displayTimeStat(2, "Avg. Time per step : ", (long int) avg_time_step);

		char eltime[200];
		char estime[200];
		char avgtime[200];
		timediff2str(eltime, (long int) elapsed_time);
		timediff2str(estime, (long int) estimated_time);
		timediff2str(avgtime, (long int) avg_time_step);
		fprintf(
				stderr,
				"[PROGRESS] %s %s (elapsed time %s, estimated time %s, avg %s), \n",
				name.c_str(), percent_text, eltime, estime, avgtime);
		//set progress value.
		if (graphicIsAvailable())
		{
			progressdialog->setValue(progress * 100);
		}
	}
}

bool ProgressBarQt::graphicIsAvailable()
{
	return (useGraphics && (getenv("DISPLAY") != NULL));
}