[u/mrichter/AliRoot.git] / TPC / Calib / AliTPCcalibTracksGain.h

#ifndef AliTPCCALIBTRACKSGAIN_H
#define AliTPCCALIBTRACKSGAIN_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
#include <TChain.h>
#include <TNamed.h>


#include <TObjArray.h>
#include <TH2D.h>
#include <TVectorD.h>
#include <TMatrixD.h>

#include <iostream>
#include <TH1F.h>

#include <AliTPCcalibBase.h>

using namespace std;

class TTreeSRedirector;
class TObjString;
class TLinearFitter;
class TProfile; 
class TProfile2D; 
class TH1F; 

class AliTPCClusterParam; 
class AliTPCParamSR; 
class AliTPCCalROC;
class AliTPCCalPad;
class AliTPCseed; 
class AliTPCclusterMI; 
class AliTrackPointArray;
class TTreeStream;
class AliTPCcalibTracksCuts;
class AliTPCFitPad;
class TGraph;
class AliTPCCClusterParam;

class AliTPCcalibTracksGain : public AliTPCcalibBase {
public:
   enum {
      kShortPads = 0,
      kMediumPads = 1,
      kLongPads = 2
   };
   enum {
      kSimpleFitter = 0,
      kSqrtFitter = 1,
      kLogFitter = 2
   };
   
  AliTPCcalibTracksGain();
  AliTPCcalibTracksGain(const AliTPCcalibTracksGain& obj);
  AliTPCcalibTracksGain(const char* name, const char* title, AliTPCcalibTracksCuts* cuts);
  virtual ~AliTPCcalibTracksGain();
  AliTPCcalibTracksGain& operator=(const AliTPCcalibTracksGain& rhs);
  virtual Long64_t        Merge(TCollection *list);
  virtual void            Process(AliTPCseed* seed);
  virtual void            Terminate();
  virtual void            Analyze();
  //
  // Tracks and cluster manipulation
  //
  Float_t         GetGain(AliTPCclusterMI *cl);
  Float_t         GetMaxNorm(AliTPCclusterMI * cl,  Float_t ky, Float_t kz);
  Float_t         GetQNorm(AliTPCclusterMI * cl,  Float_t ky, Float_t kz);
  void            AddTrack(AliTPCseed* seed);
  void            DumpTrack(AliTPCseed* track);
  Bool_t          GetDedx(AliTPCseed* track, Int_t padType,  Int_t*rows,
			  Int_t &sector, Int_t& npoints, 
			  TVectorD &dedxM, TVectorD &dedxQ, 
			  TVectorD &parY, TVectorD &parZ, TVectorD & meanPos);
  void            AddCluster(AliTPCclusterMI* cluster, Float_t momenta, Float_t mdedx, Int_t padType, Float_t xcenter, TVectorD &dedxQ, TVectorD & dedxM, Float_t fraction, Float_t fraction2, Float_t dedge, TVectorD& parY, TVectorD& parZ, TVectorD& meanPos);
  void            AddTracklet(UInt_t sector, UInt_t padType,TVectorD &dedxQ, TVectorD &dedxM,TVectorD& parY, TVectorD& parZ, TVectorD& meanPos);

  void            Add(AliTPCcalibTracksGain* cal);

  //
  // Get Derived results - gain maps
  //
  AliTPCCalPad*   CreateFitCalPad(UInt_t fitType, Bool_t undoTransformation = kFALSE, Bool_t normalizeToPadSize = kFALSE);
  AliTPCCalROC*   CreateFitCalROC(UInt_t sector, UInt_t fitType, Bool_t undoTransformation = kFALSE, Bool_t normalizeToPadSize = kFALSE);
  AliTPCCalROC*   CreateFitCalROC(UInt_t sector, UInt_t padType, TVectorD &fitParam, UInt_t fitType, Bool_t undoTransformation = kFALSE, Bool_t normalizeToPadSize = kFALSE);
  AliTPCCalROC*   CreateCombinedCalROC(const AliTPCCalROC* roc1, const AliTPCCalROC* roc2);
  //
  void            Evaluate(Bool_t robust = kFALSE, Double_t frac = -1.);
  Bool_t            GetParameters(UInt_t segment, UInt_t padType, UInt_t fitType, TVectorD &fitParam);
  void            GetErrors(UInt_t segment, UInt_t padType, UInt_t fitType, TVectorD &fitError);
  void            GetCovarianceMatrix(UInt_t segment, UInt_t padType, UInt_t fitType, TMatrixD& covMatrix);
  //
  //
  void            UpdateClusterParam(AliTPCClusterParam *param);
  TGraph *        CreateAmpGraph(Int_t ipad, Bool_t qmax);


  TLinearFitter*  GetFitter(UInt_t segment, UInt_t padType, UInt_t fitType);  
  void     Process(AliESDEvent *event) {AliTPCcalibBase::Process(event);};
  void     Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
  
public:
  //
  // Helper function
  //
  static Double_t GetPadLength(Double_t lx);
  static Int_t    GetPadType(Double_t lx);  
  //
  //
  AliTPCcalibTracksCuts* fCuts;            // cuts that are used for sieving the tracks used for calibration
  //
  // kalman fit - alignment of dEdx
  //
  //
  // Fitters
  //
  AliTPCFitPad*     fSimpleFitter;         // simple fitter for short pads
  AliTPCFitPad*     fSqrtFitter;           // sqrt fitter for medium pads
  AliTPCFitPad*     fLogFitter;            // log fitter for long pads
  //
  TLinearFitter*    fFitter0M;          // fitting of the atenuation, angular correction, and mean chamber gain
  TLinearFitter*    fFitter1M;          // fitting of the atenuation, angular correction, and mean chamber gain
  TLinearFitter*    fFitter2M;          // fitting of the atenuation, angular correction, and mean chamber gain
  TLinearFitter*    fFitter0T;          // fitting of the atenuation, angular correction, and mean chamber gain
  TLinearFitter*    fFitter1T;          // fitting of the atenuation, angular correction, and mean chamber gain
  TLinearFitter*    fFitter2T;          // fitting of the atenuation, angular correction, and mean chamber gain
  //
  // angular adn diffusion effect fitter
  // 
  TLinearFitter*    fDFitter0M;          // fitting of the atenuation, angular correction
  TLinearFitter*    fDFitter1M;          // fitting of the atenuation, angular correction
  TLinearFitter*    fDFitter2M;          // fitting of the atenuation, angular correction
  TLinearFitter*    fDFitter0T;          // fitting of the atenuation, angular correction 
  TLinearFitter*    fDFitter1T;          // fitting of the atenuation, angular correction
  TLinearFitter*    fDFitter2T;          // fitting of the atenuation, angular correction
  //
  AliTPCFitPad*     fSingleSectorFitter;   // just for debugging  
  //
  // Conters
  //
  UInt_t          fTotalTracks;         // just for debugging
  UInt_t          fAcceptedTracks;      // just for debugging
  //
  //
  //
  // Setup
  //
   static       AliTPCParamSR* fgTPCparam;              //! helper object for geometry related operations
   static const Double_t       fgkM;                    // value used in the transformation of the charge values for the logarithmic fitter
   static const char*          fgkDebugStreamFileName;  // filename of the debug stream file
   static const Bool_t         fgkUseTotalCharge;       // whether to use the cluster's total or maximum charge

   ClassDef(AliTPCcalibTracksGain, 1);
};

#endif
Commit	Line	Data
10757ee9	1	#ifndef AliTPCCALIBTRACKSGAIN_H
	2	#define AliTPCCALIBTRACKSGAIN_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
b8601924	6
10757ee9	7	#include <TChain.h>
	8	#include <TNamed.h>
	9
	10
	11	#include <TObjArray.h>
	12	#include <TH2D.h>
	13	#include <TVectorD.h>
	14	#include <TMatrixD.h>
	15
	16	#include <iostream>
	17	#include <TH1F.h>
	18
b8601924	19	#include <AliTPCcalibBase.h>
b8601924	20
10757ee9	21	using namespace std;
	22
	23	class TTreeSRedirector;
c6f3749a	24	class TObjString;
10757ee9	25	class TLinearFitter;
c6f3749a	26	class TProfile;
	27	class TProfile2D;
	28	class TH1F;
10757ee9	29
	30	class AliTPCClusterParam;
	31	class AliTPCParamSR;
	32	class AliTPCCalROC;
	33	class AliTPCCalPad;
	34	class AliTPCseed;
	35	class AliTPCclusterMI;
	36	class AliTrackPointArray;
	37	class TTreeStream;
	38	class AliTPCcalibTracksCuts;
	39	class AliTPCFitPad;
8076baa0	40	class TGraph;
8076baa0	41	class AliTPCCClusterParam;
10757ee9	42
b8601924	43	class AliTPCcalibTracksGain : public AliTPCcalibBase {
10757ee9	44	public:
	45	enum {
	46	kShortPads = 0,
	47	kMediumPads = 1,
	48	kLongPads = 2
	49	};
	50	enum {
	51	kSimpleFitter = 0,
	52	kSqrtFitter = 1,
	53	kLogFitter = 2
	54	};
	55
c6f3749a	56	AliTPCcalibTracksGain();
c6f3749a	57	AliTPCcalibTracksGain(const AliTPCcalibTracksGain& obj);
0ffd2ae1	58	AliTPCcalibTracksGain(const char* name, const char* title, AliTPCcalibTracksCuts* cuts);
c6f3749a	59	virtual ~AliTPCcalibTracksGain();
c6f3749a	60	AliTPCcalibTracksGain& operator=(const AliTPCcalibTracksGain& rhs);
b8601924	61	virtual Long64_t Merge(TCollection *list);
	62	virtual void Process(AliTPCseed* seed);
	63	virtual void Terminate();
	64	virtual void Analyze();
c6f3749a	65	//
2acad464	66	// Tracks and cluster manipulation
c6f3749a	67	//
489dce1b	68	Float_t GetGain(AliTPCclusterMI *cl);
	69	Float_t GetMaxNorm(AliTPCclusterMI * cl, Float_t ky, Float_t kz);
	70	Float_t GetQNorm(AliTPCclusterMI * cl, Float_t ky, Float_t kz);
c6f3749a	71	void AddTrack(AliTPCseed* seed);
c6f3749a	72	void DumpTrack(AliTPCseed* track);
	73	Bool_t GetDedx(AliTPCseed* track, Int_t padType, Int_t*rows,
	74	Int_t &sector, Int_t& npoints,
	75	TVectorD &dedxM, TVectorD &dedxQ,
	76	TVectorD &parY, TVectorD &parZ, TVectorD & meanPos);
	77	void AddCluster(AliTPCclusterMI* cluster, Float_t momenta, Float_t mdedx, Int_t padType, Float_t xcenter, TVectorD &dedxQ, TVectorD & dedxM, Float_t fraction, Float_t fraction2, Float_t dedge, TVectorD& parY, TVectorD& parZ, TVectorD& meanPos);
	78	void AddTracklet(UInt_t sector, UInt_t padType,TVectorD &dedxQ, TVectorD &dedxM,TVectorD& parY, TVectorD& parZ, TVectorD& meanPos);
10757ee9	79
c6f3749a	80	void Add(AliTPCcalibTracksGain* cal);
da7d274e	81
c6f3749a	82	//
	83	// Get Derived results - gain maps
	84	//
	85	AliTPCCalPad* CreateFitCalPad(UInt_t fitType, Bool_t undoTransformation = kFALSE, Bool_t normalizeToPadSize = kFALSE);
	86	AliTPCCalROC* CreateFitCalROC(UInt_t sector, UInt_t fitType, Bool_t undoTransformation = kFALSE, Bool_t normalizeToPadSize = kFALSE);
	87	AliTPCCalROC* CreateFitCalROC(UInt_t sector, UInt_t padType, TVectorD &fitParam, UInt_t fitType, Bool_t undoTransformation = kFALSE, Bool_t normalizeToPadSize = kFALSE);
	88	AliTPCCalROC* CreateCombinedCalROC(const AliTPCCalROC* roc1, const AliTPCCalROC* roc2);
	89	//
	90	void Evaluate(Bool_t robust = kFALSE, Double_t frac = -1.);
684602c8	91	Bool_t GetParameters(UInt_t segment, UInt_t padType, UInt_t fitType, TVectorD &fitParam);
c6f3749a	92	void GetErrors(UInt_t segment, UInt_t padType, UInt_t fitType, TVectorD &fitError);
c6f3749a	93	void GetCovarianceMatrix(UInt_t segment, UInt_t padType, UInt_t fitType, TMatrixD& covMatrix);
8076baa0	94	//
	95	//
	96	void UpdateClusterParam(AliTPCClusterParam *param);
	97	TGraph * CreateAmpGraph(Int_t ipad, Bool_t qmax);
	98
	99
1aa15e8d	100	TLinearFitter* GetFitter(UInt_t segment, UInt_t padType, UInt_t fitType);
	101	void Process(AliESDEvent *event) {AliTPCcalibBase::Process(event);};
	102	void Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
b8601924	103
c6f3749a	104	public:
	105	//
	106	// Helper function
	107	//
	108	static Double_t GetPadLength(Double_t lx);
	109	static Int_t GetPadType(Double_t lx);
c6f3749a	110	//
c6f3749a	111	//
c6f3749a	112	AliTPCcalibTracksCuts* fCuts; // cuts that are used for sieving the tracks used for calibration
c6f3749a	113	//
489dce1b	114	// kalman fit - alignment of dEdx
c6f3749a	115	//
c6f3749a	116	//
	117	// Fitters
	118	//
	119	AliTPCFitPad* fSimpleFitter; // simple fitter for short pads
	120	AliTPCFitPad* fSqrtFitter; // sqrt fitter for medium pads
	121	AliTPCFitPad* fLogFitter; // log fitter for long pads
684602c8	122	//
c6f3749a	123	TLinearFitter* fFitter0M; // fitting of the atenuation, angular correction, and mean chamber gain
	124	TLinearFitter* fFitter1M; // fitting of the atenuation, angular correction, and mean chamber gain
	125	TLinearFitter* fFitter2M; // fitting of the atenuation, angular correction, and mean chamber gain
	126	TLinearFitter* fFitter0T; // fitting of the atenuation, angular correction, and mean chamber gain
	127	TLinearFitter* fFitter1T; // fitting of the atenuation, angular correction, and mean chamber gain
	128	TLinearFitter* fFitter2T; // fitting of the atenuation, angular correction, and mean chamber gain
684602c8	129	//
	130	// angular adn diffusion effect fitter
	131	//
	132	TLinearFitter* fDFitter0M; // fitting of the atenuation, angular correction
	133	TLinearFitter* fDFitter1M; // fitting of the atenuation, angular correction
	134	TLinearFitter* fDFitter2M; // fitting of the atenuation, angular correction
	135	TLinearFitter* fDFitter0T; // fitting of the atenuation, angular correction
	136	TLinearFitter* fDFitter1T; // fitting of the atenuation, angular correction
	137	TLinearFitter* fDFitter2T; // fitting of the atenuation, angular correction
	138	//
489dce1b	139	AliTPCFitPad* fSingleSectorFitter; // just for debugging
c6f3749a	140	//
	141	// Conters
	142	//
	143	UInt_t fTotalTracks; // just for debugging
	144	UInt_t fAcceptedTracks; // just for debugging
c6f3749a	145	//
c6f3749a	146	//
c6f3749a	147	//
	148	// Setup
	149	//
10757ee9	150	static AliTPCParamSR* fgTPCparam; //! helper object for geometry related operations
	151	static const Double_t fgkM; // value used in the transformation of the charge values for the logarithmic fitter
	152	static const char* fgkDebugStreamFileName; // filename of the debug stream file
	153	static const Bool_t fgkUseTotalCharge; // whether to use the cluster's total or maximum charge
	154
	155	ClassDef(AliTPCcalibTracksGain, 1);
	156	};
	157
	158	#endif