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