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(std::vector< std::string > particles, const std::string& filename, const std::string& paramfile,
                    const std::string& suffx);
 
    void fitSigmavsIonz(std::vector< std::string > particles, const std::string& filename, const std::string& paramfile,
                        const std::string& suffix);
 
    void fitSigmaVsNHit(std::vector< std::string > particles, const std::string& filename, const std::string& paramsigma,
                        const std::string& suffx);
 
    void fitSigmaVsCos(std::vector< std::string > particles, const std::string& filename, const std::string& paramfile,
                       const std::string& suffx);
 
    void plotBGMonitoring(std::vector< std::string > particles, const std::string& filename, const std::string& suffix);
 
    void plotMonitoring(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);
 
    void setGraphStyle(TGraphErrors& gr, const int ic)
    {
      gr.SetMarkerColor(ic);
      gr.SetMarkerStyle(4);
      gr.SetMarkerSize(0.5);
    };
 
    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