RecoLocalTracker/SiPixelClusterizer/plugins/SiPixelArrayBuffer.h - master - githubmirror/cmssw

RecoLocalTracker/SiPixelClusterizer/plugins/SiPixelArrayBuffer.h
82 строки3 KB
Andre Govinda Stahl Leiton
Implement saturation flag in pixel cluster
02 мар 2026, 04:46
02 мар 2026, 04:464c9aa55
#ifndef RecoLocalTracker_SiPixelClusterizer_SiPixelArrayBuffer_H
#define RecoLocalTracker_SiPixelClusterizer_SiPixelArrayBuffer_H

//----------------------------------------------------------------------------
//! \class SiPixelArrayBuffer
//! \brief Class to store ADC counts during clustering.
//!
//! This class defines the buffer where the pixel ADC are stored.
//! The size is the number of rows and cols into a
//! ROC and it is set in the PixelThresholdClusterizer
//!
//! TO DO: the chip size should be obtained in some better way.
//!
//! History:
//!    Modify the indexing to col*nrows + row. 9/01 d.k.
//!    Add setSize method to adjust array size. 3/02 d.k.
//----------------------------------------------------------------------------

// We use PixelPos which is an inner class of SiPixelCluster:
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"

#include <vector>
#include <iostream>

class SiPixelArrayBuffer {
public:
  inline SiPixelArrayBuffer(int rows, int cols);
  inline SiPixelArrayBuffer() {}

  inline void setSize(int rows, int cols);
  inline int operator()(int row, int col) const;
  inline int operator()(const SiPixelCluster::PixelPos&) const;
  inline int rows() const { return nrows; }
  inline int columns() const { return ncols; }
  inline uint8_t rawADC(int row, int col) const { return pixel_rawADC[index(row, col)]; }

  inline bool inside(int row, int col) const;
  inline void set_adc(int row, int col, int adc);
  inline void set_adc(const SiPixelCluster::PixelPos&, int adc);
  inline void add_adc(int row, int col, int adc);
  inline void set_rawADC(int row, int col, uint8_t adc) { pixel_rawADC[index(row, col)] = adc; };
  int size() const { return pixel_vec.size(); }

  /// Definition of indexing within the buffer.
  int index(int row, int col) const { return col * nrows + row; }
  int index(const SiPixelCluster::PixelPos& pix) const { return index(pix.row(), pix.col()); }

private:
  uint16_t clamp(int adc) { return std::clamp<int>(adc, 0, std::numeric_limits<uint16_t>::max()); }
  std::vector<uint16_t> pixel_vec;
  std::vector<uint8_t> pixel_rawADC;
  int nrows;
  int ncols;
};

SiPixelArrayBuffer::SiPixelArrayBuffer(int rows, int cols)
    : pixel_vec(rows * cols, 0), pixel_rawADC(rows * cols, 0), nrows(rows), ncols(cols) {}

void SiPixelArrayBuffer::setSize(int rows, int cols) {
  pixel_vec.resize(rows * cols, 0);
  pixel_rawADC.resize(rows * cols, 0);
  nrows = rows;
  ncols = cols;
}

bool SiPixelArrayBuffer::inside(int row, int col) const { return (row >= 0 && row < nrows && col >= 0 && col < ncols); }

int SiPixelArrayBuffer::operator()(int row, int col) const { return pixel_vec[index(row, col)]; }

int SiPixelArrayBuffer::operator()(const SiPixelCluster::PixelPos& pix) const { return pixel_vec[index(pix)]; }

// unchecked!
void SiPixelArrayBuffer::set_adc(int row, int col, int adc) { pixel_vec[index(row, col)] = clamp(adc); }

void SiPixelArrayBuffer::set_adc(const SiPixelCluster::PixelPos& pix, int adc) { pixel_vec[index(pix)] = clamp(adc); }

void SiPixelArrayBuffer::add_adc(int row, int col, int adc) {
  auto& v = pixel_vec[index(row, col)];
  v = clamp(v + adc);
}

#endif