Initial checkin

BeefySysLib/util/BeefPerf.h

@@ -0,0 +1,331 @@
+#pragma once
+
+#ifndef BP_NOINC
+#include "../Common.h"
+#include "CritSect.h"
+#endif
+
+#include "Dictionary.h"
+#include "TLSingleton.h"
+
+#ifdef BF_PLATFORM_WINDOWS
+#include <winsock.h>
+#else
+#include <sys/types.h>          /* See NOTES */
+#include <sys/socket.h>
+typedef int SOCKET;
+#endif
+
+enum BpConnectState
+{
+	BpConnectState_NotConnected,
+	BpConnectState_Connecting,
+	BpConnectState_Connected,
+	BpConnectState_Failed
+};
+
+#ifndef BP_DISABLED
+
+NS_BF_BEGIN
+
+class Buffer
+{
+public:
+	uint8* mPtr;
+	int mBufSize;
+	int mDataSize;
+
+public:	
+	Buffer();
+	~Buffer();
+	uint8* Alloc(int size);	
+	void Free();
+	void Clear();
+};
+
+class CircularBuffer
+{
+public:
+	// Either points to an internal segment of the CircularBuffer, or if the range is discontinuous then 
+	//  it presents a temporary pointer that is later committed
+	class View
+	{
+	public:
+		uint8* mPtr;			
+		int mSrcIdx;
+		int mSrcSize;
+		uint8* mTempBuf;
+		int mTempBufSize;
+
+		CircularBuffer* mCircularBuffer;
+
+	public:
+		View();
+		~View();
+
+		void Commit(int size = -1);
+	};
+
+public:
+	uint8* mBuffer;	
+	int mTail;
+	int mDataSize;
+	int mBufSize;
+
+public:
+	CircularBuffer();
+	~CircularBuffer();
+
+	void Clear();
+	void Resize(int newSize);
+	void GrowReserve(int addSize);
+	void Grow(int addSize);
+	void GrowFront(int addSize);
+	int GetSize();
+	void MapView(int idx, int len, View& view);
+	void Read(void* ptr, int idx, int len);
+	void Write(void* ptr, int idx, int len);	
+
+	void RemoveFront(int len);	
+};
+
+class BpContext
+{
+
+};
+
+enum BpResult
+{
+	BpResult_Ok = 0,
+	BpResult_InternalError,
+	BpResult_AlreadyInitialized
+};
+
+#define BP_CHUNKSIZE 4096
+#define BP_NUMCHUNKS 8
+
+#define BP_CLIENT_VERSION 2
+
+enum BpCmd
+{
+	BpCmd_Init,
+	BpCmd_Enter,	
+	BpCmd_EnterDyn,
+	BpCmd_Leave,
+	BpCmd_LODSmallEntry,
+	BpCmd_StrEntry,
+	BpCmd_ThreadAdd,
+	BpCmd_ThreadRemove,
+	BpCmd_SetThread,	
+	BpCmd_ThreadName,	
+	BpCmd_Tick,
+	BpCmd_PrevTick,
+	BpCmd_KeepAlive,
+	BpCmd_ClockInfo,
+	BpCmd_StreamSplitInfo,
+	BpCmd_Event,
+	BpCmd_LODEvent,
+	BpCmd_Cmd
+};
+
+class BpCmdTarget
+{
+public:	
+	CritSect mCritSect;			
+	Buffer mOutBuffer;	
+	char* mThreadName;
+
+	int mCurDynStrIdx;
+	const char* mDynStrs[64];
+
+	int mCurDepth;
+
+public:
+	BpCmdTarget();
+	void Disable();
+	// Note: returned "const char*" is actually an index into a temporary string table
+	const char* DynamicString(const char* str);
+	const char* ToStrPtr(const char* str);
+
+	void Enter(const char* name);
+	void Enter(const char* name, va_list vargs);
+	void EnterF(const char* name, ...);	
+	void Leave();	
+
+	void Event(const char* name, const char* details);
+};
+
+class BpRootCmdTarget : public BpCmdTarget
+{
+public:
+	void Init();
+	void Tick();
+	void KeepAlive();
+	void AddThread(int threadId, BfpThreadId nativeThreadId);
+};
+
+class BpThreadInfo : public BpCmdTarget
+{
+public:	
+    BfpThreadId mNativeThreadId;
+	int mThreadId;	
+	bool mReadyToSend;
+	bool mHasTerminated;
+
+public:
+	BpThreadInfo();
+	~BpThreadInfo();
+	void SetThreadName(const char* name);
+	void RemoveThread();
+};
+
+struct BpZoneName
+{
+public:
+	int mIdx;
+	int mSize;
+};
+
+class BpManager
+{
+public:	
+#ifdef BF_PLATFORM_WINDOWS
+	HANDLE mMutex;
+	HANDLE mSharedMemoryFile;
+#endif
+	SOCKET mSocket;
+	BfpThread* mThread;
+	String mServerName;
+	String mSessionName;
+	String mSessionID;
+	BpRootCmdTarget mRootCmdTarget;
+	BpConnectState mConnectState;
+	SyncEvent mShutdownEvent;
+	CritSect mCritSect;
+	Array<BpThreadInfo*> mThreadInfos;
+	String mClientName;
+	TLSDtor mTLSDtor;
+	bool mCollectData;
+	bool mThreadRunning;	
+	int64 mInitTimeStamp;
+	uint mInitTickCount;
+	int mPauseCount;
+	int mInitCount;
+
+	CircularBuffer mOutBuffer;
+	int mOutBlockSizeLeft;
+	int mCurThreadId;	
+
+	Dictionary<const char*, BpZoneName> mZoneNameMap;
+
+	BF_TLS_DECLSPEC static BpThreadInfo* sBpThreadInfo;
+	static BpManager* sBpManager;
+
+	int64 mCurTick;
+
+protected:	
+	bool Connect();	
+	void FinishWorkThread();
+	void ThreadProc();	
+	static void BFP_CALLTYPE ThreadProcThunk(void* ptr);
+	uint8* StartCmd(uint8 cmd, CircularBuffer::View& view, int maxLen);
+	void EndCmd(CircularBuffer::View& view, uint8* ptr);
+	void TrySendData();
+	void LostConnection();
+	virtual BpThreadInfo* SlowGetCurThreadInfo();
+
+public:
+	virtual void* AllocBytes(int size);
+	virtual void FreeBytes(void*);
+
+public:
+	BpManager();
+	~BpManager();
+
+	void Clear();
+
+	void SetClientName(const StringImpl& clientName);
+	BpResult Init(const char* serverName, const char* sessionName);
+	void RetryConnect();
+	void Pause();
+	void Unpause();
+	void Shutdown();
+	bool IsDisconnected();
+
+	BpContext* CreateContext(const char* name);
+	void CloseContext();
+	void Tick();	
+
+	static BpThreadInfo* GetCurThreadInfo();	
+	static BpManager* Get();
+};
+
+NS_BF_END
+
+class BpAutoZone
+{
+public:
+	BpAutoZone(const char* name)
+	{
+		Beefy::BpManager::GetCurThreadInfo()->Enter(name);
+	}
+
+	~BpAutoZone()
+	{
+		Beefy::BpManager::sBpThreadInfo->Leave();
+	}
+};
+
+class BpAutoZoneF
+{
+public:
+	BpAutoZoneF(const char* name, ...)
+	{
+		va_list args;
+		va_start(args, name);
+		Beefy::BpManager::GetCurThreadInfo()->Enter(name, args);
+	}
+
+	~BpAutoZoneF()
+	{
+		Beefy::BpManager::sBpThreadInfo->Leave();
+	}
+};
+
+#define BP_ENTER(zoneName) BpEnter(name)
+
+#define BP_DYN_STR(str) BpDynStr(str)
+
+// We purposely 'incorrectly' define BP_ZONE as a variadic macro so compilation will fail rather than warn if we attempt to add params
+#define BP_TOKENPASTE(x, y) x ## y
+#define BP_TOKENPASTE2(x, y) BP_TOKENPASTE(x, y)
+
+#define BP_ZONE(zoneName, ...) BpAutoZone BP_TOKENPASTE2(bpZone_, __COUNTER__)(zoneName, ##__VA_ARGS__)
+#define BP_ZONE_F(zoneName, ...) BpAutoZoneF BP_TOKENPASTE2(bpZone_, __COUNTER__)(zoneName, ##__VA_ARGS__)
+#define BP_SHUTDOWN() BpShutdown()
+
+#else
+
+#define BP_ENTER(zoneName) do { } while (0)
+#define BP_DYN_STR(str) str
+
+#define BP_ZONE(zoneName, ...) do { } while (0)
+#define BP_ZONE_F(zoneName, ...) do { } while (0)
+#define BP_SHUTDOWN() do { } while (0)
+
+#endif
+
+BF_EXPORT void BF_CALLTYPE BpInit(const char* serverName, const char* sessionName);
+BF_EXPORT void BF_CALLTYPE BpShutdown();
+BF_EXPORT BpConnectState BF_CALLTYPE BpGetConnectState();
+BF_EXPORT void BF_CALLTYPE BpRetryConnect();
+BF_EXPORT void BF_CALLTYPE BpPause();
+BF_EXPORT void BF_CALLTYPE BpUnpause();
+BF_EXPORT void BF_CALLTYPE BpSetClientName(const char* clientName);
+BF_EXPORT void BF_CALLTYPE BpSetThreadName(const char* threadName);
+BF_EXPORT void BF_CALLTYPE BpEnter(const char* zoneName);
+BF_EXPORT void BF_CALLTYPE BpEnterF(const char* zoneName, ...);
+BF_EXPORT void BF_CALLTYPE BpLeave();
+BF_EXPORT void BF_CALLTYPE BpFrameTick();
+BF_EXPORT void BF_CALLTYPE BpEvent(const char* name, const char* details);
+BF_EXPORT const char* BF_CALLTYPE BpDynStr(const char* str);