[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 ExportTrackPoints(AliESDEvent *event);
  //
  //
  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 
  //
  //
  //
  Bool_t    fUseInnerOuter;         // flag- use Inner Outer sector for left righ alignment
  
  static AliTPCcalibAlign*   fgInstance; //! Instance of this class (singleton implementation)
private:
  AliTPCcalibAlign&  operator=(const AliTPCcalibAlign&);// not implemented

  ClassDef(AliTPCcalibAlign,3)
};


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