CDCDedxValidationAlgorithm.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 <calibration/CalibrationAlgorithm.h>
#include <framework/database/DBObjPtr.h>
 
#include <cdc/dbobjects/CDCGeometry.h>
#include <cdc/geometry/CDCGeometryPar.h>
#include <cdc/geometry/CDCGeometryParConstants.h>
 
#include <cdc/dbobjects/CDCDedxInjectionTime.h>
 
#include <vector>
#include <string>
 
#include <TPaveText.h>
#include <TH1D.h>
#include <TCanvas.h>
 

struct WireGainData {
  std::vector<double> wiregain;

  std::vector<double> layermean;
};

struct CosGainData {
  std::array<std::vector<double>, 3> cosgain;

  std::vector<double> costh;
};

struct OnedData {
  std::array<std::vector<double>, 3> oneDcorr;

  std::vector<double> enta;
};
 
namespace Belle2 {

  class CDCDedxValidationAlgorithm : public CalibrationAlgorithm {
 
  public:

    CDCDedxValidationAlgorithm();

    virtual ~CDCDedxValidationAlgorithm() {}

    void printCanvasRun(std::map<int, TH1D*>& htemp, std::string namesfx);

    void printCanvas(std::vector<TH1D*>& htemp, std::string namesfx, std::string svar);

    void fitGaussianWRange(TH1D*& temphist, std::string& status);

    void getExpRunInfo();

    void defineHisto(std::vector<TH1D*>& htemp, std::string var, std::string stype);

    void wireGain(std::vector<TH1D*>& hdedxhit);

    void printCanvasWire(std::vector<TH1D*> temp, std::string namesfx, const std::vector<double>& vdedx_mean);

    void fit(TH1D*& hist, double& mean, double& meanErr, double& sigma, double& sigmaErr);

    void bhabhaValidation();

    void radeeValidation();

    void defineTimeBins(std::vector<double>& vtlocaledges);

    void printCanvasdEdx(std::array<std::vector<TH1D*>, 2>& htemp, std::string namesfx, std::string svar);

    std::string getTimeBinLabel(const double& tedges, const int& it)
    {
      std::string label = "";
      if (tedges < 2e4)label = Form("%0.01f-%0.01fK", m_tedges[it] / 1e3, m_tedges[it + 1] / 1e3);
      else if (tedges < 1e5)label = Form("%0.0f-%0.0fK", m_tedges[it] / 1e3, m_tedges[it + 1] / 1e3);
      else label = Form("%0.01f-%0.01fM", m_tedges[it] / 1e6, m_tedges[it + 1] / 1e6);
      return label;
    }

    void setTestingPayload(const std::string& testingPayloadName)
    {
      m_testingPayloadName = testingPayloadName;
    }

    void setGlobalTag(const std::string& globalTagName)
    {
      m_GlobalTagName = globalTagName;
    }

    void setTextCosmetics(TPaveText pt, Color_t color)
    {
      pt.SetBorderSize(0);
      pt.SetShadowColor(kWhite);
      pt.SetTextColor(color);
    }

    void plotEventStats();

    void DatabaseIN(int experiment, int run);

    WireGainData getwiregain(int experiment, int run);

    CosGainData getcosgain(int experiment, int run);

    OnedData getonedgain(int experiment, int run);

    double getrungain(int experiment, int run);

    void resetDatabase();
 
  protected:

    virtual EResult calibrate() override;
 
  private:
    double m_sigmaR;  
 
    int m_dedxBins;  
    double m_dedxMin; 
    double m_dedxMax; 
 
    int m_cosBins; 
    double m_cosMin; 
    double m_cosMax; 
 
    int m_momBins; 
    double m_momMin; 
    double m_momMax; 
 
    double* m_tedges; 
    unsigned int m_tbins;  
 
    int m_eaBin; 
    double m_eaMin; 
    double m_eaMax; 
 
    std::array<std::string, 2> m_sring{"ler", "her"}; 
 
    std::string m_suffix; 
 
    DBObjPtr<CDCGeometry> m_cdcGeo; 

    std::string m_testingPayloadName = "";

    std::string m_GlobalTagName = "";

    StoreObjPtr<EventMetaData> m_EventMetaData;
 
    DBObjPtr<CDCDedxInjectionTime> m_DBInjectTime; 
  };

} // namespace Belle2