ECLTimeWalkCorrection.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 <TObject.h>
#include <limits>
 
namespace Belle2 {
  class ECLTimeWalkCorrection : public TObject {
  public:
 
    ECLTimeWalkCorrection() :
      m_par(c_expectedParametersCount)
    {
    }
 
    ~ECLTimeWalkCorrection() {}
 
    double getParam(int n) const
    {
      return m_par.at(n);
    }
    void setParam(int n, double newval)
    {
      if (n < 0 || n >= c_expectedParametersCount) {
        B2FATAL("Parameter index should be within [0, " <<
                c_expectedParametersCount - 1 << "]." <<
                LogVar("parameter index", n));
      }
      m_par[n] = newval;
    }
 
    std::vector<double> getParams() const
    {
      return m_par;
    }
    void setParams(const std::vector<double>& new_params)
    {
      if (new_params.size() != c_expectedParametersCount) {
        B2FATAL("Count of parameters must always be equal to " <<
                c_expectedParametersCount <<
                LogVar("count of parameters", new_params.size()));
      }
      m_par = new_params;
    }
 
    int getParamFormat()
    {
      return m_parFormat;
    }
 
    void setParamFormat(int newval)
    {
      m_parFormat = newval;
    }
 
  private:
    int m_parFormat = 0;
    std::vector<double> m_par;
 
    const static int c_expectedParametersCount = 6; 
 
    ClassDef(ECLTimeWalkCorrection, 1); 
  };
} // end namespace Belle2