PXDDeadPixelPar.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Benjamin Schwenker                                       *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <TObject.h>
#include <unordered_map>
#include <unordered_set>
 
 
namespace Belle2 {
  class PXDDeadPixelPar: public TObject {
  public:
    typedef std::unordered_set< unsigned int> DeadChannelSet;
 
    PXDDeadPixelPar() : m_MapSingles(), m_MapRows(), m_MapDrains(), m_MapSensors()  {}
 
    void maskSensor(unsigned short sensorID)
    {
      auto mapIter = m_MapSensors.find(sensorID);
      if (mapIter == m_MapSensors.end()) {
        // Sensor not already masked as dead. Mask it.
        m_MapSensors.insert(sensorID);
      }
    }
 
    void maskDrain(unsigned short sensorID, unsigned int drainID)
    {
      auto mapIter = m_MapDrains.find(sensorID);
      if (mapIter != m_MapDrains.end()) {
        // Already some masked drain on sensor
        auto& drains = mapIter->second;
        // Only add drain, if it is not already in
        if (drains.find(drainID) == drains.end())
          drains.insert(drainID);
      } else {
        // Create an empty set of masked drains
        PXDDeadPixelPar::DeadChannelSet drains;
        // drainID will be used to generate hash in unordered_set for quick access
        drains.insert(drainID);
        m_MapDrains[sensorID] = drains;
      }
    }
 
    void maskRow(unsigned short sensorID, unsigned int vCellID)
    {
      auto mapIter = m_MapRows.find(sensorID);
      if (mapIter != m_MapRows.end()) {
        // Already some masked rows on sensor
        auto& rows = mapIter->second;
        // Only add row, if it is not already in
        if (rows.find(vCellID) == rows.end())
          rows.insert(vCellID);
      } else {
        // Create an empty set of masked rows
        PXDDeadPixelPar::DeadChannelSet rows;
        // vCellID will be used to generate hash in unordered_set for quick access
        rows.insert(vCellID);
        m_MapRows[sensorID] = rows;
      }
    }
 
    void maskSinglePixel(unsigned short sensorID, unsigned int pixID)
    {
      auto mapIter = m_MapSingles.find(sensorID);
      if (mapIter != m_MapSingles.end()) {
        // Already some masked single pixels on sensor
        auto& singles = mapIter->second;
        // Only add pixel, if it is not already in
        if (singles.find(pixID) == singles.end())
          singles.insert(pixID);
      } else {
        // Create an empty set of masked single pixels
        PXDDeadPixelPar::DeadChannelSet singles;
        // pixID will be used to generate hash in unordered_set for quick access
        singles.insert(pixID);
        m_MapSingles[sensorID] = singles;
      }
    }
 
    bool isDeadSensor(unsigned short sensorID) const
    {
      if (m_MapSensors.find(sensorID) == m_MapSensors.end()) {
        return false;
      }
      return true;
    }
 
    bool isDeadRow(unsigned short sensorID, unsigned int vCellID) const
    {
      auto mapIter = m_MapRows.find(sensorID);
      if (mapIter != m_MapRows.end()) {
        // Found some dead rows
        auto& deadRows = mapIter->second;
        // Look if this row is dead
        if (deadRows.find(vCellID) != deadRows.end())
          return true;
      }
      return false;
    }
 
    bool isDeadDrain(unsigned short sensorID, unsigned int drainID) const
    {
      auto mapIter = m_MapDrains.find(sensorID);
      if (mapIter != m_MapDrains.end()) {
        // Found some dead drains
        auto& deadDrains = mapIter->second;
        // Look if this drain is dead
        if (deadDrains.find(drainID) != deadDrains.end())
          return true;
      }
      return false;
    }
 
    bool isDeadSinglePixel(unsigned short sensorID, unsigned int pixID) const
    {
      auto mapIter = m_MapSingles.find(sensorID);
      if (mapIter != m_MapSingles.end()) {
        // Found some dead singles
        auto& deadSingles = mapIter->second;
        // Look if this single is dead
        if (deadSingles.find(pixID) != deadSingles.end())
          return true;
      }
      return false;
    }
 
    const std::unordered_map<unsigned short, DeadChannelSet>& getDeadSinglePixelMap() const {return m_MapSingles;}
 
    const std::unordered_map<unsigned short, DeadChannelSet>& getDeadDrainMap() const {return m_MapDrains;}
 
    const std::unordered_map<unsigned short, DeadChannelSet>& getDeadRowMap() const {return m_MapRows;}
 
    const std::unordered_set< unsigned int>& getDeadSensorMap() const {return m_MapSensors;}
 
  private:
 
    std::unordered_map<unsigned short, DeadChannelSet> m_MapSingles;
 
    std::unordered_map<unsigned short, DeadChannelSet> m_MapRows;
 
    std::unordered_map<unsigned short, DeadChannelSet> m_MapDrains;
 
    std::unordered_set< unsigned int> m_MapSensors;
 
    ClassDef(PXDDeadPixelPar, 2);  
  };
} // end of namespace Belle2