SVDdEdxValidationAlgorithm.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <calibration/CalibrationAlgorithm.h>
#include <TTree.h>
#include <map>
 
namespace Belle2 {
  class SVDdEdxValidationAlgorithm : public CalibrationAlgorithm {
  public:
    SVDdEdxValidationAlgorithm();
 
    virtual ~SVDdEdxValidationAlgorithm() {}
 
    void setMonitoringPlots(bool value = false) { m_isMakePlots = value; }
 
    void setNumROCpoints(const unsigned int& value) { m_NumROCpoints = value; }
 
    void setMinROCMomentum(const double& value) { m_MomLowROC = value; }
 
    void setMaxROCMomentum(const double& value) { m_MomHighROC = value; }
 
    void setNumEffBins(const unsigned int& value) { m_NumEffBins = value; }
 
    void setMaxEffMomentum(const double& value) { m_MomHighEff = value; }
 
    void setMinEvtsPerTree(const double& value) { m_MinEvtsPerTree = value; }
 
  protected:
    virtual EResult calibrate() override;
 
  private:
    bool m_isMakePlots;                                                           
    void PlotEfficiencyPlots(const TString& PIDDetectorsName, TTree* SignalTree, TString SignalWeightName, TString SignalVarName,
                             TString SignalVarNameFull, TTree* FakeTree, TString FakeWeightName, TString FakeVarName, TString FakeVarNameFull,
                             TString PIDVarName, TString PIDCut, unsigned int nbins, double MomLow, double MomHigh);
 
    void PlotROCCurve(TTree* SignalTree, TString SignalWeightName, TString SignalVarName, TString SignalVarNameFull, TTree* FakeTree,
                      TString FakeWeightName, TString FakeVarName, TString FakeVarNameFull, TString PIDVarName);
 
    TTree* LambdaMassFit(std::shared_ptr<TTree> preselTree);  
    TTree* DstarMassFit(std::shared_ptr<TTree> preselTree); 
    int m_MinEvtsPerTree = 100;   
    unsigned int m_NumROCpoints = 250;    
    double m_MomLowROC = 0.;    
    double m_MomHighROC = 7.;    
    unsigned int m_NumEffBins = 30;    
    double m_MomHighEff = 2.5;    
  };
} // namespace Belle2