]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ITS/UPGRADE/AliITStrackerUpgrade.h
ITS-upgrade simulation/reconstruction code (C. Terrevoli, A. Mastroserio)
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITStrackerUpgrade.h
diff --git a/ITS/UPGRADE/AliITStrackerUpgrade.h b/ITS/UPGRADE/AliITStrackerUpgrade.h
new file mode 100644 (file)
index 0000000..6c38324
--- /dev/null
@@ -0,0 +1,160 @@
+#ifndef ALIITSTRACKERUPGRADE_H
+#define ALIITSTRACKERUPGRADE_H 
+
+
+
+#include "AliITStrackerMI.h"
+#include "AliITSlayerUpgrade.h"
+#include "AliITStrackMI.h"
+
+/* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+/* $Id$ */
+////////////////////////////////////////////////////
+//  ITS Upgrade Stand alone tracker class         //
+//  Authors: A.Mastroserio C.Terrevoli            //
+//  e-mail:annalisa.mastroserio@cern.ch           //
+//         cristina.terrevoli@ba.infn.it          //
+//  tracks are saved as AliITStrackV2 objects     //
+////////////////////////////////////////////////////
+
+#include <TClonesArray.h>
+
+class AliITSclusterTable;
+class AliITStrackSA;
+class AliESDVertex;
+class AliESDEvent;
+class AliITSVertexer;
+class TTree;
+class TArrayD;
+
+class AliITStrackerUpgrade : public AliITStrackerMI {
+
+
+ public:
+
+  AliITStrackerUpgrade();
+  AliITStrackerUpgrade(const AliITStrackerUpgrade& tracker);
+  AliITStrackerUpgrade& operator=(const AliITStrackerUpgrade& source);
+  virtual ~AliITStrackerUpgrade();  
+ virtual Int_t Clusters2Tracks(AliESDEvent *event);  
+ Int_t FindTracks(AliESDEvent* event,Bool_t useAllClusters=kFALSE);
+  Int_t PropagateBack(AliESDEvent *event);
+  Int_t CorrectForPipeMaterial(AliITStrackMI *t, TString direction="inward");  
+  Int_t RefitInward(AliESDEvent *event);//{return 0;}
+  AliCluster *GetCluster(Int_t index) const;
+
+  AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
+  void StoreTrack(AliITStrackV2 *t,AliESDEvent *event, Bool_t pureSA) /*const*/; 
+  Int_t FindTrackLowChiSquare() const;
+  Int_t LoadClusters(TTree *clusTree);
+  void SetVertex(AliESDVertex *vtx){fVert = vtx;}
+  void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
+
+  void SetOutwardFinding() {fInwardFlag=kFALSE;}
+  void SetInwardFinding() {fInwardFlag=kTRUE;}
+  void SetOuterStartLayer(Int_t osl = 0) {
+    if(osl>(fNLayer-2)) AliWarning("Minimum Number of layers is 2. OuterStartLayer set to Nlayers-2");
+    fOuterStartLayer = TMath::Min(fNLayer-2,osl);
+  }
+  Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
+  void SetInnerStartLayer(Int_t isl = 5) {
+    if(isl<1) AliWarning("Minimum Number of layers is 2. InnerStartLayer set to 1");
+    fInnerStartLayer = TMath::Max(1,isl);
+  }
+  Int_t GetInnerStartLayer() const {return fInnerStartLayer;}
+  void SetSAFlag(Bool_t fl){fITSStandAlone=fl;}  // StandAlone flag setter
+  Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag gett
+  void SetFixedWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
+  void SetCalculatedWindowSizes(Int_t n=10, Float_t phimin=0.002, Float_t phimax=0.0145, Float_t lambdamin=0.003, Float_t lambdamax=0.008);
+  void UnloadClusters();
+  void SetMinNPoints(Int_t np){fMinNPoints=np;}
+  Int_t GetMinNPoints() const {return fMinNPoints;}
+  void SetMinimumChargeSDDSSD(Float_t minq=0.){fMinQ=minq;}
+  enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
+
+ protected:
+
+  //Initialization
+  void Init();
+  void  CreateLayers(Int_t iEvent);
+  void ResetForFinding();
+  void UpdatePoints();
+  Bool_t SetFirstPoint(Int_t lay, Int_t clu, Double_t* primaryVertex);
+  static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
+                     Double_t x3,Double_t y3);
+
+  Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp); 
+
+  static Int_t   FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2, 
+                           Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
+  static Int_t   FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2, 
+                       Float_t x3, Float_t y3,Float_t& a, Float_t& b, 
+                       Float_t& c);
+  static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
+  static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5, 
+              Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
+              Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16, 
+              Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
+  Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, 
+                       AliITStrackSA* trs,Double_t zvertex,Int_t flagp); 
+
+  void GetCoorAngles(AliITSRecPoint* cl,Double_t &phi,Double_t &lambda,Double_t &x,Double_t &y,Double_t &z,Double_t* vertex);
+  void GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz);
+  AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
+  void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
+  Bool_t RefitAtBase(Double_t x, AliITStrackMI *track,
+                 const Int_t *clusters);
+  Int_t UpdateMI(AliITStrackMI* track, const AliITSRecPoint* cl,Double_t chi2,Int_t layer) const;
+ Int_t CorrectForLayerMaterial(AliITStrackMI *t, Int_t layerindex, Double_t oldGlobXYZ[3], TString direction="inward");
+  Double_t GetPredictedChi2MI(AliITStrackMI* track, const AliITSRecPoint *cluster,Int_t layer);
+  static Int_t GetError(Int_t layer,const AliITSRecPoint*cl,
+                        Float_t tgl,Float_t tgphitr,Float_t expQ,
+                        Float_t &erry,Float_t &errz,Float_t &covyz,
+                        Bool_t addMisalErr=kTRUE);
+  
+  static Int_t GetErrorOrigRecPoint(const AliITSRecPoint*cl,
+                                    Float_t &erry,Float_t &errz,Float_t &covyz);
+  static void GetNTeor(Int_t layer,const AliITSRecPoint* cl,
+                       Float_t tgl,Float_t tgphitr,
+                       Float_t &ny,Float_t &nz);
+
+  Int_t fNLayer;//number of layer in ITSUpgrade
+  Double_t fPhiEstimate; //Estimation of phi angle on next layer
+  Bool_t fITSStandAlone; //Tracking is performed in the ITS alone if kTRUE
+  Float_t fPoint1[2];   //! coord. of 1-st point to evaluate the curvature
+  Float_t fPoint2[2];   //! coord. of 2-nd point to evaluate the curvature
+  Float_t fPoint3[2];   //! coord. of 3-rd point to evaluate the curvature
+  Float_t fPointc[2];   //! current point coord (for curvature eval.)
+  Double_t fLambdac;    //! current value of the Lambda angle in the window
+  Double_t fPhic;       //! current value of the Phi angle in the window
+  Float_t fCoef1;       //! param. of the equation of the circ. approx a layer
+  Float_t fCoef2;       //! param. of the equation of the circ. approx a layer
+  Float_t fCoef3;       //! param. of the equation of the circ. approx a layer
+  Int_t fNloop;         //  Number of iterqations on phi and lambda windows
+  Double_t *fPhiWin;    // phi window sizes
+  Double_t *fLambdaWin; // lambda window sizes
+  AliESDVertex *fVert;        //! primary vertex
+  AliITSVertexer *fVertexer;  //! vertexer 
+  TClonesArray *fListOfTracks;   //! container for found tracks 
+  TClonesArray *fListOfSATracks; //! container for found SA tracks 
+  TTree *fITSclusters;        //! pointer to ITS tree of clusters
+  Bool_t fInwardFlag;       // set to kTRUE for inward track finding
+  Int_t fOuterStartLayer;     // Outward search for tracks with <6 points: outer layer to start from
+  Int_t fInnerStartLayer;     // Inward search for tracks with <6 points: inner layer to start from
+  Int_t fMinNPoints;        // minimum number of clusters for a track
+  Float_t fMinQ;              // lower cut on cluster charge (SDD and SSD)
+  AliITSlayerUpgrade** fLayers;
+
+  TClonesArray** fCluLayer; //! array with clusters 
+  TClonesArray** fCluCoord; //! array with cluster info
+
+  ClassDef(AliITStrackerUpgrade,1)
+};
+
+#endif
+
+