optionally dump jacobi volumes

Author: jeffamstutz

tsd/apps/interactive/demos/animatedVolume/SolverControls.cpp

@@ -4,6 +4,7 @@
 #include "SolverControls.h"
 // tsd_ui_imgui
 #include <tsd/ui/imgui/Application.h>
+#include <tsd/ui/imgui/tsd_ui_imgui.h>
 // tsd_core
 #include <tsd/core/Logging.hpp>
 // std
@@ -70,11 +71,8 @@ void SolverControls::buildUI()
     resetSolver();
   ImGui::EndDisabled();
   ImGui::Checkbox("auto update transfer function", &m_updateTF);
-
-  ImGui::Separator();
-
-  if (ImGui::Button("export .raw"))
-    exportRAW();
+  ImGui::Checkbox("dump volumes timesteps", &m_dumpVolumes);
+  ImGui::InputText("export root", &m_exportRoot);
 }
 
 void SolverControls::setField(tsd::core::SpatialFieldRef f)
@@ -226,6 +224,9 @@ void SolverControls::iterateSolver()
   tsd::core::logStatus("[jacobi solver] ...TSD update  (%i|%.2fms)",
       m_totalIterations,
       duration);
+
+  if (m_dumpVolumes)
+    exportRAW();
 }
 
 void SolverControls::exportRAW()
@@ -238,7 +239,7 @@ void SolverControls::exportRAW()
   filename.resize(100);
   filename.resize(std::snprintf(filename.data(),
       filename.size(),
-      "jacobi_%i_%zux%zux%zu_float32.raw",
+      "jacobi_%04d_%zux%zux%zu_float32.raw",
       m_totalIterations,
       nx,
       ny,
@@ -253,12 +254,13 @@ void SolverControls::exportRAW()
     m_dataHost->unmap();
   }
 
-  auto *fp = std::fopen(filename.c_str(), "wb");
+  auto location = m_exportRoot + "/" + filename;
+  auto *fp = std::fopen(location.c_str(), "wb");
   std::fwrite(
       m_dataHost->dataAs<float>(), sizeof(float), m_dataHost->size(), fp);
   std::fclose(fp);
 
-  tsd::core::logStatus("[jacobi solver] exported data to %s", filename.c_str());
+  tsd::core::logStatus("[jacobi solver] exported data to %s", location.c_str());
 }
 
 } // namespace tsd::demo

tsd/apps/interactive/demos/animatedVolume/SolverControls.h

@@ -41,6 +41,8 @@ struct SolverControls : public tsd::ui::imgui::Window
   bool m_playing{false};
   bool m_useGPUInterop{false};
   bool m_updateTF{true};
+  bool m_dumpVolumes{false};
+  std::string m_exportRoot{"./"}; // root filename for .raw exports
 };
 
 } // namespace tsd::demo

tsd/apps/interactive/demos/animatedVolume/tsdDemoAnimatedVolume.cpp

@@ -106,12 +106,12 @@ class Application : public TSDApplication
     return R"layout(
 [Window][MainDockSpace]
 Pos=0,26
-Size=1920,1105
+Size=1920,1054
 Collapsed=0
 
 [Window][Viewport]
 Pos=549,26
-Size=821,429
+Size=880,798
 Collapsed=0
 DockId=0x00000003,0
 
@@ -122,8 +122,8 @@ Collapsed=0
 DockId=0x00000004,0
 
 [Window][Log]
-Pos=549,885
-Size=821,246
+Pos=549,826
+Size=880,254
 Collapsed=0
 DockId=0x0000000E,0
 
@@ -133,32 +133,32 @@ Size=400,400
 Collapsed=0
 
 [Window][Layers]
-Pos=0,381
-Size=547,205
+Pos=0,365
+Size=547,195
 Collapsed=0
 DockId=0x0000000C,0
 
 [Window][Object Editor]
-Pos=0,588
-Size=547,543
+Pos=0,562
+Size=547,518
 Collapsed=0
 DockId=0x0000000D,0
 
 [Window][TF Editor]
-Pos=1372,26
-Size=548,553
+Pos=1431,26
+Size=489,527
 Collapsed=0
 DockId=0x00000005,0
 
 [Window][Isosurface Editor]
-Pos=1372,581
-Size=548,550
+Pos=1431,555
+Size=489,525
 Collapsed=0
 DockId=0x00000006,0
 
 [Window][Solver Controls]
 Pos=0,26
-Size=547,353
+Size=547,337
 Collapsed=0
 DockId=0x00000008,0
 
@@ -228,21 +228,24 @@ Column 1  Weight=1.0000
 Column 0  Weight=1.0000
 Column 1  Weight=1.0000
 
+[Table][0x5EB09CD8,2]
+Column 0  Weight=1.0000
+Column 1  Weight=1.0000
+
 [Docking][Data]
-DockSpace         ID=0x782A6D6B Pos=0,22 Size=1920,1057 CentralNode=1 Selected=0x13926F0B
-DockSpace         ID=0x80F5B4C5 Window=0x079D3A04 Pos=0,26 Size=1920,1105 Split=X
+DockSpace         ID=0x80F5B4C5 Window=0x079D3A04 Pos=0,26 Size=1920,1054 Split=X
   DockNode        ID=0x0000000A Parent=0x80F5B4C5 SizeRef=547,1054 Split=Y Selected=0xF64741D0
     DockNode      ID=0x00000008 Parent=0x0000000A SizeRef=547,337 Selected=0xF64741D0
     DockNode      ID=0x00000009 Parent=0x0000000A SizeRef=547,715 Split=Y Selected=0xCD8384B1
       DockNode    ID=0x0000000C Parent=0x00000009 SizeRef=547,195 Selected=0xCD8384B1
       DockNode    ID=0x0000000D Parent=0x00000009 SizeRef=547,518 Selected=0x82B4C496
   DockNode        ID=0x0000000B Parent=0x80F5B4C5 SizeRef=1371,1054 Split=X
-    DockNode      ID=0x00000001 Parent=0x0000000B SizeRef=1370,1054 Split=Y Selected=0xC450F867
-      DockNode    ID=0x00000007 Parent=0x00000001 SizeRef=821,806 Split=Y Selected=0xC450F867
+    DockNode      ID=0x00000001 Parent=0x0000000B SizeRef=880,1054 Split=Y Selected=0xC450F867
+      DockNode    ID=0x00000007 Parent=0x00000001 SizeRef=821,798 Split=Y Selected=0xC450F867
         DockNode  ID=0x00000003 Parent=0x00000007 SizeRef=821,429 CentralNode=1 Selected=0xC450F867
         DockNode  ID=0x00000004 Parent=0x00000007 SizeRef=821,426 Selected=0xA3219422
-      DockNode    ID=0x0000000E Parent=0x00000001 SizeRef=821,246 Selected=0x139FDA3F
-    DockNode      ID=0x00000002 Parent=0x0000000B SizeRef=548,1054 Split=Y Selected=0x3429FA32
+      DockNode    ID=0x0000000E Parent=0x00000001 SizeRef=821,254 Selected=0x139FDA3F
+    DockNode      ID=0x00000002 Parent=0x0000000B SizeRef=489,1054 Split=Y Selected=0x3429FA32
       DockNode    ID=0x00000005 Parent=0x00000002 SizeRef=548,527 Selected=0x3429FA32
       DockNode    ID=0x00000006 Parent=0x00000002 SizeRef=548,525 Selected=0xBCE6538B
 )layout";