CDCDedxWireGainAlgorithm.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 <algorithm>
#include <iostream>
#include <fstream>
#include <string>
#include <vector>
 
#include <TH1D.h>
#include <TCanvas.h>
#include <TMath.h>
#include <TLine.h>
#include <TStyle.h>
#include <TH1I.h>
 
#include <reconstruction/dbobjects/CDCDedxWireGain.h>
#include <reconstruction/dbobjects/CDCDedxBadWires.h>
#include <cdc/dbobjects/CDCGeometry.h>
#include <calibration/CalibrationAlgorithm.h>
#include <framework/database/DBObjPtr.h>
#include <framework/database/IntervalOfValidity.h>
 
#include <cdc/geometry/CDCGeometryPar.h>
#include <cdc/geometry/CDCGeometryParConstants.h>
 
const unsigned int c_nwireCDC = c_nSenseWires;
 
namespace Belle2 {
  class CDCDedxWireGainAlgorithm : public CalibrationAlgorithm {
 
  public:
 
    CDCDedxWireGainAlgorithm();
 
    virtual ~CDCDedxWireGainAlgorithm() {}
 
    void setMergePayload(bool value = true) {m_isMerge = value;}
 
    void enableExtraPlots(bool value = false) {m_isMakePlots = value;}
 
    void setWireBasedTruction(bool value = false) {m_isWireTruc = value;}
 
    void getExpRunInfo();
 
    void createPayload(const std::vector<double>& vdedx_tmeans);
 
    void  getTruncatedBins(TH1D* hdedxhit, unsigned int& binlow, unsigned int& binhigh);
 
    double getTruncationMean(TH1D* hdedxhit, int binlow, int binhigh);
 
    void plotLayerDist(std::array<TH1D*, 2> hdedxL);
 
    void plotWireDist(const std::vector<TH1D*>& hist, const std::vector<double>& vrel_mean);
 
    void plotWireGain(const std::vector<double>& vdedx_means, const std::vector<double>& vrel_mean, double layeravg);
 
    void plotLayerGain(const std::vector<double>& layermean, double layeravg);
 
    void plotWGPerLayer(const std::vector<double>& vdedx_means, const std::vector<double>& layermean, double layeravg);
 
    void plotEventStats();
 
  protected:
 
    virtual EResult calibrate() override;
 
 
  private:
 
    bool m_isMakePlots; 
    bool m_isMerge; 
    bool m_isWireTruc; 
    int m_dedxBins; 
    double m_dedxMin; 
    double m_dedxMax; 
    double m_truncMin; 
    double m_truncMax; 
    std::string m_suffix; 
    DBObjPtr<CDCDedxWireGain> m_DBWireGains; 
    DBObjPtr<CDCDedxBadWires> m_DBBadWires; 
    DBObjPtr<CDCGeometry> m_cdcGeo; 
  };
} // namespace Belle2