PXDClusterChargeMapPar.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>
#include <vector>
 
namespace Belle2 {
  class PXDClusterChargeMapPar: public TObject {
  public:
 
    PXDClusterChargeMapPar(int nBinsU = 4, int nBinsV = 6, float defaultValue = -1.0) : m_nBinsU(nBinsU), m_nBinsV(nBinsV),
      m_defaultValue(defaultValue), m_sensorCalibrationMap() {}
 
    unsigned short getBinsU() const { return m_nBinsU; }
 
    unsigned short getBinsV() const { return m_nBinsV; }
 
    unsigned short getGlobalID(unsigned short uBin, unsigned short vBin) const { return uBin * m_nBinsV + vBin; }
 
    void setContent(unsigned short sensorID, unsigned short globalID, float value)
    {
      auto mapIter = m_sensorCalibrationMap.find(sensorID);
      if (mapIter != m_sensorCalibrationMap.end()) {
        // Already some values stored
        auto& calVec = mapIter->second;
        // Set the value
        calVec[globalID] = value;
      } else {
        // Create a fresh calibration vector
        std::vector<float> calVec(m_nBinsU * m_nBinsV, m_defaultValue);
        // Set the value
        calVec[globalID] = value;
        // Add vector to map
        m_sensorCalibrationMap[sensorID] = calVec;
      }
    }
 
    void setContent(unsigned short sensorID, unsigned short uBin, unsigned short vBin, float value)
    {
      setContent(sensorID, getGlobalID(uBin, vBin), value);
    }
 
    float getContent(unsigned short sensorID, unsigned short globalID) const
    {
      auto mapIter = m_sensorCalibrationMap.find(sensorID);
      if (mapIter != m_sensorCalibrationMap.end()) {
        // Found sensor, return calibration value
        auto& calVec = mapIter->second;
        return calVec[globalID];
      }
      // Sensor not found, keep low profile and return default calibration value
      return m_defaultValue;
    }
 
    float getContent(unsigned short sensorID, unsigned short uBin, unsigned short vBin) const
    {
      return getContent(sensorID, getGlobalID(uBin, vBin));
    }
 
    const std::unordered_map<unsigned short, std::vector<float> >& getCalibrationMap() const {return m_sensorCalibrationMap;}
 
  private:
 
    int m_nBinsU;
 
    int m_nBinsV;
 
    float m_defaultValue;
 
    std::unordered_map<unsigned short, std::vector<float> > m_sensorCalibrationMap;
 
    ClassDef(PXDClusterChargeMapPar, 1);  
  };
} // end of namespace Belle2