NtupleGenerator_FLUKA Class Reference

#include <NtupleGenerator_FLUKA.h>

Inheritance diagram for NtupleGenerator_FLUKA:

Collaboration diagram for NtupleGenerator_FLUKA:

Public Member Functions
	NtupleGenerator_FLUKA ()
virtual	~NtupleGenerator_FLUKA ()
Bool_t	ReadEvent (FairPrimaryGenerator *)
virtual Bool_t	Init (const char *, int)
virtual Bool_t	Init (const char *)
Int_t	GetNevents ()
void	SetZ (Double_t X)

Protected Member Functions
	ClassDef (NtupleGenerator_FLUKA, 4)

Protected Attributes
Double_t	runN
Double_t	eventN
Double_t	id [1]
Double_t	generation [1]
Double_t	E [1]
Double_t	t [1]
Double_t	px [1]
Double_t	py [1]
Double_t	pz [1]
Double_t	x [1]
Double_t	y [1]
Double_t	z [1]
Double_t	w [1]
Double_t	SND_Z
TFile *	fInputFile
TTree *	fTree
int	fNevents
int	fn
TClonesArray *	primaries

Detailed Description

Definition at line 16 of file NtupleGenerator_FLUKA.h.

Constructor & Destructor Documentation

NtupleGenerator_FLUKA()

NtupleGenerator_FLUKA::NtupleGenerator_FLUKA

(

)

default constructor

Definition at line 13 of file NtupleGenerator_FLUKA.cxx.

{}

~NtupleGenerator_FLUKA()

NtupleGenerator_FLUKA::~NtupleGenerator_FLUKA ( )

virtual

destructor

Definition at line 63 of file NtupleGenerator_FLUKA.cxx.

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

Member Function Documentation

ClassDef()

NtupleGenerator_FLUKA::ClassDef ( NtupleGenerator_FLUKA  , 4    )

protected

GetNevents()

Int_t NtupleGenerator_FLUKA::GetNevents

(

)

Definition at line 114 of file NtupleGenerator_FLUKA.cxx.

{
 return fNevents;
}

Init() [1/2]

Bool_t NtupleGenerator_FLUKA::Init ( const char *  fileName )

virtual

Definition at line 16 of file NtupleGenerator_FLUKA.cxx.

                                                       {
  return Init(fileName, 0);
}

Init() [2/2]

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

virtual

Definition at line 20 of file NtupleGenerator_FLUKA.cxx.

                                                                             {
  TString fN = "";
   if (0 == strncmp("/eos",fileName,4) ) {
      fN = gSystem->Getenv("EOSSHIP");
   }
  fN+=fileName;
  fInputFile  = TFile::Open(fN);
  if (fInputFile->IsZombie()) {
    LOG(FATAL) << "NtupleGenerator_FLUKA: Error opening the Signal file" << fN;
    return kFALSE;
  }
  LOG(INFO) << "NtupleGenerator_FLUKA: Opening input file " << fN;
 
  fTree = (TTree *)fInputFile->Get("nt");
 
  fNevents = fTree->GetEntries();
  fn = firstEvent;
  primaries = 0;
  // Check the input TTree structure
  if (fTree->FindLeaf("primaries_") != nullptr) {     
     fTree->SetBranchAddress("primaries",&primaries);
  }
  else {  
    fTree->SetBranchAddress("run",&runN); // run number
    fTree->SetBranchAddress("event",&eventN); // event number
    fTree->SetBranchAddress("id",&id);                // particle id
    fTree->SetBranchAddress("generation",&generation);    //  origin generation number
    fTree->SetBranchAddress("t",&t);             // time of flight
    fTree->SetBranchAddress("E",&E);          // incoming muon energy
    fTree->SetBranchAddress("w",&w);                  // weight of event
    fTree->SetBranchAddress("x",&x);   // position
    fTree->SetBranchAddress("y",&y);
    fTree->SetBranchAddress("z",&z);
    fTree->SetBranchAddress("px",&px);   // momentum
    fTree->SetBranchAddress("py",&py);
    fTree->SetBranchAddress("pz",&pz);
  }
  return kTRUE;
}

ReadEvent()

Bool_t NtupleGenerator_FLUKA::ReadEvent

(

FairPrimaryGenerator *

cpg

)

public method ReadEvent

Definition at line 73 of file NtupleGenerator_FLUKA.cxx.

{
   fTree->GetEntry(fn);
   fn++;
   if (fn %10000==0)  {LOG(INFO)<< "reading event "<<fn;}
   if (fn > fNevents) {
     LOG(WARNING) << "No more input events";
     return kFALSE; }   
   /* Add all primary tracks in the event. Typically, muon bkg MC has single muon per event. 
      Several muon primaries per event exist in the multi-muon MC.
   */
   if (fTree->FindLeaf("primaries_") != nullptr) {
      int nf = primaries->GetEntries();
      for(int i=0; i<nf; i++){
          //casting
          struct PrimaryTrack* primary = dynamic_cast<struct PrimaryTrack*>(primaries->AddrAt(i));
          //convert time from ns to sec
          cpg->AddTrack(int(primary->id),primary->px,primary->py,primary->pz,
                        primary->x,primary->y,primary->z-SND_Z,-1,true,primary->E,
                        primary->t/1E9,primary->w,(TMCProcess)primary->generation);
          LOG(DEBUG) << "NtupleGenerator_FLUKA: add muon " << i << "," << int(primary->id)
                     <<","<<primary->px<<","<<primary->py<<","<<primary->pz<<","<<primary->x
                     <<","<<primary->y<<","<<primary->z-SND_Z<<","<<primary->generation
                     <<","<<primary->E<<","<<primary->t<<"ns,"<<primary->w;
      }
   }
   else {
     // convert time from ns to sec
     cpg->AddTrack(int(id[0]),px[0],py[0],pz[0],x[0],y[0],z[0]-SND_Z,-1,true,E[0],t[0]/1E9,w[0],(TMCProcess)generation[0]);
     LOG(DEBUG) << "NtupleGenerator_FLUKA: add muon " << id[0]<<","<<px[0]<<","<<py[0]<<","<<pz[0]<<","<<x[0]<<","<<y[0]<<","<<z[0]-SND_Z<<","<<generation[0]<<","<<E[0]<<","<<t[0]<<"ns,"<<w[0];
   }
 
   // Set the generation run and event numbers through the MC event header
   FairMCEventHeader* header = cpg->GetEvent();
   header->SetEventID(int(eventN));
   header->SetRunID(int(runN));
 
   return kTRUE;
}

SetZ()

void NtupleGenerator_FLUKA::SetZ ( Double_t  X )

inline

Definition at line 31 of file NtupleGenerator_FLUKA.h.

{  SND_Z = X; };

Member Data Documentation

E

Double_t NtupleGenerator_FLUKA::E[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

eventN

Double_t NtupleGenerator_FLUKA::eventN

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

fInputFile

TFile* NtupleGenerator_FLUKA::fInputFile

protected

Definition at line 36 of file NtupleGenerator_FLUKA.h.

fn

int NtupleGenerator_FLUKA::fn

protected

Definition at line 39 of file NtupleGenerator_FLUKA.h.

fNevents

int NtupleGenerator_FLUKA::fNevents

protected

Definition at line 38 of file NtupleGenerator_FLUKA.h.

fTree

TTree* NtupleGenerator_FLUKA::fTree

protected

Definition at line 37 of file NtupleGenerator_FLUKA.h.

generation

Double_t NtupleGenerator_FLUKA::generation[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

id

Double_t NtupleGenerator_FLUKA::id[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

primaries

TClonesArray* NtupleGenerator_FLUKA::primaries

protected

Definition at line 40 of file NtupleGenerator_FLUKA.h.

px

Double_t NtupleGenerator_FLUKA::px[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

py

Double_t NtupleGenerator_FLUKA::py[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

pz

Double_t NtupleGenerator_FLUKA::pz[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

runN

Double_t NtupleGenerator_FLUKA::runN

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

SND_Z

Double_t NtupleGenerator_FLUKA::SND_Z

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

t

Double_t NtupleGenerator_FLUKA::t[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

w

Double_t NtupleGenerator_FLUKA::w[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

x

Double_t NtupleGenerator_FLUKA::x[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

y

Double_t NtupleGenerator_FLUKA::y[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

z

Double_t NtupleGenerator_FLUKA::z[1]

protected

Definition at line 35 of file NtupleGenerator_FLUKA.h.

---

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

• shipgen/NtupleGenerator_FLUKA.h
• shipgen/NtupleGenerator_FLUKA.cxx