EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ParticleFlowReco.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ParticleFlowReco.h
1 #ifndef PARTICLEFLOWRECO_H
2 #define PARTICLEFLOWRECO_H
3 
4 //===========================================================
8 //===========================================================
9 
10 #include <fun4all/SubsysReco.h>
11 
12 #include <gsl/gsl_rng.h>
13 
14 #include <string>
15 #include <vector>
16 
17 class PHCompositeNode;
18 
20 {
21  public:
22 
23  ParticleFlowReco(const std::string &name = "ParticleFlowReco");
24 
25  ~ParticleFlowReco() override;
26 
27  int Init(PHCompositeNode *topNode) override;
28 
29  int InitRun(PHCompositeNode *topNode) override;
30 
31  int process_event(PHCompositeNode *topNode) override;
32 
33  int ResetEvent(PHCompositeNode *topNode) override;
34 
35  int EndRun(const int runnumber) override;
36 
37  int End(PHCompositeNode *topNode) override;
38 
39  int Reset(PHCompositeNode * /*topNode*/) override;
40 
41  void Print(const std::string &what = "ALL") const override;
42 
43  void set_energy_match_Nsigma( float Nsigma ) {
44 
45  _energy_match_Nsigma = Nsigma;
46 
47  }
48 
49  void set_emulated_efficiency( float eff ) {
50 
51  _emulate_efficiency = eff;
52 
53  }
54 
55  private:
56 
57  int CreateNode(PHCompositeNode *topNode);
58 
59  float calculate_dR( float, float, float, float );
60  std::pair<float, float> get_expected_signature( int );
61 
64  gsl_rng *_tr_eff;
65 
66  std::vector<float> _pflow_TRK_p;
67  std::vector<float> _pflow_TRK_eta;
68  std::vector<float> _pflow_TRK_phi;
69  std::vector< std::vector<int> > _pflow_TRK_match_EM;
70  std::vector< std::vector<int> > _pflow_TRK_match_HAD;
71 
72  // convention is ( EM index, dR value )
73  std::vector< std::vector< std::pair<int,float> > > _pflow_TRK_addtl_match_EM;
74 
75  std::vector<float> _pflow_EM_E;
76  std::vector<float> _pflow_EM_eta;
77  std::vector<float> _pflow_EM_phi;
78  std::vector< std::vector<float> > _pflow_EM_tower_eta;
79  std::vector< std::vector<float> > _pflow_EM_tower_phi;
80  std::vector< std::vector<int> > _pflow_EM_match_HAD;
81  std::vector< std::vector<int> > _pflow_EM_match_TRK;
82 
83  std::vector<float> _pflow_HAD_E;
84  std::vector<float> _pflow_HAD_eta;
85  std::vector<float> _pflow_HAD_phi;
86  std::vector< std::vector<float> > _pflow_HAD_tower_eta;
87  std::vector< std::vector<float> > _pflow_HAD_tower_phi;
88  std::vector< std::vector<int> > _pflow_HAD_match_EM;
89  std::vector< std::vector<int> > _pflow_HAD_match_TRK;
90 
91 
92 };
93 
94 #endif // PARTICLEFLOWRECO_H