DtMatrix.h

Go to the documentation of this file.

/****************************************************************************** 
** Copyright (c) 2019 MAK Technologies, Inc. 
** All rights reserved. 
******************************************************************************/ 

#pragma once

#include <vector>

#include <vrvMath/DtMatrix3.h>
#include <vrvMath/DtMatrix4.h>

namespace makVrv {

   template<class T>
   class Matrix
   {
   public:
      Matrix(size_t numRows, size_t numCols);
      virtual ~Matrix();

      std::vector<T>& operator [](size_t row);

      const std::vector<T>& operator [](size_t row) const;

      Matrix operator * (const Matrix& rhsMatrix) const;

      size_t getNumRows(void) const;
      size_t getNumCols(void) const;

      size_t getNumElements(void) const {return getNumRows()*getNumCols();}
   private:
      std::vector< std::vector<T> > myElements;
   };

   template <class T>
   Matrix<T>::Matrix(size_t numRows, size_t numCols) 
   {
      Assert(numRows>=1 && numCols >= 1);
      myElements.resize(numRows);
      for(size_t i = 0; i < myElements.size(); ++i)
      {
         myElements[i].resize(numCols);
      }
   }

   template <class T>
   Matrix<T>::~Matrix() 
   {

   }

   template <class T>
   std::vector<T>& Matrix<T>::operator [](size_t row)
   {
      return myElements[row];
   }

   template <class T>
   const std::vector<T>& Matrix<T>::operator [](size_t row) const
   {
      return myElements[row];
   }

   template <class T>
   size_t Matrix<T>::getNumRows(void) const
   {
      return myElements.size();
   }
   template <class T>
   size_t Matrix<T>::getNumCols(void) const
   {
      if (myElements.empty())
         return 0;
      return myElements[0].size();
   }

   template <class T>
   Matrix<T> Matrix<T>::operator * (const Matrix& rhsMatrix) const
   {
      const Matrix& lhsMatrix = *this;
      Assert(getNumCols()==rhsMatrix.getNumRows());
      Matrix<T> result(getNumRows(), rhsMatrix.getNumCols());

      for(size_t i = 0; i < result.getNumRows(); ++i)
      {
         for (size_t j = 0; j < result.getNumCols(); ++j)
         {
            result[i][j] = 0;
            for(size_t k = 0; k < result.getNumCols(); ++k)
            {
               result[i][j] += lhsMatrix[i][k]*rhsMatrix[k][j];
            }
         }
      }
      return result;
   }

} //namespace makVrv