TrkCuts/TrkChi2DeflTimeCut/TrkChi2DeflTimeCut.h

/*
 * TrkChi2DeflTimeCut.h
 *
 *  Created on: 6-oct-2009
 *      Author: S. Ricciarini
 */

/*! @file TrkChi2DeflTimeCut.h The TrkChi2DeflTimeCut class definition file */

#ifndef TRKCHI2DEFLTIMECUT_H_
#define TRKCHI2DEFLTIMECUT_H_

#include "../../PamCutBase/PamCutBase.h"

#include <TH2F.h>
#include <TTimeStamp.h>

/*! @brief The tracker chi2 vs |deflection| cut.
 *
 * Events whose track has been fitted with a chi2 greater than Chi2(|eta|) are discarded.
 * 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:
 *
 *   4 parameters p0...p3:
 *
 *   Chi2(|eta|) = p0 + ( p1 * |eta|^p2 * (1 + (p3 * |eta|)^2) )
 *   with eta expressed in GV^-1
 *
 *   5 parameters p0..p4
 *
 *   Chi2(|eta|) = p0 + ( p1 * |eta|^p2 * (1 + (p3 * |eta|)^p4) )
 *   with eta expressed in GV^-1
 *
 * The parameters are read from a calibration text file (prepared for a given quantile, e.g. 95%), whose format must be:
 * ...
 * first_day    last_day        p0      p1      p2      ...
 * with first_day and last_day expressed in the format YYMMDD
 * ...
 * where each row corresponds to a given time interval (first_day,last_day)
 * 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.
 * 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.
 * If the event day does not fall in any of the intervals of the calibration file, then the event is discarded. 
 *
 */

class TrkChi2DeflTimeCut: public PamCut {

public:
  /*! @brief Constructor.
   *
   * @param cutName The cut's name.
   * @param calibFile_nHitX3 Path for the calibration text file, used for events with nHitX=3.
   * @param calibFile_nHitX4 Path for the calibration text file, used for events with nHitX>=4.  
   * @param nPar Number of parameters used for the Chi2 function. Maximum allowed: 5. Default: 5.
   */
  TrkChi2DeflTimeCut(const char *cutName, const char *trkAlg, const char *calibFile_nHitX3, const char *calibFile_nHitX4, int nPar=5) :
    PamCut(cutName), _trkAlg(trkAlg), _calibFile_nHitX3(calibFile_nHitX3), _calibFile_nHitX4(calibFile_nHitX4), _nPar(nPar) {
 
    _chi2CutTable_nHitX3.open(_calibFile_nHitX3);
    while (! _chi2CutTable_nHitX3.eof()) {
    
      _chi2CutTable_nHitX3 >> _tstringtemp;
      _iDayFirst_nHitX3.push_back(_tstringtemp.Atoi());

      _chi2CutTable_nHitX3 >> _tstringtemp;
      _iDayLast_nHitX3.push_back(_tstringtemp.Atoi());
      
      if (_nPar>=0 && _nPar<=5) {
        for (int iPar=0; iPar<_nPar; iPar++) {
      
          _chi2CutTable_nHitX3 >> _tstringtemp;
          _p_nHitX3[iPar].push_back(_tstringtemp.Atof());
        
        }
      }
            
    }
    _chi2CutTable_nHitX3.close();

    _chi2CutTable_nHitX4.open(_calibFile_nHitX4);
    while (! _chi2CutTable_nHitX4.eof()) {
    
      _chi2CutTable_nHitX4 >> _tstringtemp;
      _iDayFirst_nHitX4.push_back(_tstringtemp.Atoi());

      _chi2CutTable_nHitX4 >> _tstringtemp;
      _iDayLast_nHitX4.push_back(_tstringtemp.Atoi());
      
      if (_nPar>=0 && _nPar<=5) {
        for (int iPar=0; iPar<_nPar; iPar++) {
      
          _chi2CutTable_nHitX4 >> _tstringtemp;
          _p_nHitX4[iPar].push_back(_tstringtemp.Atof());
        
        }
      }
            
    }
    _chi2CutTable_nHitX4.close();


  }

  /*! @brief Destructor. */
  ~TrkChi2DeflTimeCut() {

  }

  /*! @brief The tracker chi2 vs |deflection| check.
   *
   *
   * @param event The event to analyze.
   * @return #CUTOK if chi2 < Chi2(|eta|).
   * @return 0 if not
   */
  int Check(PamLevel2 *event);

protected:

    const char *_trkAlg;


private: 

    virtual double _GetChi2(PamLevel2 *event);

    const char* _calibFile_nHitX3;
    const char* _calibFile_nHitX4;
    Int_t _nPar;

    TString _tstringtemp;
    
    std::vector<Int_t> _iDayFirst_nHitX3;
    std::vector<Int_t> _iDayFirst_nHitX4;
    std::vector<Int_t> _iDayFirst;
    
    std::vector<Int_t> _iDayLast_nHitX3;
    std::vector<Int_t> _iDayLast_nHitX4;
    std::vector<Int_t> _iDayLast;
    
    std::vector<Double_t> _p_nHitX3[5];
    std::vector<Double_t> _p_nHitX4[5];
    std::vector<Double_t> _p[5];
    
    ifstream _chi2CutTable_nHitX3;
    ifstream _chi2CutTable_nHitX4;
    
    Double_t _pSel[5];
    
    TTimeStamp _time;

};
#endif /* TRKCHI2DEFLTIMECUT_H_ */
1	/*
2	* TrkChi2DeflTimeCut.h
3	*
4	* Created on: 6-oct-2009
5	* Author: S. Ricciarini
6	*/
7
8	/! @file TrkChi2DeflTimeCut.h The TrkChi2DeflTimeCut class definition file /
9
10	#ifndef TRKCHI2DEFLTIMECUT_H_
11	#define TRKCHI2DEFLTIMECUT_H_
12
13	#include "../../PamCutBase/PamCutBase.h"
14
15	#include <TH2F.h>
16	#include <TTimeStamp.h>
17
18	/*! @brief The tracker chi2 vs \|deflection\| cut.
19	*
20	* Events whose track has been fitted with a chi2 greater than Chi2(\|eta\|) are discarded.
21	* 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:
22	*
23	* 4 parameters p0...p3:
24	*
25	* Chi2(\|eta\|) = p0 + ( p1 * \|eta\|^p2 * (1 + (p3 * \|eta\|)^2) )
26	* 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
45	class TrkChi2DeflTimeCut: public PamCut {
46
47	public:
48	/*! @brief Constructor.
49	*
50	* @param cutName The cut's name.
51	* @param calibFile_nHitX3 Path for the calibration text file, used for events with nHitX=3.
52	* @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 trkAlg, const char calibFile_nHitX3, const char calibFile_nHitX4, int nPar=5) :
56	PamCut(cutName), _trkAlg(trkAlg), _calibFile_nHitX3(calibFile_nHitX3), _calibFile_nHitX4(calibFile_nHitX4), _nPar(nPar) {
57
58	_chi2CutTable_nHitX3.open(_calibFile_nHitX3);
59	while (! _chi2CutTable_nHitX3.eof()) {
60
61	_chi2CutTable_nHitX3 >> _tstringtemp;
62	_iDayFirst_nHitX3.push_back(_tstringtemp.Atoi());
63
64	_chi2CutTable_nHitX3 >> _tstringtemp;
65	_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_nHitX3.close();
78
79	_chi2CutTable_nHitX4.open(_calibFile_nHitX4);
80	while (! _chi2CutTable_nHitX4.eof()) {
81
82	_chi2CutTable_nHitX4 >> _tstringtemp;
83	_iDayFirst_nHitX4.push_back(_tstringtemp.Atoi());
84
85	_chi2CutTable_nHitX4 >> _tstringtemp;
86	_iDayLast_nHitX4.push_back(_tstringtemp.Atoi());
87
88	if (_nPar>=0 && _nPar<=5) {
89	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
100
101	}
102
103	/! @brief Destructor. /
104	~TrkChi2DeflTimeCut() {
105
106	}
107
108	/*! @brief The tracker chi2 vs \|deflection\| check.
109	*
110	*
111	* @param event The event to analyze.
112	* @return #CUTOK if chi2 < Chi2(\|eta\|).
113	* @return 0 if not
114	*/
115	int Check(PamLevel2 *event);
116
117	protected:
118
119	const char *_trkAlg;
120
121
122	private:
123
124	virtual double _GetChi2(PamLevel2 *event);
125
126	const char* _calibFile_nHitX3;
127	const char* _calibFile_nHitX4;
128	Int_t _nPar;
129
130	TString _tstringtemp;
131
132	std::vector<Int_t> _iDayFirst_nHitX3;
133	std::vector<Int_t> _iDayFirst_nHitX4;
134	std::vector<Int_t> _iDayFirst;
135
136	std::vector<Int_t> _iDayLast_nHitX3;
137	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 _pSel[5];
148
149	TTimeStamp _time;
150
151	};
152	#endif /* TRKCHI2DEFLTIMECUT_H_ */