ITS/AliITSReconstructor.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 // class for ITS reconstruction                                              //
  21 //                                                                           //
  22 ///////////////////////////////////////////////////////////////////////////////
  23
  24 #include "Riostream.h"
  25 #include "AliITSReconstructor.h"
  26 #include "AliRun.h"
  27 #include "AliRunLoader.h"
  28 #include "AliRawReader.h"
  29 #include "AliITSDetTypeRec.h"
  30 #include "AliITSLoader.h"
  31 #include "AliITStrackerMI.h"
  32 #include "AliITStrackerV2.h"
  33 #include "AliITStrackerSA.h"
  34 #include "AliITSVertexerIons.h"
  35 #include "AliITSVertexerFast.h"
  36 #include "AliITSVertexerPPZ.h"
  37 #include "AliITSVertexerZ.h"
  38 #include "AliESD.h"
  39 #include "AliITSpidESD.h"
  40 #include "AliITSpidESD1.h"
  41 #include "AliITSpidESD2.h"
  42 #include "AliITSInitGeometry.h"
  43
  44 ClassImp(AliITSReconstructor)
  45
  46 //___________________________________________________________________________
  47 AliITSReconstructor::AliITSReconstructor() : AliReconstructor(),
  48 fItsPID(0)
  49 {
  50   // Default constructor
  51 }
  52  //___________________________________________________________________________
  53 AliITSReconstructor::~AliITSReconstructor(){
  54 // destructor
  55   delete fItsPID;
  56 }
  57 //______________________________________________________________________
  58 AliITSReconstructor::AliITSReconstructor(const AliITSReconstructor &ob) :AliReconstructor(ob),
  59 fItsPID(ob.fItsPID)
  60 {
  61   // Copy constructor
  62 }
  63
  64 //______________________________________________________________________
  65 AliITSReconstructor& AliITSReconstructor::operator=(const AliITSReconstructor&  ob ){
  66   // Assignment operator
  67   this->~AliITSReconstructor();
  68   new(this) AliITSReconstructor(ob);
  69   return *this;
  70 }
  71 //______________________________________________________________________
  72 void AliITSReconstructor::Init(AliRunLoader *runLoader) const{
  73     // Initalize this constructor bet getting/creating the objects
  74     // nesseary for a proper ITS reconstruction.
  75     // Inputs:
  76     //    AliRunLoader *runLoader   Pointer to the run loader to allow
  77     //                              the getting of files/folders data
  78     //                              needed to do reconstruction
  79     // Output:
  80     //   none.
  81     // Return:
  82     //   none.
  83
  84     AliITSInitGeometry initgeom;
  85     AliITSgeom *geom = initgeom.CreateAliITSgeom();
  86     AliInfo(Form("Geometry name: %s",(initgeom.GetGeometryName()).Data()));
  87     AliITSLoader* loader = static_cast<AliITSLoader*>
  88         (runLoader->GetLoader("ITSLoader"));
  89     if (!loader) {
  90         Error("Init", "ITS loader not found");
  91         return;
  92     }
  93     loader->SetITSgeom(geom);
  94     return;
  95 }
  96 //_____________________________________________________________________________
  97 void AliITSReconstructor::Reconstruct(AliRunLoader* runLoader) const
  98 {
  99 // reconstruct clusters
 100
 101
 102   AliITSLoader* loader = static_cast<AliITSLoader*>(runLoader->GetLoader("ITSLoader"));
 103   if (!loader) {
 104     Error("Reconstruct", "ITS loader not found");
 105     return;
 106   }
 107   AliITSDetTypeRec* rec = new AliITSDetTypeRec(loader);
 108   rec->SetDefaults();
 109
 110   loader->LoadRecPoints("recreate");
 111   loader->LoadDigits("read");
 112   runLoader->LoadKinematics();
 113   TString option = GetOption();
 114   Bool_t clusfinder=kTRUE;   // Default: V2 cluster finder
 115   if(option.Contains("OrigCF"))clusfinder=kFALSE;
 116
 117   Int_t nEvents = runLoader->GetNumberOfEvents();
 118
 119   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
 120     runLoader->GetEvent(iEvent);
 121     if(loader->TreeR()==0x0) loader->MakeTree("R");
 122     rec->MakeBranch("R");
 123     rec->SetTreeAddress();
 124     rec->DigitsToRecPoints(iEvent,0,"All",clusfinder);
 125   }
 126
 127   loader->UnloadRecPoints();
 128   loader->UnloadDigits();
 129   runLoader->UnloadKinematics();
 130 }
 131
 132 //_________________________________________________________________
 133 void AliITSReconstructor::Reconstruct(AliRunLoader* runLoader,
 134                                       AliRawReader* rawReader) const
 135 {
 136 // reconstruct clusters
 137
 138
 139   AliITSLoader* loader = static_cast<AliITSLoader*>(runLoader->GetLoader("ITSLoader"));
 140   if (!loader) {
 141     Error("Reconstruct", "ITS loader not found");
 142     return;
 143   }
 144
 145   AliITSDetTypeRec* rec = new AliITSDetTypeRec(loader);
 146   rec->SetDefaults();
 147   rec->SetDefaultClusterFindersV2(kTRUE);
 148
 149   loader->LoadRecPoints("recreate");
 150
 151   Int_t iEvent = 0;
 152
 153   while(rawReader->NextEvent()) {
 154     runLoader->GetEvent(iEvent++);
 155     if(loader->TreeR()==0x0) loader->MakeTree("R");
 156     rec->DigitsToRecPoints(rawReader);
 157   }
 158
 159   loader->UnloadRecPoints();
 160 }
 161
 162 //_____________________________________________________________________________
 163 AliTracker* AliITSReconstructor::CreateTracker(AliRunLoader* runLoader)const
 164 {
 165 // create a ITS tracker
 166
 167
 168   TString selectedTracker = GetOption();
 169   AliTracker* tracker;
 170   if (selectedTracker.Contains("MI")) {
 171     tracker = new AliITStrackerMI(0);
 172   }
 173   else if (selectedTracker.Contains("V2")) {
 174     tracker = new AliITStrackerV2(0);
 175   }
 176   else {
 177     tracker =  new AliITStrackerSA(0);  // inherits from AliITStrackerMI
 178     AliITStrackerSA *sat=(AliITStrackerSA*)tracker;
 179     if(selectedTracker.Contains("onlyITS"))sat->SetSAFlag(kTRUE);
 180     if(sat->GetSAFlag())AliDebug(1,"Tracking Performed in ITS only\n");
 181   }
 182
 183   TString selectedPIDmethod = GetOption();
 184   AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader");
 185   if (!loader) {
 186     Error("CreateTracker", "ITS loader not found");
 187   }
 188   if(selectedPIDmethod.Contains("LandauFitPID")){
 189     loader->AdoptITSpid(new AliITSpidESD2((AliITStrackerMI*)tracker,loader));
 190   }
 191   else{
 192     Double_t parITS[] = {34., 0.15, 10.};
 193     loader->AdoptITSpid(new AliITSpidESD1(parITS));
 194   }
 195   return tracker;
 196
 197 }
 198
 199 //_____________________________________________________________________________
 200 AliVertexer* AliITSReconstructor::CreateVertexer(AliRunLoader* /*runLoader*/) const
 201 {
 202 // create a ITS vertexer
 203
 204   TString selectedVertexer = GetOption();
 205   if(selectedVertexer.Contains("ions") || selectedVertexer.Contains("IONS")){
 206     Info("CreateVertexer","a AliITSVertexerIons object has been selected\n");
 207     return new AliITSVertexerIons("null");
 208   }
 209   if(selectedVertexer.Contains("smear") || selectedVertexer.Contains("SMEAR")){
 210     Double_t smear[3]={0.005,0.005,0.01};
 211     Info("CreateVertexer","a AliITSVertexerFast object has been selected\n");
 212     return new AliITSVertexerFast(smear);
 213   }
 214   if(selectedVertexer.Contains("ppz") || selectedVertexer.Contains("PPZ")){
 215     Info("CreateVertexer","a AliITSVertexerPPZ object has been selected\n");
 216     return new AliITSVertexerPPZ("null");
 217   }
 218   // by default an AliITSVertexerZ object is instatiated
 219   Info("CreateVertexer","a AliITSVertexerZ object has been selected\n");
 220   return new AliITSVertexerZ("null");
 221 }
 222
 223 //_____________________________________________________________________________
 224 void AliITSReconstructor::FillESD(AliRunLoader* runLoader,
 225                                   AliESD* esd) const
 226 {
 227 // make PID, find V0s and cascade
 228   AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader");
 229   AliITSpidESD *pidESD = 0;
 230   TString selectedPIDmethod = GetOption();
 231   if(selectedPIDmethod.Contains("LandauFitPID")){
 232     Info("FillESD","ITS LandauFitPID option has been selected\n");
 233     pidESD=loader->GetITSpid();
 234   }
 235   else{
 236     Info("FillESD","ITS default PID\n");
 237     pidESD=loader->GetITSpid();
 238   }
 239   if(pidESD!=0){
 240     pidESD->MakePID(esd);
 241   }
 242   else {
 243     Error("FillESD","!! cannot do the PID !!\n");
 244   }
 245 }
 246
 247
 248 //_____________________________________________________________________________
 249 AliITSgeom* AliITSReconstructor::GetITSgeom(AliRunLoader* runLoader) const
 250 {
 251 // get the ITS geometry
 252
 253   if (!runLoader->GetAliRun()) runLoader->LoadgAlice();
 254   if (!runLoader->GetAliRun()) {
 255     Error("GetITSgeom", "couldn't get AliRun object");
 256     return NULL;
 257   }
 258   AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader");
 259   AliITSgeom* geom = (AliITSgeom*)loader->GetITSgeom();
 260   if(!geom){
 261     Error("GetITSgeom","no ITS geometry available");
 262     return NULL;
 263   }
 264
 265   return geom;
 266 }