HitPatternVXD.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2013 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Heck, Markus Prim                                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <bitset>
#include <utility>
#include <list>
 
namespace Belle2 {
  class HitPatternVXD {
    friend class HitPatternVXDTest_General_Test;
    friend class HitPatternVXDTest_NumberingScheme_Test;
    friend class HitPatternVXDTest_SVDSetterAndGetter_Test;
    friend class HitPatternVXDTest_PXDSetterAndGetter_Test;
 
  public:
 
    enum class PXDMode : unsigned short { normal = static_cast<unsigned short>(0), gated = 1 };
 
    HitPatternVXD() : m_pattern(0) {}
 
    explicit HitPatternVXD(const unsigned int initValue) : m_pattern(initValue) {}
 
    unsigned int getInteger() const { return m_pattern.to_ulong(); }
 
    unsigned short getNdf() const;
 
    unsigned short getNVXDLayers(const PXDMode& pxdmode = PXDMode::normal) const;
 
    // ----------------------------------------------------------------
    // ---------------- SVD LAYER FUNCTIONS ---------------------------
    // ----------------------------------------------------------------
 
    void setSVDLayer(const unsigned short layerId, unsigned short uHits, unsigned short vHits);
 
    std::pair<const unsigned short, const unsigned short> getSVDLayer(const unsigned short layerId) const;
 
    void resetSVDLayer(const unsigned short layerId);
 
    unsigned short getNSVDHits() const;
 
    unsigned short getNSVDLayers() const;
 
    short getFirstSVDLayer() const;
 
    short getLastSVDLayer() const;
 
    // ----------------------------------------------------------------
    // ---------------- PXD LAYER FUNCTIONS ---------------------------
    // ----------------------------------------------------------------
 
    void setPXDLayer(const unsigned short layerId, unsigned short nHits, const PXDMode& mode = PXDMode::normal);
 
    unsigned short getPXDLayer(const unsigned short layerId, const PXDMode& mode = PXDMode::normal) const;
 
    void resetPXDLayer(const unsigned short layerId, const PXDMode& mode = PXDMode::normal);
 
    unsigned short getNPXDHits() const;
 
    unsigned short getNPXDLayers(const PXDMode& mode = PXDMode::normal) const;
 
    short getFirstPXDLayer(const PXDMode& mode = PXDMode::normal) const;
 
    short getLastPXDLayer(const PXDMode& mode = PXDMode::normal) const;
 
    std::string __repr__() const;
 
  private:
    std::bitset<32> m_pattern;
 
    static const std::bitset<32> s_LayerMasks[6]; 
    static const std::bitset<32> s_PXDModeMasks[2]; 
    static const std::bitset<32> s_SVDuvMasks[2]; 
    static const std::bitset<32> s_infoLayerMask;  
    static const std::list<unsigned short> s_PXDLayerNumbers;  
    static const std::list<unsigned short> s_SVDLayerNumbers;  
    static const unsigned int s_svdUshift = 8; 
    static const unsigned int s_svdVshift = 10; 
    std::bitset<32> getBitset() const { return m_pattern; }
 
    std::bitset<32> getLayerMask(const unsigned short layerId) const { return HitPatternVXD::s_LayerMasks[layerId - 1]; }
 
  };
}