037c02124555bef1f44c66e1a953226cae1f7c31
[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::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 }