13#include <Kokkos_Macros.hpp>
15#include "detail/type_seq.hpp"
17#include "discrete_element.hpp"
18#include "discrete_vector.hpp"
25template <
class... DDims>
33template <
class... Tags>
44template <
class... Tags>
47 using type = TypeSeq<Tags...>;
50template <
class T,
class U>
53template <
class... DDims,
class... ODDims>
61template <
class... ODDims>
66 return DiscreteVector<ODDims...>((get<ODDims>(lhs) * get<ODDims>(rhs))...);
69template <
class... DDims>
76 using discrete_element_type = DiscreteElement<DDims...>;
81 DiscreteElement<DDims...> m_element_begin;
90 return sizeof...(DDims);
98 class = std::enable_if_t<(is_strided_discrete_domain_v<DDoms> && ...)>>
100 : m_element_begin(domains.front()...)
101 , m_extents(domains.extents()...)
102 , m_strides(domains.strides()...)
107
108
109
110
112 discrete_element_type
const& element_begin,
113 discrete_vector_type
const& extents,
114 discrete_vector_type
const& strides)
115 : m_element_begin(element_begin)
132 template <
class... ODims>
135 if (
empty() && other.empty()) {
138 return m_element_begin == other.m_element_begin && m_extents == other.m_extents
139 && m_strides == other.m_strides;
142#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
144 template <
class...
ODims>
147 return !(*
this ==
other);
153 return (1UL * ... * get<DDims>(m_extents));
166 template <
class QueryDDim>
174 return m_element_begin;
179 return discrete_element_type(
180 (DiscreteElement<DDims>(m_element_begin)
181 + (get<DDims>(m_extents) - 1) * get<DDims>(m_strides))...);
205 discrete_vector_type n1,
206 discrete_vector_type n2)
const
214 return m_element_begin + prod(dvect, m_strides);
217 template <
class... DElems>
221 sizeof...(DDims) == (0 + ... + DElems::size()),
222 "Invalid number of dimensions");
223 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
225 = ((DiscreteElement<DDims>(take<DDims>(delems...))
226 >= DiscreteElement<DDims>(m_element_begin))
229 = ((DiscreteElement<DDims>(take<DDims>(delems...))
230 < (DiscreteElement<DDims>(m_element_begin)
234 = ((((DiscreteElement<DDims>(take<DDims>(delems...))
235 - DiscreteElement<DDims>(m_element_begin))
239 return test1 && test2 && test3;
242 template <
class... DElems>
244 DElems
const&... delems)
const noexcept
247 sizeof...(DDims) == (0 + ... + DElems::size()),
248 "Invalid number of dimensions");
249 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
250 assert(contains(delems...));
252 ((DiscreteElement<DDims>(take<DDims>(delems...))
253 - DiscreteElement<DDims>(m_element_begin))
268 std::size_t N =
sizeof...(DDims),
269 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
276 std::size_t N =
sizeof...(DDims),
277 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
281 DDim0>(m_element_begin + m_extents * m_strides, m_strides);
285 std::size_t N =
sizeof...(DDims),
286 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
293 std::size_t N =
sizeof...(DDims),
294 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
298 DDim0>(m_element_begin + m_extents * m_strides, m_strides);
302 std::size_t N =
sizeof...(DDims),
303 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
310 std::size_t N =
sizeof...(DDims),
311 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
325 using discrete_element_type = DiscreteElement<>;
337 template <
class... ODDims>
344
345
346
347
349 [[
maybe_unused]] discrete_element_type
const& element_begin,
372#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
376 return !(*
this ==
other);
468template <
class... QueryDDims,
class... DDims>
473 DiscreteElement<QueryDDims...>(domain.front()),
481struct ConvertTypeSeqToStridedDiscreteDomain;
483template <
class... DDims>
484struct ConvertTypeSeqToStridedDiscreteDomain<detail::TypeSeq<DDims...>>
490using convert_type_seq_to_strided_discrete_domain_t =
491 typename ConvertTypeSeqToStridedDiscreteDomain<T>::type;
496template <
class... DDimsA,
class... DDimsB>
501 using TagSeqA = detail::TypeSeq<DDimsA...>;
502 using TagSeqB = detail::TypeSeq<DDimsB...>;
504 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
505 return detail::convert_type_seq_to_strided_discrete_domain_t<type_seq_r>(DDom_a);
511template <
class... DDimsB,
class... DDimsA>
515 using TagSeqA = detail::TypeSeq<DDimsA...>;
516 using TagSeqB = detail::TypeSeq<DDimsB...>;
518 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
519 return detail::convert_type_seq_to_strided_discrete_domain_t<type_seq_r>(DDom_a);
525template <
typename DDim1,
typename DDim2,
typename DDimA,
typename... DDimsB>
526KOKKOS_FUNCTION
constexpr std::conditional_t<
527 std::is_same_v<DDimA, DDim1>,
534 if constexpr (std::is_same_v<DDimA, DDim1>) {
544template <
typename DDim1,
typename DDim2,
typename... DDimsA,
typename... DDimsB>
550 using TagSeqA = detail::TypeSeq<DDimsA...>;
551 using TagSeqB = detail::TypeSeq<DDim1>;
552 using TagSeqC = detail::TypeSeq<DDim2>;
554 using type_seq_r =
ddc::type_seq_replace_t<TagSeqA, TagSeqB, TagSeqC>;
555 return ddc::detail::convert_type_seq_to_strided_discrete_domain_t<type_seq_r>(
556 detail::replace_dim_of_1d<
563template <
class... QueryDDims,
class... DDims>
570template <
class... QueryDDims,
class... DDims>
577template <
class... QueryDDims,
class... DDims>
588 DiscreteElement<DDim> m_value = DiscreteElement<DDim>();
593 using iterator_category = std::random_access_iterator_tag;
595 using value_type = DiscreteElement<DDim>;
597 using difference_type = std::ptrdiff_t;
602 DiscreteElement<DDim> value,
642 if (n >= difference_type(0)) {
643 m_value +=
static_cast<DiscreteElementType>(n) * m_stride;
645 m_value -=
static_cast<DiscreteElementType>(-n) * m_stride;
652 if (n >= difference_type(0)) {
653 m_value -=
static_cast<DiscreteElementType>(n) * m_stride;
655 m_value +=
static_cast<DiscreteElementType>(-n) * m_stride;
662 return m_value + n * m_stride;
669 return xx.m_value == yy.m_value;
672#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
686 return xx.m_value < yy.m_value;
735 return (yy.m_value > xx.m_value) ? (-
static_cast<difference_type>(yy.m_value - xx.m_value))
736 : (xx.m_value - yy.m_value);
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteElement()=default
KOKKOS_FUNCTION constexpr bool operator!=(DiscreteVector< OTags... > const &rhs) const noexcept
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteVector()=default
static KOKKOS_FUNCTION DiscreteVector distance_from_front() noexcept
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain & operator=(StridedDiscreteDomain &&x)=default
static KOKKOS_FUNCTION constexpr std::size_t size()
KOKKOS_FUNCTION constexpr StridedDiscreteDomain(discrete_element_type const &element_begin, discrete_vector_type const &size, discrete_vector_type const &strides)
Construct a StridedDiscreteDomain starting from element_begin with size points.
static KOKKOS_FUNCTION bool contains(DiscreteElement<>) noexcept
KOKKOS_FUNCTION constexpr StridedDiscreteDomain remove_first(discrete_vector_type n) const
static KOKKOS_FUNCTION constexpr discrete_element_type front() noexcept
KOKKOS_FUNCTION constexpr StridedDiscreteDomain remove(discrete_vector_type n1, discrete_vector_type n2) const
static KOKKOS_FUNCTION bool contains() noexcept
KOKKOS_FUNCTION constexpr StridedDiscreteDomain remove_last(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain take_last(discrete_vector_type n) const
static KOKKOS_FUNCTION constexpr discrete_vector_type extents() noexcept
static KOKKOS_FUNCTION constexpr std::size_t rank()
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain(StridedDiscreteDomain &&x)=default
static KOKKOS_FUNCTION DiscreteVector distance_from_front(DiscreteElement<>) noexcept
KOKKOS_DEFAULTED_FUNCTION constexpr StridedDiscreteDomain()=default
KOKKOS_FUNCTION constexpr operator bool()
static KOKKOS_FUNCTION constexpr bool empty() noexcept
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain(StridedDiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr DiscreteElement operator()(DiscreteVector<> const &) const noexcept
static KOKKOS_FUNCTION constexpr discrete_element_type back() noexcept
KOKKOS_FUNCTION constexpr bool operator==(StridedDiscreteDomain const &other) const
KOKKOS_DEFAULTED_FUNCTION ~StridedDiscreteDomain()=default
KOKKOS_FUNCTION constexpr StridedDiscreteDomain take_first(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain(StridedDiscreteDomain< ODDims... > const &domain)
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain & operator=(StridedDiscreteDomain const &x)=default
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain & operator=(StridedDiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr discrete_vector_type extents() const noexcept
KOKKOS_FUNCTION constexpr operator bool()
KOKKOS_FUNCTION constexpr StridedDiscreteDomain take_first(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr DiscreteVector< QueryDDim > extent() const noexcept
KOKKOS_FUNCTION constexpr discrete_element_type front() const noexcept
KOKKOS_FUNCTION bool contains(DElems const &... delems) const noexcept
KOKKOS_FUNCTION constexpr std::size_t size() const
KOKKOS_FUNCTION auto cend() const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain(DDoms const &... domains)
Construct a StridedDiscreteDomain by copies and merge of domains.
KOKKOS_FUNCTION DiscreteVector< DDims... > distance_from_front(DElems const &... delems) const noexcept
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain(StridedDiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr discrete_vector_type strides() const noexcept
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain & operator=(StridedDiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr StridedDiscreteDomain remove(discrete_vector_type n1, discrete_vector_type n2) const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain remove_first(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr discrete_element_type back() const noexcept
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain()=default
KOKKOS_FUNCTION constexpr DiscreteElement< DDims... > operator()(DiscreteVector< DDims... > const &dvect) const noexcept
friend class StridedDiscreteDomain
KOKKOS_FUNCTION auto begin() const
KOKKOS_FUNCTION auto end() const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain remove_last(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain take_last(discrete_vector_type n) const
KOKKOS_DEFAULTED_FUNCTION ~StridedDiscreteDomain()=default
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomain(StridedDiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr bool operator==(StridedDiscreteDomain< ODims... > const &other) const
KOKKOS_FUNCTION constexpr decltype(auto) operator[](std::size_t n)
static KOKKOS_FUNCTION constexpr std::size_t rank()
KOKKOS_FUNCTION auto cbegin() const
KOKKOS_FUNCTION constexpr bool empty() const noexcept
KOKKOS_FUNCTION constexpr decltype(auto) operator[](std::size_t n) const
KOKKOS_FUNCTION constexpr StridedDiscreteDomain(discrete_element_type const &element_begin, discrete_vector_type const &extents, discrete_vector_type const &strides)
Construct a StridedDiscreteDomain starting from element_begin with size points.
The top-level namespace of DDC.
KOKKOS_FUNCTION constexpr auto remove_dims_of(StridedDiscreteDomain< DDimsA... > const &DDom_a, StridedDiscreteDomain< DDimsB... > const &DDom_b) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< QueryDDims... > front(StridedDiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr DiscreteVector< QueryDDims... > extents(StridedDiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< QueryDDims... > back(StridedDiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr StridedDiscreteDomain< QueryDDims... > select(StridedDiscreteDomain< DDims... > const &domain)
KOKKOS_FUNCTION constexpr auto remove_dims_of(StridedDiscreteDomain< DDimsA... > const &DDom_a) noexcept
Remove the dimensions DDimsB from DDom_a.
constexpr bool is_strided_discrete_domain_v
KOKKOS_FUNCTION DiscreteVector< ODDims... > prod(DiscreteVector< ODDims... > const &lhs, DiscreteVector< ODDims... > const &rhs) noexcept
KOKKOS_FUNCTION constexpr auto replace_dim_of(StridedDiscreteDomain< DDimsA... > const &DDom_a, StridedDiscreteDomain< DDimsB... > const &DDom_b) noexcept
friend KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator operator-(StridedDiscreteDomainIterator i, difference_type n)
KOKKOS_FUNCTION constexpr DiscreteElement< DDim > operator*() const noexcept
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator & operator++()
friend KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator operator+(difference_type n, StridedDiscreteDomainIterator i)
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator & operator-=(difference_type n)
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator operator++(int)
friend KOKKOS_FUNCTION constexpr bool operator<(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)
KOKKOS_DEFAULTED_FUNCTION StridedDiscreteDomainIterator()=default
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator operator--(int)
friend KOKKOS_FUNCTION constexpr bool operator==(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator(DiscreteElement< DDim > value, DiscreteVector< DDim > stride)
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator & operator+=(difference_type n)
KOKKOS_FUNCTION constexpr DiscreteElement< DDim > operator[](difference_type n) const
friend KOKKOS_FUNCTION constexpr difference_type operator-(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)
KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator & operator--()
friend KOKKOS_FUNCTION constexpr bool operator>=(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)
friend KOKKOS_FUNCTION constexpr StridedDiscreteDomainIterator operator+(StridedDiscreteDomainIterator i, difference_type n)
friend KOKKOS_FUNCTION constexpr bool operator>(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)
friend KOKKOS_FUNCTION constexpr bool operator!=(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)
friend KOKKOS_FUNCTION constexpr bool operator<=(StridedDiscreteDomainIterator const &xx, StridedDiscreteDomainIterator const &yy)