io.hh

Go to the documentation of this file.

// SPDX-FileCopyrightInfo: Copyright © 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