wip: major code refactor, split most gui from logic
parent
f6318e3284
commit
c76ba69fc9
@ -0,0 +1,72 @@
|
||||
/**
|
||||
* @file code.cpp
|
||||
* @brief Contains code for algorithm-code-related UI elements and logic.
|
||||
*
|
||||
* Copyright (C) 2021 Clyne Sullivan
|
||||
*
|
||||
* Distributed under the GNU GPL v3 or later. You should have received a copy of
|
||||
* the GNU General Public License along with this program.
|
||||
* If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "imgui.h"
|
||||
#include "backends/imgui_impl_sdl.h"
|
||||
#include "backends/imgui_impl_opengl2.h"
|
||||
#include "TextEditor.h"
|
||||
|
||||
#include "config.h"
|
||||
|
||||
#include <string>
|
||||
|
||||
extern void compileEditorCode(const std::string& code);
|
||||
extern void disassembleCode();
|
||||
|
||||
TextEditor editor; // file.cpp
|
||||
|
||||
static std::string editorCompiled;
|
||||
|
||||
static void codeCompile();
|
||||
static void codeDisassemble();
|
||||
|
||||
void codeEditorInit()
|
||||
{
|
||||
editor.SetLanguageDefinition(TextEditor::LanguageDefinition::CPlusPlus());
|
||||
editor.SetPalette(TextEditor::GetLightPalette());
|
||||
}
|
||||
|
||||
void codeRenderMenu()
|
||||
{
|
||||
if (ImGui::BeginMenu("Code")) {
|
||||
if (ImGui::MenuItem("Compile code"))
|
||||
codeCompile();
|
||||
if (ImGui::MenuItem("Show disassembly"))
|
||||
codeDisassemble();
|
||||
|
||||
ImGui::EndMenu();
|
||||
}
|
||||
}
|
||||
|
||||
void codeRenderToolbar()
|
||||
{
|
||||
if (ImGui::Button("Compile"))
|
||||
codeCompile();
|
||||
}
|
||||
|
||||
void codeRenderWidgets()
|
||||
{
|
||||
editor.Render("code", {WINDOW_WIDTH - 15, 450}, true);
|
||||
}
|
||||
|
||||
static void codeCompile()
|
||||
{
|
||||
compileEditorCode(editor.GetText());
|
||||
editorCompiled = editor.GetText().compare(editorCompiled);
|
||||
}
|
||||
|
||||
static void codeDisassemble()
|
||||
{
|
||||
if (editor.GetText().compare(editorCompiled) != 0)
|
||||
codeCompile();
|
||||
disassembleCode();
|
||||
}
|
||||
|
||||
@ -0,0 +1,340 @@
|
||||
#include "imgui.h"
|
||||
#include "imgui_internal.h"
|
||||
#include "ImGuiFileDialog.h"
|
||||
|
||||
#include "stmdsp.hpp"
|
||||
|
||||
#include <array>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
#include <string_view>
|
||||
|
||||
// Used for status queries and buffer size configuration.
|
||||
extern std::shared_ptr<stmdsp::device> m_device;
|
||||
|
||||
void deviceAlgorithmUnload();
|
||||
void deviceAlgorithmUpload();
|
||||
bool deviceConnect();
|
||||
void deviceGenLoadFormula(std::string_view list);
|
||||
void deviceGenLoadList(std::string_view list);
|
||||
bool deviceGenStartToggle();
|
||||
void deviceLoadAudioFile(const std::string& file);
|
||||
void deviceLoadLogFile(const std::string& file);
|
||||
void deviceSetSampleRate(unsigned int index);
|
||||
void deviceSetInputDrawing(bool enabled);
|
||||
void deviceStart(bool measureCodeTime, bool logResults, bool drawSamples);
|
||||
void deviceUpdateDrawBufferSize(double timeframe);
|
||||
void pullFromDrawQueue(
|
||||
std::vector<stmdsp::dacsample_t>& buffer,
|
||||
decltype(buffer.begin())& bufferCursor,
|
||||
double timeframe);
|
||||
void pullFromInputDrawQueue(
|
||||
std::vector<stmdsp::dacsample_t>& buffer,
|
||||
decltype(buffer.begin())& bufferCursor,
|
||||
double timeframe);
|
||||
|
||||
static std::string sampleRatePreview = "?";
|
||||
static bool measureCodeTime = false;
|
||||
static bool logResults = false;
|
||||
static bool drawSamples = false;
|
||||
static bool popupRequestBuffer = false;
|
||||
static bool popupRequestSiggen = false;
|
||||
static bool popupRequestLog = false;
|
||||
static double drawSamplesTimeframe = 1.0; // seconds
|
||||
|
||||
static std::string getSampleRatePreview(unsigned int rate)
|
||||
{
|
||||
return std::to_string(rate / 1000) + " kHz";
|
||||
}
|
||||
|
||||
void deviceRenderMenu()
|
||||
{
|
||||
auto addMenuItem = [](const std::string& label, bool enable, auto action) {
|
||||
if (ImGui::MenuItem(label.c_str(), nullptr, false, enable)) {
|
||||
action();
|
||||
}
|
||||
};
|
||||
|
||||
if (ImGui::BeginMenu("Run")) {
|
||||
const bool isConnected = m_device ? true : false;
|
||||
const bool isRunning = isConnected && m_device->is_running();
|
||||
|
||||
static std::string connectLabel ("Connect");
|
||||
addMenuItem(connectLabel, !isConnected || !isRunning, [&] {
|
||||
if (deviceConnect()) {
|
||||
connectLabel = "Disconnect";
|
||||
sampleRatePreview =
|
||||
getSampleRatePreview(m_device->get_sample_rate());
|
||||
deviceUpdateDrawBufferSize(drawSamplesTimeframe);
|
||||
} else {
|
||||
connectLabel = "Connect";
|
||||
}
|
||||
});
|
||||
|
||||
ImGui::Separator();
|
||||
|
||||
static std::string startLabel ("Start");
|
||||
addMenuItem(startLabel, isConnected, [&] {
|
||||
startLabel = isRunning ? "Start" : "Stop";
|
||||
deviceStart(measureCodeTime, logResults, drawSamples);
|
||||
if (logResults && isRunning)
|
||||
logResults = false;
|
||||
});
|
||||
addMenuItem("Upload algorithm", isConnected && !isRunning,
|
||||
deviceAlgorithmUpload);
|
||||
addMenuItem("Unload algorithm", isConnected && !isRunning,
|
||||
deviceAlgorithmUnload);
|
||||
|
||||
ImGui::Separator();
|
||||
if (!isConnected || isRunning)
|
||||
ImGui::PushDisabled();
|
||||
|
||||
ImGui::Checkbox("Measure Code Time", &measureCodeTime);
|
||||
ImGui::Checkbox("Draw samples", &drawSamples);
|
||||
if (ImGui::Checkbox("Log results...", &logResults)) {
|
||||
if (logResults)
|
||||
popupRequestLog = true;
|
||||
}
|
||||
|
||||
if (!isConnected || isRunning)
|
||||
ImGui::PopDisabled();
|
||||
|
||||
addMenuItem("Set buffer size...", isConnected && !isRunning,
|
||||
[] { popupRequestBuffer = true; });
|
||||
|
||||
ImGui::Separator();
|
||||
|
||||
addMenuItem("Load signal generator",
|
||||
isConnected && !m_device->is_siggening(),
|
||||
[] { popupRequestSiggen = true; });
|
||||
|
||||
static std::string startSiggenLabel ("Start signal generator");
|
||||
addMenuItem(startSiggenLabel, isConnected, [&] {
|
||||
const bool running = deviceGenStartToggle();
|
||||
startSiggenLabel = running ? "Stop signal generator"
|
||||
: "Start signal generator";
|
||||
});
|
||||
|
||||
ImGui::EndMenu();
|
||||
}
|
||||
}
|
||||
|
||||
void deviceRenderToolbar()
|
||||
{
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button("Upload"))
|
||||
deviceAlgorithmUpload();
|
||||
ImGui::SameLine();
|
||||
ImGui::SetNextItemWidth(100);
|
||||
|
||||
const bool enable =
|
||||
m_device && !m_device->is_running() && !m_device->is_siggening();
|
||||
if (!enable)
|
||||
ImGui::PushDisabled();
|
||||
|
||||
if (ImGui::BeginCombo("", sampleRatePreview.c_str())) {
|
||||
extern std::array<unsigned int, 6> sampleRateInts;
|
||||
|
||||
for (const auto& r : sampleRateInts) {
|
||||
const auto s = getSampleRatePreview(r);
|
||||
if (ImGui::Selectable(s.c_str())) {
|
||||
sampleRatePreview = s;
|
||||
deviceSetSampleRate(r);
|
||||
deviceUpdateDrawBufferSize(drawSamplesTimeframe);
|
||||
}
|
||||
}
|
||||
|
||||
ImGui::EndCombo();
|
||||
}
|
||||
|
||||
if (!enable)
|
||||
ImGui::PopDisabled();
|
||||
}
|
||||
|
||||
void deviceRenderWidgets()
|
||||
{
|
||||
static std::string siggenInput;
|
||||
static int siggenOption = 0;
|
||||
|
||||
if (popupRequestSiggen) {
|
||||
popupRequestSiggen = false;
|
||||
siggenInput.clear();
|
||||
ImGui::OpenPopup("siggen");
|
||||
} else if (popupRequestBuffer) {
|
||||
popupRequestBuffer = false;
|
||||
ImGui::OpenPopup("buffer");
|
||||
} else if (popupRequestLog) {
|
||||
popupRequestLog = false;
|
||||
ImGuiFileDialog::Instance()->OpenModal(
|
||||
"ChooseFileLogGen", "Choose File", ".csv", ".");
|
||||
}
|
||||
|
||||
if (ImGui::BeginPopup("siggen")) {
|
||||
if (ImGui::RadioButton("List", &siggenOption, 0))
|
||||
siggenInput.clear();
|
||||
ImGui::SameLine();
|
||||
if (ImGui::RadioButton("Formula", &siggenOption, 1))
|
||||
siggenInput.clear();
|
||||
ImGui::SameLine();
|
||||
if (ImGui::RadioButton("Audio File", &siggenOption, 2))
|
||||
siggenInput.clear();
|
||||
|
||||
if (siggenOption == 2) {
|
||||
if (ImGui::Button("Choose File")) {
|
||||
// This dialog will override the siggen popup, closing it.
|
||||
ImGuiFileDialog::Instance()->OpenModal(
|
||||
"ChooseFileLogGen", "Choose File", ".wav", ".");
|
||||
}
|
||||
} else {
|
||||
ImGui::Text(siggenOption == 0 ? "Enter a list of numbers:"
|
||||
: "Enter a formula. f(x) = ");
|
||||
ImGui::PushStyleColor(ImGuiCol_FrameBg, {.8, .8, .8, 1});
|
||||
ImGui::InputText("", siggenInput.data(), siggenInput.size());
|
||||
ImGui::PopStyleColor();
|
||||
}
|
||||
|
||||
if (ImGui::Button("Cancel")) {
|
||||
siggenInput.clear();
|
||||
ImGui::CloseCurrentPopup();
|
||||
}
|
||||
|
||||
if (ImGui::Button("Save")) {
|
||||
switch (siggenOption) {
|
||||
case 0:
|
||||
deviceGenLoadList(siggenInput);
|
||||
break;
|
||||
case 1:
|
||||
deviceGenLoadFormula(siggenInput);
|
||||
break;
|
||||
case 2:
|
||||
break;
|
||||
}
|
||||
|
||||
ImGui::CloseCurrentPopup();
|
||||
}
|
||||
|
||||
ImGui::EndPopup();
|
||||
}
|
||||
|
||||
if (ImGui::BeginPopup("buffer")) {
|
||||
static std::string bufferSizeInput ("4096");
|
||||
ImGui::Text("Please enter a new sample buffer size (100-4096):");
|
||||
ImGui::PushStyleColor(ImGuiCol_FrameBg, {.8, .8, .8, 1});
|
||||
ImGui::InputText("",
|
||||
bufferSizeInput.data(),
|
||||
bufferSizeInput.size(),
|
||||
ImGuiInputTextFlags_CharsDecimal);
|
||||
ImGui::PopStyleColor();
|
||||
if (ImGui::Button("Save")) {
|
||||
if (m_device) {
|
||||
int n = std::clamp(std::stoi(bufferSizeInput), 100, 4096);
|
||||
m_device->continuous_set_buffer_size(n);
|
||||
}
|
||||
ImGui::CloseCurrentPopup();
|
||||
}
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button("Cancel"))
|
||||
ImGui::CloseCurrentPopup();
|
||||
ImGui::EndPopup();
|
||||
}
|
||||
|
||||
if (ImGuiFileDialog::Instance()->Display("ChooseFileLogGen",
|
||||
ImGuiWindowFlags_NoCollapse,
|
||||
ImVec2(460, 540)))
|
||||
{
|
||||
if (ImGuiFileDialog::Instance()->IsOk()) {
|
||||
auto filePathName = ImGuiFileDialog::Instance()->GetFilePathName();
|
||||
auto ext = filePathName.substr(filePathName.size() - 4);
|
||||
|
||||
if (ext.compare(".wav") == 0)
|
||||
deviceLoadAudioFile(filePathName);
|
||||
else if (ext.compare(".csv") == 0)
|
||||
deviceLoadLogFile(filePathName);
|
||||
}
|
||||
|
||||
ImGuiFileDialog::Instance()->Close();
|
||||
}
|
||||
}
|
||||
|
||||
void deviceRenderDraw()
|
||||
{
|
||||
if (drawSamples) {
|
||||
static std::vector<stmdsp::dacsample_t> buffer;
|
||||
static decltype(buffer.begin()) bufferCursor;
|
||||
static std::vector<stmdsp::dacsample_t> bufferInput;
|
||||
static decltype(bufferInput.begin()) bufferInputCursor;
|
||||
|
||||
static bool drawSamplesInput = false;
|
||||
static unsigned int yMinMax = 4095;
|
||||
|
||||
ImGui::Begin("draw", &drawSamples);
|
||||
ImGui::Text("Draw input ");
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Checkbox("", &drawSamplesInput))
|
||||
deviceSetInputDrawing(drawSamplesInput);
|
||||
ImGui::SameLine();
|
||||
ImGui::Text("Time: %0.3f sec", drawSamplesTimeframe);
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button("-", {30, 0})) {
|
||||
drawSamplesTimeframe = std::max(drawSamplesTimeframe / 2., 0.0078125);
|
||||
deviceUpdateDrawBufferSize(drawSamplesTimeframe);
|
||||
}
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button("+", {30, 0})) {
|
||||
drawSamplesTimeframe = std::min(drawSamplesTimeframe * 2, 32.);
|
||||
deviceUpdateDrawBufferSize(drawSamplesTimeframe);
|
||||
}
|
||||
ImGui::SameLine();
|
||||
ImGui::Text("Y: +/-%1.2fV", 3.3f * (static_cast<float>(yMinMax) / 4095.f));
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button(" - ", {30, 0})) {
|
||||
yMinMax = std::max(63u, yMinMax >> 1);
|
||||
}
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button(" + ", {30, 0})) {
|
||||
yMinMax = std::min(4095u, (yMinMax << 1) | 1);
|
||||
}
|
||||
|
||||
pullFromDrawQueue(buffer, bufferCursor, drawSamplesTimeframe);
|
||||
if (drawSamplesInput)
|
||||
pullFromInputDrawQueue(bufferInput, bufferInputCursor, drawSamplesTimeframe);
|
||||
|
||||
auto drawList = ImGui::GetWindowDrawList();
|
||||
ImVec2 p0 = ImGui::GetWindowPos();
|
||||
auto size = ImGui::GetWindowSize();
|
||||
p0.y += 65;
|
||||
size.y -= 70;
|
||||
drawList->AddRectFilled(p0, {p0.x + size.x, p0.y + size.y}, IM_COL32(0, 0, 0, 255));
|
||||
|
||||
const float di = static_cast<float>(buffer.size()) / size.x;
|
||||
const float dx = std::ceil(size.x / static_cast<float>(buffer.size()));
|
||||
ImVec2 pp = p0;
|
||||
float i = 0;
|
||||
while (pp.x < p0.x + size.x) {
|
||||
unsigned int idx = i;
|
||||
float n = std::clamp((buffer[idx] - 2048.) / yMinMax, -0.5, 0.5);
|
||||
i += di;
|
||||
|
||||
ImVec2 next (pp.x + dx, p0.y + size.y * (0.5 - n));
|
||||
drawList->AddLine(pp, next, ImGui::GetColorU32(IM_COL32(255, 0, 0, 255)));
|
||||
pp = next;
|
||||
}
|
||||
|
||||
if (drawSamplesInput) {
|
||||
ImVec2 pp = p0;
|
||||
float i = 0;
|
||||
while (pp.x < p0.x + size.x) {
|
||||
unsigned int idx = i;
|
||||
float n = std::clamp((bufferInput[idx] - 2048.) / yMinMax, -0.5, 0.5);
|
||||
i += di;
|
||||
|
||||
ImVec2 next (pp.x + dx, p0.y + size.y * (0.5 - n));
|
||||
drawList->AddLine(pp, next, ImGui::GetColorU32(IM_COL32(0, 0, 255, 255)));
|
||||
pp = next;
|
||||
}
|
||||
}
|
||||
|
||||
ImGui::End();
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue