1 files changed, 141 insertions, 2 deletions

diff --git a/gui/wxmain.cpp b/gui/wxmain.cpp
index 847c77a..c2a378a 100644
--- a/gui/wxmain.cpp
+++ b/gui/wxmain.cpp
@@ -1,13 +1,152 @@
 #include "wxmain.hpp"
 
 #include <wx/filename.h>
+#include <wx/sizer.h>
+
+MainFrame::MainFrame() : wxFrame(nullptr, -1, "Hello world", wxPoint(50, 50), wxSize(640, 800))
+{
+
+    auto window = new wxBoxSizer(wxVERTICAL);
+
+    auto toolbar = new wxBoxSizer(wxHORIZONTAL);
+    toolbar->Add(new wxButton(this, Id::Single, "Run"), 0, wxALL, 10);
+    toolbar->Add(new wxButton(this, Id::UploadFilter, "Upload Filter"), 0, wxALL, 10);
+    toolbar->AddStretchSpacer(2);
+    toolbar->Add(new wxButton(this, Id::ConnectDevice, "Connect"), 0, wxALL, 10);
+    window->Add(toolbar, 0, wxALL | wxEXPAND);
+
+    m_text_editor = new wxStyledTextCtrl(this, wxID_ANY, wxDefaultPosition, wxSize(600, 420));
+    prepareEditor();
+    window->Add(m_text_editor, 1, wxEXPAND | wxALL, 10);
+
+    m_signal_area = new wxControl(this, wxID_ANY, wxDefaultPosition, wxSize(600, 200));
+    window->Add(m_signal_area, 1, wxEXPAND | wxALL, 10);
+
+    SetSizerAndFit(window);
+
+    m_render_timer = new wxTimer(this, Id::RenderTimer);
+
+    Bind(wxEVT_BUTTON, &MainFrame::onSinglePressed, this, Id::Single);
+    Bind(wxEVT_BUTTON, &MainFrame::onUploadPressed, this, Id::UploadFilter);
+    Bind(wxEVT_BUTTON, &MainFrame::onConnectPressed, this, Id::ConnectDevice);
+    Bind(wxEVT_PAINT, &MainFrame::onPaint, this, wxID_ANY);
+    Bind(wxEVT_TIMER, &MainFrame::onRenderTimer, this, Id::RenderTimer);
+}
+
+void MainFrame::onPaint([[maybe_unused]] wxPaintEvent& pe)
+{
+    auto *dc = new wxClientDC(this);
+    auto region = m_signal_area->GetRect();
+    dc->SetClippingRegion(region);
+
+    dc->SetBrush(*wxBLACK_BRUSH);
+    dc->SetPen(*wxBLACK_PEN);
+    dc->DrawRectangle(region);
+
+    if (m_device_samples.size() > 0) {
+        dc->SetPen(*wxRED_PEN);
+        auto points = new wxPoint[m_device_samples.size()];
+        const float spacing = static_cast<float>(region.GetWidth()) / m_device_samples.size();
+        float x = 0;
+        for (auto ptr = points; auto sample : m_device_samples) {
+            *ptr++ = wxPoint {
+                static_cast<int>(x),
+                region.GetHeight() - sample * region.GetHeight() / 4096
+            };
+            x += spacing;
+        }
+        dc->DrawLines(m_device_samples.size(), points, region.GetX(), region.GetY());
+        delete[] points;
+    }
+}
+
+void MainFrame::onConnectPressed(wxCommandEvent& ce)
+{
+    auto button = dynamic_cast<wxButton *>(ce.GetEventObject());
+
+    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()) {
+                button->SetLabel("Disconnect");
+            } else {
+                delete m_device;
+                m_device = nullptr;
+                button->SetLabel("Connect");
+            }
+        }
+    } else {
+        delete m_device;
+        m_device = nullptr;
+        button->SetLabel("Connect");
+    }
+}
+
+void MainFrame::doSingle()
+{
+    m_device_samples_future = std::async(std::launch::async,
+                                         [this]() { return m_device->continuous_read(); });
+}
+
+void MainFrame::onSinglePressed(wxCommandEvent& ce)
+{
+    auto button = dynamic_cast<wxButton *>(ce.GetEventObject());
+
+    if (!m_render_timer->IsRunning()) {
+        if (m_device != nullptr && m_device->connected()) {
+            m_device->continuous_start();
+            m_device_samples_future = std::async(std::launch::async,
+                                                 []() { return decltype(m_device_samples)(); });
+            m_render_timer->Start(1000);
+            button->SetLabel("Stop");
+        }
+    } else {
+        m_render_timer->Stop();
+        m_device->continuous_stop();
+
+        //m_device_samples.clear();
+        //this->RefreshRect(m_signal_area->GetRect());
+
+        button->SetLabel("Run");
+    }
+}
+
+void MainFrame::onUploadPressed([[maybe_unused]] wxCommandEvent& ce)
+{
+    //wxFileDialog dialog (this, "Select filter to upload", "", "", "*.so",
+    //                     wxFD_OPEN | wxFD_FILE_MUST_EXIST);
+    //if (dialog.ShowModal() != wxID_CANCEL) {
+    if (auto file = compileEditorCode(); !file.IsEmpty()) {
+        if (wxFileInputStream file_stream (/*dialog.GetPath()*/file); file_stream.IsOk()) {
+            auto size = file_stream.GetSize();
+            auto buffer = new unsigned char[size];
+            if (m_device != nullptr && m_device->connected()) {
+                file_stream.ReadAll(buffer, size);
+                m_device->upload_filter(buffer, size);
+            }
+        }
+    }
+}
+
+void MainFrame::onRenderTimer([[maybe_unused]] wxTimerEvent& te)
+{
+    updateDrawing();
+}
+
+void MainFrame::updateDrawing()
+{
+    if (m_device_samples = m_device_samples_future.get(); m_device_samples.size() > 0)
+        this->RefreshRect(m_signal_area->GetRect());
+
+    doSingle();
+}
 
 void MainFrame::prepareEditor()
 {
     m_text_editor->SetLexer(wxSTC_LEX_CPP);
     m_text_editor->SetMarginWidth(0, 30);
     m_text_editor->SetMarginType(0, wxSTC_MARGIN_NUMBER);
-    wxFont::AddPrivateFont("./Hack-Regular.ttf");
     m_text_editor->StyleSetFaceName(wxSTC_STYLE_DEFAULT, "Hack");
     m_text_editor->StyleClearAll();
     m_text_editor->SetTabWidth(4);
@@ -34,7 +173,7 @@ void MainFrame::prepareEditor()
     m_text_editor->SetKeyWords(0,
         wxT("return for while do break continue if else goto"));
     m_text_editor->SetKeyWords(1,
-        wxT("void char short int long float double unsigned signed "
+        wxT("void char short int long auto float double unsigned signed "
             "volatile static const constexpr constinit consteval "
             "virtual final noexcept public private protected"));
     m_text_editor->SetText("void process_data(adcsample_t *samples, unsigned int size)\n{\n\t\n}\n");