Correct formula for phi, in case of current local X <0, in GetLocalXat and GetPhiZat
[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 "AliITSVertexerFixed.h"
37 #include "AliITSVertexer3D.h"
38 #include "AliITSVertexerZ.h"
39 #include "AliITSVertexerCosmics.h"
40 #include "AliESDEvent.h"
41 #include "AliITSpidESD.h"
42 #include "AliITSpidESD1.h"
43 #include "AliITSpidESD2.h"
44 #include "AliITSInitGeometry.h"
45 #include "AliITSTrackleterSPDEff.h"
46
47
48 ClassImp(AliITSReconstructor)
49
50 //___________________________________________________________________________
51 AliITSReconstructor::AliITSReconstructor() : AliReconstructor(),
52 fItsPID(0),
53 fDetTypeRec(0)
54 {
55   // Default constructor
56 }
57  //___________________________________________________________________________
58 AliITSReconstructor::~AliITSReconstructor(){
59 // destructor
60   delete fItsPID;
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     return;
100 }
101
102 //_____________________________________________________________________________
103 void AliITSReconstructor::Reconstruct(TTree *digitsTree, TTree *clustersTree) const
104 {
105 // reconstruct clusters
106
107   Int_t cluFindOpt = GetRecoParam()->GetClusterFinder();
108
109   fDetTypeRec->SetTreeAddressD(digitsTree);
110   fDetTypeRec->MakeBranch(clustersTree,"R");
111   fDetTypeRec->SetTreeAddressR(clustersTree);
112   fDetTypeRec->DigitsToRecPoints(digitsTree,clustersTree,0,"All",cluFindOpt);    
113 }
114
115 //_________________________________________________________________
116 void AliITSReconstructor::Reconstruct(AliRawReader* rawReader, TTree *clustersTree) const
117 {
118   // reconstruct clusters from raw data
119  
120   fDetTypeRec->SetDefaultClusterFindersV2(kTRUE);
121   fDetTypeRec->DigitsToRecPoints(rawReader,clustersTree);
122 }
123 //_____________________________________________________________________________
124 AliTracker* AliITSReconstructor::CreateTrackleter() const
125 {
126 // create the SPD trackeleter (for SPD PlaneEfficiency evaluation)
127   if(!GetRecoParam()->GetComputePlaneEff() && !GetRecoParam()->GetUseTrackletsPlaneEff()) return NULL;
128   //Int_t trackerOpt = GetRecoParam()->GetTracker();
129   AliTracker* trackleter;
130   trackleter = new AliITSTrackleterSPDEff();
131   AliITSTrackleterSPDEff *spdtrackleter=(AliITSTrackleterSPDEff*)trackleter;
132   // here set cuts (from RecoParam)
133   if(GetRecoParam()->GetBkgTrackletsPlaneEff()) spdtrackleter->SetReflectClusterAroundZAxisForLayer(1,kTRUE);
134   if(GetRecoParam()->GetMCTrackletsPlaneEff()) spdtrackleter->SetMC();
135   spdtrackleter->SetHistOn();
136   spdtrackleter->SetPhiWindow(GetRecoParam()->GetTrackleterPhiWindow());
137   spdtrackleter->SetZetaWindow(GetRecoParam()->GetTrackleterZetaWindow());
138   spdtrackleter->SetPhiWindowL1(GetRecoParam()->GetTrackleterPhiWindowL1());
139   spdtrackleter->SetZetaWindowL1(GetRecoParam()->GetTrackleterZetaWindowL1());
140   if(GetRecoParam()->GetUpdateOncePerEventPlaneEff()) spdtrackleter->SetUpdateOncePerEventPlaneEff();
141   //spdtrackleter->(GetRecoParam()->GetMinContVtxPlaneEff()); // to be implemented
142   return trackleter;
143 }
144
145 //_____________________________________________________________________________
146 AliTracker* AliITSReconstructor::CreateTracker() const
147 {
148 // create a ITS tracker
149
150   Int_t trackerOpt = GetRecoParam()->GetTracker();
151   AliTracker* tracker;    
152   if (trackerOpt==1) {
153     tracker = new AliITStrackerMI(0);
154     AliITStrackerMI *mit=(AliITStrackerMI*)tracker;
155     mit->SetDetTypeRec(fDetTypeRec);
156   }  
157   else if (trackerOpt==2) {
158     tracker = new AliITStrackerV2(0);
159   }
160   else {
161     tracker =  new AliITStrackerSA(0);  // inherits from AliITStrackerMI
162     AliITStrackerSA *sat=(AliITStrackerSA*)tracker;
163     sat->SetDetTypeRec(fDetTypeRec);
164     if(GetRecoParam()->GetTrackerSAOnly()) sat->SetSAFlag(kTRUE);
165     if(sat->GetSAFlag())AliDebug(1,"Tracking Performed in ITS only\n");
166         if(GetRecoParam()->GetInwardFindingSA()){
167        sat->SetInwardFinding();
168        sat->SetInnerStartLayer(GetRecoParam()->GetInnerStartLayerSA());
169         }else{
170       sat->SetOutwardFinding();
171       sat->SetOuterStartLayer(GetRecoParam()->GetOuterStartLayerSA());
172         }
173     sat->SetMinNPoints(GetRecoParam()->GetMinNPointsSA());
174   }
175
176   Int_t pidOpt = GetRecoParam()->GetPID();
177
178   AliITSReconstructor* nc = const_cast<AliITSReconstructor*>(this);
179   if(pidOpt==1){
180     Info("FillESD","ITS LandauFitPID option has been selected\n");
181     nc->fItsPID = new AliITSpidESD2((AliITStrackerMI*)tracker);
182   }
183   else{
184     Info("FillESD","ITS default PID\n");
185     Double_t parITS[] = {79.,0.13, 5.}; //IB: this is  "pp tuning"
186     nc->fItsPID = new AliITSpidESD1(parITS);
187   }
188  
189   return tracker;
190   
191 }
192
193 //_____________________________________________________________________________
194 AliVertexer* AliITSReconstructor::CreateVertexer() const
195 {
196 // create a ITS vertexer
197
198   AliITSVertexer *vptr;
199   Int_t vtxOpt = GetRecoParam()->GetVertexer();
200   if(vtxOpt==3){
201     Info("CreateVertexer","a AliITSVertexerIons object has been selected\n");
202     vptr =  new AliITSVertexerIons();
203   }
204   else if(vtxOpt==4){
205     Double_t smear[3]={GetRecoParam()->GetVertexerFastSmearX(),
206                        GetRecoParam()->GetVertexerFastSmearY(),
207                        GetRecoParam()->GetVertexerFastSmearZ()};
208     Info("CreateVertexer","a AliITSVertexerFast object has been selected\n"); 
209     vptr = new AliITSVertexerFast(smear);
210   }
211   else if(vtxOpt==1){
212     Info("CreateVertexer","a AliITSVertexerZ object has been selected\n");
213     vptr =  new AliITSVertexerZ();
214   }
215   else if(vtxOpt==2){
216     Info("CreateVertexer","a AliITSVertexerCosmics object has been selected\n");
217     vptr = new AliITSVertexerCosmics();
218   }
219   else if(vtxOpt==5){ 
220     Info("CreateVertexer","vertex is fixed in the position of the TDI\n");
221     vptr = new AliITSVertexerFixed("TDI");
222   }
223   else if(vtxOpt==6){ 
224     Info("CreateVertexer","vertex is fixed in the position of the TED\n");
225     vptr = new AliITSVertexerFixed("TED");
226   }
227   else {
228   // by default an AliITSVertexer3D object is instatiated
229     Info("CreateVertexer","a AliITSVertexer3D object has been selected\n");
230     AliITSVertexer3D*  vtxr = new AliITSVertexer3D();
231     Float_t dzw=GetRecoParam()->GetVertexer3DWideFiducialRegionZ();
232     Float_t drw=GetRecoParam()->GetVertexer3DWideFiducialRegionR();
233     vtxr->SetWideFiducialRegion(dzw,drw);
234     Float_t dzn=GetRecoParam()->GetVertexer3DNarrowFiducialRegionZ();
235     Float_t drn=GetRecoParam()->GetVertexer3DNarrowFiducialRegionR();
236     vtxr->SetNarrowFiducialRegion(dzn,drn);
237     Float_t dphil=GetRecoParam()->GetVertexer3DLooseDeltaPhiCut();
238     Float_t dphit=GetRecoParam()->GetVertexer3DTightDeltaPhiCut();
239     vtxr->SetDeltaPhiCuts(dphil,dphit);
240     Float_t dcacut=GetRecoParam()->GetVertexer3DDCACut();
241     vtxr->SetDCACut(dcacut);
242     vptr = vtxr;
243   }
244   vptr->SetDetTypeRec(fDetTypeRec);
245   return vptr;
246 }
247
248 //_____________________________________________________________________________
249 void AliITSReconstructor::FillESD(TTree * /*digitsTree*/, TTree *clustersTree, 
250                                   AliESDEvent* esd) const
251 {
252 // make PID, find V0s and cascade
253   if(fItsPID!=0) {
254     Int_t pidOpt = GetRecoParam()->GetPID();
255     if(pidOpt==1){
256       fItsPID->MakePID(clustersTree,esd);
257     }else{
258       fItsPID->MakePID(esd);
259     }
260   }
261   else {
262     Error("FillESD","!! cannot do the PID !!\n");
263   }
264 }