PXDOccupancyInfoPar.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 <TObject.h>
#include <unordered_map>
 
 
namespace Belle2 {

 
  class PXDOccupancyInfoPar: public TObject {
  public:

    PXDOccupancyInfoPar() : m_nEvents(0),  m_MapOccupancy() {}

    void setNumberOfEvents(int nEvents) { m_nEvents = nEvents; }

    int getNumberOfEvents() const { return m_nEvents; }

    void setOccupancy(unsigned short sensorID, float occupancy)
    {
      m_MapOccupancy[sensorID] = occupancy;
    }

    float getOccupancy(unsigned short sensorID) const
    {
      auto mapIter = m_MapOccupancy.find(sensorID);
      if (mapIter != m_MapOccupancy.end()) {
        return mapIter->second;
      }
      // Fallback when user asks for sensor that was not calibrated
      return -1;
    }
 
  private:

    int m_nEvents;

    std::unordered_map<unsigned short, float> m_MapOccupancy;
 
    ClassDef(PXDOccupancyInfoPar, 2);  
  };

} // end of namespace Belle2