CosmicRejBDTVars.cxx

Go to the documentation of this file.

#include "3FlavorAna/Vars/CosmicRejBDTVars.h"

#include "3FlavorAna/Vars/NumuVars.h"
#include "CAFAna/Vars/BPFVars.h"
#include "StandardRecord/Proxy/SRProxy.h"
#include "TMVA/Reader.h"
#include "Utilities/func/EnvExpand.h"

namespace ana
{
  // --- Some Vars that I want internally within my BDT computations.
  // --- Min Kal / Cos Trk Cell
  const Var kMinTrkCell([](const caf::SRProxy* sr) {
      float CosCell = kCosmicForwardCell(sr) + kCosmicBackwardCell(sr);
      float KalCell = kKalmanForwardCell(sr) + kKalmanBackwardCell(sr);
      return std::min( CosCell, KalCell );
    });

  // --- Kalman hit ratio.
  const Var kKalHitRat([](const caf::SRProxy* sr) {
      if(sr->trk.kalman.ntracks < 1) return -5.f;
      float TrkHit = kTrkNhits(sr);
      float SlcHit = kNHit(sr);
      return TrkHit / SlcHit;
    });
  // --- Max Kal Trk Start / End Y
  const Var kMaxKalTrStEnY([](const caf::SRProxy* sr) {
      return std::max(kTrkStartY(sr), kTrkEndY(sr) );
    });

  // --- BPF hit ratio.
  const Var kBPFHitRat([](const caf::SRProxy* sr) {
      float BPFHit = kBPFBestMuonNhits(sr);
      float SlcHit = kNHit(sr);
      if ( BPFHit == -5 )
        return BPFHit;
      else
        return BPFHit / SlcHit;
    });
  // --- Max BPF Start / End Y
  const Var kMaxBPFStEnY([](const caf::SRProxy* sr) {
      return std::max(kBPFBestMuonStarty(sr), kBPFBestMuonStopy(sr) );
    });

  // --- Is this a neutrino interaction?
  const Var kIsNuVar([](const caf::SRProxy* sr) {
      if (sr->mc.nnu == 0) return -1;
      if (sr->mc.nu[0].pdg > 0) { return 1; }
      else                      { return 0; }
    });
  // --- Is this a CC interaction?
  const Var kIsCCVar([](const caf::SRProxy* sr) {
      if (sr->mc.nnu == 0) return -1;
      if (sr->mc.nu[0].iscc) { return 1; }
      else                   { return 0; }
    });
  // --- Is this a NuMu interaction?
  const Var kIsNuMuVar([](const caf::SRProxy* sr) {
      if (sr->mc.nnu == 0) return -1;
      if (abs(sr->mc.nu[0].pdg) == 14) { return 1; }
      else                             { return 0; }
    });


  // ================ Period 1 BPF ================
  // FHC Period 1 BDT
  float  FHC_BPF_BDT_Per1_Vars[12] = {0};
  static TMVA::Reader* fReader_FHC_BPF_BDT_Per1 = 0;

  // ================ Period 2 BPF ================
  // FHC Period 2 BDT
  float  FHC_BPF_BDT_Per2_Vars[12] = {0};
  static TMVA::Reader* fReader_FHC_BPF_BDT_Per2 = 0;

  // ================ FHC High Gain BPF ===========
  // FHC High Gain BDT
  float  FHC_BPF_BDT_High_Vars[12] = {0};
  static TMVA::Reader* fReader_FHC_BPF_BDT_High = 0;

  // ================ RHC High Gain BPF ===========
  // RHC High Gain BDT
  float  RHC_BPF_BDT_High_Vars[12] = {0};
  static TMVA::Reader* fReader_RHC_BPF_BDT_High = 0;

  // ==============================================
  // ==============================================
  // ================ Period 1 Kal ================
  // FHC Period 1 BDT
  float  FHC_Kal_BDT_Per1_Vars[12] = {0};
  static TMVA::Reader* fReader_FHC_Kal_BDT_Per1 = 0;

  // ================ Period 2 Kal ================
  // FHC Period 2 BDT
  float  FHC_Kal_BDT_Per2_Vars[12] = {0};
  static TMVA::Reader* fReader_FHC_Kal_BDT_Per2 = 0;

  // ================ FHC High Gain Kal ===========
  // FHC High Gain BDT
  float  FHC_Kal_BDT_High_Vars[12] = {0};
  static TMVA::Reader* fReader_FHC_Kal_BDT_High = 0;

  // ================ RHC High Gain Kal ===========
  // RHC High Gain BDT
  float  RHC_Kal_BDT_High_Vars[12] = {0};
  static TMVA::Reader* fReader_RHC_Kal_BDT_High = 0;


  // -----------------------------------
  // ----- FHC_Per1_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_BPF_FHCPer1::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    FHC_BPF_BDT_Per1_Vars[0 ] = kMinTrkCell        (sr);
    FHC_BPF_BDT_Per1_Vars[1 ] = kBPFBestMuonCosNumi(sr);
    FHC_BPF_BDT_Per1_Vars[2 ] = kBPFBestMuonTrkLen (sr);
    FHC_BPF_BDT_Per1_Vars[3 ] = kMaxBPFStEnY       (sr);
    FHC_BPF_BDT_Per1_Vars[4 ] = kBPFBestMuonCosDirY(sr);
    FHC_BPF_BDT_Per1_Vars[5 ] = kBPFHitRat         (sr);
    FHC_BPF_BDT_Per1_Vars[6 ] = kBPFBestMuonPtP    (sr);
    // And my spectators
    FHC_BPF_BDT_Per1_Vars[7 ] = kIsNuVar (sr);
    FHC_BPF_BDT_Per1_Vars[8 ] = kIsCCVar (sr);
    FHC_BPF_BDT_Per1_Vars[9 ] = kCVNm    (sr);
    FHC_BPF_BDT_Per1_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    FHC_BPF_BDT_Per1_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_FHC_BPF_BDT_Per1)  InitTMVA();   
    score = fReader_FHC_BPF_BDT_Per1->EvaluateMVA("BDT");
    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_BPF_FHCPer1::InitTMVA() const
  {
    if(!fReader_FHC_BPF_BDT_Per1) fReader_FHC_BPF_BDT_Per1 = new TMVA::Reader;

    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_BPF_FHC_Per1_BDT.weights.xml";

    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &FHC_BPF_BDT_Per1_Vars[0 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "BPFCosNumi"                                         , &FHC_BPF_BDT_Per1_Vars[1 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "BPFLen"                                             , &FHC_BPF_BDT_Per1_Vars[2 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "max(BPFStY,BPFEndY)"                                , &FHC_BPF_BDT_Per1_Vars[3 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "cos(BPFDirY)"                                       , &FHC_BPF_BDT_Per1_Vars[4 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "BPFNHit/SliceNHit"                                  , &FHC_BPF_BDT_Per1_Vars[5 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddVariable( "BPFPtOverP"                                         , &FHC_BPF_BDT_Per1_Vars[6 ]);
    // And my spectators
    fReader_FHC_BPF_BDT_Per1 -> AddSpectator( "IsNu"       , &FHC_BPF_BDT_Per1_Vars[7 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddSpectator( "IsNuMu"     , &FHC_BPF_BDT_Per1_Vars[8 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddSpectator( "CVNMuon17"  , &FHC_BPF_BDT_Per1_Vars[9 ]);
    fReader_FHC_BPF_BDT_Per1 -> AddSpectator( "CVNMuon18"  , &FHC_BPF_BDT_Per1_Vars[10]);
    fReader_FHC_BPF_BDT_Per1 -> AddSpectator( "KalTrkRemID", &FHC_BPF_BDT_Per1_Vars[11]);
    fReader_FHC_BPF_BDT_Per1 -> BookMVA("BDT", pidlib);
  }

  // -----------------------------------
  // ----- FHC_Per2_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_BPF_FHCPer2::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    FHC_BPF_BDT_Per2_Vars[0 ] = kMinTrkCell        (sr);
    FHC_BPF_BDT_Per2_Vars[1 ] = kBPFBestMuonCosNumi(sr);
    FHC_BPF_BDT_Per2_Vars[2 ] = kBPFBestMuonTrkLen (sr);
    FHC_BPF_BDT_Per2_Vars[3 ] = kMaxBPFStEnY       (sr);
    FHC_BPF_BDT_Per2_Vars[4 ] = kBPFBestMuonCosDirY(sr);
    FHC_BPF_BDT_Per2_Vars[5 ] = kBPFHitRat         (sr);
    FHC_BPF_BDT_Per2_Vars[6 ] = kBPFBestMuonPtP    (sr);
    // And my spectators
    FHC_BPF_BDT_Per2_Vars[7 ] = kIsNuVar (sr);
    FHC_BPF_BDT_Per2_Vars[8 ] = kIsCCVar (sr);
    FHC_BPF_BDT_Per2_Vars[9 ] = kCVNm    (sr);
    FHC_BPF_BDT_Per2_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    FHC_BPF_BDT_Per2_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_FHC_BPF_BDT_Per2)  InitTMVA();   
    score = fReader_FHC_BPF_BDT_Per2->EvaluateMVA("BDT");
    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_BPF_FHCPer2::InitTMVA() const
  {
    if(!fReader_FHC_BPF_BDT_Per2) fReader_FHC_BPF_BDT_Per2 = new TMVA::Reader;
    
    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_BPF_FHC_Per2_BDT.weights.xml";

    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &FHC_BPF_BDT_Per2_Vars[0 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "BPFCosNumi"                                         , &FHC_BPF_BDT_Per2_Vars[1 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "BPFLen"                                             , &FHC_BPF_BDT_Per2_Vars[2 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "max(BPFStY,BPFEndY)"                                , &FHC_BPF_BDT_Per2_Vars[3 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "cos(BPFDirY)"                                       , &FHC_BPF_BDT_Per2_Vars[4 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "BPFNHit/SliceNHit"                                  , &FHC_BPF_BDT_Per2_Vars[5 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddVariable( "BPFPtOverP"                                         , &FHC_BPF_BDT_Per2_Vars[6 ]);
    // And my spectators
    fReader_FHC_BPF_BDT_Per2 -> AddSpectator( "IsNu"       , &FHC_BPF_BDT_Per2_Vars[7 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddSpectator( "IsNuMu"     , &FHC_BPF_BDT_Per2_Vars[8 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddSpectator( "CVNMuon17"  , &FHC_BPF_BDT_Per2_Vars[9 ]);
    fReader_FHC_BPF_BDT_Per2 -> AddSpectator( "CVNMuon18"  , &FHC_BPF_BDT_Per2_Vars[10]);
    fReader_FHC_BPF_BDT_Per2 -> AddSpectator( "KalTrkRemID", &FHC_BPF_BDT_Per2_Vars[11]);
    fReader_FHC_BPF_BDT_Per2 -> BookMVA("BDT", pidlib);
  }

  // -----------------------------------
  // ----- FHC_High_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_BPF_FHCHigh::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    FHC_BPF_BDT_High_Vars[0 ] = kMinTrkCell        (sr);
    FHC_BPF_BDT_High_Vars[1 ] = kBPFBestMuonCosNumi(sr);
    FHC_BPF_BDT_High_Vars[2 ] = kBPFBestMuonTrkLen (sr);
    FHC_BPF_BDT_High_Vars[3 ] = kMaxBPFStEnY       (sr);
    FHC_BPF_BDT_High_Vars[4 ] = kBPFBestMuonCosDirY(sr);
    FHC_BPF_BDT_High_Vars[5 ] = kBPFHitRat         (sr);
    FHC_BPF_BDT_High_Vars[6 ] = kBPFBestMuonPtP    (sr);
    // And my spectators
    FHC_BPF_BDT_High_Vars[7 ] = kIsNuVar (sr);
    FHC_BPF_BDT_High_Vars[8 ] = kIsCCVar (sr);
    FHC_BPF_BDT_High_Vars[9 ] = kCVNm    (sr);
    FHC_BPF_BDT_High_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    FHC_BPF_BDT_High_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_FHC_BPF_BDT_High)  InitTMVA();   
    score = fReader_FHC_BPF_BDT_High->EvaluateMVA("BDT");
    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_BPF_FHCHigh::InitTMVA() const
  {
    if(!fReader_FHC_BPF_BDT_High) fReader_FHC_BPF_BDT_High = new TMVA::Reader;
    
    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_BPF_FHC_High_BDT.weights.xml";

    fReader_FHC_BPF_BDT_High -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &FHC_BPF_BDT_High_Vars[0 ]);
    fReader_FHC_BPF_BDT_High -> AddVariable( "BPFCosNumi"                                         , &FHC_BPF_BDT_High_Vars[1 ]);
    fReader_FHC_BPF_BDT_High -> AddVariable( "BPFLen"                                             , &FHC_BPF_BDT_High_Vars[2 ]);
    fReader_FHC_BPF_BDT_High -> AddVariable( "max(BPFStY,BPFEndY)"                                , &FHC_BPF_BDT_High_Vars[3 ]);
    fReader_FHC_BPF_BDT_High -> AddVariable( "cos(BPFDirY)"                                       , &FHC_BPF_BDT_High_Vars[4 ]);
    fReader_FHC_BPF_BDT_High -> AddVariable( "BPFNHit/SliceNHit"                                  , &FHC_BPF_BDT_High_Vars[5 ]);
    fReader_FHC_BPF_BDT_High -> AddVariable( "BPFPtOverP"                                         , &FHC_BPF_BDT_High_Vars[6 ]);
    // And my spectators
    fReader_FHC_BPF_BDT_High -> AddSpectator( "IsNu"       , &FHC_BPF_BDT_High_Vars[7 ]);
    fReader_FHC_BPF_BDT_High -> AddSpectator( "IsNuMu"     , &FHC_BPF_BDT_High_Vars[8 ]);
    fReader_FHC_BPF_BDT_High -> AddSpectator( "CVNMuon17"  , &FHC_BPF_BDT_High_Vars[9 ]);
    fReader_FHC_BPF_BDT_High -> AddSpectator( "CVNMuon18"  , &FHC_BPF_BDT_High_Vars[10]);
    fReader_FHC_BPF_BDT_High -> AddSpectator( "KalTrkRemID", &FHC_BPF_BDT_High_Vars[11]);
    fReader_FHC_BPF_BDT_High -> BookMVA("BDT", pidlib);
  }

  // -----------------------------------
  // ----- RHC_High_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_BPF_RHCHigh::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    RHC_BPF_BDT_High_Vars[0 ] = kMinTrkCell        (sr);
    RHC_BPF_BDT_High_Vars[1 ] = kBPFBestMuonCosNumi(sr);
    RHC_BPF_BDT_High_Vars[2 ] = kBPFBestMuonTrkLen (sr);
    RHC_BPF_BDT_High_Vars[3 ] = kMaxBPFStEnY       (sr);
    RHC_BPF_BDT_High_Vars[4 ] = kBPFBestMuonCosDirY(sr);
    RHC_BPF_BDT_High_Vars[5 ] = kBPFHitRat         (sr);
    RHC_BPF_BDT_High_Vars[6 ] = kBPFBestMuonPtP    (sr);
    // And my spectators
    RHC_BPF_BDT_High_Vars[7 ] = kIsNuVar (sr);
    RHC_BPF_BDT_High_Vars[8 ] = kIsCCVar (sr);
    RHC_BPF_BDT_High_Vars[9 ] = kCVNm    (sr);
    RHC_BPF_BDT_High_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    RHC_BPF_BDT_High_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_RHC_BPF_BDT_High)  InitTMVA();   
    score = fReader_RHC_BPF_BDT_High->EvaluateMVA("BDT");
    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_BPF_RHCHigh::InitTMVA() const
  {
    if(!fReader_RHC_BPF_BDT_High) fReader_RHC_BPF_BDT_High = new TMVA::Reader;

    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_BPF_RHC_High_BDT.weights.xml";

    fReader_RHC_BPF_BDT_High -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &RHC_BPF_BDT_High_Vars[0 ]);
    fReader_RHC_BPF_BDT_High -> AddVariable( "BPFCosNumi"                                         , &RHC_BPF_BDT_High_Vars[1 ]);
    fReader_RHC_BPF_BDT_High -> AddVariable( "BPFLen"                                             , &RHC_BPF_BDT_High_Vars[2 ]);
    fReader_RHC_BPF_BDT_High -> AddVariable( "max(BPFStY,BPFEndY)"                                , &RHC_BPF_BDT_High_Vars[3 ]);
    fReader_RHC_BPF_BDT_High -> AddVariable( "cos(BPFDirY)"                                       , &RHC_BPF_BDT_High_Vars[4 ]);
    fReader_RHC_BPF_BDT_High -> AddVariable( "BPFNHit/SliceNHit"                                  , &RHC_BPF_BDT_High_Vars[5 ]);
    fReader_RHC_BPF_BDT_High -> AddVariable( "BPFPtOverP"                                         , &RHC_BPF_BDT_High_Vars[6 ]);
    // And my spectators
    fReader_RHC_BPF_BDT_High -> AddSpectator( "IsNu"       , &RHC_BPF_BDT_High_Vars[7 ]);
    fReader_RHC_BPF_BDT_High -> AddSpectator( "IsNuMu"     , &RHC_BPF_BDT_High_Vars[8 ]);
    fReader_RHC_BPF_BDT_High -> AddSpectator( "CVNMuon17"  , &RHC_BPF_BDT_High_Vars[9 ]);
    fReader_RHC_BPF_BDT_High -> AddSpectator( "CVNMuon18"  , &RHC_BPF_BDT_High_Vars[10]);
    fReader_RHC_BPF_BDT_High -> AddSpectator( "KalTrkRemID", &RHC_BPF_BDT_High_Vars[11]);
    fReader_RHC_BPF_BDT_High -> BookMVA("BDT", pidlib);
  }

  // ==============================================
  // ==============================================
  //            Kalman Track algorithms
  // ==============================================
  // ==============================================

  // -----------------------------------
  // ----- FHC_Per1_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_Kal_FHCPer1::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    FHC_Kal_BDT_Per1_Vars[0 ] = kMinTrkCell   (sr);
    FHC_Kal_BDT_Per1_Vars[1 ] = kCosNumi      (sr);
    FHC_Kal_BDT_Per1_Vars[2 ] = kTrkLength    (sr);
    FHC_Kal_BDT_Per1_Vars[3 ] = kMaxKalTrStEnY(sr);
    FHC_Kal_BDT_Per1_Vars[4 ] = kcosDirY      (sr);
    FHC_Kal_BDT_Per1_Vars[5 ] = kKalHitRat    (sr);
    FHC_Kal_BDT_Per1_Vars[6 ] = kNumuMuonPtP  (sr);
    // And my spectators
    FHC_Kal_BDT_Per1_Vars[7 ] = kIsNuVar (sr);
    FHC_Kal_BDT_Per1_Vars[8 ] = kIsCCVar (sr);
    FHC_Kal_BDT_Per1_Vars[9 ] = kCVNm    (sr);
    FHC_Kal_BDT_Per1_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    FHC_Kal_BDT_Per1_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_FHC_Kal_BDT_Per1)  InitTMVA();   
    score = fReader_FHC_Kal_BDT_Per1->EvaluateMVA("BDT");
    
    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_Kal_FHCPer1::InitTMVA() const
  {
    if(!fReader_FHC_Kal_BDT_Per1) fReader_FHC_Kal_BDT_Per1 = new TMVA::Reader;

    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_Kal_FHC_Per1_BDT.weights.xml";

    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &FHC_Kal_BDT_Per1_Vars[0 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "KalTrkCosNumi"                                         , &FHC_Kal_BDT_Per1_Vars[1 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "KalTrkLen"                                             , &FHC_Kal_BDT_Per1_Vars[2 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "max(KalTrkStY,KalTrkEndY)"                             , &FHC_Kal_BDT_Per1_Vars[3 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "cos(KalTrkDirY)"                                       , &FHC_Kal_BDT_Per1_Vars[4 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "KalTrkNHit/SliceNHit"                                  , &FHC_Kal_BDT_Per1_Vars[5 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddVariable( "KalTrkPtOverP"                                         , &FHC_Kal_BDT_Per1_Vars[6 ]);
    // And my spectators
    fReader_FHC_Kal_BDT_Per1 -> AddSpectator( "IsNu"       , &FHC_Kal_BDT_Per1_Vars[7 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddSpectator( "IsNuMu"     , &FHC_Kal_BDT_Per1_Vars[8 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddSpectator( "CVNMuon17"  , &FHC_Kal_BDT_Per1_Vars[9 ]);
    fReader_FHC_Kal_BDT_Per1 -> AddSpectator( "CVNMuon18"  , &FHC_Kal_BDT_Per1_Vars[10]);
    fReader_FHC_Kal_BDT_Per1 -> AddSpectator( "KalTrkRemID", &FHC_Kal_BDT_Per1_Vars[11]);
    fReader_FHC_Kal_BDT_Per1 -> BookMVA("BDT", pidlib);
  }

  // -----------------------------------
  // ----- FHC_Per2_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_Kal_FHCPer2::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    FHC_Kal_BDT_Per2_Vars[0 ] = kMinTrkCell   (sr);
    FHC_Kal_BDT_Per2_Vars[1 ] = kCosNumi      (sr);
    FHC_Kal_BDT_Per2_Vars[2 ] = kTrkLength    (sr);
    FHC_Kal_BDT_Per2_Vars[3 ] = kMaxKalTrStEnY(sr);
    FHC_Kal_BDT_Per2_Vars[4 ] = kcosDirY      (sr);
    FHC_Kal_BDT_Per2_Vars[5 ] = kKalHitRat    (sr);
    FHC_Kal_BDT_Per2_Vars[6 ] = kNumuMuonPtP  (sr);
    // And my spectators
    FHC_Kal_BDT_Per2_Vars[7 ] = kIsNuVar (sr);
    FHC_Kal_BDT_Per2_Vars[8 ] = kIsCCVar (sr);
    FHC_Kal_BDT_Per2_Vars[9 ] = kCVNm    (sr);
    FHC_Kal_BDT_Per2_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    FHC_Kal_BDT_Per2_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_FHC_Kal_BDT_Per2)  InitTMVA();   
    score = fReader_FHC_Kal_BDT_Per2->EvaluateMVA("BDT");

    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_Kal_FHCPer2::InitTMVA() const
  {
    if(!fReader_FHC_Kal_BDT_Per2) fReader_FHC_Kal_BDT_Per2 = new TMVA::Reader;
    
    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_Kal_FHC_Per2_BDT.weights.xml";

    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &FHC_Kal_BDT_Per2_Vars[0 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "KalTrkCosNumi"                                         , &FHC_Kal_BDT_Per2_Vars[1 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "KalTrkLen"                                             , &FHC_Kal_BDT_Per2_Vars[2 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "max(KalTrkStY,KalTrkEndY)"                             , &FHC_Kal_BDT_Per2_Vars[3 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "cos(KalTrkDirY)"                                       , &FHC_Kal_BDT_Per2_Vars[4 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "KalTrkNHit/SliceNHit"                                  , &FHC_Kal_BDT_Per2_Vars[5 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddVariable( "KalTrkPtOverP"                                         , &FHC_Kal_BDT_Per2_Vars[6 ]);
    // And my spectators
    fReader_FHC_Kal_BDT_Per2 -> AddSpectator( "IsNu"       , &FHC_Kal_BDT_Per2_Vars[7 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddSpectator( "IsNuMu"     , &FHC_Kal_BDT_Per2_Vars[8 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddSpectator( "CVNMuon17"  , &FHC_Kal_BDT_Per2_Vars[9 ]);
    fReader_FHC_Kal_BDT_Per2 -> AddSpectator( "CVNMuon18"  , &FHC_Kal_BDT_Per2_Vars[10]);
    fReader_FHC_Kal_BDT_Per2 -> AddSpectator( "KalTrkRemID", &FHC_Kal_BDT_Per2_Vars[11]);
    fReader_FHC_Kal_BDT_Per2 -> BookMVA("BDT", pidlib);
  }

  // -----------------------------------
  // ----- FHC_High_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_Kal_FHCHigh::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    FHC_Kal_BDT_High_Vars[0 ] = kMinTrkCell   (sr);
    FHC_Kal_BDT_High_Vars[1 ] = kCosNumi      (sr);
    FHC_Kal_BDT_High_Vars[2 ] = kTrkLength    (sr);
    FHC_Kal_BDT_High_Vars[3 ] = kMaxKalTrStEnY(sr);
    FHC_Kal_BDT_High_Vars[4 ] = kcosDirY      (sr);
    FHC_Kal_BDT_High_Vars[5 ] = kKalHitRat    (sr);
    FHC_Kal_BDT_High_Vars[6 ] = kNumuMuonPtP  (sr);
    // And my spectators
    FHC_Kal_BDT_High_Vars[7 ] = kIsNuVar (sr);
    FHC_Kal_BDT_High_Vars[8 ] = kIsCCVar (sr);
    FHC_Kal_BDT_High_Vars[9 ] = kCVNm    (sr);
    FHC_Kal_BDT_High_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    FHC_Kal_BDT_High_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_FHC_Kal_BDT_High)  InitTMVA();   
    score = fReader_FHC_Kal_BDT_High->EvaluateMVA("BDT");
    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_Kal_FHCHigh::InitTMVA() const
  {
    if(!fReader_FHC_Kal_BDT_High) fReader_FHC_Kal_BDT_High = new TMVA::Reader;
    
    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_Kal_FHC_High_BDT.weights.xml";

    fReader_FHC_Kal_BDT_High -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &FHC_Kal_BDT_High_Vars[0 ]);
    fReader_FHC_Kal_BDT_High -> AddVariable( "KalTrkCosNumi"                                         , &FHC_Kal_BDT_High_Vars[1 ]);
    fReader_FHC_Kal_BDT_High -> AddVariable( "KalTrkLen"                                             , &FHC_Kal_BDT_High_Vars[2 ]);
    fReader_FHC_Kal_BDT_High -> AddVariable( "max(KalTrkStY,KalTrkEndY)"                             , &FHC_Kal_BDT_High_Vars[3 ]);
    fReader_FHC_Kal_BDT_High -> AddVariable( "cos(KalTrkDirY)"                                       , &FHC_Kal_BDT_High_Vars[4 ]);
    fReader_FHC_Kal_BDT_High -> AddVariable( "KalTrkNHit/SliceNHit"                                  , &FHC_Kal_BDT_High_Vars[5 ]);
    fReader_FHC_Kal_BDT_High -> AddVariable( "KalTrkPtOverP"                                         , &FHC_Kal_BDT_High_Vars[6 ]);
    // And my spectators
    fReader_FHC_Kal_BDT_High -> AddSpectator( "IsNu"       , &FHC_Kal_BDT_High_Vars[7 ]);
    fReader_FHC_Kal_BDT_High -> AddSpectator( "IsNuMu"     , &FHC_Kal_BDT_High_Vars[8 ]);
    fReader_FHC_Kal_BDT_High -> AddSpectator( "CVNMuon17"  , &FHC_Kal_BDT_High_Vars[9 ]);
    fReader_FHC_Kal_BDT_High -> AddSpectator( "CVNMuon18"  , &FHC_Kal_BDT_High_Vars[10]);
    fReader_FHC_Kal_BDT_High -> AddSpectator( "KalTrkRemID", &FHC_Kal_BDT_High_Vars[11]);
    fReader_FHC_Kal_BDT_High -> BookMVA("BDT", pidlib);

  }

  // -----------------------------------
  // ----- RHC_High_BDT
  // -----------------------------------
  // ---- Operator class
  float GetBDTCosRej_Kal_RHCHigh::operator() (const caf::SRProxy* sr) const
  {
    float score = -5.;

    RHC_Kal_BDT_High_Vars[0 ] = kMinTrkCell   (sr);
    RHC_Kal_BDT_High_Vars[1 ] = kCosNumi      (sr);
    RHC_Kal_BDT_High_Vars[2 ] = kTrkLength    (sr);
    RHC_Kal_BDT_High_Vars[3 ] = kMaxKalTrStEnY(sr);
    RHC_Kal_BDT_High_Vars[4 ] = kcosDirY      (sr);
    RHC_Kal_BDT_High_Vars[5 ] = kKalHitRat    (sr);
    RHC_Kal_BDT_High_Vars[6 ] = kNumuMuonPtP  (sr);
    // And my spectators
    RHC_Kal_BDT_High_Vars[7 ] = kIsNuVar (sr);
    RHC_Kal_BDT_High_Vars[8 ] = kIsCCVar (sr);
    RHC_Kal_BDT_High_Vars[9 ] = kCVNm    (sr);
    RHC_Kal_BDT_High_Vars[10] = -5; // kCVNm2017(sr) doesn't exist anymore
    RHC_Kal_BDT_High_Vars[11] = kRemID   (sr);
    
    // Init TMVA
    if(!fReader_RHC_Kal_BDT_High)  InitTMVA();   
    score = fReader_RHC_Kal_BDT_High->EvaluateMVA("BDT");

    return score;
  }
  // ---- InitTMVA class.
  void GetBDTCosRej_Kal_RHCHigh::InitTMVA() const
  {
    if(!fReader_RHC_Kal_BDT_High) fReader_RHC_Kal_BDT_High = new TMVA::Reader;

    std::string pidlib = util::EnvExpansion("$UCANA_LIB_PATH/")+"/TMVA_MProd5_Kal_RHC_High_BDT.weights.xml";

    fReader_RHC_Kal_BDT_High -> AddVariable( "min(CosFwdCell+CosBakCell,KalTrkFwdCell+KalTrkBakCell)", &RHC_Kal_BDT_High_Vars[0 ]);
    fReader_RHC_Kal_BDT_High -> AddVariable( "KalTrkCosNumi"                                         , &RHC_Kal_BDT_High_Vars[1 ]);
    fReader_RHC_Kal_BDT_High -> AddVariable( "KalTrkLen"                                             , &RHC_Kal_BDT_High_Vars[2 ]);
    fReader_RHC_Kal_BDT_High -> AddVariable( "max(KalTrkStY,KalTrkEndY)"                             , &RHC_Kal_BDT_High_Vars[3 ]);
    fReader_RHC_Kal_BDT_High -> AddVariable( "cos(KalTrkDirY)"                                       , &RHC_Kal_BDT_High_Vars[4 ]);
    fReader_RHC_Kal_BDT_High -> AddVariable( "KalTrkNHit/SliceNHit"                                  , &RHC_Kal_BDT_High_Vars[5 ]);
    fReader_RHC_Kal_BDT_High -> AddVariable( "KalTrkPtOverP"                                         , &RHC_Kal_BDT_High_Vars[6 ]);
    // And my spectators
    fReader_RHC_Kal_BDT_High -> AddSpectator( "IsNu"       , &RHC_Kal_BDT_High_Vars[7 ]);
    fReader_RHC_Kal_BDT_High -> AddSpectator( "IsNuMu"     , &RHC_Kal_BDT_High_Vars[8 ]);
    fReader_RHC_Kal_BDT_High -> AddSpectator( "CVNMuon17"  , &RHC_Kal_BDT_High_Vars[9 ]);
    fReader_RHC_Kal_BDT_High -> AddSpectator( "CVNMuon18"  , &RHC_Kal_BDT_High_Vars[10]);
    fReader_RHC_Kal_BDT_High -> AddSpectator( "KalTrkRemID", &RHC_Kal_BDT_High_Vars[11]);
    fReader_RHC_Kal_BDT_High -> BookMVA("BDT", pidlib);
  }

}