1 files changed, 125 insertions, 0 deletions
diff --git a/source/device.cpp b/source/device.cpp
new file mode 100644
index 0000000..1dcacf0
--- /dev/null
+++ b/source/device.cpp
@@ -0,0 +1,125 @@
+/**
+ * @file device.cpp
+ * @brief Contains code for device-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 "stmdsp.hpp"
+
+#include "imgui.h"
+
+extern stmdsp::device *m_device;
+extern std::string statusMessage;
+
+static const char *sampleRateList[6] = {
+    "8 kHz",
+    "16 kHz",
+    "20 kHz",
+    "32 kHz",
+    "48 kHz",
+    "96 kHz"
+};
+static const char *sampleRatePreview = sampleRateList[0];
+
+static void deviceConnect();
+static void deviceStart();
+
+void deviceRenderMenu()
+{
+    if (ImGui::BeginMenu("Run")) {
+        static const char *connectLabel = "Connect";
+        if (ImGui::MenuItem(connectLabel)) {
+            deviceConnect();
+            connectLabel = m_device == nullptr ? "Connect" : "Disconnect";
+        }
+
+        static const char *startLabel = "Start";
+        if (ImGui::MenuItem(startLabel)) {
+            deviceStart();
+            startLabel = m_device != nullptr && m_device->is_running() ? "Stop" : "Start";
+        }
+
+/**
+TODO: Run menu:
+measure
+draw
+log
+upload
+unload
+buffer size
+load siggen
+run siggen 
+ */
+        ImGui::MenuItem("Upload algorithm");
+        ImGui::MenuItem("Unload algorithm");
+        ImGui::MenuItem("Measure Code Time");
+        ImGui::MenuItem("Draw samples");
+        ImGui::MenuItem("Log results...");
+        ImGui::MenuItem("Load signal generator");
+        ImGui::MenuItem("Start signal generator");
+
+        ImGui::EndMenu();
+    }
+}
+
+void deviceRenderToolbar()
+{
+    ImGui::SameLine();
+    ImGui::SetNextItemWidth(100);
+    if (ImGui::BeginCombo("", sampleRatePreview)) {
+        for (int i = 0; i < 6; ++i) {
+            if (ImGui::Selectable(sampleRateList[i])) {
+                sampleRatePreview = sampleRateList[i];
+                if (m_device != nullptr && !m_device->is_running())
+                    m_device->set_sample_rate(i);
+            }
+        }
+        ImGui::EndCombo();
+    }
+}
+
+void deviceConnect()
+{
+    if (m_device == nullptr) {
+        stmdsp::scanner scanner;
+        if (auto devices = scanner.scan(); devices.size() > 0) {
+            m_device = new stmdsp::device(devices.front());
+            if (m_device->connected()) {
+                sampleRatePreview = sampleRateList[m_device->get_sample_rate()];
+                statusMessage = "Connected!";
+            } else {
+                delete m_device;
+                m_device = nullptr;
+                statusMessage = "Failed to connect.";
+            }
+        } else {
+            statusMessage = "No devices found.";
+        }
+    } else {
+        delete m_device;
+        m_device = nullptr;
+        statusMessage = "Disconnected.";
+    }
+}
+
+void deviceStart()
+{
+    if (m_device == nullptr) {
+        statusMessage = "No device connected.";
+        return;
+    }
+
+    if (m_device->is_running()) {
+        m_device->continuous_stop();
+        statusMessage = "Ready.";
+    } else {
+        m_device->continuous_start();
+        statusMessage = "Running.";
+    }
+}
+