Analysis Software
Documentation for sPHENIX simulation 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 class SvtxTrack;
19 class RawCluster;
20 
22 {
23  public:
24 
25  ParticleFlowReco(const std::string &name = "ParticleFlowReco");
26 
27  ~ParticleFlowReco() override;
28 
29  int Init(PHCompositeNode *topNode) override;
30 
31  int InitRun(PHCompositeNode *topNode) override;
32 
33  int process_event(PHCompositeNode *topNode) override;
34 
35  int ResetEvent(PHCompositeNode *topNode) override;
36 
37  int EndRun(const int runnumber) override;
38 
39  int End(PHCompositeNode *topNode) override;
40 
41  int Reset(PHCompositeNode * /*topNode*/) override;
42 
43  void Print(const std::string &what = "ALL") const override;
44 
45  void set_energy_match_Nsigma( float Nsigma ) {
46 
47  _energy_match_Nsigma = Nsigma;
48 
49  }
51 
52  private:
53 
54  int CreateNode(PHCompositeNode *topNode);
55 
56  float calculate_dR( float, float, float, float );
57  std::pair<float, float> get_expected_signature( int );
58 
60 
61  std::vector<float> _pflow_TRK_p;
62  std::vector<float> _pflow_TRK_eta;
63  std::vector<float> _pflow_TRK_phi;
64  std::vector<float> _pflow_TRK_EMproj_phi;
65  std::vector<float> _pflow_TRK_EMproj_eta;
66  std::vector<float> _pflow_TRK_HADproj_phi;
67  std::vector<float> _pflow_TRK_HADproj_eta;
68  std::vector<SvtxTrack*> _pflow_TRK_trk;
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<RawCluster*> _pflow_EM_cluster;
79  std::vector< std::vector<float> > _pflow_EM_tower_eta;
80  std::vector< std::vector<float> > _pflow_EM_tower_phi;
81  std::vector< std::vector<int> > _pflow_EM_match_HAD;
82  std::vector< std::vector<int> > _pflow_EM_match_TRK;
83 
84  std::vector<float> _pflow_HAD_E;
85  std::vector<float> _pflow_HAD_eta;
86  std::vector<float> _pflow_HAD_phi;
87  std::vector<RawCluster*> _pflow_HAD_cluster;
88  std::vector< std::vector<float> > _pflow_HAD_tower_eta;
89  std::vector< std::vector<float> > _pflow_HAD_tower_phi;
90  std::vector< std::vector<int> > _pflow_HAD_match_EM;
91  std::vector< std::vector<int> > _pflow_HAD_match_TRK;
92 
93  std::string _track_map_name = "SvtxTrackMap";
94 
95 };
96 
97 #endif // PARTICLEFLOWRECO_H