add control panel buttons

Author: RobertJN64

UI/ui.cpp

@@ -1,51 +1,128 @@
 #include "imgui.h"
+#include "imgui_internal.h"
 #include "implot.h"
 #include "math.h"
+#include <string>
 
 #include <iostream>
 
-void render_loop() {
-  static int frame_counter = 0;
-  frame_counter++;
+#define DO_COUNT
+
+bool ui_do_state[DO_COUNT] = {false, false, false, false, false};
+
+// Helper to adjust brightness
+ImU32 AdjustBrightness(ImU32 color, float factor) {
+  ImVec4 c = ImGui::ColorConvertU32ToFloat4(color);
+  c.x = ImClamp(c.x * factor, 0.0f, 1.0f);
+  c.y = ImClamp(c.y * factor, 0.0f, 1.0f);
+  c.z = ImClamp(c.z * factor, 0.0f, 1.0f);
+  return ImGui::ColorConvertFloat4ToU32(c);
+}
+
+bool daq_button(const char *label, const ImVec2 &size,
+                ImU32 color, float rounding = 10.0f) {
+  ImGui::PushStyleColor(ImGuiCol_Button, color);
+  ImGui::PushStyleColor(ImGuiCol_ButtonHovered, AdjustBrightness(color, 1.2));
+  ImGui::PushStyleColor(ImGuiCol_ButtonActive, AdjustBrightness(color, 0.7));
+  ImGui::PushStyleVar(ImGuiStyleVar_FrameRounding, rounding);
+
+  bool clicked = ImGui::Button(label, size);
+
+  ImGui::PopStyleVar();
+  ImGui::PopStyleColor(3);
+
+  return clicked;
+}
+
+void toggle_button(const char *label, const ImVec2 &size, ImU32 active_color, ImU32 deactive_color, bool *control_var, float rounding = 10.0f) {
+  ImU32 color;
+  ImU32 clicked_color;
+  if (*control_var) {
+    color = active_color;
+    clicked_color = deactive_color;
+  } else {
+    color = deactive_color;
+    clicked_color = active_color;
+  }
+
+  ImGui::PushStyleColor(ImGuiCol_Button, color);
+  ImGui::PushStyleColor(ImGuiCol_ButtonHovered, AdjustBrightness(color, 1.2));
+  ImGui::PushStyleColor(ImGuiCol_ButtonActive, AdjustBrightness(color, 1.2)); // stops flickering
+  ImGui::PushStyleVar(ImGuiStyleVar_FrameRounding, rounding);
+
+  bool clicked = ImGui::Button(label, size);
+
+  ImGui::PopStyleVar();
+  ImGui::PopStyleColor(3);
+
+  if (clicked) {
+    std::cout << "toggle" << std::endl;
+    *control_var = !(*control_var);
+  }
+}
+
+void control_panel() {
+  ImGui::Text("Control Panel");
 
-  ImGuiIO io = ImGui::GetIO();
+  if (ImGui::BeginTable("start_abort", 2)) {
+    ImGui::TableNextColumn();
+    if (daq_button("START", ImVec2(-1, 50), IM_COL32(33, 112, 69, 255))) {
+      std::cout << "START" << std::endl;
+    };
+    ImGui::TableNextColumn();
+    if (daq_button("ABORT", ImVec2(-1, 50), IM_COL32(204, 0, 0, 255))) {
+      std::cout << "END" << std::endl;
+    };
+    ImGui::EndTable();
+  }
+
+  if (ImGui::BeginTable("do_control", 5)) {
+    ImGui::TableNextRow();
+    for (int i = 0; i < 5; i++) {
+      ImGui::TableSetColumnIndex(i);
+      ImGui::PushID(i);
+      std::string do_name = "DO " + std::to_string(i + 1);
+      toggle_button(do_name.c_str(), ImVec2(-1, 50), IM_COL32(33, 112, 69, 255), IM_COL32(50, 50, 50, 255), &ui_do_state[i]);
+      ImGui::PopID();
+    }
+    ImGui::EndTable();
+  }
+}
+
+void plot_panel() {
+  for (int i = 0; i < 3; i++) {
+    std::string plot_name = "Plot " + std::to_string(i + 1);
+    if (ImPlot::BeginPlot(plot_name.c_str(), ImVec2(-1, 0))) { // width = fill, height auto
+      // Example: simple sine wave
+      static float xs[1000], ys[1000];
+      for (int j = 0; j < 1000; ++j) {
+        xs[j] = j * 0.01f;
+        ys[j] = sin(xs[j] + i); // different phase for each plot
+      }
+      ImPlot::PlotLine("Sine", xs, ys, 1000);
+      ImPlot::EndPlot();
+    }
+  }
+}
 
+void render_loop() {
+  ImGuiIO &io = ImGui::GetIO();
   ImGui::SetNextWindowPos(ImVec2(0, 0));
-  ImGui::SetNextWindowSize(io.DisplaySize); // io = ImGui::GetIO()
+  ImGui::SetNextWindowSize(io.DisplaySize);
+
   ImGui::Begin("MainWindow", nullptr,
                ImGuiWindowFlags_NoTitleBar |
                    ImGuiWindowFlags_NoResize |
                    ImGuiWindowFlags_NoMove |
                    ImGuiWindowFlags_NoCollapse |
                    ImGuiWindowFlags_NoBringToFrontOnFocus |
                    ImGuiWindowFlags_NoNavFocus);
-  // Get available width for plots
-  ImVec2 avail = ImGui::GetContentRegionAvail();
-
-  // Sine Plot
-  static float xs[1000], ys[1000];
-  for (int i = 0; i < 1000; ++i) {
-    xs[i] = i * 0.01f;
-    ys[i] = sin(xs[i] + frame_counter / 100.0);
-  }
-  if (ImPlot::BeginPlot("Sine Wave")) { // half height
-    ImPlot::PlotLine("sin(x)", xs, ys, 1000);
-    ImPlot::EndPlot();
-  }
 
-  // Cosine Plot
-  for (int j = 0; j < 5; j++) {
-    for (int i = 0; i < 1000; ++i) {
-      ys[i] = cos(xs[i] + (frame_counter + j * 20) / 10.0);
-    }
-    ImGui::PushID(j);
-    if (ImPlot::BeginPlot("Cosine Wave", ImVec2(-1, 100))) { // half height
-      ImPlot::PlotLine("cos(x)", xs, ys, 1000);
-      ImPlot::EndPlot();
-    }
-    ImGui::PopID();
-  }
+  // Split into 2 sections: left (buttons) and right (plots)
+  ImGui::Columns(2, nullptr, true); // true = resizable
+  control_panel();
+  ImGui::NextColumn(); // move to right column
+  plot_panel();
 
-  ImGui::Text("%f", io.Framerate);
   ImGui::End();
 }