BOSS 7.0.4
BESIII Offline Software System
Loading...
Searching...
No Matches
InstallArea/include/MdcHoughFinder/MdcHoughFinder/Hough.h
Go to the documentation of this file.
1#ifndef HOUGHSPACE_H
2#define HOUGHSPACE_H
3
4#include "GaudiKernel/Algorithm.h"
5#include "GaudiKernel/NTuple.h"
6#include "GaudiKernel/INTupleSvc.h"
7#include "MdcGeomSvc/IMdcGeomSvc.h"
8#include "MdcGeomSvc/MdcGeoWire.h"
9#include "MdcGeomSvc/MdcGeoLayer.h"
10#include "RawDataProviderSvc/IRawDataProviderSvc.h"
11#include "RawDataProviderSvc/RawDataProviderSvc.h"
12#include "MagneticField/IMagneticFieldSvc.h"
13#include <string>
14#include <vector>
15#include <map>
16#include "TH2D.h"
17#include "TF1.h"
18#include "TGraph.h"
19#include <fstream>
20
21#include "CLHEP/Alist/AList.h"
22#include "MdcGeom/MdcDetector.h"
23#include "BField/BField.h"
24#include "MdcCalibFunSvc/MdcCalibFunSvc.h"
25#include "MdcUtilitySvc/MdcUtilitySvc.h"
26#include "MdcPrintSvc/MdcPrintSvc.h"
27//#include "CLHEP/Matrix/Vector.h"
28
29#include "MdcHoughFinder/HoughHit.h"
30#include "MdcHoughFinder/HoughHitList.h"
31#include "MdcHoughFinder/HoughGlobal.h"
32#include "MdcHoughFinder/HoughMap.h"
33
34#include "MdcRecEvent/RecMdcTrack.h"
35#include "MdcRecEvent/RecMdcHit.h"
36#include "HepPDT/ParticleDataTable.hh"
37#include "MdcTrkRecon/MdcTrack.h"
38#include "MdcTrkRecon/MdcTrackParams.h"
39#include "MdcTrkRecon/MdcTrackList.h"
40#include "TrkBase/TrkRecoTrk.h"
41#include "TrkFitter/TrkContextEv.h"
42class IBesTimerSvc;
43class BesTimer;
44
45class MdcHoughFinder:public Algorithm {
46 public:
47 MdcHoughFinder(const std::string& name, ISvcLocator* pSvcLocator);
48 StatusCode initialize();
49 StatusCode execute();
50 StatusCode finalize();
51 StatusCode beginRun();
52 StatusCode bookTuple();
53
54 private:
55 int GetMcInfo();
56 MdcDigiVec prepareDigi();
57 void dumpHoughHitList(const HoughHitList&);
58 void dumpHoughMap(const HoughMap& );
59 void diffMap(const HoughMap& ,const HoughMap&);
60 void Leastfit(vector<double>, vector<double>,double&,double&);
61 int storeTracks(RecMdcTrackCol*& a,RecMdcHitCol*& b,vector<RecMdcTrack*>& vec_trackPatTds);
62 void clearMem(MdcTrackList& list1,MdcTrackList& list2);
63 void compareHough(MdcTrackList& mdcTrackList);
64 int comparePatTsf(MdcTrackList& tracklist, RecMdcTrackCol* trackList_tds);
65 int arbitrateHits(vector<MdcTrack*>& , MdcTrackParams );
66 //int judgeHit(MdcTrackList& list ,const vector<HoughTrack>& tracklist);
67 int judgeHit(MdcTrackList& list , vector<MdcTrack*>& tracklist);
68 int judgeChargeTrack(MdcTrackList& list1 , MdcTrackList& list2);
69
70//declare property
71 std::string m_pdtFile;
72 std::string m_sigmaFile;
73 HepPDT::ParticleDataTable* m_particleTable;
74 std::map< int, const HepVector > g_tkParTruth;
75 HepPoint3D g_firstHit;
76 double t_d0;
77 double t_z0;
78 double t_tanl;
79 double t_p;
80 double t_pt;
81 double t_cos;
82
83 int m_filter;
84 std::string m_evtFile;
85 int m_debug;
86 int m_debugMap;
87 int m_debug2D;
88 int m_debug3D;
89 int m_debugTrack;
90 int m_debugStereo;
91 int m_debugZs;
92 int m_debugArbHit;
93
94 bool m_hist;
95 bool m_keepBadTdc;
96 bool m_dropHot;
97 bool m_keepUnmatch;
98 bool m_combineTracking;
99 int m_removeBadTrack;
100 double m_dropTrkDrCut;
101 double m_dropTrkDzCut;
102 double m_dropTrkPtCut;
103 double m_dropTrkChi2Cut;
104
105 int m_inputType;
106 double m_mapCharge;
107 int m_useHalf;
108 int m_mapHit;
109 int m_nBinTheta;
110 int m_nBinRho;
111 double m_rhoRange;
112 int m_npart;
113 int m_n2;
114 double m_dsigma;
115 double m_dsigma2;
116 double m_peakWidth;
117 double m_peakHigh;
118 double m_hitPro;
119
120 int m_recpos;
121 int m_recneg;
122 int m_combine;
123 double m_z0Cut_compareHough;
124
125
126 //data member
127 int t_eventNum;
128 int t_runNum;
129 BField* m_bfield;
130 double m_bunchT0;
131 TrkContextEv* m_context;
132 const MdcCalibFunSvc* m_mdcCalibFunSvc;
133 RawDataProviderSvc* m_rawDataProviderSvc;
134 MdcGeomSvc* m_mdcGeomSvc;
135 IMagneticFieldSvc* m_pIMF;
136 const MdcPrintSvc* m_mdcPrintSvc;
137 IBesTimerSvc* m_timersvc;
138 BesTimer* m_timer_all;
139
140 //Ntuple
141 NTuple::Tuple* ntuple_evt;
142 NTuple::Item<int> m_eventNum;
143 NTuple::Item<int> m_runNum;
144 NTuple::Item<int> m_nHit;
145 NTuple::Item<int> m_nSig_Axial;
146 NTuple::Array<int> m_layer;
147 NTuple::Array<int> m_cell;
148 NTuple::Array<double> m_x;
149 NTuple::Array<double> m_y;
150 NTuple::Array<double> m_z;
151 NTuple::Array<double> m_u;
152 NTuple::Array<double> m_v;
153 NTuple::Array<double> m_r;
154 NTuple::Array<double> m_uTruth;
155 NTuple::Array<double> m_vTruth;
156 NTuple::Array<double> m_rTruth;
157 NTuple::Array<double> m_xTruth;
158 NTuple::Array<double> m_yTruth;
159 NTuple::Array<double> m_type;
160
161 NTuple::Array<double> m_deltaD;
162 NTuple::Array<double> m_flt;
163 NTuple::Array<double> m_drift;
164 NTuple::Array<double> m_driftTruth;
165 NTuple::Array<int> m_slant;
166
167 NTuple::Item<int> m_xybinNum;
168 NTuple::Item<double> m_xybinMax;
169 NTuple::Array<double> m_xybinNL;
170 NTuple::Array<double> m_xybinRL;
171 NTuple::Array<double> m_xybinS;
172
173 NTuple::Item<double> m_theta_left;
174 NTuple::Item<double> m_theta_right;
175 NTuple::Item<double> m_rho_down;
176 NTuple::Item<double> m_rho_up;
177 NTuple::Item<double> m_rho;
178 NTuple::Item<double> m_theta;
179 NTuple::Item<double> m_rho_line;
180 NTuple::Item<double> m_theta_line;
181 NTuple::Item<double> m_height;
182 NTuple::Item<double> m_aver;
183 NTuple::Item<double> m_sigma;
184
185 NTuple::Item<int> m_exit_multiturn;
186
187 NTuple::Item<int> m_nMap1Pk;
188 NTuple::Item<int> m_nMap1Tk;
189 NTuple::Item<int> m_nMap2Pk;
190 NTuple::Item<int> m_nMap2Tk;
191 NTuple::Array<double> m_PkRho1;
192 NTuple::Array<double> m_PkRho2;
193 NTuple::Array<double> m_PkTheta1;
194 NTuple::Array<double> m_PkTheta2;
195 NTuple::Array<double> m_TkRho1;
196 NTuple::Array<double> m_TkRho2;
197 NTuple::Array<double> m_TkTheta1;
198 NTuple::Array<double> m_TkTheta2;
199
200 NTuple::Item<int> m_nMaxLayerSlant;
201 NTuple::Array<double> m_MaxLayerSlant;
202 NTuple::Array<double> m_MaxLayer;
203 NTuple::Item<int> m_nNoMaxLayerSlant;
204 NTuple::Array<double> m_NoMaxLayerSlant;
205 NTuple::Array<int> m_NoMaxLayerid;
206
207 NTuple::Item<int> m_MapMax;
208 NTuple::Item<int> m_nMapPk;
209 NTuple::Array<int> m_PeakOrder;
210 NTuple::Array<double> m_PeakRho;
211 NTuple::Array<double> m_PeakTheta;
212 NTuple::Array<double> m_PeakHeight;
213 NTuple::Array<double> m_PeakHit;
214 NTuple::Array<double> m_PeakHitA;
215 NTuple::Item<int> m_nMapTrk;
216 NTuple::Array<double> m_TrackRho;
217 NTuple::Array<double> m_TrackTheta;
218 NTuple::Array<double> m_TrackHitA;
219 //rec - charge
220 //global 2D
221 NTuple::Item<int> m_nTrk2D_neg;
222 NTuple::Array<double> m_pt2D_neg;
223 NTuple::Array<double> m_nHit2D_neg;
224 NTuple::Array<double> m_chi2_2D_neg;
225 //global 3D
226 NTuple::Item<int> m_nTrk3D_neg;
227 NTuple::Array<double> m_tanl_neg;
228 NTuple::Array<double> m_tanl3D_neg;
229 NTuple::Array<double> m_z0_neg;
230 NTuple::Array<double> m_z0_3D_neg;
231 NTuple::Array<double> m_pro_neg;
232 NTuple::Array<double> m_pt3D_neg;
233 NTuple::Array<double> m_nHit3D_neg;
234 NTuple::Array<double> m_chi2_3D_neg;
235
236
237 //mc
238 //NTuple::Item<int> m_nTrkMC;
239 NTuple::Item<double> m_pidTruth;
240 NTuple::Item<double> m_costaTruth;
241 NTuple::Item<double> m_ptTruth;
242 NTuple::Item<double> m_pzTruth;
243 NTuple::Item<double> m_pTruth;
244 NTuple::Item<double> m_qTruth;
245 NTuple::Item<double> m_drTruth;
246 NTuple::Item<double> m_phi0Truth;
247 NTuple::Item<double> m_omegaTruth;
248 NTuple::Item<double> m_dzTruth;
249 NTuple::Item<double> m_tanl_mc;
250 NTuple::Item<double> m_rho_mc;
251 NTuple::Item<double> m_theta_mc;
252
253 NTuple::Item<double> m_time;
254
255 NTuple::Tuple* ntuplehit;
256 NTuple::Item<int> m_evt_stereo;
257 NTuple::Item<int> m_run_stereo;
258 NTuple::Item<double> m_cos_stereo;
259 NTuple::Item<double> m_tanlTruth_stereo;
260 NTuple::Item<double> m_ncir_stereo;
261 NTuple::Item<double> m_npair_stereo;
262 NTuple::Item<double> m_tanl_stereo;
263 NTuple::Item<double> m_tanl3D_stereo;
264 NTuple::Item<double> m_z0_stereo;
265 NTuple::Item<double> m_z03D_stereo;
266 NTuple::Item<int> m_nHit_axial;
267 NTuple::Array<int> m_axial_layer;
268 NTuple::Array<int> m_axial_wire;
269 NTuple::Array<double> m_axial_deltaD;
270 NTuple::Array<double> m_axial_flt;
271 NTuple::Item<int> m_nHit_stereo;
272 NTuple::Array<int> m_stereo_layer;
273 NTuple::Array<int> m_stereo_wire;
274 NTuple::Array<int> m_stereo_style;
275 NTuple::Array<double> m_stereo_z0;
276 NTuple::Array<double> m_stereo_z1;
277 NTuple::Array<double> m_stereo_s0;
278 NTuple::Array<double> m_stereo_s1;
279 NTuple::Array<double> m_stereo_z;
280 NTuple::Array<double> m_stereo_s;
281 NTuple::Array<double> m_stereo_sTruth;
282 NTuple::Array<double> m_stereo_zTruth;
283 NTuple::Array<double> m_stereo_deltaZ;
284 NTuple::Array<int> m_stereo_nsol;
285 NTuple::Array<double> m_stereo_disToCir;
286 NTuple::Array<int> m_stereo_cirlist;
287 NTuple::Array<int> m_stereo_ambig;
288 NTuple::Array<int> m_stereo_ambig_truth;
289
290};
291bool more_pt(const HoughTrack* tracka,const HoughTrack* trackb);
292#endif
more_pt
bool more_pt(const HoughTrack *tracka, const HoughTrack *trackb)
Definition: Hough.cxx:962
RecMdcHitCol
ObjectVector< RecMdcHit > RecMdcHitCol
Definition: InstallArea/include/MdcRecEvent/MdcRecEvent/RecMdcHit.h:99
RecMdcTrackCol
ObjectVector< RecMdcTrack > RecMdcTrackCol
Definition: InstallArea/include/MdcRecEvent/MdcRecEvent/RecMdcTrack.h:79
MdcHoughFinder::execute
StatusCode execute()
Definition: Hough.cxx:250
MdcHoughFinder::initialize
StatusCode initialize()
Definition: Hough.cxx:141
MdcHoughFinder::beginRun
StatusCode beginRun()
Definition: Hough.cxx:132
MdcHoughFinder::finalize
StatusCode finalize()
Definition: Hough.cxx:623
MdcHoughFinder::bookTuple
StatusCode bookTuple()
Definition: Hough.cxx:1350