NtupleGenerator Class Reference

#include <NtupleGenerator.h>

Inheritance diagram for NtupleGenerator:

Collaboration diagram for NtupleGenerator:

Public Member Functions
	NtupleGenerator ()
virtual	~NtupleGenerator ()
Bool_t	ReadEvent (FairPrimaryGenerator *)
virtual Bool_t	Init (const char *, int)
virtual Bool_t	Init (const char *)
Int_t	GetNevents ()

Protected Member Functions
	ClassDef (NtupleGenerator, 1)

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

Detailed Description

Definition at line 11 of file NtupleGenerator.h.

Constructor & Destructor Documentation

NtupleGenerator()

NtupleGenerator::NtupleGenerator

(

)

default constructor

Definition at line 14 of file NtupleGenerator.cxx.

{}

~NtupleGenerator()

NtupleGenerator::~NtupleGenerator ( )

virtual

destructor

Definition at line 50 of file NtupleGenerator.cxx.

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

Member Function Documentation

ClassDef()

NtupleGenerator::ClassDef ( NtupleGenerator  , 1    )

protected

GetNevents()

Int_t NtupleGenerator::GetNevents

(

)

Definition at line 92 of file NtupleGenerator.cxx.

{
 return fNevents;
}

Init() [1/2]

Bool_t NtupleGenerator::Init ( const char *  fileName )

virtual

Definition at line 17 of file NtupleGenerator.cxx.

                                                 {
  return Init(fileName, 0);
}

Init() [2/2]

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

virtual

Definition at line 21 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 = (TTree *)fInputFile->Get("ntuple");
  fNevents = fTree->GetEntries();
  fn = firstEvent;
  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

)

public method ReadEvent

Definition at line 60 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 30 of file NtupleGenerator.h.

fInputFile

TFile* NtupleGenerator::fInputFile

protected

Definition at line 33 of file NtupleGenerator.h.

fLogger

FairLogger* NtupleGenerator::fLogger

protected

Definition at line 35 of file NtupleGenerator.h.

fn

int NtupleGenerator::fn

protected

Definition at line 37 of file NtupleGenerator.h.

fNevents

int NtupleGenerator::fNevents

protected

don't make it persistent, magic ROOT command

Definition at line 36 of file NtupleGenerator.h.

fTree

TTree* NtupleGenerator::fTree

protected

Definition at line 34 of file NtupleGenerator.h.

id

Int_t NtupleGenerator::id

protected

Definition at line 29 of file NtupleGenerator.h.

Nmeas

Int_t NtupleGenerator::Nmeas

protected

Definition at line 29 of file NtupleGenerator.h.

parentid

Int_t NtupleGenerator::parentid

protected

Definition at line 29 of file NtupleGenerator.h.

procid

Int_t NtupleGenerator::procid[500]

protected

Definition at line 29 of file NtupleGenerator.h.

px

Float_t NtupleGenerator::px[500]

protected

Definition at line 32 of file NtupleGenerator.h.

py

Float_t NtupleGenerator::py[500]

protected

Definition at line 32 of file NtupleGenerator.h.

pz

Float_t NtupleGenerator::pz[500]

protected

Definition at line 32 of file NtupleGenerator.h.

tof

Float_t NtupleGenerator::tof

protected

Definition at line 30 of file NtupleGenerator.h.

volid

Int_t NtupleGenerator::volid[500]

protected

Definition at line 29 of file NtupleGenerator.h.

vx

Float_t NtupleGenerator::vx[500]

protected

Definition at line 32 of file NtupleGenerator.h.

vy

Float_t NtupleGenerator::vy[500]

protected

Definition at line 32 of file NtupleGenerator.h.

vz

Float_t NtupleGenerator::vz[500]

protected

Definition at line 32 of file NtupleGenerator.h.

w

Double_t NtupleGenerator::w

protected

Definition at line 31 of file NtupleGenerator.h.

---

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

• shipgen/NtupleGenerator.h
• shipgen/NtupleGenerator.cxx