TOPCalChannelThresholdEff.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Umberto Tamponi (tamponi@to.infn.it), Marko Staric,      *
 *               Maeda Yosuke                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <TObject.h>
#include <framework/logging/Logger.h>
 
namespace Belle2 {
  class TOPCalChannelThresholdEff: public TObject {
  public:
 
    enum EStatus {
      c_Default = 0,    
      c_Calibrated = 1, 
      c_Unusable = 2    
    };
 
    TOPCalChannelThresholdEff() {}
 
    void setThrEff(int moduleID, unsigned channel, float ThrEff, short offlineThreshold)
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) {
        B2ERROR("Invalid module number, constant not set (" << ClassName() << ")");
        return;
      }
      if (channel >= c_numChannels) {
        B2ERROR("Invalid channel number, constant not set (" << ClassName() << ")");
        return;
      }
      m_ThrEff[module][channel] = ThrEff;
      m_offlineThreshold[module][channel] = offlineThreshold;
      m_status[module][channel] = c_Calibrated;
    }
 
    void setUnusable(int moduleID, unsigned channel)
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) {
        B2ERROR("Invalid module number, status not set (" << ClassName() << ")");
        return;
      }
      if (channel >= c_numChannels) {
        B2ERROR("Invalid channel number, status not set (" << ClassName() << ")");
        return;
      }
      m_status[module][channel] = c_Unusable;
    }
 
    float getThrEff(int moduleID, unsigned channel) const
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) {
        B2WARNING("Invalid module number, returning 0 (" << ClassName() << ")");
        return 0;
      }
      if (channel >= c_numChannels) {
        B2WARNING("Invalid channel number, returning 0 (" << ClassName() << ")");
        return 0;
      }
      if (m_status[module][channel] == c_Default) return 1.0;
      return m_ThrEff[module][channel];
    }
 
    bool isCalibrated(int moduleID, unsigned channel) const
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) return false;
      if (channel >= c_numChannels) return false;
      return m_status[module][channel] == c_Calibrated;
    }
 
    bool isDefault(int moduleID, unsigned channel) const
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) return false;
      if (channel >= c_numChannels) return false;
      return m_status[module][channel] == c_Default;
    }
 
    bool isUnusable(int moduleID, unsigned channel) const
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) return false;
      if (channel >= c_numChannels) return false;
      return m_status[module][channel] == c_Unusable;
    }
 
    short getOfflineThreshold(int moduleID, unsigned channel) const
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) {
        B2WARNING("Invalid module number, returning 0 (" << ClassName() << ")");
        return 0;
      }
      if (channel >= c_numChannels) {
        B2WARNING("Invalid channel number, returning 0 (" << ClassName() << ")");
        return 0;
      }
      return m_offlineThreshold[module][channel];
    }
 
  private:
 
    enum {c_numModules = 16,  
          c_numChannels = 512 
         };
 
 
    float m_ThrEff[c_numModules][c_numChannels] = {{0}};    
    EStatus m_status[c_numModules][c_numChannels] = {{c_Default}}; 
    short m_offlineThreshold[c_numModules][c_numChannels] = {{0}}; 
    ClassDef(TOPCalChannelThresholdEff, 3); 
  };
 
} // end namespace Belle2