12#include <Kokkos_Macros.hpp>
14#include "detail/type_seq.hpp"
16#include "discrete_element.hpp"
17#include "discrete_vector.hpp"
24template <
class... DDims>
32template <
class... Tags>
43template <
class... Tags>
46 using type = TypeSeq<Tags...>;
49template <
class T,
class U>
52template <
class... DDims,
class... ODDims>
53struct RebindDomain<
DiscreteDomain<DDims...>, detail::TypeSeq<ODDims...>>
60template <
class... DDims>
67 using discrete_element_type = DiscreteElement<DDims...>;
72 DiscreteElement<DDims...> m_element_begin;
74 DiscreteElement<DDims...> m_element_end;
79 return sizeof...(DDims);
85 template <
class... DDoms,
class = std::enable_if_t<(is_discrete_domain_v<DDoms> && ...)>>
87 : m_element_begin(domains.front()...)
88 , m_element_end((domains.front() + domains.extents())...)
93
94
95
97 discrete_element_type
const& element_begin,
98 discrete_vector_type
const& size)
99 : m_element_begin(element_begin)
100 , m_element_end(element_begin + size)
114 template <
class... ODims>
117 if (
empty() && other.empty()) {
120 return m_element_begin == other.m_element_begin && m_element_end == other.m_element_end;
123#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
125 template <
class...
ODims>
128 return !(*
this ==
other);
134 return (1UL * ... * extent<DDims>().value());
139 return m_element_end - m_element_begin;
142 template <
class QueryDDim>
145 return DiscreteElement<QueryDDim>(m_element_end)
146 - DiscreteElement<QueryDDim>(m_element_begin);
151 return m_element_begin;
156 return discrete_element_type((DiscreteElement<DDims>(m_element_end) - 1)...);
180 discrete_vector_type n1,
181 discrete_vector_type n2)
const
189 return m_element_begin + dvect;
192 template <
class... ODDims>
196 ((DiscreteElement<ODDims>(m_element_begin)
197 <= DiscreteElement<ODDims>(odomain.m_element_begin))
200 ((DiscreteElement<ODDims>(m_element_end)
201 >= DiscreteElement<ODDims>(odomain.m_element_end))
206 DiscreteElement<DDims...>((select_or<DDims>(
207 odomain.m_element_begin,
208 DiscreteElement<DDims>(m_element_begin)))...),
210 (select_or<DDims>(oextents,
DiscreteVector<DDims>(myextents)))...));
213 template <
class... DElems>
217 sizeof...(DDims) == (0 + ... + DElems::size()),
218 "Invalid number of dimensions");
219 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
220 return (((DiscreteElement<DDims>(take<DDims>(delems...))
221 >= DiscreteElement<DDims>(m_element_begin))
223 && ((DiscreteElement<DDims>(take<DDims>(delems...))
224 < DiscreteElement<DDims>(m_element_end))
228 template <
class... DElems>
230 DElems
const&... delems)
const noexcept
233 sizeof...(DDims) == (0 + ... + DElems::size()),
234 "Invalid number of dimensions");
235 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
237 (DiscreteElement<DDims>(take<DDims>(delems...))
238 - DiscreteElement<DDims>(m_element_begin))...);
252 std::size_t N =
sizeof...(DDims),
253 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
260 std::size_t N =
sizeof...(DDims),
261 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
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...>>>>
284 std::size_t N =
sizeof...(DDims),
285 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
292 std::size_t N =
sizeof...(DDims),
293 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
307 using discrete_element_type = DiscreteElement<>;
319 template <
class... DDoms,
class = std::enable_if_t<(is_discrete_domain_v<DDoms> && ...)>>
325
326
327
329 [[
maybe_unused]] discrete_element_type
const& element_begin,
349#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
353 return !(*
this ==
other);
414 template <
class... ODims>
452template <
class... QueryDDims,
class... DDims>
457 DiscreteElement<QueryDDims...>(domain.front()),
464struct ConvertTypeSeqToDiscreteDomain;
466template <
class... DDims>
467struct ConvertTypeSeqToDiscreteDomain<detail::TypeSeq<DDims...>>
473using convert_type_seq_to_discrete_domain_t =
typename ConvertTypeSeqToDiscreteDomain<T>::type;
479template <
typename... DDoms>
480struct cartesian_prod;
482template <
typename... DDims1,
typename... DDims2,
typename... DDomsTail>
485 using type =
typename cartesian_prod<
DiscreteDomain<DDims1..., DDims2...>, DDomsTail...>::type;
488template <
typename... DDims>
495struct cartesian_prod<>
500template <
typename... DDoms>
501using cartesian_prod_t =
typename cartesian_prod<DDoms...>::type;
504template <
class... DDimsA,
class... DDimsB>
509 using TagSeqA = detail::TypeSeq<DDimsA...>;
510 using TagSeqB = detail::TypeSeq<DDimsB...>;
512 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
513 return detail::convert_type_seq_to_discrete_domain_t<type_seq_r>(DDom_a);
519template <
class... DDimsB,
class... DDimsA>
522 using TagSeqA = detail::TypeSeq<DDimsA...>;
523 using TagSeqB = detail::TypeSeq<DDimsB...>;
525 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
526 return detail::convert_type_seq_to_discrete_domain_t<type_seq_r>(DDom_a);
530template <
typename DDom,
typename... DDims>
531using remove_dims_of_t =
decltype(remove_dims_of<DDims...>(std::declval<DDom>()));
536template <
typename DDim1,
typename DDim2,
typename DDimA,
typename... DDimsB>
537KOKKOS_FUNCTION
constexpr std::conditional_t<
538 std::is_same_v<DDimA, DDim1>,
543 [[maybe_unused]]
DiscreteDomain<DDimsB...>
const& DDom_b)
noexcept
545 if constexpr (std::is_same_v<DDimA, DDim1>) {
555template <
typename DDim1,
typename DDim2,
typename... DDimsA,
typename... DDimsB>
561 using TagSeqA = detail::TypeSeq<DDimsA...>;
562 using TagSeqB = detail::TypeSeq<DDim1>;
563 using TagSeqC = detail::TypeSeq<DDim2>;
565 using type_seq_r =
ddc::type_seq_replace_t<TagSeqA, TagSeqB, TagSeqC>;
566 return ddc::detail::convert_type_seq_to_discrete_domain_t<type_seq_r>(
567 detail::replace_dim_of_1d<
575template <
typename DDom,
typename DDim1,
typename DDim2>
576using replace_dim_of_t =
decltype(replace_dim_of<DDim1, DDim2>(
577 std::declval<DDom>(),
578 std::declval<
typename detail::RebindDomain<DDom, detail::TypeSeq<DDim2>>::type>()));
581template <
class... QueryDDims,
class... DDims>
588template <
class... QueryDDims,
class... DDims>
592 return DiscreteElement<QueryDDims...>(
DiscreteDomain<QueryDDims>(domain).front()...);
595template <
class... QueryDDims,
class... DDims>
599 return DiscreteElement<QueryDDims...>(
DiscreteDomain<QueryDDims>(domain).back()...);
606 DiscreteElement<DDim> m_value = DiscreteElement<DDim>();
609 using iterator_category = std::random_access_iterator_tag;
611 using value_type = DiscreteElement<DDim>;
613 using difference_type = std::ptrdiff_t;
655 if (n >= difference_type(0)) {
656 m_value +=
static_cast<DiscreteElementType>(n);
658 m_value -=
static_cast<DiscreteElementType>(-n);
665 if (n >= difference_type(0)) {
666 m_value -=
static_cast<DiscreteElementType>(n);
668 m_value +=
static_cast<DiscreteElementType>(-n);
682 return xx.m_value == yy.m_value;
685#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
699 return xx.m_value < yy.m_value;
748 return (yy.m_value > xx.m_value) ? (-
static_cast<difference_type>(yy.m_value - xx.m_value))
749 : (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)