1 | |
2 | |
3 | |
4 | |
5 | |
6 | |
7 | |
8 | #include <iostream> |
9 | #include <cmath> |
10 | using namespace std; |
11 | |
12 | #include <TThread.h> |
13 | |
14 | #include <TROOT.h> |
15 | |
16 | #include "DEventProcessor_trackeff_hists.h" |
17 | |
18 | #include <JANA/JApplication.h> |
19 | #include <JANA/JEventLoop.h> |
20 | |
21 | #include <DANA/DApplication.h> |
22 | #include <TRACKING/DMCTrajectoryPoint.h> |
23 | #include <PID/DChargedTrack.h> |
24 | #include <DVector2.h> |
25 | #include <particleType.h> |
26 | |
27 | |
28 | |
29 | extern "C"{ |
30 | void InitPlugin(JApplication *app){ |
31 | InitJANAPlugin(app); |
32 | app->AddProcessor(new DEventProcessor_trackeff_hists()); |
33 | } |
34 | } |
35 | |
36 | |
37 | |
38 | |
39 | |
40 | DEventProcessor_trackeff_hists::DEventProcessor_trackeff_hists() |
41 | { |
42 | trk_ptr = &trk; |
43 | MAX_TRACKS = 10; |
44 | |
45 | trkeff = NULL__null; |
46 | |
47 | pthread_mutex_init(&mutex, NULL__null); |
48 | pthread_mutex_init(&rt_mutex, NULL__null); |
49 | } |
50 | |
51 | |
52 | |
53 | |
54 | DEventProcessor_trackeff_hists::~DEventProcessor_trackeff_hists() |
55 | { |
56 | |
57 | } |
58 | |
59 | |
60 | |
61 | |
62 | jerror_t DEventProcessor_trackeff_hists::init(void) |
63 | { |
64 | pthread_mutex_lock(&mutex); |
65 | |
66 | |
67 | TDirectory *dir = (TDirectory*)gROOT->FindObject("TRACKING"); |
68 | if(!dir)dir = new TDirectoryFile("TRACKING","TRACKING"); |
69 | dir->cd(); |
70 | |
71 | |
72 | if(!trkeff){ |
73 | trkeff = new TTree("trkeff","Tracking Efficiency"); |
74 | trkeff->Branch("F","track",&trk_ptr); |
75 | } |
76 | |
77 | dir->cd("../"); |
78 | |
79 | pthread_mutex_unlock(&mutex); |
80 | |
81 | return NOERROR; |
82 | } |
83 | |
84 | |
85 | |
86 | |
87 | jerror_t DEventProcessor_trackeff_hists::brun(JEventLoop *loop, int runnumber) |
88 | { |
89 | |
90 | return NOERROR; |
91 | } |
92 | |
93 | |
94 | |
95 | |
96 | jerror_t DEventProcessor_trackeff_hists::erun(void) |
97 | { |
98 | |
99 | return NOERROR; |
100 | } |
101 | |
102 | |
103 | |
104 | |
105 | jerror_t DEventProcessor_trackeff_hists::fini(void) |
106 | { |
107 | return NOERROR; |
108 | } |
109 | |
110 | |
111 | |
112 | |
113 | jerror_t DEventProcessor_trackeff_hists::evnt(JEventLoop *loop, int eventnumber) |
114 | { |
115 | vector<const DCDCTrackHit*> cdctrackhits; |
116 | vector<const DFDCPseudo*> fdcpseudos; |
117 | vector<const DTrackCandidate*> trackcandidates; |
118 | vector<const DTrackWireBased*> trackWBs; |
119 | vector<const DChargedTrack*> trackTBs; |
120 | vector<const DTrackTimeBased*> throwns; |
121 | vector<const DTrackTimeBased*> locAssociatedTrackTimeBasedVector; |
122 | vector<const DMCTrajectoryPoint*> mctraj; |
123 | |
124 | loop->Get(cdctrackhits); |
125 | loop->Get(fdcpseudos); |
126 | loop->Get(trackcandidates); |
127 | loop->Get(trackWBs); |
128 | loop->Get(trackTBs); |
129 | loop->Get(throwns, "THROWN"); |
130 | loop->Get(mctraj); |
131 | |
132 | |
133 | |
134 | |
135 | |
136 | |
137 | vector<track_info> ti_can(MAX_TRACKS); |
138 | vector<track_info> ti_trkwb(MAX_TRACKS); |
139 | vector<track_info> ti_trktb(MAX_TRACKS); |
140 | for(unsigned int i=0; i<trackcandidates.size(); i++)FillTrackInfo(trackcandidates[i], ti_can); |
141 | for(unsigned int i=0; i<trackWBs.size(); i++)FillTrackInfo(trackWBs[i], ti_trkwb); |
142 | for(unsigned int i=0; i<trackTBs.size(); i++){ |
143 | trackTBs[i]->dChargedTrackHypotheses[0]->GetT(locAssociatedTrackTimeBasedVector); |
144 | FillTrackInfo(locAssociatedTrackTimeBasedVector[0], ti_trktb); |
145 | } |
146 | |
147 | |
148 | |
149 | vector<double> dtheta_mech(MAX_TRACKS); |
150 | vector<double> dp_mech(MAX_TRACKS); |
151 | vector<TVector3> last_p(MAX_TRACKS, TVector3(0.0, 0.0, 0.0)); |
152 | vector<int> mech_max(MAX_TRACKS,0); |
153 | for(unsigned int i=0; i<mctraj.size(); i++){ |
154 | int track = mctraj[i]->track; |
155 | int mech = mctraj[i]->mech; |
156 | double R = sqrt(pow((double)mctraj[i]->x, 2.0) + pow((double)mctraj[i]->y, 2.0)); |
157 | if(track<0 || track>=MAX_TRACKS)continue; |
158 | if(R>60.0)continue; |
159 | TVector3 p(mctraj[i]->px, mctraj[i]->py, mctraj[i]->pz); |
160 | if(mech>mech_max[track]){ |
161 | mech_max[track] = mech; |
162 | dtheta_mech[track] = p.Angle(last_p[track]); |
163 | dp_mech[track] = (p - last_p[track]).Mag(); |
164 | } |
165 | last_p[track] = p; |
166 | } |
167 | |
168 | |
169 | pthread_mutex_lock(&mutex); |
170 | |
171 | |
172 | for(unsigned int i=0; i<throwns.size(); i++){ |
173 | const DTrackTimeBased *thrown = throwns[i]; |
174 | |
175 | trk.pthrown.SetXYZ(thrown->momentum().x(), |
176 | thrown->momentum().y(), |
177 | thrown->momentum().z()); |
178 | |
179 | |
180 | GetNhits(thrown, trk.Ncdc, trk.Nfdc, trk.track); |
181 | if(trk.track<=0 || trk.track>=MAX_TRACKS)continue; |
182 | |
183 | |
184 | trk.can = ti_can[trk.track]; |
185 | trk.trkwb = ti_trkwb[trk.track]; |
186 | trk.trktb = ti_trktb[trk.track]; |
187 | |
188 | |
189 | trk.event = eventnumber; |
190 | trk.mech = mech_max[trk.track]; |
191 | trk.dtheta_mech = dtheta_mech[trk.track]; |
192 | trk.dp_mech = dp_mech[trk.track]; |
193 | trkeff->Fill(); |
194 | } |
195 | |
196 | |
197 | |
198 | pthread_mutex_unlock(&mutex); |
199 | |
200 | return NOERROR; |
201 | } |
202 | |
203 | |
204 | |
205 | |
206 | void DEventProcessor_trackeff_hists::FillTrackInfo(const DKinematicData *kd, vector<track_info> &vti) |
207 | { |
208 | |
209 | int track_no; |
210 | track_info ti; |
211 | GetTrackInfo(kd, ti, track_no); |
212 | if(track_no<0 || track_no>=MAX_TRACKS)return; |
213 | |
214 | |
215 | |
216 | int Nwires = ti.Ncdc + ti.Nfdc; |
217 | int Nwires_prev = vti[track_no].Ncdc + vti[track_no].Nfdc; |
218 | if(Nwires > Nwires_prev)vti[track_no] = ti; |
219 | } |
220 | |
221 | |
222 | |
223 | |
224 | void DEventProcessor_trackeff_hists::GetTrackInfo(const DKinematicData *kd, track_info &ti, int &track_no) |
225 | { |
226 | ti.p.SetXYZ(kd->momentum().x(),kd->momentum().y(),kd->momentum().z()); |
227 | GetNhits(kd, ti.Ncdc, ti.Nfdc, track_no); |
228 | |
229 | |
230 | |
231 | const DTrackCandidate *can = dynamic_cast<const DTrackCandidate*>(kd); |
232 | const DTrackWireBased *track = dynamic_cast<const DTrackWireBased*>(kd); |
233 | const DTrackTimeBased *part = dynamic_cast<const DTrackTimeBased*>(kd); |
234 | if(can!=NULL__null){ |
235 | ti.trk_chisq = can->chisq; |
236 | ti.trk_Ndof = can->Ndof; |
237 | }else if(track!=NULL__null){ |
238 | ti.trk_chisq = track->chisq; |
239 | ti.trk_Ndof = track->Ndof; |
240 | }else if(part!=NULL__null){ |
241 | ti.trk_chisq = part->chisq; |
242 | ti.trk_Ndof = part->Ndof; |
243 | }else{ |
244 | ti.trk_chisq = 1.0E6; |
245 | ti.trk_Ndof = -1; |
246 | } |
247 | } |
248 | |
249 | |
250 | |
251 | |
252 | void DEventProcessor_trackeff_hists::GetNhits(const DKinematicData *kd, int &Ncdc, int &Nfdc, int &track) |
253 | { |
254 | vector<const DCDCTrackHit*> cdctrackhits; |
255 | vector<const DFDCPseudo*> fdcpseudos; |
256 | |
257 | |
258 | |
259 | kd->Get(cdctrackhits); |
260 | kd->Get(fdcpseudos); |
261 | |
262 | |
263 | |
264 | |
265 | |
266 | |
267 | vector<int> cdc_track_no(MAX_TRACKS, 0); |
268 | for(unsigned int i=0; i<cdctrackhits.size(); i++){ |
| 1 | Loop condition is false. Execution continues on line 277 | |
|
269 | vector<const DMCTrackHit*> mctrackhits; |
270 | cdctrackhits[i]->Get(mctrackhits); |
271 | if(mctrackhits.size()==0)continue; |
272 | if(!mctrackhits[0]->primary)continue; |
273 | int track = mctrackhits[0]->track; |
274 | if(track>=0 && track<MAX_TRACKS)cdc_track_no[track]++; |
275 | } |
276 | |
277 | vector<int> fdc_track_no(MAX_TRACKS, 0); |
278 | for(unsigned int i=0; i<fdcpseudos.size(); i++){ |
| 2 | | Loop condition is true. Entering loop body | |
|
| 5 | | Loop condition is true. Entering loop body | |
|
| 8 | | Loop condition is true. Entering loop body | |
|
279 | vector<const DMCTrackHit*> mctrackhits; |
280 | fdcpseudos[i]->Get(mctrackhits); |
281 | if(mctrackhits.size()==0)continue; |
| |
| 4 | | Execution continues on line 278 | |
|
| |
| 7 | | Execution continues on line 278 | |
|
| |
282 | if(!mctrackhits[0]->primary)continue; |
| 10 | | Dereference of null pointer |
|
283 | int track = mctrackhits[0]->track; |
284 | if(track>=0 && track<MAX_TRACKS)fdc_track_no[track]++; |
285 | } |
286 | |
287 | |
288 | int track_with_max_hits = 0; |
289 | int tot_hits_max = cdc_track_no[0] + fdc_track_no[0]; |
290 | for(int i=1; i<MAX_TRACKS; i++){ |
291 | int tot_hits = cdc_track_no[i] + fdc_track_no[i]; |
292 | if(tot_hits > tot_hits_max){ |
293 | track_with_max_hits=i; |
294 | tot_hits_max = tot_hits; |
295 | } |
296 | } |
297 | |
298 | Ncdc = cdc_track_no[track_with_max_hits]; |
299 | Nfdc = fdc_track_no[track_with_max_hits]; |
300 | |
301 | |
302 | |
303 | track = tot_hits_max>0 ? track_with_max_hits:-1; |
304 | } |
305 | |