1 files changed, 37 insertions, 0 deletions

diff --git a/infra/pipetrace.h b/infra/pipetrace.h
new file mode 100644
index 0000000..656b9b9
--- /dev/null
+++ b/infra/pipetrace.h
@@ -0,0 +1,37 @@
+#pragma once
+
+#include <cstdint>
+#include <fmt/format.h>
+#include <ostream>
+#include <string>
+
+namespace infra {
+    struct transaction {
+        std::uint64_t record = ~(std::uint64_t)0;
+    };
+
+    struct pt {
+        static std::ostream *ptfile;
+
+        static std::uint64_t next_record;
+
+        static transaction toplevel() {
+            transaction t;
+            t.record = next_record++;
+            return t;
+        }
+
+        static transaction child(const transaction &p) {
+            transaction t;
+            t.record = next_record++;
+            if (ptfile)
+                *ptfile << fmt::format("{} parent {}\n", t.record, p.record);
+            return t;
+        }
+
+        static void event(const transaction &t, const char *event, std::uint64_t time, const std::string &data) {
+            if (ptfile)
+                *ptfile << fmt::format("@{} {} {} {}\n", time, t.record, event, data);
+        }
+    };
+}