Commit cfd0bb88 authored by Tomáš Oberhuber's avatar Tomáš Oberhuber
Browse files

Added check of correct lambda functions definition in lambda matrix factory.

parent be26a735

src/TNL/Matrices/LambdaMatrix.h

+20 −0
Original line number Diff line number Diff line
@@ -10,6 +10,7 @@ 


#pragma once



#include <type_traits>

#include <TNL/String.h>

#include <TNL/Devices/Host.h>
@@ -377,6 +378,15 @@ struct LambdaMatrixFactory 
                       CompressedRowLengthsLambda& compressedRowLengthsLambda )

   -> LambdaMatrix< MatrixElementsLambda, CompressedRowLengthsLambda, Real, Device, Index >

   {

      // TODO: fix the following asserts, they do not work in fact

      static_assert( std::is_same<

            std::enable_if_t< true, decltype(matrixElementsLambda( Index(), Index(), Index(), Index(), std::declval< Index& >(), std::declval< Real& >() ) ) >,

            void >::value,

         "Wong type of MatrixElementsLambda, it should be - matrixElementsLambda( Index rows, Index columns, Index rowIdx, Index localIdx, Index& columnIdx, Real& value )" );

      static_assert( std::is_integral<

         std::enable_if_t< true, decltype(compressedRowLengthsLambda( Index(), Index(), Index() ) ) >

          >::value ,

         "Wong type of CompressedRowLengthsLambda, it should be - matrixElementsLambda( Index rows, Index columns, Index rowIdx )" );

      return LambdaMatrix< MatrixElementsLambda, CompressedRowLengthsLambda, Real, Device, Index >(

         matrixElementsLambda,

         compressedRowLengthsLambda );
@@ -405,6 +415,16 @@ struct LambdaMatrixFactory 
                       CompressedRowLengthsLambda& compressedRowLengthsLambda )

   -> LambdaMatrix< MatrixElementsLambda, CompressedRowLengthsLambda, Real, Device, Index >

   {

      // TODO: fix the following asserts, they do not work in fact

      static_assert( std::is_same<

            std::enable_if_t< true, decltype(matrixElementsLambda( Index(), Index(), Index(), Index(), std::declval< Index& >(), std::declval< Real& >() ) ) >,

            void >::value,

         "Wong type of MatrixElementsLambda, it should be - matrixElementsLambda( Index rows, Index columns, Index rowIdx, Index localIdx, Index& columnIdx, Real& value )" );

      static_assert( std::is_integral<

         std::enable_if_t< true, decltype(compressedRowLengthsLambda( Index(), Index(), Index() ) ) >

          >::value ,

         "Wong type of CompressedRowLengthsLambda, it should be - matrixElementsLambda( Index rows, Index columns, Index rowIdx )" );



      return LambdaMatrix< MatrixElementsLambda, CompressedRowLengthsLambda, Real, Device, Index >(

         rows, columns,

         matrixElementsLambda,