matrix_vector_expressions.h
Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045 #ifndef LASS_GUARDIAN_OF_INCLUSION_NUM_IMPL_MATRIX_VECTOR_EXPRESSIONS_H
00046 #define LASS_GUARDIAN_OF_INCLUSION_NUM_IMPL_MATRIX_VECTOR_EXPRESSIONS_H
00047
00048 #include "../num_common.h"
00049
00050 namespace lass
00051 {
00052 namespace num
00053 {
00054 namespace impl
00055 {
00056
00057 template <typename T, typename VectorOperand1>
00058 class MVColumn
00059 {
00060 public:
00061 enum { lvalue = VectorOperand1::lvalue };
00062 typedef typename util::CallTraits<T>::TValue TValue;
00063 typedef typename util::CallTraits<T>::TReference TReference;
00064 typedef size_t TSize;
00065 MVColumn(VectorOperand1& iA): operand1_(iA) {}
00066 TValue operator()(TSize iI, TSize ) const { return operand1_[iI]; }
00067 TReference operator()(TSize iI, TSize ) { return operand1_[iI]; }
00068 TSize rows() const { return operand1_.size(); }\
00069 TSize columns() const { return 1; }\
00070
00071 void resize(TSize iRows, TSize iCols)
00072 {
00073 LASS_ENFORCE(iCols == 1);
00074 operand1_.resize(iRows);
00075 }
00076 private:
00077 VectorOperand1& operand1_;
00078 };
00079
00080
00081
00082 template <typename T, typename VectorOperand1>
00083 class MVDiag
00084 {
00085 public:
00086 enum { lvalue = false };
00087 typedef typename util::CallTraits<T>::TValue TValue;
00088 typedef size_t TSize;
00089 MVDiag(const VectorOperand1& iA): operand1_(iA) {}
00090 TValue operator()(TSize iI, TSize iJ) const
00091 {
00092 return iI == iJ ? operand1_[iI] : NumTraits<T>::zero;
00093 }
00094 TSize rows() const { return operand1_.size(); }\
00095 TSize columns() const { return operand1_.size(); }\
00096 private:
00097 typename VectorExpressionTraits<VectorOperand1>::TStorage operand1_;
00098 };
00099
00100
00101
00102 template <typename T, typename MatrixOperand1, typename VectorOperand2>
00103 class MVRightProd
00104 {
00105 public:
00106 enum { lvalue = false };
00107 typedef typename util::CallTraits<T>::TValue TValue;
00108 typedef size_t TSize;
00109 MVRightProd(const MatrixOperand1& iA, const VectorOperand2& iB):
00110 operand1_(iA), operand2_(iB)
00111 {
00112 LASS_ASSERT(iA.columns() == iB.size());
00113 }
00114 TValue operator[](TSize iIndex) const
00115 {
00116 const TSize n = operand1_.columns();
00117 TValue result = NumTraits<T>::zero;
00118 for (TSize k = 0; k < n; ++k)
00119 {
00120 result += operand1_(iIndex, k) * operand2_[k];
00121 }
00122 return result;
00123 }
00124 TSize size() const { return operand1_.rows(); }
00125 private:
00126 typename MatrixExpressionTraits<MatrixOperand1>::TStorage operand1_;
00127 typename VectorExpressionTraits<VectorOperand2>::TStorage operand2_;
00128 };
00129
00130
00131
00132 }
00133
00134 }
00135
00136 }
00137
00138 #endif
00139
00140