re-activate contrib code
[u/mrichter/AliRoot.git] / TRD / AliTRDdEdxBaseUtils.h
CommitLineData
6951a056 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15//
16//
17//
18// Xianguo Lu
19// lu@physi.uni-heidelberg.de
20// Xianguo.Lu@cern.ch
21//
22/*
23grep " AliTRDdEdxBaseUtils::" AliTRDdEdxBaseUtils.cxx | grep "=" -v | grep -v "[6]" | grep -v printf |wc
24grep "(" AliTRDdEdxBaseUtils.h | grep ";" | grep -v grep | grep -v ClassDef | grep -v "{" | grep -v typedef | wc
25*/
26
27
28#ifndef ALITRDDEDXBASEUTILS_H
29#define ALITRDDEDXBASEUTILS_H
30
31#ifndef TVECTORD_H
32#include "TVectorD.h"
33#endif
34
35#ifndef THNSPARSE_H
36#include "THnBase.h"
37#endif
38
39#ifndef TTREESTREAM_H
40#include "TTreeStream.h"
41#endif
42
43class TH1D;
44class TH2D;
45class TObjArray;
46
47class AliESDEvent;
48class AliESDtrack;
49class AliTRDcluster;
50class AliTRDtrackV1;
51class AliTRDseedV1;
52
53class AliTRDdEdxBaseUtils
54{
55 public:
56 //===================================================================================
57 // Math and Histogram
58 //===================================================================================
6856b6a8 59 static void BinLogX(TAxis *axis);
34e8fe1d 60 static void GetCDFCuts(const TH1D *hh, Int_t ncut, Double_t cuts[], const Double_t cdfs[], Double_t thres);
61 static Double_t GetMeanRMS(Double_t nn, Double_t sum, Double_t w2s, Double_t * grms=0x0, Double_t * gerr=0x0);
62 static Double_t TruncatedMean(Int_t nx, const Double_t xdata[], Double_t lowfrac, Double_t highfrac, Double_t * grms=0x0, Double_t * gerr=0x0, Double_t *wws=0x0);
63 static Double_t TruncatedMean(const TH1 *hh, Double_t lowfrac, Double_t highfrac, Double_t * grms=0x0, Double_t * gerr=0x0);
64 static void FitSlicesY(const TH2D *hh, TH1D *&hnor, TH1D *&hmpv, TH1D *&hwid, TH1D *&hres, Double_t thres, Double_t lowfrac, Double_t highfrac);
6951a056 65
66 //===================================================================================
67 // TRD Analysis Fast Tool
68 //===================================================================================
69 static Int_t GetNtracklet(const AliESDEvent *esd);
70 static AliTRDtrackV1 * GetTRDtrackV1(const AliESDtrack * esdtrack);
71 static Bool_t IsInSameStack(const AliTRDtrackV1 *trdtrack);
ca66da8f 72 static AliTRDseedV1 * GetFirstTracklet(const AliTRDtrackV1 *trdtrack);
fe829aa0 73 static AliTRDseedV1 * GetLastTracklet(const AliTRDtrackV1 *trdtrack);
ca66da8f 74 static void GetFirstSectorStackMomentum(const AliTRDtrackV1 *trdtrack, Int_t & isec, Int_t & istk, Double_t & mom);
75 static Double_t GetRedefinedPhi(Double_t phi);
76 static Double_t Getdydx(const AliTRDseedV1 *tracklet);
77 static Double_t Getdzdx(const AliTRDseedV1 *tracklet);
78 static Double_t Getdldx(const AliTRDseedV1 *tracklet);
6951a056 79
80 //===================================================================================
81 // Detector, Data and Control Constant
82 //===================================================================================
83
84 static Int_t NTRDchamber(){return 18*5*6;} //540
85 static Int_t NTRDtimebin(){return NTRDchamber()*31;} //16740
34e8fe1d 86 static Int_t ToDetector(Int_t gtb);
87 static Int_t ToTimeBin(Int_t gtb);
88 static Int_t ToSector(Int_t gtb);
89 static Int_t ToStack(Int_t gtb);
90 static Int_t ToLayer(Int_t gtb);
91
92 static void CheckRunB(TString listrun1kg, Int_t run, TString & type);
93 static TString GetRunType(Int_t run);
94
95 static void SetQ0Frac(Double_t q0){ fgQ0Frac = q0; }
96 static void SetQ1Frac(Double_t q1){ fgQ1Frac = q1; }
97 static void SetTimeBinCountCut(Double_t tbc){ fgTimeBinCountCut = tbc; }
98 static void SetCalibTPCnclsCut(Int_t tpc){ fgCalibTPCnclsCut = tpc; }
99 static void SetExBOn(Bool_t kon){ fgExBOn = kon; }
100 static void SetPadGainOn(Bool_t kon){ fgPadGainOn = kon; }
101 static void SetQScale(Double_t scale){ fgQScale = scale; }
6951a056 102
103 static Double_t Q0Frac(){return fgQ0Frac;}
104 static Double_t Q1Frac(){return fgQ1Frac;}
105 static Double_t TimeBinCountCut(){return fgTimeBinCountCut;}
106 static Int_t CalibTPCnclsCut(){return fgCalibTPCnclsCut;}
107 static Bool_t IsExBOn(){return fgExBOn;}
108 static Bool_t IsPadGainOn(){return fgPadGainOn;}
109 static Double_t QScale(){return fgQScale;}
110
111 static void PrintControl();
112
113 //===================================================================================
114 // dEdx Parameterization
115 //===================================================================================
fe829aa0 116 static void FastFitdEdxTR(TH1 * hh);
117
ca66da8f 118 static Double_t ALEPH(const Double_t * xx, const Double_t * par);
6951a056 119 static Double_t MeandEdx(const Double_t * xx, const Double_t * par);
120 static Double_t MeanTR(const Double_t * xx, const Double_t * par);
121 static Double_t MeandEdxTR(const Double_t * xx, const Double_t * par);
122
34e8fe1d 123 static Double_t QMeanTPC(Double_t bg);
124 static Double_t Q0MeanTRDpp(Double_t bg);
125 static Double_t Q1MeanTRDpp(Double_t bg);
126 static Double_t Q0MeanTRDPbPb(Double_t bg);
127 static Double_t Q1MeanTRDPbPb(Double_t bg);
6951a056 128
129 typedef Double_t (*FFunc)(const Double_t *xx, const Double_t *par);
130
131 static Double_t MeandEdxLogx(const Double_t * xx, const Double_t * par){return ToLogx(MeandEdx, xx, par);}
132 static Double_t MeanTRLogx(const Double_t * xx, const Double_t * par){return ToLogx(MeanTR, xx, par);}
133 static Double_t MeandEdxTRLogx(const Double_t * xx, const Double_t * par){return ToLogx(MeandEdxTR, xx, par);}
134
135 private:
136 //dEdx Parameterization
137 static Double_t ToLogx(FFunc func, const Double_t * xx, const Double_t * par);
138
139 //Control Constant
140 static Double_t fgQ0Frac; //q0frac
141 static Double_t fgQ1Frac; //q1frac
142 static Double_t fgTimeBinCountCut; //tbcut
143 static Int_t fgCalibTPCnclsCut; //tpccut
144 static Bool_t fgExBOn; //exbon
145 static Bool_t fgPadGainOn; //pad gain
146 static Double_t fgQScale; //Qscale
147
148};
149
150#endif