Line data    Source code

       1             : #pragma once
       2             : 
       3             : #include "functions/Abs.hpp"
       4             : 
       5             : #include <thrust/complex.h>
       6             : #include <thrust/count.h>
       7             : #include <thrust/limits.h>
       8             : 
       9             : namespace elsa
      10             : {
      11             :     namespace detail
      12             :     {
      13             :         struct nonZeroComp {
      14             :             template <class T>
      15             :             __host__ __device__ std::ptrdiff_t operator()(const T& arg) const noexcept
      16       27380 :             {
      17       27380 :                 return elsa::abs(arg) >= thrust::numeric_limits<T>::epsilon();
      18       27380 :             }
      19             : 
      20             :             template <class T>
      21             :             __host__ __device__ std::ptrdiff_t
      22             :                 operator()(const thrust::complex<T>& arg) const noexcept
      23       16916 :             {
      24       16916 :                 return elsa::abs(arg) >= thrust::numeric_limits<T>::epsilon();
      25       16916 :             }
      26             :         };
      27             :     } // namespace detail
      28             : 
      29             :     /// @brief compute the l0-"norm", which counts the number of non-zero elements.
      30             :     ///
      31             :     /// @ingroup reductions
      32             :     template <class InputIter>
      33             :     std::ptrdiff_t l0PseudoNorm(InputIter first, InputIter last)
      34          27 :     {
      35          27 :         return thrust::count_if(first, last, detail::nonZeroComp{});
      36          27 :     }
      37             : } // namespace elsa