NDRegion.h

Go to the documentation of this file.

//
// Class NDRegion
//   NDRegion is a simple container of N PRegion objects. It is templated
//   on the type of data (T) and the number of PRegions (Dim).
//
#ifndef IPPL_NDREGION_H
#define IPPL_NDREGION_H
 
#include <type_traits>
 
#include "Region/PRegion.h"
 
namespace ippl {
    template <typename T, unsigned Dim>
    class NDRegion {
    public:
        KOKKOS_FUNCTION
        NDRegion() {}
 
        KOKKOS_FUNCTION
        ~NDRegion() {}
 
        template <class... Args,
                  typename std::enable_if<sizeof...(Args) == Dim
                                              && (std::is_convertible_v<Args, PRegion<T>> && ...),
                                          bool>::type = true>
        KOKKOS_FUNCTION NDRegion(const Args&... args);
 
        KOKKOS_INLINE_FUNCTION NDRegion(const NDRegion<T, Dim>& nr);
 
        KOKKOS_INLINE_FUNCTION NDRegion<T, Dim>& operator=(const NDRegion<T, Dim>& nr);
 
        KOKKOS_INLINE_FUNCTION const PRegion<T>& operator[](unsigned d) const;
 
        KOKKOS_INLINE_FUNCTION PRegion<T>& operator[](unsigned d);
 
        KOKKOS_INLINE_FUNCTION NDRegion<T, Dim>& operator+=(const T t);
 
        KOKKOS_INLINE_FUNCTION NDRegion<T, Dim>& operator-=(const T t);
 
        KOKKOS_INLINE_FUNCTION NDRegion<T, Dim>& operator*=(const T t);
 
        KOKKOS_INLINE_FUNCTION NDRegion<T, Dim>& operator/=(const T t);
 
        KOKKOS_INLINE_FUNCTION bool empty() const;
 
    private:
        PRegion<T> regions_m[Dim];
    };
}  // namespace ippl
 
#include "Region/NDRegion.hpp"
 
#endif