SVDdEdxCalibrationAlgorithm.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 <TH2F.h>
#include <tuple>
 
namespace Belle2 {
  class SVDdEdxCalibrationAlgorithm : public CalibrationAlgorithm {
  public:
    SVDdEdxCalibrationAlgorithm();
 
    virtual ~SVDdEdxCalibrationAlgorithm() {}
 
    void setMonitoringPlots(bool value = false) { m_isMakePlots = value; }
 
    void setNumDEdxBins(const int& value) { m_numDEdxBins = value; }
 
    void setNumPBins(const int& value) { m_numPBins = value; }
 
    void setDEdxCutoff(const double& value) { m_dedxCutoff = value; }
 
    void setMinEvtsPerTree(const double& value) { m_MinEvtsPerTree = value; }
 
  protected:
    virtual EResult calibrate() override;
 
  private:
    bool m_isMakePlots;                                                           
    TH2F LambdaMassFit(std::shared_ptr<TTree> preselTree);                        
    std::tuple<TH2F, TH2F, TH2F> DstarMassFit(std::shared_ptr<TTree> preselTree); 
    TH2F GammaHistogram(std::shared_ptr<TTree> preselTree);                       
    int m_numDEdxBins = 100;                                                 
    int m_numPBins = 69;                                                     
    double m_dedxCutoff = 5.e6;                                              
    int m_MinEvtsPerTree =
      100;                                                 
    std::vector<double> CreatePBinningScheme()
    {
      std::vector<double> pbins;
      pbins.reserve(m_numPBins + 1);
      pbins.push_back(0.0);
      pbins.push_back(0.05);
 
      for (int iBin = 2; iBin <= m_numPBins; iBin++) {
        if (iBin <= 19)
          pbins.push_back(0.025 + 0.025 * iBin);
        else if (iBin <= 59)
          pbins.push_back(pbins.at(19) + 0.05 * (iBin - 19));
        else
          pbins.push_back(pbins.at(59) + 0.3 * (iBin - 59));
      }
 
      return pbins;
    }
  };
} // namespace Belle2