CDCDedxHadSatAlgorithm.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 <map>
#include <vector>
#include <iostream>
#include <fstream>
 
#include <calibration/CalibrationAlgorithm.h>
#include <framework/database/DBObjPtr.h>
#include <framework/gearbox/Const.h>
 
#include <cdc/dbobjects/CDCDedxHadronCor.h>
#include <cdc/calibration/CDCdEdx/HadronSaturation.h>
#include <cdc/calibration/CDCdEdx/HadronPrep.h>
 
namespace Belle2 {
  class CDCDedxHadSatAlgorithm : public CalibrationAlgorithm {
 
  public:
 
    CDCDedxHadSatAlgorithm();
 
    virtual ~CDCDedxHadSatAlgorithm() {}
 
    void setMonitoringPlots(bool value = false) {m_ismakePlots = value;}
 
    void setProtonCut(double value) {m_cut = value;}
 
    void setBGPars(const std::string& particle, double nbin, double min, double max)
    {
      m_bgpar[particle][0] = nbin;
      m_bgpar[particle][1] = min;
      m_bgpar[particle][2] = max;
    }
 
    void setCosPars(int bin, double min, double max) { m_cosBins = bin; m_cosMin = min; m_cosMax = max; }
 
    void getExpRunInfo();
 
    void createPayload();
 
    void prepareSample(std::vector< std::string >& particles, const std::string& filename, const std::string& sfx, bool correct);
 
  protected:
 
    virtual EResult calibrate() override;
 
  private:
 
    std::map<std::string, std::array<double, 3>> m_bgpar = {
      {"electron", {18, 600, 9600}},
      {"pion", {4, 2.0, 12.94}},
      {"kaon", {4, 1.10, 3.6}},
      {"muon", {8, 2.83, 28.83}},
      {"proton", {6, 0.44, 0.85}}
    };  
    int m_cosBins = 8; 
    double m_cosMin = 0; 
    double m_cosMax = 0.95; 
    double m_cut = 0.5;    
    bool m_ismakePlots; 
    std::string m_suffix; 
    DBObjPtr<CDCDedxHadronCor> m_DBHadronCor; 
  };
} // namespace Belle2