tracy/server/TracyView.hpp

116 lines
2.9 KiB
C++
Raw Normal View History

2017-09-12 23:33:50 +00:00
#ifndef __TRACYVIEW_HPP__
#define __TRACYVIEW_HPP__
#include <atomic>
2017-09-14 00:00:13 +00:00
#include <mutex>
2017-09-12 23:33:50 +00:00
#include <string>
#include <thread>
2017-09-14 00:00:13 +00:00
#include <unordered_map>
2017-09-14 00:16:51 +00:00
#include <unordered_set>
2017-09-14 00:00:13 +00:00
#include <vector>
#include "../common/tracy_lz4.hpp"
#include "../common/TracySocket.hpp"
2017-09-14 00:00:13 +00:00
#include "../common/TracyQueue.hpp"
#include "TracyEvent.hpp"
#include "TracySlab.hpp"
2017-09-21 19:57:40 +00:00
#include "TracySourceLocation.hpp"
2017-09-15 18:17:39 +00:00
#include "TracyVector.hpp"
2017-09-12 23:33:50 +00:00
namespace tracy
{
2017-09-13 21:40:28 +00:00
struct QueueItem;
2017-09-12 23:33:50 +00:00
class View
{
public:
View() : View( "127.0.0.1" ) {}
View( const char* addr );
~View();
static bool ShouldExit();
2017-09-15 00:30:22 +00:00
static void Draw();
2017-09-12 23:33:50 +00:00
private:
2017-09-21 23:15:58 +00:00
struct ThreadData
{
2017-09-21 23:58:59 +00:00
uint64_t id;
2017-09-21 23:15:58 +00:00
Vector<Event*> timeline;
};
2017-09-12 23:33:50 +00:00
void Worker();
2017-09-14 17:44:49 +00:00
void DispatchProcess( const QueueItem& ev );
void DispatchProcess( const QueueItem& ev, const char*& ptr );
void Process( const QueueItem& ev );
2017-09-14 00:00:13 +00:00
void ProcessZoneBegin( uint64_t id, const QueueZoneBegin& ev );
void ProcessZoneEnd( uint64_t id, const QueueZoneEnd& ev );
2017-09-15 22:40:51 +00:00
void ProcessFrameMark( uint64_t id );
2017-09-14 00:00:13 +00:00
2017-09-14 00:16:51 +00:00
void CheckString( uint64_t ptr );
2017-09-21 23:30:57 +00:00
void CheckThreadString( uint64_t id );
2017-09-14 17:43:40 +00:00
void AddString( uint64_t ptr, std::string&& str );
2017-09-21 23:30:57 +00:00
void AddThreadString( uint64_t id, std::string&& str );
2017-09-14 00:16:51 +00:00
2017-09-21 23:15:58 +00:00
void NewZone( Event* zone, uint64_t thread );
void UpdateZone( Event* zone );
2017-09-14 19:05:01 +00:00
2017-09-18 19:26:22 +00:00
uint64_t GetFrameTime( size_t idx ) const;
2017-09-20 20:34:56 +00:00
uint64_t GetFrameBegin( size_t idx ) const;
uint64_t GetFrameEnd( size_t idx ) const;
2017-09-18 19:26:22 +00:00
uint64_t GetLastTime() const;
2017-09-18 20:11:38 +00:00
const char* TimeToString( uint64_t ns ) const;
2017-09-21 00:10:20 +00:00
const char* GetString( uint64_t ptr ) const;
2017-09-21 23:30:57 +00:00
const char* GetThreadString( uint64_t id ) const;
2017-09-15 00:30:22 +00:00
void DrawImpl();
2017-09-18 00:37:25 +00:00
void DrawFrames();
2017-09-20 22:57:26 +00:00
void DrawZones();
2017-09-15 00:30:22 +00:00
2017-09-12 23:33:50 +00:00
std::string m_addr;
Socket m_sock;
2017-09-12 23:33:50 +00:00
std::thread m_thread;
std::atomic<bool> m_shutdown;
2017-09-17 22:31:09 +00:00
std::atomic<bool> m_connected;
std::atomic<bool> m_hasData;
2017-09-12 23:54:22 +00:00
2017-09-14 17:43:40 +00:00
// this block must be locked
2017-09-14 00:00:13 +00:00
std::mutex m_lock;
2017-09-15 22:40:51 +00:00
Vector<uint64_t> m_frames;
2017-09-21 19:57:40 +00:00
Vector<SourceLocation> m_srcFile;
2017-09-21 23:15:58 +00:00
Vector<ThreadData> m_threads;
2017-09-14 00:16:51 +00:00
std::unordered_map<uint64_t, std::string> m_strings;
2017-09-21 23:30:57 +00:00
std::unordered_map<uint64_t, std::string> m_threadNames;
2017-09-14 00:00:13 +00:00
std::mutex m_mbpslock;
std::vector<float> m_mbps;
2017-09-14 00:00:13 +00:00
// not used for vis - no need to lock
std::unordered_map<uint64_t, QueueZoneEnd> m_pendingEndZone;
std::unordered_map<uint64_t, Event*> m_openZones;
2017-09-14 00:16:51 +00:00
std::unordered_set<uint64_t> m_pendingStrings;
2017-09-21 23:30:57 +00:00
std::unordered_set<uint64_t> m_pendingThreads;
2017-09-21 19:57:40 +00:00
std::unordered_map<SourceLocation, uint32_t, SourceLocation::Hasher, SourceLocation::Comparator> m_locationRef;
2017-09-21 23:15:58 +00:00
std::unordered_map<uint64_t, uint32_t> m_threadMap;
Slab<EventSize*1024*1024> m_slab;
LZ4_streamDecode_t* m_stream;
char* m_buffer;
int m_bufferOffset;
2017-09-18 00:37:25 +00:00
int m_frameScale;
2017-09-18 22:26:40 +00:00
bool m_pause;
int m_frameStart;
2017-09-20 19:21:21 +00:00
2017-09-20 23:13:23 +00:00
int64_t m_zvStart;
int64_t m_zvEnd;
2017-09-12 23:33:50 +00:00
};
}
#endif