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 "AliRawReader.h"
  28 #include "AliITSDetTypeRec.h"
  29 #include "AliITSgeom.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 "AliITSVertexer3D.h"
  37 #include "AliITSVertexerZ.h"
  38 #include "AliITSVertexerCosmics.h"
  39 #include "AliESDEvent.h"
  40 #include "AliITSpidESD.h"
  41 #include "AliITSpidESD1.h"
  42 #include "AliITSpidESD2.h"
  43 #include "AliITSInitGeometry.h"
  44
  45
  46 ClassImp(AliITSReconstructor)
  47 AliITSRecoParam *AliITSReconstructor::fgRecoParamDefault = AliITSRecoParam::GetHighFluxParam();
  48 AliITSRecoParam *AliITSReconstructor::fgRecoParam =0;  // reconstruction parameters
  49
  50 //___________________________________________________________________________
  51 AliITSReconstructor::AliITSReconstructor() : AliReconstructor(),
  52 fItsPID(0),
  53 fDetTypeRec(0)
  54 {
  55   // Default constructor
  56   if (!fgRecoParam) {
  57     AliWarning("Using default reconstruction parameters");
  58     fgRecoParam = fgRecoParamDefault;
  59   }
  60 }
  61  //___________________________________________________________________________
  62 AliITSReconstructor::~AliITSReconstructor(){
  63 // destructor
  64   delete fItsPID;
  65   if(!fgRecoParam && (fgRecoParam != fgRecoParamDefault)) delete fgRecoParam;
  66   if(fDetTypeRec) delete fDetTypeRec;
  67 }
  68 //______________________________________________________________________
  69 AliITSReconstructor::AliITSReconstructor(const AliITSReconstructor &ob) :AliReconstructor(ob),
  70 fItsPID(ob.fItsPID),
  71 fDetTypeRec(ob.fDetTypeRec)
  72
  73 {
  74   // Copy constructor
  75 }
  76
  77 //______________________________________________________________________
  78 AliITSReconstructor& AliITSReconstructor::operator=(const AliITSReconstructor&  ob ){
  79   // Assignment operator
  80   this->~AliITSReconstructor();
  81   new(this) AliITSReconstructor(ob);
  82   return *this;
  83 }
  84
  85 //______________________________________________________________________
  86 void AliITSReconstructor::Init() {
  87     // Initalize this constructor bet getting/creating the objects
  88     // nesseary for a proper ITS reconstruction.
  89     // Inputs:
  90     //   none.
  91     // Output:
  92     //   none.
  93     // Return:
  94     //   none.
  95
  96     AliITSInitGeometry initgeom;
  97     AliITSgeom *geom = initgeom.CreateAliITSgeom();
  98     AliInfo(Form("Geometry name: %s",(initgeom.GetGeometryName()).Data()));
  99
 100     fDetTypeRec = new AliITSDetTypeRec();
 101     fDetTypeRec->SetITSgeom(geom);
 102     fDetTypeRec->SetDefaults();
 103
 104     return;
 105 }
 106
 107 //_____________________________________________________________________________
 108 void AliITSReconstructor::Reconstruct(TTree *digitsTree, TTree *clustersTree) const
 109 {
 110 // reconstruct clusters
 111
 112   TString option = GetOption();
 113   Bool_t clusfinder=kTRUE;   // Default: V2 cluster finder
 114   if(option.Contains("OrigCF"))clusfinder=kFALSE;
 115
 116   fDetTypeRec->SetTreeAddressD(digitsTree);
 117   fDetTypeRec->MakeBranch(clustersTree,"R");
 118   fDetTypeRec->SetTreeAddressR(clustersTree);
 119   fDetTypeRec->DigitsToRecPoints(digitsTree,clustersTree,0,"All",clusfinder);
 120 }
 121
 122 //_________________________________________________________________
 123 void AliITSReconstructor::Reconstruct(AliRawReader* rawReader, TTree *clustersTree) const
 124 {
 125   // reconstruct clusters from raw data
 126
 127   fDetTypeRec->SetDefaultClusterFindersV2(kTRUE);
 128   fDetTypeRec->DigitsToRecPoints(rawReader,clustersTree);
 129 }
 130
 131 //_____________________________________________________________________________
 132 AliTracker* AliITSReconstructor::CreateTracker() const
 133 {
 134 // create a ITS tracker
 135
 136
 137   TString selectedTracker = GetOption();
 138   AliTracker* tracker;
 139   if (selectedTracker.Contains("MI")) {
 140     tracker = new AliITStrackerMI(0);
 141   }
 142   else if (selectedTracker.Contains("V2")) {
 143     tracker = new AliITStrackerV2(0);
 144   }
 145   else {
 146     tracker =  new AliITStrackerSA(0);  // inherits from AliITStrackerMI
 147     AliITStrackerSA *sat=(AliITStrackerSA*)tracker;
 148     if(selectedTracker.Contains("onlyITS"))sat->SetSAFlag(kTRUE);
 149     if(sat->GetSAFlag())AliDebug(1,"Tracking Performed in ITS only\n");
 150     if(selectedTracker.Contains("cosmics")||selectedTracker.Contains("COSMICS"))
 151       sat->SetOuterStartLayer(AliITSgeomTGeo::GetNLayers()-2);
 152   }
 153
 154   TString selectedPIDmethod = GetOption();
 155   AliITSReconstructor* nc = const_cast<AliITSReconstructor*>(this);
 156   if(selectedPIDmethod.Contains("LandauFitPID")){
 157     Info("FillESD","ITS LandauFitPID option has been selected\n");
 158     nc->fItsPID = new AliITSpidESD2((AliITStrackerMI*)tracker);
 159   }
 160   else{
 161     Info("FillESD","ITS default PID\n");
 162     Double_t parITS[] = {79.,0.13, 5.}; //IB: this is  "pp tuning"
 163     nc->fItsPID = new AliITSpidESD1(parITS);
 164   }
 165
 166   return tracker;
 167
 168 }
 169
 170 //_____________________________________________________________________________
 171 AliVertexer* AliITSReconstructor::CreateVertexer() const
 172 {
 173 // create a ITS vertexer
 174
 175   TString selectedVertexer = GetOption();
 176   if(selectedVertexer.Contains("ions") || selectedVertexer.Contains("IONS")){
 177     Info("CreateVertexer","a AliITSVertexerIons object has been selected\n");
 178     return new AliITSVertexerIons("null");
 179   }
 180   if(selectedVertexer.Contains("smear") || selectedVertexer.Contains("SMEAR")){
 181     Double_t smear[3]={0.005,0.005,0.01};
 182     Info("CreateVertexer","a AliITSVertexerFast object has been selected\n");
 183     return new AliITSVertexerFast(smear);
 184   }
 185   if(selectedVertexer.Contains("3d") || selectedVertexer.Contains("3D")){
 186     Info("CreateVertexer","a AliITSVertexer3D object has been selected\n");
 187     return new AliITSVertexer3D("null");
 188   }
 189   if(selectedVertexer.Contains("cosmics") || selectedVertexer.Contains("COSMICS")){
 190     Info("CreateVertexer","a AliITSVertexerCosmics object has been selected\n");
 191     return new AliITSVertexerCosmics();
 192   }
 193   // by default an AliITSVertexerZ object is instatiated
 194   Info("CreateVertexer","a AliITSVertexerZ object has been selected\n");
 195   return new AliITSVertexerZ("null");
 196 }
 197
 198 //_____________________________________________________________________________
 199 void AliITSReconstructor::FillESD(TTree * /*digitsTree*/, TTree *clustersTree,
 200                                   AliESDEvent* esd) const
 201 {
 202 // make PID, find V0s and cascade
 203   if(fItsPID!=0) {
 204     TString selectedPIDmethod = GetOption();
 205     if(selectedPIDmethod.Contains("LandauFitPID")){
 206       fItsPID->MakePID(clustersTree,esd);
 207     }
 208     else{
 209       fItsPID->MakePID(esd);
 210     }
 211   }
 212   else {
 213     Error("FillESD","!! cannot do the PID !!\n");
 214   }
 215 }