HitPatternCDC.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 <RtypesCore.h>
 
#include <bitset>
#include <map>
#include <utility>
#include <string>
 
namespace Belle2 {
  class HitPatternCDC {
    friend class HitPatternCDCTest_settersNGetters_Test;
    friend class HitPatternCDCTest_getFirstLastLayer_Test;
 
  public:
    HitPatternCDC() : m_pattern(0) {}
 
    explicit HitPatternCDC(ULong64_t initValue) : m_pattern(initValue) {}
 
    ULong64_t getInteger() const;
 
    unsigned short getNHits() const;
 
    void setNHits(unsigned short nHits);
 
    // ----------------------------------------------------------------
    // ---------------- LAYER FUNCTIONS -------------------------------
    // ----------------------------------------------------------------
 
    void setLayer(const unsigned short layer);
 
    void resetLayer(const unsigned short layer);
 
    bool hasLayer(const unsigned short layer) const;
 
    short getFirstLayer() const;
 
    short getLastLayer() const;
 
    // ----------------------------------------------------------------
    // ---------------- SUPER LAYER FUNCTIONS -------------------------
    // ----------------------------------------------------------------
 
    bool hasSLayer(const unsigned short sLayer) const;
 
    void resetSLayer(const unsigned short sLayer);
 
    std::bitset<64> getSLayerPattern(const unsigned short sLayer);
 
    unsigned short getSLayerNHits(const unsigned short sLayer) const;
 
    unsigned short getLongestContRunInSL(const unsigned short sLayer) const;
 
    // ----------------------------------------------------------------
    // ---------------- Others ----------------------------------------
    // ----------------------------------------------------------------
 
    bool hasAxialLayer() const;
 
    bool hasStereoLayer() const;
 
    std::string __repr__() const;
 
  private:
    std::bitset<64> m_pattern;                     
    static const std::bitset<64> s_sLayerMasks[9]; 
    static const std::bitset<64> s_infoLayerMask;  
    static const std::map<unsigned short, std::pair<unsigned short, unsigned short>> s_superLayerIndices;
 
    std::bitset<64> getBitSet() const { return m_pattern; }
 
    void resetPattern() { m_pattern.reset(); }
 
  };
}