Default Branch split level set to 99
[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 Revision 1.14  2001/05/21 17:22:51  buncic
19 Fixed problem with missing AliConfig while reading galice.root
20
21 Revision 1.13  2001/05/16 14:57:22  alibrary
22 New files for folders and Stack
23
24 Revision 1.12  2001/03/12 17:47:03  hristov
25 Changes needed on Sun with CC 5.0
26
27 Revision 1.11  2001/01/26 19:58:46  hristov
28 Major upgrade of AliRoot code
29
30 Revision 1.10  2001/01/17 10:50:50  hristov
31 Corrections to destructors
32
33 Revision 1.9  2000/12/12 18:19:06  alibrary
34 Introduce consistency check when loading points
35
36 Revision 1.8  2000/11/30 07:12:48  alibrary
37 Introducing new Rndm and QA classes
38
39 Revision 1.7  2000/10/02 21:28:14  fca
40 Removal of useless dependecies via forward declarations
41
42 Revision 1.6  2000/07/12 08:56:25  fca
43 Coding convention correction and warning removal
44
45 Revision 1.5  1999/09/29 09:24:29  fca
46 Introduction of the Copyright and cvs Log
47
48 */
49
50 ///////////////////////////////////////////////////////////////////////////////
51 //                                                                           //
52 // Base class for ALICE modules. Both sensitive modules (detectors) and      //
53 // non-sensitive ones are described by this base class. This class           //
54 // supports the hit and digit trees produced by the simulation and also      //
55 // the objects produced by the reconstruction.                               //
56 //                                                                           //
57 // This class is also responsible for building the geometry of the           //
58 // detectors.                                                                //
59 //                                                                           //
60 //Begin_Html
61 /*
62 <img src="picts/AliDetectorClass.gif">
63 */
64 //End_Html
65 //                                                                           //
66 ///////////////////////////////////////////////////////////////////////////////
67
68 #include <assert.h>
69 #include <iostream.h>
70
71 #include <TTree.h>
72 #include <TBrowser.h>
73 #include <TFile.h>
74 #include <TROOT.h>
75 #include <TFolder.h>
76
77 #include "AliConfig.h"
78 #include "AliDetector.h"
79 #include "AliRun.h"
80 #include "AliHit.h"
81 #include "AliPoints.h"
82
83 // Static variables for the hit iterator routines
84 static Int_t sMaxIterHit=0;
85 static Int_t sCurIterHit=0;
86
87 ClassImp(AliDetector)
88  
89 //_____________________________________________________________________________
90 AliDetector::AliDetector()
91 {
92   //
93   // Default constructor for the AliDetector class
94   //
95   fNhits      = 0;
96   fNdigits    = 0;
97   fPoints     = 0;
98   fHits       = 0;
99   fDigits     = 0;
100   fTimeGate   = 200.e-9;
101   fBufferSize = 16000;
102   fDigitsFile = 0;
103 }
104  
105 //_____________________________________________________________________________
106 AliDetector::AliDetector(const char* name,const char *title):AliModule(name,title)
107 {
108   //
109   // Normal constructor invoked by all Detectors.
110   // Create the list for detector specific histograms
111   // Add this Detector to the global list of Detectors in Run.
112   //
113
114   fTimeGate   = 200.e-9;
115   fActive     = kTRUE;
116   fNhits      = 0;
117   fHits       = 0;
118   fDigits     = 0;
119   fNdigits    = 0;
120   fPoints     = 0;
121   fBufferSize = 16000;
122   fDigitsFile = 0;
123
124   AliConfig::Instance()->Add(this);
125 }
126  
127 //_____________________________________________________________________________
128 AliDetector::~AliDetector()
129 {
130   //
131   // Destructor
132   //
133   fNhits      = 0;
134   fNdigits    = 0;
135   //
136   // Delete space point structure
137   if (fPoints) {
138     fPoints->Delete();
139     delete fPoints;
140     fPoints     = 0;
141   }
142   // Delete digits structure
143   if (fDigits) {
144     fDigits->Delete();
145     delete fDigits;
146     fDigits     = 0;
147   }
148   if (fDigitsFile) delete [] fDigitsFile;
149 }
150
151 //_____________________________________________________________________________
152 void AliDetector::Publish(const char *dir, void *address, const char *name)
153 {
154   //
155   // Register pointer to detector objects. 
156   // 
157   TFolder *topFolder = (TFolder *)gROOT->FindObjectAny("/Folders");
158   if  (topFolder) { 
159     TFolder *folder = (TFolder *)topFolder->FindObjectAny(dir);
160     // TFolder *folder = (TFolder *)gROOT->FindObjectAny(dir);
161     if (!folder)  {
162       cerr << "Cannot register: Missing folder: " << dir << endl;
163     } else {
164       TFolder *subfolder = (TFolder *) folder->FindObjectAny(this->GetName()); 
165
166       if(!subfolder)
167          subfolder = folder->AddFolder(this->GetName(),this->GetTitle());
168       if (address) {
169         TObject **obj = (TObject **) address;
170         if ((*obj)->InheritsFrom(TCollection::Class())) {
171            TCollection *collection = (TCollection *) (*obj); 
172            if (name)
173              collection->SetName(name);
174         } 
175         subfolder->Add(*obj);
176       } 
177     }  
178   }
179 }
180
181 //_____________________________________________________________________________
182 TBranch* AliDetector::MakeBranchInTree(TTree *tree, const char* name, void* address, Int_t size,const char *file)
183
184     return(MakeBranchInTree(tree,name,0,address,size,99,file));
185 }
186
187 //_____________________________________________________________________________
188 TBranch* AliDetector::MakeBranchInTree(TTree *tree, const char* name, const char *classname, void* address,Int_t size, Int_t splitlevel, const char *file)
189
190     //
191     // Makes branch in given tree and diverts them to a separate file
192     //  
193     if (GetDebug()>1)
194       printf("* MakeBranch * Making Branch %s \n",name);
195       
196     TDirectory *cwd = gDirectory;
197     TBranch *branch = 0;
198     
199     if (classname) {
200       branch = tree->Branch(name,classname,address,size,splitlevel);
201     } else {
202       branch = tree->Branch(name,address,size);
203     }
204        
205     if (file) {
206         char * outFile = new char[strlen(gAlice->GetBaseFile())+strlen(file)+2];
207         sprintf(outFile,"%s/%s",gAlice->GetBaseFile(),file);
208         branch->SetFile(outFile);
209         TIter next( branch->GetListOfBranches());
210         while ((branch=(TBranch*)next())) {
211            branch->SetFile(outFile);
212         } 
213        delete outFile;
214         
215        cwd->cd();
216         
217        if (GetDebug()>1)
218            printf("* MakeBranch * Diverting Branch %s to file %s\n",name,file);
219     }
220     char *folder = 0;
221         
222     if (!strncmp(tree->GetName(),"TreeE",5)) folder = "RunMC/Event/Data";
223     if (!strncmp(tree->GetName(),"TreeK",5)) folder = "RunMC/Event/Data";
224     if (!strncmp(tree->GetName(),"TreeH",5)) folder = "RunMC/Event/Data";
225     if (!strncmp(tree->GetName(),"TreeD",5)) folder = "Run/Event/Data";
226     if (!strncmp(tree->GetName(),"TreeS",5)) folder = "Run/Event/Data";
227     if (!strncmp(tree->GetName(),"TreeR",5)) folder = "Run/Event/RecData";
228
229     if (folder) {
230       if (GetDebug())
231           printf("%15s: Publishing %s to %s\n",ClassName(),name,folder);
232       Publish(folder,address,name);
233     }  
234     return branch;
235 }
236
237 //_____________________________________________________________________________
238 void AliDetector::Browse(TBrowser *b)
239 {
240   //
241   // Insert Detector objects in the list of objects to be browsed
242   //
243   char name[64];
244   if( fHits == 0) return;
245   TObject *obj;
246   Int_t i, nobjects;
247   //
248   nobjects = fHits->GetEntries();
249   for (i=0;i<nobjects;i++) {
250     obj = fHits->At(i);
251     sprintf(name,"%s_%d",obj->GetName(),i);
252     b->Add(obj, &name[0]);
253   }
254 }
255
256 //_____________________________________________________________________________
257 void AliDetector::Copy(AliDetector &det) const
258 {
259   //
260   // Copy *this onto det -- not implemented
261   //
262   Fatal("Copy","Not implemented~\n");
263 }
264
265 //_____________________________________________________________________________
266 void AliDetector::FinishRun()
267 {
268   //
269   // Procedure called at the end of a run.
270   //
271 }
272
273 //_____________________________________________________________________________
274 AliHit* AliDetector::FirstHit(Int_t track)
275 {
276   //
277   // Initialise the hit iterator
278   // Return the address of the first hit for track
279   // If track>=0 the track is read from disk
280   // while if track<0 the first hit of the current
281   // track is returned
282   // 
283   if(track>=0) {
284     gAlice->ResetHits();
285     gAlice->TreeH()->GetEvent(track);
286   }
287   //
288   sMaxIterHit=fHits->GetEntriesFast();
289   sCurIterHit=0;
290   if(sMaxIterHit) return (AliHit*) fHits->UncheckedAt(0);
291   else            return 0;
292 }
293
294 //_____________________________________________________________________________
295 AliHit* AliDetector::NextHit()
296 {
297   //
298   // Return the next hit for the current track
299   //
300   if(sMaxIterHit) {
301     if(++sCurIterHit<sMaxIterHit) 
302       return (AliHit*) fHits->UncheckedAt(sCurIterHit);
303     else        
304       return 0;
305   } else {
306     printf("* AliDetector::NextHit * Hit Iterator called without calling FistHit before\n");
307     return 0;
308   }
309 }
310
311 //_____________________________________________________________________________
312 void AliDetector::LoadPoints(Int_t)
313 {
314   //
315   // Store x, y, z of all hits in memory
316   //
317   if (fHits == 0) return;
318   //
319   Int_t nhits = fHits->GetEntriesFast();
320   if (nhits == 0) return;
321   Int_t tracks = gAlice->GetNtrack();
322   if (fPoints == 0) fPoints = new TObjArray(tracks);
323   AliHit *ahit;
324   //
325   Int_t *ntrk=new Int_t[tracks];
326   Int_t *limi=new Int_t[tracks];
327   Float_t **coor=new Float_t*[tracks];
328   for(Int_t i=0;i<tracks;i++) {
329     ntrk[i]=0;
330     coor[i]=0;
331     limi[i]=0;
332   }
333   //
334   AliPoints *points = 0;
335   Float_t *fp=0;
336   Int_t trk;
337   Int_t chunk=nhits/4+1;
338   //
339   // Loop over all the hits and store their position
340   for (Int_t hit=0;hit<nhits;hit++) {
341     ahit = (AliHit*)fHits->UncheckedAt(hit);
342     trk=ahit->GetTrack();
343     assert(trk<=tracks);
344     if(ntrk[trk]==limi[trk]) {
345       //
346       // Initialise a new track
347       fp=new Float_t[3*(limi[trk]+chunk)];
348       if(coor[trk]) {
349         memcpy(fp,coor[trk],sizeof(Float_t)*3*limi[trk]);
350         delete [] coor[trk];
351       }
352       limi[trk]+=chunk;
353       coor[trk] = fp;
354     } else {
355       fp = coor[trk];
356     }
357     fp[3*ntrk[trk]  ] = ahit->X();
358     fp[3*ntrk[trk]+1] = ahit->Y();
359     fp[3*ntrk[trk]+2] = ahit->Z();
360     ntrk[trk]++;
361   }
362   //
363   for(trk=0; trk<tracks; ++trk) {
364     if(ntrk[trk]) {
365       points = new AliPoints();
366       points->SetMarkerColor(GetMarkerColor());
367       points->SetMarkerSize(GetMarkerSize());
368       points->SetDetector(this);
369       points->SetParticle(trk);
370       points->SetPolyMarker(ntrk[trk],coor[trk],GetMarkerStyle());
371       fPoints->AddAt(points,trk);
372       delete [] coor[trk];
373       coor[trk]=0;
374     }
375   }
376   delete [] coor;
377   delete [] ntrk;
378   delete [] limi;
379 }
380
381 //_____________________________________________________________________________
382 void AliDetector::MakeBranch(Option_t *option, const char *file)
383 {
384   //
385   // Create a new branch in the current Root Tree
386   // The branch of fHits is automatically split
387   //
388  
389   char branchname[10];
390   sprintf(branchname,"%s",GetName());
391   //
392   // Get the pointer to the header
393   const char *cH = strstr(option,"H");
394   //
395   if (fHits && gAlice->TreeH() && cH) {
396     MakeBranchInTree(gAlice->TreeH(), 
397                      branchname, &fHits, fBufferSize, file) ;              
398   }     
399   
400   const char *cD = strstr(option,"D");
401
402   if (cD) {
403     if (file) {
404        fDigitsFile = new char[strlen (file)];
405        strcpy(fDigitsFile,file);
406     }
407   }
408 }
409
410 //_____________________________________________________________________________
411 void AliDetector::ResetDigits()
412 {
413   //
414   // Reset number of digits and the digits array
415   //
416   fNdigits   = 0;
417   if (fDigits)   fDigits->Clear();
418 }
419
420 //_____________________________________________________________________________
421 void AliDetector::ResetHits()
422 {
423   //
424   // Reset number of hits and the hits array
425   //
426   fNhits   = 0;
427   if (fHits)   fHits->Clear();
428 }
429
430 //_____________________________________________________________________________
431 void AliDetector::ResetPoints()
432 {
433   //
434   // Reset array of points
435   //
436   if (fPoints) {
437     fPoints->Delete();
438     delete fPoints;
439     fPoints = 0;
440   }
441 }
442
443 //_____________________________________________________________________________
444 void AliDetector::SetTreeAddress()
445 {
446   //
447   // Set branch address for the Hits and Digits Trees
448   //
449   TBranch *branch;
450   char branchname[20];
451   sprintf(branchname,"%s",GetName());
452   //
453   // Branch address for hit tree
454   TTree *treeH = gAlice->TreeH();
455   if (treeH && fHits) {
456     branch = treeH->GetBranch(branchname);
457     if (branch) branch->SetAddress(&fHits);
458   }
459   //
460   // Branch address for digit tree
461   TTree *treeD = gAlice->TreeD();
462   if (treeD && fDigits) {
463     branch = treeD->GetBranch(branchname);
464     if (branch) branch->SetAddress(&fDigits);
465   }
466 }
467
468