WebSVN – gelsvn – Diff – /branches/wireframe-experiments/src/CGLA/ArithMatFloat.h


#ifndef __CGLA_ARITHMATFLOAT_H__
#define __CGLA_ARITHMATFLOAT_H__
 
#include <vector>
#include <iostream>
#include <numeric>
 
#include "CGLA.h"
 
 
namespace CGLA 
{
 
  /** \brief Basic class template for matrices.
		
  In this template a matrix is defined as an array of vectors. This may
  not in all cases be the most efficient but it has the advantage that 
  it is possible to use the double subscripting notation:
		
  T x = m[i][j]
 
  This template should be used through inheritance just like the 
  vector template */
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    class ArithMatFloat
    { 
    public:
 
      /// Horizontal vector type
      typedef HVT HVectorType;
 
      /// Vertical vector type
      typedef VVT VVectorType;
 
      /// The type of a matrix element
      typedef typename HVT::ScalarType ScalarType;
		
    protected:
 
      /// The actual contents of the matrix.
      HVT data[ROWS];
 
    protected:
 
      /// Construct 0 matrix
      ArithMatFloat() 
	{
#ifndef NDEBUG
	  std::fill_n(data, ROWS, HVT(CGLA_INIT_VALUE));
#endif
	}
 
      /// Construct a matrix where all entries are the same.
      explicit ArithMatFloat(ScalarType x)
	{
	  std::fill_n(data, ROWS, HVT(x));
	}
 
      /// Construct a matrix where all rows are the same.
      explicit ArithMatFloat(HVT _a)
	{
	  std::fill_n(data, ROWS, _a);
	}
 
      /// Construct a matrix with two rows.
      ArithMatFloat(HVT _a, HVT _b)
	{
	  assert(ROWS==2);
	  data[0] = _a;
	  data[1] = _b;
	}
 
      /// Construct a matrix with three rows.
      ArithMatFloat(HVT _a, HVT _b, HVT _c)
	{
	  assert(ROWS==3);
	  data[0] = _a;
	  data[1] = _b;
	  data[2] = _c;
	}
 
      /// Construct a matrix with four rows.
      ArithMatFloat(HVT _a, HVT _b, HVT _c, HVT _d)
	{
	  assert(ROWS==4);
	  data[0] = _a;
	  data[1] = _b;
	  data[2] = _c;
	  data[3] = _d;
	}
		
    public:
 
      /// Get vertical dimension of matrix 
      static unsigned int get_v_dim() {return VVT::get_dim();}
 
      /// Get horizontal dimension of matrix
      static unsigned int get_h_dim() {return HVT::get_dim();}
 
 
      /** Get const pointer to data array.
	  This function may be useful when interfacing with some other API 
	  such as OpenGL (TM). */
      const ScalarType* get() const 
	{
	  return data[0].get();
	}
 
      /** Get pointer to data array.
	  This function may be useful when interfacing with some other API 
	  such as OpenGL (TM). */
      ScalarType* get()
	{
	  return data[0].get();
	}
 
      //----------------------------------------------------------------------
      // index operators
      //----------------------------------------------------------------------
 
      /// Const index operator. Returns i'th row of matrix.
      const HVT& operator [] ( unsigned int i ) const
	{
	  assert(i<ROWS);
	  return data[i];
	}
 
      /// Non-const index operator. Returns i'th row of matrix.
      HVT& operator [] ( unsigned int i ) 
	{
	  assert(i<ROWS);
	  return data[i];
	}
 
      //----------------------------------------------------------------------
 
      /// Equality operator. 
      bool operator==(const MT& v) const 
	{
	  return std::inner_product(data, &data[ROWS], &v[0], true,
				    std::logical_and<bool>(), std::equal_to<HVT>());
	}
 
      /// Inequality operator.
      bool operator!=(const MT& v) const 
	{
	  return !(*this==v);
	}
 
      //----------------------------------------------------------------------
 
      /// Multiply scalar onto matrix. All entries are multiplied by scalar.
      const MT operator * (ScalarType k) const
	{
	  MT v_new;
	  std::transform(data, &data[ROWS], &v_new[0], std::bind2nd(std::multiplies<HVT>(), k));
	  return v_new;
	}
 
      /// Divide all entries in matrix by scalar.
      const MT operator / (ScalarType k) const
	{
	  MT v_new;
	  std::transform(data, &data[ROWS], &v_new[0], std::bind2nd(std::divides<HVT>(), k));
	  return v_new;      
	}
 
      /// Assignment multiplication of matrix by scalar.
      const MT& operator *=(ScalarType k) 
	{
	  std::transform(data, &data[ROWS], data, std::bind2nd(std::multiplies<HVT>(), k));
	  return static_cast<const MT&>(*this);
	}
 
      /// Assignment division of matrix by scalar.
      const MT& operator /=(ScalarType k) 
	{ 
	  std::transform(data, &data[ROWS], data, std::bind2nd(std::divides<HVT>(), k));
	  return static_cast<const MT&>(*this);
	}
 
      //----------------------------------------------------------------------
 
      /// Add two matrices. 
      const MT operator + (const MT& m1) const
	{
	  MT v_new;
	  std::transform(data, &data[ROWS], &m1[0], &v_new[0], std::plus<HVT>());
	  return v_new;
	}
 
      /// Subtract two matrices.
      const MT operator - (const MT& m1) const
	{
	  MT v_new;
	  std::transform(data, &data[ROWS], &m1[0], &v_new[0], std::minus<HVT>());
	  return v_new;
	}
 
      /// Assigment addition of matrices.
      const MT& operator +=(const MT& v) 
	{
	  std::transform(data, &data[ROWS], &v[0], data, std::plus<HVT>());
	  return static_cast<const MT&>(*this);
	}
 
      /// Assigment subtraction of matrices.
      const MT& operator -=(const MT& v) 
	{
	  std::transform(data, &data[ROWS], &v[0], data, std::minus<HVT>());
	  return static_cast<const MT&>(*this);
	}
 
      //----------------------------------------------------------------------
 
      /// Negate matrix.
      const MT operator - () const
	{
	  MT v_new;
	  std::transform(data, &data[ROWS], &v_new[0], std::negate<HVT>());
	  return v_new;
	}
    };
 
  /// Multiply scalar onto matrix
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    inline const MT operator * (double k, const ArithMatFloat<VVT,HVT,MT,ROWS>& v) 
    {
      return v * k;
    }
 
  /// Multiply scalar onto matrix
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    inline const MT operator * (float k, const ArithMatFloat<VVT,HVT,MT,ROWS>& v) 
    {
      return v * k;
    }
 
  /// Multiply scalar onto matrix
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    inline const MT operator * (int k, const ArithMatFloat<VVT,HVT,MT,ROWS>& v) 
    {
      return v * k;
    }
 
  /// Multiply vector onto matrix 
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    inline VVT operator*(const ArithMatFloat<VVT,HVT,MT,ROWS>& m,const HVT& v) 
    {
      VVT v2;
      for(unsigned int i=0;i<ROWS;i++) v2[i] = dot(m[i], v);
      return v2;
    }
 
 
#ifndef WIN32
  /** Multiply two arbitrary matrices. 
      In principle, this function could return a matrix, but in general
      the new matrix will be of a type that is different from either of
      the two matrices that are multiplied together. We do not want to 
      return an ArithMatFloat - so it seems best to let the return value be
      a reference arg.
		
      This template can only be instantiated if the dimensions of the
      matrices match -- i.e. if the multiplication can actually be
      carried out. This is more type safe than the win32 version below.
  */
 
  template <class VVT, class HVT, 
    class HV1T, class VV2T,
    class MT1, class MT2, class MT,
    unsigned int ROWS1, unsigned int ROWS2>
    inline void mul(const ArithMatFloat<VVT,HV1T,MT1,ROWS1>& m1,
		    const ArithMatFloat<VV2T,HVT,MT2,ROWS2>& m2,
		    ArithMatFloat<VVT,HVT,MT,ROWS1>& m)
    {
      unsigned int cols = ArithMatFloat<VVT,HVT,MT,ROWS1>::get_h_dim();
      for(unsigned int i=0;i<ROWS1;i++)
	for(unsigned int j=0;j<cols;j++)
	  {
	    m[i][j] = 0;
	    for(unsigned int k=0;k<ROWS2;k++)
	      m[i][j] += m1[i][k] * m2[k][j]; 
	  }
    }
 
 
  /** Transpose. See the discussion on mul if you are curious as to why
      I don't simply return the transpose. */
  template <class VVT, class HVT, class M1T, class M2T, unsigned int ROWS, unsigned int COLS>
    inline void transpose(const ArithMatFloat<VVT,HVT,M1T,ROWS>& m,
			  ArithMatFloat<HVT,VVT,M2T,COLS>& m_new)
    {
      for(unsigned int i=0;i<M2T::get_v_dim();++i)
	for(unsigned int j=0;j<M2T::get_h_dim();++j)
	  m_new[i][j] = m[j][i];
    }
 
#else
 
  //----------------- win32 -------------------------------
  // Visual studio is not good at deducing the args. to these template functions.
  // This means that you can call the two functions below with 
  // matrices of wrong dimension.
 
  template <class M1, class M2, class M>
    inline void mul(const M1& m1, const M2& m2, M& m)
    {
      unsigned int cols = M::get_h_dim();
      unsigned int rows1 = M1::get_v_dim();
      unsigned int rows2 = M2::get_v_dim();
 
      for(unsigned int i=0;i<rows1;++i)
	for(unsigned int j=0;j<cols;++j)
	  {
	    m[i][j] = 0;
	    for(unsigned int k=0;k<rows2;++k)
	      m[i][j] += m1[i][k] * m2[k][j];
	  }
    }
 
 
  /** Transpose. See the discussion on mul if you are curious as to why
      I don't simply return the transpose. */
  template <class M1, class M2>
    inline void transpose(const M1& m1, M2& m2)
    {
      for(unsigned int i=0;i<M2::get_v_dim();++i)
	for(unsigned int j=0;j<M2::get_h_dim();++j)
	  m2[i][j] = m1[j][i];
    }
 
#endif
 
  /** Compute the outer product of a and b: a * transpose(b). This is 
      a matrix with a::rows and b::columns. */
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    void outer_product(const VVT& a, const HVT& b, 
		       ArithMatFloat<VVT,HVT,MT,ROWS>& m)
    {
      unsigned int R = VVT::get_dim();
      unsigned int C = HVT::get_dim();
      for(unsigned int i=0;i<R;++i)
	for(unsigned int j=0;j<C;++j)
	  {
	    m[i][j] = a[i] * b[j];
	  }
    }
 
  /** Compute the outer product of a and b using an arbitrary 
      binary operation: op(a, transpose(b)). This is 
      a matrix with a::rows and b::columns. */
  template <class VVT, class HVT, class MT, int ROWS, class BinOp>
    void outer_product(const VVT& a, const HVT& b, 
		       ArithMatFloat<VVT,HVT,MT,ROWS>& m, BinOp op)
    {
      int R = VVT::get_dim();
      int C = HVT::get_dim();
      for(int i=0;i<R;++i)
	for(int j=0;j<C;++j)
	  {
	    m[i][j] = op(a[i], b[j]);
	  }
    }
 
  /** Copy a matrix to another matrix, cell by cell.
      This conversion that takes a const matrix as first argument
      (source) and a non-const matrix as second argument
      (destination). The contents of the first matrix is simply copied
      to the second matrix. 
			
      However, if the first matrix is	larger than the second,
      the cells outside the range of the destination are simply not
      copied. If the destination is larger, the cells outside the 
      range of the source matrix are not touched.
 
      An obvious use of this function is to copy a 3x3 rotation matrix
      into a 4x4 transformation matrix.
  */
 
  template <class M1, class M2>
    void copy_matrix(const M1& inmat, M2& outmat)
    {
      const unsigned int R = s_min(inmat.get_v_dim(), outmat.get_v_dim());
      const unsigned int C = s_min(inmat.get_h_dim(), outmat.get_h_dim());
      for(unsigned int i=0;i<R;++i)
	for(unsigned int j=0;j<C;++j)
	  outmat[i][j] = inmat[i][j];
    }
 
  /** Put to operator */
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    inline std::ostream& 
    operator<<(std::ostream&os, const ArithMatFloat<VVT,HVT,MT,ROWS>& m)
    {
      os << "[\n";
      for(unsigned int i=0;i<ROWS;i++) os << "  " << m[i] << "\n";
      os << "]\n";
      return os;
    }
 
  /** Get from operator */
  template <class VVT, class HVT, class MT, unsigned int ROWS>
    inline std::istream& operator>>(std::istream&is, 
				    const ArithMatFloat<VVT,HVT,MT,ROWS>& m)
    {
      for(unsigned int i=0;i<ROWS;i++) is>>m[i];
      return is;
    }
}
#endif
 

Subversion Repositories gelsvn

(root)/branches/wireframe-experiments/src/CGLA/ArithMatFloat.h – Rev 382 → 393

Rev 382		Rev 393
1	`#ifndef __CGLA_ARITHMATFLOAT_H__`	1	`#ifndef __CGLA_ARITHMATFLOAT_H__`
2	`#define __CGLA_ARITHMATFLOAT_H__`	2	`#define __CGLA_ARITHMATFLOAT_H__`
3		3
4	`#include <vector>`	4	`#include <vector>`
5	`#include <iostream>`	5	`#include <iostream>`
6	`#include <numeric>`	6	`#include <numeric>`
7		7
8	`#include "CGLA.h"`	8	`#include "CGLA.h"`
9		9
10		10
11	`namespace CGLA`	11	`namespace CGLA`
12	`{`	12	`{`
13		13
14	`/** \brief Basic class template for matrices.`	14	`/** \brief Basic class template for matrices.`
15		15
16	`In this template a matrix is defined as an array of vectors. This may`	16	`In this template a matrix is defined as an array of vectors. This may`
17	`not in all cases be the most efficient but it has the advantage that`	17	`not in all cases be the most efficient but it has the advantage that`
18	`it is possible to use the double subscripting notation:`	18	`it is possible to use the double subscripting notation:`
19		19
20	`T x = m[i][j]`	20	`T x = m[i][j]`
21		21
22	`This template should be used through inheritance just like the`	22	`This template should be used through inheritance just like the`
23	`vector template */`	23	`vector template */`
24	`template <class VVT, class HVT, class MT, unsigned int ROWS>`	24	`template <class VVT, class HVT, class MT, unsigned int ROWS>`
25	`class ArithMatFloat`	25	`class ArithMatFloat`
26	`{`	26	`{`
27	`public:`	27	`public:`
28		28
29	`/// Horizontal vector type`	29	`/// Horizontal vector type`
30	`typedef HVT HVectorType;`	30	`typedef HVT HVectorType;`
31		31
32	`/// Vertical vector type`	32	`/// Vertical vector type`
33	`typedef VVT VVectorType;`	33	`typedef VVT VVectorType;`
34		34
35	`/// The type of a matrix element`	35	`/// The type of a matrix element`
36	`typedef typename HVT::ScalarType ScalarType;`	36	`typedef typename HVT::ScalarType ScalarType;`
37		37
38	`protected:`	38	`protected:`
39		39
40	`/// The actual contents of the matrix.`	40	`/// The actual contents of the matrix.`