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>
54struct RebindDomain<
DiscreteDomain<DDims...>, detail::TypeSeq<ODDims...>>
61template <
class... DDims>
68 using discrete_element_type = DiscreteElement<DDims...>;
73 DiscreteElement<DDims...> m_element_begin;
75 DiscreteElement<DDims...> m_element_end;
80 return sizeof...(DDims);
86 template <
class... DDoms,
class = std::enable_if_t<(is_discrete_domain_v<DDoms> && ...)>>
88 : m_element_begin(domains.front()...)
89 , m_element_end((domains.front() + domains.extents())...)
94
95
96
98 discrete_element_type
const& element_begin,
99 discrete_vector_type
const& size)
100 : m_element_begin(element_begin)
101 , m_element_end(element_begin + size)
115 template <
class... ODims>
118 if (
empty() && other.empty()) {
121 return m_element_begin == other.m_element_begin && m_element_end == other.m_element_end;
124#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
126 template <
class...
ODims>
129 return !(*
this ==
other);
135 return (1UL * ... * extent<DDims>().value());
140 return m_element_end - m_element_begin;
143 template <
class QueryDDim>
146 return DiscreteElement<QueryDDim>(m_element_end)
147 - DiscreteElement<QueryDDim>(m_element_begin);
152 return m_element_begin;
157 return discrete_element_type((DiscreteElement<DDims>(m_element_end) - 1)...);
181 discrete_vector_type n1,
182 discrete_vector_type n2)
const
190 return m_element_begin + dvect;
193 template <
class... ODDims>
196 assert(((DiscreteElement<ODDims>(m_element_begin)
197 <= DiscreteElement<ODDims>(odomain.m_element_begin))
199 assert(((DiscreteElement<ODDims>(m_element_end)
200 >= DiscreteElement<ODDims>(odomain.m_element_end))
205 DiscreteElement<DDims...>((select_or<DDims>(
206 odomain.m_element_begin,
207 DiscreteElement<DDims>(m_element_begin)))...),
209 (select_or<DDims>(oextents,
DiscreteVector<DDims>(myextents)))...));
212 template <
class... DElems>
216 sizeof...(DDims) == (0 + ... + DElems::size()),
217 "Invalid number of dimensions");
218 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
219 return (((DiscreteElement<DDims>(take<DDims>(delems...))
220 >= DiscreteElement<DDims>(m_element_begin))
222 && ((DiscreteElement<DDims>(take<DDims>(delems...))
223 < DiscreteElement<DDims>(m_element_end))
227 template <
class... DElems>
229 DElems
const&... delems)
const noexcept
232 sizeof...(DDims) == (0 + ... + DElems::size()),
233 "Invalid number of dimensions");
234 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
236 (DiscreteElement<DDims>(take<DDims>(delems...))
237 - DiscreteElement<DDims>(m_element_begin))...);
251 std::size_t N =
sizeof...(DDims),
252 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
259 std::size_t N =
sizeof...(DDims),
260 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
267 std::size_t N =
sizeof...(DDims),
268 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
275 std::size_t N =
sizeof...(DDims),
276 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
283 std::size_t N =
sizeof...(DDims),
284 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
291 std::size_t N =
sizeof...(DDims),
292 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
306 using discrete_element_type = DiscreteElement<>;
318 template <
class... DDoms,
class = std::enable_if_t<(is_discrete_domain_v<DDoms> && ...)>>
324
325
326
328 [[
maybe_unused]] discrete_element_type
const& element_begin,
348#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
352 return !(*
this ==
other);
413 template <
class... ODims>
451template <
class... QueryDDims,
class... DDims>
456 DiscreteElement<QueryDDims...>(domain.front()),
463struct ConvertTypeSeqToDiscreteDomain;
465template <
class... DDims>
466struct ConvertTypeSeqToDiscreteDomain<detail::TypeSeq<DDims...>>
472using convert_type_seq_to_discrete_domain_t =
typename ConvertTypeSeqToDiscreteDomain<T>::type;
478template <
typename... DDoms>
479struct cartesian_prod;
481template <
typename... DDims1,
typename... DDims2,
typename... DDomsTail>
484 using type =
typename cartesian_prod<
DiscreteDomain<DDims1..., DDims2...>, DDomsTail...>::type;
487template <
typename... DDims>
494struct cartesian_prod<>
499template <
typename... DDoms>
500using cartesian_prod_t =
typename cartesian_prod<DDoms...>::type;
503template <
class... DDimsA,
class... DDimsB>
508 using TagSeqA = detail::TypeSeq<DDimsA...>;
509 using TagSeqB = detail::TypeSeq<DDimsB...>;
511 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
512 return detail::convert_type_seq_to_discrete_domain_t<type_seq_r>(DDom_a);
518template <
class... DDimsB,
class... DDimsA>
521 using TagSeqA = detail::TypeSeq<DDimsA...>;
522 using TagSeqB = detail::TypeSeq<DDimsB...>;
524 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
525 return detail::convert_type_seq_to_discrete_domain_t<type_seq_r>(DDom_a);
529template <
typename DDom,
typename... DDims>
530using remove_dims_of_t =
decltype(remove_dims_of<DDims...>(std::declval<DDom>()));
535template <
typename DDim1,
typename DDim2,
typename DDimA,
typename... DDimsB>
536KOKKOS_FUNCTION
constexpr std::conditional_t<
537 std::is_same_v<DDimA, DDim1>,
542 [[maybe_unused]]
DiscreteDomain<DDimsB...>
const& DDom_b)
noexcept
544 if constexpr (std::is_same_v<DDimA, DDim1>) {
554template <
typename DDim1,
typename DDim2,
typename... DDimsA,
typename... DDimsB>
560 using TagSeqA = detail::TypeSeq<DDimsA...>;
561 using TagSeqB = detail::TypeSeq<DDim1>;
562 using TagSeqC = detail::TypeSeq<DDim2>;
564 using type_seq_r =
ddc::type_seq_replace_t<TagSeqA, TagSeqB, TagSeqC>;
565 return ddc::detail::convert_type_seq_to_discrete_domain_t<type_seq_r>(
566 detail::replace_dim_of_1d<
574template <
typename DDom,
typename DDim1,
typename DDim2>
575using replace_dim_of_t =
decltype(replace_dim_of<DDim1, DDim2>(
576 std::declval<DDom>(),
577 std::declval<
typename detail::RebindDomain<DDom, detail::TypeSeq<DDim2>>::type>()));
580template <
class... QueryDDims,
class... DDims>
587template <
class... QueryDDims,
class... DDims>
591 return DiscreteElement<QueryDDims...>(
DiscreteDomain<QueryDDims>(domain).front()...);
594template <
class... QueryDDims,
class... DDims>
598 return DiscreteElement<QueryDDims...>(
DiscreteDomain<QueryDDims>(domain).back()...);
605 DiscreteElement<DDim> m_value = DiscreteElement<DDim>();
608 using iterator_category = std::random_access_iterator_tag;
610 using value_type = DiscreteElement<DDim>;
612 using difference_type = std::ptrdiff_t;
654 if (n >= difference_type(0)) {
655 m_value +=
static_cast<DiscreteElementType>(n);
657 m_value -=
static_cast<DiscreteElementType>(-n);
664 if (n >= difference_type(0)) {
665 m_value -=
static_cast<DiscreteElementType>(n);
667 m_value +=
static_cast<DiscreteElementType>(-n);
681 return xx.m_value == yy.m_value;
684#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
698 return xx.m_value < yy.m_value;
747 return (yy.m_value > xx.m_value) ? (-
static_cast<difference_type>(yy.m_value - xx.m_value))
748 : (xx.m_value - yy.m_value);
static KOKKOS_FUNCTION constexpr discrete_vector_type extents() noexcept
KOKKOS_DEFAULTED_FUNCTION ~DiscreteDomain()=default
KOKKOS_FUNCTION constexpr DiscreteDomain(discrete_element_type const &element_begin, discrete_vector_type const &size)
Construct a DiscreteDomain starting from element_begin with size points.
KOKKOS_FUNCTION constexpr DiscreteDomain remove(discrete_vector_type n1, discrete_vector_type n2) const
KOKKOS_FUNCTION constexpr DiscreteDomain take_last(discrete_vector_type n) const
static KOKKOS_FUNCTION constexpr std::size_t rank()
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain(DiscreteDomain const &x)=default
static KOKKOS_FUNCTION bool contains() noexcept
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteDomain()=default
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain & operator=(DiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr operator bool()
static KOKKOS_FUNCTION constexpr std::size_t size()
static KOKKOS_FUNCTION constexpr bool empty() noexcept
KOKKOS_FUNCTION constexpr DiscreteDomain remove_last(discrete_vector_type n) const
static KOKKOS_FUNCTION bool contains(DiscreteElement<>) noexcept
static KOKKOS_FUNCTION constexpr discrete_element_type back() noexcept
static KOKKOS_FUNCTION DiscreteVector distance_from_front(DiscreteElement<>) noexcept
static KOKKOS_FUNCTION constexpr discrete_element_type front() noexcept
KOKKOS_FUNCTION constexpr DiscreteDomain take_first(discrete_vector_type n) const
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain & operator=(DiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr DiscreteDomain(DDoms const &... domains)
Construct a DiscreteDomain by copies and merge of domains.
KOKKOS_FUNCTION constexpr bool operator==(DiscreteDomain const &other) const
KOKKOS_FUNCTION constexpr DiscreteDomain remove_first(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr DiscreteDomain restrict_with(DiscreteDomain< ODims... > const &) const
KOKKOS_FUNCTION constexpr DiscreteElement operator()(DiscreteVector<> const &) const noexcept
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain(DiscreteDomain &&x)=default
static KOKKOS_FUNCTION DiscreteVector distance_from_front() noexcept
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain(DiscreteDomain const &x)=default
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain(DiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr DiscreteDomain remove_last(discrete_vector_type n) const
KOKKOS_FUNCTION bool contains(DElems const &... delems) const noexcept
KOKKOS_FUNCTION constexpr DiscreteDomain take_last(discrete_vector_type n) const
KOKKOS_FUNCTION auto begin() const
KOKKOS_FUNCTION constexpr DiscreteDomain remove(discrete_vector_type n1, discrete_vector_type n2) const
KOKKOS_FUNCTION constexpr auto restrict_with(DiscreteDomain< ODDims... > const &odomain) const
static KOKKOS_FUNCTION constexpr std::size_t rank()
KOKKOS_FUNCTION constexpr bool empty() const noexcept
KOKKOS_FUNCTION constexpr std::size_t size() const
friend class DiscreteDomain
KOKKOS_FUNCTION constexpr discrete_vector_type extents() const noexcept
KOKKOS_FUNCTION constexpr DiscreteVector< QueryDDim > extent() const noexcept
KOKKOS_FUNCTION constexpr discrete_element_type front() const noexcept
KOKKOS_FUNCTION constexpr DiscreteDomain(DDoms const &... domains)
Construct a DiscreteDomain by copies and merge of domains.
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain & operator=(DiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr DiscreteDomain remove_first(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr bool operator==(DiscreteDomain< ODims... > const &other) const
KOKKOS_FUNCTION auto cbegin() const
KOKKOS_FUNCTION auto end() const
KOKKOS_FUNCTION constexpr DiscreteDomain take_first(discrete_vector_type n) const
KOKKOS_DEFAULTED_FUNCTION ~DiscreteDomain()=default
KOKKOS_FUNCTION auto cend() const
KOKKOS_FUNCTION constexpr decltype(auto) operator[](std::size_t n) const
KOKKOS_FUNCTION DiscreteVector< DDims... > distance_from_front(DElems const &... delems) const noexcept
KOKKOS_FUNCTION constexpr discrete_element_type back() const noexcept
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain & operator=(DiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr DiscreteElement< DDims... > operator()(DiscreteVector< DDims... > const &dvect) const noexcept
KOKKOS_DEFAULTED_FUNCTION DiscreteDomain()=default
KOKKOS_FUNCTION constexpr operator bool()
KOKKOS_FUNCTION constexpr DiscreteDomain(discrete_element_type const &element_begin, discrete_vector_type const &size)
Construct a DiscreteDomain starting from element_begin with size points.
KOKKOS_FUNCTION constexpr decltype(auto) operator[](std::size_t n)
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteElement()=default
KOKKOS_FUNCTION constexpr bool operator!=(DiscreteVector< OTags... > const &rhs) const noexcept
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteVector()=default
The top-level namespace of DDC.
KOKKOS_FUNCTION constexpr DiscreteElement< QueryDDims... > back(DiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr auto remove_dims_of(DiscreteDomain< DDimsA... > const &DDom_a) noexcept
Remove the dimensions DDimsB from DDom_a.
KOKKOS_FUNCTION constexpr DiscreteElement< QueryDDims... > front(DiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr DiscreteVector< QueryDDims... > extents(DiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr auto remove_dims_of(DiscreteDomain< DDimsA... > const &DDom_a, DiscreteDomain< DDimsB... > const &DDom_b) noexcept
KOKKOS_FUNCTION constexpr auto replace_dim_of(DiscreteDomain< DDimsA... > const &DDom_a, DiscreteDomain< DDimsB... > const &DDom_b) noexcept
constexpr bool is_discrete_domain_v
KOKKOS_FUNCTION constexpr DiscreteDomain< QueryDDims... > select(DiscreteDomain< DDims... > const &domain)
KOKKOS_FUNCTION constexpr DiscreteDomainIterator operator--(int)
friend KOKKOS_FUNCTION constexpr bool operator<=(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)
KOKKOS_FUNCTION constexpr DiscreteElement< DDim > operator*() const noexcept
friend KOKKOS_FUNCTION constexpr bool operator>=(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)
KOKKOS_DEFAULTED_FUNCTION DiscreteDomainIterator()=default
KOKKOS_FUNCTION constexpr DiscreteElement< DDim > operator[](difference_type n) const
KOKKOS_FUNCTION constexpr DiscreteDomainIterator & operator--()
KOKKOS_FUNCTION constexpr DiscreteDomainIterator operator++(int)
friend KOKKOS_FUNCTION constexpr DiscreteDomainIterator operator-(DiscreteDomainIterator i, difference_type n)
friend KOKKOS_FUNCTION constexpr DiscreteDomainIterator operator+(DiscreteDomainIterator i, difference_type n)
friend KOKKOS_FUNCTION constexpr bool operator>(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)
friend KOKKOS_FUNCTION constexpr bool operator!=(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)
friend KOKKOS_FUNCTION constexpr bool operator<(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)
KOKKOS_FUNCTION constexpr DiscreteDomainIterator & operator+=(difference_type n)
KOKKOS_FUNCTION constexpr DiscreteDomainIterator & operator++()
friend KOKKOS_FUNCTION constexpr DiscreteDomainIterator operator+(difference_type n, DiscreteDomainIterator i)
friend KOKKOS_FUNCTION constexpr bool operator==(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)
KOKKOS_FUNCTION constexpr DiscreteDomainIterator(DiscreteElement< DDim > value)
KOKKOS_FUNCTION constexpr DiscreteDomainIterator & operator-=(difference_type n)
friend KOKKOS_FUNCTION constexpr difference_type operator-(DiscreteDomainIterator const &xx, DiscreteDomainIterator const &yy)