CONEXStructures.h

Go to the documentation of this file.

#ifndef _io_CONEXStructures_h_
#define _io_CONEXStructures_h_

// #if _CONEX2R_VERSION >= 565
// #include <ConexDynamicInterface.h>
// #endif

#include <evt/ShowerSimData.h>
#include <evt/GenParticle.h>

#include <vector>


namespace evt {
  class ShowerSimData;
}

namespace io {

  class CONEXHeader {
  public:
    int fHighEnergyModel = 0;
    int fLowEnergyModel = 0;
    float fHighLowEnergy = 0;
    int fSeed = 0;
    int fParticleID = 0;
    float fVersion = 0;
    float fROOTVersion = 0;
    // cutoffs for MC part of CONEX
    float fECutHadrons = 0;
    float fECutEM = 0;
    // threshold MC --> cascade equation
    float fEThrHadrons = 0;
    float fEThrMuons = 0;
    float fEThrEM = 0;
    // cutoffs for longitudinal profiles
    float fECutLongProfHadrons = 0;
    float fECutLongProfMuons = 0;
    float fECutLongProfElectrons = 0;
    float fECutLongProfPhotons = 0;
    float fDelX = 0;
    float fFirstDepth = 0;

  };


  class CONEXShower {

  public:
    CONEXShower(const int nX);
    ~CONEXShower();

    void SetVectorLength(const int n);

    float fLogE = 0;
    float fZenith = 0;
    float fAzimuth = 0;
    float fXfirst = 0;
    float fX0 = 0;
    float fXmax = 0;
    float fNmax = 0;
    float fP1 = 0;
    float fP2 = 0;
    float fP3 = 0;
    float fChi2 = 0;
    float fHfirst = 0;
    int fNX = 0;
    float* fX = nullptr;      // grammage (starting from outputVersion >= 2.0
    float* fN = nullptr;      // charged particles
    float* fdEdX = nullptr;   // energy deposit
    float* fNmu = nullptr;    // muons
    float* fNgam = nullptr;   // gammas
    float* fNele = nullptr;   // electrons
    float* fMuProd = nullptr;    // muon production rate (X)
    float fGroundEnergy[3] = { 0 }; // energy deposit at ground

  private:
    // implement if needed, but don't use compiler-defaults
    CONEXShower& operator=(const CONEXShower&);
    CONEXShower(const CONEXShower&);

  };


  // Data about interactions of leading particles

  class CONEXLeadingParticles {

  public:
    int GetParentId() const { return fParentId; }
    double GetParentEnergy() const { return fParentEnergy; }
    double GetEnergyCM() const { return fEnergyCM; }
    double GetKinel() const { return fKinel; }
    int GetMultiplicity() const { return fMultiplicity; }
    int GetTargetMass() const { return fTargetMass; }
    double GetDepth() const { return fDepth; }
    double GetHeight() const { return fHeight; }

    std::vector<int> GetParticleIds() const { return fParticleIds; }
    std::vector<double> GetParticleEnergies() const { return fParticleEnergies; }

    void SetParentId(const int parentId) { fParentId = parentId; }
    void SetParentEnergy(const double parentEnergy) { fParentEnergy = parentEnergy; }
    void SetEnergyCM(const double energyCM) { fEnergyCM = energyCM; }
    void SetKinel(const double kinel) { fKinel = kinel; }
    void SetMultiplicity(const int mult) { fMultiplicity = mult; }
    void SetTargetMass(const int targetMass) { fTargetMass = targetMass; }
    void SetDepth(const double depth) { fDepth = depth; }
    void SetHeight(const double height) { fHeight = height; }

    void AddParticleId(const double pId) { fParticleIds.push_back(pId); }
    void AddParticleEnergy(const double penergy) { fParticleEnergies.push_back(penergy); }
    void AddInteractionNumber(const int intnumber) { fInteractionNumber.push_back(intnumber); }
    void AddParticleMass(const double pmass) { fParticleMass.push_back(pmass); }

  private:
    int fParentId;           // parent id
    double fParentEnergy;    // parent energy [GeV]
    double fEnergyCM;   // center-of-mass energy [GeV]
    double fKinel;      // inelasticity
    int fMultiplicity;          // multiplicity
    int fTargetMass;          // target nucleus mass
    double fDepth;      // depth
    double fHeight;     // height

    std::vector<int> fParticleIds;         // Ids of particles in the interaction
    std::vector<double> fParticleEnergies; // Energies of particles in the interaction
    std::vector<double> fParticleMass;     // Masses of particles in the interaction
    std::vector<int> fInteractionNumber;                 // Interaction number

  };


  // utility function to fill ShowerSimData from CONEXHeader and CONEXShower
  // (last parameters allows to shift shower by grammage of deltaX (for syst.
  //  FD studies))
  void FillShowerSimDataFromConex(const CONEXHeader& conexHeader,
                                  const CONEXShower& conexShower,
                                  evt::ShowerSimData&);

  void FillShowerProfileDataFromConex(const CONEXHeader& conexHeader,
                                      const CONEXShower& conexShower,
                                      const std::vector<io::CONEXLeadingParticles> conexLPvector,
                                      const float conexFileVersion,
                                      evt::ShowerSimData&,
                                      const double xShift = 0);

  void FillInteractionData(evt::GenParticle& interactionTree,
                           const std::vector<io::CONEXLeadingParticles>& lpVector,
                           unsigned counter = 0);

}


#endif