SiPixelArrayBuffer.h

Go to the documentation of this file.

#ifndef RecoLocalTracker_SiPixelClusterizer_SiPixelArrayBuffer_H
#define RecoLocalTracker_SiPixelClusterizer_SiPixelArrayBuffer_H
 
//----------------------------------------------------------------------------
//----------------------------------------------------------------------------
 
// 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 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);
  int size() const { return pixel_vec.size(); }
 
  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:
  std::vector<int> pixel_vec;  // TO DO: any benefit in using shorts instead?
  int nrows;
  int ncols;
};
 
SiPixelArrayBuffer::SiPixelArrayBuffer(int rows, int cols) : pixel_vec(rows * cols, 0), nrows(rows), ncols(cols) {}
 
void SiPixelArrayBuffer::setSize(int rows, int cols) {
  pixel_vec.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)] = adc; }
 
void SiPixelArrayBuffer::set_adc(const SiPixelCluster::PixelPos& pix, int adc) { pixel_vec[index(pix)] = adc; }
 
void SiPixelArrayBuffer::add_adc(int row, int col, int adc) { pixel_vec[index(row, col)] += adc; }
 
#endif