Add custom source flag to rebuild source on apply

ViennaTools · Jun 27, 2024 · a6c7c2d · a6c7c2d
1 parent 225313a
commit a6c7c2d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/include/viennaray/rayTrace.hpp b/include/viennaray/rayTrace.hpp
@@ -43,7 +43,7 @@ template <class NumericType, int D> class Trace {
     Vec3D<Vec3D<NumericType>> orthonormalBasis;
     if (usePrimaryDirection_)
       orthonormalBasis = rayInternal::getOrthonormalBasis(primaryDirection_);
-    if (!pSource_)
+    if (!useCustomSource)
       pSource_ = std::make_shared<SourceRandom<NumericType, D>>(
           boundingBox, pParticle_->getSourceDistributionPower(), traceSettings,
           geometry_.getNumPoints(), usePrimaryDirection_, orthonormalBasis);
@@ -132,10 +132,14 @@ template <class NumericType, int D> class Trace {
   /// the raySource class.
   void setSource(std::shared_ptr<Source<NumericType>> source) {
     pSource_ = source;
+    useCustomSource = true;
   }
 
   /// Reset the source to the default random source.
-  void resetSource() { pSource_.reset(); }
+  void resetSource() {
+    pSource_.reset();
+    useCustomSource = false;
+  }
 
   void enableProgressBar() { printProgress_ = true; }
 
@@ -382,6 +386,7 @@ template <class NumericType, int D> class Trace {
 
   bool usePrimaryDirection_ = false;
   bool useRandomSeeds_ = false;
+  bool useCustomSource = false;
   size_t runNumber_ = 0;
   bool calcFlux_ = true;
   bool checkError_ = true;