#include <NtupleGenerator.h>

Inheritance diagram for NtupleGenerator:

Collaboration diagram for NtupleGenerator:

Public Member Functions
	NtupleGenerator ()

	~NtupleGenerator () override

Bool_t	ReadEvent (FairPrimaryGenerator *) override

Bool_t	Init (const char *, int) override

Bool_t	Init (const char *) override

Int_t	GetNevents ()

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 Attributes
Int_t	id

Int_t	Nmeas

Int_t	volid [500]

Int_t	procid [500]

Int_t	parentid

Float_t	Ezero

Float_t	tof

Double_t	w

Float_t	px [500]

Float_t	py [500]

Float_t	pz [500]

Float_t	vx [500]

Float_t	vy [500]

Float_t	vz [500]

TFile *	fInputFile

TTree *	fTree

FairLogger *	fLogger

int	fNevents
	don't make it persistent, magic ROOT command

int	fn

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

Int_t	firstEvent = 0

Detailed Description

Definition at line 15 of file NtupleGenerator.h.

Constructor & Destructor Documentation

◆ NtupleGenerator()

NtupleGenerator::NtupleGenerator ( )

default constructor

Definition at line 20 of file NtupleGenerator.cxx.

20{}

◆ ~NtupleGenerator()

NtupleGenerator::~NtupleGenerator ( )

override

destructor

Definition at line 58 of file NtupleGenerator.cxx.

                                  {
  // cout << "destroy Ntuple" << endl;
  fInputFile->Close();
  fInputFile->Delete();
  delete fInputFile;
}

Member Function Documentation

◆ GetNevents()

Int_t NtupleGenerator::GetNevents ( )

Definition at line 106 of file NtupleGenerator.cxx.

106{ return fNevents; }

NtupleGenerator::fNevents

int fNevents

don't make it persistent, magic ROOT command

Definition: NtupleGenerator.h:39

◆ Init() [1/2]

Bool_t NtupleGenerator::Init ( const char * fileName )

overridevirtual

Implements SHiP::Generator.

Definition at line 23 of file NtupleGenerator.cxx.

23{ return Init(fileName, 0); }

NtupleGenerator::Init

Bool_t Init(const char *, int) override

Definition: NtupleGenerator.cxx:25

◆ Init() [2/2]

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

overridevirtual

Implements SHiP::Generator.

Definition at line 25 of file NtupleGenerator.cxx.

                                                                       {
  cout << "Info NtupleGenerator: Opening input file " << fileName << endl;
  fInputFile = new TFile(fileName);
  if (fInputFile->IsZombie()) {
    cout << "-E NtupleGenerator: Error opening the Signal file" << fileName
         << endl;
  }
  fTree = dynamic_cast<TTree*>(fInputFile->Get("ntuple"));
  fNevents = fTree->GetEntries();
  fn = startEvent;
  fTree->SetBranchAddress("id", &id);  // particle id
  if (fTree->FindBranch("parentid")) {
    fTree->SetBranchAddress("parentid", &parentid);
  }  // parent id
  if (fTree->FindBranch("tof")) {
    fTree->SetBranchAddress("tof", &tof);
  }  // time of flight
  fTree->SetBranchAddress("Nmeas", &Nmeas);  // number of Geant4 points
  fTree->SetBranchAddress("Ezero", &Ezero);  // incoming muon energy
  fTree->SetBranchAddress("w", &w);          // weight of event
  fTree->SetBranchAddress("x", &vx);         // position
  fTree->SetBranchAddress("y", &vy);
  fTree->SetBranchAddress("z", &vz);
  fTree->SetBranchAddress("px", &px);  // momentum
  fTree->SetBranchAddress("py", &py);
  fTree->SetBranchAddress("pz", &pz);
  fTree->SetBranchAddress("volid", &volid);    // which volume
  fTree->SetBranchAddress("procid", &procid);  // which process
  return kTRUE;
}

◆ ReadEvent()

Bool_t NtupleGenerator::ReadEvent ( FairPrimaryGenerator * cpg )

override

Definition at line 67 of file NtupleGenerator.cxx.

                                                           {
  while (fn < fNevents) {
    fTree->GetEntry(fn);
    fn++;
    if (fn % 10000 == 0) {
      cout << "reading event " << fn << endl;
    }
    // test if muon survives:
    Int_t i = Nmeas - 3;
    Float_t r2 = (vx[i] * vx[i] + vy[i] * vy[i]);
    if (procid[Nmeas - 1] == 2 && r2 < 9) {
      break;
    }
  }
  if (fn == fNevents) {
    cout << "No more input events" << endl;
    return kFALSE;
  }
  TDatabasePDG* pdgBase = TDatabasePDG::Instance();
  Double_t mass = pdgBase->GetParticle(id)->Mass();
  Double_t e =
      TMath::Sqrt(px[0] * px[0] + py[0] * py[0] + pz[0] * pz[0] + mass * mass);
  tof = 0;
  // first, original muon
  cpg->AddTrack(id, px[0], py[0], pz[0], vx[0] * 100., vy[0] * 100.,
                vz[0] * 100., -1., false, e, tof, w);
  Int_t i = Nmeas - 1;
  // second, surviving muon, extrapolate back to end of muon shield, z=20m
  Double_t zscor = 20.;
  Double_t lam = (zscor - vz[i]) / pz[i];
  Double_t xscor = vx[i] + lam * px[i];
  Double_t yscor = vy[i] + lam * py[i];
  e = TMath::Sqrt(px[i] * px[i] + py[i] * py[i] + pz[i] * pz[i] + mass * mass);
  cpg->AddTrack(id, px[i], py[i], pz[i], xscor * 100., yscor * 100.,
                zscor * 100., 0, true, e, tof, w);
  return kTRUE;
}

Member Data Documentation

◆ Ezero

Float_t NtupleGenerator::Ezero

protected

Definition at line 33 of file NtupleGenerator.h.

◆ fInputFile

TFile* NtupleGenerator::fInputFile

protected

Definition at line 36 of file NtupleGenerator.h.

◆ fLogger

FairLogger* NtupleGenerator::fLogger

protected

Definition at line 38 of file NtupleGenerator.h.

◆ fn

int NtupleGenerator::fn

protected

Definition at line 40 of file NtupleGenerator.h.

◆ fNevents

int NtupleGenerator::fNevents

protected

don't make it persistent, magic ROOT command

Definition at line 39 of file NtupleGenerator.h.

◆ fTree

TTree* NtupleGenerator::fTree

protected

Definition at line 37 of file NtupleGenerator.h.

◆ id

Int_t NtupleGenerator::id

protected

Definition at line 32 of file NtupleGenerator.h.

◆ Nmeas

Int_t NtupleGenerator::Nmeas

protected

Definition at line 32 of file NtupleGenerator.h.

◆ parentid

Int_t NtupleGenerator::parentid

protected

Definition at line 32 of file NtupleGenerator.h.

◆ procid

Int_t NtupleGenerator::procid[500]

protected

Definition at line 32 of file NtupleGenerator.h.

◆ px

Float_t NtupleGenerator::px[500]

protected

Definition at line 35 of file NtupleGenerator.h.

◆ py

Float_t NtupleGenerator::py[500]

protected

Definition at line 35 of file NtupleGenerator.h.

◆ pz

Float_t NtupleGenerator::pz[500]

protected

Definition at line 35 of file NtupleGenerator.h.

◆ tof

Float_t NtupleGenerator::tof

protected

Definition at line 33 of file NtupleGenerator.h.

◆ volid

Int_t NtupleGenerator::volid[500]

protected

Definition at line 32 of file NtupleGenerator.h.

◆ vx

Float_t NtupleGenerator::vx[500]

protected

Definition at line 35 of file NtupleGenerator.h.

◆ vy

Float_t NtupleGenerator::vy[500]

protected

Definition at line 35 of file NtupleGenerator.h.

◆ vz

Float_t NtupleGenerator::vz[500]

protected

Definition at line 35 of file NtupleGenerator.h.

◆ w

Double_t NtupleGenerator::w

protected

Definition at line 34 of file NtupleGenerator.h.

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

shipgen/NtupleGenerator.h
shipgen/NtupleGenerator.cxx

Public Member Functions

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

◆ NtupleGenerator()

◆ ~NtupleGenerator()

Member Function Documentation

◆ GetNevents()

◆ Init() [1/2]

◆ Init() [2/2]

◆ ReadEvent()

Member Data Documentation

◆ Ezero

◆ fInputFile

◆ fLogger

◆ fn

◆ fNevents

◆ fTree

◆ id

◆ Nmeas

◆ parentid

◆ procid

◆ px

◆ py

◆ pz

◆ tof

◆ volid

◆ vx

◆ vy

◆ vz

◆ w