PXDMaskedPixelPar.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 PXDMaskedPixelPar: public TObject {
  public:
    typedef std::unordered_set< unsigned int> MaskedSinglePixelsSet;
 
    PXDMaskedPixelPar() : m_MapSingles() {}
    ~PXDMaskedPixelPar() {}
 
    void maskSinglePixel(unsigned short sensorID, unsigned int pixID)
    {
      auto mapIterSingles = m_MapSingles.find(sensorID);
      if (mapIterSingles != m_MapSingles.end()) {
        // Already some masked single pixels on sensor
        auto& singles = mapIterSingles->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
        PXDMaskedPixelPar::MaskedSinglePixelsSet singles;
        // pixID will be used to generate hash in unordered_set for quick access
        singles.insert(pixID);
        m_MapSingles[sensorID] = singles;
      }
    }
 
    bool pixelOK(unsigned short sensorID, unsigned int pixID) const
    {
      auto mapIterSingles = m_MapSingles.find(sensorID);
      if (mapIterSingles != m_MapSingles.end()) {
        // Found some masked single pixels on sensor
        auto& singles = mapIterSingles->second;
        // Look if this is a single masked pixel
        if (singles.find(pixID) != singles.end())
          return false;
      }
      // Pixel not found in the mask => pixel OK
      return true;
    }
 
    const std::unordered_map<unsigned short, MaskedSinglePixelsSet>& getMaskedPixelMap() const {return m_MapSingles;}
 
  private:
 
    std::unordered_map<unsigned short, MaskedSinglePixelsSet> m_MapSingles;
 
    ClassDef(PXDMaskedPixelPar, 1);  
  };
} // end of namespace Belle2