SVDClusterTimeShifterAlgorithm.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 "TMath.h"
 
#include <optional>
 
namespace Belle2 {
  class SVDClusterTimeShifterAlgorithm : public CalibrationAlgorithm {
  public:
 
    explicit SVDClusterTimeShifterAlgorithm(const std::string& str);
 
    virtual ~SVDClusterTimeShifterAlgorithm() {}
 
    void setMaximumAllowedShift(const float& value) {m_maximumAllowedShift = value;}
 
    void setTimeAlgorithm(const std::vector<TString>& lst) {m_timeAlgorithms = lst;}
 
    void setMinEntries(const int& minEntries) {m_minEntries = minEntries;}
 
  protected:
 
    virtual EResult calibrate() override;
 
  private:
 
    std::string m_id = ""; 
    float m_maximumAllowedShift = 15; 
    int m_minEntries = 1000; 
    std::vector<TString> m_timeAlgorithms = {"CoG3", "ELS3", "CoG6"}; 
  };
 
  inline double singleGaus(const double* x, const double* par)
  {
    double N = std::fabs(par[0]);
    double a = par[1];
    double b = par[2];
    double e = std::fabs(par[3]);
    return N * TMath::Gaus(x[0], a, b, true) + e;
  };
 
  inline double doubleGaus(const double* x, const double* par)
  {
    double N = std::fabs(par[0]);
    double f = std::fabs(par[1]);
    double a = par[2];
    double b = par[3];
    double c = par[4];
    double d = par[5];
    double e = std::fabs(par[6]);
    return (N * (f * TMath::Gaus(x[0], a, b) + (1 - f) * TMath::Gaus(x[0], c, d)) /
            (TMath::Sqrt(2. * TMath::Pi()) * ((b - d) * f + d)) + e);
  };
 
} // namespace Belle2