Added operator<< and setValue for SparseMatrixRowView

src/TNL/Matrices/SparseMatrixRowView.h

@@ -10,8 +10,12 @@
 
 #pragma once
 
+#include <ostream>
+
+#include <TNL/Cuda/CudaCallable.h>
+
 namespace TNL {
-   namespace Matrices {
+namespace Matrices {
 
 template< typename SegmentView,
           typename ValuesView,
@@ -52,6 +56,10 @@ class SparseMatrixRowView
       __cuda_callable__
       RealType& getValue( const IndexType localIdx );
 
+      __cuda_callable__
+      void setValue( const IndexType localIdx,
+                     const RealType& value );
+
       __cuda_callable__
       void setElement( const IndexType localIdx,
                        const IndexType column,
@@ -64,7 +72,14 @@ class SparseMatrixRowView
 
       ColumnsIndexesViewType columnIndexes;
 };
-   } // namespace Matrices
+
+template< typename SegmentView,
+          typename ValuesView,
+          typename ColumnsIndexesView,
+          bool isBinary_ >
+std::ostream& operator<<( std::ostream& str, const SparseMatrixRowView< SegmentView, ValuesView, ColumnsIndexesView, isBinary_ >& row );
+
+} // namespace Matrices
 } // namespace TNL
 
 #include <TNL/Matrices/SparseMatrixRowView.hpp>

src/TNL/Matrices/SparseMatrixRowView.hpp

@@ -11,9 +11,10 @@
 #pragma once
 
 #include <TNL/Matrices/SparseMatrixRowView.h>
+#include <TNL/Assert.h>
 
 namespace TNL {
-   namespace Matrices {
+namespace Matrices {
 
 template< typename SegmentView,
           typename ValuesView,
@@ -89,6 +90,22 @@ getValue( const IndexType localIdx ) -> RealType&
    return values[ segmentView.getGlobalIndex( localIdx ) ];
 }
 
+template< typename SegmentView,
+          typename ValuesView,
+          typename ColumnsIndexesView,
+          bool isBinary_ >
+__cuda_callable__ void
+SparseMatrixRowView< SegmentView, ValuesView, ColumnsIndexesView, isBinary_ >::
+setValue( const IndexType localIdx,
+          const RealType& value )
+{
+   TNL_ASSERT_LT( localIdx, this->getSize(), "Local index exceeds matrix row capacity." );
+   if( ! isBinary() ) {
+      const IndexType globalIdx = segmentView.getGlobalIndex( localIdx );
+      values[ globalIdx ] = value;
+   }
+}
+
 template< typename SegmentView,
           typename ValuesView,
           typename ColumnsIndexesView,
@@ -106,6 +123,17 @@ setElement( const IndexType localIdx,
       values[ globalIdx ] = value;
 }
 
+template< typename SegmentView,
+          typename ValuesView,
+          typename ColumnsIndexesView,
+          bool isBinary_ >
+std::ostream& operator<<( std::ostream& str, const SparseMatrixRowView< SegmentView, ValuesView, ColumnsIndexesView, isBinary_ >& row )
+{
+   using NonConstIndex = std::remove_const_t< typename SparseMatrixRowView< SegmentView, ValuesView, ColumnsIndexesView, isBinary_ >::IndexType >;
+   for( NonConstIndex i = 0; i < row.getSize(); i++ )
+      str << " [ " << row.getColumnIndex( i ) << " ] = " << row.getValue( i ) << ", ";
+   return str;
+}
 
-   } // namespace Matrices
+} // namespace Matrices
 } // namespace TNL