]> git.uio.no Git - u/mrichter/AliRoot.git/blame - HLT/EVE/AliHLTEveMuon.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
compilation warnings fixed
[u/mrichter/AliRoot.git] / HLT / EVE / AliHLTEveMuon.cxx
CommitLineData
34ba7866 1/**************************************************************************\r
2 * This file is property of and copyright by the ALICE HLT Project *\r
3 * ALICE Experiment at CERN, All rights reserved. *\r
4 * *\r
5 * Primary Authors: Svein Lindal <slindal@fys.uio.no > *\r
6 * for The ALICE HLT Project. *\r
7 * *\r
8 * Permission to use, copy, modify and distribute this software and its *\r
9 * documentation strictly for non-commercial purposes is hereby granted *\r
10 * without fee, provided that the above copyright notice appears in all *\r
11 * copies and that both the copyright notice and this permission notice *\r
12 * appear in the supporting documentation. The authors make no claims *\r
13 * about the suitability of this software for any purpose. It is *\r
14 * provided "as is" without express or implied warranty. *\r
15 **************************************************************************/\r
16\r
17/// @file AliHLTEvePhos.cxx\r
18/// @author Svein Lindal <slindal@fys.uio.no>\r
19/// @brief Muon processor for the HLT EVE display\r
20\r
21#include "AliHLTEveMuon.h"\r
22#include "AliHLTMUONDataBlockReader.h"\r
23#include "AliHLTHOMERBlockDesc.h"\r
24#include "TCanvas.h"\r
25#include "TEveStraightLineSet.h"\r
26#include "TEvePointSet.h"\r
27#include "AliEveHOMERManager.h"\r
28#include "TEveManager.h"\r
29\r
30\r
31#include "TEveVSDStructs.h"\r
32#include "TGeoGlobalMagField.h"\r
33#include "AliESDMuonTrack.h"\r
34#include "AliESDMuonCluster.h"\r
35#include "AliEveMUONTrack.h"\r
36#include "AliHLTMUONConstants.h"\r
37#include "AliHLTMUONUtils.h"\r
38#include "AliMUONVCluster.h"\r
39#include "AliMUONConstants.h"\r
40#include "TEveTrackPropagator.h"\r
41\r
42using namespace std;\r
43\r
44class AliHLTMUONUtils;\r
45class AliEveMuonTrack;\r
46\r
47ClassImp(AliHLTEveMuon);\r
48\r
49AliHLTEveMuon::AliHLTEveMuon() : \r
50 AliHLTEveBase(),\r
51 fFullTrackList(NULL),\r
52 fTracks(NULL),\r
53 fClusters(NULL)\r
54{\r
55 // Constructor.\r
933fb19e 56 SetMaxHistograms(6);\r
34ba7866 57}\r
58\r
59AliHLTEveMuon::~AliHLTEveMuon()\r
60{\r
61 //Destructor\r
62 if (fFullTrackList)\r
63 delete fFullTrackList;\r
64 fFullTrackList = NULL;\r
65 \r
66 if (fTracks)\r
67 delete fTracks;\r
68 fTracks = NULL;\r
69\r
70 if(fClusters)\r
71 delete fClusters;\r
72 fClusters = NULL;\r
73}\r
74\r
75\r
76void AliHLTEveMuon::ProcessBlock(AliHLTHOMERBlockDesc * block) {\r
77 //See header file for documentation\r
78 if ( (block->GetDataType().CompareTo("RECHITS") == 0) || (block->GetDataType().CompareTo("TRIGRECS") == 0) ) {\r
79 if(!fClusters) {\r
80 fClusters = CreateClusters();\r
81 fEventManager->GetEveManager()->AddElement(fClusters);\r
82 }\r
83 ProcessClusters( block, fClusters );\r
84 \r
85 }else if(block->GetDataType().CompareTo("MANTRACK") == 0){\r
86 \r
87 if ( !fTracks ) {\r
88 fTracks = CreateTrackSet(); \r
89 fEventManager->GetEveManager()->AddElement(fTracks);\r
90 gEve->AddElement(fTracks);\r
91 }\r
92 \r
93 ProcessTracks( block, fTracks );\r
94\r
95 }else if(block->GetDataType().CompareTo("TRACKS") == 0){\r
96 \r
97 if ( !fFullTrackList ) {\r
98 fFullTrackList = CreateFullTrackList(); \r
99 fEventManager->GetEveManager()->AddElement(fFullTrackList); \r
100 gEve->AddElement(fFullTrackList);\r
101 }\r
102 \r
103 ProcessFullTracks( block, fFullTrackList );\r
0e77fbbf 104\r
105 } else if(block->GetDataType().CompareTo("ROOTHIST") == 0) {\r
106 ProcessHistogram(block);\r
107 }\r
34ba7866 108 \r
109}\r
110\r
111TEvePointSet * AliHLTEveMuon::CreateClusters() {\r
112 //See header file for documentation\r
113 TEvePointSet * ps = new TEvePointSet("MUON RecHits");\r
114 ps->SetMainColor(kBlue);\r
115 ps->SetMarkerStyle(20);\r
116 return ps;\r
117}\r
118\r
119TEveStraightLineSet * AliHLTEveMuon::CreateTrackSet() {\r
120 // See header file\r
121 TEveStraightLineSet * ls = new TEveStraightLineSet("MUON Tracks");\r
122 ls->SetMainColor(kRed);\r
123 ls->SetLineWidth(3);\r
124 return ls;\r
125}\r
126\r
127TEveTrackList * AliHLTEveMuon::CreateFullTrackList(){\r
128 // See header file\r
129 TEveTrackList * ls = new TEveTrackList("MUON Full Tracks");\r
130 ls->SetMainColor(kBlue);\r
131 return ls;\r
132}\r
133\r
134void AliHLTEveMuon::ProcessHistogram(AliHLTHOMERBlockDesc * block ) {\r
135 //See header file for documentation\r
136 if(!fCanvas) {\r
137 fCanvas = CreateCanvas("MUON QA", "MUON QA");\r
138 fCanvas->Divide(3, 2);\r
139 }\r
140 AddHistogramsToCanvas(block, fCanvas, fHistoCount);\r
141}\r
142\r
143void AliHLTEveMuon::UpdateElements() {\r
144 //See header file for documentation\r
145 if(fCanvas) fCanvas->Update();\r
146 if(fClusters) fClusters->ResetBBox();\r
147 if(fTracks) fTracks->ElementChanged();\r
148}\r
149\r
150void AliHLTEveMuon::ResetElements(){\r
151 //See header file for documentation\r
152 fHistoCount = 0;\r
153 \r
154 if ( fClusters ) fClusters->Reset();\r
155 if ( fTracks ){\r
156 fTracks->Destroy();\r
157 fTracks = NULL;\r
158 }\r
159 if ( fFullTrackList ){\r
160 fFullTrackList->Destroy();\r
161 fFullTrackList = NULL;\r
162 }\r
163\r
164\r
165}\r
166\r
167void AliHLTEveMuon::ProcessClusters(AliHLTHOMERBlockDesc * block, TEvePointSet * clusters) {\r
168 //See header file for documentation\r
169 unsigned long size = block->GetSize();\r
170 Int_t * buffer ;\r
171 \r
172 buffer = (Int_t *)block->GetData();\r
173 //cout<<"block size : "<<size<<", buffer : "<<buffer<<", DataType : "<<block->GetDataType()<<endl;\r
174\r
175 if(block->GetDataType().CompareTo("RECHITS") == 0){\r
176 \r
177 AliHLTMUONRecHitsBlockReader trackblock((char*)buffer, size);\r
178 const AliHLTMUONRecHitStruct* hit = trackblock.GetArray();\r
179 \r
180 for(AliHLTUInt32_t ientry = 0; ientry < trackblock.Nentries(); ientry++){\r
181 if(hit->fX!=0.0 && hit->fY!=0.0 && hit->fZ!=0.0)\r
182 clusters->SetNextPoint(hit->fX,hit->fY,hit->fZ);\r
183 hit++;\r
184 \r
185 }// track hit loop\r
186 }\r
187 \r
188 else{// if rechits\r
189 // if(!strcmp((BlockType(ULong64_t(reader->GetBlockDataType(i)))).Data(),"TRIGRECS")){\r
190 \r
191 AliHLTMUONTriggerRecordsBlockReader trigblock(buffer, size);\r
192 const AliHLTMUONTriggerRecordStruct* trigrec = trigblock.GetArray();\r
193 for(AliHLTUInt32_t ientry = 0; ientry < trigblock.Nentries(); ientry++){\r
194 \r
195 const AliHLTMUONRecHitStruct* hit = &trigrec->fHit[0];\r
196 for(AliHLTUInt32_t ch = 0; ch < 4; ch++)\r
197 {\r
198 if(hit->fX!=0.0 && hit->fY!=0.0 && hit->fZ!=0.0)\r
199 clusters->SetNextPoint(hit->fX,hit->fY,hit->fZ);\r
200 hit++;\r
201 }// trig chamber loop\r
202 trigrec++;\r
203 }//trig hit loop\r
204 }//else trigger\r
205 \r
206}\r
207\r
208void AliHLTEveMuon::ProcessTracks(AliHLTHOMERBlockDesc * block, TEveStraightLineSet * tracks) {\r
209 //See header file for documentation \r
210 unsigned long size = block->GetSize();\r
211 Int_t * buffer = (Int_t *)block->GetData();\r
212 AliHLTMUONRecHitStruct hit1,hit2;\r
213 hit1.fX = hit1.fY = hit1.fZ = hit2.fX = hit2.fY = hit2.fZ = 0;\r
214 Int_t ch1=0, ch2=0;\r
215 Float_t x0=0.0,y0=0.0,z0=0.0;\r
216 Float_t x3=0.0,y3=0.0,z3=0.0;\r
217 if(block->GetDataType().CompareTo("MANTRACK") == 0){ \r
218 AliHLTMUONMansoTracksBlockReader mantrackblock(buffer, size);\r
219 const AliHLTMUONMansoTrackStruct* mtrack = mantrackblock.GetArray();\r
220 for(AliHLTUInt32_t ientry = 0; ientry < mantrackblock.Nentries(); ientry++){\r
221 const AliHLTMUONRecHitStruct* hit = &mtrack->fHit[0];\r
222 for(AliHLTUInt32_t ch = 0; ch < 4; ch++){\r
223 // cout << setw(10) << left << ch + 7 << setw(0);\r
224 // cout << setw(13) << left << hit->fX << setw(0);\r
225 // cout << setw(13) << left << hit->fY << setw(0);\r
226 // cout << hit->fZ << setw(0) << endl;\r
227 if(hit->fZ != 0.0){\r
228 if(ch==0 || ch==1){\r
229 hit1 = *hit; ch1 = ch+6;\r
230 }else{\r
231 hit2 = *hit; ch2 = ch+6;\r
232 }\r
233 }\r
234 hit++;\r
235 }// trig chamber loop\r
236 // printf("ch : %d, (X,Y,Z) : (%f,%f,%f)\n",ch1,hit1.fX,hit1.fY,hit1.fZ);\r
237 // printf("ch : %d, (X,Y,Z) : (%f,%f,%f)\n",ch2,hit2.fX,hit2.fY,hit2.fZ);\r
238 // meminfo();\r
239 z3 = AliMUONConstants::DefaultChamberZ(ch2+4);\r
240 y3 = hit1.fY - (hit1.fZ-z3)*(hit1.fY - hit2.fY)/(hit1.fZ - hit2.fZ) ;\r
241 x3 = hit1.fX - (hit1.fZ-z3)*(hit1.fX - hit2.fX)/(hit1.fZ - hit2.fZ) ;\r
242\r
243 z0 = AliMUONConstants::DefaultChamberZ(ch1);\r
244 y0 = hit1.fY - (hit1.fZ-z0)*(hit1.fY - hit2.fY)/(hit1.fZ - hit2.fZ) ;\r
245 x0 = hit1.fX - (hit1.fZ-z0)*(hit1.fX - hit2.fX)/(hit1.fZ - hit2.fZ) ;\r
246 \r
247\r
248 tracks->AddLine(x0,y0,z0,x3,y3,z3);\r
249 mtrack++;\r
250 }\r
251 // cout<<"NofManso Tracks : "<<mantrackblock.Nentries()<<endl;\r
252 }\r
253}\r
254\r
255int AliHLTEveMuon::MakeMUONESDTrack(AliESDMuonTrack *muonESDTrack, const AliHLTMUONTrackStruct *muonHLTTrack)\r
256{\r
257 // See header for documentation\r
258 AliHLTUInt32_t clusterIndex = 0; // for the cluster unique ID. \r
259 AliHLTMUONParticleSign sign;\r
260 bool hitset[16];\r
261 AliHLTMUONUtils::UnpackTrackFlags(\r
262 muonHLTTrack->fFlags, sign, hitset\r
263 );\r
264 \r
265 TVector3 mom(muonHLTTrack->fPx, muonHLTTrack->fPy, muonHLTTrack->fPz);\r
266 if (mom.Mag() != 0)\r
267 muonESDTrack->SetInverseBendingMomentum(muonHLTTrack->fInverseBendingMomentum);\r
268 else\r
269 muonESDTrack->SetInverseBendingMomentum(0.);\r
270 muonESDTrack->SetThetaX(muonHLTTrack->fThetaX);\r
271 muonESDTrack->SetThetaY(muonHLTTrack->fThetaY);\r
272 muonESDTrack->SetZ(muonHLTTrack->fZ);\r
273 muonESDTrack->SetBendingCoor(muonHLTTrack->fY);\r
274 muonESDTrack->SetNonBendingCoor(muonHLTTrack->fX);\r
275\r
276 //printf("(X,Y,Z) : (%8.3f,%8.3f,%8.3f)\n",muonHLTTrack->fX,muonHLTTrack->fY,muonHLTTrack->fZ);\r
277\r
278 muonESDTrack->SetChi2(muonHLTTrack->fChi2);\r
279\r
280 Int_t nHits = 0;\r
281 for (int i = 0; i < 16; i++)\r
282 {\r
283 if (not hitset[i]) continue;\r
284 \r
285 AliHLTUInt8_t chamber;\r
286 AliHLTUInt16_t detElemId;\r
287 AliHLTMUONUtils::UnpackRecHitFlags((muonHLTTrack->fHit[i]).fFlags, chamber, detElemId);\r
288 \r
289 AliESDMuonCluster cluster;\r
290 cluster.SetUniqueID(AliMUONVCluster::BuildUniqueID(chamber, detElemId, clusterIndex++));\r
291 cluster.SetXYZ((muonHLTTrack->fHit[i]).fX, (muonHLTTrack->fHit[i]).fY, (muonHLTTrack->fHit[i]).fZ);\r
292 cluster.SetErrXY( // Use nominal values.\r
293 AliHLTMUONConstants::DefaultNonBendingReso(),\r
294 AliHLTMUONConstants::DefaultBendingReso()\r
295 );\r
296 cluster.SetCharge(-1.); // Indicate no total charge calculated.\r
297 cluster.SetChi2(-1.); // Indicate no fit made.\r
298 muonESDTrack->AddCluster(cluster);\r
299 nHits++;\r
300 muonESDTrack->AddInMuonClusterMap(chamber);\r
301 }\r
302 \r
303 muonESDTrack->SetNHit(nHits);\r
304\r
305 return 0;\r
306}\r
307\r
308Int_t AliHLTEveMuon::ProcessFullTracks(AliHLTHOMERBlockDesc * block, TEveTrackList * fullTracks) {\r
309\r
310 // See header for documentation \r
311\r
312 Int_t iResult = 0;\r
313\r
314 Double_t b[3], x[3];\r
315 x[0] = 0.0 ; x[1] = 0.0 ; x[2] = -950.0;\r
316 TGeoGlobalMagField::Instance()->Field(x,b);\r
317 //" Field at (0.0, 0.0, -950.0) [at the middle of dipole magnet] \r
318 //should be (6.79, 0.03, -0.17) or similar value with change of sign"\r
319 if(TMath::AreEqualAbs(b[0],0.0,1.0e-5) and TMath::AreEqualAbs(b[1],0.0,1.0e-5) and TMath::AreEqualAbs(b[2],0.0,1.0e-5)){\r
320 printf("At (X,Y,Z) : (%6.2lf,%6.2lf,%6.2lf) Field (Bx,By,Bz) is (%6.2lf,%6.2lf,%6.2lf)\n",\r
321 x[0],x[1],x[2],b[0],b[1],b[2]); \r
322 cerr<<"Magnetic field is not properly set, MUON tracking will not possble"<<endl;\r
323 return 1;\r
324 }\r
325 \r
326 \r
327\r
328 TEveRecTrack rt;\r
329 \r
330 unsigned long size = block->GetSize();\r
331 Int_t * buffer = (Int_t *)block->GetData();\r
332\r
333 AliHLTMUONTracksBlockReader muontrackblock(buffer, size);\r
334 const AliHLTMUONTrackStruct* mtrack = muontrackblock.GetArray();\r
335 //cout<<"NofTracks : "<<muontrackblock.Nentries()<<endl;\r
336 for(AliHLTUInt32_t ientry = 0; ientry < muontrackblock.Nentries(); ientry++){\r
337 \r
338 AliESDMuonTrack *muonESDTrack = new AliESDMuonTrack();\r
339 MakeMUONESDTrack(muonESDTrack,mtrack);\r
340 if(muonESDTrack->GetNHit()==0){\r
341 muonESDTrack->Delete();\r
342 continue;\r
343 }\r
344 \r
345 rt.fLabel = ientry;\r
346 AliEveMUONTrack* track = new AliEveMUONTrack(&rt, fullTracks->GetPropagator());\r
347 track->MakeESDTrack(muonESDTrack);\r
348 //track->SetTitle(Form("HLT Track : %d, pt : %lf",ientry,TMath::Sqrt(((mtrack->fPx * mtrack->fPx) + (mtrack->fPy * mtrack->fPy)))));\r
349 track->SetName(Form("HLT Track : %d, pt : %lf",ientry,TMath::Sqrt(((mtrack->fPx * mtrack->fPx) + (mtrack->fPy * mtrack->fPy)))));\r
350 gEve->AddElement(track, fullTracks);\r
351 \r
352 mtrack++;\r
353 }//track loop\r
354 \r
355 return iResult;\r
356\r
357}\r