TOPCalChannelPulseHeight.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2017 - 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>
#include <top/dbobjects/TOPPulseHeightPar.h>
 
namespace Belle2 {
  class TOPCalChannelPulseHeight: public TObject {
  public:
 
    enum EStatus {
      c_Default = 0,    
      c_Calibrated = 1, 
      c_Unusable = 2    
    };
 
    TOPCalChannelPulseHeight()
    {}
 
    void setParameters(int moduleID, unsigned channel, double x0, double p1, double p2)
    {
      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;
      }
      if (x0 <= 0) {
        B2ERROR("Invalid parameter value x0, constant not set (" << ClassName() << ")."
                << LogVar("x0", x0)
                << LogVar("slot", moduleID)
                << LogVar("channel", channel));
        return;
      }
      if (p1 < 0) {
        B2ERROR("Invalid parameter value p1, constant not set (" << ClassName() << ")."
                << LogVar("p1", p1)
                << LogVar("slot", moduleID)
                << LogVar("channel", channel));
        return;
      }
      if (p2 <= 0) {
        B2ERROR("Invalid parameter value p2, constant not set (" << ClassName() << ")."
                << LogVar("p2", p2)
                << LogVar("slot", moduleID)
                << LogVar("channel", channel));
        return;
      }
      m_par[module][channel].x0 = x0;
      m_par[module][channel].p1 = p1;
      m_par[module][channel].p2 = p2;
      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;
    }
 
    const TOPPulseHeightPar& getParameters(int moduleID, unsigned channel) const
    {
      unsigned module = moduleID - 1;
      if (module >= c_numModules) {
        B2WARNING("Invalid slot number, "
                  "returning parameters of slot 1 channel 0 (" << ClassName() << ")"
                  << LogVar("slot", moduleID));
        return m_par[0][0];
      }
      if (channel >= c_numChannels) {
        B2WARNING("Invalid channel, "
                  "returning parameters of channel 0 (" << ClassName() << ")"
                  << LogVar("channel", channel));
        return m_par[module][0];
      }
      return m_par[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 
    };
 
    TOPPulseHeightPar m_par[c_numModules][c_numChannels]; 
    EStatus m_status[c_numModules][c_numChannels] = {{c_Default}}; 
    ClassDef(TOPCalChannelPulseHeight, 3); 
  };
 
} // end namespace Belle2