--- PamCut/TrkCuts/TrkChi2DeflTimeCut/TrkChi2DeflTimeCut.cpp 2009/10/23 15:34:31 1.1 +++ PamCut/TrkCuts/TrkChi2DeflTimeCut/TrkChi2DeflTimeCut.cpp 2011/05/27 14:01:39 1.5 @@ -7,46 +7,69 @@ #include "TrkChi2DeflTimeCut.h" +double TrkChi2DeflTimeCut::_GetChi2(PamLevel2 *event) { + return event->GetTrack(0)->GetTrkTrack()->chi2; +} + int TrkChi2DeflTimeCut::Check(PamLevel2 *event) { - Double_t chi2 = event->GetTrack(0)->GetTrkTrack()->chi2; + double chi2 = _GetChi2(event); if (chi2 < 0.) return 0; - Double_t eta2 = event->GetTrack(0)->GetTrkTrack()->GetDeflection(); //Sign doesn't matter - eta2 *= eta2; // Now eta2 is deflection^2 - -#ifdef DEBUGPAMCUT - - h_trk_chi2_defl[0]->Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), pow(chi2,0.25)); - -#endif + Double_t etaMod = fabs(event->GetTrack(0)->GetTrkTrack()->GetDeflection()); _time.Set(event->GetOrbitalInfo()->absTime, kFALSE, 0, kFALSE); // TTimestamp::GetDate() returns date in format YYYYMMDD so to compare it - // with 4-digits dates (YYMM) we must subtract 20000000 and then divide by 100 (divide by integer) + // with 6-digits dates (YYMMDD) we must subtract 20000000 - Int_t iMonthSel = (_time.GetDate(kFALSE) - 20000000)/100; // note: GetDate returns an unsigned integer - - for (UInt_t i=0; i<_iMonth.size(); i++) { - if (_iMonth[i]==iMonthSel) { - _p0sel=_p0[i]; - _p1sel=_p1[i]; - _p2sel=_p2[i]; -// cout << "FOUND: month " << iMonthSel << endl; -// cout << _p0sel << " " << _p1sel << " " << _p2sel << endl; + Int_t iDaySel = (_time.GetDate(kFALSE) - 20000000); // note: GetDate returns an unsigned integer + + bool found = false; + for (UInt_t i = 0; i < _iDayFirst.size(); i++) { + if (_iDayFirst[i] <= iDaySel && iDaySel <= _iDayLast[i]) { + for (int iPar=0; iPar<_nPar; iPar++) { + _pSel[iPar] = _p[iPar][i]; + } + +// cout << "FOUND: day " << iDaySel << endl; +// cout << _iDayFirst[i] << " " << _iDayLast[i] << endl; +// cout << _pSel[0] << " " << _pSel[1] << " " << _pSel[2] << " " << _pSel[3] << endl; + + found = true; break; } } - - if (chi2 > _p0sel + _p1sel * eta2 + _p2sel * eta2 * eta2) { + + if (!found) return 0; + + double chi2max; + + if (_nPar==4) { + + chi2max=_pSel[0] + _pSel[1] * pow(etaMod, _pSel[2]) * (1. + pow(_pSel[3] * etaMod, 2)); + +// cout << "chi2max " << chi2max << endl; + + } + else if (_nPar==5) { + chi2max=_pSel[0] + _pSel[1] * pow(etaMod, _pSel[2]) * (1. + pow(_pSel[3] * etaMod, _pSel[4])); + } + else { + chi2max=-1; } -#ifdef DEBUGPAMCUT +// cout << "chi2 " << chi2 << endl; + + if (chi2 > chi2max) { + +// cout << "KO" << endl; + + return 0; - h_trk_chi2_defl[1]->Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), pow(chi2,0.25)); + } -#endif +// cout << "OK" << endl; return CUTOK;