13#include <Kokkos_Macros.hpp>
15#include "detail/macros.hpp"
16#include "detail/type_seq.hpp"
18#include "discrete_vector.hpp"
30template <
class... Tags>
41template <
class... Tags>
42struct ToTypeSeq<DiscreteElement<Tags...>>
44 using type = TypeSeq<Tags...>;
50
51using DiscreteElementType = std::size_t;
54KOKKOS_FUNCTION constexpr DiscreteElementType
const&
uid(DiscreteElement<Tag>
const& tuple)
noexcept
60KOKKOS_FUNCTION constexpr DiscreteElementType&
uid(DiscreteElement<Tag>& tuple)
noexcept
65template <
class QueryTag,
class... Tags>
67 DiscreteElement<Tags...>
const& tuple)
noexcept
69 return tuple.
template uid<QueryTag>();
72template <
class QueryTag,
class... Tags>
73KOKKOS_FUNCTION constexpr DiscreteElementType&
uid(DiscreteElement<Tags...>& tuple)
noexcept
75 return tuple.
template uid<QueryTag>();
78template <
class QueryTag,
class... Tags>
80 DiscreteElement<Tags...>
const& arr,
81 DiscreteElement<QueryTag>
const& default_value)
noexcept
83 if constexpr (in_tags_v<QueryTag, detail::TypeSeq<Tags...>>) {
84 return DiscreteElement<QueryTag>(arr);
90template <
class... QueryTags,
class... Tags>
92 DiscreteElement<Tags...>
const& arr)
noexcept
94 return DiscreteElement<QueryTags...>(arr);
97template <
class... QueryTags,
class... Tags>
99 DiscreteElement<Tags...>&& arr)
noexcept
101 return DiscreteElement<QueryTags...>(std::move(arr));
110 is_discrete_element_v<HeadDElem> && (is_discrete_element_v<TailDElems> && ...),
113KOKKOS_FUNCTION constexpr auto const&
take(HeadDElem
const& head, TailDElems
const&... tail)
115 DDC_IF_NVCC_THEN_PUSH_AND_SUPPRESS(implicit_return_from_non_void_function)
116 if constexpr (type_seq_contains_v<detail::TypeSeq<QueryTag>, to_type_seq_t<HeadDElem>>) {
118 (!type_seq_contains_v<detail::TypeSeq<QueryTag>, to_type_seq_t<TailDElems>> && ...),
119 "ERROR: tag redundant");
122 static_assert(
sizeof...(TailDElems) > 0,
"ERROR: tag not found");
123 return take<QueryTag>(tail...);
131template <
class... Tags>
132KOKKOS_FUNCTION
constexpr std::array<DiscreteElementType,
sizeof...(Tags)>& array(
133 DiscreteElement<Tags...>& v)
noexcept
139template <
class... Tags>
140KOKKOS_FUNCTION
constexpr std::array<DiscreteElementType,
sizeof...(Tags)>
const& array(
141 DiscreteElement<Tags...>
const& v)
noexcept
151 return DiscreteElement<DDim>(0);
155
156
157
158template <
class... Tags>
161 using tags_seq = detail::TypeSeq<Tags...>;
163 friend KOKKOS_FUNCTION
constexpr std::array<DiscreteElementType,
sizeof...(Tags)>& detail::
164 array<Tags...>(DiscreteElement<Tags...>& v)
noexcept;
166 friend KOKKOS_FUNCTION
constexpr std::array<DiscreteElementType,
sizeof...(Tags)>
const&
167 detail::array<Tags...>(DiscreteElement<Tags...>
const& v)
noexcept;
170 std::array<DiscreteElementType,
sizeof...(Tags)> m_values;
173 using value_type = DiscreteElementType;
177 return sizeof...(Tags);
183 KOKKOS_DEFAULTED_FUNCTION
constexpr DiscreteElement(DiscreteElement
const&) =
default;
185 KOKKOS_DEFAULTED_FUNCTION
constexpr DiscreteElement(DiscreteElement&&) =
default;
187 template <
class... DElems,
class = std::enable_if_t<(is_discrete_element_v<DElems> && ...)>>
189 : m_values {take<Tags>(delems...).
template uid<Tags>()...}
195 class = std::enable_if_t<(!is_discrete_element_v<Params> && ...)>,
196 class = std::enable_if_t<(std::is_convertible_v<Params, DiscreteElementType> && ...)>,
197 class = std::enable_if_t<
sizeof...(Params) ==
sizeof...(Tags)>>
199 : m_values {
static_cast<DiscreteElementType>(params)...}
205 KOKKOS_DEFAULTED_FUNCTION DiscreteElement&
operator=(DiscreteElement
const& other) =
default;
207 KOKKOS_DEFAULTED_FUNCTION DiscreteElement&
operator=(DiscreteElement&& other) =
default;
209 template <
class QueryTag>
212 static_assert(in_tags_v<QueryTag, tags_seq>,
"requested Tag absent from DiscreteElement");
213 return m_values[type_seq_rank_v<QueryTag, tags_seq>];
216 template <
class QueryTag>
219 static_assert(in_tags_v<QueryTag, tags_seq>,
"requested Tag absent from DiscreteElement");
220 return m_values[type_seq_rank_v<QueryTag, tags_seq>];
223 template <std::size_t N =
sizeof...(Tags)>
229 template <std::size_t N =
sizeof...(Tags)>
230 KOKKOS_FUNCTION constexpr std::enable_if_t<N == 1, value_type
const&>
uid()
const noexcept
235 template <std::size_t N =
sizeof...(Tags),
class = std::enable_if_t<N == 1>>
242 template <std::size_t N =
sizeof...(Tags),
class = std::enable_if_t<N == 1>>
245 DiscreteElement
const tmp = *
this;
250 template <std::size_t N =
sizeof...(Tags),
class = std::enable_if_t<N == 1>>
257 template <std::size_t N =
sizeof...(Tags),
class = std::enable_if_t<N == 1>>
260 DiscreteElement
const tmp = *
this;
265 template <
class... OTags>
268 static_assert(((type_seq_contains_v<detail::TypeSeq<OTags>, tags_seq>) && ...));
269 ((m_values[type_seq_rank_v<OTags, tags_seq>] += rhs.
template get<OTags>()), ...);
275 std::size_t N =
sizeof...(Tags),
276 class = std::enable_if_t<N == 1>,
277 class = std::enable_if_t<std::is_integral_v<IntegralType>>>
284 template <
class... OTags>
287 static_assert(((type_seq_contains_v<detail::TypeSeq<OTags>, tags_seq>) && ...));
288 ((m_values[type_seq_rank_v<OTags, tags_seq>] -= rhs.
template get<OTags>()), ...);
294 std::size_t N =
sizeof...(Tags),
295 class = std::enable_if_t<N == 1>,
296 class = std::enable_if_t<std::is_integral_v<IntegralType>>>
304inline std::ostream& operator<<(std::ostream& out, DiscreteElement<>
const&)
310template <
class Head,
class... Tags>
311std::ostream& operator<<(std::ostream& out, DiscreteElement<Head, Tags...>
const& arr)
314 out << uid<Head>(arr);
315 ((out <<
", " << uid<Tags>(arr)), ...);
321template <
class... Tags,
class... OTags>
323 DiscreteElement<Tags...>
const& lhs,
324 DiscreteElement<OTags...>
const& rhs)
noexcept
326 return ((lhs.
template uid<Tags>() == rhs.
template uid<Tags>()) && ...);
329#if !defined(__cpp_impl_three_way_comparison) || __cpp_impl_three_way_comparison < 201902L
342 DiscreteElement<Tag>
const& lhs,
343 DiscreteElement<Tag>
const& rhs)
345 return lhs.uid() < rhs.uid();
350 DiscreteElement<Tag>
const& lhs,
351 DiscreteElement<Tag>
const& rhs)
353 return lhs.uid() <= rhs.uid();
358 DiscreteElement<Tag>
const& lhs,
359 DiscreteElement<Tag>
const& rhs)
361 return lhs.uid() > rhs.uid();
366 DiscreteElement<Tag>
const& lhs,
367 DiscreteElement<Tag>
const& rhs)
369 return lhs.uid() >= rhs.uid();
374template <
class... Tags,
class... OTags>
376 DiscreteElement<Tags...>
const& lhs,
379 using detail::TypeSeq;
380 static_assert(((type_seq_contains_v<TypeSeq<OTags>, TypeSeq<Tags...>>) && ...));
381 DiscreteElement<Tags...> result(lhs);
389 class = std::enable_if_t<std::is_integral_v<IntegralType>>,
390 class = std::enable_if_t<!is_discrete_vector_v<IntegralType>>>
392 DiscreteElement<Tag>
const& lhs,
393 IntegralType
const& rhs)
395 DiscreteElement<Tag> result(lhs);
400template <
class... Tags,
class... OTags>
402 DiscreteElement<Tags...>
const& lhs,
405 using detail::TypeSeq;
406 static_assert(((type_seq_contains_v<TypeSeq<OTags>, TypeSeq<Tags...>>) && ...));
407 DiscreteElement<Tags...> result(lhs);
415 class = std::enable_if_t<std::is_integral_v<IntegralType>>,
416 class = std::enable_if_t<!is_discrete_vector_v<IntegralType>>>
418 DiscreteElement<Tag>
const& lhs,
419 IntegralType
const& rhs)
421 DiscreteElement<Tag> result(lhs);
428template <
class... Tags,
class... OTags>
430 DiscreteElement<Tags...>
const& lhs,
431 DiscreteElement<OTags...>
const& rhs)
433 static_assert(type_seq_same_v<detail::TypeSeq<Tags...>, detail::TypeSeq<OTags...>>);
434 return DiscreteVector<Tags...>((uid<Tags>(lhs) - uid<Tags>(rhs))...);
KOKKOS_FUNCTION constexpr DiscreteElement & operator+=(DiscreteVector< OTags... > const &rhs)
KOKKOS_FUNCTION constexpr value_type const & uid() const noexcept
KOKKOS_FUNCTION constexpr value_type & uid() noexcept
KOKKOS_DEFAULTED_FUNCTION ~DiscreteElement()=default
KOKKOS_FUNCTION constexpr DiscreteElement & operator++()
KOKKOS_FUNCTION constexpr DiscreteElement & operator-=(IntegralType const &rhs)
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteElement(DiscreteElement const &)=default
KOKKOS_FUNCTION constexpr DiscreteElement(DElems const &... delems) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement & operator-=(DiscreteVector< OTags... > const &rhs)
KOKKOS_FUNCTION constexpr DiscreteElement operator++(int)
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteElement()=default
KOKKOS_FUNCTION constexpr DiscreteElement & operator--()
KOKKOS_DEFAULTED_FUNCTION DiscreteElement & operator=(DiscreteElement const &other)=default
KOKKOS_FUNCTION constexpr DiscreteElement operator--(int)
KOKKOS_FUNCTION constexpr std::enable_if_t< N==1, value_type & > uid() noexcept
static KOKKOS_FUNCTION constexpr std::size_t size() noexcept
KOKKOS_FUNCTION constexpr std::enable_if_t< N==1, value_type const & > uid() const noexcept
KOKKOS_FUNCTION constexpr DiscreteElement & operator+=(IntegralType const &rhs)
KOKKOS_DEFAULTED_FUNCTION DiscreteElement & operator=(DiscreteElement &&other)=default
KOKKOS_FUNCTION constexpr DiscreteElement(Params const &... params) noexcept
KOKKOS_DEFAULTED_FUNCTION constexpr DiscreteElement(DiscreteElement &&)=default
KOKKOS_FUNCTION constexpr bool operator!=(DiscreteVector< OTags... > const &rhs) const noexcept
The top-level namespace of DDC.
KOKKOS_FUNCTION constexpr bool operator<(DiscreteElement< Tag > const &lhs, DiscreteElement< Tag > const &rhs)
KOKKOS_FUNCTION constexpr bool operator==(DiscreteElement< Tags... > const &lhs, DiscreteElement< OTags... > const &rhs) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< QueryTags... > select(DiscreteElement< Tags... > &&arr) noexcept
constexpr DiscreteElement< DDim > create_reference_discrete_element() noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< QueryTags... > select(DiscreteElement< Tags... > const &arr) noexcept
KOKKOS_FUNCTION constexpr auto const & take(HeadDElem const &head, TailDElems const &... tail)
Returns a reference towards the DiscreteElement that contains the QueryTag.
KOKKOS_FUNCTION constexpr DiscreteElement< Tag > operator-(DiscreteElement< Tag > const &lhs, IntegralType const &rhs)
KOKKOS_FUNCTION constexpr DiscreteVector< Tags... > operator-(DiscreteElement< Tags... > const &lhs, DiscreteElement< OTags... > const &rhs)
binary operator: -
KOKKOS_FUNCTION constexpr DiscreteElement< QueryTag > select_or(DiscreteElement< Tags... > const &arr, DiscreteElement< QueryTag > const &default_value) noexcept
KOKKOS_FUNCTION constexpr DiscreteElementType const & uid(DiscreteElement< Tag > const &tuple) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< Tags... > operator+(DiscreteElement< Tags... > const &lhs, DiscreteVector< OTags... > const &rhs)
right external binary operators: +, -
constexpr bool is_discrete_element_v
KOKKOS_FUNCTION constexpr bool operator>=(DiscreteElement< Tag > const &lhs, DiscreteElement< Tag > const &rhs)
KOKKOS_FUNCTION constexpr bool operator<=(DiscreteElement< Tag > const &lhs, DiscreteElement< Tag > const &rhs)
KOKKOS_FUNCTION constexpr DiscreteElementType & uid(DiscreteElement< Tags... > &tuple) noexcept
KOKKOS_FUNCTION constexpr bool operator>(DiscreteElement< Tag > const &lhs, DiscreteElement< Tag > const &rhs)
KOKKOS_FUNCTION constexpr DiscreteElement< Tag > operator+(DiscreteElement< Tag > const &lhs, IntegralType const &rhs)
KOKKOS_FUNCTION constexpr DiscreteElementType & uid(DiscreteElement< Tag > &tuple) noexcept
KOKKOS_FUNCTION constexpr DiscreteElement< Tags... > operator-(DiscreteElement< Tags... > const &lhs, DiscreteVector< OTags... > const &rhs)
KOKKOS_FUNCTION constexpr DiscreteElementType const & uid(DiscreteElement< Tags... > const &tuple) noexcept