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