1 #ifndef ALI_TPC_CORRECTION_H
2 #define ALI_TPC_CORRECTION_H
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
7 ////////////////////////////////////////////////////////////////////////////////
9 // AliTPCCorrection class //
11 // This class provides a general framework to deal with space point //
12 // distortions. An correction class which inherits from here is for example //
13 // AliTPCExBBShape or AliTPCExBTwist //
15 // General functions are (for example): //
16 // CorrectPoint(x,roc) where x is the vector of inital positions in //
17 // cartesian coordinates and roc represents the Read Out chamber number //
18 // according to the offline naming convention. The vector x is overwritten //
19 // with the corrected coordinates. //
21 // An alternative usage would be CorrectPoint(x,roc,dx), which leaves the //
22 // vector x untouched, put returns the distortions via the vector dx //
24 // The class allows "effective Omega Tau" corrections to be shifted to the //
25 // single distortion classes. //
27 // Note: This class is normally used via the class AliTPCComposedCorrection //
29 // date: 27/04/2010 //
30 // Authors: Magnus Mager, Stefan Rossegger, Jim Thomas //
31 ////////////////////////////////////////////////////////////////////////////////
39 class TTreeSRedirector;
40 class AliExternalTrackParam;
46 class AliTPCCorrection : public TNamed {
48 enum CompositionType {kParallel,kQueue};
51 AliTPCCorrection(const char *name,const char *title);
52 virtual ~AliTPCCorrection();
55 // functions to correct a space point
56 void CorrectPoint ( Float_t x[],const Short_t roc);
57 void CorrectPoint (const Float_t x[],const Short_t roc,Float_t xp[]);
58 virtual void GetCorrection(const Float_t x[],const Short_t roc,Float_t dx[]);
60 // functions to distort a space point
61 void DistortPoint ( Float_t x[],const Short_t roc);
62 void DistortPoint (const Float_t x[],const Short_t roc,Float_t xp[]);
63 virtual void GetDistortion(const Float_t x[],const Short_t roc,Float_t dx[]);
65 // initialization and update functions
67 virtual void Update(const TTimeStamp &timeStamp);
69 // convenience functions
70 virtual void Print(Option_t* option="") const;
72 TH2F* CreateHistoDRinXY (Float_t z=10.,Int_t nx=100,Int_t ny=100);
73 TH2F* CreateHistoDRPhiinXY(Float_t z=10.,Int_t nx=100,Int_t nphi=100);
74 TH2F* CreateHistoDRinZR (Float_t phi=0.,Int_t nZ=100,Int_t nR=100);
75 TH2F* CreateHistoDRPhiinZR(Float_t phi=0.,Int_t nZ=100,Int_t nR=100);
76 TTree* CreateDistortionTree(Double_t step=5);
77 static void MakeDistortionMap(THnSparse * his0, TTreeSRedirector *pcstream, const char* hname, Int_t run);
78 // normally called directly in the correction classes which inherit from this class
79 virtual void SetOmegaTauT1T2(Float_t omegaTau,Float_t t1,Float_t t2);
80 AliExternalTrackParam * FitDistortedTrack(AliExternalTrackParam & trackIn, Double_t refX, Int_t dir,TTreeSRedirector *pcstream);
81 void StoreInOCDB(Int_t startRun, Int_t endRun, const char *comment=0);
82 static void MakeTrackDistortionTree(TTree *tinput, Int_t dtype, Int_t ptype, const TObjArray * corrArray, Int_t step=1, Bool_t debug=0);
83 static void MakeLaserDistortionTree(TTree* tree, TObjArray *corrArray, Int_t itype);
85 TH2F* CreateTH2F(const char *name,const char *title,
86 const char *xlabel,const char *ylabel,const char *zlabel,
87 Int_t nbinsx,Double_t xlow,Double_t xup,
88 Int_t nbinsy,Double_t ylow,Double_t yup);
90 static const Double_t fgkTPCZ0; // nominal gating grid position
91 static const Double_t fgkIFCRadius; // Mean Radius of the Inner Field Cage ( 82.43 min, 83.70 max) (cm)
92 static const Double_t fgkOFCRadius; // Mean Radius of the Outer Field Cage (252.55 min, 256.45 max) (cm)
93 static const Double_t fgkZOffSet; // Offset from CE: calculate all distortions closer to CE as if at this point
94 static const Double_t fgkCathodeV; // Cathode Voltage (volts)
95 static const Double_t fgkGG; // Gating Grid voltage (volts)
97 enum {kNR= 92}; // Number of R points in the table for interpolating distortion data
98 enum {kNZ= 270}; // Number of Z points in the table for interpolating distortion data
99 static const Double_t fgkRList[kNR]; // points in the radial direction (for the lookup table)
100 static const Double_t fgkZList[kNZ]; // points in the z direction (for the lookup table)
102 // Simple Interpolation functions: e.g. with tricubic interpolation (not yet in TH3)
103 Int_t fJLow; // variable to help in the interpolation
104 Int_t fKLow; // variable to help in the interpolation
105 void Interpolate2DEdistortion( const Int_t order, const Double_t r, const Double_t z,
106 const Double_t er[kNZ][kNR], Double_t &erValue );
107 Double_t Interpolate( const Double_t xArray[], const Double_t yArray[],
108 const Int_t order, const Double_t x );
109 void Search( const Int_t n, const Double_t xArray[], const Double_t x, Int_t &low );
110 virtual Int_t IsPowerOfTwo ( Int_t i ) const ;
112 // Algorithms to solve the laplace or possion equation
113 void PoissonRelaxation2D(TMatrixD &arrayV, const TMatrixD &chargeDensity, TMatrixD &arrayErOverEz, const Int_t rows, const Int_t columns, const Int_t iterations );
117 Double_t fT1; // tensor term of wt - T1
118 Double_t fT2; // tensor term of wt - T2
120 ClassDef(AliTPCCorrection,2);