Only delay OpticalPhotons if discard macro is set (#132)

* Only delay OpticalPhotons if discard macro is set --------- Co-authored-by: Manuel Huber <[email protected]>
legend-exp · Oct 17, 2024 · 6cc4359 · 6cc4359
1 parent 6054276
commit 6cc4359
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/RMGGermaniumOutputScheme.cc b/src/RMGGermaniumOutputScheme.cc
@@ -167,8 +167,10 @@ std::optional<G4ClassificationOfNewTrack> RMGGermaniumOutputScheme::StackingActi
   // we are only interested in stacking optical photons into stage 1 after stage 0 finished.
   if (stage != 0) return std::nullopt;
 
-  // defer tracking of optical photons, irrespective of our settings.
-  if (aTrack->GetDefinition() == G4OpticalPhoton::OpticalPhotonDefinition()) return fWaiting;
+  // defer tracking of optical photons.
+  if (fDiscardPhotonsIfNoGermaniumEdep &&
+      aTrack->GetDefinition() == G4OpticalPhoton::OpticalPhotonDefinition())
+    return fWaiting;
   return std::nullopt;
 }
 

diff --git a/src/RMGIsotopeFilterOutputScheme.cc b/src/RMGIsotopeFilterOutputScheme.cc
@@ -65,7 +65,9 @@ std::optional<G4ClassificationOfNewTrack> RMGIsotopeFilterOutputScheme::
   if (stage != 0) return std::nullopt;
 
   // defer tracking of optical photons.
-  if (aTrack->GetDefinition() == G4OpticalPhoton::OpticalPhotonDefinition()) return fWaiting;
+  if (fDiscardPhotonsIfIsotopeNotProduced &&
+      aTrack->GetDefinition() == G4OpticalPhoton::OpticalPhotonDefinition())
+    return fWaiting;
   return std::nullopt;
 }