TRestGeant4PrimaryGeneratorInfo.h

 
#ifndef REST_TRESTGEANT4PRIMARYGENERATORINFO_H
#define REST_TRESTGEANT4PRIMARYGENERATORINFO_H
 
#include <TF1.h>
#include <TF2.h>
#include <TString.h>
#include <TVector3.h>
 
#include <string>
 
namespace TRestGeant4PrimaryGeneratorTypes {
 
enum class SpatialGeneratorTypes {
    CUSTOM,
    VOLUME,
    SURFACE,
    POINT,
    COSMIC,
    SOURCE,
};
 
std::string SpatialGeneratorTypesToString(const SpatialGeneratorTypes&);
SpatialGeneratorTypes StringToSpatialGeneratorTypes(const std::string&);
 
enum class SpatialGeneratorShapes {
    GDML,
    WALL,
    CIRCLE,
    BOX,
    SPHERE,
    CYLINDER,
};
 
std::string SpatialGeneratorShapesToString(const SpatialGeneratorShapes&);
SpatialGeneratorShapes StringToSpatialGeneratorShapes(const std::string&);
 
enum class EnergyDistributionTypes {
    TH1D,
    TH2D,
    FORMULA,
    FORMULA2,
    MONO,
    FLAT,
    LOG,
};
 
std::string EnergyDistributionTypesToString(const EnergyDistributionTypes&);
EnergyDistributionTypes StringToEnergyDistributionTypes(const std::string&);
 
enum class EnergyDistributionFormulas {
    COSMIC_NEUTRONS,
    COSMIC_GAMMAS,
};
 
std::string EnergyDistributionFormulasToString(const EnergyDistributionFormulas&);
EnergyDistributionFormulas StringToEnergyDistributionFormulas(const std::string&);
TF1 EnergyDistributionFormulasToRootFormula(const EnergyDistributionFormulas&);
 
enum class AngularDistributionTypes {
    TH1D,
    TH2D,
    FORMULA,
    FORMULA2,
    ISOTROPIC,
    FLUX,
    BACK_TO_BACK,
};
 
std::string AngularDistributionTypesToString(const AngularDistributionTypes&);
AngularDistributionTypes StringToAngularDistributionTypes(const std::string&);
 
enum class AngularDistributionFormulas {
    COS2,
    COS3,
};
 
std::string AngularDistributionFormulasToString(const AngularDistributionFormulas&);
AngularDistributionFormulas StringToAngularDistributionFormulas(const std::string&);
TF1 AngularDistributionFormulasToRootFormula(const AngularDistributionFormulas&);
 
enum class EnergyAndAngularDistributionFormulas {
    COSMIC_MUONS,
};
 
std::string EnergyAndAngularDistributionFormulasToString(const EnergyAndAngularDistributionFormulas&);
EnergyAndAngularDistributionFormulas StringToEnergyAndAngularDistributionFormulas(const std::string&);
 
TF2 EnergyAndAngularDistributionFormulasToRootFormula(const EnergyAndAngularDistributionFormulas&);
 
}  // namespace TRestGeant4PrimaryGeneratorTypes
 
class TiXmlElement;
 
class TRestGeant4PrimaryGeneratorInfo {
    ClassDef(TRestGeant4PrimaryGeneratorInfo, 2);
 
   public:
    void Print() const;
 
   private:
    TString fSpatialGeneratorType = "point";
 
    TString fSpatialGeneratorShape;
 
    TString fSpatialGeneratorFrom;
 
    TVector3 fSpatialGeneratorPosition;
 
    TVector3 fSpatialGeneratorRotationAxis;
 
    Double_t fSpatialGeneratorRotationValue = 0;
 
    TVector3 fSpatialGeneratorSize;
 
    TString fSpatialGeneratorSpatialDensityFunction;
 
    TVector3 fSpatialGeneratorWorldSize;
 
   public:
    inline TString GetSpatialGeneratorType() const { return fSpatialGeneratorType; }
 
    inline TString GetSpatialGeneratorShape() const { return fSpatialGeneratorShape; }
 
    inline TString GetSpatialGeneratorFrom() const { return fSpatialGeneratorFrom; }
 
    inline TVector3 GetSpatialGeneratorPosition() const { return fSpatialGeneratorPosition; }
 
    inline TVector3 GetSpatialGeneratorRotationAxis() const { return fSpatialGeneratorRotationAxis; }
 
    inline Double_t GetSpatialGeneratorRotationValue() const { return fSpatialGeneratorRotationValue; }
 
    inline TVector3 GetSpatialGeneratorSize() const { return fSpatialGeneratorSize; }
 
    inline TVector3 GetSpatialGeneratorWorldSize() const { return fSpatialGeneratorWorldSize; }
 
    inline Double_t GetSpatialGeneratorCosmicRadius() const { return fSpatialGeneratorWorldSize.Mag(); }
 
    inline Double_t GetSpatialGeneratorCosmicSurfaceTermCm2() const {
        const auto radius = GetSpatialGeneratorCosmicRadius();
        return M_PI * radius * radius * 0.01;  // cm2
    }
 
    inline TString GetSpatialGeneratorSpatialDensityFunction() const {
        return fSpatialGeneratorSpatialDensityFunction;
    }
 
    friend class TRestGeant4Metadata;
    friend class DetectorConstruction;
};
 
#endif  // REST_TRESTGEANT4PRIMARYGENERATORINFO_H