PndKFParticleFinderPID Class Reference

#include <PndKFParticleFinderPID.h>

Inheritance diagram for PndKFParticleFinderPID:

Public Member Functions
	PndKFParticleFinderPID (const char *name="PndKFParticleFinderPID", Int_t iVerbose=0)
	~PndKFParticleFinderPID ()
void	SetMCTrackBranchName (const TString &name)
void	SetChargedTrackBranchName (const TString &name)
void	SetNeutralTrackBranchName (const TString &name)
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)
virtual void	Finish ()
void	SetPIDMode (int mode)
void	UseNoPID ()
void	UseMCPID ()
void	UseRecoPID ()
const std::vector< int > &	GetPID () const

Private Member Functions
void	SetMCPID ()
void	SetRecoPID ()
	ClassDef (PndKFParticleFinderPID, 1)

Private Attributes
TString	fChargedTrackBranchName
TString	fNeutralTrackBranchName
	Name of the input TCA. More...
TString	fMCTracksBranchName
	Name of the input TCA. More...
TClonesArray *	fChargedTrackArray
	Name of the input TCA with MC tracks. More...
TClonesArray *	fNeutralTrackArray
TClonesArray *	fMCTrackArray
Int_t	fPIDMode
std::vector< int >	fPID

Detailed Description

Definition at line 17 of file PndKFParticleFinderPID.h.

Constructor & Destructor Documentation

PndKFParticleFinderPID::PndKFParticleFinderPID	(	const char *	name = "PndKFParticleFinderPID",
		Int_t	iVerbose = 0
	)

Definition at line 17 of file PndKFParticleFinderPID.cxx.

                                                                              :
  FairTask(name, iVerbose), fChargedTrackBranchName(""), fNeutralTrackBranchName(""), fMCTracksBranchName(""),
  fChargedTrackArray(0), fNeutralTrackArray(0), fMCTrackArray(0),
  fPIDMode(0), fPID(0)
{
}

PndKFParticleFinderPID::~PndKFParticleFinderPID

(

)

Definition at line 24 of file PndKFParticleFinderPID.cxx.

{
}

Member Function Documentation

PndKFParticleFinderPID::ClassDef ( PndKFParticleFinderPID  , 1    )

private

void PndKFParticleFinderPID::Exec ( Option_t *  opt )

virtual

Definition at line 68 of file PndKFParticleFinderPID.cxx.

References fChargedTrackArray, fPID, fPIDMode, SetMCPID(), and SetRecoPID().

{ 
  fPID.clear();
  
  Int_t nTracks=fChargedTrackArray->GetEntriesFast();
  fPID.resize(nTracks, -1);

  if(fPIDMode==1)
    SetMCPID();
  if(fPIDMode==2)
    SetRecoPID();
}

void PndKFParticleFinderPID::Finish ( )

virtual

Definition at line 81 of file PndKFParticleFinderPID.cxx.

{
}

const std::vector<int>& PndKFParticleFinderPID::GetPID ( ) const

inline

Definition at line 36 of file PndKFParticleFinderPID.h.

References fPID.

Referenced by PndKFParticleFinder::Exec().

{ return fPID; }

InitStatus PndKFParticleFinderPID::Init ( )

virtual

Definition at line 28 of file PndKFParticleFinderPID.cxx.

References fChargedTrackArray, fChargedTrackBranchName, fMCTrackArray, fMCTracksBranchName, fNeutralTrackArray, fNeutralTrackBranchName, and fPIDMode.

{
  //Get ROOT Manager
  FairRootManager* ioman= FairRootManager::Instance();
  
  if(ioman==0)
  {
    Error("PndKFParticleFinderPID::Init","RootManager not instantiated!");
    return kERROR;
  }
  
  // Get charged tracks
  fChargedTrackArray=(TClonesArray*) ioman->GetObject(fChargedTrackBranchName);
  if(fChargedTrackArray==0)
  {
    Error("PndKFParticleFinderPID::Init","Charged track array not found! Stop the execution.");
    return kERROR;
  }
  
  // Get neutral tracks
  fNeutralTrackArray=(TClonesArray*) ioman->GetObject(fNeutralTrackBranchName);
  if(fNeutralTrackArray==0)
  {
    Error("PndKFParticleFinderPID::Init","Neutral track array not found! Will continue without gamma reconstruction.");
  }
  
  if(fPIDMode==1)
  {
    //MC Tracks
    fMCTrackArray=(TClonesArray*) ioman->GetObject(fMCTracksBranchName);
    if(fMCTrackArray==0)
    {
      Error("StandaloneHitGenerator::Init","MC track array not found! Stop the execution.");
      return kERROR;
    }
  }
  
  return kSUCCESS;
}

void PndKFParticleFinderPID::SetChargedTrackBranchName ( const TString &  name )

inline

Definition at line 25 of file PndKFParticleFinderPID.h.

References fChargedTrackBranchName, and name.

Referenced by kfparticle().

{ fChargedTrackBranchName = name;  }

void PndKFParticleFinderPID::SetMCPID ( )

private

Definition at line 85 of file PndKFParticleFinderPID.cxx.

References CAMath::Abs(), fChargedTrackArray, fMCTrackArray, fPID, PndPidCandidate::GetMcIndex(), and PndMCTrack::GetPdgCode().

Referenced by Exec().

{ 
  Int_t nTracks=fChargedTrackArray->GetEntriesFast();
  for(int iTr=0; iTr<nTracks; iTr++)
  {
    PndPidCandidate *inTrack = (PndPidCandidate*)fChargedTrackArray->At(iTr);
    if(!inTrack) continue;
    Int_t iMC = inTrack->GetMcIndex();
    
    if(iMC >= 0)
    {
      PndMCTrack *pndMCTrack = (PndMCTrack*)fMCTrackArray->At(iMC);
      if(TMath::Abs(pndMCTrack->GetPdgCode()) > 2500)
        continue;
      fPID[iTr] = pndMCTrack->GetPdgCode();
    }
  }
}

void PndKFParticleFinderPID::SetMCTrackBranchName ( const TString &  name )

inline

Definition at line 24 of file PndKFParticleFinderPID.h.

References fMCTracksBranchName, and name.

Referenced by kfparticle().

{ fMCTracksBranchName = name;  }

void PndKFParticleFinderPID::SetNeutralTrackBranchName ( const TString &  name )

inline

Definition at line 26 of file PndKFParticleFinderPID.h.

References fNeutralTrackBranchName, and name.

Referenced by kfparticle().

{ fNeutralTrackBranchName = name;  }

void PndKFParticleFinderPID::SetPIDMode ( int  mode )

inline

Definition at line 32 of file PndKFParticleFinderPID.h.

References fPIDMode, and mode.

Referenced by kfparticle().

{ fPIDMode = mode; }

void PndKFParticleFinderPID::SetRecoPID ( )

private

Definition at line 104 of file PndKFParticleFinderPID.cxx.

Referenced by Exec().

{
}

void PndKFParticleFinderPID::UseMCPID ( )

inline

Definition at line 34 of file PndKFParticleFinderPID.h.

References fPIDMode.

{ fPIDMode = 1; }

void PndKFParticleFinderPID::UseNoPID ( )

inline

Definition at line 33 of file PndKFParticleFinderPID.h.

References fPIDMode.

{ fPIDMode = 0; }

void PndKFParticleFinderPID::UseRecoPID ( )

inline

Definition at line 35 of file PndKFParticleFinderPID.h.

References fPIDMode.

{ fPIDMode = 2; }

Member Data Documentation

TClonesArray* PndKFParticleFinderPID::fChargedTrackArray

private

Name of the input TCA with MC tracks.

Definition at line 50 of file PndKFParticleFinderPID.h.

Referenced by Exec(), Init(), and SetMCPID().

TString PndKFParticleFinderPID::fChargedTrackBranchName

private

Definition at line 45 of file PndKFParticleFinderPID.h.

Referenced by Init(), and SetChargedTrackBranchName().

TClonesArray* PndKFParticleFinderPID::fMCTrackArray

private

Definition at line 52 of file PndKFParticleFinderPID.h.

Referenced by Init(), and SetMCPID().

TString PndKFParticleFinderPID::fMCTracksBranchName

private

Name of the input TCA.

Definition at line 47 of file PndKFParticleFinderPID.h.

Referenced by Init(), and SetMCTrackBranchName().

TClonesArray* PndKFParticleFinderPID::fNeutralTrackArray

private

Definition at line 51 of file PndKFParticleFinderPID.h.

Referenced by Init().

TString PndKFParticleFinderPID::fNeutralTrackBranchName

private

Name of the input TCA.

Definition at line 46 of file PndKFParticleFinderPID.h.

Referenced by Init(), and SetNeutralTrackBranchName().

std::vector<int> PndKFParticleFinderPID::fPID

private

Definition at line 56 of file PndKFParticleFinderPID.h.

Referenced by Exec(), GetPID(), and SetMCPID().

Int_t PndKFParticleFinderPID::fPIDMode

private

Definition at line 55 of file PndKFParticleFinderPID.h.

Referenced by Exec(), Init(), SetPIDMode(), UseMCPID(), UseNoPID(), and UseRecoPID().

---

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

• PndKFParticleFinderPID.h
• PndKFParticleFinderPID.cxx