EMCAL/AliEMCALReconstructioner.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 /* $Id$ */
  17
  18 //_________________________________________________________________________
  19 //*--
  20 //*-- Author: Gines Martinez & Yves Schutz (SUBATECH)
  21 //*-- Compleetely redesigned by Dmitri Peressounko (SUBATECH & RRC KI) March 2001
  22 /////////////////////////////////////////////////////////////////////////////////////
  23 //  Wrapping class for reconstruction. Allows to produce reconstruction from
  24 //  different steps: from previously produced hits,sdigits, etc. Each new reconstruction
  25 //  flow (e.g. digits, made from them RecPoints,
  26 //  subsequently made RecParticles) are distinguished by the title of created branches. One can
  27 //  use this title as a comment, see use case below.
  28 //  Thanks to getters, one can set
  29 //  parameters to reconstruction briks. The full set of parameters is saved in the
  30 //  corresponding branch: e.g. parameters of clusterizer are stored in branch
  31 //  TreeR::AliEMCALClusterizer with the same title as the branch containing the RecPoints. //  TTree does not support overwriting, therefore one can not produce several
  32 //  branches with the same names and titles - use different titles.
  33 //
  34 //  Use case:
  35 //
  36 //  root [0] AliEMCALReconstructioner * r = new AliEMCALReconstructioner("galice.root")
  37 //              //  Set the header file
  38 //  root [1] r->ExecuteTask()
  39 //              //  Make full chain of reconstruction
  40 //
  41 //              // One can specify the title for each branch
  42 //  root [2] r->SetBranchFileName("RecPoints","RecPoints1") ;
  43 //
  44 //             // One can change parameters of reconstruction algorithms
  45 //  root [3] r->GetClusterizer()->SetEmcLocalMaxCut(0.02)
  46 //
  47 //             // One can specify the starting point of the reconstruction and title of all
  48 //             // branches produced in this pass
  49 //  root [4] r->StartFrom("AliEMCALClusterizer","Local max cut 0.02")
  50 //             // means that will use already generated Digits and produce only RecPoints,
  51 //             // and RecParticles
  52 //
  53 //             // And finally one can call ExecuteTask() with the following options
  54 //  root [5] r->ExecuteTask("debug all timing")
  55 //             // deb     - prints the numbers of RecPoints and RecParticles
  56 //             // deb all - prints in addition list of RecPoints and RecParticles
  57 //             // timing  - prints benchmarking results
  58 ///////////////////////////////////////////////////////////////////////////////////////////////////
  59
  60 // --- ROOT system ---
  61
  62 // --- Standard library ---
  63
  64 // --- AliRoot header files ---
  65 #include "AliESD.h"
  66 #include "AliESDCaloTrack.h"
  67 #include "AliEMCALReconstructioner.h"
  68 #include "AliEMCALClusterizerv1.h"
  69 #include "AliEMCALPIDv1.h"
  70 #include "AliEMCALGetter.h"
  71
  72 ClassImp(AliEMCALReconstructioner)
  73
  74 //____________________________________________________________________________
  75   AliEMCALReconstructioner::AliEMCALReconstructioner():TTask("AliEMCALReconstructioner","")
  76 {
  77   // ctor
  78   fClusterizer = 0 ;
  79   fPID         = 0 ;
  80
  81   fIsInitialized = kFALSE ;
  82
  83 }
  84
  85 //____________________________________________________________________________
  86 AliEMCALReconstructioner::AliEMCALReconstructioner(const char* evFoldName,const char * branchName):
  87 TTask("AliEMCALReconstructioner",evFoldName)
  88 {
  89   // ctor
  90
  91   AliEMCALGetter::Instance(evFoldName) ;
  92
  93   fRecPointBranch=branchName ;
  94   fClusterizer = new AliEMCALClusterizerv1(evFoldName, GetTitle());
  95   Add(fClusterizer);
  96
  97   fRecPartBranch=branchName ;
  98   fPID         = new AliEMCALPIDv1(evFoldName, GetTitle());
  99   Add(fPID);
 100
 101   fIsInitialized = kTRUE ;
 102 }
 103 //____________________________________________________________________________
 104 void AliEMCALReconstructioner::Exec(Option_t *opt)
 105 {
 106   //check, if the names of branches, which should be made coincide with already
 107   //existing
 108   if (!opt)
 109     return ;
 110   if(!fIsInitialized)
 111     Init() ;
 112 }
 113 //____________________________________________________________________________
 114 void AliEMCALReconstructioner:: Clusters2Tracks(Int_t ievent, AliESD *event)
 115 {
 116   // Convert EMCAL reconstructed particles into ESD object for event# ievent.
 117   // ESD object is returned as an argument event
 118
 119   if(!fIsInitialized) Init() ;
 120
 121   fClusterizer->SetEventRange(ievent,ievent);
 122   fClusterizer->ExecuteTask();
 123
 124   fPID        ->SetEventRange(ievent,ievent);
 125   fPID        ->ExecuteTask();
 126
 127   AliEMCALGetter *gime = AliEMCALGetter::Instance();
 128   TClonesArray *recParticles = gime->RecParticles();
 129   Int_t nOfRecParticles = recParticles->GetEntries();
 130   for (Int_t recpart=0; recpart<nOfRecParticles; recpart++) {
 131     AliESDCaloTrack *ct = new AliESDCaloTrack((AliEMCALRecParticle*)recParticles->At(recpart));
 132     event->AddCaloTrack(ct);
 133   }
 134
 135 }
 136 //____________________________________________________________________________
 137  void AliEMCALReconstructioner::Init()
 138 {
 139   // initiliaze Reconstructioner if necessary: we can not do this in default constructor
 140
 141   if(!fIsInitialized){
 142
 143     fRecPointBranch="Default" ;
 144     fClusterizer = new AliEMCALClusterizerv1(GetTitle(),fRecPointBranch.Data());
 145     Add(fClusterizer) ;
 146
 147     fRecPartBranch="Default";
 148     fPID         = new AliEMCALPIDv1(GetTitle(),fRecPartBranch.Data()) ;
 149     Add(fPID) ;
 150
 151     fIsInitialized = kTRUE ;
 152
 153   }
 154 }
 155 //____________________________________________________________________________
 156 AliEMCALReconstructioner::~AliEMCALReconstructioner()
 157 {
 158   // Delete data members if any
 159 }
 160
 161 //____________________________________________________________________________
 162 void AliEMCALReconstructioner::Print()const {
 163   // Print reconstructioner data
 164
 165   TString message ;
 166   message  = "-----------------AliEMCALReconstructioner---------------\n" ;
 167   message += " Reconstruction of the header file %s\n" ;
 168   message += " with the following modules:\n" ;
 169
 170   if(fClusterizer->IsActive()){
 171     message += "   (+)   %s to branch %s\n" ;
 172   }
 173
 174   if(fPID->IsActive()){
 175     message += "   (+)   %s to branch %s\n" ;
 176   }
 177   Info("Print", message.Data(),
 178        GetTitle(),
 179        fClusterizer->GetName(), fRecPointBranch.Data(),
 180        fPID->GetName(), fRecPartBranch.Data() ) ;
 181 }
 182
 183 //____________________________________________________________________________
 184 void AliEMCALReconstructioner::SetEventRange(Int_t first, Int_t last)
 185 {
 186   // Set the event range to process
 187   fFirstEvent=first;
 188   fLastEvent=last;
 189   fClusterizer->SetEventRange(fFirstEvent, fLastEvent) ;
 190   fPID->SetEventRange(fFirstEvent, fLastEvent) ;
 191 }