/**

 * vim: set ts=4 :

 * =============================================================================

 * SourceMod

 * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.

 * =============================================================================

 *

 * This program is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License, version 3.0, 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 General Public License along with

 * this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to link the

 * code of this program (as well as its derivative works) to "Half-Life 2," the

 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

 * by the Valve Corporation.  You must obey the GNU General Public License in

 * all respects for all other code used.  Additionally, AlliedModders LLC grants

 * this exception to all derivative works.  AlliedModders LLC defines further

 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

 * or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#ifndef _INCLUDE_SOURCEMOD_PLUGIN_PROFILER_H_

#define _INCLUDE_SOURCEMOD_PLUGIN_PROFILER_H_

#include <sp_vm_api.h>

#include <sm_platform.h>

#include <sm_trie_tpl.h>

#include <sh_vector.h>

#include <sh_stack.h>

#include <stdio.h>

#include "sm_globals.h"

#include "sm_srvcmds.h"

using namespace SourcePawn;

using namespace SourceHook;

struct prof_point_t

{

#if defined PLATFORM_WINDOWS

	LARGE_INTEGER value;

#elif defined PLATFORM_POSIX

	struct timeval value;

#endif

	bool is_set;

};

struct prof_atom_t

{

	int atom_type;			/* Type of object we're profiling */

	int atom_serial;		/* Serial number, if appropriate */

	sp_context_t *ctx;		/* Plugin context. */

	const char *name;		/* Name of the function */

	prof_point_t start;		/* Start time */

	prof_point_t end;		/* End time */

	double base_time;		/* Known time from children or pausing. */

};

struct prof_atom_report_t

{

	char atom_name[256];	/* Full name to shove to logs */

	double total_time;		/* Total time spent executing, in s */

	unsigned int num_calls;	/* Number of invocations */

	double min_time;		/* Min time spent in one call, in s */

	double max_time;		/* Max time spent in one call, in s */

};

class ProfileReport

{

public:

	~ProfileReport();

public:

	void SaveAtom(const prof_atom_t &atom);

	size_t GetNumReports();

	prof_atom_report_t *GetReport(size_t i);

	void Clear();

private:

	KTrie<prof_atom_report_t *> m_ReportLookup;

	CVector<prof_atom_report_t *> m_Reports;

};

class ProfileEngine :

	public SMGlobalClass,

	public IRootConsoleCommand,

	public IProfiler

{

public:

	ProfileEngine();

public:

	bool IsEnabled();

	bool GenerateReport(FILE *fp);

	void Clear();

public: //SMGlobalClass

	void OnSourceModAllInitialized();

	void OnSourceModShutdown();

public: //IRootConsoleCommand

	void OnRootConsoleCommand(const char *cmdname, const CCommand &command);

public: //IProfiler

	void OnNativeBegin(IPluginContext *pContext, sp_native_t *native);

	void OnNativeEnd() ;

	void OnFunctionBegin(IPluginContext *pContext, const char *name);

	void OnFunctionEnd();

	int OnCallbackBegin(IPluginContext *pContext, sp_public_t *pubfunc);

	void OnCallbackEnd(int serial);

private:

	void PushProfileStack(IPluginContext *ctx, int type, const char *name);

	void PopProfileStack(ProfileReport *reporter);

	void PauseParent();

	void ResumeParent(double addTime);

	void WriteReport(FILE *fp, ProfileReport *report, const char *name);

private:

	CStack<prof_atom_t> m_AtomStack;

	ProfileReport m_Callbacks;

	ProfileReport m_Functions;

	ProfileReport m_Natives;

	int m_serial;

	prof_point_t m_ProfStart;

};

extern ProfileEngine g_Profiler;

#endif //_INCLUDE_SOURCEMOD_PLUGIN_PROFILER_H_