Add timeline and replay feature
This commit is contained in:
parent
d3f046f0d0
commit
d71a584f19
109
main.cpp
109
main.cpp
@ -9,40 +9,45 @@
|
|||||||
#include "imgui_impl_opengl3.h"
|
#include "imgui_impl_opengl3.h"
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
|
||||||
// Forward declarations
|
|
||||||
void generateRandomColors(unsigned char& r, unsigned char& g, unsigned char& b);
|
void generateRandomColors(unsigned char& r, unsigned char& g, unsigned char& b);
|
||||||
void sendDMXData(HANDLE hSerial, unsigned char* data, int length);
|
void sendDMXData(HANDLE hSerial, unsigned char* data, int length);
|
||||||
void setupImGui(GLFWwindow* window);
|
void setupImGui(GLFWwindow* window);
|
||||||
void renderImGui();
|
void renderImGui();
|
||||||
|
|
||||||
// DMX data buffer
|
|
||||||
|
// ui timeline stuff
|
||||||
|
std::vector<std::tuple<float, unsigned char, unsigned char, unsigned char>> markers; // Time, R, G, B
|
||||||
|
float timeline_position = 0.0f;
|
||||||
|
bool is_playing = false;
|
||||||
|
float playback_start_time = 0.0f;
|
||||||
|
float current_time = 0.0f;
|
||||||
|
float timeline_duration = 10.0f; // 10 seconds by default
|
||||||
|
|
||||||
|
|
||||||
const int num_channels = 512;
|
const int num_channels = 512;
|
||||||
unsigned char buffer[num_channels] = { 0 };
|
unsigned char buffer[num_channels] = { 0 };
|
||||||
|
|
||||||
// Function to initialize OpenGL and create a window
|
|
||||||
GLFWwindow* initOpenGL();
|
GLFWwindow* initOpenGL();
|
||||||
|
|
||||||
// Serial port handle
|
|
||||||
HANDLE hSerial;
|
HANDLE hSerial;
|
||||||
std::atomic<bool> running(true);
|
std::atomic<bool> running(true);
|
||||||
|
|
||||||
void dmxThreadFunc() {
|
void dmxThreadFunc() {
|
||||||
while (running) {
|
while (running) {
|
||||||
// Send DMX data with a refresh rate of 25ms
|
// 25ms refresh rate
|
||||||
sendDMXData(hSerial, buffer, num_channels);
|
sendDMXData(hSerial, buffer, num_channels);
|
||||||
std::this_thread::sleep_for(std::chrono::milliseconds(25));
|
std::this_thread::sleep_for(std::chrono::milliseconds(25));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
// Open the serial port
|
|
||||||
hSerial = CreateFileA("\\\\.\\COM3", GENERIC_WRITE, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
|
hSerial = CreateFileA("\\\\.\\COM3", GENERIC_WRITE, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
|
||||||
if (hSerial == INVALID_HANDLE_VALUE) {
|
if (hSerial == INVALID_HANDLE_VALUE) {
|
||||||
std::cerr << "Error opening COM3 port" << std::endl;
|
std::cerr << "Error opening COM3 port" << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Configure the serial port
|
|
||||||
DCB dcbSerialParams = { 0 };
|
DCB dcbSerialParams = { 0 };
|
||||||
dcbSerialParams.DCBlength = sizeof(dcbSerialParams);
|
dcbSerialParams.DCBlength = sizeof(dcbSerialParams);
|
||||||
if (!GetCommState(hSerial, &dcbSerialParams)) {
|
if (!GetCommState(hSerial, &dcbSerialParams)) {
|
||||||
@ -51,9 +56,9 @@ int main() {
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
dcbSerialParams.BaudRate = 250000; // DMX512 uses 250000 baud
|
dcbSerialParams.BaudRate = 250000;
|
||||||
dcbSerialParams.ByteSize = 8;
|
dcbSerialParams.ByteSize = 8;
|
||||||
dcbSerialParams.StopBits = TWOSTOPBITS; // DMX512 uses 2 stop bits
|
dcbSerialParams.StopBits = TWOSTOPBITS; // <-- don't forget about this
|
||||||
dcbSerialParams.Parity = NOPARITY;
|
dcbSerialParams.Parity = NOPARITY;
|
||||||
|
|
||||||
if (!SetCommState(hSerial, &dcbSerialParams)) {
|
if (!SetCommState(hSerial, &dcbSerialParams)) {
|
||||||
@ -62,7 +67,6 @@ int main() {
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set timeouts
|
|
||||||
COMMTIMEOUTS timeouts = { 0 };
|
COMMTIMEOUTS timeouts = { 0 };
|
||||||
timeouts.ReadIntervalTimeout = 50;
|
timeouts.ReadIntervalTimeout = 50;
|
||||||
timeouts.ReadTotalTimeoutConstant = 50;
|
timeouts.ReadTotalTimeoutConstant = 50;
|
||||||
@ -76,33 +80,24 @@ int main() {
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Initialize OpenGL and create a window
|
|
||||||
GLFWwindow* window = initOpenGL();
|
GLFWwindow* window = initOpenGL();
|
||||||
if (!window) {
|
if (!window) {
|
||||||
CloseHandle(hSerial);
|
CloseHandle(hSerial);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Setup ImGui
|
|
||||||
setupImGui(window);
|
setupImGui(window);
|
||||||
|
|
||||||
// Start the DMX thread
|
|
||||||
std::thread dmxThread(dmxThreadFunc);
|
std::thread dmxThread(dmxThreadFunc);
|
||||||
|
|
||||||
// Main loop
|
|
||||||
while (!glfwWindowShouldClose(window)) {
|
while (!glfwWindowShouldClose(window)) {
|
||||||
// Poll events
|
|
||||||
glfwPollEvents();
|
glfwPollEvents();
|
||||||
|
|
||||||
// Start the ImGui frame
|
|
||||||
ImGui_ImplOpenGL3_NewFrame();
|
ImGui_ImplOpenGL3_NewFrame();
|
||||||
ImGui_ImplGlfw_NewFrame();
|
ImGui_ImplGlfw_NewFrame();
|
||||||
ImGui::NewFrame();
|
ImGui::NewFrame();
|
||||||
|
|
||||||
// Render ImGui components
|
|
||||||
renderImGui();
|
renderImGui();
|
||||||
|
|
||||||
// Rendering
|
|
||||||
ImGui::Render();
|
ImGui::Render();
|
||||||
int display_w, display_h;
|
int display_w, display_h;
|
||||||
glfwGetFramebufferSize(window, &display_w, &display_h);
|
glfwGetFramebufferSize(window, &display_w, &display_h);
|
||||||
@ -111,15 +106,12 @@ int main() {
|
|||||||
glClear(GL_COLOR_BUFFER_BIT);
|
glClear(GL_COLOR_BUFFER_BIT);
|
||||||
ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
|
ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
|
||||||
|
|
||||||
// Swap buffers
|
|
||||||
glfwSwapBuffers(window);
|
glfwSwapBuffers(window);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Stop the DMX thread
|
|
||||||
running = false;
|
running = false;
|
||||||
dmxThread.join();
|
dmxThread.join();
|
||||||
|
|
||||||
// Cleanup
|
|
||||||
ImGui_ImplOpenGL3_Shutdown();
|
ImGui_ImplOpenGL3_Shutdown();
|
||||||
ImGui_ImplGlfw_Shutdown();
|
ImGui_ImplGlfw_Shutdown();
|
||||||
ImGui::DestroyContext();
|
ImGui::DestroyContext();
|
||||||
@ -137,13 +129,12 @@ void generateRandomColors(unsigned char& r, unsigned char& g, unsigned char& b)
|
|||||||
}
|
}
|
||||||
|
|
||||||
void sendDMXData(HANDLE hSerial, unsigned char* data, int length) {
|
void sendDMXData(HANDLE hSerial, unsigned char* data, int length) {
|
||||||
// Start with a DMX break condition
|
// break condition
|
||||||
EscapeCommFunction(hSerial, SETBREAK);
|
EscapeCommFunction(hSerial, SETBREAK);
|
||||||
std::this_thread::sleep_for(std::chrono::milliseconds(1)); // Break duration
|
std::this_thread::sleep_for(std::chrono::milliseconds(1));
|
||||||
EscapeCommFunction(hSerial, CLRBREAK);
|
EscapeCommFunction(hSerial, CLRBREAK);
|
||||||
std::this_thread::sleep_for(std::chrono::milliseconds(1)); // Mark-after-break duration
|
std::this_thread::sleep_for(std::chrono::milliseconds(1)); // this is the mab
|
||||||
|
|
||||||
// Write the start code followed by the DMX data
|
|
||||||
DWORD bytesWritten;
|
DWORD bytesWritten;
|
||||||
unsigned char startCode = 0;
|
unsigned char startCode = 0;
|
||||||
WriteFile(hSerial, &startCode, 1, &bytesWritten, NULL);
|
WriteFile(hSerial, &startCode, 1, &bytesWritten, NULL);
|
||||||
@ -151,13 +142,11 @@ void sendDMXData(HANDLE hSerial, unsigned char* data, int length) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
GLFWwindow* initOpenGL() {
|
GLFWwindow* initOpenGL() {
|
||||||
// Initialize GLFW
|
|
||||||
if (!glfwInit()) {
|
if (!glfwInit()) {
|
||||||
std::cerr << "Failed to initialize GLFW" << std::endl;
|
std::cerr << "Failed to initialize GLFW" << std::endl;
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create a windowed mode window and its OpenGL context
|
|
||||||
GLFWwindow* window = glfwCreateWindow(1280, 720, "DMX Controller", NULL, NULL);
|
GLFWwindow* window = glfwCreateWindow(1280, 720, "DMX Controller", NULL, NULL);
|
||||||
if (!window) {
|
if (!window) {
|
||||||
std::cerr << "Failed to create GLFW window" << std::endl;
|
std::cerr << "Failed to create GLFW window" << std::endl;
|
||||||
@ -165,11 +154,9 @@ GLFWwindow* initOpenGL() {
|
|||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Make the window's context current
|
|
||||||
glfwMakeContextCurrent(window);
|
glfwMakeContextCurrent(window);
|
||||||
glfwSwapInterval(1); // Enable vsync
|
glfwSwapInterval(1); // I think this is vsync??
|
||||||
|
|
||||||
// Initialize GLEW
|
|
||||||
if (glewInit() != GLEW_OK) {
|
if (glewInit() != GLEW_OK) {
|
||||||
std::cerr << "Failed to initialize GLEW" << std::endl;
|
std::cerr << "Failed to initialize GLEW" << std::endl;
|
||||||
glfwDestroyWindow(window);
|
glfwDestroyWindow(window);
|
||||||
@ -181,15 +168,12 @@ GLFWwindow* initOpenGL() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void setupImGui(GLFWwindow* window) {
|
void setupImGui(GLFWwindow* window) {
|
||||||
// Setup ImGui context
|
|
||||||
IMGUI_CHECKVERSION();
|
IMGUI_CHECKVERSION();
|
||||||
ImGui::CreateContext();
|
ImGui::CreateContext();
|
||||||
ImGuiIO& io = ImGui::GetIO(); (void)io;
|
ImGuiIO& io = ImGui::GetIO(); (void)io;
|
||||||
|
|
||||||
// Setup ImGui style
|
|
||||||
ImGui::StyleColorsDark();
|
ImGui::StyleColorsDark();
|
||||||
|
|
||||||
// Setup Platform/Renderer bindings
|
|
||||||
ImGui_ImplGlfw_InitForOpenGL(window, true);
|
ImGui_ImplGlfw_InitForOpenGL(window, true);
|
||||||
ImGui_ImplOpenGL3_Init("#version 130");
|
ImGui_ImplOpenGL3_Init("#version 130");
|
||||||
}
|
}
|
||||||
@ -213,5 +197,62 @@ void renderImGui() {
|
|||||||
buffer[1] = static_cast<unsigned char>(channel2);
|
buffer[1] = static_cast<unsigned char>(channel2);
|
||||||
buffer[2] = static_cast<unsigned char>(channel3);
|
buffer[2] = static_cast<unsigned char>(channel3);
|
||||||
|
|
||||||
|
// Allow user to set the total duration of the timeline
|
||||||
|
ImGui::SliderFloat("Timeline Duration (seconds)", &timeline_duration, 1.0f, 300.0f);
|
||||||
|
|
||||||
|
// Timeline UI with time scale
|
||||||
|
ImGui::Text("Timeline");
|
||||||
|
ImGui::SliderFloat("##Timeline", &timeline_position, 0.0f, timeline_duration, "%.2f sec");
|
||||||
|
|
||||||
|
if (ImGui::Button("Add Marker")) {
|
||||||
|
markers.emplace_back(timeline_position / timeline_duration, buffer[0], buffer[1], buffer[2]);
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::SameLine();
|
||||||
|
if (ImGui::Button(is_playing ? "Pause" : "Play")) {
|
||||||
|
is_playing = !is_playing;
|
||||||
|
if (is_playing) {
|
||||||
|
playback_start_time = static_cast<float>(glfwGetTime());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Display markers with time indication
|
||||||
|
for (const auto& marker : markers) {
|
||||||
|
float marker_position = std::get<0>(marker) * timeline_duration;
|
||||||
|
ImGui::Text("Marker at %.2f sec: R=%d, G=%d, B=%d",
|
||||||
|
marker_position,
|
||||||
|
std::get<1>(marker),
|
||||||
|
std::get<2>(marker),
|
||||||
|
std::get<3>(marker));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Playback logic with time synchronization
|
||||||
|
if (is_playing) {
|
||||||
|
current_time = static_cast<float>(glfwGetTime()) - playback_start_time;
|
||||||
|
for (const auto& marker : markers) {
|
||||||
|
float marker_position = std::get<0>(marker) * timeline_duration;
|
||||||
|
if (current_time >= marker_position) {
|
||||||
|
buffer[0] = std::get<1>(marker);
|
||||||
|
buffer[1] = std::get<2>(marker);
|
||||||
|
buffer[2] = std::get<3>(marker);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update the timeline slider position according to current playback time
|
||||||
|
timeline_position = current_time;
|
||||||
|
if (timeline_position >= timeline_duration) {
|
||||||
|
is_playing = false; // Stop playback at the end
|
||||||
|
timeline_position = timeline_duration;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Draw time scale
|
||||||
|
ImGui::Separator();
|
||||||
|
ImGui::Text("Time Scale:");
|
||||||
|
for (float t = 0.0f; t <= timeline_duration; t += timeline_duration / 10.0f) {
|
||||||
|
ImGui::SameLine();
|
||||||
|
ImGui::Text("%.1f sec", t);
|
||||||
|
}
|
||||||
|
|
||||||
ImGui::End();
|
ImGui::End();
|
||||||
}
|
}
|
Loading…
x
Reference in New Issue
Block a user