diff --git a/.gitignore b/.gitignore index f18e0257..baa59397 100644 --- a/.gitignore +++ b/.gitignore @@ -5,6 +5,9 @@ x64 Release Debug +_build +_compiler +tools/* *.d *.o *.so @@ -28,3 +31,7 @@ vcpkg/* !vcpkg/install_vcpkg_dependencies.bat .deps/ .dirstamp +.vscode/ + +/_*/** +/**/__pycache__/** diff --git a/import-chrome/src/import-chrome.cpp b/import-chrome/src/import-chrome.cpp index a5186e4d..d17653a9 100644 --- a/import-chrome/src/import-chrome.cpp +++ b/import-chrome/src/import-chrome.cpp @@ -6,6 +6,7 @@ #include #include #include +#include #include "json.hpp" @@ -56,6 +57,7 @@ int main( int argc, char** argv ) std::vector timeline; std::vector messages; std::vector plots; + std::unordered_map threadNames; if( j.is_object() && j.contains( "traceEvents" ) ) { @@ -156,6 +158,13 @@ int main( int argc, char** argv ) } } } + else if (type == "M") + { + if (v.contains("name") && v["name"] == "thread_name" && v.contains("args") && v["args"].is_object() && v["args"].contains("name")) + { + threadNames[v["tid"].get()] = v["args"]["name"].get(); + } + } } std::stable_sort( timeline.begin(), timeline.end(), [] ( const auto& l, const auto& r ) { return l.timestamp < r.timestamp; } ); @@ -185,11 +194,15 @@ int main( int argc, char** argv ) printf( "\33[2KProcessing...\r" ); fflush( stdout ); - auto program = input; - while( *program ) program++; - program--; - while( program > input && ( *program != '/' || *program != '\\' ) ) program--; - tracy::Worker worker( program, timeline, messages, plots ); + auto&& getFilename = [](const char* in) { + auto out = in; + while (*out) ++out; + --out; + while (out > in && (*out != '/' || *out != '\\')) out--; + return out; + }; + + tracy::Worker worker( getFilename(output), getFilename(input), timeline, messages, plots, threadNames ); auto w = std::unique_ptr( tracy::FileWrite::Open( output, clev ) ); if( !w ) diff --git a/server/TracyWorker.cpp b/server/TracyWorker.cpp index f3e13bba..fde2f4d7 100644 --- a/server/TracyWorker.cpp +++ b/server/TracyWorker.cpp @@ -275,10 +275,11 @@ Worker::Worker( const char* addr, uint16_t port ) m_threadNet = std::thread( [this] { SetThreadName( "Tracy Network" ); Network(); } ); } -Worker::Worker( const std::string& program, const std::vector& timeline, const std::vector& messages, const std::vector& plots ) +Worker::Worker( const char* name, const char* program, const std::vector& timeline, const std::vector& messages, const std::vector& plots, const std::unordered_map& threadNames ) : m_hasData( true ) , m_delay( 0 ) , m_resolution( 0 ) + , m_captureName( name ) , m_captureProgram( program ) , m_captureTime( 0 ) , m_pid( 0 ) @@ -428,9 +429,19 @@ Worker::Worker( const std::string& program, const std::vectorsecond.c_str()); + AddThreadString(t.first, buf, len); + } + else + { + char buf[64]; + sprintf( buf, "%" PRIu64, t.first ); + AddThreadString( t.first, buf, strlen( buf ) ); + } } m_data.framesBase = m_data.frames.Retrieve( 0, [this] ( uint64_t name ) { @@ -512,6 +523,8 @@ Worker::Worker( FileRead& f, EventType::Type eventMask, bool bgTasks ) char tmp[1024]; f.Read( tmp, sz ); m_captureName = std::string( tmp, tmp+sz ); + if (m_captureName.empty()) + m_captureName = f.GetFilename(); } { f.Read( sz ); diff --git a/server/TracyWorker.hpp b/server/TracyWorker.hpp index 11443c6a..28b2403e 100644 --- a/server/TracyWorker.hpp +++ b/server/TracyWorker.hpp @@ -10,6 +10,7 @@ #include #include #include +#include #include #include "../common/TracyForceInline.hpp" @@ -394,7 +395,7 @@ public: }; Worker( const char* addr, uint16_t port ); - Worker( const std::string& program, const std::vector& timeline, const std::vector& messages, const std::vector& plots ); + Worker( const char* name, const char* program, const std::vector& timeline, const std::vector& messages, const std::vector& plots, const std::unordered_map& threadNames ); Worker( FileRead& f, EventType::Type eventMask = EventType::All, bool bgTasks = true ); ~Worker();