PXDAnalyticGainCalibrationAlgorithm.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2021 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors:  Qingyuan Liu                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
#include <calibration/CalibrationAlgorithm.h>
#include <vxd/dataobjects/VxdID.h>
#include <vector>
 
// ROOT
#include <TH1.h>
 
 
namespace Belle2 {
  class  PXDAnalyticGainCalibrationAlgorithm : public CalibrationAlgorithm {
  public:
 
    PXDAnalyticGainCalibrationAlgorithm();
 
    virtual ~PXDAnalyticGainCalibrationAlgorithm() {}
 
    int minClusters;
 
    float safetyFactor;
 
    bool forceContinue;
 
    int strategy;
 
    bool useChargeRatioHistogram;
 
    bool correctForward;
 
    void setBoundaries(const std::vector<Calibration::ExpRun>& boundaries) {m_requestedBoundaries = boundaries;}
 
    std::vector<Calibration::ExpRun> getBoundaries() const {return m_requestedBoundaries;}
 
  protected:
 
    virtual EResult calibrate() override;
 
    virtual bool isBoundaryRequired(const Calibration::ExpRun& /*currentRun*/) override;
 
    //  We reset the temporary boundaries list to be the same as the requested one.
    virtual void boundaryFindingSetup(std::vector<Calibration::ExpRun> /*runs*/, int /*iteration = 0*/) override
    {
      m_boundaries = m_requestedBoundaries;
    };
 
  private:
 
    double EstimateGain(VxdID sensorID, unsigned short uBin, unsigned short vBin, TH1* hist = nullptr);
 
    //double CalculateMedian(std::vector<double>& signals);
 
    //double CalculateMedian(TH1* hist);
 
    //double FitLandau(std::vector<double>& signals);
 
    //double FitLandau(TH1* hist);
 
    double GetCurrentGainFromDB(VxdID sensorID, unsigned short uBin, unsigned short vBin);
 
    std::vector<Calibration::ExpRun> m_requestedBoundaries;
 
  };
} // namespace Belle2