| // This file is part of Eigen, a lightweight C++ template library | 
 | // for linear algebra. | 
 | // | 
 | // Copyright (C) 2008 Gael Guennebaud <g.gael@free.fr> | 
 | // | 
 | // Eigen is free software; you can redistribute it and/or | 
 | // modify it under the terms of the GNU Lesser General Public | 
 | // License as published by the Free Software Foundation; either | 
 | // version 3 of the License, or (at your option) any later version. | 
 | // | 
 | // Alternatively, 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 2 of | 
 | // the License, or (at your option) any later version. | 
 | // | 
 | // Eigen 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 Lesser General Public License or the | 
 | // GNU General Public License for more details. | 
 | // | 
 | // You should have received a copy of the GNU Lesser General Public | 
 | // License and a copy of the GNU General Public License along with | 
 | // Eigen. If not, see <http://www.gnu.org/licenses/>. | 
 |  | 
 | #ifndef EIGEN_SPARSE_CWISE_BINARY_OP_H | 
 | #define EIGEN_SPARSE_CWISE_BINARY_OP_H | 
 |  | 
 | // Here we have to handle 3 cases: | 
 | //  1 - sparse op dense | 
 | //  2 - dense op sparse | 
 | //  3 - sparse op sparse | 
 | // We also need to implement a 4th iterator for: | 
 | //  4 - dense op dense | 
 | // Finally, we also need to distinguish between the product and other operations : | 
 | //                configuration      returned mode | 
 | //  1 - sparse op dense    product      sparse | 
 | //                         generic      dense | 
 | //  2 - dense op sparse    product      sparse | 
 | //                         generic      dense | 
 | //  3 - sparse op sparse   product      sparse | 
 | //                         generic      sparse | 
 | //  4 - dense op dense     product      dense | 
 | //                         generic      dense | 
 |  | 
 | template<> struct ei_promote_storage_type<Dense,Sparse> | 
 | { typedef Sparse ret; }; | 
 |  | 
 | template<> struct ei_promote_storage_type<Sparse,Dense> | 
 | { typedef Sparse ret; }; | 
 |  | 
 | template<typename BinaryOp, typename Lhs, typename Rhs> | 
 | class CwiseBinaryOpImpl<BinaryOp, Lhs, Rhs, Sparse> | 
 |   : public SparseMatrixBase<CwiseBinaryOp<BinaryOp, Lhs, Rhs> > | 
 | { | 
 |   public: | 
 |     class InnerIterator; | 
 |     typedef CwiseBinaryOp<BinaryOp, Lhs, Rhs> Derived; | 
 |     EIGEN_SPARSE_PUBLIC_INTERFACE(Derived) | 
 | }; | 
 |  | 
 | template<typename BinaryOp, typename Lhs, typename Rhs, typename Derived, | 
 |   typename _LhsStorageMode = typename ei_traits<Lhs>::StorageKind, | 
 |   typename _RhsStorageMode = typename ei_traits<Rhs>::StorageKind> | 
 | class ei_sparse_cwise_binary_op_inner_iterator_selector; | 
 |  | 
 | template<typename BinaryOp, typename Lhs, typename Rhs> | 
 | class CwiseBinaryOpImpl<BinaryOp,Lhs,Rhs,Sparse>::InnerIterator | 
 |   : public ei_sparse_cwise_binary_op_inner_iterator_selector<BinaryOp,Lhs,Rhs,typename CwiseBinaryOpImpl<BinaryOp,Lhs,Rhs,Sparse>::InnerIterator> | 
 | { | 
 |   public: | 
 |     typedef typename Lhs::Index Index; | 
 |     typedef ei_sparse_cwise_binary_op_inner_iterator_selector< | 
 |       BinaryOp,Lhs,Rhs, InnerIterator> Base; | 
 |  | 
 |     EIGEN_STRONG_INLINE InnerIterator(const CwiseBinaryOpImpl& binOp, Index outer) | 
 |       : Base(binOp.derived(),outer) | 
 |     {} | 
 | }; | 
 |  | 
 | /*************************************************************************** | 
 | * Implementation of inner-iterators | 
 | ***************************************************************************/ | 
 |  | 
 | // template<typename T> struct ei_func_is_conjunction { enum { ret = false }; }; | 
 | // template<typename T> struct ei_func_is_conjunction<ei_scalar_product_op<T> > { enum { ret = true }; }; | 
 |  | 
 | // TODO generalize the ei_scalar_product_op specialization to all conjunctions if any ! | 
 |  | 
 | // sparse - sparse  (generic) | 
 | template<typename BinaryOp, typename Lhs, typename Rhs, typename Derived> | 
 | class ei_sparse_cwise_binary_op_inner_iterator_selector<BinaryOp, Lhs, Rhs, Derived, Sparse, Sparse> | 
 | { | 
 |     typedef CwiseBinaryOp<BinaryOp, Lhs, Rhs> CwiseBinaryXpr; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::Scalar Scalar; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::_LhsNested _LhsNested; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::_RhsNested _RhsNested; | 
 |     typedef typename _LhsNested::InnerIterator LhsIterator; | 
 |     typedef typename _RhsNested::InnerIterator RhsIterator; | 
 |     typedef typename Lhs::Index Index; | 
 |  | 
 |   public: | 
 |  | 
 |     EIGEN_STRONG_INLINE ei_sparse_cwise_binary_op_inner_iterator_selector(const CwiseBinaryXpr& xpr, Index outer) | 
 |       : m_lhsIter(xpr.lhs(),outer), m_rhsIter(xpr.rhs(),outer), m_functor(xpr.functor()) | 
 |     { | 
 |       this->operator++(); | 
 |     } | 
 |  | 
 |     EIGEN_STRONG_INLINE Derived& operator++() | 
 |     { | 
 |       if (m_lhsIter && m_rhsIter && (m_lhsIter.index() == m_rhsIter.index())) | 
 |       { | 
 |         m_id = m_lhsIter.index(); | 
 |         m_value = m_functor(m_lhsIter.value(), m_rhsIter.value()); | 
 |         ++m_lhsIter; | 
 |         ++m_rhsIter; | 
 |       } | 
 |       else if (m_lhsIter && (!m_rhsIter || (m_lhsIter.index() < m_rhsIter.index()))) | 
 |       { | 
 |         m_id = m_lhsIter.index(); | 
 |         m_value = m_functor(m_lhsIter.value(), Scalar(0)); | 
 |         ++m_lhsIter; | 
 |       } | 
 |       else if (m_rhsIter && (!m_lhsIter || (m_lhsIter.index() > m_rhsIter.index()))) | 
 |       { | 
 |         m_id = m_rhsIter.index(); | 
 |         m_value = m_functor(Scalar(0), m_rhsIter.value()); | 
 |         ++m_rhsIter; | 
 |       } | 
 |       else | 
 |       { | 
 |         m_value = 0; // this is to avoid a compilation warning | 
 |         m_id = -1; | 
 |       } | 
 |       return *static_cast<Derived*>(this); | 
 |     } | 
 |  | 
 |     EIGEN_STRONG_INLINE Scalar value() const { return m_value; } | 
 |  | 
 |     EIGEN_STRONG_INLINE Index index() const { return m_id; } | 
 |     EIGEN_STRONG_INLINE Index row() const { return Lhs::IsRowMajor ? m_lhsIter.row() : index(); } | 
 |     EIGEN_STRONG_INLINE Index col() const { return Lhs::IsRowMajor ? index() : m_lhsIter.col(); } | 
 |  | 
 |     EIGEN_STRONG_INLINE operator bool() const { return m_id>=0; } | 
 |  | 
 |   protected: | 
 |     LhsIterator m_lhsIter; | 
 |     RhsIterator m_rhsIter; | 
 |     const BinaryOp& m_functor; | 
 |     Scalar m_value; | 
 |     Index m_id; | 
 | }; | 
 |  | 
 | // sparse - sparse  (product) | 
 | template<typename T, typename Lhs, typename Rhs, typename Derived> | 
 | class ei_sparse_cwise_binary_op_inner_iterator_selector<ei_scalar_product_op<T>, Lhs, Rhs, Derived, Sparse, Sparse> | 
 | { | 
 |     typedef ei_scalar_product_op<T> BinaryFunc; | 
 |     typedef CwiseBinaryOp<BinaryFunc, Lhs, Rhs> CwiseBinaryXpr; | 
 |     typedef typename CwiseBinaryXpr::Scalar Scalar; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::_LhsNested _LhsNested; | 
 |     typedef typename _LhsNested::InnerIterator LhsIterator; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::_RhsNested _RhsNested; | 
 |     typedef typename _RhsNested::InnerIterator RhsIterator; | 
 |     typedef typename Lhs::Index Index; | 
 |   public: | 
 |  | 
 |     EIGEN_STRONG_INLINE ei_sparse_cwise_binary_op_inner_iterator_selector(const CwiseBinaryXpr& xpr, Index outer) | 
 |       : m_lhsIter(xpr.lhs(),outer), m_rhsIter(xpr.rhs(),outer), m_functor(xpr.functor()) | 
 |     { | 
 |       while (m_lhsIter && m_rhsIter && (m_lhsIter.index() != m_rhsIter.index())) | 
 |       { | 
 |         if (m_lhsIter.index() < m_rhsIter.index()) | 
 |           ++m_lhsIter; | 
 |         else | 
 |           ++m_rhsIter; | 
 |       } | 
 |     } | 
 |  | 
 |     EIGEN_STRONG_INLINE Derived& operator++() | 
 |     { | 
 |       ++m_lhsIter; | 
 |       ++m_rhsIter; | 
 |       while (m_lhsIter && m_rhsIter && (m_lhsIter.index() != m_rhsIter.index())) | 
 |       { | 
 |         if (m_lhsIter.index() < m_rhsIter.index()) | 
 |           ++m_lhsIter; | 
 |         else | 
 |           ++m_rhsIter; | 
 |       } | 
 |       return *static_cast<Derived*>(this); | 
 |     } | 
 |  | 
 |     EIGEN_STRONG_INLINE Scalar value() const { return m_functor(m_lhsIter.value(), m_rhsIter.value()); } | 
 |  | 
 |     EIGEN_STRONG_INLINE Index index() const { return m_lhsIter.index(); } | 
 |     EIGEN_STRONG_INLINE Index row() const { return m_lhsIter.row(); } | 
 |     EIGEN_STRONG_INLINE Index col() const { return m_lhsIter.col(); } | 
 |  | 
 |     EIGEN_STRONG_INLINE operator bool() const { return (m_lhsIter && m_rhsIter); } | 
 |  | 
 |   protected: | 
 |     LhsIterator m_lhsIter; | 
 |     RhsIterator m_rhsIter; | 
 |     const BinaryFunc& m_functor; | 
 | }; | 
 |  | 
 | // sparse - dense  (product) | 
 | template<typename T, typename Lhs, typename Rhs, typename Derived> | 
 | class ei_sparse_cwise_binary_op_inner_iterator_selector<ei_scalar_product_op<T>, Lhs, Rhs, Derived, Sparse, Dense> | 
 | { | 
 |     typedef ei_scalar_product_op<T> BinaryFunc; | 
 |     typedef CwiseBinaryOp<BinaryFunc, Lhs, Rhs> CwiseBinaryXpr; | 
 |     typedef typename CwiseBinaryXpr::Scalar Scalar; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::_LhsNested _LhsNested; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::RhsNested RhsNested; | 
 |     typedef typename _LhsNested::InnerIterator LhsIterator; | 
 |     typedef typename Lhs::Index Index; | 
 |     enum { IsRowMajor = (int(Lhs::Flags)&RowMajorBit)==RowMajorBit }; | 
 |   public: | 
 |  | 
 |     EIGEN_STRONG_INLINE ei_sparse_cwise_binary_op_inner_iterator_selector(const CwiseBinaryXpr& xpr, Index outer) | 
 |       : m_rhs(xpr.rhs()), m_lhsIter(xpr.lhs(),outer), m_functor(xpr.functor()), m_outer(outer) | 
 |     {} | 
 |  | 
 |     EIGEN_STRONG_INLINE Derived& operator++() | 
 |     { | 
 |       ++m_lhsIter; | 
 |       return *static_cast<Derived*>(this); | 
 |     } | 
 |  | 
 |     EIGEN_STRONG_INLINE Scalar value() const | 
 |     { return m_functor(m_lhsIter.value(), | 
 |                        m_rhs.coeff(IsRowMajor?m_outer:m_lhsIter.index(),IsRowMajor?m_lhsIter.index():m_outer)); } | 
 |  | 
 |     EIGEN_STRONG_INLINE Index index() const { return m_lhsIter.index(); } | 
 |     EIGEN_STRONG_INLINE Index row() const { return m_lhsIter.row(); } | 
 |     EIGEN_STRONG_INLINE Index col() const { return m_lhsIter.col(); } | 
 |  | 
 |     EIGEN_STRONG_INLINE operator bool() const { return m_lhsIter; } | 
 |  | 
 |   protected: | 
 |     const RhsNested m_rhs; | 
 |     LhsIterator m_lhsIter; | 
 |     const BinaryFunc m_functor; | 
 |     const Index m_outer; | 
 | }; | 
 |  | 
 | // sparse - dense  (product) | 
 | template<typename T, typename Lhs, typename Rhs, typename Derived> | 
 | class ei_sparse_cwise_binary_op_inner_iterator_selector<ei_scalar_product_op<T>, Lhs, Rhs, Derived, Dense, Sparse> | 
 | { | 
 |     typedef ei_scalar_product_op<T> BinaryFunc; | 
 |     typedef CwiseBinaryOp<BinaryFunc, Lhs, Rhs> CwiseBinaryXpr; | 
 |     typedef typename CwiseBinaryXpr::Scalar Scalar; | 
 |     typedef typename ei_traits<CwiseBinaryXpr>::_RhsNested _RhsNested; | 
 |     typedef typename _RhsNested::InnerIterator RhsIterator; | 
 |     typedef typename Lhs::Index Index; | 
 |  | 
 |     enum { IsRowMajor = (int(Rhs::Flags)&RowMajorBit)==RowMajorBit }; | 
 |   public: | 
 |  | 
 |     EIGEN_STRONG_INLINE ei_sparse_cwise_binary_op_inner_iterator_selector(const CwiseBinaryXpr& xpr, Index outer) | 
 |       : m_xpr(xpr), m_rhsIter(xpr.rhs(),outer), m_functor(xpr.functor()), m_outer(outer) | 
 |     {} | 
 |  | 
 |     EIGEN_STRONG_INLINE Derived& operator++() | 
 |     { | 
 |       ++m_rhsIter; | 
 |       return *static_cast<Derived*>(this); | 
 |     } | 
 |  | 
 |     EIGEN_STRONG_INLINE Scalar value() const | 
 |     { return m_functor(m_xpr.lhs().coeff(IsRowMajor?m_outer:m_rhsIter.index(),IsRowMajor?m_rhsIter.index():m_outer), m_rhsIter.value()); } | 
 |  | 
 |     EIGEN_STRONG_INLINE Index index() const { return m_rhsIter.index(); } | 
 |     EIGEN_STRONG_INLINE Index row() const { return m_rhsIter.row(); } | 
 |     EIGEN_STRONG_INLINE Index col() const { return m_rhsIter.col(); } | 
 |  | 
 |     EIGEN_STRONG_INLINE operator bool() const { return m_rhsIter; } | 
 |  | 
 |   protected: | 
 |     const CwiseBinaryXpr& m_xpr; | 
 |     RhsIterator m_rhsIter; | 
 |     const BinaryFunc& m_functor; | 
 |     const Index m_outer; | 
 | }; | 
 |  | 
 |  | 
 | /*************************************************************************** | 
 | * Implementation of SparseMatrixBase and SparseCwise functions/operators | 
 | ***************************************************************************/ | 
 |  | 
 | // template<typename Derived> | 
 | // template<typename OtherDerived> | 
 | // EIGEN_STRONG_INLINE const CwiseBinaryOp<ei_scalar_difference_op<typename ei_traits<Derived>::Scalar>, | 
 | //                                  Derived, OtherDerived> | 
 | // SparseMatrixBase<Derived>::operator-(const SparseMatrixBase<OtherDerived> &other) const | 
 | // { | 
 | //   return CwiseBinaryOp<ei_scalar_difference_op<Scalar>, | 
 | //                        Derived, OtherDerived>(derived(), other.derived()); | 
 | // } | 
 |  | 
 | template<typename Derived> | 
 | template<typename OtherDerived> | 
 | EIGEN_STRONG_INLINE Derived & | 
 | SparseMatrixBase<Derived>::operator-=(const SparseMatrixBase<OtherDerived> &other) | 
 | { | 
 |   return *this = derived() - other.derived(); | 
 | } | 
 |  | 
 | // template<typename Derived> | 
 | // template<typename OtherDerived> | 
 | // EIGEN_STRONG_INLINE const CwiseBinaryOp<ei_scalar_sum_op<typename ei_traits<Derived>::Scalar>, Derived, OtherDerived> | 
 | // SparseMatrixBase<Derived>::operator+(const SparseMatrixBase<OtherDerived> &other) const | 
 | // { | 
 | //   return CwiseBinaryOp<ei_scalar_sum_op<Scalar>, Derived, OtherDerived>(derived(), other.derived()); | 
 | // } | 
 |  | 
 | template<typename Derived> | 
 | template<typename OtherDerived> | 
 | EIGEN_STRONG_INLINE Derived & | 
 | SparseMatrixBase<Derived>::operator+=(const SparseMatrixBase<OtherDerived>& other) | 
 | { | 
 |   return *this = derived() + other.derived(); | 
 | } | 
 |  | 
 | // template<typename ExpressionType> | 
 | // template<typename OtherDerived> | 
 | // EIGEN_STRONG_INLINE const EIGEN_SPARSE_CWISE_PRODUCT_RETURN_TYPE | 
 | // SparseCwise<ExpressionType>::operator*(const SparseMatrixBase<OtherDerived> &other) const | 
 | // { | 
 | //   return EIGEN_SPARSE_CWISE_PRODUCT_RETURN_TYPE(_expression(), other.derived()); | 
 | // } | 
 |  | 
 | template<typename Derived> | 
 | template<typename OtherDerived> | 
 | EIGEN_STRONG_INLINE const EIGEN_SPARSE_CWISE_PRODUCT_RETURN_TYPE | 
 | SparseMatrixBase<Derived>::cwiseProduct(const MatrixBase<OtherDerived> &other) const | 
 | { | 
 |   return EIGEN_SPARSE_CWISE_PRODUCT_RETURN_TYPE(derived(), other.derived()); | 
 | } | 
 |  | 
 | // template<typename ExpressionType> | 
 | // template<typename OtherDerived> | 
 | // EIGEN_STRONG_INLINE const EIGEN_SPARSE_CWISE_BINOP_RETURN_TYPE(ei_scalar_quotient_op) | 
 | // SparseCwise<ExpressionType>::operator/(const SparseMatrixBase<OtherDerived> &other) const | 
 | // { | 
 | //   return EIGEN_SPARSE_CWISE_BINOP_RETURN_TYPE(ei_scalar_quotient_op)(_expression(), other.derived()); | 
 | // } | 
 | // | 
 | // template<typename ExpressionType> | 
 | // template<typename OtherDerived> | 
 | // EIGEN_STRONG_INLINE const EIGEN_SPARSE_CWISE_BINOP_RETURN_TYPE(ei_scalar_quotient_op) | 
 | // SparseCwise<ExpressionType>::operator/(const MatrixBase<OtherDerived> &other) const | 
 | // { | 
 | //   return EIGEN_SPARSE_CWISE_BINOP_RETURN_TYPE(ei_scalar_quotient_op)(_expression(), other.derived()); | 
 | // } | 
 |  | 
 | // template<typename ExpressionType> | 
 | // template<typename OtherDerived> | 
 | // inline ExpressionType& SparseCwise<ExpressionType>::operator*=(const SparseMatrixBase<OtherDerived> &other) | 
 | // { | 
 | //   return m_matrix.const_cast_derived() = _expression() * other.derived(); | 
 | // } | 
 |  | 
 |  | 
 | #endif // EIGEN_SPARSE_CWISE_BINARY_OP_H |