merged gui

src/gui/input/Widget.cpp

@@ -0,0 +1,89 @@
+#include "Widget.h"
+#include "linalg/linalg.h"
+#include "consts.h"
+#include <cstdio>
+#include <stdlib.h>
+
+
+Widget::Widget(GLFWwindow* window) {
+  IMGUI_CHECKVERSION();
+  ImGui::CreateContext();
+  this->io = ImGui::GetIO();
+
+  ImGui_ImplGlfw_InitForOpenGL(window, true);
+  ImGui_ImplOpenGL3_Init();
+
+  // this->cameraPos = Point3::init(50.0f, -50.0f, -75.0f);  // Camera for partially trimmed data set
+  this->cameraPos = Point3::init(300.0f, 200.0f, -700.0f);  // Camera for full data set
+  //
+  Vec3 h_center = Vec3::init((float)VOLUME_WIDTH/2.0f, (float)VOLUME_HEIGHT/2.0f, (float)VOLUME_DEPTH/2.0f);
+  this->cameraDir = (h_center - this->cameraPos).normalize();
+
+  this->bgColor = Color3::init(0.1f, 0.1f, 0.1f);
+  this->lightPos = Point3::init(1.5, 2.0, -1.0);
+
+  this->fps = (char*)malloc(512*sizeof(char));
+  this->paused = true;
+  this->renderOnce = false;
+};
+
+void Widget::tick(double fps) {
+  if (this->renderOnce) {
+    this->renderOnce = false;
+    this->paused = true;
+  }
+
+  ImGui_ImplOpenGL3_NewFrame();
+  ImGui_ImplGlfw_NewFrame();
+  ImGui::NewFrame();
+  
+  // input widgets
+  float min = -1, max = 1;
+
+  ImGui::Begin("Light Controls");
+  ImGui::DragScalar("X coordinate", ImGuiDataType_Double, &this->lightPos.x, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("Y coordinate", ImGuiDataType_Double, &this->lightPos.y, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("Z coordinate", ImGuiDataType_Double, &this->lightPos.z, 0.005f, &min, &max, "%.3f");
+  ImGui::End();
+
+  ImGui::Begin("Miscellaneous");
+  if (ImGui::Button(this->paused ? "Unpause" : "Pause")) this->paused = !this->paused;
+  ImGui::SameLine();
+  if (ImGui::Button("render Once")) {
+    this->paused = !this->paused;
+    this->renderOnce = true;
+  }
+  sprintf(this->fps, "%.3f fps\n", fps);
+  ImGui::Text(this->fps);
+  ImGui::End();
+
+  ImGui::Begin("Camera Controls");
+  ImGui::DragScalar("X position", ImGuiDataType_Double, &this->cameraPos.x, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("Y position", ImGuiDataType_Double, &this->cameraPos.y, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("Z position", ImGuiDataType_Double, &this->cameraPos.z, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("X direction", ImGuiDataType_Double, &this->cameraDir.x, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("Y direction", ImGuiDataType_Double, &this->cameraDir.y, 0.005f, &min, &max, "%.3f");
+  ImGui::DragScalar("Z direction", ImGuiDataType_Double, &this->cameraDir.z, 0.005f, &min, &max, "%.3f");
+  ImGui::End();
+  
+  copyToDevice();
+}
+
+void Widget::render() {
+  ImGui::Render();
+  ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
+}
+
+void Widget::copyToDevice() {
+  cudaMemcpyToSymbol(&d_cameraPos, &this->cameraPos, sizeof(Point3));
+  cudaMemcpyToSymbol(&d_cameraDir, &this->cameraDir, sizeof(Vec3));
+  cudaMemcpyToSymbol(&d_lightPos, &this->lightPos, sizeof(Point3));
+  cudaMemcpyToSymbol(&d_backgroundColor, &this->bgColor, sizeof(Color3));
+}
+
+Widget::~Widget() {
+  ImGui_ImplOpenGL3_Shutdown();
+  ImGui_ImplGlfw_Shutdown();
+  ImGui::DestroyContext();
+  free(this->fps);
+}

src/gui/input/Widget.h

@@ -0,0 +1,33 @@
+#ifndef WIDGET_H
+#define WIDGET_H
+
+#include "../include/imgui/imgui.h"
+#include "../include/imgui/backends/imgui_impl_glfw.h"
+#include "../include/imgui/backends/imgui_impl_opengl3.h"
+#include <GLFW/glfw3.h>
+#include "linalg/linalg.h"
+
+class Widget {
+public:
+  Point3 cameraDir;
+  Vec3 cameraPos;
+  Point3 lightPos;
+  Color3 bgColor;
+
+  bool paused;
+  bool renderOnce;
+  char* fps;
+
+  ImGuiIO io;
+
+  void tick(double fps);
+  void render();
+  void copyToDevice();
+
+  Widget(GLFWwindow* window);
+  ~Widget();
+};
+
+
+
+#endif // WIDGET_H