// ============================================================================
//  Copyright (C) 2001-2012 by Jozef Starosczyk
//  ixen@copyhandler.com
//
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU Library General Public License
//  (version 2) as published by the Free Software Foundation;
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU Library General Public
//  License along with this program; if not, write to the
//  Free Software Foundation, Inc.,
//  59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
// ============================================================================
/// @file  TSubTaskStatsSnapshot.cpp
/// @date  2012/2/26
/// @brief Contains class responsible for holding sub task stats.
// ============================================================================
#include "stdafx.h"
#include "TSubTaskStatsSnapshot.h"
#include "DataBuffer.h"
#include <boost/numeric/conversion/cast.hpp>
#include "MathFunctions.h"

BEGIN_CHCORE_NAMESPACE

///////////////////////////////////////////////////////////////////////////////////
// class TSubTaskStats
TSubTaskStatsSnapshot::TSubTaskStatsSnapshot() :
	m_bSubTaskIsRunning(false),
	m_ullTotalSize(0),
	m_ullProcessedSize(0),
	m_stTotalCount(0),
	m_stProcessedCount(0),
	m_iCurrentBufferIndex(TBufferSizes::eBuffer_Default),
	m_strCurrentPath(0),
	m_timeElapsed(0),
	m_dSizeSpeed(0),
	m_dCountSpeed(0),
	m_ullCurrentItemProcessedSize(0),
	m_ullCurrentItemTotalSize(0)
{
}

TSubTaskStatsSnapshot::TSubTaskStatsSnapshot(const TSubTaskStatsSnapshot& rSrc) :
	m_bSubTaskIsRunning(rSrc.m_bSubTaskIsRunning),
	m_ullTotalSize(rSrc.m_ullTotalSize),
	m_ullProcessedSize(rSrc.m_ullProcessedSize),
	m_stTotalCount(rSrc.m_stTotalCount),
	m_stProcessedCount(rSrc.m_stProcessedCount),
	m_iCurrentBufferIndex(rSrc.m_iCurrentBufferIndex),
	m_strCurrentPath(rSrc.m_strCurrentPath),
	m_timeElapsed(rSrc.m_timeElapsed),
	m_dSizeSpeed(rSrc.m_dSizeSpeed),
	m_dCountSpeed(rSrc.m_dCountSpeed),
	m_ullCurrentItemProcessedSize(rSrc.m_ullCurrentItemProcessedSize),
	m_ullCurrentItemTotalSize(rSrc.m_ullCurrentItemTotalSize)
{
}

TSubTaskStatsSnapshot& TSubTaskStatsSnapshot::operator=(const TSubTaskStatsSnapshot& rSrc)
{
	m_bSubTaskIsRunning = rSrc.m_bSubTaskIsRunning;
	m_ullTotalSize = rSrc.m_ullTotalSize;
	m_ullProcessedSize = rSrc.m_ullProcessedSize;
	m_stTotalCount = rSrc.m_stTotalCount;
	m_stProcessedCount = rSrc.m_stProcessedCount;
	m_iCurrentBufferIndex = rSrc.m_iCurrentBufferIndex;
	m_strCurrentPath = rSrc.m_strCurrentPath;
	m_timeElapsed = rSrc.m_timeElapsed;
	m_dSizeSpeed = rSrc.m_dSizeSpeed;
	m_dCountSpeed = rSrc.m_dCountSpeed;
	m_ullCurrentItemProcessedSize = rSrc.m_ullCurrentItemProcessedSize;
	m_ullCurrentItemTotalSize = rSrc.m_ullCurrentItemTotalSize;

	return *this;
}

void TSubTaskStatsSnapshot::Clear()
{
	m_bSubTaskIsRunning = false;
	m_ullTotalSize = 0;
	m_ullProcessedSize = 0;
	m_stTotalCount = 0;
	m_stProcessedCount = 0;
	m_iCurrentBufferIndex = TBufferSizes::eBuffer_Default;
	m_strCurrentPath = 0;
	m_timeElapsed = 0;
	m_dSizeSpeed = 0;
	m_dCountSpeed = 0;
	m_ullCurrentItemProcessedSize = 0;
	m_ullCurrentItemTotalSize = 0;
}

double TSubTaskStatsSnapshot::CalculateProgressInPercent() const
{
	if(m_ullTotalSize != 0)
		return Math::Div64(m_ullProcessedSize, m_ullTotalSize);
	else
		return 0.0;
}

unsigned long long TSubTaskStatsSnapshot::GetEstimatedTotalTime() const
{
	double dProgress = CalculateProgressInPercent();
	if(dProgress == 0.0)
		return std::numeric_limits<unsigned long long>::max();
	else
		return (unsigned long long)(m_timeElapsed * (1.0 / dProgress));
}

void TSubTaskStatsSnapshot::SetSizeSpeed(double dSizeSpeed)
{
	m_dSizeSpeed = dSizeSpeed;
}

void TSubTaskStatsSnapshot::SetCountSpeed(double dCountSpeed)
{
	m_dCountSpeed = dCountSpeed;
}

double TSubTaskStatsSnapshot::GetAvgSizeSpeed() const
{
	if(m_timeElapsed)
		return Math::Div64(m_ullProcessedSize, m_timeElapsed / 1000);
	else
		return 0.0;
}

double TSubTaskStatsSnapshot::GetAvgCountSpeed() const
{
	if(m_timeElapsed)
		return Math::Div64(m_stProcessedCount, m_timeElapsed / 1000);
	else
		return 0.0;
}

END_CHCORE_NAMESPACE