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 "AliITStrackerSA.h"
  33 #include "AliITSVertexerIons.h"
  34 #include "AliITSVertexerFast.h"
  35 #include "AliITSVertexerPPZ.h"
  36 #include "AliITSVertexerZ.h"
  37 #include "AliESD.h"
  38 #include "AliITSpidESD.h"
  39 #include "AliITSpidESD1.h"
  40 #include "AliITSpidESD2.h"
  41 #include "AliV0vertexer.h"
  42 #include "AliCascadeVertexer.h"
  43 #include "AliITSInitGeometry.h"
  44
  45 ClassImp(AliITSReconstructor)
  46
  47 //___________________________________________________________________________
  48 AliITSReconstructor::AliITSReconstructor() : AliReconstructor(),
  49 fItsPID(0)
  50 {
  51   // Default constructor
  52 }
  53  //___________________________________________________________________________
  54 AliITSReconstructor::~AliITSReconstructor(){
  55 // destructor
  56   delete fItsPID;
  57 }
  58 //______________________________________________________________________
  59 AliITSReconstructor::AliITSReconstructor(const AliITSReconstructor &ob) :AliReconstructor(ob),
  60 fItsPID(ob.fItsPID)
  61 {
  62   // Copy constructor
  63 }
  64
  65 //______________________________________________________________________
  66 AliITSReconstructor& AliITSReconstructor::operator=(const AliITSReconstructor&  ob ){
  67   // Assignment operator
  68   this->~AliITSReconstructor();
  69   new(this) AliITSReconstructor(ob);
  70   return *this;
  71 }
  72 //______________________________________________________________________
  73 void AliITSReconstructor::Init(AliRunLoader *runLoader) const{
  74     // Initalize this constructor bet getting/creating the objects
  75     // nesseary for a proper ITS reconstruction.
  76     // Inputs:
  77     //    AliRunLoader *runLoader   Pointer to the run loader to allow
  78     //                              the getting of files/folders data
  79     //                              needed to do reconstruction
  80     // Output:
  81     //   none.
  82     // Return:
  83     //   none.
  84
  85     AliITSInitGeometry *initgeom = new AliITSInitGeometry("AliITSvPPRasymmFMD",
  86                                                           2);
  87     AliITSgeom *geom = initgeom->CreateAliITSgeom();
  88     delete initgeom; // once created, do not need initgeom any more.
  89     AliITSLoader* loader = static_cast<AliITSLoader*>
  90         (runLoader->GetLoader("ITSLoader"));
  91     if (!loader) {
  92         Error("Init", "ITS loader not found");
  93         return;
  94     }
  95     loader->SetITSgeom(geom);
  96     return;
  97 }
  98 //_____________________________________________________________________________
  99 void AliITSReconstructor::Reconstruct(AliRunLoader* runLoader) const
 100 {
 101 // reconstruct clusters
 102
 103
 104   AliITSLoader* loader = static_cast<AliITSLoader*>(runLoader->GetLoader("ITSLoader"));
 105   if (!loader) {
 106     Error("Reconstruct", "ITS loader not found");
 107     return;
 108   }
 109   AliITSDetTypeRec* rec = new AliITSDetTypeRec(loader);
 110   rec->SetDefaults();
 111
 112   loader->LoadRecPoints("recreate");
 113   loader->LoadDigits("read");
 114   runLoader->LoadKinematics();
 115   TString option = GetOption();
 116   Bool_t clusfinder=kTRUE;   // Default: V2 cluster finder
 117   if(option.Contains("OrigCF"))clusfinder=kFALSE;
 118
 119   Int_t nEvents = runLoader->GetNumberOfEvents();
 120
 121   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
 122     runLoader->GetEvent(iEvent);
 123     if(loader->TreeR()==0x0) loader->MakeTree("R");
 124     rec->MakeBranch("R");
 125     rec->SetTreeAddress();
 126     rec->DigitsToRecPoints(iEvent,0,"All",clusfinder);
 127   }
 128
 129   loader->UnloadRecPoints();
 130   loader->UnloadDigits();
 131   runLoader->UnloadKinematics();
 132 }
 133
 134 //_________________________________________________________________
 135 void AliITSReconstructor::Reconstruct(AliRunLoader* runLoader,
 136                                       AliRawReader* rawReader) const
 137 {
 138 // reconstruct clusters
 139
 140
 141   AliITSLoader* loader = static_cast<AliITSLoader*>(runLoader->GetLoader("ITSLoader"));
 142   if (!loader) {
 143     Error("Reconstruct", "ITS loader not found");
 144     return;
 145   }
 146
 147   AliITSDetTypeRec* rec = new AliITSDetTypeRec(loader);
 148   rec->SetDefaults();
 149   rec->SetDefaultClusterFindersV2(kTRUE);
 150
 151   loader->LoadRecPoints("recreate");
 152
 153   Int_t iEvent = 0;
 154
 155   while(rawReader->NextEvent()) {
 156     runLoader->GetEvent(iEvent++);
 157     if(loader->TreeR()==0x0) loader->MakeTree("R");
 158     rec->DigitsToRecPoints(rawReader);
 159   }
 160
 161   loader->UnloadRecPoints();
 162 }
 163
 164 //_____________________________________________________________________________
 165 AliTracker* AliITSReconstructor::CreateTracker(AliRunLoader* runLoader)const
 166 {
 167 // create a ITS tracker
 168
 169
 170   AliITSgeom* geom = GetITSgeom(runLoader);
 171   TString selectedTracker = GetOption();
 172   AliTracker* tracker;
 173   if (selectedTracker.Contains("MI")) {
 174     tracker = new AliITStrackerMI(geom);
 175   }
 176   else {
 177     tracker =  new AliITStrackerSA(geom);  // inherits from AliITStrackerMI
 178   }
 179
 180   TString selectedPIDmethod = GetOption();
 181   AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader");
 182   if (!loader) {
 183     Error("CreateTracker", "ITS loader not found");
 184   }
 185   if(selectedPIDmethod.Contains("LandauFitPID")){
 186     loader->AdoptITSpid(new AliITSpidESD2((AliITStrackerMI*)tracker,loader));
 187   }
 188   else{
 189     Double_t parITS[] = {34., 0.15, 10.};
 190     loader->AdoptITSpid(new AliITSpidESD1(parITS));
 191   }
 192   return tracker;
 193
 194 }
 195
 196 //_____________________________________________________________________________
 197 AliVertexer* AliITSReconstructor::CreateVertexer(AliRunLoader* /*runLoader*/) const
 198 {
 199 // create a ITS vertexer
 200
 201   TString selectedVertexer = GetOption();
 202   if(selectedVertexer.Contains("ions") || selectedVertexer.Contains("IONS")){
 203     Info("CreateVertexer","a AliITSVertexerIons object has been selected\n");
 204     return new AliITSVertexerIons("null");
 205   }
 206   if(selectedVertexer.Contains("smear") || selectedVertexer.Contains("SMEAR")){
 207     Double_t smear[3]={0.005,0.005,0.01};
 208     Info("CreateVertexer","a AliITSVertexerFast object has been selected\n");
 209     return new AliITSVertexerFast(smear);
 210   }
 211   if(selectedVertexer.Contains("ppz") || selectedVertexer.Contains("PPZ")){
 212     Info("CreateVertexer","a AliITSVertexerPPZ object has been selected\n");
 213     return new AliITSVertexerPPZ("null");
 214   }
 215   // by default an AliITSVertexerZ object is instatiated
 216   Info("CreateVertexer","a AliITSVertexerZ object has been selected\n");
 217   return new AliITSVertexerZ("null");
 218 }
 219
 220 //_____________________________________________________________________________
 221 void AliITSReconstructor::FillESD(AliRunLoader* runLoader,
 222                                   AliESD* esd) const
 223 {
 224 // make PID, find V0s and cascade
 225   AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader");
 226   AliITSpidESD *pidESD = 0;
 227   TString selectedPIDmethod = GetOption();
 228   if(selectedPIDmethod.Contains("LandauFitPID")){
 229     Info("FillESD","ITS LandauFitPID option has been selected\n");
 230     pidESD=loader->GetITSpid();
 231   }
 232   else{
 233     Info("FillESD","ITS default PID\n");
 234     pidESD=loader->GetITSpid();
 235   }
 236   if(pidESD!=0){
 237     pidESD->MakePID(esd);
 238   }
 239   else {
 240     Error("FillESD","!! cannot do the PID !!\n");
 241   }
 242   // V0 finding
 243   Double_t cuts[]={33,  // max. allowed chi2
 244                    0.16,// min. allowed negative daughter's impact parameter
 245                    0.05,// min. allowed positive daughter's impact parameter
 246                    0.08,// max. allowed DCA between the daughter tracks
 247                    0.99,// max. allowed cosine of V0's pointing angle
 248                    0.9,  // min. radius of the fiducial volume
 249                    2.9   // max. radius of the fiducial volume
 250   };
 251   AliV0vertexer vtxer(cuts);
 252   Double_t vtx[3], cvtx[6];
 253   esd->GetVertex()->GetXYZ(vtx);
 254   esd->GetVertex()->GetSigmaXYZ(cvtx);
 255   vtxer.SetVertex(vtx);
 256   vtxer.Tracks2V0vertices(esd);
 257
 258   // cascade finding
 259   Double_t cts[]={33.,    // max. allowed chi2
 260                   0.05,   // min. allowed V0 impact parameter
 261                   0.008,  // window around the Lambda mass
 262                   0.035,  // min. allowed bachelor's impact parameter
 263                   0.10,   // max. allowed DCA between a V0 and a track
 264                   0.9985, //max. allowed cosine of the cascade pointing angle
 265                   0.9,    // min. radius of the fiducial volume
 266                   2.9     // max. radius of the fiducial volume
 267   };
 268   AliCascadeVertexer cvtxer=AliCascadeVertexer(cts);
 269   cvtxer.SetVertex(vtx);
 270   cvtxer.V0sTracks2CascadeVertices(esd);
 271 }
 272
 273
 274 //_____________________________________________________________________________
 275 AliITSgeom* AliITSReconstructor::GetITSgeom(AliRunLoader* runLoader) const
 276 {
 277 // get the ITS geometry
 278
 279   if (!runLoader->GetAliRun()) runLoader->LoadgAlice();
 280   if (!runLoader->GetAliRun()) {
 281     Error("GetITSgeom", "couldn't get AliRun object");
 282     return NULL;
 283   }
 284   AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader");
 285   AliITSgeom* geom = (AliITSgeom*)loader->GetITSgeom();
 286   if(!geom){
 287     Error("GetITSgeom","no ITS geometry available");
 288     return NULL;
 289   }
 290
 291   return geom;
 292 }