address error

rest-for-physics · Oct 3, 2024 · c5acba6 · c5acba6
1 parent aa10e1b
commit c5acba6
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/TRestGeant4Metadata.cxx b/src/TRestGeant4Metadata.cxx
@@ -983,18 +983,19 @@ Double_t TRestGeant4Metadata::GetCosmicIntensityInCountsPerSecond() const {
 
 Double_t TRestGeant4Metadata::GetEquivalentSimulatedTime() const {
     const auto type = ToLower(fGeant4PrimaryGeneratorInfo.GetSpatialGeneratorType());
-    const auto shape = ToLower(fGeant4PrimaryGeneratorInfo.GetSpatialGeneratorShape());
 
     double scalingFactor = 1.0;
     if (type == "cosmic") {
         // get the cosmic generator
         auto cosmicSource = dynamic_cast<TRestGeant4ParticleSourceCosmics*>(GetParticleSource());
-        if (cosmicSource == nullptr) {
-            throw std::runtime_error("Cosmic source not found");
+        if (cosmicSource != nullptr) {
+            scalingFactor =
+                cosmicSource
+                    ->GetEnergyRangeScalingFactor();  // number less than 1, to account for energy range
+            if (scalingFactor < 0 || scalingFactor > 1) {
+                throw std::runtime_error("Energy range scaling factor must be between 0 and 1");
+            }
         }
-        scalingFactor =
-            cosmicSource->GetEnergyRangeScalingFactor();  // number less than 1, to account for energy range
-                                                          // vs full range of the hist (is 1 if full range)
     }
 
     // counts per seconds should be reduced proportionally to the range we are sampling