/*
 * 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 maximum value of chi2 associated to a certain
 * deflection eta; in current implementation it is parameterized as:
 *
 *   Chi2(eta) = p0 + p1 * eta^2 + p2 * eta ^4
 *
 * where p0, p1 and p2 are parameters. These are read from a file, and are supposed to be
 * computed month by month.
 *
 */

class TrkChi2DeflTimeCut: public PamCut {

public:
  /*! @brief Constructor.
   *
   * @param cutName The cut's name.
   * @param calibFile Path for the calibration file. The format of the file's rows is:
   *                  YYMM efficiency p0 p1 p2
   */
  TrkChi2DeflTimeCut(const char *cutName, const char *calibFile) :
    PamCut(cutName), _calibFile(calibFile) {
 
    _chi2CutTable.open(_calibFile);
    while (! _chi2CutTable.eof()) {
    
      _chi2CutTable >> _tstringtemp;
      _iMonth.push_back(_tstringtemp.Atoi());
      
      _chi2CutTable >> _tstringtemp;
      
      _chi2CutTable >> _tstringtemp;
      _p0.push_back(_tstringtemp.Atof());
      
      _chi2CutTable >> _tstringtemp;
      _p1.push_back(_tstringtemp.Atof());

      _chi2CutTable >> _tstringtemp;
      _p2.push_back(_tstringtemp.Atof());      
      
    }
    _chi2CutTable.close();

#ifdef DEBUGPAMCUT

    TString hId;
    TString hTitle;

    for (UInt_t j = 0; j < 2; j++) {
      hId.Form("h_trk_chi2_vs_defl_%i_", j);
      hId.Append(TString(cutName));
      hTitle.Form("TRK chi2 vs defl (%i)", j);
      h_trk_chi2_defl[j] = new TH2F(hId.Data(), hTitle.Data(), 50, 0, 2.5, 50, 0, 20);
    }

#endif

  }

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

#ifdef DEBUGPAMCUT

    for (UInt_t j = 0; j < 2; j++) {
      h_trk_chi2_defl[j]->Write();
    }

#endif

  }

  /*! @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);

private:

#ifdef DEBUGPAMCUT

  TH2F* h_trk_chi2_defl[2];

#endif


    virtual double _GetChi2(PamLevel2 *event);

    const char* _calibFile;

    TString _tstringtemp;
    
    std::vector<Int_t> _iMonth;
    std::vector<Double_t> _p0;
    std::vector<Double_t> _p1;
    std::vector<Double_t> _p2;

    ifstream _chi2CutTable;
    
    Double_t _p0sel, _p1sel, _p2sel;
    
    TTimeStamp _time;

};
#endif /* TRKCHI2DEFLTIMECUT_H_ */