Adaption to new fluka common blocks (E. Futo)
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALRecPoint.cxx
CommitLineData
ab48128d 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15/* $Id$ */
16//_________________________________________________________________________
17// Base Class for EMCAL Reconstructed Points
18// Why should I put meaningless comments
19// just to satisfy
20// the code checker
21//*-- Author: Gines Martinez (SUBATECH)
22
23// --- ROOT system ---
24#include "TPad.h"
25#include "TClonesArray.h"
26
27// --- Standard library ---
ab48128d 28#include <stdio.h>
29
30// --- AliRoot header files ---
31
32#include "AliEMCALGeometry.h"
33#include "AliEMCALDigit.h"
34#include "AliEMCALRecPoint.h"
35#include "AliEMCALGetter.h"
36
37ClassImp(AliEMCALRecPoint)
38
39
40//____________________________________________________________________________
41AliEMCALRecPoint::AliEMCALRecPoint()
42 : AliRecPoint()
43{
44 // ctor
45
692088ae 46 fMaxTrack = 0 ;
ab48128d 47 fTheta = fPhi = 0. ;
692088ae 48 fEMCALArm = 0;
aad8e277 49 fPRESection = fECALSection = fHCALSection = kFALSE ;
ab48128d 50}
51
52//____________________________________________________________________________
53AliEMCALRecPoint::AliEMCALRecPoint(const char * opt) : AliRecPoint(opt)
54{
55 // ctor
56
57 fMaxTrack = 200 ;
58 fTheta = fPhi = 0. ;
59 fEMCALArm = 1;
60
61}
62
63//____________________________________________________________________________
64Int_t AliEMCALRecPoint::DistancetoPrimitive(Int_t px, Int_t py)
65{
66 // Compute distance from point px,py to a AliEMCALRecPoint considered as a Tmarker
67 // Compute the closest distance of approach from point px,py to this marker.
68 // The distance is computed in pixels units.
69
70 TVector3 pos(0.,0.,0.) ;
71 GetLocalPosition( pos) ;
72 Float_t x = pos.X() ;
73 Float_t y = pos.Z() ;
74 const Int_t kMaxDiff = 10;
75 Int_t pxm = gPad->XtoAbsPixel(x);
76 Int_t pym = gPad->YtoAbsPixel(y);
77 Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
78
79 if (dist > kMaxDiff) return 9999;
80 return dist;
81}
82
83//___________________________________________________________________________
84 void AliEMCALRecPoint::Draw(Option_t *option)
85 {
86 // Draw this AliEMCALRecPoint with its current attributes
87
88 AppendPad(option);
89 }
90
91//______________________________________________________________________________
92void AliEMCALRecPoint::ExecuteEvent(Int_t event, Int_t px, Int_t py)
93{
94 // Execute action corresponding to one event
95 // This member function is called when a AliEMCALRecPoint is clicked with the locator
96 //
97 // If Left button is clicked on AliEMCALRecPoint, the digits are switched on
98 // and switched off when the mouse button is released.
99
100 // static Int_t pxold, pyold;
101
102 static TGraph * digitgraph = 0 ;
103 static TPaveText* clustertext = 0 ;
104
105 if (!gPad->IsEditable()) return;
106
107 switch (event) {
108
109
110 case kButton1Down:{
111 AliEMCALDigit * digit ;
112 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
113 AliEMCALGeometry * emcalgeom = const_cast<AliEMCALGeometry*>(gime->EMCALGeometry());
114
115 Int_t iDigit;
116 Int_t relid[4] ;
117
118 const Int_t kMulDigit=AliEMCALRecPoint::GetDigitsMultiplicity() ;
119 Float_t * xi = new Float_t [kMulDigit] ;
120 Float_t * zi = new Float_t [kMulDigit] ;
121
122 for(iDigit = 0; iDigit < kMulDigit; iDigit++) {
123 Fatal("AliEMCALRecPoint::ExecuteEvent", " -> Something wrong with the code");
124 digit = 0 ; //dynamic_cast<AliEMCALDigit *>((fDigitsList)[iDigit]);
125 emcalgeom->AbsToRelNumbering(digit->GetId(), relid) ;
126 emcalgeom->PosInAlice(relid, xi[iDigit], zi[iDigit]) ;
127 }
128
129 if (!digitgraph) {
130 digitgraph = new TGraph(fMulDigit,xi,zi);
131 digitgraph-> SetMarkerStyle(5) ;
132 digitgraph-> SetMarkerSize(1.) ;
133 digitgraph-> SetMarkerColor(1) ;
134 digitgraph-> Draw("P") ;
135 }
136 if (!clustertext) {
137
138 TVector3 pos(0.,0.,0.) ;
139 GetLocalPosition(pos) ;
140 clustertext = new TPaveText(pos.X()-10,pos.Z()+10,pos.X()+50,pos.Z()+35,"") ;
141 Text_t line1[40] ;
142 Text_t line2[40] ;
143 sprintf(line1,"Energy=%1.2f GeV",GetEnergy()) ;
144 sprintf(line2,"%d Digits",GetDigitsMultiplicity()) ;
145 clustertext ->AddText(line1) ;
146 clustertext ->AddText(line2) ;
147 clustertext ->Draw("");
148 }
149 gPad->Update() ;
150 Print() ;
151 delete[] xi ;
152 delete[] zi ;
153 }
154
155break;
156
157 case kButton1Up:
158 if (digitgraph) {
159 delete digitgraph ;
160 digitgraph = 0 ;
161 }
162 if (clustertext) {
163 delete clustertext ;
164 clustertext = 0 ;
165 }
166
167 break;
168
169 }
170}
171//____________________________________________________________________________
172void AliEMCALRecPoint::EvalAll(Float_t logWeight,TClonesArray * digits) {
173 //evaluates (if necessary) all RecPoint data members
174
175 EvalPrimaries(digits) ;
176}
177
178//____________________________________________________________________________
179void AliEMCALRecPoint::EvalEMCALArm(AliEMCALDigit * digit)
180{
181 // Returns the EMCAL module in which the RecPoint is found
182
183
184 if( fEMCALArm == 0){
185 Int_t relid[4] ;
186
187 AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
188 AliEMCALGeometry * emcalgeom = const_cast<AliEMCALGeometry*>(gime->EMCALGeometry());
189
190 emcalgeom->AbsToRelNumbering(digit->GetId(), relid) ;
191 fEMCALArm = relid[0];
192 }
193}
194
195//______________________________________________________________________________
196void AliEMCALRecPoint::EvalPrimaries(TClonesArray * digits)
197{
198 // Constructs the list of primary particles (tracks) which have contributed to this RecPoint
199
200 AliEMCALDigit * digit ;
201 Int_t * tempo = new Int_t[fMaxTrack] ;
202
203 Int_t index ;
204 for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
205 digit = dynamic_cast<AliEMCALDigit *>(digits->At( fDigitsList[index] )) ;
206 Int_t nprimaries = digit->GetNprimary() ;
207 Int_t * newprimaryarray = new Int_t[nprimaries] ;
208 Int_t ii ;
209 for ( ii = 0 ; ii < nprimaries ; ii++)
210 newprimaryarray[ii] = digit->GetPrimary(ii+1) ;
211
212 Int_t jndex ;
213 for ( jndex = 0 ; jndex < nprimaries ; jndex++ ) { // all primaries in digit
214 if ( fMulTrack > fMaxTrack ) {
215 fMulTrack = - 1 ;
9859bfc0 216 Error("GetNprimaries", "increase fMaxTrack ") ;
ab48128d 217 break ;
218 }
219 Int_t newprimary = newprimaryarray[jndex] ;
220 Int_t kndex ;
221 Bool_t already = kFALSE ;
222 for ( kndex = 0 ; kndex < fMulTrack ; kndex++ ) { //check if not already stored
223 if ( newprimary == tempo[kndex] ){
224 already = kTRUE ;
225 break ;
226 }
227 } // end of check
228 if ( !already) { // store it
229 tempo[fMulTrack] = newprimary ;
230 fMulTrack++ ;
231 } // store it
232 } // all primaries in digit
233 delete newprimaryarray ;
234 } // all digits
235
236
237 fTracksList = new Int_t[fMulTrack] ;
238 for(index = 0; index < fMulTrack; index++)
239 fTracksList[index] = tempo[index] ;
240
241 delete tempo ;
242
243}
7ee5c5be 244
ab48128d 245//____________________________________________________________________________
246void AliEMCALRecPoint::GetGlobalPosition(TVector3 & gpos) const
247{
248 // returns the position of the cluster in the global reference system of ALICE
ab48128d 249
250 AliEMCALGeometry * emcalgeom = AliEMCALGetter::GetInstance()->EMCALGeometry();
251 gpos.SetX(fPhi) ;
aad8e277 252 if ( IsInECAL() )
253 gpos.SetY(emcalgeom->GetIP2ECALSection()) ;
254 else if ( IsInPRE() )
255 gpos.SetY(emcalgeom->GetIP2PRESection()) ;
256 else if ( IsInHCAL() )
257 gpos.SetY(emcalgeom->GetIP2HCALSection()) ;
7ee5c5be 258 else
aad8e277 259 Fatal("GetGlobalPosition", "Unexpected tower section") ;
ab48128d 260 gpos.SetZ(fTheta) ;
ab48128d 261}
262
aad8e277 263//____________________________________________________________________________
264void AliEMCALRecPoint::GetLocalPosition(TVector3 & lpos) const
265{
266 // returns the position of the cluster in the global reference system of ALICE
267
268 lpos.SetX(fLocPos.X()) ;
269 lpos.SetY(fLocPos.Y()) ;
270 lpos.SetZ(fLocPos.Z()) ;
271}
ab48128d 272
273//______________________________________________________________________________
274void AliEMCALRecPoint::Paint(Option_t *)
275{
276 // Paint this ALiRecPoint as a TMarker with its current attributes
277
278 TVector3 pos(0.,0.,0.) ;
279 GetLocalPosition(pos) ;
280 Coord_t x = pos.X() ;
281 Coord_t y = pos.Z() ;
282 Color_t markercolor = 1 ;
283 Size_t markersize = 1. ;
284 Style_t markerstyle = 5 ;
285
286 if (!gPad->IsBatch()) {
287 gVirtualX->SetMarkerColor(markercolor) ;
288 gVirtualX->SetMarkerSize (markersize) ;
289 gVirtualX->SetMarkerStyle(markerstyle) ;
290 }
291 gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
292 gPad->PaintPolyMarker(1,&x,&y,"") ;
293}