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