SimpleMatrix.h

Go to the documentation of this file.

#ifndef RecoBTag_Analysis_SimpleMatrix_h
#define RecoBTag_Analysis_SimpleMatrix_h
 
#include <memory>
#include <vector>
 
namespace btag {
 
  template <typename T>
  class SimpleMatrix {
  public:
    typedef T value_type;
    typedef typename std::vector<T>::size_type size_type;
 
    SimpleMatrix(size_type rows, size_type cols) : width(cols), height(rows), container(rows * cols) {}
 
    ~SimpleMatrix() {}
 
    inline size_type rows() const { return height; }
    inline size_type cols() const { return width; }
    inline size_type size() const { return container.size(); }
 
    inline double &operator()(size_type row, size_type col) { return container[index(row, col)]; }
    inline double operator()(size_type row, size_type col) const { return container[index(row, col)]; }
 
    inline double &operator[](size_type index) { return container[index]; }
    inline double operator[](size_type index) const { return container[index]; }
 
    inline size_type row(size_type index) const { return index / width; }
    inline size_type col(size_type index) const { return index % width; }
 
  protected:
    size_type index(size_type row, size_type col) const { return row * width + col; }
 
  private:
    size_type width, height;
    std::vector<T> container;
  };
 
}  // namespace btag
 
#endif  // GeneratorEvent_Analysis_SimpleMatrix_h