]>
Commit | Line | Data |
---|---|---|
e1e03704 | 1 | // $Id$ |
2 | ||
3 | //************************************************************************** | |
4 | //* This file is property of and copyright by the ALICE HLT Project * | |
5 | //* ALICE Experiment at CERN, All rights reserved. * | |
6 | //* * | |
7 | //* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> * | |
8 | //* for The ALICE HLT Project. * | |
9 | //* * | |
10 | //* Permission to use, copy, modify and distribute this software and its * | |
11 | //* documentation strictly for non-commercial purposes is hereby granted * | |
12 | //* without fee, provided that the above copyright notice appears in all * | |
13 | //* copies and that both the copyright notice and this permission notice * | |
14 | //* appear in the supporting documentation. The authors make no claims * | |
15 | //* about the suitability of this software for any purpose. It is * | |
16 | //* provided "as is" without express or implied warranty. * | |
17 | //************************************************************************** | |
18 | ||
19 | /// @file AliHLTTrackGeometry.cxx | |
20 | /// @author Matthias Richter | |
21 | /// @date 2011-05-20 | |
22 | /// @brief Desciption of a track by a sequence of track points | |
23 | /// | |
24 | ||
25 | #include "AliHLTTrackGeometry.h" | |
26 | #include "AliHLTSpacePointContainer.h" | |
27 | #include "TObjArray.h" | |
28 | #include "TMarker.h" | |
29 | #include "TMath.h" | |
54ff4c01 | 30 | #include "TH2.h" |
e1e03704 | 31 | #include <memory> |
32 | #include <iostream> | |
33 | #include <algorithm> | |
34 | ||
35 | /** ROOT macro for the implementation of ROOT specific class methods */ | |
36 | ClassImp(AliHLTTrackGeometry) | |
37 | ||
38 | AliHLTTrackGeometry::AliHLTTrackGeometry() | |
39 | : TObject(), AliHLTLogging() | |
40 | , fTrackPoints() | |
54ff4c01 | 41 | , fSelectionMasks() |
e1e03704 | 42 | , fTrackId(-1) |
54ff4c01 | 43 | , fVerbosity(0) |
e1e03704 | 44 | { |
45 | /// standard constructor | |
46 | } | |
47 | ||
48 | AliHLTTrackGeometry::AliHLTTrackGeometry(const AliHLTTrackGeometry& src) | |
49 | : TObject(src), AliHLTLogging() | |
50 | , fTrackPoints(src.fTrackPoints) | |
54ff4c01 | 51 | , fSelectionMasks(src.fSelectionMasks) |
e1e03704 | 52 | , fTrackId(src.fTrackId) |
54ff4c01 | 53 | , fVerbosity(src.fVerbosity) |
e1e03704 | 54 | { |
55 | /// copy constructor | |
56 | } | |
57 | ||
58 | AliHLTTrackGeometry& AliHLTTrackGeometry::operator=(const AliHLTTrackGeometry& src) | |
59 | { | |
60 | /// assignment operator | |
61 | if (this!=&src) { | |
62 | fTrackPoints.assign(src.fTrackPoints.begin(), src.fTrackPoints.end()); | |
54ff4c01 | 63 | fSelectionMasks.assign(src.fSelectionMasks.begin(), src.fSelectionMasks.end()); |
e1e03704 | 64 | fTrackId=src.fTrackId; |
54ff4c01 | 65 | fVerbosity=src.fVerbosity; |
e1e03704 | 66 | } |
67 | return *this; | |
68 | } | |
69 | ||
70 | AliHLTTrackGeometry::~AliHLTTrackGeometry() | |
71 | { | |
72 | /// destructor | |
73 | } | |
74 | ||
54ff4c01 | 75 | int AliHLTTrackGeometry::AddTrackPoint(const AliHLTTrackPoint& point, AliHLTUInt32_t selectionMask) |
e1e03704 | 76 | { |
77 | /// add a track point to the list | |
78 | vector<AliHLTTrackPoint>::const_iterator element = find(fTrackPoints.begin(), fTrackPoints.end(), point); | |
79 | if (element==fTrackPoints.end()) { | |
80 | fTrackPoints.push_back(point); | |
54ff4c01 | 81 | if (std::find(fSelectionMasks.begin(), fSelectionMasks.end(), selectionMask)==fSelectionMasks.end()) { |
82 | fSelectionMasks.push_back(selectionMask); | |
83 | } | |
e1e03704 | 84 | } else { |
85 | HLTError("track point of id %08x already existing", point.GetId()); | |
86 | return -EEXIST; | |
87 | } | |
88 | return 0; | |
89 | } | |
90 | ||
91 | void AliHLTTrackGeometry::Clear(Option_t * /*option*/) | |
92 | { | |
93 | // internal cleanup | |
94 | } | |
95 | ||
96 | void AliHLTTrackGeometry::Print(Option_t *option) const | |
97 | { | |
98 | // print info | |
99 | Print(cout, option); | |
100 | } | |
101 | ||
102 | void AliHLTTrackGeometry::Print(ostream& out, Option_t */*option*/) const | |
103 | { | |
104 | // print to stream | |
105 | out << "AliHLTTrackGeometry::Print" << endl; | |
106 | } | |
107 | ||
108 | void AliHLTTrackGeometry::Draw(Option_t *option) | |
109 | { | |
110 | /// Inherited from TObject, draw the track | |
111 | float scale=250; | |
112 | float center[2]={0.5,0.5}; | |
113 | int markerColor=1; | |
114 | int markerSize=1; | |
3488f70b | 115 | int verbosity=0; |
e1e03704 | 116 | |
117 | TString strOption(option); | |
118 | std::auto_ptr<TObjArray> tokens(strOption.Tokenize(" ")); | |
119 | if (!tokens.get()) return; | |
120 | for (int i=0; i<tokens->GetEntriesFast(); i++) { | |
121 | if (!tokens->At(i)) continue; | |
122 | const char* key=""; | |
123 | TString arg=tokens->At(i)->GetName(); | |
124 | ||
125 | key="scale="; | |
126 | if (arg.BeginsWith(key)) { | |
127 | arg.ReplaceAll(key, ""); | |
128 | scale=arg.Atof(); | |
129 | continue; | |
130 | } | |
131 | key="centerx="; | |
132 | if (arg.BeginsWith(key)) { | |
133 | arg.ReplaceAll(key, ""); | |
134 | center[0]=arg.Atof(); | |
135 | continue; | |
136 | } | |
137 | key="centery="; | |
138 | if (arg.BeginsWith(key)) { | |
139 | arg.ReplaceAll(key, ""); | |
140 | center[1]=arg.Atof(); | |
141 | continue; | |
142 | } | |
143 | ||
144 | key="markercolor="; | |
145 | if (arg.BeginsWith(key)) { | |
146 | arg.ReplaceAll(key, ""); | |
147 | markerColor=arg.Atoi(); | |
148 | continue; | |
149 | } | |
150 | ||
151 | key="markersize="; | |
152 | if (arg.BeginsWith(key)) { | |
153 | arg.ReplaceAll(key, ""); | |
154 | markerSize=arg.Atoi(); | |
155 | continue; | |
156 | } | |
3488f70b | 157 | |
158 | key="verbosity="; | |
159 | if (arg.BeginsWith(key)) { | |
160 | arg.ReplaceAll(key, ""); | |
161 | verbosity=arg.Atoi(); | |
162 | continue; | |
163 | } | |
e1e03704 | 164 | } |
165 | ||
166 | bool bFirstPoint=true; | |
167 | float firstalpha=0.0; | |
168 | for (vector<AliHLTTrackPoint>::const_iterator point=fTrackPoints.begin(); | |
169 | point!=fTrackPoints.end(); | |
170 | point++) { | |
171 | float alpha=GetPlaneAlpha(point->GetId()); | |
172 | float r=GetPlaneR(point->GetId()); | |
173 | float cosa=TMath::Cos(alpha); | |
174 | float sina=TMath::Sin(alpha); | |
175 | float x = r*sina + point->GetU()*cosa; | |
176 | float y =-r*cosa + point->GetU()*sina; | |
3488f70b | 177 | if (verbosity>0) { |
178 | HLTInfo("ID 0x%08x: x=% .4f y=% .4f alpha=% .4f", point->GetId(), r, point->GetU(), alpha); | |
179 | } | |
e1e03704 | 180 | int color=markerColor; |
181 | if (bFirstPoint) { | |
182 | bFirstPoint=false; | |
183 | TMarker* m=new TMarker(x/(2*scale)+center[0], y/(2*scale)+center[1], 29); | |
184 | m->SetMarkerSize(2); | |
185 | m->SetMarkerColor(2); | |
186 | m->Draw("same"); | |
187 | firstalpha=alpha; | |
188 | } else { | |
409faf4b | 189 | color+=int(9*TMath::Abs(alpha-firstalpha)/TMath::Pi()); |
e1e03704 | 190 | } |
191 | TMarker* m=new TMarker(x/(2*scale)+center[0], y/(2*scale)+center[1], point->GetV()>0?2:5); | |
192 | m->SetMarkerColor(color); | |
193 | m->SetMarkerSize(markerSize); | |
194 | m->Draw("same"); | |
195 | } | |
196 | } | |
197 | ||
c5efb946 | 198 | int AliHLTTrackGeometry::SetAssociatedSpacePoint(UInt_t planeId, UInt_t spacepointId, int /*status*/, float dU, float dV) |
e1e03704 | 199 | { |
200 | /// set the spacepoint associated with a track point | |
201 | vector<AliHLTTrackPoint>::iterator element = find(fTrackPoints.begin(), fTrackPoints.end(), planeId); | |
202 | if (element==fTrackPoints.end()) return -ENOENT; | |
c5efb946 | 203 | element->AddAssociatedSpacePoint(spacepointId, dU, dV); |
e1e03704 | 204 | return 0; |
205 | } | |
206 | ||
207 | int AliHLTTrackGeometry::GetAssociatedSpacePoint(UInt_t planeId, UInt_t& spacepointId) const | |
208 | { | |
209 | /// get the spacepoint associated with a track point | |
210 | /// return status flag if found, -ENOENT if no associated spacepoint found | |
211 | vector<AliHLTTrackPoint>::const_iterator element = find(fTrackPoints.begin(), fTrackPoints.end(), planeId); | |
212 | if (element==fTrackPoints.end()) return -ENOENT; | |
213 | if (!element->HaveAssociatedSpacePoint()) return -ENODATA; | |
c5efb946 | 214 | spacepointId=(element->GetSpacepoints())[0].fId; |
215 | return 0; | |
216 | } | |
217 | ||
218 | int AliHLTTrackGeometry::RegisterTrackPoints(AliHLTTrackGrid* /*pGrid*/) const | |
219 | { | |
220 | /// default implementation, nothing to do | |
221 | return -ENOSYS; | |
222 | } | |
223 | ||
224 | int AliHLTTrackGeometry::FillTrackPoints(AliHLTTrackGrid* /*pGrid*/) const | |
225 | { | |
226 | /// default implementation, nothing to do | |
227 | return -ENOSYS; | |
e1e03704 | 228 | } |
229 | ||
230 | const AliHLTTrackGeometry::AliHLTTrackPoint* AliHLTTrackGeometry::GetTrackPoint(AliHLTUInt32_t id) const | |
231 | { | |
232 | /// get const pointer to track point | |
233 | vector<AliHLTTrackPoint>::const_iterator element = find(fTrackPoints.begin(), fTrackPoints.end(), id); | |
234 | if (element==fTrackPoints.end()) return NULL; | |
235 | return &(*element); | |
236 | } | |
237 | ||
238 | AliHLTTrackGeometry::AliHLTTrackPoint* AliHLTTrackGeometry::GetTrackPoint(AliHLTUInt32_t id) | |
239 | { | |
240 | /// get const pointer to track point | |
241 | vector<AliHLTTrackPoint>::iterator element = find(fTrackPoints.begin(), fTrackPoints.end(), id); | |
242 | if (element==fTrackPoints.end()) return NULL; | |
243 | return &(*element); | |
244 | } | |
245 | ||
54ff4c01 | 246 | AliHLTSpacePointContainer* AliHLTTrackGeometry::ConvertToSpacePoints(bool /*bAssociated*/) const |
3488f70b | 247 | { |
248 | /// create a collection of all points | |
249 | HLTError("implementation of child method missing"); | |
250 | return NULL; | |
251 | } | |
252 | ||
e1e03704 | 253 | int AliHLTTrackGeometry::AssociateSpacePoints(AliHLTSpacePointContainer& points) |
254 | { | |
255 | /// associate the track space points to the calculated track points | |
256 | vector<AliHLTUInt32_t> ids; | |
257 | points.GetClusterIDs(ids); | |
258 | if (ids.size()>0) return 0; | |
259 | int result=AssociateSpacePoints(&ids[0], ids.size(), points); | |
260 | if (result>0) { | |
54ff4c01 | 261 | HLTInfo("associated %d of %d space point(s) to track points", result, ids.size()); |
e1e03704 | 262 | } |
263 | return result; | |
264 | } | |
265 | ||
266 | int AliHLTTrackGeometry::AssociateSpacePoints(const AliHLTUInt32_t* trackpoints, AliHLTUInt32_t nofPoints, AliHLTSpacePointContainer& points) | |
267 | { | |
268 | /// associate the track space points to the calculated track points | |
269 | if (nofPoints==0) return 0; | |
270 | if (trackpoints==NULL) return -EINVAL; | |
271 | int count=0; | |
272 | for (int i=nofPoints-1; i>=0; i--) { | |
273 | if (!points.Check(trackpoints[i])) { | |
274 | HLTWarning("can not find point id %08x", trackpoints[i]); | |
275 | continue; | |
276 | } | |
277 | float xyz[3]={points.GetX(trackpoints[i]), points.GetY(trackpoints[i]), points.GetZ(trackpoints[i])}; | |
278 | AliHLTUInt32_t planeId=0; | |
279 | int result=FindMatchingTrackPoint(trackpoints[i], xyz, planeId); | |
280 | if (result<0) { | |
0ca33cc6 | 281 | if (GetVerbosity()>0) HLTWarning("no associated track point found for space point id %08x x=%f y=%f z=%f", trackpoints[i], xyz[0], xyz[1], xyz[2]); |
e1e03704 | 282 | continue; |
283 | } else if (result==0) { | |
284 | HLTWarning("associated track point for space pointid %08x x=%f y=%f z=%f occupied", trackpoints[i], xyz[0], xyz[1], xyz[2]); | |
285 | continue; | |
286 | } | |
54ff4c01 | 287 | vector<AliHLTTrackPoint>::const_iterator element = find(fTrackPoints.begin(), fTrackPoints.end(), planeId); |
288 | SetAssociatedSpacePoint(planeId, trackpoints[i], 1, xyz[1]-element->GetU(), xyz[2]-element->GetV()); | |
e1e03704 | 289 | if (points.GetTrackID(trackpoints[i])<0 && GetTrackId()>=0) { |
290 | points.SetTrackID(GetTrackId(), trackpoints[i]); | |
291 | HLTDebug("associating unused cluster %08x with track %d", trackpoints[i], GetTrackId()); | |
292 | } | |
293 | count++; | |
294 | } | |
295 | return count; | |
296 | } | |
297 | ||
298 | int AliHLTTrackGeometry::AssociateUnusedSpacePoints(AliHLTSpacePointContainer& points) | |
299 | { | |
300 | /// associate the track space points to the calculated track points | |
301 | int count=0; | |
54ff4c01 | 302 | for (vector<AliHLTUInt32_t>::iterator mask=fSelectionMasks.begin(); |
303 | mask!=fSelectionMasks.end(); mask++) { | |
304 | int subcount=0; | |
305 | const vector<AliHLTUInt32_t>* selectedPoints=points.GetClusterIDs(*mask); | |
306 | if (!selectedPoints) { | |
307 | HLTWarning("space point collection does not contain data for mask 0x%08x", *mask); | |
e1e03704 | 308 | continue; |
309 | } | |
54ff4c01 | 310 | for (vector<AliHLTUInt32_t>::const_iterator id=selectedPoints->begin(); |
311 | id!=selectedPoints->end(); id++) { | |
312 | if (points.GetTrackID(*id)>=0) continue; | |
313 | float xyz[3]={points.GetX(*id), points.GetY(*id), points.GetZ(*id)}; | |
314 | AliHLTUInt32_t planeId=0; | |
315 | int result=FindMatchingTrackPoint(*id, xyz, planeId); | |
316 | if (result<0) { | |
317 | //HLTWarning("no associated track point found for space point id %08x x=%f y=%f z=%f", *id, xyz[0], xyz[1], xyz[2]); | |
318 | continue; | |
319 | } else if (result==0) { | |
320 | //HLTWarning("associated track point for space pointid %08x x=%f y=%f z=%f occupied", *id, xyz[0], xyz[1], xyz[2]); | |
321 | continue; | |
322 | } | |
323 | SetAssociatedSpacePoint(planeId, *id, 1); | |
324 | if (points.GetTrackID(*id)<0 && GetTrackId()>=0) { | |
325 | points.SetTrackID(GetTrackId(), *id); | |
326 | HLTDebug("associating unused cluster %08x with track %d", *id, GetTrackId()); | |
327 | } | |
328 | subcount++; | |
e1e03704 | 329 | } |
54ff4c01 | 330 | if (fVerbosity>0) { |
331 | HLTInfo("associated %d of %d spacepoint(s) from selection 0x%08x to track %d", | |
332 | subcount, selectedPoints->size(), *mask, GetTrackId()); | |
333 | } | |
334 | count+=subcount; | |
e1e03704 | 335 | } |
336 | return count; | |
337 | } | |
338 | ||
54ff4c01 | 339 | int AliHLTTrackGeometry::FillResidual(int coordinate, TH2* histo) const |
340 | { | |
341 | // fill residual histogram | |
342 | const vector<AliHLTTrackPoint>& trackPoints=TrackPoints(); | |
343 | for (vector<AliHLTTrackPoint>::const_iterator trackpoint=trackPoints.begin(); | |
344 | trackpoint!=trackPoints.end(); trackpoint++) { | |
0ca33cc6 | 345 | if (!trackpoint->HaveAssociatedSpacePoint()) continue; |
c5efb946 | 346 | for (vector<AliHLTTrackSpacepoint>::const_iterator sp=(trackpoint->GetSpacepoints()).begin(); |
347 | sp!=(trackpoint->GetSpacepoints()).end(); sp++) { | |
348 | histo->Fill(GetPlaneR(trackpoint->GetId()), sp->GetResidual(coordinate)); | |
349 | } | |
54ff4c01 | 350 | } |
351 | return 0; | |
352 | } | |
353 | ||
e1e03704 | 354 | ostream& operator<<(ostream &out, const AliHLTTrackGeometry& p) |
355 | { | |
356 | p.Print(out); | |
357 | return out; | |
358 | } |