15 |
#include <TH2F.h> |
#include <TH2F.h> |
16 |
#include <TTimeStamp.h> |
#include <TTimeStamp.h> |
17 |
|
|
18 |
/*! @brief The tracker chi2 vs deflection cut. |
/*! @brief The tracker chi2 vs |deflection| cut. |
19 |
* |
* |
20 |
* Events whose track has been fitted with a chi2 greater than Chi2(eta) are discarded. |
* Events whose track has been fitted with a chi2 greater than Chi2(|eta|) are discarded. |
21 |
* Chi2(eta) is a function which gives the maximum value of chi2 associated to a certain |
* Chi2(|eta|) is a function which gives the value of chi2 distribution, corresponding to a given quantile (depending on the calibration text file), as a function of |deflection| (|eta|); in current implementation it is parameterized as: |
|
* deflection eta; in current implementation it is parameterized as: |
|
22 |
* |
* |
23 |
* Chi2(eta) = p0 + p1 * eta^2 + p2 * eta ^4 |
* 4 parameters p0...p3: |
24 |
* |
* |
25 |
* where p0, p1 and p2 are parameters. These are read from a file, and are supposed to be |
* Chi2(|eta|) = p0 + ( p1 * |eta|^p2 * (1 + (p3 * |eta|)^2) ) |
26 |
* computed month by month. |
* with eta expressed in GV^-1 |
27 |
|
* |
28 |
|
* 5 parameters p0..p4 |
29 |
|
* |
30 |
|
* Chi2(|eta|) = p0 + ( p1 * |eta|^p2 * (1 + (p3 * |eta|)^p4) ) |
31 |
|
* with eta expressed in GV^-1 |
32 |
|
* |
33 |
|
* The parameters are read from a calibration text file (prepared for a given quantile, e.g. 95%), whose format must be: |
34 |
|
* ... |
35 |
|
* first_day last_day p0 p1 p2 ... |
36 |
|
* with first_day and last_day expressed in the format YYMMDD |
37 |
|
* ... |
38 |
|
* where each row corresponds to a given time interval (first_day,last_day) |
39 |
|
* The calibration text file can vary with nHitX of the event: in the current implementation, a text file must be specified for events with nHitX=3 and for events with nHitX>=4. |
40 |
|
* For a given event the associated day is determined; the FIRST (starting from first row) time interval of the calibration file, which contains the event day, gives the parameters to be inserted in the Chi2(|eta|) for the given event. |
41 |
|
* If the event day does not fall in any of the intervals of the calibration file, then the event is discarded. |
42 |
* |
* |
43 |
*/ |
*/ |
44 |
|
|
48 |
/*! @brief Constructor. |
/*! @brief Constructor. |
49 |
* |
* |
50 |
* @param cutName The cut's name. |
* @param cutName The cut's name. |
51 |
* @param calibFile Path for the calibration file. The format of the file's rows is: |
* @param calibFile_nHitX3 Path for the calibration text file, used for events with nHitX=3. |
52 |
* YYMM efficiency p0 p1 p2 |
* @param calibFile_nHitX4 Path for the calibration text file, used for events with nHitX>=4. |
53 |
|
* @param nPar Number of parameters used for the Chi2 function. Maximum allowed: 5. Default: 5. |
54 |
*/ |
*/ |
55 |
TrkChi2DeflTimeCut(const char *cutName, const char *calibFile) : |
TrkChi2DeflTimeCut(const char *cutName, const char *trkAlg, const char *calibFile_nHitX3, const char *calibFile_nHitX4, int nPar=5) : |
56 |
PamCut(cutName), _calibFile(calibFile) { |
PamCut(cutName), _trkAlg(trkAlg), _calibFile_nHitX3(calibFile_nHitX3), _calibFile_nHitX4(calibFile_nHitX4), _nPar(nPar) { |
57 |
|
|
58 |
_chi2CutTable.open(_calibFile); |
_chi2CutTable_nHitX3.open(_calibFile_nHitX3); |
59 |
while (! _chi2CutTable.eof()) { |
while (! _chi2CutTable_nHitX3.eof()) { |
60 |
|
|
61 |
_chi2CutTable >> _tstringtemp; |
_chi2CutTable_nHitX3 >> _tstringtemp; |
62 |
_iMonth.push_back(_tstringtemp.Atoi()); |
_iDayFirst_nHitX3.push_back(_tstringtemp.Atoi()); |
|
|
|
|
_chi2CutTable >> _tstringtemp; |
|
|
|
|
|
_chi2CutTable >> _tstringtemp; |
|
|
_p0.push_back(_tstringtemp.Atof()); |
|
|
|
|
|
_chi2CutTable >> _tstringtemp; |
|
|
_p1.push_back(_tstringtemp.Atof()); |
|
63 |
|
|
64 |
_chi2CutTable >> _tstringtemp; |
_chi2CutTable_nHitX3 >> _tstringtemp; |
65 |
_p2.push_back(_tstringtemp.Atof()); |
_iDayLast_nHitX3.push_back(_tstringtemp.Atoi()); |
66 |
|
|
67 |
|
if (_nPar>=0 && _nPar<=5) { |
68 |
|
for (int iPar=0; iPar<_nPar; iPar++) { |
69 |
|
|
70 |
|
_chi2CutTable_nHitX3 >> _tstringtemp; |
71 |
|
_p_nHitX3[iPar].push_back(_tstringtemp.Atof()); |
72 |
|
|
73 |
|
} |
74 |
|
} |
75 |
|
|
76 |
} |
} |
77 |
_chi2CutTable.close(); |
_chi2CutTable_nHitX3.close(); |
|
|
|
|
#ifdef DEBUGPAMCUT |
|
78 |
|
|
79 |
TString hId; |
_chi2CutTable_nHitX4.open(_calibFile_nHitX4); |
80 |
TString hTitle; |
while (! _chi2CutTable_nHitX4.eof()) { |
81 |
|
|
82 |
|
_chi2CutTable_nHitX4 >> _tstringtemp; |
83 |
|
_iDayFirst_nHitX4.push_back(_tstringtemp.Atoi()); |
84 |
|
|
85 |
for (UInt_t j = 0; j < 2; j++) { |
_chi2CutTable_nHitX4 >> _tstringtemp; |
86 |
hId.Form("h_trk_chi2_vs_defl_%i_", j); |
_iDayLast_nHitX4.push_back(_tstringtemp.Atoi()); |
87 |
hId.Append(TString(cutName)); |
|
88 |
hTitle.Form("TRK chi2 vs defl (%i)", j); |
if (_nPar>=0 && _nPar<=5) { |
89 |
h_trk_chi2_defl[j] = new TH2F(hId.Data(), hTitle.Data(), 50, 0, 2.5, 50, 0, 20); |
for (int iPar=0; iPar<_nPar; iPar++) { |
90 |
|
|
91 |
|
_chi2CutTable_nHitX4 >> _tstringtemp; |
92 |
|
_p_nHitX4[iPar].push_back(_tstringtemp.Atof()); |
93 |
|
|
94 |
|
} |
95 |
|
} |
96 |
|
|
97 |
} |
} |
98 |
|
_chi2CutTable_nHitX4.close(); |
99 |
|
|
|
#endif |
|
100 |
|
|
101 |
} |
} |
102 |
|
|
103 |
/*! @brief Destructor. */ |
/*! @brief Destructor. */ |
104 |
~TrkChi2DeflTimeCut() { |
~TrkChi2DeflTimeCut() { |
105 |
|
|
|
#ifdef DEBUGPAMCUT |
|
|
|
|
|
for (UInt_t j = 0; j < 2; j++) { |
|
|
h_trk_chi2_defl[j]->Write(); |
|
|
} |
|
|
|
|
|
#endif |
|
|
|
|
106 |
} |
} |
107 |
|
|
108 |
/*! @brief The tracker chi2 vs deflection check. |
/*! @brief The tracker chi2 vs |deflection| check. |
109 |
* |
* |
110 |
* |
* |
111 |
* @param event The event to analyze. |
* @param event The event to analyze. |
112 |
* @return #CUTOK if chi2 < Chi2(eta). |
* @return #CUTOK if chi2 < Chi2(|eta|). |
113 |
* @return 0 if not |
* @return 0 if not |
114 |
*/ |
*/ |
115 |
int Check(PamLevel2 *event); |
int Check(PamLevel2 *event); |
116 |
|
|
117 |
private: |
protected: |
118 |
|
|
119 |
#ifdef DEBUGPAMCUT |
const char *_trkAlg; |
120 |
|
|
|
TH2F* h_trk_chi2_defl[2]; |
|
|
|
|
|
#endif |
|
121 |
|
|
122 |
|
private: |
123 |
|
|
124 |
virtual double _GetChi2(PamLevel2 *event); |
virtual double _GetChi2(PamLevel2 *event); |
125 |
|
|
126 |
const char* _calibFile; |
const char* _calibFile_nHitX3; |
127 |
|
const char* _calibFile_nHitX4; |
128 |
|
Int_t _nPar; |
129 |
|
|
130 |
TString _tstringtemp; |
TString _tstringtemp; |
131 |
|
|
132 |
std::vector<Int_t> _iMonth; |
std::vector<Int_t> _iDayFirst_nHitX3; |
133 |
std::vector<Double_t> _p0; |
std::vector<Int_t> _iDayFirst_nHitX4; |
134 |
std::vector<Double_t> _p1; |
std::vector<Int_t> _iDayFirst; |
135 |
std::vector<Double_t> _p2; |
|
136 |
|
std::vector<Int_t> _iDayLast_nHitX3; |
137 |
ifstream _chi2CutTable; |
std::vector<Int_t> _iDayLast_nHitX4; |
138 |
|
std::vector<Int_t> _iDayLast; |
139 |
|
|
140 |
|
std::vector<Double_t> _p_nHitX3[5]; |
141 |
|
std::vector<Double_t> _p_nHitX4[5]; |
142 |
|
std::vector<Double_t> _p[5]; |
143 |
|
|
144 |
|
ifstream _chi2CutTable_nHitX3; |
145 |
|
ifstream _chi2CutTable_nHitX4; |
146 |
|
|
147 |
Double_t _p0sel, _p1sel, _p2sel; |
Double_t _pSel[5]; |
148 |
|
|
149 |
TTimeStamp _time; |
TTimeStamp _time; |
150 |
|
|