#ifndef ALI_TPC_CORRECTION_H
#define ALI_TPC_CORRECTION_H
#include <TNamed.h>
#include "TMatrixD.h"
#include "TMatrixF.h"
class TH2F;
class TTimeStamp;
class TCollection;
class TTreeSRedirector;
class AliExternalTrackParam;
class TTree;
class THnSparse;
class AliESDVertex;
class AliTPCCorrection : public TNamed {
public:
enum CompositionType {kParallel,kQueue};
AliTPCCorrection();
AliTPCCorrection(const char *name,const char *title);
virtual ~AliTPCCorrection();
virtual Bool_t AddCorrectionCompact(AliTPCCorrection* corr, Double_t weight);
void CorrectPoint ( Float_t x[], Short_t roc);
void CorrectPointLocal(Float_t x[], Short_t roc);
void CorrectPoint (const Float_t x[], Short_t roc,Float_t xp[]);
virtual void GetCorrection(const Float_t x[], Short_t roc,Float_t dx[]);
virtual void GetCorrectionDz(const Float_t x[], Short_t roc,Float_t dx[], Float_t delta);
virtual void GetCorrectionIntegralDz(const Float_t x[], Short_t roc,Float_t dx[], Float_t delta);
void DistortPoint ( Float_t x[], Short_t roc);
void DistortPointLocal(Float_t x[], Short_t roc);
void DistortPoint (const Float_t x[], Short_t roc,Float_t xp[]);
virtual void GetDistortion(const Float_t x[], Short_t roc,Float_t dx[]);
virtual void GetDistortionDz(const Float_t x[], Short_t roc,Float_t dx[], Float_t delta);
virtual void GetDistortionIntegralDz(const Float_t x[], Short_t roc,Float_t dx[], Float_t delta);
virtual void Init();
virtual void Update(const TTimeStamp &timeStamp);
virtual void SetCorrScaleFactor(Float_t ) { ; }
virtual Float_t GetCorrScaleFactor() const { return 1.; }
virtual void Print(Option_t* option="") const;
TH2F* CreateHistoDRinXY (Float_t z=10.,Int_t nx=100,Int_t ny=100);
TH2F* CreateHistoDRPhiinXY(Float_t z=10.,Int_t nx=100,Int_t nphi=100);
TH2F* CreateHistoDZinXY (Float_t z=10.,Int_t nx=100,Int_t ny=100);
TH2F* CreateHistoDRinZR (Float_t phi=0.,Int_t nZ=100,Int_t nR=100);
TH2F* CreateHistoDRPhiinZR(Float_t phi=0.,Int_t nZ=100,Int_t nR=100);
TH2F* CreateHistoDZinZR (Float_t phi=0.,Int_t nZ=100,Int_t nR=100);
TTree* CreateDistortionTree(Double_t step=5);
static void MakeDistortionMap(THnSparse * his0, TTreeSRedirector *pcstream, const char* hname, Int_t run, Float_t refX, Int_t type, Int_t integ=1);
static void MakeDistortionMapCosmic(THnSparse * his0, TTreeSRedirector *pcstream, const char* hname, Int_t run, Float_t refX, Int_t type);
static void MakeDistortionMapSector(THnSparse * his0, TTreeSRedirector *pcstream, const char* hname, Int_t run, Int_t type);
virtual void SetOmegaTauT1T2(Float_t omegaTau,Float_t t1,Float_t t2);
AliExternalTrackParam * FitDistortedTrack(AliExternalTrackParam & trackIn, Double_t refX, Int_t dir,TTreeSRedirector *pcstream);
void StoreInOCDB(Int_t startRun, Int_t endRun, const char *comment=0);
static void MakeTrackDistortionTree(TTree *tinput, Int_t dtype, Int_t ptype, const TObjArray * corrArray, Int_t step=1, Int_t offset=0, Bool_t debug=0);
static void MakeSectorDistortionTree(TTree *tinput, Int_t dtype, Int_t ptype, const TObjArray * corrArray, Int_t step=1, Int_t offset=0, Bool_t debug=0);
static void MakeLaserDistortionTreeOld(TTree* tree, TObjArray *corrArray, Int_t itype);
static void MakeLaserDistortionTree(TTree* tree, TObjArray *corrArray, Int_t itype);
void FastSimDistortedVertex(Double_t orgVertex[3], Int_t nTracks, AliESDVertex &aV, AliESDVertex &avOrg, AliESDVertex &cV, AliESDVertex &cvOrg, TTreeSRedirector * const pcstream, Double_t etaCuts);
static void AddVisualCorrection(AliTPCCorrection* corr, Int_t position);
static AliTPCCorrection* GetVisualCorrection(Int_t position);
static Double_t GetCorrSector(Double_t sector, Double_t r, Double_t kZ, Int_t axisType, Int_t corrType=0);
static Double_t GetCorrXYZ(Double_t gx, Double_t gy, Double_t gz, Int_t axisType, Int_t corrType=0);
static Double_t GetCorrXYZDz(Double_t gx, Double_t gy, Double_t gz, Int_t axisType, Int_t corrType=0,Double_t delta=5);
static Double_t GetCorrXYZIntegrateZ(Double_t gx, Double_t gy, Double_t gz, Int_t axisType, Int_t corrType=0, Double_t delta=5);
static Double_t GetDistXYZ(Double_t gx, Double_t gy, Double_t gz, Int_t axisType, Int_t corrType=0);
static Double_t GetDistXYZDz(Double_t gx, Double_t gy, Double_t gz, Int_t axisType, Int_t corrType=0,Double_t delta=5);
static Double_t GetDistXYZIntegrateZ(Double_t gx, Double_t gy, Double_t gz, Int_t axisType, Int_t corrType=0, Double_t delta=5);
protected:
TH2F* CreateTH2F(const char *name,const char *title,
const char *xlabel,const char *ylabel,const char *zlabel,
Int_t nbinsx,Double_t xlow,Double_t xup,
Int_t nbinsy,Double_t ylow,Double_t yup);
static const Double_t fgkTPCZ0;
static const Double_t fgkIFCRadius;
static const Double_t fgkOFCRadius;
static const Double_t fgkZOffSet;
static const Double_t fgkCathodeV;
static const Double_t fgkGG;
static const Double_t fgkdvdE;
static const Double_t fgkEM;
static const Double_t fgke0;
enum {kNR= 72 };
enum {kNPhi= 18*10+1};
enum {kNZ= 166};
Double_t fgkRList[kNR];
Double_t fgkPhiList[kNPhi];
Double_t fgkZList[kNZ];
Int_t fILow, fJLow, fKLow;
void Interpolate2DEdistortion( Int_t order, Double_t r, Double_t z,
const Double_t er[kNZ][kNR], Double_t &erValue );
void Interpolate3DEdistortion( Int_t order, Double_t r, Float_t phi, Double_t z,
const Double_t er[kNZ][kNPhi][kNR], const Double_t ephi[kNZ][kNPhi][kNR],
const Double_t ez[kNZ][kNPhi][kNR],
Double_t &erValue, Double_t &ephiValue, Double_t &ezValue);
Double_t Interpolate2DTable( Int_t order, Double_t x, Double_t y,
Int_t nx, Int_t ny, const Double_t xv[], const Double_t yv[],
const TMatrixD &array );
Double_t Interpolate3DTable( Int_t order, Double_t x, Double_t y, Double_t z,
Int_t nx, Int_t ny, Int_t nz,
const Double_t xv[], const Double_t yv[], const Double_t zv[],
TMatrixD **arrayofArrays );
Double_t Interpolate( const Double_t xArray[], const Double_t yArray[],
Int_t order, Double_t x );
void Search( Int_t n, const Double_t xArray[], Double_t x, Int_t &low );
Float_t Interpolate2DTable( Int_t order, Double_t x, Double_t y,
Int_t nx, Int_t ny, const Double_t xv[], const Double_t yv[],
const TMatrixF &array );
Float_t Interpolate3DTable( Int_t order, Double_t x, Double_t y, Double_t z,
Int_t nx, Int_t ny, Int_t nz,
const Double_t xv[], const Double_t yv[], const Double_t zv[],
TMatrixF **arrayofArrays );
Float_t Interpolate( const Double_t xArray[], const Float_t yArray[],
Int_t order, Double_t x );
virtual Int_t IsPowerOfTwo ( Int_t i ) const ;
void PoissonRelaxation2D(TMatrixD &arrayV, TMatrixD &chargeDensity,
TMatrixD &arrayErOverEz, TMatrixD &arrayDeltaEz,
Int_t rows, Int_t columns, Int_t iterations,
Bool_t rocDisplacement = kTRUE);
void PoissonRelaxation3D( TMatrixD **arrayofArrayV, TMatrixD **arrayofChargeDensities,
TMatrixD **arrayofEroverEz, TMatrixD **arrayofEPhioverEz, TMatrixD **arrayofEz,
Int_t rows, Int_t columns, Int_t phislices,
Float_t deltaphi, Int_t iterations, Int_t summetry,
Bool_t rocDisplacement = kTRUE);
void SetIsLocal(Bool_t isLocal){fIsLocal=isLocal;}
Bool_t IsLocal() const { return fIsLocal;}
protected:
Double_t fT1;
Double_t fT2;
Bool_t fIsLocal;
static TObjArray *fgVisualCorrection;
private:
AliTPCCorrection(const AliTPCCorrection &);
AliTPCCorrection &operator=(const AliTPCCorrection &);
void InitLookUpfulcrums();
ClassDef(AliTPCCorrection,5);
};
#endif