1 files changed, 77 insertions, 0 deletions
diff --git a/src/ui/ExportUtils.h b/src/ui/ExportUtils.h
new file mode 100644
index 0000000..d655d08
--- /dev/null
+++ b/src/ui/ExportUtils.h
@@ -0,0 +1,77 @@
+#pragma once
+
+#ifdef BUILD_UI
+
+#include <cstdint>
+#include <cstdio>
+#include <filesystem>
+#include <string>
+#include <unordered_map>
+
+#include "imgui.h"
+
+namespace UI {
+
+// Exports land under torch-exports/ mirroring the asset path.
+inline std::filesystem::path ExportFilePath(const std::string& assetName, const char* ext) {
+    std::filesystem::path path = std::filesystem::path("torch-exports") / (assetName + "." + ext);
+    std::error_code ec;
+    std::filesystem::create_directories(path.parent_path(), ec);
+    return path;
+}
+
+inline bool WriteWavFile(const std::filesystem::path& path, const int16_t* samples, size_t frames, int channels,
+                         int rate) {
+    FILE* f = fopen(path.string().c_str(), "wb");
+    if (f == nullptr) {
+        return false;
+    }
+    const uint32_t dataSize = (uint32_t)(frames * channels * 2);
+    const uint32_t riffSize = 36 + dataSize;
+    const uint16_t fmt = 1, ch = (uint16_t)channels, bits = 16;
+    const uint16_t align = (uint16_t)(channels * 2);
+    const uint32_t rate32 = (uint32_t)rate, byteRate = rate32 * align, fmtSize = 16;
+    fwrite("RIFF", 1, 4, f);
+    fwrite(&riffSize, 4, 1, f);
+    fwrite("WAVE", 1, 4, f);
+    fwrite("fmt ", 1, 4, f);
+    fwrite(&fmtSize, 4, 1, f);
+    fwrite(&fmt, 2, 1, f);
+    fwrite(&ch, 2, 1, f);
+    fwrite(&rate32, 4, 1, f);
+    fwrite(&byteRate, 4, 1, f);
+    fwrite(&align, 2, 1, f);
+    fwrite(&bits, 2, 1, f);
+    fwrite("data", 1, 4, f);
+    fwrite(&dataSize, 4, 1, f);
+    fwrite(samples, 2, frames * channels, f);
+    fclose(f);
+    return true;
+}
+
+// Last export result per asset; shown as a "saved" marker with the full
+// path in the tooltip.
+inline std::unordered_map<std::string, std::string>& ExportResults() {
+    static std::unordered_map<std::string, std::string> results;
+    return results;
+}
+
+inline void NoteExport(const std::string& assetName, const std::string& result) {
+    ExportResults()[assetName] = result;
+}
+
+inline void DrawExportMarker(const std::string& assetName) {
+    const auto it = ExportResults().find(assetName);
+    if (it == ExportResults().end()) {
+        return;
+    }
+    ImGui::SameLine();
+    ImGui::TextDisabled("saved");
+    if (ImGui::IsItemHovered()) {
+        ImGui::SetTooltip("%s", it->second.c_str());
+    }
+}
+
+} // namespace UI
+
+#endif // BUILD_UI