EnergyMask.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 <vector>
#include <cmath>
 
namespace Belle2 {
  namespace TOP {
 
    class EnergyMask {
 
    public:
 
      EnergyMask(double dyde, double dydL, double dydx, double dy, double dL, double dx, double dE);
 
      static void setNMin(unsigned Nmin) {s_Nmin = Nmin;}
 
      static void setNMax(unsigned Nmax) {s_Nmax = Nmax;}
 
      double getStep() const {return m_dE;}
 
      double getFullWidth() const {return m_A + m_B + m_C;}
 
      const std::vector<double>& getMask() const {return m_mask;}
 
      double getMask(double E) const;
 
      double getMask(int i) const;
 
      double getMask(int i, double fract) const;
 
      double getEnergyDifference(int i) {return i * m_dE;}
 
    private:
 
      double twoSquareConvolution(double E) const;
 
      double threeSquareConvolution(double E) const;
 
      double mask(int i) const;
 
      double m_dE = 0; 
      double m_Wy = 0; 
      double m_A = 0;  
      double m_B = 0;  
      double m_C = 0;  
      std::vector<double> m_mask; 
      static unsigned s_Nmin; 
      static unsigned s_Nmax; 
    };
 
 
    inline double EnergyMask::mask(int i) const
    {
      unsigned k = std::abs(i);
      if (k < m_mask.size()) return m_mask[k];
      return 0;
    }
 
 
  } // namespace TOP
} // namespace Belle2