1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
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
|
