matrix.h

Go to the documentation of this file.

//
// APFEL++ 2017
//
// Author: Valerio Bertone: valerio.bertone@cern.ch
//
 
#pragma once
 
#include <vector>
#include <array>
#include <stddef.h>
#include <cstddef>
 
namespace apfel
{
  template<typename T>
  class matrix
  {
  public:
    matrix(size_t const& row = 0, size_t const& col = 0, std::vector<T> const& d = {});
 
    matrix(matrix<T> const& m);
 
    void resize(size_t const& row, size_t const& col, T const& v = 0);
 
    void set(T const& v);
 
    size_t const& size(size_t const& dim) const { return _size[dim]; }
 
    std::array<size_t, 2> const& size() const { return _size; }
 
    std::vector<T> const& data() const { return _data; }
 
    T&       operator () (size_t const& i, size_t const& j)       { return _data[i * _size[1] + j]; }
    T const& operator () (size_t const& i, size_t const& j) const { return _data[i * _size[1] + j]; }
    matrix<T>& operator  = (matrix<T> const& m);
    matrix<T>& operator += (matrix<T> const& m);
    matrix<T>& operator -= (matrix<T> const& m);
    matrix<T>& operator *= (double const& f);
    matrix<T>& operator /= (double const& f);
    matrix<T>& operator *= (matrix<T> const& m);
  private:
    std::array<size_t, 2> _size; 
    std::vector<T>        _data; 
  };
 
  template<class T>
  matrix<T> operator + (matrix<T> lhs, matrix<T> const& rhs); 
  template<class T>
  matrix<T> operator - (matrix<T> lhs, matrix<T> const& rhs); 
  template<class T>
  matrix<T> operator * (double const& s, matrix<T> rhs);      
  template<class T>
  matrix<T> operator * (matrix<T> lhs, double const& s);      
  template<class T>
  matrix<T> operator / (matrix<T> lhs, double const& s);      
  template<class T>
  matrix<T> operator * (matrix<T> lhs, matrix<T> const& rhs); 
}