ClusterPattern.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_ITSMFT_CLUSTERPATTERN_H
#define ALICEO2_ITSMFT_CLUSTERPATTERN_H
#include <Rtypes.h>
#include <array>
#include <iosfwd>
#include <gsl/gsl>
#include "DataFormatsITSMFT/CompCluster.h"
 
namespace o2
{
namespace itsmft
{
class ClusterTopology;
class TopologyDictionary;
class BuildTopologyDictionary;
 
class ClusterPattern
{
 public:
  static constexpr uint8_t MaxRowSpan = 128;
  static constexpr uint8_t MaxColSpan = 128;
  static constexpr int MaxPatternBits = MaxRowSpan * MaxColSpan;
  static constexpr int MaxPatternBytes = MaxPatternBits / 8;
  static constexpr int SpanMask = 0x7fff;
  static constexpr int TruncateMask = 0x8000;
 
  ClusterPattern();
  ClusterPattern(int nRow, int nCol, const unsigned char patt[MaxPatternBytes]);
 
  template <class iterator>
  void acquirePattern(iterator& pattIt)
  {
    mBitmap[0] = *pattIt++;
    mBitmap[1] = *pattIt++;
    int nbits = mBitmap[0] * mBitmap[1];
    int nBytes = nbits / 8;
    if (((nbits) % 8) != 0) {
      nBytes++;
    }
    memcpy(&mBitmap[2], &(*pattIt), nBytes);
    pattIt += nBytes;
  }
 
  template <class iterator>
  static void skipPattern(iterator& pattIt)
  {
    unsigned char b0 = *pattIt++, b1 = *pattIt++;
    int nbits = b0 * b1;
    pattIt += nbits / 8 + (((nbits) % 8) != 0);
  }
 
  template <class iterator>
  ClusterPattern(iterator& pattIt)
  {
    acquirePattern(pattIt);
  }
 
  static constexpr int kExtendedPatternBytes = MaxPatternBytes + 2;
  const std::array<unsigned char, kExtendedPatternBytes>& getPattern() const { return mBitmap; }
  unsigned char getByte(int n) const;
  int getRowSpan() const { return (int)mBitmap[0]; }
  int getColumnSpan() const { return (int)mBitmap[1]; }
  int getUsedBytes() const;
  int getNPixels() const;
  friend std::ostream& operator<<(std::ostream& os, const ClusterPattern& top);
  void print() const;
  void setPattern(int nRow, int nCol, const unsigned char patt[MaxPatternBytes]);
  void setPattern(const unsigned char bitmask[kExtendedPatternBytes]);
  static int getCOG(int rowSpan, int colSpan, const unsigned char patt[MaxPatternBytes], float& xCOG, float& zCOG);
  int getCOG(float& xCOG, float& zCOG) const { return ClusterPattern::getCOG(getRowSpan(), getColumnSpan(), mBitmap.data() + 2, xCOG, zCOG); }
 
  bool isSet(int row, int col) const
  {
    const auto bmap = mBitmap.data() + 2;
    int pos = row * getColumnSpan() + col;
    return pos < getColumnSpan() * getRowSpan() && (bmap[pos >> 3] & (0x1 << (7 - (pos % 8))));
  }
 
  void resetPixel(int row, int col)
  {
    const auto bmap = mBitmap.data() + 2;
    int pos = row * getColumnSpan() + col;
    if (pos < getColumnSpan() * getRowSpan()) {
      bmap[pos >> 3] &= 0xff & ~(0x1 << (7 - (pos % 8)));
    }
  }
 
  void setPixel(int row, int col)
  {
    const auto bmap = mBitmap.data() + 2;
    int pos = row * getColumnSpan() + col;
    if (pos < getColumnSpan() * getRowSpan()) {
      bmap[pos >> 3] |= 0x1 << (7 - (pos % 8));
    }
  }
 
  template <typename Processor>
  void process(Processor procRowCol);
 
  friend ClusterTopology;
  friend TopologyDictionary;
  friend BuildTopologyDictionary;
 
 private:
 
  std::array<unsigned char, kExtendedPatternBytes> mBitmap; 
 
  ClassDefNV(ClusterPattern, 1);
};
 
template <typename Processor>
void ClusterPattern::process(Processor procRowCol)
{
  auto cspan = getColumnSpan(), rspan = getRowSpan();
  uint32_t nBits = cspan * rspan;
  uint32_t nBytes = (nBits >> 3) + (nBits % 8 != 0);
  const auto bmap = mBitmap.data() + 2;
  uint16_t ic = 0, ir = 0;
  for (unsigned int i = 0; i < nBytes; i++) {
    int s = 128; // 0b10000000
    while (s > 0) {
      if ((bmap[i] & s) != 0) {
        procRowCol(ir, ic);
      }
      ic++;
      s >>= 1;
      if (uint32_t(ir + 1) * ic == nBits) {
        break;
      }
      if (ic == cspan) {
        ic = 0;
        ir++;
      }
    }
    if (uint32_t(ir + 1) * ic == nBits) {
      break;
    }
  }
}
 
} // namespace itsmft
} // namespace o2
#endif /* ALICEO2_ITS_CLUSTERPATTERN_H */