HitPatternCDC.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 <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 __str__() 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(); }
 
  };
}