dvander@914: /**
dvander@914:  * vim: set ts=4 :
dvander@914:  * ===============================================================
dvander@914:  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@914:  * ===============================================================
dvander@914:  *
dvander@914:  * This file is not open source and may not be copied without explicit
dvander@914:  * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@914:  * in whole or significant part.
dvander@914:  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@914:  *
dvander@914:  * Version: $Id$
dvander@914:  */
dvander@914: 
dvander@914: #include "sm_globals.h"
dvander@914: #include "HandleSys.h"
dvander@914: 
dvander@914: //Note: Do not add this to Linux yet, i haven't done the HPET timing research (if even available)
dvander@914: //nonetheless we need accurate counting
dvander@914: #if defined PLATFORM_LINUX
dvander@914: #error "Not supported"
dvander@914: #endif
dvander@914: 
dvander@914: struct Profiler
dvander@914: {
dvander@914: 	Profiler()
dvander@914: 	{
dvander@914: 		started = false;
dvander@914: 		stopped = false;
dvander@914: 	}
dvander@914: #if defined PLATFORM_WINDOWS
dvander@914: 	LARGE_INTEGER start;
dvander@914: 	LARGE_INTEGER end;
dvander@914: 	double freq;
dvander@914: #endif
dvander@914: 	bool started;
dvander@914: 	bool stopped;
dvander@914: };
dvander@914: 
dvander@914: HandleType_t g_ProfilerType = 0;
dvander@914: 
dvander@914: class ProfilerHelpers : 
dvander@914: 	public SMGlobalClass,
dvander@914: 	public IHandleTypeDispatch
dvander@914: {
dvander@914: public:
dvander@914: 	void OnSourceModAllInitialized()
dvander@914: 	{
dvander@914: 		g_ProfilerType = g_HandleSys.CreateType("Profiler", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@914: 	}
dvander@914: 	void OnSourceModShutdown()
dvander@914: 	{
dvander@914: 		g_HandleSys.RemoveType(g_ProfilerType, g_pCoreIdent);
dvander@914: 	}
dvander@914: 	void OnHandleDestroy(HandleType_t type, void *object)
dvander@914: 	{
dvander@914: 		Profiler *prof = (Profiler *)object;
dvander@914: 		delete prof;
dvander@914: 	}
dvander@914: } s_ProfHelpers;
dvander@914: 
dvander@914: static cell_t CreateProfiler(IPluginContext *pContext, const cell_t *params)
dvander@914: {
dvander@914: 	Profiler *p = new Profiler();
dvander@914: 
dvander@914: #if defined PLATFORM_WINDOWS
dvander@914: 	LARGE_INTEGER qpf;
dvander@914: 	QueryPerformanceFrequency(&qpf);
dvander@914: 	p->freq = 1.0 / (double)(qpf.QuadPart);
dvander@914: #endif
dvander@914: 
dvander@914: 	Handle_t hndl = g_HandleSys.CreateHandle(g_ProfilerType, p, pContext->GetIdentity(), g_pCoreIdent, NULL);
dvander@914: 	if (hndl == BAD_HANDLE)
dvander@914: 	{
dvander@914: 		delete p;
dvander@914: 	}
dvander@914: 
dvander@914: 	return hndl;
dvander@914: }
dvander@914: 
dvander@914: static cell_t StartProfiling(IPluginContext *pContext, const cell_t *params)
dvander@914: {
dvander@914: 	Handle_t hndl = params[1];
dvander@914: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@914: 	HandleError err;
dvander@914: 	Profiler *prof;
dvander@914: 
dvander@914: 	if ((err = g_HandleSys.ReadHandle(hndl, g_ProfilerType, &sec, (void **)&prof))
dvander@914: 		!= HandleError_None)
dvander@914: 	{
dvander@914: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@914: 	}
dvander@914: 
dvander@914: #if defined PLATFORM_WINDOWS
dvander@914: 	QueryPerformanceCounter(&prof->start);
dvander@914: #endif
dvander@914: 
dvander@914: 	prof->started = true;
dvander@914: 	prof->stopped = false;
dvander@914: 
dvander@914: 	return 1;
dvander@914: }
dvander@914: 
dvander@914: static cell_t StopProfiling(IPluginContext *pContext, const cell_t *params)
dvander@914: {
dvander@914: 	Handle_t hndl = params[1];
dvander@914: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@914: 	HandleError err;
dvander@914: 	Profiler *prof;
dvander@914: 
dvander@914: 	if ((err = g_HandleSys.ReadHandle(hndl, g_ProfilerType, &sec, (void **)&prof))
dvander@914: 		!= HandleError_None)
dvander@914: 	{
dvander@914: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@914: 	}
dvander@914: 
dvander@914: 	if (!prof->started)
dvander@914: 	{
dvander@914: 		return pContext->ThrowNativeError("Profiler was never started");
dvander@914: 	}
dvander@914: 
dvander@914: #if defined PLATFORM_WINDOWS
dvander@914: 	QueryPerformanceCounter(&prof->end);
dvander@914: #endif
dvander@914: 
dvander@914: 	prof->started = false;
dvander@914: 	prof->stopped = true;
dvander@914: 
dvander@914: 	return 1;
dvander@914: }
dvander@914: 
dvander@914: static cell_t GetProfilerTime(IPluginContext *pContext, const cell_t *params)
dvander@914: {
dvander@914: 	Handle_t hndl = params[1];
dvander@914: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@914: 	HandleError err;
dvander@914: 	Profiler *prof;
dvander@914: 
dvander@914: 	if ((err = g_HandleSys.ReadHandle(hndl, g_ProfilerType, &sec, (void **)&prof))
dvander@914: 		!= HandleError_None)
dvander@914: 	{
dvander@914: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@914: 	}
dvander@914: 
dvander@914: 	if (!prof->stopped)
dvander@914: 	{
dvander@914: 		return pContext->ThrowNativeError("Profiler was never stopped");
dvander@914: 	}
dvander@914: 
dvander@914: 	float fTime;
dvander@914: 
dvander@914: #if defined PLATFORM_WINDOWS
dvander@914: 	LONGLONG diff = prof->end.QuadPart - prof->start.QuadPart;
dvander@914: 	double seconds = diff * prof->freq;
dvander@914: 	fTime = (float)seconds;
dvander@914: #endif
dvander@914: 
dvander@914: 	return sp_ftoc(fTime);
dvander@914: }
dvander@914: 
dvander@914: REGISTER_NATIVES(profilerNatives)
dvander@914: {
dvander@914: 	{"CreateProfiler",			CreateProfiler},
dvander@914: 	{"GetProfilerTime",			GetProfilerTime},
dvander@914: 	{"StartProfiling",			StartProfiling},
dvander@914: 	{"StopProfiling",			StopProfiling},
dvander@914: 	{NULL,						NULL},
dvander@914: };