255 lines
9.6 KiB
C++
255 lines
9.6 KiB
C++
/**
|
|
* Furnace Tracker - multi-system chiptune tracker
|
|
* Copyright (C) 2021-2026 tildearrow and contributors
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
// bulk of the code is from https://github.com/Eknous-P/unscope/blob/spectrum/src/gui/wins/spectrum.cpp
|
|
|
|
#include "gui.h"
|
|
#include "imgui_internal.h"
|
|
#include "IconsFontAwesome4.h"
|
|
#include <fftw3.h>
|
|
#include <math.h> // fmod
|
|
|
|
inline float scaleFuncLog(float x) {
|
|
constexpr float base=100;
|
|
return log((base-1)*x+1.0f)/log(base);
|
|
}
|
|
|
|
inline float scaleFuncDb(float y) {
|
|
return log10(y)*20.0f/70.0f+1;
|
|
}
|
|
|
|
void FurnaceGUI::drawSpectrum() {
|
|
if (nextWindow==GUI_WINDOW_SPECTRUM) {
|
|
spectrumOpen=true;
|
|
ImGui::SetNextWindowFocus();
|
|
nextWindow=GUI_WINDOW_NOTHING;
|
|
}
|
|
if (!spectrumOpen) return;
|
|
if (ImGui::Begin("Spectrum",&spectrumOpen,globalWinFlags,_("Spectrum"))) {
|
|
ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding,ImVec2(0.0f,0.0f));
|
|
ImDrawList* dl=ImGui::GetWindowDrawList();
|
|
ImVec2 origin=ImGui::GetWindowPos(), size=ImGui::GetWindowSize();
|
|
float titleBar=ImGui::GetCurrentWindow()->TitleBarHeight;
|
|
origin.y+=titleBar;
|
|
size.y-=titleBar;
|
|
// x scale labels (precalc)
|
|
if (spectrum.showXScale) size.y-=ImGui::GetFontSize();
|
|
// y scale labels
|
|
if (spectrum.showYScale) {
|
|
float padding=ImGui::GetStyle().FramePadding.x;
|
|
ImVec2 p1, textSize;
|
|
char buf[16];
|
|
textSize=ImGui::CalcTextSize("-100");
|
|
textSize.x+=5.f;
|
|
int lineOffset=spectrum.yOffset*8;
|
|
for (int z=lineOffset; z<=7+lineOffset; z++) {
|
|
p1.y=origin.y+size.y*((z+1)/8.f-spectrum.yOffset)-textSize.y/2.0f;
|
|
p1.x=origin.x+padding;
|
|
snprintf(buf,16,"-%2d",(z+1)*10);
|
|
dl->AddText(p1,ImGui::GetColorU32(ImGuiCol_Text),buf);
|
|
}
|
|
origin.x+=textSize.x+padding;
|
|
size.x-=textSize.x+padding;
|
|
}
|
|
// y grid
|
|
if (spectrum.showYGrid) {
|
|
int lines=((size.y>450)?16:8);
|
|
float offset=fmod(spectrum.yOffset*lines,1.0f)/lines;
|
|
for (unsigned char z=1; z<=lines; z++) {
|
|
dl->AddLine(
|
|
origin+ImVec2(0,size.y*((float)z/lines-offset)),
|
|
origin+ImVec2(size.x,size.y*((float)z/lines-offset)),
|
|
0x55ffffff,
|
|
dpiScale);
|
|
}
|
|
}
|
|
// x grid
|
|
if (spectrum.showXGrid || spectrum.showXScale) {
|
|
char buf[16];
|
|
float pos=0, prevPos=0, i;
|
|
for (size_t j=0; j<spectrum.frequencies.size(); j++) {
|
|
i=spectrum.frequencies[j];
|
|
ImU32 color=0x22ffffff;
|
|
pos=spectrum.xZoom*size.x*(scaleFuncLog(i/(e->getAudioDescGot().rate/2.0f))-spectrum.xOffset);
|
|
if (pos>size.x) break;
|
|
if ((j%9)==0) color=0x55ffffff;
|
|
if (spectrum.showXScale) {
|
|
if (i>=1000) {
|
|
snprintf(buf,16,"%dk",(int)i/1000);
|
|
} else {
|
|
snprintf(buf,16,"%d",(int)i);
|
|
}
|
|
float x=ImGui::CalcTextSize(buf).x;
|
|
if (pos-prevPos>x || (j%9)==0) dl->AddText(
|
|
origin+ImVec2(pos-x/2,size.y),
|
|
ImGui::GetColorU32(ImGuiCol_Text),
|
|
buf
|
|
);
|
|
}
|
|
if (spectrum.showXGrid) dl->AddLine(
|
|
origin+ImVec2(pos,0),
|
|
origin+ImVec2(pos,size.y),
|
|
color,
|
|
dpiScale);
|
|
prevPos=pos;
|
|
}
|
|
}
|
|
int chans=e->getAudioDescGot().outChans;
|
|
if (spectrum.update) {
|
|
spectrum.update=false;
|
|
spectrum.running=true;
|
|
for (int i=0; i<DIV_MAX_OUTPUTS; i++) {
|
|
if (spectrum.buffer[i]) {
|
|
fftw_free(spectrum.buffer[i]);
|
|
spectrum.buffer[i]=NULL;
|
|
}
|
|
if (spectrum.in[i]) {
|
|
delete[] spectrum.in[i];
|
|
spectrum.in[i]=NULL;
|
|
}
|
|
if (spectrum.plan[i]) {
|
|
fftw_free(spectrum.plan[i]);
|
|
spectrum.plan[i]=NULL;
|
|
}
|
|
if (spectrum.plot[i]) {
|
|
delete[] spectrum.plot[i];
|
|
spectrum.plot[i]=NULL;
|
|
}
|
|
}
|
|
for (int i=0; i<(spectrum.mono?1:chans); i++) {
|
|
spectrum.buffer[i]=(fftw_complex*)fftw_malloc(sizeof(fftw_complex)*spectrum.bins);
|
|
if (!spectrum.buffer[i]) spectrum.running=false;
|
|
spectrum.in[i]=new double[spectrum.bins];
|
|
if (!spectrum.in[i]) spectrum.running=false;
|
|
spectrum.plan[i]=fftw_plan_dft_r2c_1d(spectrum.bins,spectrum.in[i],spectrum.buffer[i],FFTW_ESTIMATE);
|
|
if (!spectrum.plan[i]) spectrum.running=false;
|
|
spectrum.plot[i]=new ImVec2[spectrum.bins/2];
|
|
if (!spectrum.plot[i]) spectrum.running=false;
|
|
}
|
|
spectrum.frequencies.clear();
|
|
float freq;
|
|
float maxRate=e->getAudioDescGot().rate/2;
|
|
for (int j=10; j<maxRate; j*=10) {
|
|
for (int i=1; i<10; i++) {
|
|
freq=i*j;
|
|
if (freq>maxRate) break;
|
|
spectrum.frequencies.push_back((int)freq);
|
|
}
|
|
if (freq>maxRate) break;
|
|
}
|
|
if (spectrum.running) logV("spectrum ready!");
|
|
}
|
|
if (spectrum.running) {
|
|
for (int z=spectrum.mono?0:(chans-1); z>=0; z--) {
|
|
if (!spectrum.buffer[z]) {
|
|
spectrum.buffer[z]=(fftw_complex*)fftw_malloc(sizeof(fftw_complex)*spectrum.bins);
|
|
if (!spectrum.buffer[z]) spectrum.running=false;
|
|
}
|
|
if (!spectrum.in[z]) {
|
|
spectrum.in[z]=new double[spectrum.bins];
|
|
if (!spectrum.in[z]) spectrum.running=false;
|
|
}
|
|
if (!spectrum.plan[z]) {
|
|
spectrum.plan[z]=fftw_plan_dft_r2c_1d(spectrum.bins,spectrum.in[z],spectrum.buffer[z],FFTW_ESTIMATE);
|
|
if (!spectrum.plan[z]) spectrum.running=false;
|
|
}
|
|
if (!spectrum.plot[z]) {
|
|
spectrum.plot[z]=new ImVec2[spectrum.bins/2];
|
|
if (!spectrum.plot[z]) spectrum.running=false;
|
|
}
|
|
if (!spectrum.running) {
|
|
logE("oh no what why");
|
|
break;
|
|
}
|
|
// get buffer
|
|
memset(spectrum.in[z],0,sizeof(double)*spectrum.bins);
|
|
int needle=e->oscReadPos-spectrum.bins;
|
|
|
|
for (int j=0; j<spectrum.bins; j++) {
|
|
int pos=(needle+j)&0x7fff;
|
|
double sample=0.0;
|
|
if (spectrum.mono) {
|
|
for (int i=0; i<chans; i++) {
|
|
sample+=e->oscBuf[i][pos];
|
|
}
|
|
sample/=chans;
|
|
} else {
|
|
sample+=e->oscBuf[z][pos];
|
|
}
|
|
spectrum.in[z][j]=sample*(0.5*(1.0-cos(2.0*M_PI*j/(spectrum.bins-1))));
|
|
}
|
|
fftw_execute(spectrum.plan[z]);
|
|
unsigned int count=0;
|
|
double mag=0.0f;
|
|
float x=0.0f, y=0.0f;
|
|
count=spectrum.bins/2;
|
|
for (unsigned int i=0; i<count; i++) {
|
|
x=spectrum.xZoom*size.x*(scaleFuncLog((float)i/count)-spectrum.xOffset);
|
|
mag=2.0f*sqrt(spectrum.buffer[z][i][0]*spectrum.buffer[z][i][0]+spectrum.buffer[z][i][1]*spectrum.buffer[z][i][1])/count;
|
|
y=1.0-scaleFuncDb(mag);
|
|
spectrum.plot[z][i].x=origin.x+x;
|
|
spectrum.plot[z][i].y=origin.y+size.y*(y-spectrum.yOffset);
|
|
}
|
|
ImGui::PushClipRect(origin,origin+size,true);
|
|
dl->AddPolyline(spectrum.plot[z],count,ImGui::GetColorU32(uiColors[spectrum.mono?GUI_COLOR_OSC_WAVE:GUI_COLOR_OSC_WAVE_CH0+z]),0,dpiScale);
|
|
dl->PathFillConcave(ImGui::GetColorU32(uiColors[spectrum.mono?GUI_COLOR_OSC_WAVE:GUI_COLOR_OSC_WAVE_CH0+z]));
|
|
ImGui::PopClipRect();
|
|
}
|
|
}
|
|
ImGui::PopStyleVar();
|
|
if (ImGui::IsWindowHovered()) {
|
|
ImGui::SetCursorPosX(ImGui::GetWindowSize().x-ImGui::GetStyle().ItemSpacing.x-ImGui::CalcTextSize(ICON_FA_BARS).x);
|
|
ImGui::TextUnformatted(ICON_FA_BARS "##spectrumSettings");
|
|
if (ImGui::IsItemHovered(ImGuiHoveredFlags_DelayShort)) {
|
|
ImGui::SetTooltip(_("Spectrum settings"));
|
|
}
|
|
}
|
|
if (ImGui::BeginPopupContextItem("spectrumSettingsPopup",ImGuiPopupFlags_MouseButtonLeft)) {
|
|
if (ImGui::Checkbox(_("Mono##spec"),&spectrum.mono)) {
|
|
spectrum.update=true;
|
|
}
|
|
if (ImGui::InputScalar(_("Bins##spec"),ImGuiDataType_U32,&spectrum.bins)) {
|
|
if (spectrum.bins<32) spectrum.bins=32;
|
|
if (spectrum.bins>32768) spectrum.bins=32768;
|
|
spectrum.update=true;
|
|
}
|
|
ImGui::Separator();
|
|
if (ImGui::SliderFloat(("X Zoom##spec"),&spectrum.xZoom,1.0f,10.0f)) {
|
|
if (spectrum.xZoom<1.0f) spectrum.xZoom=1.0f;
|
|
if (spectrum.xZoom>10.0f) spectrum.xZoom=10.0f;
|
|
}
|
|
if (ImGui::SliderFloat(_("X Offset##spec"),&spectrum.xOffset,0.0f,1.0f)) {
|
|
if (spectrum.xOffset<0.0f) spectrum.xOffset=0.0f;
|
|
if (spectrum.xOffset>1.0f) spectrum.xOffset=1.0f;
|
|
}
|
|
ImGui::Separator();
|
|
if (ImGui::SliderFloat(_("Y Offset##spec"),&spectrum.yOffset,0.0f,1.0f)) {
|
|
if (spectrum.yOffset<0.0f) spectrum.yOffset=0.0f;
|
|
if (spectrum.yOffset>1.0f) spectrum.yOffset=1.0f;
|
|
}
|
|
ImGui::Separator();
|
|
ImGui::Checkbox(_("Show X Grid##spec"),&spectrum.showXGrid);
|
|
ImGui::Checkbox(_("Show Y Grid##spec"),&spectrum.showYGrid);
|
|
ImGui::Checkbox(_("Show X Scale##spec"),&spectrum.showXScale);
|
|
ImGui::Checkbox(_("Show Y Scale##spec"),&spectrum.showYScale);
|
|
}
|
|
}
|
|
if (ImGui::IsWindowFocused(ImGuiFocusedFlags_ChildWindows)) curWindow=GUI_WINDOW_SPECTRUM;
|
|
ImGui::End();
|
|
}
|