SVD3SampleCoGTimeCalibrationAlgorithm.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 <optional>
 
namespace Belle2 {
  class SVD3SampleCoGTimeCalibrationAlgorithm : public CalibrationAlgorithm {
  public:
 
    explicit SVD3SampleCoGTimeCalibrationAlgorithm(const std::string& str);
 
    virtual ~SVD3SampleCoGTimeCalibrationAlgorithm() {}
 
    void setAllowedTimeShift(float value) {m_allowedTimeShift = value;}
 
    float getAllowedTimeShift() {return m_allowedTimeShift;}
 
    void setMinEntries(int minEntries) {m_minEntries = minEntries;}
 
    int getMinEntries() {return m_minEntries;}
 
    void setTwoLineSelectionParameters(bool linearCutsOnCoG3, float interceptUpperLine, float angularCoefficientUpperLine,
                                       float interceptLowerLine,
                                       float angularCoefficientLowerLine)
    {
      m_applyLinearCutsToRemoveBkg = linearCutsOnCoG3;
      m_interceptUpperLine =  interceptUpperLine;
      m_angularCoefficientUpperLine =  angularCoefficientUpperLine;
      m_interceptLowerLine =  interceptLowerLine;
      m_angularCoefficientLowerLine =  angularCoefficientLowerLine;
    }
 
    std::vector<float> getTwoLineSelectionParameters()
    {
      std::vector<float> twoLineSelectionParameters;
      twoLineSelectionParameters.push_back(m_interceptUpperLine);
      twoLineSelectionParameters.push_back(m_angularCoefficientUpperLine);
      twoLineSelectionParameters.push_back(m_interceptLowerLine);
      twoLineSelectionParameters.push_back(m_angularCoefficientLowerLine);
 
      return twoLineSelectionParameters;
    }
 
  protected:
 
    virtual EResult calibrate() override;
 
    virtual bool isBoundaryRequired(const Calibration::ExpRun& currentRun) override;
 
    virtual void boundaryFindingSetup(std::vector<Calibration::ExpRun> /*runs*/, int /*iteration = 0*/) override
    {
      m_previousRawTimeMeanL3V.reset();
    }
 
 
  private:
 
    std::string m_id; 
    std::optional<float> m_previousRawTimeMeanL3V; 
    float m_allowedTimeShift = 2.; 
    float m_minEntries = 10000; 
    // straight lines used to reject off-time clusters from 2D-distribution
    bool m_applyLinearCutsToRemoveBkg = false; 
    float m_interceptUpperLine =
      -94.0; 
    float m_angularCoefficientUpperLine =
      1.264; 
    float m_interceptLowerLine =
      -134.0; 
    float m_angularCoefficientLowerLine =
      1.264; 
  };
} // namespace Belle2