tracy/server/TracyView.hpp

244 lines
7.7 KiB
C++
Raw Normal View History

2017-09-12 23:33:50 +00:00
#ifndef __TRACYVIEW_HPP__
#define __TRACYVIEW_HPP__
#include <atomic>
#include <map>
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 "TracyCharUtil.hpp"
2017-09-14 00:00:13 +00:00
#include "TracyEvent.hpp"
#include "TracySlab.hpp"
2017-09-15 18:17:39 +00:00
#include "TracyVector.hpp"
2017-09-12 23:33:50 +00:00
struct ImVec2;
2017-09-12 23:33:50 +00:00
namespace tracy
{
2017-09-13 21:40:28 +00:00
struct QueueItem;
2017-09-30 14:58:02 +00:00
class FileRead;
2017-09-30 14:20:08 +00:00
class FileWrite;
2017-09-13 21:40:28 +00:00
2017-09-12 23:33:50 +00:00
class View
{
public:
View() : View( "127.0.0.1" ) {}
View( const char* addr );
2017-09-30 14:58:02 +00:00
View( FileRead& f );
2017-09-12 23:33:50 +00:00
~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-10-22 11:56:05 +00:00
enum class Namespace : uint8_t
{
Full,
Mid,
Short
};
2017-09-12 23:33:50 +00:00
void Worker();
2017-11-10 18:15:00 +00:00
void DispatchProcess( const QueueItem& ev, char*& ptr );
2017-09-14 17:44:49 +00:00
void ServerQuery( uint8_t type, uint64_t data );
2017-09-14 17:44:49 +00:00
void Process( const QueueItem& ev );
2017-10-03 14:41:32 +00:00
void ProcessZoneBegin( const QueueZoneBegin& ev );
void ProcessZoneBeginAllocSrcLoc( const QueueZoneBegin& ev );
2017-10-03 14:41:32 +00:00
void ProcessZoneEnd( const QueueZoneEnd& ev );
void ProcessFrameMark( const QueueFrameMark& ev );
void ProcessZoneText( const QueueZoneText& ev );
void ProcessZoneName( const QueueZoneName& ev );
2017-11-11 16:56:41 +00:00
void ProcessZoneNameLiteral( const QueueZoneName& ev );
void ProcessLockWait( const QueueLockWait& ev );
void ProcessLockObtain( const QueueLockObtain& ev );
void ProcessLockRelease( const QueueLockRelease& ev );
2017-10-06 15:05:14 +00:00
void ProcessLockMark( const QueueLockMark& ev );
2017-10-13 01:36:59 +00:00
void ProcessPlotData( const QueuePlotData& ev );
2017-10-21 10:39:26 +00:00
void ProcessMessage( const QueueMessage& ev );
void ProcessMessageLiteral( const QueueMessage& ev );
2017-11-11 18:44:09 +00:00
void ProcessGpuNewContext( const QueueGpuNewContext& ev );
2017-11-11 20:09:48 +00:00
void ProcessGpuZoneBegin( const QueueGpuZoneBegin& ev );
void ProcessGpuZoneEnd( const QueueGpuZoneEnd& ev );
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 );
void CheckSourceLocation( uint64_t ptr );
void AddString( uint64_t ptr, char* str, size_t sz );
void AddThreadString( uint64_t id, char* str, size_t sz );
2017-11-10 18:15:00 +00:00
void AddCustomString( uint64_t ptr, char* str, size_t sz );
2017-10-03 14:41:32 +00:00
void AddSourceLocation( const QueueSourceLocation& srcloc );
void AddSourceLocationPayload( uint64_t ptr, char* data, size_t sz );
2017-09-14 00:16:51 +00:00
2017-11-10 18:15:00 +00:00
StringLocation StoreString( char* str, size_t sz );
uint32_t ShrinkSourceLocation( uint64_t srcloc );
2017-10-21 10:39:26 +00:00
void InsertMessageData( MessageData* msg, uint64_t thread );
2017-10-21 11:14:20 +00:00
ThreadData* NoticeThread( uint64_t thread );
2017-10-22 13:37:24 +00:00
void NewZone( ZoneEvent* zone, uint64_t thread );
void UpdateZone( ZoneEvent* zone );
2017-09-14 19:05:01 +00:00
2017-10-22 14:15:27 +00:00
void InsertZone( ZoneEvent* zone, Vector<ZoneEvent*>& vec );
2017-10-08 21:03:38 +00:00
void InsertLockEvent( LockMap& lockmap, LockEvent* lev, uint64_t thread );
void UpdateLockCount( LockMap& lockmap, size_t pos );
2017-10-04 16:51:50 +00:00
2017-10-13 01:36:59 +00:00
void InsertPlot( PlotData* plot, int64_t time, double val );
2017-10-19 19:04:11 +00:00
void InsertPlot( PlotData* plot, PlotItem* item );
2017-11-10 18:41:37 +00:00
void HandlePlotName( uint64_t name, char* str, size_t sz );
2017-10-19 21:26:21 +00:00
void HandlePostponedPlots();
2017-10-13 01:36:59 +00:00
2017-10-28 19:50:06 +00:00
int64_t GetFrameTime( size_t idx ) const;
int64_t GetFrameBegin( size_t idx ) const;
int64_t GetFrameEnd( size_t idx ) const;
int64_t GetLastTime() const;
2017-10-22 13:37:24 +00:00
int64_t GetZoneEnd( const ZoneEvent& ev ) const;
2017-09-21 00:10:20 +00:00
const char* GetString( uint64_t ptr ) const;
const char* GetString( const StringRef& ref ) const;
2017-09-21 23:30:57 +00:00
const char* GetThreadString( uint64_t id ) const;
const SourceLocation& GetSourceLocation( int32_t srcloc ) const;
2017-10-22 11:56:05 +00:00
const char* ShortenNamespace( const char* name ) const;
2017-09-15 00:30:22 +00:00
void DrawImpl();
2017-09-30 12:37:21 +00:00
void DrawConnection();
2017-09-18 00:37:25 +00:00
void DrawFrames();
bool DrawZoneFrames();
2017-09-20 22:57:26 +00:00
void DrawZones();
2017-10-22 13:37:24 +00:00
int DrawZoneLevel( const Vector<ZoneEvent*>& vec, bool hover, double pxns, const ImVec2& wpos, int offset, int depth );
2017-10-08 21:03:38 +00:00
int DrawLocks( uint64_t tid, bool hover, double pxns, const ImVec2& wpos, int offset, LockHighlight& highlight );
2017-09-29 19:49:14 +00:00
void DrawZoneInfoWindow();
2017-10-13 13:15:57 +00:00
int DrawPlots( int offset, double pxns, const ImVec2& wpos, bool hover );
2017-10-18 20:29:59 +00:00
void DrawPlotPoint( const ImVec2& wpos, float x, float y, int offset, uint32_t color, bool hover, bool hasPrev, double val, double prev, bool merged );
2017-10-13 11:32:23 +00:00
void DrawOptions();
2017-10-14 12:36:30 +00:00
void DrawMessages();
2017-09-15 00:30:22 +00:00
2017-10-12 20:27:17 +00:00
void HandleZoneViewMouse( int64_t timespan, const ImVec2& wpos, float w, double& pxns );
2017-10-22 13:37:24 +00:00
uint32_t GetZoneColor( const ZoneEvent& ev );
uint32_t GetZoneColor( const SourceLocation& srcloc );
2017-10-22 13:37:24 +00:00
uint32_t GetZoneHighlight( const ZoneEvent& ev, bool migration );
float GetZoneThickness( const ZoneEvent& ev );
2017-10-01 17:31:22 +00:00
2017-10-22 13:37:24 +00:00
void ZoomToZone( const ZoneEvent& ev );
void ZoneTooltip( const ZoneEvent& ev );
2017-10-22 14:15:27 +00:00
const ZoneEvent* GetZoneParent( const ZoneEvent& zone ) const;
2017-09-29 19:57:00 +00:00
2017-10-22 13:37:24 +00:00
TextData* GetTextData( ZoneEvent& zone );
const TextData* GetTextData( const ZoneEvent& zone ) const;
2017-09-30 14:20:08 +00:00
void Write( FileWrite& f );
2017-10-22 13:37:24 +00:00
void WriteTimeline( FileWrite& f, const Vector<ZoneEvent*>& vec );
void ReadTimeline( FileRead& f, Vector<ZoneEvent*>& vec );
2017-09-30 14:20:08 +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-30 14:58:02 +00:00
bool m_staticView;
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-10-28 19:50:06 +00:00
Vector<int64_t> m_frames;
Vector<ThreadData*> m_threads;
2017-10-13 01:36:59 +00:00
Vector<PlotData*> m_plots;
2017-10-14 12:28:04 +00:00
Vector<MessageData*> m_messages;
Vector<TextData*> m_textData;
2017-11-11 18:44:09 +00:00
Vector<GpuCtxData*> m_gpuData;
std::unordered_map<uint64_t, const char*> m_strings;
std::unordered_map<uint64_t, const char*> m_threadNames;
std::unordered_map<uint64_t, SourceLocation> m_sourceLocation;
std::vector<uint64_t> m_sourceLocationExpand;
std::map<uint32_t, LockMap> m_lockMap;
2017-09-25 21:38:49 +00:00
uint64_t m_zonesCnt;
2017-09-14 00:00:13 +00:00
Vector<const char*> m_stringData;
std::unordered_map<const char*, uint32_t, charutil::Hasher, charutil::Comparator> m_stringMap;
Vector<SourceLocation*> m_sourceLocationPayload;
std::unordered_map<SourceLocation*, uint32_t, SourceLocationHasher, SourceLocationComparator> m_sourceLocationPayloadMap;
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
2017-10-22 13:37:24 +00:00
std::unordered_map<uint64_t, std::vector<ZoneEvent*>> m_zoneStack;
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;
std::unordered_set<uint64_t> m_pendingSourceLocation;
2017-11-11 14:41:21 +00:00
std::unordered_map<uint64_t, StringLocation> m_pendingCustomStrings;
2017-09-21 23:15:58 +00:00
std::unordered_map<uint64_t, uint32_t> m_threadMap;
2017-10-13 01:36:59 +00:00
std::unordered_map<uint64_t, uint32_t> m_plotMap;
2017-11-10 18:41:37 +00:00
std::unordered_map<const char*, uint32_t, charutil::Hasher, charutil::Comparator> m_plotRev;
2017-10-13 01:36:59 +00:00
std::unordered_map<uint64_t, PlotData*> m_pendingPlots;
std::unordered_map<uint64_t, uint32_t> m_sourceLocationShrink;
std::unordered_map<uint64_t, int32_t> m_pendingSourceLocationPayload;
2017-10-22 13:36:22 +00:00
Slab<64*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-26 19:49:50 +00:00
int64_t m_zvStartNext;
int64_t m_zvEndNext;
2017-10-28 19:36:47 +00:00
int64_t m_delay;
int64_t m_resolution;
double m_timerMul;
std::string m_captureName;
2017-09-29 19:49:14 +00:00
int8_t m_lastCpu;
2017-10-28 19:50:06 +00:00
int m_zvHeight;
int m_zvScroll;
2017-10-12 20:27:17 +00:00
2017-10-22 13:37:24 +00:00
const ZoneEvent* m_zoneInfoWindow;
const ZoneEvent* m_zoneHighlight;
2017-10-08 21:03:38 +00:00
LockHighlight m_lockHighlight;
2017-10-14 13:47:06 +00:00
const MessageData* m_msgHighlight;
2017-10-13 11:32:23 +00:00
bool m_drawRegion;
2017-10-18 21:36:50 +00:00
int64_t m_regionStart;
int64_t m_regionEnd;
2017-10-13 11:32:23 +00:00
bool m_showOptions;
2017-10-14 12:36:30 +00:00
bool m_showMessages;
2017-10-13 11:32:23 +00:00
bool m_drawZones;
bool m_drawLocks;
2017-10-13 12:54:32 +00:00
bool m_drawPlots;
2017-10-22 11:32:27 +00:00
bool m_onlyContendedLocks;
2017-10-18 16:48:51 +00:00
2017-10-22 11:56:05 +00:00
Namespace m_namespace;
2017-10-18 16:48:51 +00:00
bool m_terminate;
2017-09-12 23:33:50 +00:00
};
}
#endif