Merge pull request #94 from rest-for-physics/lobis-cosmic-energy-dist

New energy distribution formula for cosmics

Author: lobis

inc/TRestGeant4PrimaryGeneratorInfo.h

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

src/TRestGeant4PrimaryGeneratorInfo.cxx

@@ -163,8 +163,8 @@ string TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToString(
     switch (type) {
         case EnergyDistributionFormulas::COSMIC_NEUTRONS:
             return "CosmicNeutrons";
-        case EnergyDistributionFormulas::COSMIC_GAMMAS:
-            return "CosmicGammas";
+        case EnergyDistributionFormulas::COSMIC:
+            return "Cosmic";
     }
     cout << "TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToString - Error - Unknown energy "
             "distribution formula"
@@ -177,10 +177,9 @@ EnergyDistributionFormulas TRestGeant4PrimaryGeneratorTypes::StringToEnergyDistr
     if (TString(type).EqualTo(EnergyDistributionFormulasToString(EnergyDistributionFormulas::COSMIC_NEUTRONS),
                               TString::ECaseCompare::kIgnoreCase)) {
         return EnergyDistributionFormulas::COSMIC_NEUTRONS;
-    } else if (TString(type).EqualTo(
-                   EnergyDistributionFormulasToString(EnergyDistributionFormulas::COSMIC_GAMMAS),
-                   TString::ECaseCompare::kIgnoreCase)) {
-        return EnergyDistributionFormulas::COSMIC_GAMMAS;
+    } else if (TString(type).EqualTo(EnergyDistributionFormulasToString(EnergyDistributionFormulas::COSMIC),
+                                     TString::ECaseCompare::kIgnoreCase)) {
+        return EnergyDistributionFormulas::COSMIC;
     } else {
         cout << "TRestGeant4PrimaryGeneratorTypes::StringToEnergyDistributionFormulas - Error - Unknown "
                 "energyDistributionFormulas: "
@@ -207,8 +206,13 @@ TF1 TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToRootFormula(
             distribution.GetXaxis()->SetTitle("Energy (keV)");
             return distribution;
         }
-        case EnergyDistributionFormulas::COSMIC_GAMMAS:
-            exit(1);
+        case EnergyDistributionFormulas::COSMIC:
+            const char* title = "Cosmic distribution approximation";
+            auto distribution = TF1(title, "TMath::Power(x, -2.7)", 1E2, 1E9);
+            distribution.SetNormalized(true);  // Normalized, not really necessary
+            distribution.SetTitle(title);
+            distribution.GetXaxis()->SetTitle("Energy (keV)");
+            return distribution;
     }
     cout << "TRestGeant4PrimaryGeneratorTypes::EnergyDistributionFormulasToRootFormula - Error - Unknown "
             "energy distribution formula"