CDCDedxCosineAlgorithm.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 <cdc/dbobjects/CDCDedxCosineCor.h>
#include <calibration/CalibrationAlgorithm.h>
#include <framework/database/DBObjPtr.h>
#include <TH1D.h>
#include <TLegend.h>
 
namespace Belle2 {

  class CDCDedxCosineAlgorithm : public CalibrationAlgorithm {
 
  public:
 
    static constexpr int m_kNGroups = 3; 

    CDCDedxCosineAlgorithm();

    virtual ~CDCDedxCosineAlgorithm() {}

    void setMethodSep(bool value = true) {isMethodSep = value;}

    void setMonitoringPlots(bool value = false) {isMakePlots = value;}

    void setMergePayload(bool value = true) {isMergePayload = value;}

    void setFitWidth(double value = 2.5) {m_sigLim = value;}

    void setCosineBins(unsigned int value = 100) {m_cosBin = value;}

    void setCosineRange(double min = -1.0, double max = 1.0) {m_cosMin = min; m_cosMax = max;}

    void setHistBins(int value = 250) {m_dedxBin = value;}

    void setHistRange(double min = 0.0, double max = 5.0) {m_dedxMin = min; m_dedxMax = max;}

    void setSuffix(const std::string& value) {m_suffix = value;}

    void getExpRunInfo();

    void defineHisto(std::vector<TH1D*>& hdedx, const std::string& tag, const std::string& chargeLabel);

    TH1D* defineCosthHist(const std::string& tag);

    void fitGaussianWithRange(TH1D*& temphist, TString& status);

    void createPayload(std::vector<double> cosine);

    void plotdedxHist(std::vector<TH1D*>& hDedxCos_all, std::vector<TH1D*>& hDedxCos_neg, std::vector<TH1D*>& hDedxCos_pos);

    void setHist(TH1D* h, int color, const char* title,
                 double ymin, double ymax, int marker = 20)
    {
      h->SetLineColor(color);
      h->SetLineWidth(2);
      h->SetMarkerColor(color);
      h->SetMarkerStyle(marker);
      h->SetMarkerSize(0.7);
      h->SetStats(0);
 
      if (strlen(title) > 0)
        h->SetTitle(title);
 
      h->GetYaxis()->SetRangeUser(ymin, ymax);
    }

    void plotCosThetaDist(TH1D* hCosth_all, TH1D* hCosth_pos, TH1D* hCosth_neg);

    void plotFitResults(const std::vector<std::vector<double>>& dedxAll,  const std::vector<std::vector<double>>& dedxNeg,
                        const std::vector<std::vector<double>>& dedxPos);

    void plotEventStats();

    void plotConstants();

    unsigned int getRepresentativeLayer(unsigned int igroup) const
    {
      static const std::array<unsigned int, m_kNGroups> repLayer = {1, 9, 17};
      return repLayer.at(igroup);
    }
 
  protected:

    virtual EResult calibrate() override;
 
  private:

 
    struct FitValues {
      double mean = 1.0;  
      double meanErr = 0.0;  
      double sigma = 0.0;  
      double sigmaErr = 0.0;  
      TString status;  
    };

    FitValues fitHistogram(TH1D*& hist);
 
    bool isMethodSep; 
    bool isMakePlots; 
    bool isMergePayload; 
    double m_sigLim; 
    unsigned int m_cosBin; 
    double m_cosMin; 
    double m_cosMax; 
 
    int m_dedxBin; 
    double m_dedxMin; 
    double m_dedxMax; 
 
    std::string m_suffix; 
 
    std::vector<std::vector<double>> m_coscors; 
 
    const std::array<std::string, m_kNGroups> m_label = {"SL0", "SL1", "SL2-8"}; 
 
    DBObjPtr<CDCDedxCosineCor> m_DBCosineCor; 
  };

} // namespace Belle2