[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 AliTPCseed;
class TGraphErrors;

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();
  virtual void Process(AliTPCseed *track);
  virtual void Analyze();
  virtual void Terminate();  
  virtual Long64_t Merge(TCollection* list);
  //
  virtual void EvalFitters();
  TH1 * GetHisto(HistoType type, Int_t s1, Int_t s2, Bool_t force=kFALSE);
  void  MakeTree(const char *fname="alignTree.root");
  TGraphErrors * MakeGraph(Int_t sec0, Int_t sec1, Int_t dsec, 
			   Int_t i0, Int_t i1, FitType type); 
  void ProcessTracklets(const AliExternalTrackParam &t1,
			const AliExternalTrackParam &t2,
			const AliTPCseed * seed,
			Int_t s1,Int_t s2);
  inline Int_t GetIndex(Int_t s1,Int_t s2){return 72*s1+s2;}
  //
  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);
  
  void ProcessDiff(const AliExternalTrackParam &t1,
		   const AliExternalTrackParam &t2,
		   const AliTPCseed *seed,
		   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);};
  void     Process(AliESDEvent *event){AliTPCcalibBase::Process(event);}
private:
  void FillHisto(const AliExternalTrackParam &t1,
			const AliExternalTrackParam &t2,
			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);
  TLinearFitter* GetOrMakeFitter12(Int_t s1,Int_t s2);
  TLinearFitter* GetOrMakeFitter9(Int_t s1,Int_t s2);
  TLinearFitter* GetOrMakeFitter6(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
  Int_t fPoints[72*72];        // number of points in the fitter
  ClassDef(AliTPCcalibAlign,1)
};


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)]);
}


#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;
e4042305	18	class AliTPCseed;
8b3c60d8	19	class TGraphErrors;
9318a5b4	20
e4042305	21	class AliTPCcalibAlign:public AliTPCcalibBase {
9318a5b4	22	public:
bb6bc8f6	23	enum HistoType {kY=0, kZ =1, kPhi=2, kTheta=3,
	24	kYPhi=4, kZTheta=5,
	25	kYz=6,kZz=7,kPhiZ=8,kThetaZ=9};
	26	enum FitType{ k6=0, k9=1, k12=2};
9318a5b4	27	AliTPCcalibAlign();
e149f26d	28	AliTPCcalibAlign(const Text_t name, const Text_t title);
bb6bc8f6	29	AliTPCcalibAlign(const AliTPCcalibAlign &align);
bb6bc8f6	30	//
9318a5b4	31	virtual ~AliTPCcalibAlign();
e4042305	32	virtual void Process(AliTPCseed *track);
7eaa723e	33	virtual void Analyze();
7eaa723e	34	virtual void Terminate();
ae0ac7be	35	virtual Long64_t Merge(TCollection* list);
7eaa723e	36	//
7eaa723e	37	virtual void EvalFitters();
8b3c60d8	38	TH1 * GetHisto(HistoType type, Int_t s1, Int_t s2, Bool_t force=kFALSE);
	39	void MakeTree(const char *fname="alignTree.root");
	40	TGraphErrors * MakeGraph(Int_t sec0, Int_t sec1, Int_t dsec,
	41	Int_t i0, Int_t i1, FitType type);
e4042305	42	void ProcessTracklets(const AliExternalTrackParam &t1,
e4042305	43	const AliExternalTrackParam &t2,
967eae0d	44	const AliTPCseed * seed,
e4042305	45	Int_t s1,Int_t s2);
972cf6f2	46	inline Int_t GetIndex(Int_t s1,Int_t s2){return 72*s1+s2;}
	47	//
	48	inline TLinearFitter* GetFitter12(Int_t s1,Int_t s2);
	49	inline TLinearFitter* GetFitter9(Int_t s1,Int_t s2);
	50	inline TLinearFitter* GetFitter6(Int_t s1,Int_t s2);
	51	//
9318a5b4	52	Bool_t GetTransformation12(Int_t s1,Int_t s2,TMatrixD &a);
	53	Bool_t GetTransformation9(Int_t s1,Int_t s2,TMatrixD &a);
	54	Bool_t GetTransformation6(Int_t s1,Int_t s2,TMatrixD &a);
967eae0d	55
	56	void ProcessDiff(const AliExternalTrackParam &t1,
	57	const AliExternalTrackParam &t2,
	58	const AliTPCseed *seed,
	59	Int_t s1,Int_t s2);
	60
972cf6f2	61	// Bool_t GetTransformationCovar12(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
	62	// Bool_t GetTransformationCovar9(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
	63	// Bool_t GetTransformationCovar6(Int_t s1,Int_t s2,TMatrixD &a, Bool_t norm=kFALSE);
ae0ac7be	64	void Add(AliTPCcalibAlign * align);
108953e9	65	Int_t *GetPoints() {return fPoints;}
1aa15e8d	66	void Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
1aa15e8d	67	void Process(AliESDEvent *event){AliTPCcalibBase::Process(event);}
9318a5b4	68	private:
972cf6f2	69	void FillHisto(const AliExternalTrackParam &t1,
	70	const AliExternalTrackParam &t2,
	71	Int_t s1,Int_t s2);
9318a5b4	72
972cf6f2	73	void Process12(const Double_t t1, const Double_t t2,
	74	TLinearFitter *fitter);
	75	void Process9(Double_t t1, Double_t t2, TLinearFitter *fitter);
	76	void Process6(Double_t t1, Double_t t2, TLinearFitter *fitter);
	77	TLinearFitter* GetOrMakeFitter12(Int_t s1,Int_t s2);
	78	TLinearFitter* GetOrMakeFitter9(Int_t s1,Int_t s2);
	79	TLinearFitter* GetOrMakeFitter6(Int_t s1,Int_t s2);
bb6bc8f6	80	TObjArray fDphiHistArray; // array of residual histograms phi -kPhi
	81	TObjArray fDthetaHistArray; // array of residual histograms theta -kTheta
	82	TObjArray fDyHistArray; // array of residual histograms y -kY
	83	TObjArray fDzHistArray; // array of residual histograms z -kZ
	84	//
	85	TObjArray fDyPhiHistArray; // array of residual histograms y -kYPhi
	86	TObjArray fDzThetaHistArray; // array of residual histograms z-z -kZTheta
	87	//
	88	TObjArray fDphiZHistArray; // array of residual histograms phi -kPhiz
	89	TObjArray fDthetaZHistArray; // array of residual histograms theta -kThetaz
	90	TObjArray fDyZHistArray; // array of residual histograms y -kYz
	91	TObjArray fDzZHistArray; // array of residual histograms z -kZz
	92	//
	93	//
972cf6f2	94	TObjArray fFitterArray12; // array of fitters
	95	TObjArray fFitterArray9; // array of fitters
	96	TObjArray fFitterArray6; // array of fitters
	97	Int_t fPoints[72*72]; // number of points in the fitter
9318a5b4	98	ClassDef(AliTPCcalibAlign,1)
	99	};
	100
972cf6f2	101
	102	TLinearFitter* AliTPCcalibAlign::GetFitter12(Int_t s1,Int_t s2) {
	103	return static_cast<TLinearFitter*>(fFitterArray12[GetIndex(s1,s2)]);
	104	}
	105	TLinearFitter* AliTPCcalibAlign::GetFitter9(Int_t s1,Int_t s2) {
	106	return static_cast<TLinearFitter*>(fFitterArray9[GetIndex(s1,s2)]);
	107	}
	108	TLinearFitter* AliTPCcalibAlign::GetFitter6(Int_t s1,Int_t s2) {
	109	return static_cast<TLinearFitter*>(fFitterArray6[GetIndex(s1,s2)]);
	110	}
	111
	112
	113
	114
	115
9318a5b4	116	#endif