AddForceCentralGravPotential.cpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//

 
#include "shammodels/common/modules/AddForceCentralGravPotential.hpp"
#include "shambackends/kernel_call_distrib.hpp"
#include "shamrock/patch/PatchDataField.hpp"
#include "shamsys/NodeInstance.hpp"
 
namespace shammodels::common::modules {
 
    template<class Tvec>
    void AddForceCentralGravPotential<Tvec>::_impl_evaluate_internal() {
 
        __shamrock_stack_entry();
 
        auto edges = get_edges();
 
        edges.spans_positions.check_sizes(edges.sizes.indexes);
        edges.spans_accel_ext.ensure_sizes(edges.sizes.indexes);
 
        Tscal cmass = edges.central_mass.data;
        Tscal G     = edges.constant_G.data;
        Tvec cpos   = edges.central_pos.data;
 
        sham::distributed_data_kernel_call(
            shamsys::instance::get_compute_scheduler_ptr(),
            sham::DDMultiRef{edges.spans_positions.get_spans()},
            sham::DDMultiRef{edges.spans_accel_ext.get_spans()},
            edges.sizes.indexes,
            [mGM = -cmass * G, cpos](u32 gid, const Tvec *xyz, Tvec *axyz_ext) {
                Tvec r_a       = xyz[gid] - cpos;
                Tscal abs_ra   = sycl::length(r_a);
                Tscal abs_ra_3 = abs_ra * abs_ra * abs_ra;
                axyz_ext[gid] += mGM * r_a / abs_ra_3;
            });
    }
 
    template<class Tvec>
    inline std::string AddForceCentralGravPotential<Tvec>::_impl_get_tex() const {
        std::string tex = R"tex(
                 Add force (central gravitational potential)
 
                 \begin{align}
                 {spans_accel_ext}_i = -{constant_G} * {central_mass} * {central_pos}_i / {spans_positions}_i^3
                 \end{align}
             )tex";
 
        replace_edges_tex_symbols(tex);
 
        return tex;
    }
 
    template class shammodels::common::modules::AddForceCentralGravPotential<f64_3>;
 
} // namespace shammodels::common::modules