HadronCalibration.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 <string>
#include <sstream>
#include <fstream>
#include <math.h>
#include <iostream>
#include <stdlib.h>
#include <vector>
 
#include <TFile.h>
#include <TTree.h>
#include <TGraphErrors.h>
#include <TLine.h>
#include <TMultiGraph.h>
#include <TCanvas.h>
#include <TLegend.h>
#include <TF1.h>
 
#include <TStyle.h>
#include <TPaveText.h>
#include <framework/gearbox/Const.h>
#include <cdc/utilities/CDCDedxMeanPred.h>
#include <cdc/utilities/CDCDedxWidgetCurve.h>
#include <cdc/utilities/CDCDedxSigmaPred.h>
 
#include <cdc/calibration/CDCdEdx/HadronBgPrep.h>
 
namespace Belle2 {

  class HadronCalibration {
 
  public:

    HadronCalibration();

    virtual ~HadronCalibration() {};

    void fitBGCurve(const std::vector< std::string >& particles, const std::string& filename, const std::string& paramfile,
                    const std::string& suffx, const bool makeIterationSummary);

    void fitSigmavsIonz(const std::vector< std::string >& particles, const std::string& filename, const std::string& paramfile,
                        const std::string& suffix, const bool makeIterationSummary);

    static void fitSigmaVsNHit(const std::vector< std::string >& particles, const std::string& filename, const std::string& paramsigma,
                               const std::string& suffx, const bool makeIterationSummary);

    static void fitSigmaVsCos(const std::vector< std::string >& particles, const std::string& filename, const std::string& paramfile,
                              const std::string& suffx, const bool makeIterationSummary);

    void plotBGMonitoring(const std::vector<std::string >& particles, const std::string& filename, const std::string& suffix);

    void plotMonitoring(const std::vector< std::string >& particles, const std::string& filename, const std::string& sname,
                        const std::string& title,
                        const std::string& sx, const std::string& sy);

    static void setGraphStyle(TGraphErrors& gr, const int ic)
    {
      gr.SetMarkerColor(ic);
      gr.SetMarkerStyle(4);
      gr.SetMarkerSize(0.5);
    };

    static void setFitterStyle(TF1*& fitt, const int ic, const int il)
    {
      fitt->SetLineColor(ic);
      fitt->SetLineWidth(1);
      fitt->SetLineStyle(il);
    };
 
  private:
 
    HadronBgPrep m_prep; 
  };

} // namespace Belle2