io.hh

Go to the documentation of this file.

// SPDX-FileCopyrightInfo: Copyright (C) DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
#ifndef DUNE_COMMON_SIMD_IO_HH
#define DUNE_COMMON_SIMD_IO_HH
 
#include <ios>
#include <type_traits>
 
#include <dune/common/rangeutilities.hh>
#include <dune/common/simd/simd.hh>
#include <dune/common/typetraits.hh>
 
namespace Dune {
 
  namespace SimdImpl {
 
    template<class T>
    class Inserter {
      T value_;
 
    public:
      Inserter(const T &value) : value_(value) {}
 
      template<class Stream,
               class = std::enable_if_t<std::is_base_of<std::ios_base,
                                                        Stream>::value> >
      friend Stream& operator<<(Stream &out, const Inserter &ins)
      {
        const char *sep = "<";
        for(auto l : range(Simd::lanes(ins.value_)))
        {
          out << sep << autoCopy(Simd::lane(l, ins.value_));
          sep = ", ";
        }
        out << '>';
        return out;
      }
    };
 
    template<class V, class = std::enable_if_t<Simd::lanes<V>() != 1> >
    Inserter<V> io(const V &v)
    {
      return { v };
    }
 
    template<class V, class = std::enable_if_t<Simd::lanes<V>() == 1> >
    Simd::Scalar<V> io(const V &v)
    {
      return Simd::lane(0, v);
    }
 
  }
 
  namespace Simd {
 
 
    template<class V>
    auto vio(const V &v)
    {
      return SimdImpl::Inserter<V>{ v };
    }
 
 
    template<class V>
    auto io(const V &v)
    {
      return SimdImpl::io(v);
    }
 
 
 
  } // namespace Simd
} // namespace Dune
 
#endif // DUNE_COMMON_SIMD_IO_HH