10#include <initializer_list>
16#include <Kokkos_Core.hpp>
24struct NonUniformBSplinesBase
36
37
38
39
40
41
42
43
44template <
class CDim, std::size_t D>
47 static_assert(D > 0,
"Parameter `D` must be positive");
51 using continuous_dimension_type = CDim;
57
58
59
60 static constexpr std::size_t
degree()
noexcept
66
67
68
71 return CDim::PERIODIC;
75
76
77
84
85
86
87
88 template <
class DDim,
class MemorySpace>
91 template <
class ODDim,
class OMemorySpace>
105 using discrete_element_type = DiscreteElement<DDim>;
114 ddc::DiscreteElement<DDim> m_reference;
120
121
122
123
124
125
126 Impl(std::initializer_list<
ddc::Coordinate<CDim>> breaks)
127 :
Impl(breaks.begin(), breaks.end())
132
133
134
135
136
137
138 explicit Impl(std::vector<
ddc::Coordinate<CDim>>
const& breaks)
139 :
Impl(breaks.begin(), breaks.end())
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161 template <
class RandomIt>
162 Impl(RandomIt breaks_begin, RandomIt breaks_end);
165
166
167
168 template <
class OriginMemorySpace>
169 explicit Impl(
Impl<DDim, OriginMemorySpace>
const& impl)
170 : m_knot_domain(impl.m_knot_domain)
171 , m_break_point_domain(impl.m_break_point_domain)
172 , m_reference(impl.m_reference)
177
178
179
183
184
185
192
193
194
195
199
200
201
202
206
207
208
209
210
211
212
213
214
215
216
217 KOKKOS_INLINE_FUNCTION discrete_element_type
218 eval_basis(DSpan1D values,
ddc::Coordinate<CDim>
const& x)
const;
221
222
223
224
225
226
227
228
229
230
231
232 KOKKOS_INLINE_FUNCTION discrete_element_type
233 eval_deriv(DSpan1D derivs,
ddc::Coordinate<CDim>
const& x)
const;
236
237
238
239
240
241
242
243
244
245
246
247
250 ddc::Coordinate<CDim>
const& x,
251 std::size_t n)
const;
254
255
256
257
258
259
260
261
262 KOKKOS_INLINE_FUNCTION
ddc::DiscreteElement<knot_discrete_dimension_type>
265 return m_knot_domain.front() + (ix - m_reference).value();
269
270
271
272
273
274
275
276
277 KOKKOS_INLINE_FUNCTION
ddc::DiscreteElement<knot_discrete_dimension_type>
285
286
287
288 KOKKOS_INLINE_FUNCTION
ddc::Coordinate<CDim>
rmin()
const noexcept
290 return ddc::coordinate(m_break_point_domain.front());
294
295
296
297 KOKKOS_INLINE_FUNCTION
ddc::Coordinate<CDim>
rmax()
const noexcept
299 return ddc::coordinate(m_break_point_domain.back());
303
304
305
306 KOKKOS_INLINE_FUNCTION
double length()
const noexcept
312
313
314
315
316
317
318
319 KOKKOS_INLINE_FUNCTION std::size_t
size()
const noexcept
325
326
327
328 KOKKOS_INLINE_FUNCTION discrete_domain_type
full_domain()
const
330 return discrete_domain_type(m_reference, discrete_vector_type(
size()));
334
335
336
340 return m_break_point_domain;
344
345
346
347
348
349 KOKKOS_INLINE_FUNCTION std::size_t
npoints()
const noexcept
351 return m_knot_domain.size() - 2 *
degree();
355
356
357
358
359
360 KOKKOS_INLINE_FUNCTION std::size_t
nbasis()
const noexcept
366
367
368
369
370
371
372 KOKKOS_INLINE_FUNCTION std::size_t
ncells()
const noexcept
378 KOKKOS_INLINE_FUNCTION discrete_element_type get_first_bspline_in_cell(
379 ddc::DiscreteElement<knot_discrete_dimension_type>
const& ic)
const
381 return m_reference + (ic - m_break_point_domain.front()).value();
385
386
387
388
389 KOKKOS_INLINE_FUNCTION
ddc::DiscreteElement<knot_discrete_dimension_type> find_cell_start(
390 ddc::Coordinate<CDim>
const& x)
const;
400
401
402
403
410concept non_uniform_bsplines = is_non_uniform_bsplines_v<DDim>;
416template <
class RandomIt>
418 RandomIt
const breaks_begin,
419 RandomIt
const breaks_end)
421 ddc::DiscreteElement<knot_discrete_dimension_type>(0),
423 (breaks_end - breaks_begin)
425 , m_break_point_domain(
426 ddc::DiscreteElement<knot_discrete_dimension_type>(
degree()),
428 (breaks_end - breaks_begin)))
429 , m_reference(
ddc::create_reference_discrete_element<DDim>())
431 std::vector<
ddc::Coordinate<CDim>> knots((breaks_end - breaks_begin) + 2 *
degree());
434 for (RandomIt it = breaks_begin; it < breaks_end; ++it) {
438 ddc::Coordinate<CDim>
const rmin = knots[
degree()];
439 ddc::Coordinate<CDim>
const rmax = knots[(breaks_end - breaks_begin) +
degree() - 1];
444 double const period = rmax - rmin;
445 for (std::size_t i = 1; i <
degree() + 1; ++i) {
451 for (std::size_t i = 1; i <
degree() + 1; ++i) {
456 ddc::init_discrete_space<knot_discrete_dimension_type>(knots);
459template <
class CDim, std::size_t D>
460template <
class DDim,
class MemorySpace>
462 Impl<DDim, MemorySpace>::
eval_basis(DSpan1D values,
ddc::Coordinate<CDim>
const& x)
const
464 KOKKOS_ASSERT(values.size() == D + 1)
471 KOKKOS_ASSERT(values.size() ==
degree() + 1)
474 ddc::DiscreteElement<knot_discrete_dimension_type>
const icell = find_cell_start(x);
476 KOKKOS_ASSERT(icell >= m_break_point_domain.front())
477 KOKKOS_ASSERT(icell <= m_break_point_domain.back())
478 KOKKOS_ASSERT(
ddc::coordinate(icell) - x <=
length() * 1e-14)
479 KOKKOS_ASSERT(x -
ddc::coordinate(icell + 1) <=
length() * 1e-14)
484 for (std::size_t j = 0; j <
degree(); ++j) {
485 left[j] = x -
ddc::coordinate(icell - j);
486 right[j] =
ddc::coordinate(icell + j + 1) - x;
488 for (std::size_t r = 0; r < j + 1; ++r) {
489 temp = values[r] / (right[r] + left[j - r]);
490 values[r] = saved + right[r] * temp;
491 saved = left[j - r] * temp;
493 values[j + 1] = saved;
496 return get_first_bspline_in_cell(icell);
499template <
class CDim, std::size_t D>
500template <
class DDim,
class MemorySpace>
502 Impl<DDim, MemorySpace>::
eval_deriv(DSpan1D derivs,
ddc::Coordinate<CDim>
const& x)
const
509 KOKKOS_ASSERT(derivs.size() ==
degree() + 1)
512 ddc::DiscreteElement<knot_discrete_dimension_type>
const icell = find_cell_start(x);
514 KOKKOS_ASSERT(icell >= m_break_point_domain.front())
515 KOKKOS_ASSERT(icell <= m_break_point_domain.back())
516 KOKKOS_ASSERT(
ddc::coordinate(icell) <= x)
517 KOKKOS_ASSERT(
ddc::coordinate(icell + 1) >= x)
522
523
524
525
529 for (std::size_t j = 0; j <
degree() - 1; ++j) {
530 left[j] = x -
ddc::coordinate(icell - j);
531 right[j] =
ddc::coordinate(icell + j + 1) - x;
533 for (std::size_t r = 0; r < j + 1; ++r) {
534 temp = derivs[r] / (right[r] + left[j - r]);
535 derivs[r] = saved + right[r] * temp;
536 saved = left[j - r] * temp;
538 derivs[j + 1] = saved;
542
543
544
546 / (
ddc::coordinate(icell + 1) -
ddc::coordinate(icell + 1 -
degree()));
548 for (std::size_t j = 1; j <
degree(); ++j) {
551 / (
ddc::coordinate(icell + j + 1) -
ddc::coordinate(icell + j + 1 -
degree()));
552 derivs[j] = temp - saved;
556 return get_first_bspline_in_cell(icell);
559template <
class CDim, std::size_t D>
560template <
class DDim,
class MemorySpace>
562 Impl<DDim, MemorySpace>::eval_basis_and_n_derivs(
563 ddc::DSpan2D
const derivs,
564 ddc::Coordinate<CDim>
const& x,
565 std::size_t
const n)
const
570 std::array<
double, 2 * (
degree() + 1)> a_ptr;
571 Kokkos::mdspan<
double, Kokkos::extents<std::size_t,
degree() + 1, 2>>
const a(a_ptr.data());
574 Kokkos::mdspan<
double, Kokkos::extents<std::size_t,
degree() + 1,
degree() + 1>>
const ndu(
581 KOKKOS_ASSERT(derivs.extent(0) == 1 +
degree())
582 KOKKOS_ASSERT(derivs.extent(1) == 1 + n)
585 ddc::DiscreteElement<knot_discrete_dimension_type>
const icell = find_cell_start(x);
587 KOKKOS_ASSERT(icell >= m_break_point_domain.front())
588 KOKKOS_ASSERT(icell <= m_break_point_domain.back())
589 KOKKOS_ASSERT(
ddc::coordinate(icell) <= x)
590 KOKKOS_ASSERT(
ddc::coordinate(icell + 1) >= x)
602 DDC_MDSPAN_ACCESS_OP(ndu, 0, 0) = 1.0;
603 for (std::size_t j = 0; j <
degree(); ++j) {
604 left[j] = x -
ddc::coordinate(icell - j);
605 right[j] =
ddc::coordinate(icell + j + 1) - x;
607 for (std::size_t r = 0; r < j + 1; ++r) {
610 DDC_MDSPAN_ACCESS_OP(ndu, r, j + 1) = 1.0 / (right[r] + left[j - r]);
613 temp = DDC_MDSPAN_ACCESS_OP(ndu, j, r) * DDC_MDSPAN_ACCESS_OP(ndu, r, j + 1);
614 DDC_MDSPAN_ACCESS_OP(ndu, j + 1, r) = saved + right[r] * temp;
615 saved = left[j - r] * temp;
617 DDC_MDSPAN_ACCESS_OP(ndu, j + 1, j + 1) = saved;
620 for (std::size_t j = 0; j <
degree() + 1; ++j) {
621 DDC_MDSPAN_ACCESS_OP(derivs, j, 0) = DDC_MDSPAN_ACCESS_OP(ndu,
degree(), j);
624 for (
int r = 0; r <
static_cast<
int>(
degree() + 1); ++r) {
627 DDC_MDSPAN_ACCESS_OP(a, 0, 0) = 1.0;
628 for (
int k = 1; k <
static_cast<
int>(n + 1); ++k) {
630 int const rk = r - k;
633 DDC_MDSPAN_ACCESS_OP(a, 0, s2)
634 = DDC_MDSPAN_ACCESS_OP(a, 0, s1) * DDC_MDSPAN_ACCESS_OP(ndu, rk, pk + 1);
635 d = DDC_MDSPAN_ACCESS_OP(a, 0, s2) * DDC_MDSPAN_ACCESS_OP(ndu, pk, rk);
637 int const j1 = rk > -1 ? 1 : (-rk);
638 int const j2 = (r - 1) <= pk ? k : (
degree() - r + 1);
639 for (
int j = j1; j < j2; ++j) {
640 DDC_MDSPAN_ACCESS_OP(a, j, s2)
641 = (DDC_MDSPAN_ACCESS_OP(a, j, s1) - DDC_MDSPAN_ACCESS_OP(a, j - 1, s1))
642 * DDC_MDSPAN_ACCESS_OP(ndu, rk + j, pk + 1);
643 d += DDC_MDSPAN_ACCESS_OP(a, j, s2) * DDC_MDSPAN_ACCESS_OP(ndu, pk, rk + j);
646 DDC_MDSPAN_ACCESS_OP(a, k, s2) = -DDC_MDSPAN_ACCESS_OP(a, k - 1, s1)
647 * DDC_MDSPAN_ACCESS_OP(ndu, r, pk + 1);
648 d += DDC_MDSPAN_ACCESS_OP(a, k, s2) * DDC_MDSPAN_ACCESS_OP(ndu, pk, r);
650 DDC_MDSPAN_ACCESS_OP(derivs, r, k) = d;
651 Kokkos::kokkos_swap(s1, s2);
656 for (
int k = 1; k <
static_cast<
int>(n + 1); ++k) {
657 for (std::size_t i = 0; i < derivs.extent(0); ++i) {
658 DDC_MDSPAN_ACCESS_OP(derivs, i, k) *= r;
663 return get_first_bspline_in_cell(icell);
666template <
class CDim, std::size_t D>
667template <
class DDim,
class MemorySpace>
670 D>::
Impl<DDim, MemorySpace>::find_cell_start(
ddc::Coordinate<CDim>
const& x)
const
676 return m_break_point_domain.front();
679 return m_break_point_domain.back() - 1;
683 ddc::DiscreteElement<knot_discrete_dimension_type> low = m_break_point_domain.front();
684 ddc::DiscreteElement<knot_discrete_dimension_type> high = m_break_point_domain.back();
685 ddc::DiscreteElement<knot_discrete_dimension_type> icell = low + (high - low) / 2;
686 while (x <
ddc::coordinate(icell) || x >=
ddc::coordinate(icell + 1)) {
687 if (x <
ddc::coordinate(icell)) {
692 icell = low + (high - low) / 2;
friend class DiscreteDomain
KOKKOS_FUNCTION constexpr bool operator!=(DiscreteVector< OTags... > const &rhs) const noexcept
The top-level namespace of DDC.
constexpr bool is_non_uniform_bsplines_v
Indicates if a tag corresponds to non-uniform B-splines or not.