KLMStripEfficiency.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
 
/* KLM headers. */
#include <klm/dataobjects/KLMElementNumbers.h>
 
/* ROOT headers. */
#include <TObject.h>
 
/* C++ headers. */
#include <map>
#include <limits>
 
namespace Belle2 {
  class KLMStripEfficiency: public TObject {
  public:
 
    KLMStripEfficiency()
    {
    }
 
    ~KLMStripEfficiency()
    {
    }
 
    void setEfficiency(KLMChannelNumber stripId, float efficiency, float efficiencyError = 0.)
    {
      m_efficiency.insert(std::pair<KLMChannelNumber, float>(stripId, efficiency));
      m_efficiencyError.insert(std::pair<KLMChannelNumber, float>(stripId, efficiencyError));
    }
 
    void setBarrelEfficiency(int section, int sector, int layer, int plane, int strip, float efficiency, float efficiencyError = 0.)
    {
      const KLMElementNumbers* elementNumbers = &(KLMElementNumbers::Instance());
      KLMChannelNumber stripId = elementNumbers->channelNumberBKLM(section, sector, layer, plane, strip);
      setEfficiency(stripId, efficiency, efficiencyError);
    }
 
    void setEndcapEfficiency(int section, int sector, int layer, int plane, int strip, float efficiency, float efficiencyError = 0.)
    {
      const KLMElementNumbers* elementNumbers = &(KLMElementNumbers::Instance());
      KLMChannelNumber stripId = elementNumbers->channelNumberEKLM(section, sector, layer, plane, strip);
      setEfficiency(stripId, efficiency, efficiencyError);
    }
 
    float getEfficiency(KLMChannelNumber stripId) const
    {
      auto search = m_efficiency.find(stripId);
      if (search == m_efficiency.end())
        return std::numeric_limits<float>::quiet_NaN();
      return search->second;
    }
 
    float getBarrelEfficiency(int section, int sector, int layer, int plane, int strip) const
    {
      const KLMElementNumbers* elementNumbers = &(KLMElementNumbers::Instance());
      KLMChannelNumber stripId = elementNumbers->channelNumberBKLM(section, sector, layer, plane, strip);
      return getEfficiency(stripId);
    }
 
    float getEndcapEfficiency(int section, int sector, int layer, int plane, int strip) const
    {
      const KLMElementNumbers* elementNumbers = &(KLMElementNumbers::Instance());
      KLMChannelNumber stripId = elementNumbers->channelNumberEKLM(section, sector, layer, plane, strip);
      return getEfficiency(stripId);
    }
 
    float getEfficiencyError(KLMChannelNumber stripId) const
    {
      auto search = m_efficiencyError.find(stripId);
      if (search == m_efficiencyError.end())
        return std::numeric_limits<float>::quiet_NaN();
      return search->second;
    }
 
    float getBarrelEfficiencyError(int section, int sector, int layer, int plane, int strip) const
    {
      const KLMElementNumbers* elementNumbers = &(KLMElementNumbers::Instance());
      KLMChannelNumber stripId = elementNumbers->channelNumberBKLM(section, sector, layer, plane, strip);
      return getEfficiencyError(stripId);
    }
 
    float getEndcapEfficiencyError(int section, int sector, int layer, int plane, int strip) const
    {
      const KLMElementNumbers* elementNumbers = &(KLMElementNumbers::Instance());
      KLMChannelNumber stripId = elementNumbers->channelNumberEKLM(section, sector, layer, plane, strip);
      return getEfficiencyError(stripId);
    }
 
  private:
 
    std::map<KLMChannelNumber, float> m_efficiency;
 
    std::map<KLMChannelNumber, float> m_efficiencyError;
 
    ClassDef(KLMStripEfficiency, 1);
 
  };
 
}