11#include <Kokkos_Core.hpp>
45template <
class ChunkType>
46void display(
double time, ChunkType temp)
53 / temp.domain().size();
54 std::cout << std::fixed << std::setprecision(3);
55 std::cout <<
"At t = " << time <<
",\n";
56 std::cout <<
" * mean temperature = " << mean_temp <<
"\n";
59 = temp[ddc::get_domain<DDimY>(temp).front()
60 + ddc::get_domain<DDimY>(temp).size() / 2];
61 std::cout <<
" * temperature[y:"
62 << ddc::get_domain<DDimY>(temp).size() / 2 <<
"] = {";
65 ddc::get_domain<DDimX>(temp),
67 std::cout << std::setw(6) << temp_slice(ix);
69 std::cout <<
" }" << std::endl;
75int main(
int argc,
char** argv)
84 double const x_start = -1.;
86 double const x_end = 1.;
88 size_t const nb_x_points = 10;
90 double const kx = .01;
92 double const y_start = -1.;
94 double const y_end = 1.;
96 size_t const nb_y_points = 100;
98 double const ky = .002;
100 double const start_time = 0.;
102 double const end_time = 10.;
104 ptrdiff_t
const t_output_period = 10;
116 auto const [x_domain, ghosted_x_domain, x_pre_ghost, x_post_ghost]
128 x_domain_begin(x_domain.front(), x_post_ghost.extents());
132 x_domain.back() - x_pre_ghost.extents() + 1,
133 x_pre_ghost.extents());
142 auto const [y_domain, ghosted_y_domain, y_pre_ghost, y_post_ghost]
152 y_domain_begin(y_domain.front(), y_post_ghost.extents());
156 y_domain.back() - y_pre_ghost.extents() + 1,
157 y_pre_ghost.extents());
182 .5 / (kx * invdx2_max + ky * invdy2_max)};
186 std::ceil((end_time - start_time) / max_dt) + .2};
204 DDimY>(ghosted_x_domain, ghosted_y_domain),
211 DDimY>(ghosted_x_domain, ghosted_y_domain),
217 = ghosted_last_temp.span_view();
227 ghosted_initial_temp(ixy)
228 = 9.999 * ((x * x + y * y) < 0.25);
235 DDimY>(ghosted_x_domain, ghosted_y_domain),
243 ghosted_temp[x_domain][y_domain]);
249 for (
auto const iter :
256 ghosted_last_temp[x_pre_ghost][y_domain],
257 ghosted_last_temp[y_domain][x_domain_end]);
259 ghosted_last_temp[y_domain][x_post_ghost],
260 ghosted_last_temp[y_domain][x_domain_begin]);
262 ghosted_last_temp[x_domain][y_pre_ghost],
263 ghosted_last_temp[x_domain][y_domain_end]);
265 ghosted_last_temp[x_domain][y_post_ghost],
266 ghosted_last_temp[x_domain][y_domain_begin]);
273 ghosted_next_temp[x_domain][y_domain]};
286 = ddc::select<DDimX>(ixy);
288 = ddc::select<DDimY>(ixy);
291 double const dx_m = 0.5 * (dx_l + dx_r);
294 double const dy_m = 0.5 * (dy_l + dy_r);
295 next_temp(ix, iy) = last_temp(ix, iy);
297 += kx * ddc::step<DDimT>()
298 * (dx_l * last_temp(ix + 1, iy)
299 - 2.0 * dx_m * last_temp(ix, iy)
300 + dx_r * last_temp(ix - 1, iy))
301 / (dx_l * dx_m * dx_r);
303 += ky * ddc::step<DDimT>()
304 * (dy_l * last_temp(ix, iy + 1)
305 - 2.0 * dy_m * last_temp(ix, iy)
306 + dy_r * last_temp(ix, iy - 1))
307 / (dy_l * dy_m * dy_r);
312 if (iter - last_output >= t_output_period) {
316 ghosted_temp[x_domain][y_domain]);
322 std::swap(ghosted_last_temp, ghosted_next_temp);
327 if (last_output < time_domain.
back()) {
330 ghosted_temp[x_domain][y_domain]);
Definition: discrete_domain.hpp:24
constexpr discrete_element_type back() const noexcept
Definition: discrete_domain.hpp:120
constexpr discrete_element_type front() const noexcept
Definition: discrete_domain.hpp:115
constexpr DiscreteDomain remove_first(mlength_type n) const
Definition: discrete_domain.hpp:135
A DiscreteElement identifies an element of the discrete dimension.
Definition: discrete_element.hpp:125
A DiscreteVector is a vector in the discrete dimension.
Definition: discrete_vector.hpp:228
Definition: kokkos_allocator.hpp:14
Definition: scope_guard.hpp:14
constexpr parallel_device_policy parallel_device
Definition: for_each.hpp:198
constexpr serial_host_policy serial_host
Definition: for_each.hpp:196
DDC_INLINE_FUNCTION Coordinate< CDim > distance_at_left(DiscreteElement< NonUniformPointSampling< CDim > > i)
Definition: non_uniform_point_sampling.hpp:144
DDC_INLINE_FUNCTION Coordinate< typename DDim::continuous_dimension_type... > coordinate(DiscreteElement< DDim... > const &c)
Definition: coordinate_md.hpp:12
auto deepcopy(ChunkDst &&dst, ChunkSrc &&src)
Copy the content of a borrowed chunk into another.
Definition: deepcopy.hpp:19
void init_discrete_space(Args &&... args)
Initialize (emplace) a global singleton discrete space.
Definition: discrete_space.hpp:140
T transform_reduce(serial_host_policy policy, DiscreteDomain< DDims... > const &domain, T neutral, BinaryReductionOp &&reduce, UnaryTransformOp &&transform) noexcept
A reduction over a nD domain using the Serial execution policy.
Definition: transform_reduce.hpp:267
DDC_INLINE_FUNCTION Coordinate< CDim > distance_at_right(DiscreteElement< NonUniformPointSampling< CDim > > i)
Definition: non_uniform_point_sampling.hpp:151
void for_each(serial_host_policy, DiscreteDomain< DDims... > const &domain, Functor &&f) noexcept
iterates over a nD domain using the serial execution policy
Definition: for_each.hpp:127
detail::TaggedVector< CoordinateElement, CDims... > Coordinate
A Coordinate represents a coordinate in the continuous space.
Definition: coordinate.hpp:21
Definition: chunk_span.hpp:26
Definition: reducer.hpp:96
Definition: reducer.hpp:10