Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-06-11 02:03:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add save_trace MCP tool for snapshotting live or loaded captures.

Browse Source 
- save_worker binding: wraps Worker::Write under
  Worker::ObtainLockForMainThread() so live instances yield their
  receive thread cooperatively for the save's duration — the same
  pattern View::Save uses in the GUI.
- save_trace MCP tool: defaults to async_mode=True for multi-GB
  traces; reuses the existing Task/executor machinery so callers
  poll via the task tool. Path resolution mirrors load_capture.
- manual/tracy.tex: add save_trace bullet to the MCP tool list.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alan Tse
2026-05-19 01:23:31 -07:00
parent 1f1738c221
commit 33fe84532e
3 changed files with 132 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
python/bindings/ServerModule.cpp
View File

@@ -11,6 +11,7 @@
# pragma GCC diagnostic ignored "-Wnarrowing"
#endif
#include "../../server/TracyFileRead.hpp"
#include "../../server/TracyFileWrite.hpp"
#include "../../server/TracyWorker.hpp"
#ifdef _MSC_VER
# pragma warning( pop )
@@ -1084,6 +1085,32 @@ PYBIND11_MODULE( TracyServerBindings, m )
m.def( "create_worker_from_file", []( std::shared_ptr<FileRead> f ) {
return std::make_unique<Worker>( *f );
} );
// -------------------------------------------------------------------------
// FileWrite — snapshot a Worker (live or loaded) to a .tracy file.
//
// Mirrors capture.cpp and View::Save: open a Zstd FileWrite, call
// Worker::Write under the main-thread data lock (so a live receive thread
// yields cooperatively rather than racing with our reads), then Finish to
// drain the compression streams before returning the size pair.
//
// Defaults match the standalone capture tool: Zstd level 3, 4 streams,
// fiDict=false. Returns (uncompressed_bytes, compressed_bytes).
// -------------------------------------------------------------------------
m.def( "save_worker", []( Worker& w, const char* path, int level, int streams, bool fiDict ) {
auto f = std::unique_ptr<FileWrite>( FileWrite::Open( path, FileCompression::Zstd, level, streams ) );
if( !f ) throw std::runtime_error( "Could not open output file for writing" );
std::pair<size_t, size_t> stats;
{
py::gil_scoped_release release;
auto lock = w.ObtainLockForMainThread();
w.Write( *f, fiDict );
f->Finish();
stats = f->GetCompressionStatistics();
}
return py::make_tuple( stats.first, stats.second );
}, py::arg( "worker" ), py::arg( "path" ),
py::arg( "level" ) = 3, py::arg( "streams" ) = 4, py::arg( "fi_dict" ) = false );
}
} // namespace tracy