dvander@914: /**
dvander@914: * vim: set ts=4 :
dvander@1231: * ================================================================
dvander@1231: * SourceMod
dvander@1231: * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved.
dvander@1231: * ================================================================
dvander@914: *
dvander@1231: * This program is free software; you can redistribute it and/or
dvander@1231: * modify it under the terms of the GNU General Public License,
dvander@1231: * version 3.0, as published by the Free Software Foundation.
dvander@1231: *
dvander@1231: * This program is distributed in the hope that it will be useful,
dvander@1231: * but WITHOUT ANY WARRANTY; without even the implied warranty of
dvander@1231: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
dvander@1231: * GNU General Public License for more details.
dvander@1231: *
dvander@1231: * You should have received a copy of the GNU General Public License
dvander@1231: * along with this program. If not, see .
dvander@1231: *
dvander@1231: * As a special exception, AlliedModders LLC gives you permission to
dvander@1231: * link the code of this program (as well as its derivative works) to
dvander@1231: * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any
dvander@1231: * Game MODs that run on software by the Valve Corporation. You must
dvander@1231: * obey the GNU General Public License in all respects for all other
dvander@1231: * code used. Additionally, AlliedModders LLC grants this exception
dvander@1231: * to all derivative works. AlliedModders LLC defines further
dvander@1231: * exceptions, found in LICENSE.txt (as of this writing, version
dvander@1231: * JULY-31-2007), or .
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: };