core/Matrix.hpp

Go to the documentation of this file.

/*

  This file is part of OpenFLUID software
  Copyright(c) 2007, INRA - Montpellier SupAgro


 == GNU General Public License Usage ==

  OpenFLUID is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  OpenFLUID is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with OpenFLUID. If not, see <http://www.gnu.org/licenses/>.


 == Other Usage ==

  Other Usage means a use of OpenFLUID that is inconsistent with the GPL
  license, and requires a written agreement between You and INRA.
  Licensees for Other Usage of OpenFLUID may use this file in accordance
  with the terms contained in the written agreement between You and INRA.
  
*/



/**
  \file Matrix.hpp
  \brief Header of ...

  \author Jean-Christophe FABRE <fabrejc@supagro.inra.fr>
 */


#ifndef __MATRIX_HPP___
#define __MATRIX_HPP___



#include <iostream>
#include <openfluid/dllexport.hpp>
#include <openfluid/base/FrameworkException.hpp>
#include <boost/multi_array.hpp>

namespace openfluid { namespace core {


/**
  Template class for matrix data
*/
template <class T>
class DLLEXPORT Matrix
{
  protected :

    boost::multi_array<T, 2> m_Data;

    unsigned long m_ColsNbr;

    unsigned long m_RowsNbr;


  public :

    /**
      Default constructor, creates an empty Matrix
    */
    Matrix();

    /**
      Copy constructor
     */
    Matrix(const Matrix &Matrix);

    /**
      Constructor, creates a Matrix containing Size elements
    */
    Matrix(unsigned long ColsNbr,unsigned long RowsNbr);

    /**
      Constructor, creates a Matrix containing Size elements, initialized with value InitValue
    */
    Matrix(unsigned long ColsNbr,unsigned long RowsNbr, T InitValue);

    /**
      Destructor
    */
    virtual ~Matrix() {};

    /**
      Returns the number of columns of the Matrix
      @return number of columns
    */
    inline unsigned long getColsNbr() const { return m_ColsNbr; };

    /**
      Returns the number of rows of the Matrix
      @return number of rows
    */
    inline unsigned long getRowsNbr() const { return m_RowsNbr; };

    /**
      Returns the full size of the Matrix (number of columns x number of rows)
      @return size of the Matrix
    */
    inline unsigned long getSize() const { return (m_ColsNbr * m_RowsNbr); };

    /**
      Returns the full size of the Matrix
    */
    inline unsigned long size() const { return (m_ColsNbr * m_RowsNbr); };


    /**
      Returns a pointer to the content of the Matrix (like C arrays)
    */
    T* getData() const { return (T*)(m_Data.data()); };

    /**
      Sets data from a pointer to a content (like C arrays)
    */
    void setData(T* Data);

    /**
      Returns the element of the Matrix for index Index
    */
    T getElement(unsigned long ColIndex, unsigned long RowIndex) const;

    /**
      Returns the element of the Matrix for index Index
    */
    inline T at(unsigned long ColIndex, unsigned long RowIndex) const { return getElement(ColIndex,RowIndex); };

    /**
      Returns the element of the Matrix for index Index
    */
    inline T get(unsigned long ColIndex, unsigned long RowIndex) const { return getElement(ColIndex,RowIndex); };


    /**
      Sets a new value for element at the given index
    */
    void setElement(unsigned long ColIndex, unsigned long RowIndex, T Element);

    /**
      Sets a new value for element at the given index
    */
    inline void set(unsigned long ColIndex, unsigned long RowIndex, T Element) { setElement(ColIndex,RowIndex,Element); };

    /**
      Allocation operator
    */
    Matrix<T>& operator = (const Matrix &A);

    /**
      Fills the Matrix with given value
    */
    void fill(const T& Val);

    /**
      Clears the Matrix (empty and size is 0)
    */
    void clear();

};

// =====================================================================
// =====================================================================

template <class T>
Matrix<T>::Matrix():
  m_Data(boost::extents[0][0],boost::fortran_storage_order()),
  m_ColsNbr(0),m_RowsNbr(0)
{
}


// =====================================================================
// =====================================================================

template <class T>
Matrix<T>::Matrix(const Matrix &A):
  m_Data(boost::extents[A.m_ColsNbr][A.m_RowsNbr],boost::fortran_storage_order()),
  m_ColsNbr(A.m_ColsNbr),m_RowsNbr(A.m_RowsNbr)
{
  m_Data = A.m_Data;
}


// =====================================================================
// =====================================================================

template <class T>
Matrix<T>::Matrix(unsigned long ColsNbr, unsigned long RowsNbr) :
  m_Data(boost::extents[ColsNbr][RowsNbr],boost::fortran_storage_order()),
  m_ColsNbr(ColsNbr),m_RowsNbr(RowsNbr)
{
  fill(0);
}


// =====================================================================
// =====================================================================


template <class T>
Matrix<T>::Matrix(unsigned long ColsNbr, unsigned long RowsNbr, T InitValue) :
  m_Data(boost::extents[ColsNbr][RowsNbr],boost::fortran_storage_order()),
  m_ColsNbr(ColsNbr),m_RowsNbr(RowsNbr)
{
  fill(InitValue);
}


// =====================================================================
// =====================================================================


template <class T>
void Matrix<T>::setData(T* Data)
{
  for (unsigned long j=0; j < m_RowsNbr;j++)
  {
    for (unsigned long i=0; i < m_ColsNbr;i++)
    {
      m_Data[i][j] = Data[i+(j*m_ColsNbr)];
    }
  }
}


// =====================================================================
// =====================================================================


template <class T>
T Matrix<T>::getElement(unsigned long ColIndex, unsigned long RowIndex) const
{
  if (ColIndex >= m_ColsNbr || RowIndex >= m_RowsNbr) throw openfluid::base::FrameworkException("Matrix::getElement","element access range error");
  return m_Data[ColIndex][RowIndex];
}


// =====================================================================
// =====================================================================


template <class T>
void Matrix<T>::setElement(unsigned long ColIndex, unsigned long RowIndex, T Element)
{
  if (ColIndex >= m_ColsNbr || RowIndex >= m_RowsNbr) throw openfluid::base::FrameworkException("Matrix::setElement","element access range error");
  m_Data[ColIndex][RowIndex] = Element;
}



// =====================================================================
// =====================================================================

template <class T>
Matrix<T>& Matrix<T>::operator=(const Matrix &A)
{

  if (this == &A) return *this; // in case somebody tries assign array to itself

  m_Data.resize(boost::extents[A.m_ColsNbr][A.m_RowsNbr]);
  m_Data = A.m_Data;
  m_ColsNbr = A.m_ColsNbr;
  m_RowsNbr = A.m_RowsNbr;

  return *this;
}


// =====================================================================
// =====================================================================


template <class T>
void Matrix<T>::fill(const T& Val)
{
  for (unsigned long i=0;i<m_ColsNbr;i++)
    for (unsigned long j=0;j<m_RowsNbr;j++)
      m_Data[i][j] = Val;
}


// =====================================================================
// =====================================================================

template <class T>
void Matrix<T>::clear()
{
  m_Data.resize(boost::extents[0][0]);
  m_ColsNbr=0;
  m_RowsNbr=0;
}



} }

#endif /* __MATRIX_HPP___ */