#include <NtupleGenerator_FLUKA.h>

Inheritance diagram for NtupleGenerator_FLUKA:

Collaboration diagram for NtupleGenerator_FLUKA:

[legend]

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, 3)

Protected Attributes
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 12 of file NtupleGenerator_FLUKA.cxx.

12{}

◆ ~NtupleGenerator_FLUKA()

NtupleGenerator_FLUKA::~NtupleGenerator_FLUKA ( )

virtual

destructor

Definition at line 60 of file NtupleGenerator_FLUKA.cxx.

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

Member Function Documentation

◆ ClassDef()

NtupleGenerator_FLUKA::ClassDef	(	NtupleGenerator_FLUKA	,
		3
	)

protected

◆ GetNevents()

Int_t NtupleGenerator_FLUKA::GetNevents ( )

Definition at line 106 of file NtupleGenerator_FLUKA.cxx.

{
 return fNevents;
}

◆ Init() [1/2]

Bool_t NtupleGenerator_FLUKA::Init ( const char * fileName )

virtual

Definition at line 15 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 19 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("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 70 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));
          // what to do with generation info?   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 {
     // what to do with generation info?   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];
   }
 
   return kTRUE;
}

◆ SetZ()

void NtupleGenerator_FLUKA::SetZ ( Double_t X )

inline

Definition at line 31 of file NtupleGenerator_FLUKA.h.

31{ SND_Z = X; };

geometry_config.X

X

Definition geometry_config.py:517

Member Data Documentation

◆ E

Double_t NtupleGenerator_FLUKA::E[1]

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.

◆ 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

Public Member Functions

Protected Member Functions

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

◆ NtupleGenerator_FLUKA()

◆ ~NtupleGenerator_FLUKA()

Member Function Documentation

◆ ClassDef()

◆ GetNevents()

◆ Init() [1/2]

◆ Init() [2/2]

◆ ReadEvent()

◆ SetZ()

Member Data Documentation

◆ E

◆ fInputFile

◆ fn

◆ fNevents

◆ fTree

◆ generation

◆ id

◆ primaries

◆ px

◆ py

◆ pz

◆ SND_Z

◆ t

◆ w

◆ x

◆ y

◆ z