13#include <Kokkos_Core.hpp>
14#include <Kokkos_StdAlgorithms.hpp>
16#include "detail/kokkos.hpp"
17#include "detail/tagged_vector.hpp"
18#include "detail/type_seq.hpp"
20#include "discrete_element.hpp"
21#include "discrete_vector.hpp"
26struct SparseDiscreteDomainIterator;
28template <
class... DDims>
36template <
class... Tags>
47template <
class... Tags>
50 using type = TypeSeq<Tags...>;
53template <
class T,
class U>
56template <
class... DDims,
class... ODDims>
62template <
class InputIt1,
class InputIt2>
63KOKKOS_FUNCTION
bool equal(InputIt1 first1, InputIt1 last1, InputIt2 first2)
65 for (; first1 != last1; ++first1, ++first2) {
66 if (!(*first1 == *first2)) {
76 class T =
typename std::iterator_traits<ForwardIt>::value_type,
78KOKKOS_FUNCTION ForwardIt lower_bound(ForwardIt first, ForwardIt last, T
const& value, Compare comp)
81 typename std::iterator_traits<ForwardIt>::difference_type count = last - first;
84 typename std::iterator_traits<ForwardIt>::difference_type
const step = count / 2;
87 if (comp(*it, value)) {
100 class T =
typename std::iterator_traits<ForwardIt>::value_type,
102KOKKOS_FUNCTION
bool binary_search(ForwardIt first, ForwardIt last, T
const& value, Compare comp)
104 first = ::
ddc::detail::lower_bound(first, last, value, comp);
105 return (!(first == last) && !(comp(value, *first)));
111 KOKKOS_FUNCTION DiscreteElementType
112 operator()(DiscreteElement<DDim>
const& delem)
const noexcept
120template <
class... DDims>
127 using discrete_element_type = DiscreteElement<DDims...>;
132 detail::TaggedVector<Kokkos::View<DiscreteElementType*, Kokkos::SharedSpace>, DDims...> m_views;
137 return sizeof...(DDims);
143 template <
class... DDoms,
class = std::enable_if_t<(is_sparse_discrete_domain_v<DDoms> && ...)>>
145 : m_views(domains.m_views...)
150
151
153 Kokkos::View<DiscreteElement<DDims>*, Kokkos::SharedSpace>
const&... views)
155 ((m_views[type_seq_rank_v<DDims, detail::TypeSeq<DDims...>>]
156 = Kokkos::View<DiscreteElementType*, Kokkos::SharedSpace>(views.label(), views.size())),
158 Kokkos::DefaultExecutionSpace
const execution_space;
159 ((Kokkos::Experimental::transform(
160 "SparseDiscreteDomainCtor",
163 m_views[type_seq_rank_v<DDims, detail::TypeSeq<DDims...>>],
164 detail::GetUidFn<DDims>())),
166 execution_space.fence(
"SparseDiscreteDomainCtor");
180 template <
class... ODims>
183 if (
empty() && other.empty()) {
186 if (m_views.size() != other.m_views.size()) {
189 for (std::size_t i = 0; i < m_views.size(); ++i) {
190 if (m_views[i].size() != other.m_views[i].size()) {
194 equal(m_views[i].data(),
195 m_views[i].data() + m_views[i].size(),
196 other.m_views[i].data())) {
203#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
205 template <
class...
ODims>
208 return !(*
this ==
other);
214 return (1UL * ... * get<DDims>(m_views).size());
219 return discrete_vector_type(get<DDims>(m_views).size()...);
227 template <
class QueryDDim>
235 return discrete_element_type(get<DDims>(m_views)(0)...);
240 return discrete_element_type(get<DDims>(m_views)(get<DDims>(m_views).size() - 1)...);
264 discrete_vector_type n1,
265 discrete_vector_type n2)
const
273 return m_views(get<DDims>(dvect)...);
276 template <
class... DElems>
280 sizeof...(DDims) == (0 + ... + DElems::size()),
281 "Invalid number of dimensions");
282 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
283 return (detail::binary_search(
284 get<DDims>(m_views).data(),
285 get<DDims>(m_views).data() + get<DDims>(m_views).size(),
286 uid<DDims>(take<DDims>(delems...)),
291 template <
class... DElems>
293 DElems
const&... delems)
const noexcept
296 sizeof...(DDims) == (0 + ... + DElems::size()),
297 "Invalid number of dimensions");
298 static_assert((is_discrete_element_v<DElems> && ...),
"Expected DiscreteElements");
299 assert(contains(delems...));
301 (detail::lower_bound(
302 get<DDims>(m_views).data(),
303 get<DDims>(m_views).data() + get<DDims>(m_views).size(),
304 uid<DDims>(take<DDims>(delems...)),
306 - get<DDims>(m_views).data())...);
320 std::size_t N =
sizeof...(DDims),
321 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
324 return Kokkos::Experimental::begin(get<DDim0>(m_views));
328 std::size_t N =
sizeof...(DDims),
329 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
332 return Kokkos::Experimental::end(get<DDim0>(m_views));
336 std::size_t N =
sizeof...(DDims),
337 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
340 return Kokkos::Experimental::cbegin(get<DDim0>(m_views));
344 std::size_t N =
sizeof...(DDims),
345 class DDim0 = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
348 return Kokkos::Experimental::cend(get<DDim0>(m_views));
352 std::size_t N =
sizeof...(DDims),
353 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
360 std::size_t N =
sizeof...(DDims),
361 class = std::enable_if_t<N == 1, std::tuple_element_t<0, std::tuple<DDims...>>>>
375 using discrete_element_type = DiscreteElement<>;
387 template <
class... ODDims>
410#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
414 return !(*
this ==
other);
506template <
class... QueryDDims,
class... DDims>
511 DiscreteElement<QueryDDims...>(domain.front()),
512 DiscreteElement<QueryDDims...>(domain.extents()));
518struct ConvertTypeSeqToSparseDiscreteDomain;
520template <
class... DDims>
521struct ConvertTypeSeqToSparseDiscreteDomain<detail::TypeSeq<DDims...>>
527using convert_type_seq_to_sparse_discrete_domain_t =
528 typename ConvertTypeSeqToSparseDiscreteDomain<T>::type;
533template <
class... DDimsA,
class... DDimsB>
538 using TagSeqA = detail::TypeSeq<DDimsA...>;
539 using TagSeqB = detail::TypeSeq<DDimsB...>;
541 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
542 return detail::convert_type_seq_to_sparse_discrete_domain_t<type_seq_r>(DDom_a);
548template <
class... DDimsB,
class... DDimsA>
552 using TagSeqA = detail::TypeSeq<DDimsA...>;
553 using TagSeqB = detail::TypeSeq<DDimsB...>;
555 using type_seq_r = type_seq_remove_t<TagSeqA, TagSeqB>;
556 return detail::convert_type_seq_to_sparse_discrete_domain_t<type_seq_r>(DDom_a);
562template <
typename DDim1,
typename DDim2,
typename DDimA,
typename... DDimsB>
563KOKKOS_FUNCTION
constexpr std::conditional_t<
564 std::is_same_v<DDimA, DDim1>,
571 if constexpr (std::is_same_v<DDimA, DDim1>) {
581template <
typename DDim1,
typename DDim2,
typename... DDimsA,
typename... DDimsB>
587 using TagSeqA = detail::TypeSeq<DDimsA...>;
588 using TagSeqB = detail::TypeSeq<DDim1>;
589 using TagSeqC = detail::TypeSeq<DDim2>;
591 using type_seq_r =
ddc::type_seq_replace_t<TagSeqA, TagSeqB, TagSeqC>;
592 return ddc::detail::convert_type_seq_to_sparse_discrete_domain_t<type_seq_r>(
593 detail::replace_dim_of_1d<
600template <
class... QueryDDims,
class... DDims>
607template <
class... QueryDDims,
class... DDims>
614template <
class... QueryDDims,
class... DDims>
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteElement()=default
KOKKOS_FUNCTION constexpr bool operator!=(DiscreteVector< OTags... > const &rhs) const noexcept
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteVector()=default
KOKKOS_FUNCTION constexpr bool operator==(SparseDiscreteDomain const &other) const
static KOKKOS_FUNCTION bool contains(DiscreteElement<>) noexcept
static KOKKOS_FUNCTION constexpr std::size_t rank()
KOKKOS_DEFAULTED_FUNCTION constexpr SparseDiscreteDomain()=default
KOKKOS_FUNCTION constexpr SparseDiscreteDomain(SparseDiscreteDomain< ODDims... > const &domain)
KOKKOS_FUNCTION constexpr DiscreteElement operator()(DiscreteVector<> const &) const noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain take_last(discrete_vector_type n) const
static KOKKOS_FUNCTION DiscreteVector distance_from_front() noexcept
static KOKKOS_FUNCTION constexpr std::size_t size()
static KOKKOS_FUNCTION constexpr bool empty() noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain remove(discrete_vector_type n1, discrete_vector_type n2) const
static KOKKOS_FUNCTION DiscreteVector distance_from_front(DiscreteElement<>) noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain remove_first(discrete_vector_type n) const
static KOKKOS_FUNCTION constexpr discrete_element_type back() noexcept
static KOKKOS_FUNCTION bool contains() noexcept
KOKKOS_FUNCTION constexpr operator bool()
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain & operator=(SparseDiscreteDomain const &x)=default
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain & operator=(SparseDiscreteDomain &&x)=default
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain(SparseDiscreteDomain const &x)=default
KOKKOS_DEFAULTED_FUNCTION ~SparseDiscreteDomain()=default
KOKKOS_FUNCTION constexpr SparseDiscreteDomain take_first(discrete_vector_type n) const
static KOKKOS_FUNCTION constexpr discrete_element_type front() noexcept
static KOKKOS_FUNCTION constexpr discrete_vector_type extents() noexcept
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain(SparseDiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr SparseDiscreteDomain remove_last(discrete_vector_type n) const
KOKKOS_FUNCTION constexpr discrete_element_type back() const noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain remove_first(discrete_vector_type n) const
static KOKKOS_FUNCTION constexpr std::size_t rank()
KOKKOS_FUNCTION constexpr decltype(auto) operator[](std::size_t n) const
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain & operator=(SparseDiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr SparseDiscreteDomain take_last(discrete_vector_type n) const
KOKKOS_FUNCTION DiscreteVector< DDims... > distance_from_front(DElems const &... delems) const noexcept
KOKKOS_FUNCTION constexpr auto discrete_elements() const noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain remove(discrete_vector_type n1, discrete_vector_type n2) const
KOKKOS_FUNCTION constexpr SparseDiscreteDomain(DDoms const &... domains)
Construct a SparseDiscreteDomain by copies and merge of domains.
SparseDiscreteDomain(Kokkos::View< DiscreteElement< DDims > *, Kokkos::SharedSpace > const &... views)
Construct a SparseDiscreteDomain with Kokkos::View explicitly listing the discrete elements.
KOKKOS_FUNCTION auto cbegin() const
KOKKOS_FUNCTION constexpr decltype(auto) operator[](std::size_t n)
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain & operator=(SparseDiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr std::size_t size() const
KOKKOS_FUNCTION constexpr bool operator==(SparseDiscreteDomain< ODims... > const &other) const
KOKKOS_FUNCTION bool contains(DElems const &... delems) const noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain take_first(discrete_vector_type n) const
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain()=default
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain(SparseDiscreteDomain &&x)=default
KOKKOS_FUNCTION constexpr discrete_element_type front() const noexcept
KOKKOS_FUNCTION constexpr operator bool()
KOKKOS_FUNCTION constexpr bool empty() const noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< DDims... > operator()(DiscreteVector< DDims... > const &dvect) const noexcept
KOKKOS_DEFAULTED_FUNCTION SparseDiscreteDomain(SparseDiscreteDomain const &x)=default
KOKKOS_FUNCTION constexpr SparseDiscreteDomain remove_last(discrete_vector_type n) const
KOKKOS_FUNCTION auto end() const
KOKKOS_FUNCTION auto cend() const
friend class SparseDiscreteDomain
KOKKOS_FUNCTION auto begin() const
KOKKOS_FUNCTION constexpr discrete_vector_type extents() const noexcept
KOKKOS_FUNCTION constexpr DiscreteVector< QueryDDim > extent() const noexcept
KOKKOS_DEFAULTED_FUNCTION ~SparseDiscreteDomain()=default
The top-level namespace of DDC.
KOKKOS_FUNCTION constexpr DiscreteElement< QueryDDims... > back(SparseDiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr SparseDiscreteDomain< QueryDDims... > select(SparseDiscreteDomain< DDims... > const &domain)
KOKKOS_FUNCTION constexpr DiscreteVector< QueryDDims... > extents(SparseDiscreteDomain< DDims... > const &domain) noexcept
KOKKOS_FUNCTION constexpr auto replace_dim_of(SparseDiscreteDomain< DDimsA... > const &DDom_a, SparseDiscreteDomain< DDimsB... > const &DDom_b) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< QueryDDims... > front(SparseDiscreteDomain< DDims... > const &domain) noexcept
constexpr bool is_sparse_discrete_domain_v
KOKKOS_FUNCTION constexpr auto remove_dims_of(SparseDiscreteDomain< DDimsA... > const &DDom_a) noexcept
Remove the dimensions DDimsB from DDom_a.
KOKKOS_FUNCTION constexpr auto remove_dims_of(SparseDiscreteDomain< DDimsA... > const &DDom_a, SparseDiscreteDomain< DDimsB... > const &DDom_b) noexcept