Fixes for building of DA (Anshul)
[u/mrichter/AliRoot.git] / TPC / AliTPCcalibTracksGain.h
CommitLineData
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>
20
10757ee9 21using namespace std;
22
23class TTreeSRedirector;
c6f3749a 24class TObjString;
10757ee9 25class TLinearFitter;
c6f3749a 26class TProfile;
27class TProfile2D;
28class TH1F;
10757ee9 29
30class AliTPCClusterParam;
31class AliTPCParamSR;
32class AliTPCCalROC;
33class AliTPCCalPad;
34class AliTPCseed;
35class AliTPCclusterMI;
36class AliTrackPointArray;
37class TTreeStream;
38class AliTPCcalibTracksCuts;
39class AliTPCFitPad;
8076baa0 40class TGraph;
41class AliTPCCClusterParam;
10757ee9 42
b8601924 43class AliTPCcalibTracksGain : public AliTPCcalibBase {
10757ee9 44public:
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();
57 AliTPCcalibTracksGain(const AliTPCcalibTracksGain& obj);
0ffd2ae1 58 AliTPCcalibTracksGain(const char* name, const char* title, AliTPCcalibTracksCuts* cuts);
c6f3749a 59 virtual ~AliTPCcalibTracksGain();
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
82 //
c6f3749a 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 104public:
105 //
106 // Helper function
107 //
108 static Double_t GetPadLength(Double_t lx);
109 static Int_t GetPadType(Double_t lx);
c6f3749a 110 //
111 //
c6f3749a 112 AliTPCcalibTracksCuts* fCuts; // cuts that are used for sieving the tracks used for calibration
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 //
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