aff0b6c80bba8d317d814575e09d284defc0b03a
[u/mrichter/AliRoot.git] / TRD / AliTRDclusterizer.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.11.6.2  2002/10/11 07:26:37  hristov
19 Updating VirtualMC to v3-09-02
20
21 Revision 1.13  2002/09/18 09:20:53  cblume
22 Write the parameter class into the cluster file
23
24 Revision 1.11  2001/11/27 08:50:33  hristov
25 BranchOld replaced by Branch
26
27 Revision 1.10  2001/11/14 10:50:45  cblume
28 Changes in digits IO. Add merging of summable digits
29
30 Revision 1.9  2001/10/21 18:30:02  hristov
31 Several pointers were set to zero in the default constructors to avoid memory management problems
32
33 Revision 1.8  2001/05/07 08:06:44  cblume
34 Speedup of the code. Create only AliTRDcluster
35
36 Revision 1.7  2001/03/30 14:40:14  cblume
37 Update of the digitization parameter
38
39 Revision 1.6  2000/11/01 14:53:20  cblume
40 Merge with TRD-develop
41
42
43 Revision 1.1.4.5  2000/10/15 23:40:01  cblume
44 Remove AliTRDconst
45
46 Revision 1.1.4.4  2000/10/06 16:49:46  cblume
47 Made Getters const
48
49 Revision 1.1.4.3  2000/10/04 16:34:58  cblume
50 Replace include files by forward declarations
51
52 Revision 1.1.4.2  2000/09/22 14:49:49  cblume
53 Adapted to tracking code
54
55 Revision 1.5  2000/10/02 21:28:19  fca
56 Removal of useless dependecies via forward declarations
57
58 Revision 1.4  2000/06/09 11:10:07  cblume
59 Compiler warnings and coding conventions, next round
60
61 Revision 1.3  2000/06/08 18:32:58  cblume
62 Make code compliant to coding conventions
63
64 Revision 1.2  2000/05/08 16:17:27  cblume
65 Merge TRD-develop
66
67 Revision 1.1.4.1  2000/05/08 15:08:03  cblume
68 Remove the class AliTRDcluster
69
70 Revision 1.4  2000/06/09 11:10:07  cblume
71 Compiler warnings and coding conventions, next round
72
73 Revision 1.3  2000/06/08 18:32:58  cblume
74 Make code compliant to coding conventions
75
76 Revision 1.2  2000/05/08 16:17:27  cblume
77 Merge TRD-develop
78
79 Revision 1.1.4.1  2000/05/08 15:08:03  cblume
80 Remove the class AliTRDcluster
81
82 Revision 1.1  2000/02/28 18:57:58  cblume
83 Add new TRD classes
84
85 */
86
87 ///////////////////////////////////////////////////////////////////////////////
88 //                                                                           //
89 //  TRD cluster finder base class                                            //
90 //                                                                           //
91 ///////////////////////////////////////////////////////////////////////////////
92
93 #include <TROOT.h>
94 #include <TTree.h>
95 #include <TFile.h>
96
97 #include "AliRun.h"
98 #include "AliTRD.h"
99 #include "AliTRDclusterizer.h"
100 #include "AliTRDcluster.h"
101 #include "AliTRDrecPoint.h"
102 #include "AliTRDgeometry.h"
103 #include "AliTRDparameter.h"
104
105 ClassImp(AliTRDclusterizer)
106
107 //_____________________________________________________________________________
108 AliTRDclusterizer::AliTRDclusterizer():TNamed()
109 {
110   //
111   // AliTRDclusterizer default constructor
112   //
113
114   fInputFile   = NULL;
115   fOutputFile  = NULL;
116   fClusterTree = NULL;
117   fTRD         = 0;
118   fEvent       = 0;
119   fVerbose     = 0;
120   fPar         = 0;
121
122 }
123
124 //_____________________________________________________________________________
125 AliTRDclusterizer::AliTRDclusterizer(const Text_t* name, const Text_t* title)
126                   :TNamed(name, title)
127 {
128   //
129   // AliTRDclusterizer default constructor
130   //
131
132   fInputFile   = NULL;
133   fOutputFile  = NULL;
134   fClusterTree = NULL;
135   fEvent       = 0;
136   fVerbose     = 0;
137   fPar         = 0;
138
139 }
140
141 //_____________________________________________________________________________
142 AliTRDclusterizer::AliTRDclusterizer(const AliTRDclusterizer &c)
143 {
144   //
145   // AliTRDclusterizer copy constructor
146   //
147
148   ((AliTRDclusterizer &) c).Copy(*this);
149
150 }
151
152 //_____________________________________________________________________________
153 AliTRDclusterizer::~AliTRDclusterizer()
154 {
155   //
156   // AliTRDclusterizer destructor
157   //
158
159   if (fInputFile) {
160     fInputFile->Close();
161     delete fInputFile;
162   }
163
164   if (fOutputFile) {
165     fOutputFile->Close();
166     delete fOutputFile;
167   }
168
169   if (fClusterTree) {
170     delete fClusterTree;
171   }
172
173 }
174
175 //_____________________________________________________________________________
176 AliTRDclusterizer &AliTRDclusterizer::operator=(const AliTRDclusterizer &c)
177 {
178   //
179   // Assignment operator
180   //
181
182   if (this != &c) ((AliTRDclusterizer &) c).Copy(*this);
183   return *this;
184
185 }
186
187 //_____________________________________________________________________________
188 void AliTRDclusterizer::Copy(TObject &c)
189 {
190   //
191   // Copy function
192   //
193
194   ((AliTRDclusterizer &) c).fInputFile   = NULL;
195   ((AliTRDclusterizer &) c).fOutputFile  = NULL;
196   ((AliTRDclusterizer &) c).fClusterTree = NULL;
197   ((AliTRDclusterizer &) c).fEvent       = 0;  
198   ((AliTRDclusterizer &) c).fVerbose     = fVerbose;  
199   ((AliTRDclusterizer &) c).fPar         = 0;
200
201 }
202
203 //_____________________________________________________________________________
204 Bool_t AliTRDclusterizer::Open(const Char_t *name, Int_t nEvent)
205 {
206   //
207   // Opens the AliROOT file. Output and input are in the same file
208   //
209
210   OpenInput(name,nEvent);
211   OpenOutput(name);
212
213   return kTRUE;
214
215 }
216
217 //_____________________________________________________________________________
218 Bool_t AliTRDclusterizer::Open(const Char_t *inname, const Char_t *outname
219                               , Int_t nEvent)
220 {
221   //
222   // Opens the AliROOT file. Output and input are in different files
223   //
224
225   OpenInput(inname,nEvent);
226   OpenOutput(outname);
227
228   return kTRUE;
229
230 }
231
232 //_____________________________________________________________________________
233 Bool_t AliTRDclusterizer::OpenOutput(const Char_t *name)
234 {
235   //
236   // Open the output file
237   //
238
239   TDirectory *savedir = NULL;
240
241   if (!fInputFile) return kFALSE;
242
243   if (strcmp(name,fInputFile->GetName()) != 0) {
244     savedir = gDirectory;
245     printf("AliTRDclusterizer::OpenOutput -- ");
246     printf("Open the output file %s.\n",name);
247     fOutputFile = new TFile(name,"RECREATE");
248   }
249
250   // Create a tree for the cluster
251   Char_t treeName[12];
252   sprintf(treeName,"TreeR%d_TRD",fEvent);
253   fClusterTree = new TTree(treeName,"TRD cluster");
254   TObjArray *ioArray = 0;
255   fClusterTree->Branch("TRDcluster","TObjArray",&ioArray,32000,0);
256
257   if (savedir) {
258     savedir->cd();
259   }
260
261   return kTRUE;
262
263 }
264
265 //_____________________________________________________________________________
266 Bool_t AliTRDclusterizer::OpenInput(const Char_t *name, Int_t nEvent)
267 {
268   //
269   // Opens a ROOT-file with TRD-hits and reads in the digits-tree
270   //
271
272   // Connect the AliRoot file containing Geometry, Kine, and Hits
273   fInputFile = (TFile*) gROOT->GetListOfFiles()->FindObject(name);
274   if (!fInputFile) {
275     printf("AliTRDclusterizer::OpenInput -- ");
276     printf("Open the ALIROOT-file %s.\n",name);
277     fInputFile = new TFile(name,"UPDATE");
278   }
279   else {
280     printf("AliTRDclusterizer::OpenInput -- ");
281     printf("%s is already open.\n",name);
282   }
283
284   // Get AliRun object from file
285   gAlice = (AliRun *) fInputFile->Get("gAlice");
286   if (!(gAlice)) {
287     printf("AliTRDclusterizer::OpenInput -- ");
288     printf("Could not find AliRun object.\n");
289     return kFALSE;
290   }
291
292   fEvent = nEvent;
293
294   // Import the Trees for the event nEvent in the file
295   Int_t nparticles = gAlice->GetEvent(fEvent);
296   if (nparticles <= 0) {
297     printf("AliTRDclusterizer::OpenInput -- ");
298     printf("No entries in the trees for event %d.\n",fEvent);
299     return kFALSE;
300   }
301
302   // Get the TRD object
303   fTRD = (AliTRD*) gAlice->GetDetector("TRD"); 
304   if (!fTRD) {
305     printf("AliTRDclusterizer::OpenInput -- ");
306     printf("No TRD detector object found\n");
307     return kFALSE;
308   }
309
310   return kTRUE;
311
312 }
313
314 //_____________________________________________________________________________
315 Bool_t AliTRDclusterizer::WriteClusters(Int_t det)
316 {
317   //
318   // Fills TRDcluster branch in the tree with the clusters 
319   // found in detector = det. For det=-1 writes the tree. 
320   //
321
322   if ((det < -1) || (det >= AliTRDgeometry::Ndet())) {
323     printf("AliTRDclusterizer::WriteClusters -- ");
324     printf("Unexpected detector index %d.\n",det);
325     return kFALSE;
326   }
327  
328   TDirectory *savedir = gDirectory;
329
330   if (fOutputFile) {
331     fOutputFile->cd();
332   }
333
334   TBranch *branch = fClusterTree->GetBranch("TRDcluster");
335   if (!branch) {
336     TObjArray *ioArray = 0;
337     branch = fClusterTree->Branch("TRDcluster","TObjArray",&ioArray,32000,0);
338   }
339
340   if ((det >= 0) && (det < AliTRDgeometry::Ndet())) {
341
342     Int_t nRecPoints = fTRD->RecPoints()->GetEntriesFast();
343     TObjArray *detRecPoints = new TObjArray(400);
344
345     for (Int_t i = 0; i < nRecPoints; i++) {
346       AliTRDcluster *c = (AliTRDcluster *) fTRD->RecPoints()->UncheckedAt(i);
347       if (det == c->GetDetector()) {
348         detRecPoints->AddLast(c);
349       }
350       else {
351         printf("AliTRDclusterizer::WriteClusters --");
352         printf("Attempt to write a cluster with unexpected detector index\n");
353       }
354     }
355
356     branch->SetAddress(&detRecPoints);
357     fClusterTree->Fill();
358
359     return kTRUE;
360
361   }
362
363   if (det == -1) {
364
365     printf("AliTRDclusterizer::WriteClusters -- ");
366     printf("Writing the cluster tree %-18s for event %d.\n"
367           ,fClusterTree->GetName(),fEvent);
368
369     fClusterTree->Write();
370
371     AliTRDgeometry *geo = fTRD->GetGeometry();
372     geo->SetName("TRDgeometry");
373     geo->Write();
374     fPar->Write();
375      
376     return kTRUE;  
377
378   }
379   
380   savedir->cd();
381
382   printf("AliTRDclusterizer::WriteClusters -- ");
383   printf("Unexpected detector index %d.\n",det);
384  
385   return kFALSE;  
386
387 }
388
389
390