[u/mrichter/AliRoot.git] / TPC / AliTPCcalibAlign.h

#ifndef ALITPCCALIBALIGN_H
#define ALITPCCALIBALIGN_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

////
////
////

#include "TObject.h"
#include "TObjArray.h"
#include "TLinearFitter.h"
#include "AliTPCcalibBase.h"
#include "TH1.h"

class AliExternalTrackParam;
class AliExternalComparison;
class AliTPCseed;
class TGraphErrors;
class TTree;
class THnSparse;
class AliTPCPointCorrection;
class TFile;

class AliTPCcalibAlign:public AliTPCcalibBase {
public:
  enum HistoType {kY=0, kZ =1, kPhi=2, kTheta=3, 
		  kYPhi=4, kZTheta=5, 
		  kYz=6,kZz=7,kPhiZ=8,kThetaZ=9};
  enum FitType{ k6=0, k9=1, k12=2};
  AliTPCcalibAlign();
  AliTPCcalibAlign(const Text_t *name, const Text_t *title);
  AliTPCcalibAlign(const AliTPCcalibAlign &align);
  //
  virtual ~AliTPCcalibAlign();
  void     Process(AliESDEvent *event);
  virtual void Process(AliTPCseed *track);
  virtual void Analyze();
  virtual void Terminate();  
  virtual Long64_t Merge(TCollection* list);
  //
  //
  void MakeReportDy(TFile *output); 
  void MakeReportDyPhi(TFile *output);
  //
  void UpdatePointCorrection(AliTPCPointCorrection * correction);
  //
  virtual void EvalFitters(Int_t minPoints=20);
  TH1 * GetHisto(HistoType type, Int_t s1, Int_t s2, Bool_t force=kFALSE);
  void  MakeTree(const char *fname="alignTree.root", Int_t minPoints=20);
  TGraphErrors * MakeGraph(Int_t sec0, Int_t sec1, Int_t dsec, 
			   Int_t i0, Int_t i1, FitType type); 
  Int_t  RefitLinear(const AliTPCseed * seed, Int_t isec, Double_t *fitParam, Int_t refSector, TMatrixD &param, TMatrixD&covar, Double_t xRef, Bool_t both=kFALSE);
  
  void ProcessTracklets(const AliExternalTrackParam &t1,
			const AliExternalTrackParam &t2,
			const AliTPCseed * seed,
			Int_t s1,Int_t s2);
  
  void UpdateAlignSector(const AliTPCseed * seed,Int_t isec); 
  inline Int_t GetIndex(Int_t s1,Int_t s2){return 72*s1+s2;}
  //
  inline const TMatrixD     * GetTransformation(Int_t s1,Int_t s2, Int_t fitType);
  //
  inline TLinearFitter* GetFitter12(Int_t s1,Int_t s2);
  inline TLinearFitter* GetFitter9(Int_t s1,Int_t s2);
  inline TLinearFitter* GetFitter6(Int_t s1,Int_t s2);
  //
  Bool_t GetTransformation12(Int_t s1,Int_t s2,TMatrixD &a);
  Bool_t GetTransformation9(Int_t s1,Int_t s2,TMatrixD &a);
  Bool_t GetTransformation6(Int_t s1,Int_t s2,TMatrixD &a);
  Int_t  AcceptTracklet(const AliExternalTrackParam &tp1,
			const AliExternalTrackParam &tp2);
  Int_t  AcceptTracklet(const Double_t *t1,
			const Double_t *t2);

  void ProcessDiff(const AliExternalTrackParam &t1,
		   const AliExternalTrackParam &t2,
		   const AliTPCseed *seed,
		   Int_t s1,Int_t s2);
  void ProcessAlign(Double_t * t1, Double_t * t2, Int_t s1,Int_t s2);

//   Bool_t GetTransformationCovar12(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
//   Bool_t GetTransformationCovar9(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
//   Bool_t GetTransformationCovar6(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
  void Add(AliTPCcalibAlign * align);
  Int_t *GetPoints() {return fPoints;}
  void     Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
  TLinearFitter* GetOrMakeFitter12(Int_t s1,Int_t s2);
  TLinearFitter* GetOrMakeFitter9(Int_t s1,Int_t s2);
  TLinearFitter* GetOrMakeFitter6(Int_t s1,Int_t s2);
  void Process12(const Double_t *t1, const Double_t *t2,
		 TLinearFitter *fitter);
  void Process9(Double_t *t1, Double_t *t2, TLinearFitter *fitter);
  void Process6(Double_t *t1, Double_t *t2, TLinearFitter *fitter);
  void ProcessTree(TTree * tree, AliExternalComparison *comp=0);
  void GlobalAlign6(Int_t minPoints, Float_t sysError, Int_t niter);
  //
  // Cluster comparison Part
  //
  //
  // For visualization and test purposes
  //
  Double_t Correct(Int_t type, Int_t value, Int_t s1, Int_t s2, Double_t x, Double_t y, Double_t z, Double_t phi,Double_t theta); 
  static Double_t SCorrect(Int_t type, Int_t value, Int_t s1, Int_t s2, Double_t x, Double_t y, Double_t z, Double_t phi,Double_t theta){return Instance()->Correct(type,value,s1,s2,x,y,z,phi,theta);}
  static AliTPCcalibAlign* Instance();
  void SetInstance(AliTPCcalibAlign*param){fgInstance = param;}
  static void Constrain1Pt(AliExternalTrackParam &t1, const AliExternalTrackParam &t2, Bool_t noField);
  void SetNoField(Bool_t noField){ fNoField=noField;}

  //
  // Kalman fileter for sectors
  //
  void MakeSectorKalman();
  void UpdateSectorKalman(Int_t sector, Int_t quadrant0, Int_t quadrant1,  TMatrixD *p0, TMatrixD *c0, TMatrixD *p1, TMatrixD *c1);
  void UpdateSectorKalman(TMatrixD &par0, TMatrixD &cov0, TMatrixD &para1, TMatrixD &cov1);
  Double_t GetCorrectionSector(Int_t coord, Int_t sector, Double_t lx, Double_t ly, Double_t lz); 
  static Double_t SGetCorrectionSector(Int_t coord, Int_t sector, Double_t lx, Double_t ly, Double_t lz); 

  //
  // Kalman filter for full TPC
  //
  void MakeKalman();
  void UpdateKalman(Int_t sector0, Int_t sector1,  TMatrixD &p0, TMatrixD &c0, TMatrixD &p1, TMatrixD &c1);
  void UpdateKalman(TMatrixD &par0, TMatrixD &cov0, TMatrixD &para1, TMatrixD &cov1);
  //
  //private:
  static Int_t CheckCovariance(TMatrixD &covar);
public:
  
  void FillHisto(const Double_t *t1,
		 const Double_t *t2,
		 Int_t s1,Int_t s2);

  TObjArray fDphiHistArray;    // array of residual histograms  phi      -kPhi
  TObjArray fDthetaHistArray;  // array of residual histograms  theta    -kTheta
  TObjArray fDyHistArray;      // array of residual histograms  y        -kY
  TObjArray fDzHistArray;      // array of residual histograms  z        -kZ
  //
  TObjArray fDyPhiHistArray;      // array of residual histograms  y     -kYPhi
  TObjArray fDzThetaHistArray;    // array of residual histograms  z-z   -kZTheta
  //
  TObjArray fDphiZHistArray;      // array of residual histograms  phi   -kPhiz
  TObjArray fDthetaZHistArray;    // array of residual histograms  theta -kThetaz
  TObjArray fDyZHistArray;        // array of residual histograms  y     -kYz
  TObjArray fDzZHistArray;        // array of residual histograms  z     -kZz
  //
  //
  TObjArray fFitterArray12;    // array of fitters
  TObjArray fFitterArray9;     // array of fitters
  TObjArray fFitterArray6;     // array of fitters
  //
  TObjArray fMatrixArray12;    // array of transnformtation matrix
  TObjArray fMatrixArray9;     // array of transnformtation matrix
  TObjArray fMatrixArray6;     // array of transnformtation matrix 
  //
  //
  TObjArray fCombinedMatrixArray6;      // array  combeined transformation matrix
  //
  AliExternalComparison  *fCompTracklet;  //tracklet comparison
  //
  Int_t fPoints[72*72];        // number of points in the fitter 
  Bool_t fNoField;             // flag - no field data
  // refernce x
  Double_t fXIO;               // OROC-IROC boundary
  Double_t fXmiddle;           // center of the TPC sector local X
  Double_t fXquadrant;         // x quadrant
  //
  // Kalman filter for sector internal  alignemnt
  //
  TObjArray fArraySectorIntParam; // array of sector alignment parameters
  TObjArray fArraySectorIntCovar; // array of sector alignment covariances 
  //
  // Kalman filter for global alignment
  //
  TMatrixD  *fSectorParamA;     // Kalman parameter   for A side
  TMatrixD  *fSectorCovarA;     // Kalman covariance  for A side 
  TMatrixD  *fSectorParamC;     // Kalman parameter   for A side
  TMatrixD  *fSectorCovarC;     // Kalman covariance  for A side 


  static AliTPCcalibAlign*   fgInstance; //! Instance of this class (singleton implementation)
private:
  AliTPCcalibAlign&  operator=(const AliTPCcalibAlign&);// not implemented

  ClassDef(AliTPCcalibAlign,2)
};


TLinearFitter* AliTPCcalibAlign::GetFitter12(Int_t s1,Int_t s2) {
  return static_cast<TLinearFitter*>(fFitterArray12[GetIndex(s1,s2)]);
}
TLinearFitter* AliTPCcalibAlign::GetFitter9(Int_t s1,Int_t s2) {
  return static_cast<TLinearFitter*>(fFitterArray9[GetIndex(s1,s2)]);
}
TLinearFitter* AliTPCcalibAlign::GetFitter6(Int_t s1,Int_t s2) {
  return static_cast<TLinearFitter*>(fFitterArray6[GetIndex(s1,s2)]);
}

const TMatrixD * AliTPCcalibAlign::GetTransformation(Int_t s1,Int_t s2, Int_t fitType){
  if (fitType==0) return static_cast<TMatrixD*>(fMatrixArray6[GetIndex(s1,s2)]);
  if (fitType==1) return static_cast<TMatrixD*>(fMatrixArray9[GetIndex(s1,s2)]);
  if (fitType==2) return static_cast<TMatrixD*>(fMatrixArray12[GetIndex(s1,s2)]);
  return 0;
}


#endif
Commit	Line	Data
9318a5b4	1	#ifndef ALITPCCALIBALIGN_H
	2	#define ALITPCCALIBALIGN_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////
	8	////
	9	////
	10
	11	#include "TObject.h"
	12	#include "TObjArray.h"
	13	#include "TLinearFitter.h"
e4042305	14	#include "AliTPCcalibBase.h"
e4042305	15	#include "TH1.h"
9318a5b4	16
9318a5b4	17	class AliExternalTrackParam;
1d82fc56	18	class AliExternalComparison;
e4042305	19	class AliTPCseed;
8b3c60d8	20	class TGraphErrors;
6f387311	21	class TTree;
1d82fc56	22	class THnSparse;
774a5ee9	23	class AliTPCPointCorrection;
774a5ee9	24	class TFile;
9318a5b4	25
e4042305	26	class AliTPCcalibAlign:public AliTPCcalibBase {
9318a5b4	27	public:
bb6bc8f6	28	enum HistoType {kY=0, kZ =1, kPhi=2, kTheta=3,
	29	kYPhi=4, kZTheta=5,
	30	kYz=6,kZz=7,kPhiZ=8,kThetaZ=9};
	31	enum FitType{ k6=0, k9=1, k12=2};
9318a5b4	32	AliTPCcalibAlign();
e149f26d	33	AliTPCcalibAlign(const Text_t name, const Text_t title);
bb6bc8f6	34	AliTPCcalibAlign(const AliTPCcalibAlign &align);
bb6bc8f6	35	//
9318a5b4	36	virtual ~AliTPCcalibAlign();
774a5ee9	37	void Process(AliESDEvent *event);
e4042305	38	virtual void Process(AliTPCseed *track);
7eaa723e	39	virtual void Analyze();
7eaa723e	40	virtual void Terminate();
ae0ac7be	41	virtual Long64_t Merge(TCollection* list);
7eaa723e	42	//
774a5ee9	43	//
	44	void MakeReportDy(TFile *output);
	45	void MakeReportDyPhi(TFile *output);
	46	//
	47	void UpdatePointCorrection(AliTPCPointCorrection * correction);
	48	//
	49	virtual void EvalFitters(Int_t minPoints=20);
8b3c60d8	50	TH1 * GetHisto(HistoType type, Int_t s1, Int_t s2, Bool_t force=kFALSE);
774a5ee9	51	void MakeTree(const char *fname="alignTree.root", Int_t minPoints=20);
8b3c60d8	52	TGraphErrors * MakeGraph(Int_t sec0, Int_t sec1, Int_t dsec,
8b3c60d8	53	Int_t i0, Int_t i1, FitType type);
774a5ee9	54	Int_t RefitLinear(const AliTPCseed * seed, Int_t isec, Double_t *fitParam, Int_t refSector, TMatrixD &param, TMatrixD&covar, Double_t xRef, Bool_t both=kFALSE);
774a5ee9	55
e4042305	56	void ProcessTracklets(const AliExternalTrackParam &t1,
e4042305	57	const AliExternalTrackParam &t2,
967eae0d	58	const AliTPCseed * seed,
e4042305	59	Int_t s1,Int_t s2);
774a5ee9	60
774a5ee9	61	void UpdateAlignSector(const AliTPCseed * seed,Int_t isec);
972cf6f2	62	inline Int_t GetIndex(Int_t s1,Int_t s2){return 72*s1+s2;}
972cf6f2	63	//
6f387311	64	inline const TMatrixD * GetTransformation(Int_t s1,Int_t s2, Int_t fitType);
6f387311	65	//
972cf6f2	66	inline TLinearFitter* GetFitter12(Int_t s1,Int_t s2);
	67	inline TLinearFitter* GetFitter9(Int_t s1,Int_t s2);
	68	inline TLinearFitter* GetFitter6(Int_t s1,Int_t s2);
	69	//
9318a5b4	70	Bool_t GetTransformation12(Int_t s1,Int_t s2,TMatrixD &a);
	71	Bool_t GetTransformation9(Int_t s1,Int_t s2,TMatrixD &a);
	72	Bool_t GetTransformation6(Int_t s1,Int_t s2,TMatrixD &a);
1d82fc56	73	Int_t AcceptTracklet(const AliExternalTrackParam &tp1,
6f387311	74	const AliExternalTrackParam &tp2);
774a5ee9	75	Int_t AcceptTracklet(const Double_t *t1,
774a5ee9	76	const Double_t *t2);
6f387311	77
967eae0d	78	void ProcessDiff(const AliExternalTrackParam &t1,
	79	const AliExternalTrackParam &t2,
	80	const AliTPCseed *seed,
	81	Int_t s1,Int_t s2);
6f387311	82	void ProcessAlign(Double_t * t1, Double_t * t2, Int_t s1,Int_t s2);
967eae0d	83
972cf6f2	84	// Bool_t GetTransformationCovar12(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
	85	// Bool_t GetTransformationCovar9(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
	86	// Bool_t GetTransformationCovar6(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
ae0ac7be	87	void Add(AliTPCcalibAlign * align);
108953e9	88	Int_t *GetPoints() {return fPoints;}
1aa15e8d	89	void Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
6f387311	90	TLinearFitter* GetOrMakeFitter12(Int_t s1,Int_t s2);
	91	TLinearFitter* GetOrMakeFitter9(Int_t s1,Int_t s2);
	92	TLinearFitter* GetOrMakeFitter6(Int_t s1,Int_t s2);
972cf6f2	93	void Process12(const Double_t t1, const Double_t t2,
	94	TLinearFitter *fitter);
	95	void Process9(Double_t t1, Double_t t2, TLinearFitter *fitter);
	96	void Process6(Double_t t1, Double_t t2, TLinearFitter *fitter);
1d82fc56	97	void ProcessTree(TTree * tree, AliExternalComparison *comp=0);
	98	void GlobalAlign6(Int_t minPoints, Float_t sysError, Int_t niter);
	99	//
	100	// Cluster comparison Part
	101	//
6f387311	102	//
	103	// For visualization and test purposes
	104	//
	105	Double_t Correct(Int_t type, Int_t value, Int_t s1, Int_t s2, Double_t x, Double_t y, Double_t z, Double_t phi,Double_t theta);
	106	static Double_t SCorrect(Int_t type, Int_t value, Int_t s1, Int_t s2, Double_t x, Double_t y, Double_t z, Double_t phi,Double_t theta){return Instance()->Correct(type,value,s1,s2,x,y,z,phi,theta);}
	107	static AliTPCcalibAlign* Instance();
	108	void SetInstance(AliTPCcalibAlign*param){fgInstance = param;}
1d82fc56	109	static void Constrain1Pt(AliExternalTrackParam &t1, const AliExternalTrackParam &t2, Bool_t noField);
1d82fc56	110	void SetNoField(Bool_t noField){ fNoField=noField;}
774a5ee9	111
	112	//
	113	// Kalman fileter for sectors
	114	//
	115	void MakeSectorKalman();
	116	void UpdateSectorKalman(Int_t sector, Int_t quadrant0, Int_t quadrant1, TMatrixD p0, TMatrixD c0, TMatrixD p1, TMatrixD c1);
	117	void UpdateSectorKalman(TMatrixD &par0, TMatrixD &cov0, TMatrixD &para1, TMatrixD &cov1);
	118	Double_t GetCorrectionSector(Int_t coord, Int_t sector, Double_t lx, Double_t ly, Double_t lz);
	119	static Double_t SGetCorrectionSector(Int_t coord, Int_t sector, Double_t lx, Double_t ly, Double_t lz);
	120
	121	//
	122	// Kalman filter for full TPC
	123	//
	124	void MakeKalman();
	125	void UpdateKalman(Int_t sector0, Int_t sector1, TMatrixD &p0, TMatrixD &c0, TMatrixD &p1, TMatrixD &c1);
	126	void UpdateKalman(TMatrixD &par0, TMatrixD &cov0, TMatrixD &para1, TMatrixD &cov1);
	127	//
	128	//private:
	129	static Int_t CheckCovariance(TMatrixD &covar);
	130	public:
6f387311	131
774a5ee9	132	void FillHisto(const Double_t *t1,
774a5ee9	133	const Double_t *t2,
6f387311	134	Int_t s1,Int_t s2);
6f387311	135
bb6bc8f6	136	TObjArray fDphiHistArray; // array of residual histograms phi -kPhi
	137	TObjArray fDthetaHistArray; // array of residual histograms theta -kTheta
	138	TObjArray fDyHistArray; // array of residual histograms y -kY
	139	TObjArray fDzHistArray; // array of residual histograms z -kZ
	140	//
	141	TObjArray fDyPhiHistArray; // array of residual histograms y -kYPhi
	142	TObjArray fDzThetaHistArray; // array of residual histograms z-z -kZTheta
	143	//
	144	TObjArray fDphiZHistArray; // array of residual histograms phi -kPhiz
	145	TObjArray fDthetaZHistArray; // array of residual histograms theta -kThetaz
	146	TObjArray fDyZHistArray; // array of residual histograms y -kYz
	147	TObjArray fDzZHistArray; // array of residual histograms z -kZz
	148	//
	149	//
972cf6f2	150	TObjArray fFitterArray12; // array of fitters
	151	TObjArray fFitterArray9; // array of fitters
	152	TObjArray fFitterArray6; // array of fitters
6f387311	153	//
	154	TObjArray fMatrixArray12; // array of transnformtation matrix
	155	TObjArray fMatrixArray9; // array of transnformtation matrix
1d82fc56	156	TObjArray fMatrixArray6; // array of transnformtation matrix
	157	//
	158	//
	159	TObjArray fCombinedMatrixArray6; // array combeined transformation matrix
	160	//
	161	AliExternalComparison *fCompTracklet; //tracklet comparison
6f387311	162	//
6f387311	163	Int_t fPoints[72*72]; // number of points in the fitter
774a5ee9	164	Bool_t fNoField; // flag - no field data
	165	// refernce x
	166	Double_t fXIO; // OROC-IROC boundary
	167	Double_t fXmiddle; // center of the TPC sector local X
	168	Double_t fXquadrant; // x quadrant
	169	//
	170	// Kalman filter for sector internal alignemnt
	171	//
	172	TObjArray fArraySectorIntParam; // array of sector alignment parameters
	173	TObjArray fArraySectorIntCovar; // array of sector alignment covariances
	174	//
	175	// Kalman filter for global alignment
	176	//
	177	TMatrixD *fSectorParamA; // Kalman parameter for A side
	178	TMatrixD *fSectorCovarA; // Kalman covariance for A side
	179	TMatrixD *fSectorParamC; // Kalman parameter for A side
	180	TMatrixD *fSectorCovarC; // Kalman covariance for A side
	181
	182
6f387311	183	static AliTPCcalibAlign* fgInstance; //! Instance of this class (singleton implementation)
774a5ee9	184	private:
	185	AliTPCcalibAlign& operator=(const AliTPCcalibAlign&);// not implemented
	186
1d82fc56	187	ClassDef(AliTPCcalibAlign,2)
9318a5b4	188	};
9318a5b4	189
972cf6f2	190
	191	TLinearFitter* AliTPCcalibAlign::GetFitter12(Int_t s1,Int_t s2) {
	192	return static_cast<TLinearFitter*>(fFitterArray12[GetIndex(s1,s2)]);
	193	}
	194	TLinearFitter* AliTPCcalibAlign::GetFitter9(Int_t s1,Int_t s2) {
	195	return static_cast<TLinearFitter*>(fFitterArray9[GetIndex(s1,s2)]);
	196	}
	197	TLinearFitter* AliTPCcalibAlign::GetFitter6(Int_t s1,Int_t s2) {
	198	return static_cast<TLinearFitter*>(fFitterArray6[GetIndex(s1,s2)]);
	199	}
	200
6f387311	201	const TMatrixD * AliTPCcalibAlign::GetTransformation(Int_t s1,Int_t s2, Int_t fitType){
	202	if (fitType==0) return static_cast<TMatrixD*>(fMatrixArray6[GetIndex(s1,s2)]);
	203	if (fitType==1) return static_cast<TMatrixD*>(fMatrixArray9[GetIndex(s1,s2)]);
	204	if (fitType==2) return static_cast<TMatrixD*>(fMatrixArray12[GetIndex(s1,s2)]);
1d82fc56	205	return 0;
6f387311	206	}
972cf6f2	207
	208
	209
9318a5b4	210	#endif