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 "AliESDEvent.h"
  29 #include "AliESDpid.h"
  30 #include "AliITSgeom.h"
  31 #include "AliITSLoader.h"
  32 #include "AliITStrackerMI.h"
  33 #include "AliITStrackerV2.h"
  34 #include "AliITStrackerSA.h"
  35 #include "AliITSVertexerFast.h"
  36 #include "AliITSVertexerFixed.h"
  37 #include "AliITSVertexer3D.h"
  38 #include "AliITSVertexerZ.h"
  39 #include "AliITSVertexerCosmics.h"
  40 #include "AliITSInitGeometry.h"
  41 #include "AliITSTrackleterSPDEff.h"
  42 #include "AliITSMultReconstructor.h"
  43
  44 ClassImp(AliITSReconstructor)
  45
  46 //___________________________________________________________________________
  47 AliITSReconstructor::AliITSReconstructor() : AliReconstructor(),
  48 fDetTypeRec(0)
  49 {
  50   // Default constructor
  51 }
  52  //___________________________________________________________________________
  53 AliITSReconstructor::~AliITSReconstructor(){
  54 // destructor
  55   if(fDetTypeRec) delete fDetTypeRec;
  56 }
  57 //____________________________________________________________________________
  58 void AliITSReconstructor::GetPidSettings(AliESDpid *ESDpid) {
  59   //
  60   // pass PID settings from AliITSRecoParam to AliESDpid
  61   //
  62   Int_t pidOpt = GetRecoParam()->GetPID();
  63
  64   if(pidOpt==1){
  65     AliDebug(1,"ITS LandauFitPID option has been selected\n");
  66     ESDpid->SetITSPIDmethod(AliESDpid::kITSLikelihood);
  67   }
  68   else{
  69     AliDebug(1,"ITS default PID\n");
  70     ESDpid->SetITSPIDmethod(AliESDpid::kITSTruncMean);
  71   }
  72 }
  73 //______________________________________________________________________
  74 void AliITSReconstructor::Init() {
  75     // Initalize this constructor bet getting/creating the objects
  76     // nesseary for a proper ITS reconstruction.
  77     // Inputs:
  78     //   none.
  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
  88     fDetTypeRec = new AliITSDetTypeRec();
  89     fDetTypeRec->SetITSgeom(geom);
  90     fDetTypeRec->SetDefaults();
  91
  92     return;
  93 }
  94
  95 //_____________________________________________________________________________
  96 void AliITSReconstructor::Reconstruct(TTree *digitsTree, TTree *clustersTree) const
  97 {
  98 // reconstruct clusters
  99
 100   Int_t cluFindOpt = GetRecoParam()->GetClusterFinder();
 101   TString recopt = GetRecoParam()->GetOptReco();
 102   fDetTypeRec->SetTreeAddressD(digitsTree);
 103   fDetTypeRec->MakeBranch(clustersTree,"R");
 104   fDetTypeRec->SetTreeAddressR(clustersTree);
 105   fDetTypeRec->DigitsToRecPoints(digitsTree,clustersTree,0,recopt.Data(),cluFindOpt);
 106 }
 107
 108 //_________________________________________________________________
 109 void AliITSReconstructor::Reconstruct(AliRawReader* rawReader, TTree *clustersTree) const
 110 {
 111   // reconstruct clusters from raw data
 112   TString recopt = GetRecoParam()->GetOptReco();
 113   fDetTypeRec->SetDefaultClusterFindersV2(kTRUE);
 114   fDetTypeRec->DigitsToRecPoints(rawReader,clustersTree,recopt.Data());
 115 }
 116
 117 //_____________________________________________________________________________
 118 AliTrackleter* AliITSReconstructor::CreateMultFinder() const
 119 {
 120   // create the SPD trackeleter for mult. reconstruction
 121   AliITSMultReconstructor* multReco = new AliITSMultReconstructor();
 122   multReco->SetHistOn(kFALSE);
 123   multReco->SetDetTypeRec(fDetTypeRec);
 124   return multReco;
 125 }
 126
 127 //_____________________________________________________________________________
 128 AliTracker* AliITSReconstructor::CreateTrackleter() const
 129 {
 130 // create the SPD trackeleter (for SPD PlaneEfficiency evaluation)
 131   if(!GetRecoParam()->GetComputePlaneEff() || !GetRecoParam()->GetUseTrackletsPlaneEff()) return NULL;
 132   //Int_t trackerOpt = GetRecoParam()->GetTracker();
 133   AliTracker* trackleter;
 134   trackleter = new AliITSTrackleterSPDEff();
 135   AliITSTrackleterSPDEff *spdtrackleter=(AliITSTrackleterSPDEff*)trackleter;
 136   // here set cuts (from RecoParam)
 137   if(GetRecoParam()->GetBkgTrackletsPlaneEff()) spdtrackleter->SetReflectClusterAroundZAxisForLayer(1,kTRUE);
 138   if(GetRecoParam()->GetMCTrackletsPlaneEff()) spdtrackleter->SetMC();
 139   spdtrackleter->SetHistOn();
 140   spdtrackleter->SetPhiWindowL2(GetRecoParam()->GetTrackleterPhiWindowL2());
 141   spdtrackleter->SetZetaWindowL2(GetRecoParam()->GetTrackleterZetaWindowL2());
 142   spdtrackleter->SetPhiWindowL1(GetRecoParam()->GetTrackleterPhiWindowL1());
 143   spdtrackleter->SetZetaWindowL1(GetRecoParam()->GetTrackleterZetaWindowL1());
 144   if(GetRecoParam()->GetUpdateOncePerEventPlaneEff()) spdtrackleter->SetUpdateOncePerEventPlaneEff();
 145   spdtrackleter->SetMinContVtx(GetRecoParam()->GetMinContVtxPlaneEff());
 146   return trackleter;
 147 }
 148
 149 //_____________________________________________________________________________
 150 AliTracker* AliITSReconstructor::CreateTracker() const
 151 {
 152 // create a ITS tracker
 153
 154   Int_t trackerOpt = GetRecoParam()->GetTracker();
 155   AliTracker* tracker;
 156   if (trackerOpt==1) {
 157     tracker = new AliITStrackerMI(0);
 158     AliITStrackerMI *mit=(AliITStrackerMI*)tracker;
 159     mit->SetDetTypeRec(fDetTypeRec);
 160   }
 161   else if (trackerOpt==2) {
 162     tracker = new AliITStrackerV2(0);
 163   }
 164   else {
 165     tracker =  new AliITStrackerSA(0);  // inherits from AliITStrackerMI
 166     AliITStrackerSA *sat=(AliITStrackerSA*)tracker;
 167     sat->SetDetTypeRec(fDetTypeRec);
 168     if(GetRecoParam()->GetTrackerSAOnly()) sat->SetSAFlag(kTRUE);
 169     if(sat->GetSAFlag())AliDebug(1,"Tracking Performed in ITS only\n");
 170         if(GetRecoParam()->GetInwardFindingSA()){
 171        sat->SetInwardFinding();
 172        sat->SetInnerStartLayer(GetRecoParam()->GetInnerStartLayerSA());
 173         }else{
 174       sat->SetOutwardFinding();
 175       sat->SetOuterStartLayer(GetRecoParam()->GetOuterStartLayerSA());
 176         }
 177     sat->SetMinNPoints(GetRecoParam()->GetMinNPointsSA());
 178   }
 179
 180   return tracker;
 181
 182 }
 183
 184 //_____________________________________________________________________________
 185 AliVertexer* AliITSReconstructor::CreateVertexer() const
 186 {
 187 // create a ITS vertexer
 188
 189   AliITSVertexer *vptr = NULL;
 190   Int_t vtxOpt = GetRecoParam()->GetVertexer();
 191   if(vtxOpt==3){
 192     AliFatal("Option AliITSVertexerIons is no longer supported");
 193     return vptr;
 194   }
 195   else if(vtxOpt==4){
 196     Double_t smear[3]={GetRecoParam()->GetVertexerFastSmearX(),
 197                        GetRecoParam()->GetVertexerFastSmearY(),
 198                        GetRecoParam()->GetVertexerFastSmearZ()};
 199     AliDebug(1,"AliITSVertexerFast has been selected");
 200     vptr = new AliITSVertexerFast(smear);
 201   }
 202   else if(vtxOpt==1){
 203     AliDebug(1,"AliITSVertexerZ has been selected");
 204     AliITSVertexerZ* vtxrz =  new AliITSVertexerZ();
 205     Int_t pileupAlgo=GetRecoParam()->GetSPDVertexerPileupAlgo();
 206     if(pileupAlgo==3) vtxrz->SetSearchForPileup(kFALSE);
 207     vptr = vtxrz;
 208   }
 209   else if(vtxOpt==2){
 210     AliDebug(1,"AliITSVertexerCosmics has been selected");
 211     vptr = new AliITSVertexerCosmics();
 212   }
 213   else if(vtxOpt==5){
 214     AliDebug(1,"Vertex is fixed in the position of the TDI\n");
 215     vptr = new AliITSVertexerFixed("TDI");
 216   }
 217   else if(vtxOpt==6){
 218     AliDebug(1,"Vertex is fixed in the position of the TED\n");
 219     vptr = new AliITSVertexerFixed("TED");
 220   }
 221   else {
 222   // by default an AliITSVertexer3D object is instatiated
 223     AliITSVertexer3D*  vtxr = new AliITSVertexer3D();
 224     Float_t dzw=GetRecoParam()->GetVertexer3DWideFiducialRegionZ();
 225     Float_t drw=GetRecoParam()->GetVertexer3DWideFiducialRegionR();
 226     vtxr->SetWideFiducialRegion(dzw,drw);
 227     Float_t dzn=GetRecoParam()->GetVertexer3DNarrowFiducialRegionZ();
 228     Float_t drn=GetRecoParam()->GetVertexer3DNarrowFiducialRegionR();
 229     vtxr->SetNarrowFiducialRegion(dzn,drn);
 230     Float_t dphil=GetRecoParam()->GetVertexer3DLooseDeltaPhiCut();
 231     Float_t dphit=GetRecoParam()->GetVertexer3DTightDeltaPhiCut();
 232     vtxr->SetDeltaPhiCuts(dphil,dphit);
 233     Float_t dcacut=GetRecoParam()->GetVertexer3DDCACut();
 234     vtxr->SetDCACut(dcacut);
 235     Int_t pileupAlgo=GetRecoParam()->GetSPDVertexerPileupAlgo();
 236     vtxr->SetPileupAlgo(pileupAlgo);
 237     UChar_t highmultAlgo=GetRecoParam()->GetSPDVertexerHighMultAlgo();
 238     vtxr->SetHighMultAlgo(highmultAlgo);
 239     AliDebug(1,Form("AliITSVertexer3D with pileup algo %d has been selected",pileupAlgo));
 240     vptr = vtxr;
 241   }
 242   vptr->SetDetTypeRec(fDetTypeRec);
 243   return vptr;
 244 }
 245
 246 //_____________________________________________________________________________
 247 void AliITSReconstructor::FillESD(TTree * /*digitsTree*/, TTree * /*clustersTree*/,
 248                                   AliESDEvent* /* esd */) const
 249 {
 250 // make PID, find V0s and cascade
 251 /* Now done in AliESDpid
 252   if(fItsPID!=0) fItsPID->MakePID(esd);
 253   else AliError("!! cannot do the PID !!");
 254 */
 255 }