PndKFParticleFinder Class Reference

#include <PndKFParticleFinder.h>

Inheritance diagram for PndKFParticleFinder:

Public Member Functions
	PndKFParticleFinder (const char *name="PndKFParticleFinder", Int_t iVerbose=0)
	~PndKFParticleFinder ()
void	SetPVToZero ()
void	ReconstructSinglePV ()
void	RconstructMultiplePV ()
void	SetChargedTrackBranchName (const TString &name)
void	SetNeutralTrackBranchName (const TString &name)
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)
const KFParticleTopoReconstructor *	GetTopoReconstructor () const
void	SetPIDInformation (PndKFParticleFinderPID *pid)
void	SetPrimaryProbCut (float prob)

Private Member Functions
double	InversedChi2Prob (double p, int ndf) const
const PndKFParticleFinder &	operator= (const PndKFParticleFinder &)
	PndKFParticleFinder (const PndKFParticleFinder &)
	ClassDef (PndKFParticleFinder, 1)

Private Attributes
TString	fChargedTrackBranchName
TString	fNeutralTrackBranchName
	Name of the input TCA. More...
TClonesArray *	fChargedTrackArray
	Name of the input TCA. More...
TClonesArray *	fNeutralTrackArray
TClonesArray *	fEmcBumps
KFParticleTopoReconstructor *	fTopoReconstructor
int	fPVFindMode
PndKFParticleFinderPID *	fPID

Detailed Description

Definition at line 17 of file PndKFParticleFinder.h.

Constructor & Destructor Documentation

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

Definition at line 28 of file PndKFParticleFinder.cxx.

References fTopoReconstructor, and SetPrimaryProbCut().

                                                                        :
  FairTask(name, iVerbose), fChargedTrackBranchName(""), fNeutralTrackBranchName(""), fChargedTrackArray(0), fNeutralTrackArray(0), fEmcBumps(0),
  fTopoReconstructor(0), fPVFindMode(0), fPID(0)
{
  fTopoReconstructor = new KFParticleTopoReconstructor;
  // set default cuts
  SetPrimaryProbCut(0.0001); // 0.01% to consider primary track as a secondary;
}

PndKFParticleFinder::~PndKFParticleFinder

(

)

Definition at line 37 of file PndKFParticleFinder.cxx.

References fTopoReconstructor.

{
  if(fTopoReconstructor) delete fTopoReconstructor;
}

PndKFParticleFinder::PndKFParticleFinder ( const PndKFParticleFinder &  )

private

Member Function Documentation

PndKFParticleFinder::ClassDef ( PndKFParticleFinder  , 1    )

private

void PndKFParticleFinder::Exec ( Option_t *  opt )

virtual

Definition at line 78 of file PndKFParticleFinder.cxx.

References PndEmcCluster::energy(), fChargedTrackArray, fEmcBumps, fNeutralTrackArray, fPID, fPVFindMode, fTopoReconstructor, PndPidCandidate::GetCharge(), PndPidCandidate::GetChiSquared(), PndPidCandidate::GetDegreesOfFreedom(), PndPidCandidate::GetEmcIndex(), PndPidCandidate::GetEmcQuality(), PndPidCandidate::GetErrorP7(), PndPidCandidate::GetLastHit(), PndPidCandidate::GetMomentum(), PndKFParticleFinderPID::GetPID(), PndPidCandidate::GetPosition(), ok, par, KFPVertex::SetChi2(), KFPVertex::SetCovarianceMatrix(), KFPVertex::SetNContributors(), KFPVertex::SetXYZ(), and timer.

{
  Int_t ntracks=0;//fChargedTrackArray->GetEntriesFast();

  vector<PndPidCandidate> vRTracks(fChargedTrackArray->GetEntriesFast());
  vector<int> pdg(fChargedTrackArray->GetEntriesFast(), -1);
  vector<int> trackId(fChargedTrackArray->GetEntriesFast(), -1);
    
  for(int iTr=0; iTr<fChargedTrackArray->GetEntriesFast(); iTr++)
  {
    PndPidCandidate *inTrack = (PndPidCandidate*)fChargedTrackArray->At(iTr);
  
    bool ok = 1;
    ok = ok && std::isfinite(inTrack->GetPosition().x());
    ok = ok && std::isfinite(inTrack->GetPosition().y());
    ok = ok && std::isfinite(inTrack->GetPosition().z());
    ok = ok && std::isfinite(inTrack->GetMomentum().x());
    ok = ok && std::isfinite(inTrack->GetMomentum().y());
    ok = ok && std::isfinite(inTrack->GetMomentum().z());
    
    float cov[21];
    cov[ 0] = inTrack->GetErrorP7()[0];
    cov[ 1] = inTrack->GetErrorP7()[1];
    cov[ 2] = inTrack->GetErrorP7()[2];
    cov[ 3] = inTrack->GetErrorP7()[3];
    cov[ 4] = inTrack->GetErrorP7()[4];
    cov[ 5] = inTrack->GetErrorP7()[5];
    cov[ 6] = inTrack->GetErrorP7()[6];
    cov[ 7] = inTrack->GetErrorP7()[7];
    cov[ 8] = inTrack->GetErrorP7()[8];
    cov[ 9] = inTrack->GetErrorP7()[18];
    cov[10] = inTrack->GetErrorP7()[9];
    cov[11] = inTrack->GetErrorP7()[10];
    cov[12] = inTrack->GetErrorP7()[11];
    cov[13] = inTrack->GetErrorP7()[19];
    cov[14] = inTrack->GetErrorP7()[20];
    cov[15] = inTrack->GetErrorP7()[12];
    cov[16] = inTrack->GetErrorP7()[13];
    cov[17] = inTrack->GetErrorP7()[14];
    cov[18] = inTrack->GetErrorP7()[21];
    cov[19] = inTrack->GetErrorP7()[22];
    cov[20] = inTrack->GetErrorP7()[23];
    
      
    for(unsigned short iC=0; iC<21; iC++)
      ok = ok && std::isfinite(cov[iC]);
//     ok = ok && (cov[0] < 100. && cov[0] > 0.)
//             && (cov[2] < 100. && cov[2] > 0.)
//             && (cov[5] < 100. && cov[5] > 0.)
//             && (cov[9] < 100. && cov[9] > 0.)
//             && (cov[14] < 100. && cov[14] > 0.)
//             && (cov[20] < 100. && cov[20] > 0.);
    ok = ok && (cov[0] > 0.)
            && (cov[2] > 0.)
            && (cov[5] > 0.)
            && (cov[9] > 0.)
            && (cov[14] > 0.)
            && (cov[20] > 0.);
            
    ok = ok && inTrack->GetChiSquared() < 20*inTrack->GetDegreesOfFreedom();
    if(!ok) continue;
    
    if(fPID)
    {
      if(fPID->GetPID()[iTr] == -2) continue; 
      pdg[ntracks] = fPID->GetPID()[iTr];
    }
    vRTracks[ntracks] = *inTrack;
    trackId[ntracks] = iTr;
    
    ntracks++;
  }
  
  vRTracks.resize(ntracks);
  pdg.resize(ntracks);
  trackId.resize(ntracks);
  
  KFPTrackVector tracks;
  tracks.Resize(ntracks);
  //fill vector with tracks
  for(Int_t iTr=0; iTr<ntracks; iTr++)
  {
    PndPidCandidate *inTrack = (PndPidCandidate*)fChargedTrackArray->At(trackId[iTr]);
    
    float par[6] = {0.f};
    par[0] = inTrack->GetPosition().x();
    par[1] = inTrack->GetPosition().y();
    par[2] = inTrack->GetPosition().z();
    par[3] = inTrack->GetMomentum().x();
    par[4] = inTrack->GetMomentum().y();
    par[5] = inTrack->GetMomentum().z();
    
    Int_t q = inTrack->GetCharge();
    
    float cov[21];
    cov[ 0] = inTrack->GetErrorP7()[0];
    cov[ 1] = inTrack->GetErrorP7()[1];
    cov[ 2] = inTrack->GetErrorP7()[2];
    cov[ 3] = inTrack->GetErrorP7()[3];
    cov[ 4] = inTrack->GetErrorP7()[4];
    cov[ 5] = inTrack->GetErrorP7()[5];
    cov[ 6] = inTrack->GetErrorP7()[6];
    cov[ 7] = inTrack->GetErrorP7()[7];
    cov[ 8] = inTrack->GetErrorP7()[8];
    cov[ 9] = inTrack->GetErrorP7()[18];
    cov[10] = inTrack->GetErrorP7()[9];
    cov[11] = inTrack->GetErrorP7()[10];
    cov[12] = inTrack->GetErrorP7()[11];
    cov[13] = inTrack->GetErrorP7()[19];
    cov[14] = inTrack->GetErrorP7()[20];
    cov[15] = inTrack->GetErrorP7()[12];
    cov[16] = inTrack->GetErrorP7()[13];
    cov[17] = inTrack->GetErrorP7()[14];
    cov[18] = inTrack->GetErrorP7()[21];
    cov[19] = inTrack->GetErrorP7()[22];
    cov[20] = inTrack->GetErrorP7()[23];
    
    for(Int_t iP=0; iP<6; iP++)
      tracks.SetParameter(par[iP], iP, iTr);    
    for(Int_t iC=0; iC<21; iC++)
      tracks.SetCovariance(cov[iC], iC, iTr);
    tracks.SetId(trackId[iTr], iTr);
    if(fPID)
      tracks.SetPDG(fPID->GetPID()[iTr], iTr);
    else
      tracks.SetPDG(-1, iTr);
    tracks.SetQ(q, iTr);

    if(fPVFindMode == 0)
    {    
      tracks.SetPVIndex(-1, iTr);
    }
    else
      tracks.SetPVIndex(-1, iTr);
  }

    
  //fill vector with Emc clusters
  KFPEmcCluster emcClusters;
  if(fNeutralTrackArray)
  {
    int nClusters = 0;
    vector<int> clusterId(fNeutralTrackArray->GetEntriesFast());
    for(int iClust=0; iClust<fNeutralTrackArray->GetEntriesFast(); iClust++)
    {
      PndPidCandidate *inTrack = (PndPidCandidate*)fNeutralTrackArray->At(iClust);
      int iBump = inTrack->GetEmcIndex();
      PndEmcBump *inBump = (PndEmcBump*) fEmcBumps->At(iBump);
      
      if(inTrack->GetEmcQuality()<400) continue;
      if(inBump->energy() < 0.02) continue;
      
      clusterId[nClusters] = iClust;
      nClusters++;
    }
    
    emcClusters.Resize(nClusters);
    for(int iClust2=0; iClust2<nClusters; iClust2++)
    {
      int iClust = clusterId[iClust2];
      PndPidCandidate *inTrack = (PndPidCandidate*)fNeutralTrackArray->At(iClust);
      int iBump = inTrack->GetEmcIndex();
      PndEmcBump *inBump = (PndEmcBump*) fEmcBumps->At(iBump);
      
      float par[4] = {0.f};
      par[0] = inTrack->GetLastHit().x();
      par[1] = inTrack->GetLastHit().y();
      par[2] = inTrack->GetLastHit().z();
      par[3] = inBump->energy();
      float cov[10] = {0.f};
      cov[ 0] = inTrack->GetErrorP7()[0];
      cov[ 1] = inTrack->GetErrorP7()[1];
      cov[ 2] = inTrack->GetErrorP7()[2];
      cov[ 3] = inTrack->GetErrorP7()[3];
      cov[ 4] = inTrack->GetErrorP7()[4];
      cov[ 5] = inTrack->GetErrorP7()[5];
      cov[ 6] = inTrack->GetErrorP7()[15];
      cov[ 7] = inTrack->GetErrorP7()[16];
      cov[ 8] = inTrack->GetErrorP7()[17];
      cov[ 9] = inTrack->GetErrorP7()[27];
            
      for(Int_t iP=0; iP<4; iP++)
        emcClusters.SetParameter(par[iP], iP, iClust2);    
      for(Int_t iC=0; iC<10; iC++)
        emcClusters.SetCovariance(cov[iC], iC, iClust2);
      emcClusters.SetId(iClust+fChargedTrackArray->GetEntriesFast(), iClust2);
    }
  }
  
  TStopwatch timer;
  timer.Start();
  
  fTopoReconstructor->Init(tracks);
  if(fNeutralTrackArray)
  {
    fTopoReconstructor->SetEmcClusters(&emcClusters);
  }
  
  if(fPVFindMode == 0)
  {
    KFPVertex primVtx_tmp;
    primVtx_tmp.SetXYZ(0,0,0);
    primVtx_tmp.SetCovarianceMatrix( 0,0,0,0,0,0 );
    primVtx_tmp.SetNContributors( 0 );
    primVtx_tmp.SetChi2( -100 );

    vector<int> pvTrackIds;
    KFVertex pv(primVtx_tmp);
    fTopoReconstructor->AddPV(pv, pvTrackIds);
  }
  else if(fPVFindMode == 1)
    fTopoReconstructor->ReconstructPrimVertex();
  else if(fPVFindMode == 2)
    fTopoReconstructor->ReconstructPrimVertex(0);
  
  fTopoReconstructor->SortTracks();
  fTopoReconstructor->ReconstructParticles();
  
  timer.Stop();
  fTopoReconstructor->SetTime(timer.RealTime());
}

const KFParticleTopoReconstructor* PndKFParticleFinder::GetTopoReconstructor ( ) const

inline

Definition at line 34 of file PndKFParticleFinder.h.

References fTopoReconstructor.

Referenced by kfparticle().

{ return fTopoReconstructor; }

InitStatus PndKFParticleFinder::Init ( )

virtual

Definition at line 42 of file PndKFParticleFinder.cxx.

References Double_t, fChargedTrackArray, fChargedTrackBranchName, fEmcBumps, fNeutralTrackArray, fNeutralTrackBranchName, and fTopoReconstructor.

{
  //Get ROOT Manager
  FairRootManager* ioman= FairRootManager::Instance();
  
  if(ioman==0)
  {
    Error("PndKFParticleFinder::Init","RootManager not instantiated!");
    return kERROR;
  }
  
  // Get input collection
  fChargedTrackArray=(TClonesArray*) ioman->GetObject(fChargedTrackBranchName);
  if(fChargedTrackArray==0)
  {
    Error("PndKFParticleFinder::Init","track-array not found!");
    return kERROR;
  }

  if(!(fNeutralTrackBranchName.IsNull()))
  {
    fNeutralTrackArray = (TClonesArray*) ioman->GetObject(fNeutralTrackBranchName);
  }
  fEmcBumps = (TClonesArray*) ioman->GetObject("EmcBump");
  
  //set magnetic field for KF Particle
  FairField* MF=FairRunAna::Instance()->GetField();
  Double_t xyz[3]={0,0,0};
  Double_t fieldValue[3];
  MF->Field(xyz,fieldValue);
  double Bz = fieldValue[2];
  fTopoReconstructor->SetField(-Bz);
  
  return kSUCCESS;
}

double PndKFParticleFinder::InversedChi2Prob ( double  p, int  ndf  ) const

private

Definition at line 300 of file PndKFParticleFinder.cxx.

References CAMath::Abs(), and f.

Referenced by SetPrimaryProbCut().

{
  double epsilon = 1.e-14;
  double chi2Left = 0.f;
  double chi2Right = 10000.f;
  
  double probLeft = p - TMath::Prob(chi2Left, ndf);
  
  double chi2Centr = (chi2Left+chi2Right)/2.f;
  double probCentr = p - TMath::Prob( chi2Centr, ndf);
  
  while( TMath::Abs(chi2Right-chi2Centr)/chi2Centr > epsilon )
  {
    if(probCentr * probLeft > 0.f)
    {
      chi2Left = chi2Centr;
      probLeft = probCentr;
    }
    else
    {
      chi2Right = chi2Centr;
    }
    
    chi2Centr = (chi2Left+chi2Right)/2.f;
    probCentr = p - TMath::Prob( chi2Centr, ndf);
  }
  
  return chi2Centr;
}

const PndKFParticleFinder& PndKFParticleFinder::operator= ( const PndKFParticleFinder &  )

private

void PndKFParticleFinder::RconstructMultiplePV ( )

inline

Definition at line 26 of file PndKFParticleFinder.h.

References fPVFindMode.

{ fPVFindMode = 2; }

void PndKFParticleFinder::ReconstructSinglePV ( )

inline

Definition at line 25 of file PndKFParticleFinder.h.

References fPVFindMode.

{ fPVFindMode = 1; }

void PndKFParticleFinder::SetChargedTrackBranchName ( const TString &  name )

inline

Definition at line 28 of file PndKFParticleFinder.h.

References fChargedTrackBranchName, and name.

Referenced by kfparticle().

{ fChargedTrackBranchName = name;  }

void PndKFParticleFinder::SetNeutralTrackBranchName ( const TString &  name )

inline

Definition at line 29 of file PndKFParticleFinder.h.

References fNeutralTrackBranchName, and name.

Referenced by kfparticle().

{ fNeutralTrackBranchName = name;  }

void PndKFParticleFinder::SetPIDInformation ( PndKFParticleFinderPID *  pid )

inline

Definition at line 36 of file PndKFParticleFinder.h.

References fPID, and pid().

Referenced by kfparticle().

{ fPID = pid; }

void PndKFParticleFinder::SetPrimaryProbCut

(

float

prob

)

Definition at line 330 of file PndKFParticleFinder.cxx.

References fTopoReconstructor, and InversedChi2Prob().

Referenced by PndKFParticleFinder().

{ 
  fTopoReconstructor->SetChi2PrimaryCut( InversedChi2Prob(prob, 2) );
}

void PndKFParticleFinder::SetPVToZero ( )

inline

Definition at line 24 of file PndKFParticleFinder.h.

References fPVFindMode.

{ fPVFindMode = 0; }

Member Data Documentation

TClonesArray* PndKFParticleFinder::fChargedTrackArray

private

Name of the input TCA.

Definition at line 53 of file PndKFParticleFinder.h.

Referenced by Exec(), and Init().

TString PndKFParticleFinder::fChargedTrackBranchName

private

Definition at line 49 of file PndKFParticleFinder.h.

Referenced by Init(), and SetChargedTrackBranchName().

TClonesArray* PndKFParticleFinder::fEmcBumps

private

Definition at line 55 of file PndKFParticleFinder.h.

Referenced by Exec(), and Init().

TClonesArray* PndKFParticleFinder::fNeutralTrackArray

private

Definition at line 54 of file PndKFParticleFinder.h.

Referenced by Exec(), and Init().

TString PndKFParticleFinder::fNeutralTrackBranchName

private

Name of the input TCA.

Definition at line 50 of file PndKFParticleFinder.h.

Referenced by Init(), and SetNeutralTrackBranchName().

PndKFParticleFinderPID* PndKFParticleFinder::fPID

private

Definition at line 62 of file PndKFParticleFinder.h.

Referenced by Exec(), and SetPIDInformation().

int PndKFParticleFinder::fPVFindMode

private

Definition at line 59 of file PndKFParticleFinder.h.

Referenced by Exec(), RconstructMultiplePV(), ReconstructSinglePV(), and SetPVToZero().

KFParticleTopoReconstructor* PndKFParticleFinder::fTopoReconstructor

private

Definition at line 58 of file PndKFParticleFinder.h.

Referenced by Exec(), GetTopoReconstructor(), Init(), PndKFParticleFinder(), SetPrimaryProbCut(), and ~PndKFParticleFinder().

---

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

• PndKFParticleFinder.h
• PndKFParticleFinder.cxx