Merge pull request #139 from rest-for-physics/environmental-gammas-formula

Environmental gammas

Author: lobis

inc/TRestGeant4PrimaryGeneratorInfo.h

@@ -48,7 +48,12 @@ enum class EnergyDistributionTypes {
 std::string EnergyDistributionTypesToString(const EnergyDistributionTypes&);
 EnergyDistributionTypes StringToEnergyDistributionTypes(const std::string&);
 
-enum class EnergyDistributionFormulas { COSMIC_NEUTRONS, COSMIC_GAMMAS, FISSION_NEUTRONS_U238 };
+enum class EnergyDistributionFormulas {
+    COSMIC_NEUTRONS,
+    COSMIC_GAMMAS,
+    FISSION_NEUTRONS_U238,
+    ENVIRONMENTAL_GAMMAS
+};
 
 std::string EnergyDistributionFormulasToString(const EnergyDistributionFormulas&);
 EnergyDistributionFormulas StringToEnergyDistributionFormulas(const std::string&);

src/TRestGeant4PrimaryGeneratorInfo.cxx

@@ -176,6 +176,8 @@ string TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToString(
             return "CosmicGammas";
         case EnergyDistributionFormulas::FISSION_NEUTRONS_U238:
             return "FissionNeutronsU238";
+        case EnergyDistributionFormulas::ENVIRONMENTAL_GAMMAS:
+            return "EnvironmentalGammas";
     }
     cout << "TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToString - Error - Unknown energy "
             "distribution formula"
@@ -196,6 +198,10 @@ EnergyDistributionFormulas TRestGeant4PrimaryGeneratorTypes::StringToEnergyDistr
                    EnergyDistributionFormulasToString(EnergyDistributionFormulas::FISSION_NEUTRONS_U238),
                    TString::ECaseCompare::kIgnoreCase)) {
         return EnergyDistributionFormulas::FISSION_NEUTRONS_U238;
+    } else if (TString(type).EqualTo(
+                   EnergyDistributionFormulasToString(EnergyDistributionFormulas::ENVIRONMENTAL_GAMMAS),
+                   TString::ECaseCompare::kIgnoreCase)) {
+        return EnergyDistributionFormulas::ENVIRONMENTAL_GAMMAS;
     } else {
         cout << "TRestGeant4PrimaryGeneratorTypes::StringToEnergyDistributionFormulas - Error - Unknown "
                 "energyDistributionFormulas: "
@@ -235,6 +241,15 @@ TF1 TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToRootFormula(
             distribution.GetXaxis()->SetTitle("Energy (keV)");
             return distribution;
         }
+        case EnergyDistributionFormulas::ENVIRONMENTAL_GAMMAS: {
+            // Environmental gamma radiation approximation
+            const char* title = "Environmental Gammas";
+            auto distribution = TF1(title, "(1 / TMath::Power(x, 1.5)) * TMath::Exp(-x / 1500.)", 1, 10000);
+            distribution.SetNormalized(true);
+            distribution.SetTitle(title);
+            distribution.GetXaxis()->SetTitle("Energy (keV)");
+            return distribution;
+        }
     }
     cout << "TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToRootFormula - Error - Unknown "
             "energy distribution formula"