35#ifndef _BLAZE_MATH_EXPRESSIONS_SMATSMATADDEXPR_H_
36#define _BLAZE_MATH_EXPRESSIONS_SMATSMATADDEXPR_H_
93 :
public MatMatAddExpr< SparseMatrix< SMatSMatAddExpr<MT1,MT2>, false > >
113 static constexpr bool returnExpr = !IsTemporary_v<RN1> && !IsTemporary_v<RN2>;
126 template<
typename T1,
typename T2,
typename T3 >
127 static constexpr bool UseSymmetricKernel_v =
128 ( IsColumnMajorMatrix_v<T1> && IsSymmetric_v<T2> && IsSymmetric_v<T3> );
139 template<
typename MT >
140 static constexpr bool UseSMPAssign_v = MT::smpAssignable;
213 if( i >=
lhs_.rows() ) {
216 if( j >=
lhs_.columns() ) {
228 inline size_t rows() const noexcept {
239 return lhs_.columns();
249 return lhs_.nonZeros() +
rhs_.nonZeros();
260 return lhs_.nonZeros(i) +
rhs_.nonZeros(i);
290 template<
typename T >
291 inline bool canAlias(
const T* alias )
const noexcept {
292 return (
lhs_.canAlias( alias ) ||
rhs_.canAlias( alias ) );
302 template<
typename T >
303 inline bool isAliased(
const T* alias )
const noexcept {
304 return (
lhs_.isAliased( alias ) ||
rhs_.isAliased( alias ) );
326 template<
typename MT
335 assign( *lhs, rhs.
lhs_ );
338 addAssign( *lhs, rhs.
rhs_ );
349 for(
size_t i=0UL; i<(*lhs).rows(); ++i ) {
350 const auto end( B.end(i) );
351 for(
auto element=B.begin(i); element!=
end; ++element ) {
352 if(
isDefault( (*lhs)(i,element->index()) ) )
353 (*lhs)(i,element->index()) = element->value();
355 (*lhs)(i,element->index()) += element->value();
375 template<
typename MT >
394 (*lhs).reserve( A.nonZeros() + B.nonZeros() );
397 for(
size_t i=0UL; i<(*lhs).rows(); ++i )
399 const auto lend( A.end(i) );
400 const auto rend( B.end(i) );
402 auto l( A.begin(i) );
403 auto r( B.begin(i) );
405 while( l != lend && r != rend )
407 if( l->index() < r->index() ) {
408 (*lhs).append( i, l->index(), l->value() );
411 else if( l->index() > r->index() ) {
412 (*lhs).append( i, r->index(), r->value() );
416 (*lhs).append( i, l->index(), l->value() + r->value() );
423 (*lhs).append( i, l->index(), l->value() );
428 (*lhs).append( i, r->index(), r->value() );
432 (*lhs).finalize( i );
450 template<
typename MT >
451 friend inline auto assign( SparseMatrix<MT,true>& lhs,
const SMatSMatAddExpr& rhs )
452 -> DisableIf_t< UseSymmetricKernel_v<MT,MT1,MT2> >
471 const size_t m( A.rows() );
472 const size_t n( A.columns() );
475 std::vector<size_t> nonzeros( n, 0UL );
476 for(
size_t i=0UL; i<m; ++i )
478 const auto lend( A.end(i) );
479 const auto rend( B.end(i) );
481 auto l( A.begin(i) );
482 auto r( B.begin(i) );
484 while( l != lend && r != rend )
486 if( l->index() < r->index() ) {
487 ++nonzeros[l->index()];
490 else if( l->index() > r->index() ) {
491 ++nonzeros[r->index()];
495 ++nonzeros[l->index()];
502 ++nonzeros[l->index()];
507 ++nonzeros[r->index()];
513 for(
size_t j=0UL; j<n; ++j ) {
514 (*lhs).reserve( j, nonzeros[j] );
518 for(
size_t i=0UL; i<m; ++i )
520 const auto lend( A.end(i) );
521 const auto rend( B.end(i) );
523 auto l( A.begin(i) );
524 auto r( B.begin(i) );
526 while( l != lend && r != rend )
528 if( l->index() < r->index() ) {
529 (*lhs).append( i, l->index(), l->value() );
532 else if( l->index() > r->index() ) {
533 (*lhs).append( i, r->index(), r->value() );
537 (*lhs).append( i, l->index(), l->value() + r->value() );
544 (*lhs).append( i, l->index(), l->value() );
549 (*lhs).append( i, r->index(), r->value() );
569 template<
typename MT >
570 friend inline auto assign( SparseMatrix<MT,true>& lhs,
const SMatSMatAddExpr& rhs )
571 -> EnableIf_t< UseSymmetricKernel_v<MT,MT1,MT2> >
597 template<
typename MT
599 friend inline void addAssign( DenseMatrix<MT,SO>& lhs,
const SMatSMatAddExpr& rhs )
606 addAssign( *lhs, rhs.lhs_ );
607 addAssign( *lhs, rhs.rhs_ );
628 template<
typename MT
630 friend inline void subAssign( DenseMatrix<MT,SO>& lhs,
const SMatSMatAddExpr& rhs )
637 subAssign( *lhs, rhs.lhs_ );
638 subAssign( *lhs, rhs.rhs_ );
659 template<
typename MT
661 friend inline void schurAssign( DenseMatrix<MT,SO>& lhs,
const SMatSMatAddExpr& rhs )
673 schurAssign( *lhs, tmp );
712 template<
typename MT
715 -> EnableIf_t< UseSMPAssign_v<MT> >
746 template<
typename MT
749 -> EnableIf_t< UseSMPAssign_v<MT> >
781 template<
typename MT
784 -> EnableIf_t< UseSMPAssign_v<MT> >
846template<
typename MT1
848 ,
DisableIf_t< ( ( IsZero_v<MT1> || IsZero_v<MT2> ) &&
849 IsSame_v< ElementType_t<MT1>, ElementType_t<MT2> > ) ||
850 ( IsZero_v<MT1> && IsZero_v<MT2> ) >* =
nullptr >
851inline const SMatSMatAddExpr<MT1,MT2>
852 smatsmatadd(
const SparseMatrix<MT1,false>& lhs,
const SparseMatrix<MT2,false>& rhs )
859 return SMatSMatAddExpr<MT1,MT2>( *lhs, *rhs );
879template<
typename MT1
881 , EnableIf_t< !IsZero_v<MT1> && IsZero_v<MT2> &&
882 IsSame_v< ElementType_t<MT1>, ElementType_t<MT2> > >* =
nullptr >
884 smatsmatadd(
const SparseMatrix<MT1,false>& lhs,
const SparseMatrix<MT2,false>& rhs )
913template<
typename MT1
915 , EnableIf_t< IsZero_v<MT1> && !IsZero_v<MT2> &&
916 IsSame_v< ElementType_t<MT1>, ElementType_t<MT2> > >* =
nullptr >
918 smatsmatadd(
const SparseMatrix<MT1,false>& lhs,
const SparseMatrix<MT2,false>& rhs )
945template<
typename MT1
947 , EnableIf_t< IsZero_v<MT1> && IsZero_v<MT2> >* =
nullptr >
949 smatsmatadd(
const SparseMatrix<MT1,false>& lhs,
const SparseMatrix<MT2,false>& rhs )
958 using ReturnType =
const AddTrait_t< ResultType_t<MT1>, ResultType_t<MT2> >;
963 return ReturnType( (*lhs).rows(), (*lhs).columns() );
995template<
typename MT1
1002 if( (*lhs).rows() != (*rhs).rows() || (*lhs).columns() != (*rhs).columns() ) {
1006 return smatsmatadd( *lhs, *rhs );
Header file for the addition trait.
Header file for auxiliary alias declarations.
typename T::CompositeType CompositeType_t
Alias declaration for nested CompositeType type definitions.
Definition: Aliases.h:110
typename T::ReturnType ReturnType_t
Alias declaration for nested ReturnType type definitions.
Definition: Aliases.h:470
typename T::ResultType ResultType_t
Alias declaration for nested ResultType type definitions.
Definition: Aliases.h:450
typename T::ElementType ElementType_t
Alias declaration for nested ElementType type definitions.
Definition: Aliases.h:190
typename T::OppositeType OppositeType_t
Alias declaration for nested OppositeType type definitions.
Definition: Aliases.h:310
typename T::TransposeType TransposeType_t
Alias declaration for nested TransposeType type definitions.
Definition: Aliases.h:550
Header file for run time assertion macros.
Header file for the EnableIf class template.
Header file for the function trace functionality.
Header file for the If class template.
Header file for the IntegralConstant class template.
Header file for the IsColumnMajorMatrix type trait.
Header file for the isDefault shim.
Header file for the IsExpression type trait class.
Header file for the IsResizable type trait.
Header file for the IsSame and IsStrictlySame type traits.
Header file for the IsSymmetric type trait.
Header file for the IsTemporary type trait class.
Header file for the MAYBE_UNUSED function template.
Constraints on the storage order of matrix types.
Constraint on the data type.
Constraint on the data type.
Base class for dense matrices.
Definition: DenseMatrix.h:82
Expression object for sparse matrix-sparse matrix additions.
Definition: SMatSMatAddExpr.h:95
RightOperand rhs_
Right-hand side sparse matrix of the addition expression.
Definition: SMatSMatAddExpr.h:311
AddTrait_t< RT1, RT2 > ResultType
Result type for expression template evaluations.
Definition: SMatSMatAddExpr.h:152
TransposeType_t< ResultType > TransposeType
Transpose type for expression template evaluations.
Definition: SMatSMatAddExpr.h:154
SMatSMatAddExpr(const MT1 &lhs, const MT2 &rhs) noexcept
Constructor for the SMatSMatAddExpr class.
Definition: SMatSMatAddExpr.h:181
bool canAlias(const T *alias) const noexcept
Returns whether the expression can alias with the given address alias.
Definition: SMatSMatAddExpr.h:291
static constexpr bool smpAssignable
Compilation switch for the expression template assignment strategy.
Definition: SMatSMatAddExpr.h:172
LeftOperand leftOperand() const noexcept
Returns the left-hand side sparse matrix operand.
Definition: SMatSMatAddExpr.h:269
size_t nonZeros() const
Returns the number of non-zero elements in the sparse matrix.
Definition: SMatSMatAddExpr.h:248
decltype(std::declval< RN1 >()+std::declval< RN2 >()) ExprReturnType
Expression return type for the subscript operator.
Definition: SMatSMatAddExpr.h:116
ResultType_t< MT1 > RT1
Result type of the left-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:98
OppositeType_t< ResultType > OppositeType
Result type with opposite storage order for expression template evaluations.
Definition: SMatSMatAddExpr.h:153
CompositeType_t< MT1 > CT1
Composite type of the left-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:100
LeftOperand lhs_
Left-hand side sparse matrix of the addition expression.
Definition: SMatSMatAddExpr.h:310
RightOperand rightOperand() const noexcept
Returns the right-hand side sparse matrix operand.
Definition: SMatSMatAddExpr.h:279
const If_t< returnExpr, ExprReturnType, ElementType > ReturnType
Return type for expression template evaluations.
Definition: SMatSMatAddExpr.h:158
If_t< IsExpression_v< MT1 >, const MT1, const MT1 & > LeftOperand
Composite type of the left-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:164
ElementType_t< ResultType > ElementType
Resulting element type.
Definition: SMatSMatAddExpr.h:155
size_t rows() const noexcept
Returns the current number of rows of the matrix.
Definition: SMatSMatAddExpr.h:228
ReturnType at(size_t i, size_t j) const
Checked access to the matrix elements.
Definition: SMatSMatAddExpr.h:212
ReturnType_t< MT2 > RN2
Return type of the right-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:103
size_t columns() const noexcept
Returns the current number of columns of the matrix.
Definition: SMatSMatAddExpr.h:238
size_t nonZeros(size_t i) const
Returns the number of non-zero elements in the specified row.
Definition: SMatSMatAddExpr.h:259
bool isAliased(const T *alias) const noexcept
Returns whether the expression is aliased with the given address alias.
Definition: SMatSMatAddExpr.h:303
static constexpr bool returnExpr
Compilation switch for the selection of the subscript operator return type.
Definition: SMatSMatAddExpr.h:113
If_t< IsExpression_v< MT2 >, const MT2, const MT2 & > RightOperand
Composite type of the right-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:167
ResultType_t< MT2 > RT2
Result type of the right-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:99
ReturnType operator()(size_t i, size_t j) const
2D-access to the matrix elements.
Definition: SMatSMatAddExpr.h:197
const ResultType CompositeType
Data type for composite expression templates.
Definition: SMatSMatAddExpr.h:161
ReturnType_t< MT1 > RN1
Return type of the left-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:102
CompositeType_t< MT2 > CT2
Composite type of the right-hand side sparse matrix expression.
Definition: SMatSMatAddExpr.h:101
Base class for sparse matrices.
Definition: SparseMatrix.h:77
Constraint on the data type.
Constraint on the data type.
Constraint on the data type.
Header file for the Computation base class.
Header file for the MatMatAddExpr base class.
Header file for the SparseMatrix base class.
decltype(auto) trans(const DenseMatrix< MT, SO > &dm)
Calculation of the transpose of the given dense matrix.
Definition: DMatTransExpr.h:766
decltype(auto) serial(const DenseMatrix< MT, SO > &dm)
Forces the serial evaluation of the given dense matrix expression dm.
Definition: DMatSerialExpr.h:812
bool isDefault(const DiagonalMatrix< MT, SO, DF > &m)
Returns whether the given diagonal matrix is in default state.
Definition: DiagonalMatrix.h:169
#define BLAZE_CONSTRAINT_MUST_NOT_BE_SYMMETRIC_MATRIX_TYPE(T)
Constraint on the data type.
Definition: Symmetric.h:79
#define BLAZE_CONSTRAINT_MUST_BE_ROW_MAJOR_MATRIX_TYPE(T)
Constraint on the data type.
Definition: RowMajorMatrix.h:61
#define BLAZE_CONSTRAINT_MUST_NOT_REQUIRE_EVALUATION(T)
Constraint on the data type.
Definition: RequiresEvaluation.h:81
#define BLAZE_CONSTRAINT_MUST_BE_SPARSE_MATRIX_TYPE(T)
Constraint on the data type.
Definition: SparseMatrix.h:61
#define BLAZE_CONSTRAINT_MUST_FORM_VALID_MATMATADDEXPR(T1, T2)
Constraint on the data type.
Definition: MatMatAddExpr.h:103
#define BLAZE_CONSTRAINT_MUST_BE_ZERO_TYPE(T)
Constraint on the data type.
Definition: Zero.h:61
typename AddTrait< T1, T2 >::Type AddTrait_t
Auxiliary alias declaration for the AddTrait class template.
Definition: AddTrait.h:163
constexpr bool IsResizable_v
Auxiliary variable template for the IsResizable type trait.
Definition: IsResizable.h:136
MT::Iterator end(Matrix< MT, SO > &matrix, size_t i)
Returns an iterator just past the last element of row/column i.
Definition: Matrix.h:584
#define BLAZE_INTERNAL_ASSERT(expr, msg)
Run time assertion macro for internal checks.
Definition: Assert.h:101
auto smpSubAssign(Matrix< MT1, SO1 > &lhs, const Matrix< MT2, SO2 > &rhs) -> EnableIf_t< IsDenseMatrix_v< MT1 > >
Default implementation of the SMP subtraction assignment of a matrix to dense matrix.
Definition: DenseMatrix.h:162
auto smpSchurAssign(Matrix< MT1, SO1 > &lhs, const Matrix< MT2, SO2 > &rhs) -> EnableIf_t< IsDenseMatrix_v< MT1 > >
Default implementation of the SMP Schur product assignment of a matrix to dense matrix.
Definition: DenseMatrix.h:194
auto smpAddAssign(Matrix< MT1, SO1 > &lhs, const Matrix< MT2, SO2 > &rhs) -> EnableIf_t< IsDenseMatrix_v< MT1 > >
Default implementation of the SMP addition assignment of a matrix to a dense matrix.
Definition: DenseMatrix.h:131
constexpr bool IsSame_v
Auxiliary variable template for the IsSame type trait.
Definition: IsSame.h:159
constexpr void MAYBE_UNUSED(const Args &...)
Suppression of unused parameter warnings.
Definition: MaybeUnused.h:81
typename If< Condition >::template Type< T1, T2 > If_t
Auxiliary alias template for the If class template.
Definition: If.h:108
#define BLAZE_THROW_OUT_OF_RANGE(MESSAGE)
Macro for the emission of a std::out_of_range exception.
Definition: Exception.h:331
#define BLAZE_THROW_INVALID_ARGUMENT(MESSAGE)
Macro for the emission of a std::invalid_argument exception.
Definition: Exception.h:235
typename EnableIf<!Condition, T >::Type DisableIf_t
Auxiliary type for the EnableIf class template.
Definition: EnableIf.h:175
#define BLAZE_FUNCTION_TRACE
Function trace macro.
Definition: FunctionTrace.h:94
Header file for the exception macros of the math module.
Header file for all forward declarations for expression class templates.
Header file for the serial shim.
Base class for all compute expression templates.
Definition: Computation.h:68
Base class for all matrix/matrix addition expression templates.
Definition: MatMatAddExpr.h:68
Header file for the IsZero type trait.
Header file for basic type definitions.