CbmRoot
Loading...
Searching...
No Matches
Public Member Functions | Private Member Functions | Private Attributes | List of all members
ATKFParticleFinder Class Reference

#include <ATKFParticleFinder.h>

Collaboration diagram for ATKFParticleFinder:
[legend]

Public Member Functions

void InitInput (const std::string &file_name, const std::string &tree_name)
 
void InitOutput (const std::string &file_name="KFPF_output.root")
 
void Finish ()
 
void SetCuts (CutsContainer cuts)
 
void SetPIDMode (int pid)
 
void WriteCandidates (const KFParticleTopoReconstructor *TR)
 
void Run (int n_events=-1)
 

Private Member Functions

void InitTopoReconstructor ()
 
std::vector< float > GetCovMatrixCbm (const AnalysisTree::Track &track) const
 

Private Attributes

std::string in_file_name_
 
std::string in_tree_name_
 
AnalysisTree::Chain * in_chain_ {nullptr}
 
AnalysisTree::EventHeader * rec_event_header_ {nullptr}
 
AnalysisTree::TrackDetector * kf_tracks_ {nullptr}
 
AnalysisTree::Particles * sim_particles_ {nullptr}
 
AnalysisTree::Matching * kf2sim_tracks_ {nullptr}
 
TFile * out_file_ {nullptr}
 
TTree * out_tree_ {nullptr}
 
AnalysisTree::Configuration out_config_
 
AnalysisTree::Particles * particles_reco_ {nullptr}
 
int q_field_id_ {-999}
 
int par_field_id_ {-999}
 
int mf_field_id_ {-999}
 
int cov_field_id_ {-999}
 
int passcuts_field_id_ {-999}
 
int vtx_chi2_field_id_ {-999}
 
int nhits_field_id_ {-999}
 
int nhits_mvd_field_id_ {-999}
 
int daughter1_id_field_id_ {-999}
 
int daughter2_id_field_id_ {-999}
 
CutsContainer cuts_
 
KFParticleTopoReconstructor * topo_reconstructor_ {nullptr}
 
int pid_mode_ {1}
 

Detailed Description

Definition at line 23 of file ATKFParticleFinder.h.

Member Function Documentation

◆ Finish()

void ATKFParticleFinder::Finish ( )

Definition at line 84 of file ATKFParticleFinder.cxx.

References out_file_, out_tree_, and topo_reconstructor_.

Referenced by Run().

◆ GetCovMatrixCbm()

std::vector< float > ATKFParticleFinder::GetCovMatrixCbm ( const AnalysisTree::Track & track) const
private

Definition at line 205 of file ATKFParticleFinder.cxx.

References cov_field_id_, par_field_id_, q_field_id_, and sqrt().

Referenced by InitTopoReconstructor().

◆ InitInput()

void ATKFParticleFinder::InitInput ( const std::string & file_name,
const std::string & tree_name )

Definition at line 10 of file ATKFParticleFinder.cxx.

References cov_field_id_, cuts_, CutsContainer::GetCutChi2Geo(), CutsContainer::GetCutChi2Prim(), CutsContainer::GetCutDistance(), CutsContainer::GetCutLdL(), CutsContainer::GetCutLDown(), in_chain_, kf2sim_tracks_, kf_tracks_, mf_field_id_, nhits_field_id_, nhits_mvd_field_id_, par_field_id_, passcuts_field_id_, q_field_id_, rec_event_header_, sim_particles_, topo_reconstructor_, and vtx_chi2_field_id_.

◆ InitOutput()

void ATKFParticleFinder::InitOutput ( const std::string & file_name = "KFPF_output.root")

Definition at line 61 of file ATKFParticleFinder.cxx.

References daughter1_id_field_id_, daughter2_id_field_id_, out_config_, out_file_, out_tree_, and particles_reco_.

◆ InitTopoReconstructor()

void ATKFParticleFinder::InitTopoReconstructor ( )
private

Definition at line 116 of file ATKFParticleFinder.cxx.

References GetCovMatrixCbm(), kf2sim_tracks_, kf_tracks_, mf_field_id_, nhits_mvd_field_id_, par_field_id_, passcuts_field_id_, pid_mode_, q_field_id_, rec_event_header_, sim_particles_, topo_reconstructor_, and vtx_chi2_field_id_.

Referenced by Run().

◆ Run()

void ATKFParticleFinder::Run ( int n_events = -1)

Definition at line 93 of file ATKFParticleFinder.cxx.

References Finish(), in_chain_, InitTopoReconstructor(), topo_reconstructor_, and WriteCandidates().

◆ SetCuts()

void ATKFParticleFinder::SetCuts ( CutsContainer cuts)
inline

Definition at line 28 of file ATKFParticleFinder.h.

References cuts_.

◆ SetPIDMode()

void ATKFParticleFinder::SetPIDMode ( int pid)
inline

Definition at line 30 of file ATKFParticleFinder.h.

References pid_mode_.

◆ WriteCandidates()

void ATKFParticleFinder::WriteCandidates ( const KFParticleTopoReconstructor * TR)

Definition at line 186 of file ATKFParticleFinder.cxx.

References daughter1_id_field_id_, daughter2_id_field_id_, out_config_, out_tree_, and particles_reco_.

Referenced by Run().

Member Data Documentation

◆ cov_field_id_

int ATKFParticleFinder::cov_field_id_ {-999}
private

Definition at line 53 of file ATKFParticleFinder.h.

Referenced by GetCovMatrixCbm(), and InitInput().

◆ cuts_

CutsContainer ATKFParticleFinder::cuts_
private

Definition at line 62 of file ATKFParticleFinder.h.

Referenced by InitInput(), and SetCuts().

◆ daughter1_id_field_id_

int ATKFParticleFinder::daughter1_id_field_id_ {-999}
private

Definition at line 59 of file ATKFParticleFinder.h.

Referenced by InitOutput(), and WriteCandidates().

◆ daughter2_id_field_id_

int ATKFParticleFinder::daughter2_id_field_id_ {-999}
private

Definition at line 60 of file ATKFParticleFinder.h.

Referenced by InitOutput(), and WriteCandidates().

◆ in_chain_

AnalysisTree::Chain* ATKFParticleFinder::in_chain_ {nullptr}
private

Definition at line 39 of file ATKFParticleFinder.h.

Referenced by InitInput(), and Run().

◆ in_file_name_

std::string ATKFParticleFinder::in_file_name_
private

Definition at line 37 of file ATKFParticleFinder.h.

◆ in_tree_name_

std::string ATKFParticleFinder::in_tree_name_
private

Definition at line 38 of file ATKFParticleFinder.h.

◆ kf2sim_tracks_

AnalysisTree::Matching* ATKFParticleFinder::kf2sim_tracks_ {nullptr}
private

Definition at line 43 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ kf_tracks_

AnalysisTree::TrackDetector* ATKFParticleFinder::kf_tracks_ {nullptr}
private

Definition at line 41 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ mf_field_id_

int ATKFParticleFinder::mf_field_id_ {-999}
private

Definition at line 52 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ nhits_field_id_

int ATKFParticleFinder::nhits_field_id_ {-999}
private

Definition at line 56 of file ATKFParticleFinder.h.

Referenced by InitInput().

◆ nhits_mvd_field_id_

int ATKFParticleFinder::nhits_mvd_field_id_ {-999}
private

Definition at line 57 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ out_config_

AnalysisTree::Configuration ATKFParticleFinder::out_config_
private

Definition at line 47 of file ATKFParticleFinder.h.

Referenced by InitOutput(), and WriteCandidates().

◆ out_file_

TFile* ATKFParticleFinder::out_file_ {nullptr}
private

Definition at line 45 of file ATKFParticleFinder.h.

Referenced by Finish(), and InitOutput().

◆ out_tree_

TTree* ATKFParticleFinder::out_tree_ {nullptr}
private

Definition at line 46 of file ATKFParticleFinder.h.

Referenced by Finish(), InitOutput(), and WriteCandidates().

◆ par_field_id_

int ATKFParticleFinder::par_field_id_ {-999}
private

Definition at line 51 of file ATKFParticleFinder.h.

Referenced by GetCovMatrixCbm(), InitInput(), and InitTopoReconstructor().

◆ particles_reco_

AnalysisTree::Particles* ATKFParticleFinder::particles_reco_ {nullptr}
private

Definition at line 48 of file ATKFParticleFinder.h.

Referenced by InitOutput(), and WriteCandidates().

◆ passcuts_field_id_

int ATKFParticleFinder::passcuts_field_id_ {-999}
private

Definition at line 54 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ pid_mode_

int ATKFParticleFinder::pid_mode_ {1}
private

Definition at line 67 of file ATKFParticleFinder.h.

Referenced by InitTopoReconstructor(), and SetPIDMode().

◆ q_field_id_

int ATKFParticleFinder::q_field_id_ {-999}
private

Definition at line 50 of file ATKFParticleFinder.h.

Referenced by GetCovMatrixCbm(), InitInput(), and InitTopoReconstructor().

◆ rec_event_header_

AnalysisTree::EventHeader* ATKFParticleFinder::rec_event_header_ {nullptr}
private

Definition at line 40 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ sim_particles_

AnalysisTree::Particles* ATKFParticleFinder::sim_particles_ {nullptr}
private

Definition at line 42 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

◆ topo_reconstructor_

KFParticleTopoReconstructor* ATKFParticleFinder::topo_reconstructor_ {nullptr}
private

Definition at line 63 of file ATKFParticleFinder.h.

Referenced by Finish(), InitInput(), InitTopoReconstructor(), and Run().

◆ vtx_chi2_field_id_

int ATKFParticleFinder::vtx_chi2_field_id_ {-999}
private

Definition at line 55 of file ATKFParticleFinder.h.

Referenced by InitInput(), and InitTopoReconstructor().

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