#include <PndFlukaGenerator.h>

Inheritance diagram for PndFlukaGenerator:

Public Member Functions
	PndFlukaGenerator ()

	PndFlukaGenerator (const Char_t *fileName)

virtual	~PndFlukaGenerator ()

virtual Bool_t	ReadEvent (FairPrimaryGenerator *primGen)

Protected Member Functions
Bool_t	ReadRootEvent (FairPrimaryGenerator *primGen)

Bool_t	Init ()

Private Member Functions
Bool_t	SetBranchAddresses ()

void	CloseInput ()
	number of particles in event More...

	ClassDef (PndFlukaGenerator, 2)

Private Attributes
Int_t	iEvent

Int_t	iEntry
	Event number. More...

TString	fFileName
	Root tree entry number. More...

TFile *	fInputRootFile
	Input file name. More...

TTree *	fInputTree
	Pointer to input file. More...

Double_t	fpx
	Pointer to input tree. More...

Double_t	fpy

Double_t	fpz

Double_t	fvx
	Momentum of particle. More...

Double_t	fvy

Double_t	fvz

Int_t	fpid
	Start Vertex of particle. More...

Int_t	fnpart
	PDG code of particle. More...

Detailed Description

Definition at line 31 of file PndFlukaGenerator.h.

Constructor & Destructor Documentation

PndFlukaGenerator::PndFlukaGenerator ( )

Default constructor (should not be used)

Definition at line 29 of file PndFlukaGenerator.cxx.

References fInputRootFile, fInputTree, iEntry, and iEvent.

                                      {
   iEvent     = 0;
   iEntry     = 0;
   fInputRootFile = NULL;
   fInputTree = NULL;
 }

PndFlukaGenerator::PndFlukaGenerator ( const Char_t * fileName )

Standard constructor

Parameters

fileName The input root file name

Definition at line 38 of file PndFlukaGenerator.cxx.

References fFileName, Init(), and TString.

                                                            {
   
   fFileName  = TString(fileName);
   Init();
   
 }

PndFlukaGenerator::~PndFlukaGenerator ( )

virtual

Destructor

Definition at line 46 of file PndFlukaGenerator.cxx.

References CloseInput().

                                       {
   CloseInput();
 
 }

Member Function Documentation

PndFlukaGenerator::ClassDef	(	PndFlukaGenerator	,
		2
	)

private

void PndFlukaGenerator::CloseInput ( )

private

number of particles in event

Private method CloseInput. Just for convenience. Closes the input file properly. Called from destructor and from ReadEvent.

Definition at line 170 of file PndFlukaGenerator.cxx.

References fFileName, and fInputRootFile.

Referenced by ReadRootEvent(), and ~PndFlukaGenerator().

                                    {
   if (fInputRootFile) 
     {
       cout << "-I PndFlukaGenerator: Closing ROOT input file " << fFileName.Data() << endl;
       fInputRootFile->Close();
       delete fInputRootFile;
       fInputRootFile = NULL;
     }
 }

Bool_t PndFlukaGenerator::Init ( )

protected

Definition at line 52 of file PndFlukaGenerator.cxx.

References fFileName, fInputRootFile, fInputTree, iEntry, iEvent, and SetBranchAddresses().

Referenced by PndFlukaGenerator().

 {
   iEvent     = 0;
   iEntry     = 0;  
 
   cout << "-I PndFlukaGenerator: Opening input file " << fFileName.Data() << endl;
   
   // open ROOT input file
   
   fInputRootFile = new TFile(fFileName,"READ");
   if (fInputRootFile->IsZombie())
     Fatal("PndFlukaGenerator","Cannot open ROOT input file.");
   
   fInputTree = (TTree*) fInputRootFile->Get("ResultsTree");
   if (!SetBranchAddresses()) 
     Fatal("PndFlukaGenerator","Incompatible ROOT input file!");
   return kTRUE;  
 }

Bool_t PndFlukaGenerator::ReadEvent ( FairPrimaryGenerator * primGen )

virtual

Reads on event from the input file and pushes the tracks onto the stack. Abstract method in base class.

Parameters

primGen pointer to the FairPrimaryGenerator

Definition at line 110 of file PndFlukaGenerator.cxx.

References fInputRootFile, and ReadRootEvent().

 {
   if ( ! fInputRootFile ) {
     cout << "-E PndFlukaGenerator: Input ROOT file not open!" << endl;
     return kFALSE;
   }
   
   return ReadRootEvent(primGen);
   
 
 }

Bool_t PndFlukaGenerator::ReadRootEvent ( FairPrimaryGenerator * primGen )

protected

Definition at line 125 of file PndFlukaGenerator.cxx.

References CloseInput(), fInputRootFile, fInputTree, fnpart, fpid, fpx, fpy, fpz, fvx, fvy, fvz, g, and iEntry.

Referenced by ReadEvent().

 {
   // Check for number of events in input file
 
   //Int_t nentries = fInputTree->GetEntries(); //[R.K. 01/2017] unused variable?
 
   if ( iEntry > fInputTree->GetEntries() ) {
     cout << "-E PndFlukaGenerator: No more events in input file!" << endl;
     CloseInput();
     return kFALSE;
   }
   
   // preserve orginal TDirectory
   TFile  *g=gFile;
   fInputRootFile->cd();
   g->cd();
   
   Int_t evmul=0;
 
   do 
     { 
       fInputTree->GetEntry(iEntry);
 
       evmul++;
       
       if ( fnpart < evmul ){
         break;
       }
 
 //      cout << "Pid= " << fpid << endl;
 //      cout << "momentum " << fpx << " " << fpy << " " << fpz <<endl;
 //      cout << "vertex   " << fvx << " " << fvy << " " << fvz <<endl;
 
       primGen->AddTrack(fpid,fpx,fpy,fpz,fvx,fvy,fvz);
 
       iEntry++;
 
     } while (evmul <= fnpart);
 
   return kTRUE;
   
 }

Bool_t PndFlukaGenerator::SetBranchAddresses ( )

private

Definition at line 73 of file PndFlukaGenerator.cxx.

References fInputTree, fnpart, fpid, fpx, fpy, fpz, fvx, fvy, and fvz.

Referenced by Init().

 {
     if (0==fInputTree) return false;
     
     fInputTree->SetMakeClass(1);
 
     fInputTree->SetBranchStatus("*",0);
     
     fInputTree->SetBranchStatus("Npart",1);
     //    fInputTree->SetBranchStatus("Weight",1);
     fInputTree->SetBranchStatus("pid",1);
     //    fInputTree->SetBranchStatus("NofInt",1);
     //    fInputTree->SetBranchStatus("etot",1);
     //    fInputTree->SetBranchStatus("ptot",1);
     fInputTree->SetBranchStatus("px",1);
     fInputTree->SetBranchStatus("py",1);
     fInputTree->SetBranchStatus("pz",1);
     //    fInputTree->SetBranchStatus("time",1);
     fInputTree->SetBranchStatus("xx",1);
     fInputTree->SetBranchStatus("yy",1);
     fInputTree->SetBranchStatus("zz",1);
 
     
     fInputTree->SetBranchAddress("Npart",&fnpart);
     fInputTree->SetBranchAddress("pid",&fpid);
     fInputTree->SetBranchAddress("px",&fpx);
     fInputTree->SetBranchAddress("py",&fpy);
     fInputTree->SetBranchAddress("pz",&fpz);
     fInputTree->SetBranchAddress("xx",&fvx);
     fInputTree->SetBranchAddress("yy",&fvy);
     fInputTree->SetBranchAddress("zz",&fvz);
 
    
     return true;
 }