Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ResonanceJetTagging.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ResonanceJetTagging.h
1 #ifndef RESONANCEJETTAGGING_H__
2 #define RESONANCEJETTAGGING_H__
3 
4 #include <jetbase/Jetv2.h>
5 
6 #include <fun4all/SubsysReco.h>
7 
8 #include <fastjet/JetDefinition.hh>
9 
10 #include <map> // for map
11 #include <string> // for string
12 #include <utility> // for pair
13 #include <vector>
14 
16 class PHCompositeNode;
17 class SvtxTrack;
18 class PHG4Particlev2;
19 class PHG4Particle;
20 class ParticleFlowElement;
21 class JetContainerv1;
22 
23 namespace CLHEP
24 {
25  class Hep3Vector;
26 }
27 namespace HepMC
28 {
29  class GenParticle;
30 }
31 namespace fastjet
32 {
33  class PseudoJet;
34 }
35 
37 class ResonanceJetTagging : public SubsysReco
38 {
39  public:
40  enum ALGO
41  {
42  ANTIKT = 0,
43  KT = 1,
44  CAMBRIDGE = 2
45  };
46 
47  enum RECOMB
48  {
49  E_SCHEME = 0,
50  PT_SCHEME = 1,
51  PT2_SCHEME = 2,
52  ET_SCHEME = 3,
53  ET2_SCHEME = 4
54  };
55 
56  enum TAG
57  {
58  D0 = 0,
59  D0TOK3PI = 1,
60  DPLUS = 2,
61  DSTAR = 3,
62  JPSY = 4,
63  K0 = 5,
64  GAMMA = 6,
65  ELECTRON = 7,
66  LAMBDAC = 8
67  };
68 
70  ResonanceJetTagging(const std::string &name = "ResonanceJetTagging", const TAG tag = TAG::D0, const std::string &KFparticle_Container_name = "");
71 
72  // Destructor
73  virtual ~ResonanceJetTagging();
74 
76  int Init(PHCompositeNode *);
77 
79  int process_event(PHCompositeNode *);
80 
82  int End(PHCompositeNode *);
83 
84  // Particle flow
85  void setParticleFlowEtaAcc(double etamin, double etamax)
86  {
87  m_particleflow_mineta = etamin;
88  m_particleflow_maxeta = etamax;
89  }
90  void setParticleFlowMinEta(double etamin) { m_particleflow_mineta = etamin; }
91  void setParticleFlowMaxEta(double etamax) { m_particleflow_maxeta = etamax; }
92  double getParticleFlowMinEta() { return m_particleflow_mineta; }
93  double getParticleFlowMaxEta() { return m_particleflow_maxeta; }
94 
95  // tracks
96  void setTrackPtAcc(double ptmin, double ptmax)
97  {
98  m_track_minpt = ptmin;
99  m_track_maxpt = ptmax;
100  }
101  void setTrackMinPt(double ptmin) { m_track_minpt = ptmin; }
102  void setTrackMaxPt(double ptmax) { m_track_maxpt = ptmax; }
103  double getTrackMinPt() { return m_track_minpt; }
104  double getTrackMaxPt() { return m_track_maxpt; }
105 
106  void setTrackEtaAcc(double etamin, double etamax)
107  {
108  m_track_mineta = etamin;
109  m_track_maxeta = etamax;
110  }
111  void setTrackMinEta(double etamin) { m_track_mineta = etamin; }
112  void setTrackMaxEta(double etamax) { m_track_maxeta = etamax; }
113  double getTrackMinEta() { return m_track_mineta; }
114  double getTrackMaxEta() { return m_track_maxeta; }
115 
116  // EMCal clusters
117  void setEMCalClusterPtAcc(double ptmin, double ptmax)
118  {
119  m_EMCal_cluster_minpt = ptmin;
120  m_EMCal_cluster_maxpt = ptmax;
121  }
122  void setEMCalClusterMinPt(double ptmin) { m_EMCal_cluster_minpt = ptmin; }
123  void setEMCalClusterMaxPt(double ptmax) { m_EMCal_cluster_maxpt = ptmax; }
124  double getEMCalClusterMinPt() { return m_EMCal_cluster_minpt; }
125  double getEMCalClusterMaxPt() { return m_EMCal_cluster_maxpt; }
126 
127  void setEMCalClusterEtaAcc(double etamin, double etamax)
128  {
129  m_EMCal_cluster_mineta = etamin;
130  m_EMCal_cluster_maxeta = etamax;
131  }
132  void setEMCalClusterMinEta(double etamin) { m_EMCal_cluster_mineta = etamin; }
133  void setEMCalClusterMaxEta(double etamax) { m_EMCal_cluster_maxeta = etamax; }
134  double getEMCalClusterMinEta() { return m_EMCal_cluster_mineta; }
135  double getEMCalClusterMaxEta() { return m_EMCal_cluster_maxeta; }
136 
137  // HCal clusters
138  void setHCalClusterPtAcc(double ptmin, double ptmax)
139  {
140  m_HCal_cluster_minpt = ptmin;
141  m_HCal_cluster_maxpt = ptmax;
142  }
143  void setHCalClusterMinPt(double ptmin) { m_HCal_cluster_minpt = ptmin; }
144  void setHCalClusterMaxPt(double ptmax) { m_HCal_cluster_maxpt = ptmax; }
145  double getHCalClusterMinPt() { return m_HCal_cluster_minpt; }
146  double getHCalClusterMaxPt() { return m_HCal_cluster_maxpt; }
147 
148  void setHCalClusterEtaAcc(double etamin, double etamax)
149  {
150  m_HCal_cluster_mineta = etamin;
151  m_HCal_cluster_maxeta = etamax;
152  }
153  void setHCalClusterMinEta(double etamin) { m_HCal_cluster_mineta = etamin; }
154  void setHCalClusterMaxEta(double etamax) { m_HCal_cluster_maxeta = etamax; }
155  double getHCalClusterMinEta() { return m_HCal_cluster_mineta; }
156  double getHCalClusterMaxEta() { return m_HCal_cluster_maxeta; }
157 
158  // Set/Get add Tracks and Clusters
159  void setAddParticleFlow(bool b) { m_add_particleflow = b; }
160  bool getAddParticleFlow() { return m_add_particleflow; }
161  void setAddTracks(bool b) { m_add_tracks = b; }
162  bool getAddTracks() { return m_add_tracks; }
163  void setAddEMCalClusters(bool b) { m_add_EMCal_clusters = b; }
164  bool getAddEMCalClusters() { return m_add_EMCal_clusters; }
165  void setAddHCalClusters(bool b) { m_add_HCal_clusters = b; }
166  bool getAddHCalClusters() { return m_add_HCal_clusters; }
167 
168  // Jet settings
169  void setR(double r) { m_jetr = r; }
170  double getR(double /*r*/) { return m_jetr; }
171  void setJetAlgo(ALGO jetalgo)
172  {
173  switch (jetalgo)
174  {
175  case ALGO::ANTIKT:
176  m_jetalgo = fastjet::antikt_algorithm;
177  break;
178  case ALGO::KT:
179  m_jetalgo = fastjet::kt_algorithm;
180  break;
181  case ALGO::CAMBRIDGE:
182  m_jetalgo = fastjet::cambridge_algorithm;
183  break;
184  }
185  }
186  fastjet::JetAlgorithm getJetAlgo() { return m_jetalgo; }
187  void setRecombScheme(RECOMB recomb_scheme)
188  {
189  switch (recomb_scheme)
190  {
191  case RECOMB::E_SCHEME:
192  m_recomb_scheme = fastjet::E_scheme;
193  break;
194  case RECOMB::PT_SCHEME:
195  m_recomb_scheme = fastjet::pt_scheme;
196  break;
197  case RECOMB::PT2_SCHEME:
198  m_recomb_scheme = fastjet::pt2_scheme;
199  break;
200  case RECOMB::ET_SCHEME:
201  m_recomb_scheme = fastjet::Et_scheme;
202  break;
203  case RECOMB::ET2_SCHEME:
204  m_recomb_scheme = fastjet::Et2_scheme;
205  break;
206  }
207  }
208  fastjet::RecombinationScheme getRecombScheme() { return m_recomb_scheme; }
209  void setJetParameters(double r, ALGO jetalgo, RECOMB recomb_scheme)
210  {
211  setR(r);
212  setJetAlgo(jetalgo);
213  setRecombScheme(recomb_scheme);
214  }
215  void setJetContainerName(const std::string &n) { m_jetcontainer_name = n; }
216  std::string getJetContainerName() { return m_jetcontainer_name; }
217  void setDoRecunstructed(bool b) { m_dorec = b; }
218  bool getDoRecunstructed() { return m_dorec; }
219  void setDoTruth(bool b) { m_dotruth = b; }
220  bool getDoTruth() { return m_dotruth; }
221 
222  private:
224  std::string m_jetcontainer_name;
225 
227  double m_particleflow_mineta;
228  double m_particleflow_maxeta;
229 
231  double m_track_minpt;
232  double m_track_maxpt;
233  double m_track_mineta;
234  double m_track_maxeta;
235 
237  double m_EMCal_cluster_minpt;
238  double m_EMCal_cluster_maxpt;
239  double m_EMCal_cluster_mineta;
240  double m_EMCal_cluster_maxeta;
241 
243  double m_HCal_cluster_minpt;
244  double m_HCal_cluster_maxpt;
245  double m_HCal_cluster_mineta;
246  double m_HCal_cluster_maxeta;
247 
249  bool m_add_particleflow;
250  bool m_add_tracks;
251  bool m_add_EMCal_clusters;
252  bool m_add_HCal_clusters;
253 
254  // jet settings variables
255  double m_jetr;
256  fastjet::JetAlgorithm m_jetalgo;
257  fastjet::RecombinationScheme m_recomb_scheme;
258 
259  JetContainerv1 *m_taggedJetContainer = nullptr;
260  JetContainerv1 *m_truth_taggedJetContainer = nullptr;
261 
262  int m_tag_pdg;
263  unsigned int m_jet_id = 0;
264  unsigned int m_truth_jet_id = 0;
265  bool m_dorec;
266  bool m_dotruth;
267  int m_nDaughters;
268  TAG m_tag_particle;
269  std::string m_KFparticle_name;
270 
272  int tagHFHadronic(PHCompositeNode *topNode);
273  void findTaggedJets(PHCompositeNode *topNode, PHG4Particlev2 *Tag, const std::vector<PHG4Particlev2*> &TagDecays);
274  void addParticleFlow(PHCompositeNode *topNode, std::vector<fastjet::PseudoJet> &particles, const std::vector<PHG4Particlev2*> &TagDecays, std::map<int, std::pair<Jet::SRC, int>> &fjMap);
275  void addTracks(PHCompositeNode *topNode, std::vector<fastjet::PseudoJet> &particles, const std::vector<PHG4Particlev2*> &TagDecays, std::map<int, std::pair<Jet::SRC, int>> &fjMap);
276  void addClusters(PHCompositeNode *topNode, std::vector<fastjet::PseudoJet> &particles, std::map<int, std::pair<Jet::SRC, int>> &fjMap);
277  void findMCTaggedJets(PHCompositeNode *topNode);
278 
279  bool isAcceptableParticleFlow(ParticleFlowElement *pfPart);
280  bool isAcceptableTrack(SvtxTrack *track);
281  bool isAcceptableEMCalCluster(CLHEP::Hep3Vector &E_vec_cluster);
282  bool isAcceptableHCalCluster(CLHEP::Hep3Vector &E_vec_cluster);
283  bool isDecay(HepMC::GenParticle *particle, const std::vector<PHG4Particlev2*> &decays);
284  bool isDecay(SvtxTrack *track, const std::vector<PHG4Particlev2*> &decays);
285  int createJetNode(PHCompositeNode *topNode);
286 };
287 
288 #endif