1c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This file is part of Eigen, a lightweight C++ template library 2c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// for linear algebra. 3c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 4c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Copyright (C) 2008-2010 Gael Guennebaud <gael.guennebaud@inria.fr> 5c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 6c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This Source Code Form is subject to the terms of the Mozilla 7c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Public License v. 2.0. If a copy of the MPL was not distributed 8c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 9c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 10c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#ifndef EIGEN_SPARSEDENSEPRODUCT_H 11c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#define EIGEN_SPARSEDENSEPRODUCT_H 12c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 13c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathnamespace Eigen { 14c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 15c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs, int InnerSize> struct SparseDenseProductReturnType 16c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 17c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef SparseTimeDenseProduct<Lhs,Rhs> Type; 18c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 19c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 20c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs> struct SparseDenseProductReturnType<Lhs,Rhs,1> 21c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 227faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef typename internal::conditional< 237faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Lhs::IsRowMajor, 247faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez SparseDenseOuterProduct<Rhs,Lhs,true>, 257faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez SparseDenseOuterProduct<Lhs,Rhs,false> >::type Type; 26c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 27c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 28c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs, int InnerSize> struct DenseSparseProductReturnType 29c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 30c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef DenseTimeSparseProduct<Lhs,Rhs> Type; 31c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 32c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 33c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs> struct DenseSparseProductReturnType<Lhs,Rhs,1> 34c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 357faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef typename internal::conditional< 367faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Rhs::IsRowMajor, 377faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez SparseDenseOuterProduct<Rhs,Lhs,true>, 387faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez SparseDenseOuterProduct<Lhs,Rhs,false> >::type Type; 39c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 40c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 41c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathnamespace internal { 42c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 43c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs, bool Tr> 44c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct traits<SparseDenseOuterProduct<Lhs,Rhs,Tr> > 45c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 46c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef Sparse StorageKind; 47c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename scalar_product_traits<typename traits<Lhs>::Scalar, 487faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typename traits<Rhs>::Scalar>::ReturnType Scalar; 49c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::Index Index; 50c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::Nested LhsNested; 51c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Rhs::Nested RhsNested; 52c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename remove_all<LhsNested>::type _LhsNested; 53c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename remove_all<RhsNested>::type _RhsNested; 54c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 55c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath enum { 56c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath LhsCoeffReadCost = traits<_LhsNested>::CoeffReadCost, 57c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath RhsCoeffReadCost = traits<_RhsNested>::CoeffReadCost, 58c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 59c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath RowsAtCompileTime = Tr ? int(traits<Rhs>::RowsAtCompileTime) : int(traits<Lhs>::RowsAtCompileTime), 60c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath ColsAtCompileTime = Tr ? int(traits<Lhs>::ColsAtCompileTime) : int(traits<Rhs>::ColsAtCompileTime), 61c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath MaxRowsAtCompileTime = Tr ? int(traits<Rhs>::MaxRowsAtCompileTime) : int(traits<Lhs>::MaxRowsAtCompileTime), 62c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath MaxColsAtCompileTime = Tr ? int(traits<Lhs>::MaxColsAtCompileTime) : int(traits<Rhs>::MaxColsAtCompileTime), 63c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 64c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Flags = Tr ? RowMajorBit : 0, 65c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 66c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath CoeffReadCost = LhsCoeffReadCost + RhsCoeffReadCost + NumTraits<Scalar>::MulCost 67c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath }; 68c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 69c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 70c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} // end namespace internal 71c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 72c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs, bool Tr> 73c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass SparseDenseOuterProduct 74c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : public SparseMatrixBase<SparseDenseOuterProduct<Lhs,Rhs,Tr> > 75c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 76c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 77c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 78c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef SparseMatrixBase<SparseDenseOuterProduct> Base; 79c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_DENSE_PUBLIC_INTERFACE(SparseDenseOuterProduct) 80c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef internal::traits<SparseDenseOuterProduct> Traits; 81c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 82c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath private: 83c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 84c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Traits::LhsNested LhsNested; 85c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Traits::RhsNested RhsNested; 86c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Traits::_LhsNested _LhsNested; 87c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Traits::_RhsNested _RhsNested; 88c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 89c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 90c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 91c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath class InnerIterator; 92c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 93c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE SparseDenseOuterProduct(const Lhs& lhs, const Rhs& rhs) 94c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : m_lhs(lhs), m_rhs(rhs) 95c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 96c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STATIC_ASSERT(!Tr,YOU_MADE_A_PROGRAMMING_MISTAKE); 97c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 98c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 99c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE SparseDenseOuterProduct(const Rhs& rhs, const Lhs& lhs) 100c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : m_lhs(lhs), m_rhs(rhs) 101c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 102c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STATIC_ASSERT(Tr,YOU_MADE_A_PROGRAMMING_MISTAKE); 103c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 104c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 105c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE Index rows() const { return Tr ? m_rhs.rows() : m_lhs.rows(); } 106c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE Index cols() const { return Tr ? m_lhs.cols() : m_rhs.cols(); } 107c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 108c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE const _LhsNested& lhs() const { return m_lhs; } 109c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE const _RhsNested& rhs() const { return m_rhs; } 110c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 111c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath protected: 112c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath LhsNested m_lhs; 113c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath RhsNested m_rhs; 114c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 115c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 116c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs, bool Transpose> 117c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass SparseDenseOuterProduct<Lhs,Rhs,Transpose>::InnerIterator : public _LhsNested::InnerIterator 118c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 119c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename _LhsNested::InnerIterator Base; 1207faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typedef typename SparseDenseOuterProduct::Index Index; 121c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 122c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_STRONG_INLINE InnerIterator(const SparseDenseOuterProduct& prod, Index outer) 1237faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez : Base(prod.lhs(), 0), m_outer(outer), m_factor(get(prod.rhs(), outer, typename internal::traits<Rhs>::StorageKind() )) 1247faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez { } 125c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 126c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath inline Index outer() const { return m_outer; } 1277faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez inline Index row() const { return Transpose ? m_outer : Base::index(); } 1287faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez inline Index col() const { return Transpose ? Base::index() : m_outer; } 129c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 130c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath inline Scalar value() const { return Base::value() * m_factor; } 131c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 132c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath protected: 1337faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez static Scalar get(const _RhsNested &rhs, Index outer, Dense = Dense()) 1347faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez { 1357faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez return rhs.coeff(outer); 1367faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez } 1377faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 1387faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez static Scalar get(const _RhsNested &rhs, Index outer, Sparse = Sparse()) 1397faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez { 1407faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez typename Traits::_RhsNested::InnerIterator it(rhs, outer); 1417faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez if (it && it.index()==0) 1427faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez return it.value(); 1437faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 1447faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez return Scalar(0); 1457faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez } 1467faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez 1477faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index m_outer; 148c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar m_factor; 149c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 150c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 151c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathnamespace internal { 152c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs> 153c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct traits<SparseTimeDenseProduct<Lhs,Rhs> > 154c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : traits<ProductBase<SparseTimeDenseProduct<Lhs,Rhs>, Lhs, Rhs> > 155c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 156c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef Dense StorageKind; 157c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef MatrixXpr XprKind; 158c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 159c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 160c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename SparseLhsType, typename DenseRhsType, typename DenseResType, 161c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath int LhsStorageOrder = ((SparseLhsType::Flags&RowMajorBit)==RowMajorBit) ? RowMajor : ColMajor, 162c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath bool ColPerCol = ((DenseRhsType::Flags&RowMajorBit)==0) || DenseRhsType::ColsAtCompileTime==1> 163c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct sparse_time_dense_product_impl; 164c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 165c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename SparseLhsType, typename DenseRhsType, typename DenseResType> 166c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct sparse_time_dense_product_impl<SparseLhsType,DenseRhsType,DenseResType, RowMajor, true> 167c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 168c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<SparseLhsType>::type Lhs; 169c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseRhsType>::type Rhs; 170c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseResType>::type Res; 171c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::Index Index; 172c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::InnerIterator LhsInnerIterator; 1737faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez static void run(const SparseLhsType& lhs, const DenseRhsType& rhs, DenseResType& res, const typename Res::Scalar& alpha) 174c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 175c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(Index c=0; c<rhs.cols(); ++c) 176c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 1777faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez Index n = lhs.outerSize(); 178c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(Index j=0; j<n; ++j) 179c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 180c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typename Res::Scalar tmp(0); 181c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(LhsInnerIterator it(lhs,j); it ;++it) 182c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath tmp += it.value() * rhs.coeff(it.index(),c); 183c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath res.coeffRef(j,c) = alpha * tmp; 184c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 185c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 186c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 187c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 188c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 189c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename SparseLhsType, typename DenseRhsType, typename DenseResType> 190c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct sparse_time_dense_product_impl<SparseLhsType,DenseRhsType,DenseResType, ColMajor, true> 191c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 192c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<SparseLhsType>::type Lhs; 193c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseRhsType>::type Rhs; 194c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseResType>::type Res; 195c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::InnerIterator LhsInnerIterator; 196c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::Index Index; 1977faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez static void run(const SparseLhsType& lhs, const DenseRhsType& rhs, DenseResType& res, const typename Res::Scalar& alpha) 198c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 199c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(Index c=0; c<rhs.cols(); ++c) 200c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 201c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(Index j=0; j<lhs.outerSize(); ++j) 202c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 203c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typename Res::Scalar rhs_j = alpha * rhs.coeff(j,c); 204c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(LhsInnerIterator it(lhs,j); it ;++it) 205c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath res.coeffRef(it.index(),c) += it.value() * rhs_j; 206c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 207c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 208c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 209c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 210c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 211c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename SparseLhsType, typename DenseRhsType, typename DenseResType> 212c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct sparse_time_dense_product_impl<SparseLhsType,DenseRhsType,DenseResType, RowMajor, false> 213c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 214c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<SparseLhsType>::type Lhs; 215c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseRhsType>::type Rhs; 216c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseResType>::type Res; 217c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::InnerIterator LhsInnerIterator; 218c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::Index Index; 2197faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez static void run(const SparseLhsType& lhs, const DenseRhsType& rhs, DenseResType& res, const typename Res::Scalar& alpha) 220c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 221c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(Index j=0; j<lhs.outerSize(); ++j) 222c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 223c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typename Res::RowXpr res_j(res.row(j)); 224c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(LhsInnerIterator it(lhs,j); it ;++it) 225c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath res_j += (alpha*it.value()) * rhs.row(it.index()); 226c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 227c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 228c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 229c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 230c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename SparseLhsType, typename DenseRhsType, typename DenseResType> 231c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct sparse_time_dense_product_impl<SparseLhsType,DenseRhsType,DenseResType, ColMajor, false> 232c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 233c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<SparseLhsType>::type Lhs; 234c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseRhsType>::type Rhs; 235c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename internal::remove_all<DenseResType>::type Res; 236c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::InnerIterator LhsInnerIterator; 237c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef typename Lhs::Index Index; 2387faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez static void run(const SparseLhsType& lhs, const DenseRhsType& rhs, DenseResType& res, const typename Res::Scalar& alpha) 239c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 240c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(Index j=0; j<lhs.outerSize(); ++j) 241c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 242c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typename Rhs::ConstRowXpr rhs_j(rhs.row(j)); 243c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for(LhsInnerIterator it(lhs,j); it ;++it) 244c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath res.row(it.index()) += (alpha*it.value()) * rhs_j; 245c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 246c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 247c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 248c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 249c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename SparseLhsType, typename DenseRhsType, typename DenseResType,typename AlphaType> 250c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathinline void sparse_time_dense_product(const SparseLhsType& lhs, const DenseRhsType& rhs, DenseResType& res, const AlphaType& alpha) 251c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 252c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath sparse_time_dense_product_impl<SparseLhsType,DenseRhsType,DenseResType>::run(lhs, rhs, res, alpha); 253c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 254c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 255c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} // end namespace internal 256c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 257c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs> 258c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass SparseTimeDenseProduct 259c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : public ProductBase<SparseTimeDenseProduct<Lhs,Rhs>, Lhs, Rhs> 260c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 261c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 262c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_PRODUCT_PUBLIC_INTERFACE(SparseTimeDenseProduct) 263c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 264c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath SparseTimeDenseProduct(const Lhs& lhs, const Rhs& rhs) : Base(lhs,rhs) 265c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath {} 266c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 2677faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez template<typename Dest> void scaleAndAddTo(Dest& dest, const Scalar& alpha) const 268c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 269c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath internal::sparse_time_dense_product(m_lhs, m_rhs, dest, alpha); 270c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 271c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 272c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath private: 273c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath SparseTimeDenseProduct& operator=(const SparseTimeDenseProduct&); 274c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 275c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 276c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 277c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// dense = dense * sparse 278c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathnamespace internal { 279c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs> 280c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct traits<DenseTimeSparseProduct<Lhs,Rhs> > 281c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : traits<ProductBase<DenseTimeSparseProduct<Lhs,Rhs>, Lhs, Rhs> > 282c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 283c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef Dense StorageKind; 284c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 285c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} // end namespace internal 286c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 287c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Lhs, typename Rhs> 288c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass DenseTimeSparseProduct 289c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath : public ProductBase<DenseTimeSparseProduct<Lhs,Rhs>, Lhs, Rhs> 290c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 291c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath public: 292c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_PRODUCT_PUBLIC_INTERFACE(DenseTimeSparseProduct) 293c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 294c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath DenseTimeSparseProduct(const Lhs& lhs, const Rhs& rhs) : Base(lhs,rhs) 295c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath {} 296c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 2977faaa9f3f0df9d23790277834d426c3d992ac3baCarlos Hernandez template<typename Dest> void scaleAndAddTo(Dest& dest, const Scalar& alpha) const 298c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 299c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Transpose<const _LhsNested> lhs_t(m_lhs); 300c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Transpose<const _RhsNested> rhs_t(m_rhs); 301c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Transpose<Dest> dest_t(dest); 302c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath internal::sparse_time_dense_product(rhs_t, lhs_t, dest_t, alpha); 303c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 304c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 305c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath private: 306c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath DenseTimeSparseProduct& operator=(const DenseTimeSparseProduct&); 307c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 308c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 309c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// sparse * dense 310c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename Derived> 311c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<typename OtherDerived> 312c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathinline const typename SparseDenseProductReturnType<Derived,OtherDerived>::Type 313c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan KamathSparseMatrixBase<Derived>::operator*(const MatrixBase<OtherDerived> &other) const 314c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 315c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return typename SparseDenseProductReturnType<Derived,OtherDerived>::Type(derived(), other.derived()); 316c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 317c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 318c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} // end namespace Eigen 319c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 320c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#endif // EIGEN_SPARSEDENSEPRODUCT_H 321