lili/ggwave
1
0
Fork 0
mirror of https://github.com/ggerganov/ggwave.git synced 2026-03-24 20:51:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

spectrogram : add simple color scaling control

Browse Source
This commit is contained in:
Georgi Gerganov
2021-02-13 08:10:04 +02:00
parent 2774040fdd
commit ff5c569071
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/spectrogram/main.cpp
View File

@@ -34,6 +34,8 @@ constexpr int g_nSamplesPerFrame = 1024;
int g_binMin = 40; int g_binMin = 40;
int g_binMax = 40 + 96; int g_binMax = 40 + 96;
float g_scale = 5.0;
bool g_showControls = true; bool g_showControls = true;
int g_freqDataHead = 0; int g_freqDataHead = 0;
@@ -449,7 +451,7 @@ int main(int argc, char** argv) {
if (k >= g_freqDataSize) k -= g_freqDataSize; if (k >= g_freqDataSize) k -= g_freqDataSize;
auto v = g_freqData[g_binMin + i].mag[k]; auto v = g_freqData[g_binMin + i].mag[k];
ImVec4 c = { 0.0f, 1.0f, 0.0, 0.0f }; ImVec4 c = { 0.0f, 1.0f, 0.0, 0.0f };
c.w = v/(5.0*sum); c.w = v/(g_scale*sum);
drawList->AddRectFilled({ p0.x + j*dx, p0.y + i*dy }, { p0.x + j*dx + dx - 1, p0.y + i*dy + dy - 1 }, ImGui::ColorConvertFloat4ToU32(c)); drawList->AddRectFilled({ p0.x + j*dx, p0.y + i*dy }, { p0.x + j*dx + dx - 1, p0.y + i*dy + dy - 1 }, ImGui::ColorConvertFloat4ToU32(c));
} }
} }
@@ -471,6 +473,7 @@ int main(int argc, char** argv) {
ImGui::Text("Press 'c' to hide/show this window"); ImGui::Text("Press 'c' to hide/show this window");
ImGui::DragInt("Min", &g_binMin, 1, 0, g_binMax - 1); ImGui::DragInt("Min", &g_binMin, 1, 0, g_binMax - 1);
ImGui::DragInt("Max", &g_binMax, 1, g_binMin + 1, g_nSamplesPerFrame/2); ImGui::DragInt("Max", &g_binMax, 1, g_binMin + 1, g_nSamplesPerFrame/2);
ImGui::DragFloat("Scale", &g_scale, 1.0f, 1.0f, 1000.0f);
if (ImGui::Button("Pause")) { if (ImGui::Button("Pause")) {
togglePause = true; togglePause = true;
} }