added convenience function to obtain map and model from simulator object

uos · Apr 23, 2024 · 1afa2d5 · 1afa2d5
1 parent 6442834
commit 1afa2d5
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 0 deletions.
diff --git a/src/rmagine_optix/include/rmagine/simulation/O1DnSimulatorOptix.hpp b/src/rmagine_optix/include/rmagine/simulation/O1DnSimulatorOptix.hpp
@@ -155,6 +155,16 @@ class O1DnSimulatorOptix {
     template<typename BundleT>
     void preBuildProgram();
 
+    inline Memory<O1DnModel_<VRAM_CUDA>, VRAM_CUDA> model() const
+    {
+        return m_model_d;
+    }
+
+    inline OptixMapPtr map() const 
+    {
+        return m_map;
+    }
+
     // Problems:
     // - a lot of copies
     // 

diff --git a/src/rmagine_optix/include/rmagine/simulation/OnDnSimulatorOptix.hpp b/src/rmagine_optix/include/rmagine/simulation/OnDnSimulatorOptix.hpp
@@ -165,6 +165,16 @@ class OnDnSimulatorOptix {
     template<typename BundleT>
     void preBuildProgram();
 
+    inline Memory<OnDnModel_<VRAM_CUDA>, VRAM_CUDA> model() const
+    {
+        return m_model_d;
+    }
+
+    inline OptixMapPtr map() const 
+    {
+        return m_map;
+    }
+
     // Problems:
     // - a lot of copies
     // 

diff --git a/src/rmagine_optix/include/rmagine/simulation/PinholeSimulatorOptix.hpp b/src/rmagine_optix/include/rmagine/simulation/PinholeSimulatorOptix.hpp
@@ -155,6 +155,16 @@ class PinholeSimulatorOptix {
     template<typename BundleT>
     void preBuildProgram();
 
+    inline Memory<PinholeModel, VRAM_CUDA> model() const
+    {
+        return m_model;
+    }
+
+    inline OptixMapPtr map() const 
+    {
+        return m_map;
+    }
+
     // Problems:
     // - a lot of copies
     // 

diff --git a/src/rmagine_optix/include/rmagine/simulation/SphereSimulatorOptix.hpp b/src/rmagine_optix/include/rmagine/simulation/SphereSimulatorOptix.hpp
@@ -158,6 +158,16 @@ class SphereSimulatorOptix {
     template<typename BundleT>
     void preBuildProgram();
 
+    inline Memory<SphericalModel, VRAM_CUDA> model() const
+    {
+        return m_model;
+    }
+
+    inline OptixMapPtr map() const 
+    {
+        return m_map;
+    }
+
     // Problems:
     // - a lot of copies
     //