Introduction of the Copyright and cvs Log
[u/mrichter/AliRoot.git] / STEER / AliDetector.cxx
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
16 /*
17 $Log$
18 */
19
20 ///////////////////////////////////////////////////////////////////////////////
21 //                                                                           //
22 // Base class for ALICE modules. Both sensitive modules (detectors) and      //
23 // non-sensitive ones are described by this base class. This class           //
24 // supports the hit and digit trees produced by the simulation and also      //
25 // the objects produced by the reconstruction.                               //
26 //                                                                           //
27 // This class is also responsible for building the geometry of the           //
28 // detectors.                                                                //
29 //                                                                           //
30 //Begin_Html
31 /*
32 <img src="picts/AliDetectorClass.gif">
33 */
34 //End_Html
35 //                                                                           //
36 ///////////////////////////////////////////////////////////////////////////////
37 #include "AliDetector.h"
38 #include "AliRun.h"
39 #include "AliHit.h"
40 #include "AliPoints.h"
41 #include <TClass.h>
42 #include <TNode.h>
43 #include <TRandom.h>
44
45 // Static variables for the hit iterator routines
46 static Int_t sMaxIterHit=0;
47 static Int_t sCurIterHit=0;
48
49 ClassImp(AliDetector)
50  
51 //_____________________________________________________________________________
52 AliDetector::AliDetector()
53 {
54   //
55   // Default constructor for the AliDetector class
56   //
57   fNhits      = 0;
58   fNdigits    = 0;
59   fPoints     = 0;
60   fHits       = 0;
61   fDigits     = 0;
62   fTimeGate   = 200.e-9;
63   fBufferSize = 16000;
64 }
65  
66 //_____________________________________________________________________________
67 AliDetector::AliDetector(const char* name,const char *title):AliModule(name,title)
68 {
69   //
70   // Normal constructor invoked by all Detectors.
71   // Create the list for detector specific histograms
72   // Add this Detector to the global list of Detectors in Run.
73   //
74
75   fTimeGate   = 200.e-9;
76   fActive     = kTRUE;
77   fNhits      = 0;
78   fHits       = 0;
79   fDigits     = 0;
80   fNdigits    = 0;
81   fPoints     = 0;
82   fBufferSize = 16000;
83 }
84  
85 //_____________________________________________________________________________
86 AliDetector::~AliDetector()
87 {
88   //
89   // Destructor
90   //
91   fNhits      = 0;
92   fNdigits    = 0;
93   //
94   // Delete space point structure
95   if (fPoints) fPoints->Delete();
96   delete fPoints;
97   fPoints     = 0;
98 }
99  
100 //_____________________________________________________________________________
101 void AliDetector::Browse(TBrowser *b)
102 {
103   //
104   // Insert Detector objects in the list of objects to be browsed
105   //
106   char name[64];
107   if( fHits == 0) return;
108   TObject *obj;
109   Int_t i, nobjects;
110   //
111   nobjects = fHits->GetEntries();
112   for (i=0;i<nobjects;i++) {
113     obj = fHits->At(i);
114     sprintf(name,"%s_%d",obj->GetName(),i);
115     b->Add(obj, &name[0]);
116   }
117 }
118
119 //_____________________________________________________________________________
120 void AliDetector::FinishRun()
121 {
122   //
123   // Procedure called at the end of a run.
124   //
125 }
126
127 //_____________________________________________________________________________
128 AliHit* AliDetector::FirstHit(Int_t track)
129 {
130   //
131   // Initialise the hit iterator
132   // Return the address of the first hit for track
133   // If track>=0 the track is read from disk
134   // while if track<0 the first hit of the current
135   // track is returned
136   // 
137   if(track>=0) {
138     gAlice->ResetHits();
139     gAlice->TreeH()->GetEvent(track);
140   }
141   //
142   sMaxIterHit=fHits->GetEntriesFast();
143   sCurIterHit=0;
144   if(sMaxIterHit) return (AliHit*) fHits->UncheckedAt(0);
145   else            return 0;
146 }
147
148 //_____________________________________________________________________________
149 AliHit* AliDetector::NextHit()
150 {
151   //
152   // Return the next hit for the current track
153   //
154   if(sMaxIterHit) {
155     if(++sCurIterHit<sMaxIterHit) 
156       return (AliHit*) fHits->UncheckedAt(sCurIterHit);
157     else        
158       return 0;
159   } else {
160     printf("* AliDetector::NextHit * Hit Iterator called without calling FistHit before\n");
161     return 0;
162   }
163 }
164
165 //_____________________________________________________________________________
166 void AliDetector::LoadPoints(Int_t)
167 {
168   //
169   // Store x, y, z of all hits in memory
170   //
171   if (fHits == 0) return;
172   //
173   Int_t nhits = fHits->GetEntriesFast();
174   if (nhits == 0) return;
175   Int_t tracks = gAlice->GetNtrack();
176   if (fPoints == 0) fPoints = new TObjArray(tracks);
177   AliHit *ahit;
178   //
179   Int_t *ntrk=new Int_t[tracks];
180   Int_t *limi=new Int_t[tracks];
181   Float_t **coor=new Float_t*[tracks];
182   for(Int_t i=0;i<tracks;i++) {
183     ntrk[i]=0;
184     coor[i]=0;
185     limi[i]=0;
186   }
187   //
188   AliPoints *points = 0;
189   Float_t *fp=0;
190   Int_t trk;
191   Int_t chunk=nhits/4+1;
192   //
193   // Loop over all the hits and store their position
194   for (Int_t hit=0;hit<nhits;hit++) {
195     ahit = (AliHit*)fHits->UncheckedAt(hit);
196     trk=ahit->GetTrack();
197     if(ntrk[trk]==limi[trk]) {
198       //
199       // Initialise a new track
200       fp=new Float_t[3*(limi[trk]+chunk)];
201       if(coor[trk]) {
202         memcpy(fp,coor[trk],sizeof(Float_t)*3*limi[trk]);
203         delete [] coor[trk];
204       }
205       limi[trk]+=chunk;
206       coor[trk] = fp;
207     } else {
208       fp = coor[trk];
209     }
210     fp[3*ntrk[trk]  ] = ahit->fX;
211     fp[3*ntrk[trk]+1] = ahit->fY;
212     fp[3*ntrk[trk]+2] = ahit->fZ;
213     ntrk[trk]++;
214   }
215   //
216   for(trk=0; trk<tracks; ++trk) {
217     if(ntrk[trk]) {
218       points = new AliPoints();
219       points->SetMarkerColor(GetMarkerColor());
220       points->SetMarkerSize(GetMarkerSize());
221       points->SetDetector(this);
222       points->SetParticle(trk);
223       points->SetPolyMarker(ntrk[trk],coor[trk],GetMarkerStyle());
224       fPoints->AddAt(points,trk);
225       delete [] coor[trk];
226       coor[trk]=0;
227     }
228   }
229   delete [] coor;
230   delete [] ntrk;
231   delete [] limi;
232 }
233
234 //_____________________________________________________________________________
235 void AliDetector::MakeBranch(Option_t *option)
236 {
237   //
238   // Create a new branch in the current Root Tree
239   // The branch of fHits is automatically split
240   //
241   char branchname[10];
242   sprintf(branchname,"%s",GetName());
243   //
244   // Get the pointer to the header
245   char *H = strstr(option,"H");
246   //
247   if (fHits   && gAlice->TreeH() && H) {
248     gAlice->TreeH()->Branch(branchname,&fHits, fBufferSize);
249     printf("* AliDetector::MakeBranch * Making Branch %s for hits\n",branchname);
250   }     
251 }
252
253 //_____________________________________________________________________________
254 void AliDetector::ResetDigits()
255 {
256   //
257   // Reset number of digits and the digits array
258   //
259   fNdigits   = 0;
260   if (fDigits)   fDigits->Clear();
261 }
262
263 //_____________________________________________________________________________
264 void AliDetector::ResetHits()
265 {
266   //
267   // Reset number of hits and the hits array
268   //
269   fNhits   = 0;
270   if (fHits)   fHits->Clear();
271 }
272
273 //_____________________________________________________________________________
274 void AliDetector::ResetPoints()
275 {
276   //
277   // Reset array of points
278   //
279   if (fPoints) {
280     fPoints->Delete();
281     delete fPoints;
282     fPoints = 0;
283   }
284 }
285
286 //_____________________________________________________________________________
287 void AliDetector::SetTreeAddress()
288 {
289   //
290   // Set branch address for the Hits and Digits Trees
291   //
292   TBranch *branch;
293   char branchname[20];
294   sprintf(branchname,"%s",GetName());
295   //
296   // Branch address for hit tree
297   TTree *treeH = gAlice->TreeH();
298   if (treeH && fHits) {
299     branch = treeH->GetBranch(branchname);
300     if (branch) branch->SetAddress(&fHits);
301   }
302   //
303   // Branch address for digit tree
304   TTree *treeD = gAlice->TreeD();
305   if (treeD && fDigits) {
306     branch = treeD->GetBranch(branchname);
307     if (branch) branch->SetAddress(&fDigits);
308   }
309 }
310
311 //_____________________________________________________________________________
312 void AliDetector::Streamer(TBuffer &R__b)
313 {
314   //
315   // Stream an object of class Detector.
316   //
317   if (R__b.IsReading()) {
318     Version_t R__v = R__b.ReadVersion(); if (R__v) { }
319     TNamed::Streamer(R__b);
320     TAttLine::Streamer(R__b);
321     TAttMarker::Streamer(R__b);
322     AliModule::Streamer(R__b);
323     R__b >> fTimeGate;
324     R__b >> fIshunt;
325     //R__b >> fNhits;
326     //
327     // Stream the pointers but not the TClonesArrays
328     R__b >> fHits; // diff
329     R__b >> fDigits; // diff
330     
331   } else {
332     R__b.WriteVersion(AliDetector::IsA());
333     TNamed::Streamer(R__b);
334     TAttLine::Streamer(R__b);
335     TAttMarker::Streamer(R__b);
336     AliModule::Streamer(R__b);
337     R__b << fTimeGate;
338     R__b << fIshunt;
339     //R__b << fNhits;
340     //
341     // Stream the pointers but not the TClonesArrays
342     R__b << fHits; // diff
343     R__b << fDigits; // diff
344   }
345 }
346