Skip to content
Snippets Groups Projects
Select Git revision
  • 2564c14ee6e1800bf97e461ff13c8f4a6e78729b
  • develop default protected
  • feature/gmsh-reader
  • origin/stage/bouguettaia
  • feature/kinetic-schemes
  • feature/reconstruction
  • feature/local-dt-fsi
  • feature/composite-scheme-sources
  • feature/composite-scheme-other-fluxes
  • feature/serraille
  • feature/variational-hydro
  • feature/composite-scheme
  • hyperplastic
  • feature/polynomials
  • feature/gks
  • feature/implicit-solver-o2
  • feature/coupling_module
  • feature/implicit-solver
  • feature/merge-local-dt-fsi
  • master protected
  • feature/escobar-smoother
  • v0.5.0 protected
  • v0.4.1 protected
  • v0.4.0 protected
  • v0.3.0 protected
  • v0.2.0 protected
  • v0.1.0 protected
  • Kidder
  • v0.0.4 protected
  • v0.0.3 protected
  • v0.0.2 protected
  • v0 protected
  • v0.0.1 protected
33 results

Array.hpp

Blame
  • Array.hpp 1.87 KiB
    #ifndef ARRAY_HPP
    #define ARRAY_HPP
    
    #include <Kokkos_Core.hpp>
    
    template <typename DataType>
    class Array
    {
     private:
      Kokkos::View<DataType*> m_values;
    
      // Allows const version to access our data
      friend Array<std::add_const_t<DataType>>;
    
     public:
      KOKKOS_INLINE_FUNCTION
      size_t size() const
      {
        return m_values.extent(0);
      }
    
      KOKKOS_INLINE_FUNCTION
      DataType& operator[](const size_t& i)
      {
        return m_values[i];
      }
    
      KOKKOS_INLINE_FUNCTION
      DataType& operator[](const size_t& i) const
      {
        return m_values[i];
      }
    
      KOKKOS_INLINE_FUNCTION
      DataType& operator()(const size_t& i)
      {
        return m_values(i);
      }
    
      KOKKOS_INLINE_FUNCTION
      DataType& operator()(const size_t& i) const
      {
        return m_values(i);
      }
    
      KOKKOS_INLINE_FUNCTION
      Array(const size_t& size)
          : m_values("anonymous", size)
      {
        static_assert(not std::is_const<DataType>(),
                      "Cannot build Array of const data of given size");
      }
    
      template <typename DataType2>
      KOKKOS_INLINE_FUNCTION
      Array& operator=(const Array<DataType2>& array)
      {
        // ensures that DataType is the same as source DataType2
        static_assert(std::is_same<std::remove_const_t<DataType>, std::remove_const_t<DataType2>>(),
                      "Cannot assign Array of different type");
        // ensures that const is not lost through copy
        static_assert(((std::is_const<DataType2>() and std::is_const<DataType>())
                       or not std::is_const<DataType2>()),
                      "Cannot assign Array of const to  Array of non-const");
        m_values = array.m_values;
        return *this;
      }
    
      KOKKOS_INLINE_FUNCTION
      Array& operator=(const Array&) = default;
    
      KOKKOS_INLINE_FUNCTION
      Array& operator=(Array&&) = default;
    
      KOKKOS_INLINE_FUNCTION
      Array() = default;
    
      KOKKOS_INLINE_FUNCTION
      Array(const Array&) = default;
    
      KOKKOS_INLINE_FUNCTION
      ~Array() = default;
    };
    
    #endif // ARRAY_HPP