TOPCalChannelRQE.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <TObject.h>
#include <framework/logging/Logger.h>
 
namespace Belle2 {
  class TOPCalChannelRQE: public TObject {
  public:
 
    enum EStatus {
      c_Default = 0,    
      c_Calibrated = 1, 
      c_Unusable = 2    
    };
 
    TOPCalChannelRQE() {}
 
    void setRQE(int moduleID, unsigned channel, double relQE)
    {
      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_relQE[module][channel] = relQE;
      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;
    }
 
    double getRQE(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_relQE[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;
    }
 
  private:
 
    enum {
      c_numModules = 16,  
      c_numChannels = 512 
    };
 
    // Note: if initialization list is too short others are set to 0 (see comment in TOPCalChannelMask)
    float m_relQE[c_numModules][c_numChannels] = {{0.0}};    
    EStatus m_status[c_numModules][c_numChannels] = {{c_Default}}; 
    ClassDef(TOPCalChannelRQE, 1); 
  };
 
} // end namespace Belle2