#include <EvtCalcGenerator.h>

Inheritance diagram for EvtCalcGenerator:

Collaboration diagram for EvtCalcGenerator:

Public Member Functions
	EvtCalcGenerator ()

	~EvtCalcGenerator () override

Bool_t	ReadEvent (FairPrimaryGenerator *) override

Bool_t	Init (const char *, int) override

Bool_t	Init (const char *) override

Int_t	GetNevents ()

void	SetPositions (Double_t zTa, Double_t zDV)

Double_t	GetNdaughters (const std::unique_ptr< TTree > &) const

Double_t	GetMotherPx (const std::unique_ptr< TTree > &) const

Double_t	GetMotherPy (const std::unique_ptr< TTree > &) const

Double_t	GetMotherPz (const std::unique_ptr< TTree > &) const

Double_t	GetMotherE (const std::unique_ptr< TTree > &) const

Double_t	GetVx (const std::unique_ptr< TTree > &) const

Double_t	GetVy (const std::unique_ptr< TTree > &) const

Double_t	GetVz (const std::unique_ptr< TTree > &) const

Double_t	GetDecayProb (const std::unique_ptr< TTree > &) const

Double_t	GetDauPx (const std::unique_ptr< TTree > &, int) const

Double_t	GetDauPy (const std::unique_ptr< TTree > &, int) const

Double_t	GetDauPz (const std::unique_ptr< TTree > &, int) const

Double_t	GetDauE (const std::unique_ptr< TTree > &, int) const

Double_t	GetDauPDG (const std::unique_ptr< TTree > &, int) const

Public Member Functions inherited from SHiP::Generator
	Generator ()

virtual	~Generator ()

virtual Bool_t	Init (const char *, int)=0

virtual Bool_t	Init (const char *)=0

virtual Bool_t	Init (const std::vector< std::string > &inFiles, int startNumber)

virtual Bool_t	Init (const std::vector< std::string > &inFiles)

virtual void	UseExternalFile (std::string x, Int_t i)

virtual void	UseExternalFile (std::vector< std::string > &inFiles, Int_t i)

Protected Types
enum class	BranchIndices { MotherPx = 0 , MotherPy = 1 , MotherPz = 2 , MotherE = 3 , DecayProb = 6 , Vx = 7 , Vy = 8 , Vz = 9 }

Protected Member Functions
Double_t	GetBranchValue (const std::unique_ptr< TTree > &, unsigned) const

Double_t	GetDaughterValue (const std::unique_ptr< TTree > &tree, int, int) const

Protected Attributes
Double_t	ztarget

Double_t	zDecayVolume

std::unique_ptr< TFile >	fInputFile

std::unique_ptr< TTree >	fTree

std::vector< double >	branchVars

int	fNevents

int	fn

int	nBranches

int	Ndau

Protected Attributes inherited from SHiP::Generator
std::optional< std::string >	fextFile

Int_t	firstEvent = 0

Detailed Description

Definition at line 16 of file EvtCalcGenerator.h.

Member Enumeration Documentation

◆ BranchIndices

enum class EvtCalcGenerator::BranchIndices

strongprotected

Enumerator
MotherPx
MotherPy
MotherPz
MotherE
DecayProb
Vx
Vy
Vz

Definition at line 65 of file EvtCalcGenerator.h.

                           {
    MotherPx = 0,
    MotherPy = 1,
    MotherPz = 2,
    MotherE = 3,
    DecayProb = 6,
    Vx = 7,
    Vy = 8,
    Vz = 9
  };

Constructor & Destructor Documentation

◆ EvtCalcGenerator()

EvtCalcGenerator::EvtCalcGenerator ( )

default constructor

Definition at line 17 of file EvtCalcGenerator.cxx.

17{}

◆ ~EvtCalcGenerator()

EvtCalcGenerator::~EvtCalcGenerator ( )

override

destructor

Definition at line 47 of file EvtCalcGenerator.cxx.

47{}

Member Function Documentation

◆ GetBranchValue()

Double_t EvtCalcGenerator::GetBranchValue	(	const std::unique_ptr< TTree > &	tree,
		unsigned	index
	)		const

protected

Definition at line 50 of file EvtCalcGenerator.cxx.

                                                                {
  if (index < branchVars.size()) {
    return branchVars[index];
  } else {
    throw std::out_of_range("Branch index out of range");
  }
}

◆ GetDauE()

Double_t EvtCalcGenerator::GetDauE	(	const std::unique_ptr< TTree > &	tree,
		int	dauID
	)		const

Definition at line 120 of file EvtCalcGenerator.cxx.

                                                    {
  return GetDaughterValue(tree, dauID, 3);
}

◆ GetDaughterValue()

Double_t EvtCalcGenerator::GetDaughterValue	(	const std::unique_ptr< TTree > &	tree,
		int	dauID,
		int	offset
	)		const

protected

Definition at line 59 of file EvtCalcGenerator.cxx.

                                                                         {
  int baseIndex = 10 + (dauID * 6);
  return GetBranchValue(tree, baseIndex + offset);
}

◆ GetDauPDG()

Double_t EvtCalcGenerator::GetDauPDG	(	const std::unique_ptr< TTree > &	tree,
		int	dauID
	)		const

Definition at line 124 of file EvtCalcGenerator.cxx.

                                                      {
  return GetDaughterValue(tree, dauID, 5);
}

◆ GetDauPx()

Double_t EvtCalcGenerator::GetDauPx	(	const std::unique_ptr< TTree > &	tree,
		int	dauID
	)		const

Definition at line 108 of file EvtCalcGenerator.cxx.

                                                     {
  return GetDaughterValue(tree, dauID, 0);
}

◆ GetDauPy()

Double_t EvtCalcGenerator::GetDauPy	(	const std::unique_ptr< TTree > &	tree,
		int	dauID
	)		const

Definition at line 112 of file EvtCalcGenerator.cxx.

                                                     {
  return GetDaughterValue(tree, dauID, 1);
}

◆ GetDauPz()

Double_t EvtCalcGenerator::GetDauPz	(	const std::unique_ptr< TTree > &	tree,
		int	dauID
	)		const

Definition at line 116 of file EvtCalcGenerator.cxx.

                                                     {
  return GetDaughterValue(tree, dauID, 2);
}

◆ GetDecayProb()

Double_t EvtCalcGenerator::GetDecayProb ( const std::unique_ptr< TTree > & tree ) const

Definition at line 102 of file EvtCalcGenerator.cxx.

                                            {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::DecayProb));
}

◆ GetMotherE()

Double_t EvtCalcGenerator::GetMotherE ( const std::unique_ptr< TTree > & tree ) const

Definition at line 85 of file EvtCalcGenerator.cxx.

                                            {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::MotherE));
}

◆ GetMotherPx()

Double_t EvtCalcGenerator::GetMotherPx ( const std::unique_ptr< TTree > & tree ) const

Definition at line 73 of file EvtCalcGenerator.cxx.

                                            {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::MotherPx));
}

◆ GetMotherPy()

Double_t EvtCalcGenerator::GetMotherPy ( const std::unique_ptr< TTree > & tree ) const

Definition at line 77 of file EvtCalcGenerator.cxx.

                                            {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::MotherPy));
}

◆ GetMotherPz()

Double_t EvtCalcGenerator::GetMotherPz ( const std::unique_ptr< TTree > & tree ) const

Definition at line 81 of file EvtCalcGenerator.cxx.

                                            {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::MotherPz));
}

◆ GetNdaughters()

Double_t EvtCalcGenerator::GetNdaughters ( const std::unique_ptr< TTree > & tree ) const

Definition at line 67 of file EvtCalcGenerator.cxx.

                                            {
  return GetBranchValue(tree, nBranches - 1);
}

◆ GetNevents()

Int_t EvtCalcGenerator::GetNevents ( )

inline

Definition at line 30 of file EvtCalcGenerator.h.

30{ return fNevents; }

EvtCalcGenerator::fNevents

int fNevents

Definition: EvtCalcGenerator.h:76

◆ GetVx()

Double_t EvtCalcGenerator::GetVx ( const std::unique_ptr< TTree > & tree ) const

Definition at line 91 of file EvtCalcGenerator.cxx.

                                                                       {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::Vx));
}

◆ GetVy()

Double_t EvtCalcGenerator::GetVy ( const std::unique_ptr< TTree > & tree ) const

Definition at line 94 of file EvtCalcGenerator.cxx.

                                                                       {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::Vy));
}

◆ GetVz()

Double_t EvtCalcGenerator::GetVz ( const std::unique_ptr< TTree > & tree ) const

Definition at line 97 of file EvtCalcGenerator.cxx.

                                                                       {
  return GetBranchValue(tree, static_cast<int>(BranchIndices::Vz));
}

◆ Init() [1/2]

Bool_t EvtCalcGenerator::Init ( const char * fileName )

overridevirtual

Implements SHiP::Generator.

Definition at line 20 of file EvtCalcGenerator.cxx.

                                                  {
  return Init(fileName, 0);
}

◆ Init() [2/2]

Bool_t EvtCalcGenerator::Init	(	const char *	fileName,
		int	startEvent
	)

overridevirtual

Implements SHiP::Generator.

Definition at line 24 of file EvtCalcGenerator.cxx.

                                                                        {
  fInputFile = std::unique_ptr<TFile>(TFile::Open(fileName, "read"));
  LOGF(info, "Info EvtCalcGenerator: Opening input file %s", fileName);
 
  fTree =
      std::unique_ptr<TTree>(dynamic_cast<TTree*>(fInputFile->Get("LLP_tree")));
  fNevents = fTree->GetEntries();
  fn = startEvent;
 
  auto* branches = fTree->GetListOfBranches();
  nBranches = branches->GetEntries();
  branchVars.resize(nBranches);
 
  for (int i = 0; i < nBranches; ++i) {
    auto* branch = dynamic_cast<TBranch*>(branches->At(i));
    if (fTree->FindBranch(branch->GetName())) {
      fTree->SetBranchAddress(branch->GetName(), &branchVars[i]);
    }
  }
 
  return kTRUE;
}

◆ ReadEvent()

Bool_t EvtCalcGenerator::ReadEvent ( FairPrimaryGenerator * cpg )

override

Definition at line 130 of file EvtCalcGenerator.cxx.

                                                            {
  if (fn == fNevents) {
    LOG(warning) << "End of input file. Rewind.";
    fn = 0;
  }
 
  fTree->GetEntry(fn);
  fn++;
  if (fn % 100 == 0) LOGF(info, "Info EvtCalcGenerator: event nr %d", fn);
 
  Ndau = GetNdaughters(fTree);
  // Vertex coordinates in the SHiP reference frame, expressed in [cm]
  Double_t space_unit_conv = 100.;                                    // m to cm
  Double_t coord_shift = (zDecayVolume - ztarget) / space_unit_conv;  // units m
  Double_t vx_transf = GetVx(fTree) * space_unit_conv;  // units cm
  Double_t vy_transf = GetVy(fTree) * space_unit_conv;  // units cm
  Double_t vz_transf =
      (GetVz(fTree) - coord_shift) * space_unit_conv;  // units cm
 
  Double_t c = 2.99792458e+10;  // speed of light [cm/s]
  Double_t tof = TMath::Sqrt(vx_transf * vx_transf + vy_transf * vy_transf +
                             vz_transf * vz_transf) /
                 c;
  Double_t decay_prob = GetDecayProb(fTree);
  Double_t pdg_dau = 0;
 
  // Mother LLP
  Bool_t wanttracking = false;
  Double_t pdg_llp = 999;  // Geantino, placeholder
 
  cpg->AddTrack(pdg_llp, GetMotherPx(fTree), GetMotherPy(fTree),
                GetMotherPz(fTree), vx_transf, vy_transf, vz_transf, -1.,
                wanttracking, GetMotherE(fTree), tof, decay_prob);
 
  wanttracking = true;
 
  // Secondaries
  for (int iPart = 0; iPart < Ndau; ++iPart) {
    pdg_dau = GetDauPDG(fTree, iPart);
    if (pdg_dau != -999) {
      cpg->AddTrack(pdg_dau, GetDauPx(fTree, iPart), GetDauPy(fTree, iPart),
                    GetDauPz(fTree, iPart), vx_transf, vy_transf, vz_transf, 0.,
                    wanttracking, GetDauE(fTree, iPart), tof, decay_prob);
    }
  }
 
  return kTRUE;
}

◆ SetPositions()

void EvtCalcGenerator::SetPositions	(	Double_t	zTa,
		Double_t	zDV
	)

inline

Definition at line 31 of file EvtCalcGenerator.h.

                                                {
    ztarget = zTa;       // units cm (midpoint)
    zDecayVolume = zDV;  // units cm (midpoint)
  }

Member Data Documentation

◆ branchVars

std::vector<double> EvtCalcGenerator::branchVars

protected

Definition at line 63 of file EvtCalcGenerator.h.

◆ fInputFile

std::unique_ptr<TFile> EvtCalcGenerator::fInputFile

protected

Definition at line 61 of file EvtCalcGenerator.h.

◆ fn

int EvtCalcGenerator::fn

protected

Definition at line 77 of file EvtCalcGenerator.h.

◆ fNevents

int EvtCalcGenerator::fNevents

protected

Definition at line 76 of file EvtCalcGenerator.h.

◆ fTree

std::unique_ptr<TTree> EvtCalcGenerator::fTree

protected

Definition at line 62 of file EvtCalcGenerator.h.

◆ nBranches

int EvtCalcGenerator::nBranches

protected

Definition at line 78 of file EvtCalcGenerator.h.

◆ Ndau

int EvtCalcGenerator::Ndau

protected

Definition at line 79 of file EvtCalcGenerator.h.

◆ zDecayVolume

Double_t EvtCalcGenerator::zDecayVolume

protected

Definition at line 60 of file EvtCalcGenerator.h.

◆ ztarget

Double_t EvtCalcGenerator::ztarget

protected

Definition at line 60 of file EvtCalcGenerator.h.

The documentation for this class was generated from the following files:

shipgen/EvtCalcGenerator.h
shipgen/EvtCalcGenerator.cxx

Public Member Functions

Protected Types

Protected Member Functions

Protected Attributes

Detailed Description

Member Enumeration Documentation

◆ BranchIndices

Constructor & Destructor Documentation

◆ EvtCalcGenerator()

◆ ~EvtCalcGenerator()

Member Function Documentation

◆ GetBranchValue()

◆ GetDauE()

◆ GetDaughterValue()

◆ GetDauPDG()

◆ GetDauPx()

◆ GetDauPy()

◆ GetDauPz()

◆ GetDecayProb()

◆ GetMotherE()

◆ GetMotherPx()

◆ GetMotherPy()

◆ GetMotherPz()

◆ GetNdaughters()

◆ GetNevents()

◆ GetVx()

◆ GetVy()

◆ GetVz()

◆ Init() [1/2]

◆ Init() [2/2]

◆ ReadEvent()

◆ SetPositions()

Member Data Documentation

◆ branchVars

◆ fInputFile

◆ fn

◆ fNevents

◆ fTree

◆ nBranches

◆ Ndau

◆ zDecayVolume

◆ ztarget