HadronPrep.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 <TH1D.h>
#include <TH2F.h>
#include <TF1.h>
#include <TCanvas.h>
#include <TLine.h>
#include <TLegend.h>
#include<TGraphErrors.h>
 
#include <calibration/CalibrationAlgorithm.h>
#include <framework/database/DBObjPtr.h>
#include <cdc/utilities/CDCDedxHadSat.h>
#include <framework/gearbox/Const.h>
 
#include <cdc/calibration/CDCdEdx/HadronBgPrep.h>
 
namespace Belle2 {
  class HadronPrep {
 
  public:
 
    HadronPrep();
 
    virtual ~HadronPrep() {};
 
    HadronPrep(int bgbins, double upperbg, double lowerbg, int cosbins, double uppercos, double lowercos, double cut);
 
    void prepareSample(std::shared_ptr<TTree> hadron, TFile*& outfile, const std::string& suffix,
                       const std::string& pdg, bool ismakePlots, bool correct);
 
    void defineHisto(std::vector<TH1F*>& htemp, const std::string& title, const std::string& pdg);
 
    void plotDist(std::map<int, std::vector<TH1F*>>& hist, const std::string& sname, const std::string& pdg);
 
    void plotDist(std::vector<TH2F*>& hist, const std::string& sname, const std::string& pdg);
 
    void setPars(TFile*& outfile, std::map<int, std::vector<TH1F*>>& hdedx_bgcosth, const std::string& pdg);
 
    void plotGraph(const std::string& sname, const std::string& pdg);
 
    void clear();
 
  private:
 
    std::map<int, std::vector<double>> m_sumcos;  
    std::map<int, std::vector<double>> m_sumbg; 
    std::map<int, std::vector<double>> m_means; 
    std::map<int, std::vector<double>> m_errors;  
    std::map<int, std::vector<int>> m_sumsize; 
    double m_dedxmax = 0.0; 
    double m_dedxmin = 99999999.0; 
    int m_bgBins; 
    double m_bgMin; 
    double m_bgMax; 
    int m_cosBins; 
    double m_cosMin; 
    double m_cosMax; 
    double m_cut;  
  };
} // namespace Belle2