// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010  Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "stdmisc.h"
#include "nel/misc/cpu_time_stat.h"

#include <nel/misc/common.h>

#include <string>
#include <vector>

#include <cstdio>

#ifdef NL_OS_UNIX
#include <unistd.h>
#else
#include <process.h>
#endif

using namespace std;

namespace NLMISC
{


// Get absolute ticks value for the whole cpu set
bool	CCPUTimeStat::getCPUTicks(uint64& user, uint64& nice, uint64& system, uint64& idle, uint64& iowait)
{
#ifdef NL_OS_UNIX

	const char*	statfile = "/proc/stat";
	FILE*		f = fopen(statfile, "r");

	if (f == NULL)
		return false;

	// /proc/stat
			// cpu  [user]     [nice]    [system]    [idle]     [iowait]   [irq] [softirq]
	fscanf(f, "cpu  %"NL_I64"u %"NL_I64"u %"NL_I64"u %"NL_I64"u %"NL_I64"u", &user, &nice, &system, &idle, &iowait);

	fclose(f);

	return true;
#else
	return false;
#endif
}



// Get absolute ticks values for a specified pid
bool	CCPUTimeStat::getPIDTicks(uint64& utime, uint64& stime, uint64& cutime, uint64& cstime, uint pid)
{
#ifdef NL_OS_UNIX

	std::string	statfile = NLMISC::toString("/proc/%u/stat", pid);
	FILE*	f = fopen(statfile.c_str(), "r");

	if (f == NULL)
		return false;

	// /proc/<pid>/stat
			// pid com sta ppi pgi ses tty tpg fla mif maf cmi cma [utime]    [stime]    [cutime]   [cstime] ...
	fscanf(f, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %"NL_I64"u %"NL_I64"u %"NL_I64"u %"NL_I64"u", &utime, &stime, &cutime, &cstime);

	fclose(f);

	return true;
#else
	utime = 0;
	stime = 0;
	cutime = 0;
	cstime = 0;
	nlunreferenced(pid);
	return false;
#endif
}

CCPUTimeStat::CCPUTimeStat() :  _PID(0), _FirstTime(true), _LastCPUTicks(0), _LastPIDTicks(0)
{
#ifdef NL_OS_UNIX

	_PID = getpid();

#else

	_PID = _getpid();

#endif

}


// Peek measure
void	CCPUTimeStat::peekMeasures()
{
	NLMISC::TTime	ctime = NLMISC::CTime::getLocalTime();

	uint64	u, n, s, i, io;
	uint64	ut, st, cut, cst;
	if (getCPUTicks(u, n, s, i, io) && getPIDTicks(ut, st, cut, cst, _PID))
	{
		uint64	cpuTicks = u+n+s+i+io;
		uint64	pidTicks = ut+st+cut+cst;

		// only compute diff
		if (_LastCPUTicks == cpuTicks || _LastPIDTicks == pidTicks)
			return;

		_LastCPUTicks = cpuTicks;
		_CPUUser.computeDiff(u);
		_CPUNice.computeDiff(n);
		_CPUSystem.computeDiff(s);
		_CPUIdle.computeDiff(i);
		_CPUIOWait.computeDiff(io);

		_LastPIDTicks = pidTicks;
		_PIDUTime.computeDiff(ut);
		_PIDSTime.computeDiff(st);
		_PIDCUTime.computeDiff(cut);
		_PIDCSTime.computeDiff(cst);

		if (!_FirstTime)
		{
			uint32	cpuTotal = _CPUUser.Diff+_CPUNice.Diff+_CPUSystem.Diff+_CPUIdle.Diff+_CPUIOWait.Diff;

			_CPUUser.computeLoad(cpuTotal, ctime);
			_CPUNice.computeLoad(cpuTotal, ctime);
			_CPUSystem.computeLoad(cpuTotal, ctime);
			_CPUIdle.computeLoad(cpuTotal, ctime);
			_CPUIOWait.computeLoad(cpuTotal, ctime);

			uint32	pidTotal = _PIDUTime.Diff+_PIDSTime.Diff+_PIDCUTime.Diff+_PIDCSTime.Diff;
			if (pidTotal > cpuTotal)
			{
				// EEK! should not happen!!
				cpuTotal = pidTotal;
			}

			_PIDUTime.computeLoad(cpuTotal, ctime);
			_PIDSTime.computeLoad(cpuTotal, ctime);
			_PIDCUTime.computeLoad(cpuTotal, ctime);
			_PIDCSTime.computeLoad(cpuTotal, ctime);
		}

		_FirstTime = false;
	}
}


}