Model.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 "shambase/aliases_float.hpp"
#include "shambase/exception.hpp"
#include "shambase/memory.hpp"
#include "shambase/stacktrace.hpp"
#include "shambase/string.hpp"
#include "shambackends/type_traits.hpp"
#include "shamcomm/logs.hpp"
#include "shammath/crystalLattice.hpp"
#include "shammath/sphkernels.hpp"
#include "shammodels/common/setup/generators.hpp"
#include "shammodels/sph/Model.hpp"
#include "shammodels/sph/SinkPartStruct.hpp"
#include "shammodels/sph/io/Phantom2Shamrock.hpp"
#include "shammodels/sph/io/PhantomDump.hpp"
#include "shammodels/sph/modules/ParticleReordering.hpp"
#include "shammodels/sph/sink_edges_helper.hpp"
#include "shamrock/patch/PatchDataLayer.hpp"
#include "shamrock/scheduler/DataInserterUtility.hpp"
#include "shamrock/scheduler/PatchScheduler.hpp"
#include "shamsys/NodeInstance.hpp"
#include "shamsys/legacy/log.hpp"
#include <functional>
#include <random>
#include <stdexcept>
#include <utility>
#include <vector>
 
template<class Tvec, template<class> class SPHKernel>
f64 shammodels::sph::Model<Tvec, SPHKernel>::evolve_once_time_expl(f64 t_curr, f64 dt_input) {
    auto tmp = solver.evolve_once_time_expl(t_curr, dt_input);
    solver.print_timestep_logs();
    return tmp;
}
 
template<class Tvec, template<class> class SPHKernel>
shammodels::sph::TimestepLog shammodels::sph::Model<Tvec, SPHKernel>::timestep() {
    return solver.evolve_once();
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::init() {
 
    if (solver.solver_config.scheduler_conf.split_load_value == 0) {
        throw shambase::make_except_with_loc<std::invalid_argument>(
            "Scheduler load value should be greater than 0");
    }
 
    solver.init_required_fields();
    ctx.init_sched(
        solver.solver_config.scheduler_conf.split_load_value,
        solver.solver_config.scheduler_conf.merge_load_value);
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    sched.add_root_patch();
 
    shamlog_debug_ln("Sys", "build local scheduler tables");
    sched.owned_patch_id = sched.patch_list.build_local();
    sched.patch_list.build_local_idx_map();
    sched.update_local_load_value([&](shamrock::patch::Patch p) {
        return sched.patch_data.owned_data.get(p.id_patch).get_obj_cnt();
    });
    solver.init_ghost_layout();
 
    solver.ensure_time_state_edges();
 
    ensure_sink_edges<Tvec>(sched.synchronized_data);
 
    // must be bone after time state edges are ensured (it will connect to it)
    solver.init_solver_graph();
}
 
REGISTER_IDATAEDGESERIALIZABLE(shamrock::solvergraph::IDataEdgeSerializable<f64>);
REGISTER_IDATAEDGESERIALIZABLE(shamrock::solvergraph::IDataEdgeSerializable<std::vector<f64>>);
REGISTER_IDATAEDGESERIALIZABLE(shamrock::solvergraph::IDataEdgeSerializable<std::vector<f64_3>>);
 
template<class Tvec, template<class> class SPHKernel>
u64 shammodels::sph::Model<Tvec, SPHKernel>::get_total_part_count() {
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
    return shamalgs::collective::allreduce_sum(sched.get_rank_count());
}
 
template<class Tvec, template<class> class SPHKernel>
f64 shammodels::sph::Model<Tvec, SPHKernel>::total_mass_to_part_mass(f64 totmass) {
    return totmass / get_total_part_count();
}
 
template<class Tvec, template<class> class SPHKernel>
auto shammodels::sph::Model<Tvec, SPHKernel>::get_closest_part_to(Tvec pos) -> Tvec {
    StackEntry stack_loc{};
 
    using namespace shamrock::patch;
 
    Tvec best_dr     = shambase::VectorProperties<Tvec>::get_max();
    Tscal best_dist2 = shambase::VectorProperties<Tscal>::get_max();
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    sched.for_each_patchdata_nonempty([&](const Patch, PatchDataLayer &pdat) {
        auto acc = pdat.get_field<Tvec>(0).get_buf().copy_to_stdvec();
 
        u32 cnt = pdat.get_obj_cnt();
 
        for (u32 i = 0; i < cnt; i++) {
            Tvec tmp    = acc[i];
            Tvec dr     = tmp - pos;
            Tscal dist2 = sycl::dot(dr, dr);
            if (dist2 < best_dist2) {
                best_dr    = dr;
                best_dist2 = dist2;
            }
        }
    });
 
    std::vector<Tvec> list_dr{};
    shamalgs::collective::vector_allgatherv(std::vector<Tvec>{best_dr}, list_dr, MPI_COMM_WORLD);
 
    // reset distances because if two rank find the same distance the return value won't be the same
    // this bug took me a whole day to fix, aaaaaaaaaaaaah !!!!!
    // maybe this should be moved somewhere else to prevent similar issues
    // TODO (in a year maybe XD )
    best_dr    = shambase::VectorProperties<Tvec>::get_max();
    best_dist2 = shambase::VectorProperties<Tscal>::get_max();
 
    for (Tvec tmp : list_dr) {
        Tvec dr     = tmp - pos;
        Tscal dist2 = sycl::dot(dr, dr);
        if (dist2 < best_dist2) {
            best_dr    = dr;
            best_dist2 = dist2;
        }
    }
 
    return pos + best_dr;
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::remap_positions(std::function<Tvec(Tvec)> map) {
    StackEntry stack_loc{};
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
    sched.for_each_patchdata_nonempty([&](const Patch, PatchDataLayer &pdat) {
        auto &xyz = pdat.get_field<Tvec>(0).get_buf();
        auto acc  = xyz.copy_to_stdvec();
 
        u32 cnt = pdat.get_obj_cnt();
 
        for (u32 i = 0; i < cnt; i++) {
            acc[i] = map(acc[i]);
        }
 
        xyz.copy_from_stdvec(acc);
    });
 
    modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
        .update_load_balancing();
    sched.scheduler_step(false, false);
 
    {
        StackEntry stack_loc{};
        SerialPatchTree<Tvec> sptree(
            sched.patch_tree, sched.get_sim_box().get_patch_transform<Tvec>());
        shamrock::ReattributeDataUtility reatrib(sched);
        sptree.attach_buf();
        reatrib.reatribute_patch_objects(sptree, "xyz");
        sched.check_patchdata_locality_correctness();
    }
 
    modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
        .update_load_balancing();
    sched.scheduler_step(true, true);
 
    {
        StackEntry stack_loc{};
        SerialPatchTree<Tvec> sptree(
            sched.patch_tree, sched.get_sim_box().get_patch_transform<Tvec>());
 
        shamrock::ReattributeDataUtility reatrib(sched);
        sptree.attach_buf();
        reatrib.reatribute_patch_objects(sptree, "xyz");
        sched.check_patchdata_locality_correctness();
    }
}
 
template<class Tvec>
inline void post_insert_data(PatchScheduler &sched) {
    StackEntry stack_loc{};
 
    // logger::raw_ln(sched.dump_status());
    sched.scheduler_step(false, false);
 
    /*
            if(shamcomm::world_rank() == 7){
                logger::raw_ln(sched.dump_status());
            }
    */
 
    auto [m, M] = sched.get_box_tranform<Tvec>();
 
    {
        StackEntry stack_loc{};
        SerialPatchTree<Tvec> sptree(
            sched.patch_tree, sched.get_sim_box().get_patch_transform<Tvec>());
        shamrock::ReattributeDataUtility reatrib(sched);
        sptree.attach_buf();
        reatrib.reatribute_patch_objects(sptree, "xyz");
        sched.check_patchdata_locality_correctness();
    }
 
    sched.scheduler_step(true, true);
 
    {
        StackEntry stack_loc{};
        SerialPatchTree<Tvec> sptree(
            sched.patch_tree, sched.get_sim_box().get_patch_transform<Tvec>());
 
        shamrock::ReattributeDataUtility reatrib(sched);
        sptree.attach_buf();
        reatrib.reatribute_patch_objects(sptree, "xyz");
        sched.check_patchdata_locality_correctness();
    }
 
    std::string log = "";
 
    using namespace shamrock::patch;
 
    u32 smallest_count = u32_max;
    u32 largest_count  = 0;
 
    sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
        u32 tmp        = pdat.get_obj_cnt();
        smallest_count = sham::min(tmp, smallest_count);
        largest_count  = sham::max(tmp, largest_count);
    });
 
    smallest_count = shamalgs::collective::allreduce_min(smallest_count);
    largest_count  = shamalgs::collective::allreduce_max(largest_count);
 
    if (shamcomm::world_rank() == 0) {
        logger::info_ln(
            "Model", "current particle counts : min = ", smallest_count, "max = ", largest_count);
    }
 
    // sched.for_each_local_patchdata([&](const Patch p, PatchData &pdat) {
    //     log += shambase::format(
    //         "\n    patch id={}, N={} particles", p.id_patch, pdat.get_obj_cnt());
    // });
    //
    // std::string log_gathered = "";
    // shamalgs::collective::gather_str(log, log_gathered);
    //
    // if (shamcomm::world_rank() == 0)
    //     logger::info_ln("Model", "current particle counts : ", log_gathered);
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::push_particle(
    std::vector<Tvec> &part_pos_insert,
    std::vector<Tscal> &part_hpart_insert,
    std::vector<Tscal> &part_u_insert) {
    StackEntry stack_loc{};
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    std::string log = "";
 
    sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
        PatchCoordTransform<Tvec> ptransf = sched.get_sim_box().get_patch_transform<Tvec>();
 
        shammath::CoordRange<Tvec> patch_coord = ptransf.to_obj_coord(p);
 
        std::vector<Tvec> vec_acc;
        std::vector<Tscal> hpart_acc;
        std::vector<Tscal> u_acc;
        for (u32 i = 0; i < part_pos_insert.size(); i++) {
            Tvec r  = part_pos_insert[i];
            Tscal u = part_u_insert[i];
            if (patch_coord.contain_pos(r)) {
                vec_acc.push_back(r);
                hpart_acc.push_back(part_hpart_insert[i]);
                u_acc.push_back(u);
            }
        }
 
        if (vec_acc.size() == 0) {
            return;
        }
 
        log += shambase::format(
            "\n  rank = {}  patch id={}, add N={} particles, coords = {} {}",
            shamcomm::world_rank(),
            p.id_patch,
            vec_acc.size(),
            patch_coord.lower,
            patch_coord.upper);
 
        PatchDataLayer tmp(sched.get_layout_ptr_old());
        tmp.resize(vec_acc.size());
        tmp.fields_raz();
 
        {
            u32 len = vec_acc.size();
            PatchDataField<Tvec> &f
                = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("xyz"));
            sycl::buffer<Tvec> buf(vec_acc.data(), len);
            f.override(buf, len);
        }
 
        {
            u32 len = vec_acc.size();
            PatchDataField<Tscal> &f
                = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("hpart"));
            sycl::buffer<Tscal> buf(hpart_acc.data(), len);
            f.override(buf, len);
        }
 
        {
            u32 len = u_acc.size();
            PatchDataField<Tscal> &f
                = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("uint"));
            sycl::buffer<Tscal> buf(u_acc.data(), len);
            f.override(buf, len);
        }
 
        pdat.insert_elements(tmp);
 
        sched.check_patchdata_locality_correctness();
 
        std::string log_gathered = "";
        shamalgs::collective::gather_str(log, log_gathered);
 
        if (shamcomm::world_rank() == 0) {
            logger::info_ln("Model", "Push particles : ", log_gathered);
        }
        log = "";
 
        modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .update_load_balancing();
 
        post_insert_data<Tvec>(sched);
    });
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::push_particle_mhd(
    std::vector<Tvec> &part_pos_insert,
    std::vector<Tscal> &part_hpart_insert,
    std::vector<Tscal> &part_u_insert,
    std::vector<Tvec> &part_B_on_rho_insert,
    std::vector<Tscal> &part_psi_on_ch_insert) {
    StackEntry stack_loc{};
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    std::string log = "";
 
    sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
        PatchCoordTransform<Tvec> ptransf = sched.get_sim_box().get_patch_transform<Tvec>();
 
        shammath::CoordRange<Tvec> patch_coord = ptransf.to_obj_coord(p);
 
        std::vector<Tvec> vec_acc;
        std::vector<Tscal> hpart_acc;
        std::vector<Tscal> u_acc;
        std::vector<Tvec> B_on_rho_acc;
        std::vector<Tscal> psi_on_ch_acc;
        for (u32 i = 0; i < part_pos_insert.size(); i++) {
            Tvec r  = part_pos_insert[i];
            Tscal u = part_u_insert[i];
            if (patch_coord.contain_pos(r)) {
                vec_acc.push_back(r);
                hpart_acc.push_back(part_hpart_insert[i]);
                u_acc.push_back(u);
                B_on_rho_acc.push_back(part_B_on_rho_insert[i]);
                psi_on_ch_acc.push_back(part_psi_on_ch_insert[i]);
            }
        }
 
        if (vec_acc.size() == 0) {
            return;
        }
 
        log += shambase::format(
            "\n  rank = {}  patch id={}, add N={} particles, coords = {} {}",
            shamcomm::world_rank(),
            p.id_patch,
            vec_acc.size(),
            patch_coord.lower,
            patch_coord.upper);
 
        PatchDataLayer tmp(sched.get_layout_ptr_old());
        tmp.resize(vec_acc.size());
        tmp.fields_raz();
 
        {
            u32 len = vec_acc.size();
            PatchDataField<Tvec> &f
                = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("xyz"));
            sycl::buffer<Tvec> buf(vec_acc.data(), len);
            f.override(buf, len);
        }
 
        {
            u32 len = vec_acc.size();
            PatchDataField<Tscal> &f
                = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("hpart"));
            sycl::buffer<Tscal> buf(hpart_acc.data(), len);
            f.override(buf, len);
        }
 
        {
            u32 len = u_acc.size();
            PatchDataField<Tscal> &f
                = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("uint"));
            sycl::buffer<Tscal> buf(u_acc.data(), len);
            f.override(buf, len);
        }
 
        {
            u32 len = vec_acc.size();
            PatchDataField<Tvec> &f
                = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("B/rho"));
            sycl::buffer<Tvec> buf(B_on_rho_acc.data(), len);
            f.override(buf, len);
        }
 
        {
            u32 len = vec_acc.size();
            PatchDataField<Tscal> &f
                = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("psi/ch"));
            sycl::buffer<Tscal> buf(psi_on_ch_acc.data(), len);
            f.override(buf, len);
        }
 
        pdat.insert_elements(tmp);
 
        sched.check_patchdata_locality_correctness();
 
        std::string log_gathered = "";
        shamalgs::collective::gather_str(log, log_gathered);
 
        if (shamcomm::world_rank() == 0) {
            logger::info_ln("Model", "Push particles MHD : ", log_gathered);
        }
        log = "";
 
        modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .update_load_balancing();
 
        post_insert_data<Tvec>(sched);
    });
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::add_cube_hcp_3d(
    Tscal dr, std::pair<Tvec, Tvec> _box) {
    shambase::Timer time_setup;
    time_setup.start();
 
    StackEntry stack_loc{};
 
    shammath::CoordRange<Tvec> box = _box;
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    using Lattice     = shammath::LatticeHCP<Tvec>;
    using LatticeIter = typename shammath::LatticeHCP<Tvec>::IteratorDiscontinuous;
 
    auto [idxs_min, idxs_max] = Lattice::get_box_index_bounds(dr, box.lower, box.upper);
 
    LatticeIter gen = LatticeIter(dr, idxs_min, idxs_max);
 
    u64 acc_count = 0;
 
    std::string log = "";
    while (!gen.is_done()) {
 
        // loc maximum count of insert part
        u64 loc_sum_ins_cnt = 0;
        // sum_node( loc_sum_ins_cnt )
        u64 max_loc_sum_ins_cnt = 0;
 
        do {
            std::vector<Tvec> to_ins = gen.next_n(sched.crit_patch_split * 2);
            acc_count += to_ins.size();
 
            sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
                PatchCoordTransform<Tvec> ptransf = sched.get_sim_box().get_patch_transform<Tvec>();
 
                shammath::CoordRange<Tvec> patch_coord = ptransf.to_obj_coord(p);
 
                std::vector<Tvec> vec_acc;
                for (Tvec r : to_ins) {
                    if (patch_coord.contain_pos(r)) {
                        vec_acc.push_back(r);
                    }
                }
 
                // update max insert_count
                loc_sum_ins_cnt += vec_acc.size();
 
                if (vec_acc.size() == 0) {
                    return;
                }
 
                log += shambase::format(
                    "\n  rank = {}  patch id={}, add N={} particles, coords = {} {}",
                    shamcomm::world_rank(),
                    p.id_patch,
                    vec_acc.size(),
                    patch_coord.lower,
                    patch_coord.upper);
 
                // reserve space to avoid allocating during copy
                pdat.reserve(vec_acc.size());
 
                PatchDataLayer tmp(sched.get_layout_ptr_old());
                tmp.resize(vec_acc.size());
                tmp.fields_raz();
 
                {
                    u32 len = vec_acc.size();
                    PatchDataField<Tvec> &f
                        = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("xyz"));
                    // sycl::buffer<Tvec> buf(vec_acc.data(), len);
                    f.override(vec_acc, len);
                }
 
                {
                    PatchDataField<Tscal> &f
                        = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("hpart"));
                    f.override(dr);
                }
 
                pdat.insert_elements(tmp);
            });
 
            max_loc_sum_ins_cnt = shamalgs::collective::allreduce_max(loc_sum_ins_cnt);
 
            if (shamcomm::world_rank() == 0) {
                logger::info_ln(
                    "Model",
                    "--> insertion loop : max loc insert count = ",
                    max_loc_sum_ins_cnt,
                    "sum =",
                    acc_count);
            }
        } while (!gen.is_done() && max_loc_sum_ins_cnt < sched.crit_patch_split * 8);
 
        sched.check_patchdata_locality_correctness();
 
        // if(logger::details::loglevel >= shamcomm::logs::log_info){
        //     std::string log_gathered = "";
        //     shamalgs::collective::gather_str(log, log_gathered);
        //
        //     if (shamcomm::world_rank() == 0) {
        //         shamlog_debug_ln("Model", "Push particles : ", log_gathered);
        //     }
        // }
        log = "";
 
        modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .update_load_balancing();
        post_insert_data<Tvec>(sched);
    }
 
    if (true) {
        modules::ParticleReordering<Tvec, u32, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .reorder_particles();
    }
 
    time_setup.stop();
    if (shamcomm::world_rank() == 0) {
        logger::info_ln("Model", "add_cube_hcp took :", time_setup.elapsed_sec(), "s");
    }
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::add_cube_hcp_3d_v2(
    Tscal dr, std::pair<Tvec, Tvec> _box) {
    shambase::Timer time_setup;
    time_setup.start();
    StackEntry stack_loc{};
 
    shammath::CoordRange<Tvec> box = _box;
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    using Lattice     = shammath::LatticeHCP<Tvec>;
    using LatticeIter = typename shammath::LatticeHCP<Tvec>::IteratorDiscontinuous;
 
    auto [idxs_min, idxs_max] = Lattice::get_box_index_bounds(dr, box.lower, box.upper);
 
    LatticeIter gen = LatticeIter(dr, idxs_min, idxs_max);
 
    shamrock::DataInserterUtility inserter(sched);
 
    auto push_current_data = [&](std::vector<Tvec> pos_data) {
        PatchDataLayer tmp(sched.get_layout_ptr_old());
        tmp.resize(pos_data.size());
        tmp.fields_raz();
 
        {
            u32 len = pos_data.size();
            PatchDataField<Tvec> &f
                = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("xyz"));
            // sycl::buffer<Tvec> buf(pos_data.data(), len);
            f.override(pos_data, len);
        }
 
        {
            PatchDataField<Tscal> &f
                = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("hpart"));
            f.override(dr);
        }
 
        inserter.push_patch_data<Tvec>(tmp, "xyz", sched.crit_patch_split * 8, [&]() {
            modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(
                ctx, solver.solver_config, solver.storage)
                .update_load_balancing();
        });
        pos_data.clear();
    };
 
    u32 insert_step = sched.crit_patch_split * 8;
 
    auto [bmin, bmax] = sched.patch_data.sim_box.get_bounding_box<Tvec>();
 
    auto has_pdat = [&]() {
        bool ret = false;
        sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
            ret = true;
        });
        return ret;
    };
 
    // Every MPI rank should be synchroneous on gen state
    while (!gen.is_done()) {
 
        u64 loc_gen_count = (has_pdat()) ? insert_step : 0;
 
        auto gen_info = shamalgs::collective::fetch_view(loc_gen_count);
 
        u64 skip_start = gen_info.head_offset;
        u64 gen_cnt    = loc_gen_count;
        u64 skip_end   = gen_info.total_byte_count - loc_gen_count - gen_info.head_offset;
 
        shamlog_debug_ln(
            "Gen",
            "generate : ",
            skip_start,
            gen_cnt,
            skip_end,
            "total",
            skip_start + gen_cnt + skip_end);
        gen.skip(skip_start);
        auto tmp = gen.next_n(gen_cnt);
        gen.skip(skip_end);
 
        std::vector<Tvec> pos_data;
        for (Tvec r : tmp) {
            if (Patch::is_in_patch_converted(r, bmin, bmax)) {
                pos_data.push_back(r);
            }
        }
 
        push_current_data(pos_data);
 
        shamlog_debug_ln("Gen", "gen.is_done()", gen.is_done());
    }
 
    time_setup.stop();
    if (shamcomm::world_rank() == 0) {
        logger::info_ln("Model", "add_cube_hcp took :", time_setup.elapsed_sec(), "s");
    }
}
 
template<class Tvec>
class BigDiscUtils {
    public:
    using Tscal = shambase::VecComponent<Tvec>;
    using Out   = generic::setup::generators::DiscOutput<Tscal>;
 
    class DiscIterator {
        bool done = false;
        Tvec center;
        Tscal central_mass;
        u64 Npart;
        Tscal r_in;
        Tscal r_out;
        Tscal disc_mass;
        Tscal p;
        Tscal H_r_in;
        Tscal q;
        Tscal G;
 
        u64 current_index;
 
        std::mt19937 eng;
 
        std::function<Tscal(Tscal)> sigma_profile;
        std::function<Tscal(Tscal)> cs_profile;
        std::function<Tscal(Tscal)> rot_profile;
        std::function<Tscal(Tscal)> vel_full_corr;
 
        public:
        DiscIterator(
            Tvec center,
            Tscal central_mass,
            u64 Npart,
            Tscal r_in,
            Tscal r_out,
            Tscal disc_mass,
            Tscal p,
            Tscal H_r_in,
            Tscal q,
            Tscal G,
            std::mt19937 eng,
            std::function<Tscal(Tscal)> sigma_profile,
            std::function<Tscal(Tscal)> cs_profile,
            std::function<Tscal(Tscal)> rot_profile)
            : current_index(0), Npart(Npart), center(center), central_mass(central_mass),
              r_in(r_in), r_out(r_out), disc_mass(disc_mass), p(p), H_r_in(H_r_in), q(q), G(G),
              eng(eng), sigma_profile(sigma_profile), cs_profile(cs_profile),
              rot_profile(rot_profile) {
 
            if (Npart == 0) {
                done = true;
            }
        }
 
        inline bool is_done() { return done; } // just to make sure the result is not tempered with
 
        inline generic::setup::generators::DiscOutput<Tscal> next() {
 
            constexpr Tscal _2pi = 2 * shambase::constants::pi<Tscal>;
 
            auto f_func = [&](Tscal r) {
                return r * sigma_profile(r);
            };
 
            Tscal fmax = f_func(r_out);
 
            auto find_r = [&]() {
                while (true) {
                    Tscal u2 = shamalgs::primitives::mock_value<Tscal>(eng, 0, fmax);
                    Tscal r  = shamalgs::primitives::mock_value<Tscal>(eng, r_in, r_out);
                    if (u2 < f_func(r)) {
                        return r;
                    }
                }
            };
 
            auto theta  = shamalgs::primitives::mock_value<Tscal>(eng, 0, _2pi);
            auto Gauss  = shamalgs::random::mock_gaussian<Tscal>(eng);
            Tscal aspin = 2.;
 
            Tscal r = find_r();
 
            Tscal vk    = rot_profile(r);
            Tscal cs    = cs_profile(r);
            Tscal sigma = sigma_profile(r);
 
            Tscal Omega_Kep = sycl::sqrt(G * central_mass / (r * r * r));
 
            // Tscal H_r = cs/vk;
            // Tscal H =  H_r * r;
            Tscal H = sycl::sqrt(2.) * 3. * cs
                      / Omega_Kep; // factor taken from phantom, to fasten thermalizing
 
            Tscal z = H * Gauss;
 
            auto pos = sycl::vec<Tscal, 3>{r * sycl::cos(theta), z, r * sycl::sin(theta)};
 
            auto etheta = sycl::vec<Tscal, 3>{-pos.z(), 0, pos.x()};
            etheta /= sycl::length(etheta);
 
            auto vel = vk * etheta;
 
            // Tscal rho = (sigma / (H * shambase::constants::pi2_sqrt<Tscal>))*
            //     sycl::exp(- z*z / (2*H*H));
 
            Tscal fs  = 1. - sycl::sqrt(r_in / r);
            Tscal rho = (sigma * fs) * sycl::exp(-z * z / (2 * H * H));
 
            Out out{pos, vel, cs, rho};
 
            // increase counter + check if finished
            current_index++;
            if (current_index == Npart) {
                done = true;
            }
 
            return out;
        }
 
        inline std::vector<Out> next_n(u32 nmax) {
            std::vector<Out> ret{};
            for (u32 i = 0; i < nmax; i++) {
                if (done) {
                    break;
                }
 
                ret.push_back(next());
            }
            return ret;
        }
    };
};
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::add_big_disc_3d(
    Tvec center,
    Tscal central_mass,
    u32 Npart,
    Tscal r_in,
    Tscal r_out,
    Tscal disc_mass,
    Tscal p,
    Tscal H_r_in,
    Tscal q,
    std::mt19937 eng) {
 
    Tscal eos_gamma;
    using Config              = SolverConfig;
    using SolverConfigEOS     = typename Config::EOSConfig;
    using SolverEOS_Adiabatic = typename SolverConfigEOS::Adiabatic;
    if (SolverEOS_Adiabatic *eos_config
        = std::get_if<SolverEOS_Adiabatic>(&solver.solver_config.eos_config.config)) {
 
        eos_gamma = eos_config->gamma;
 
    } else {
        // dirty hack for disc setup in locally isothermal
        eos_gamma = 2;
        // shambase::throw_unimplemented();
    }
 
    auto sigma_profile = [=](Tscal r) {
        // we setup with an adimensional mass since it is monte carlo
        constexpr Tscal sigma_0 = 1;
        return sigma_0 * sycl::pow(r / r_in, -p);
    };
 
    auto cs_law = [=](Tscal r) {
        return sycl::pow(r / r_in, -q);
    };
 
    auto kep_profile = [&](Tscal r) {
        Tscal G = solver.solver_config.get_constant_G();
        return sycl::sqrt(G * central_mass / r);
    };
 
    auto rot_profile = [&](Tscal r) -> Tscal {
        // carefull: needs r in cylindrical
        Tscal G       = solver.solver_config.get_constant_G();
        Tscal c       = solver.solver_config.get_constant_c();
        Tscal aspin   = 2.;
        Tscal term    = G * central_mass / r;
        Tscal term_fs = 1. - sycl::sqrt(r_in / r);
        Tscal term_pr
            = -sycl::pown(cs_law(r), 2) * (1.5 + p + q); // NO CORRECTION from fs term, bad response
        Tscal term_bh = 0.; //- (2. * aspin / sycl::pow(c, 3)) * sycl::pow(G * central_mass / r, 2);
        Tscal det     = sycl::pown(term_bh, 2) + 4. * (term + term_pr);
        Tscal Rg      = G * central_mass / sycl::pown(c, 2);
        Tscal vkep    = sqrt(G * central_mass / r);
 
        Tscal vphi = 0.5 * (term_bh + sycl::sqrt(det));
 
        return vphi;
    };
 
    auto cs_profile = [&](Tscal r) {
        Tscal cs_in = (H_r_in * r_in / r) * kep_profile(r_in); // H_r_in*rot_profile(r_in);
        return cs_law(r) * cs_in;
    };
 
    auto get_hfact = []() -> Tscal {
        return Kernel::hfactd;
    };
 
    auto int_rho_h = [&](Tscal h) -> Tscal {
        return shamrock::sph::rho_h(solver.solver_config.gpart_mass, h, Kernel::hfactd);
    };
 
    Tscal part_mass = disc_mass / Npart;
 
    shambase::Timer time_setup;
    time_setup.start();
 
    StackEntry stack_loc{};
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    using Out   = generic::setup::generators::DiscOutput<Tscal>;
    using DIter = typename BigDiscUtils<Tvec>::DiscIterator;
 
    Tscal G   = solver.solver_config.get_constant_G();
    DIter gen = DIter(
        center,
        central_mass,
        Npart,
        r_in,
        r_out,
        disc_mass,
        p,
        H_r_in,
        q,
        G,
        eng,
        sigma_profile,
        cs_profile,
        rot_profile);
 
    u64 acc_count = 0;
 
    std::string log = "";
    while (!gen.is_done()) {
 
        // loc maximum count of insert part
        u64 loc_sum_ins_cnt = 0;
        // sum_node( loc_sum_ins_cnt )
        u64 max_loc_sum_ins_cnt = 0;
 
        do {
            std::vector<Out> to_ins = gen.next_n(sched.crit_patch_split * 2);
            acc_count += to_ins.size();
 
            sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
                PatchCoordTransform<Tvec> ptransf = sched.get_sim_box().get_patch_transform<Tvec>();
 
                shammath::CoordRange<Tvec> patch_coord = ptransf.to_obj_coord(p);
 
                std::vector<Out> part_list;
                for (Out r : to_ins) {
                    if (patch_coord.contain_pos(r.pos)) {
                        // add all part to insert in a vector
                        part_list.push_back(r);
                    }
                }
 
                // update max insert_count
                loc_sum_ins_cnt += part_list.size();
 
                if (part_list.size() == 0) {
                    return;
                }
 
                log += shambase::format(
                    "\n  rank = {}  patch id={}, add N={} particles, coords = {} {}",
                    shamcomm::world_rank(),
                    p.id_patch,
                    part_list.size(),
                    patch_coord.lower,
                    patch_coord.upper);
 
                // extract the pos from part_list
                std::vector<Tvec> vec_pos;
                std::vector<Tvec> vec_vel;
                std::vector<Tscal> vec_u;
                std::vector<Tscal> vec_h;
                std::vector<Tscal> vec_cs;
 
                for (Out o : part_list) {
                    vec_pos.push_back(o.pos);
                    vec_vel.push_back(o.velocity);
                    vec_u.push_back(o.cs * o.cs / (/*solver.eos_gamma * */ (eos_gamma - 1)));
                    vec_h.push_back(shamrock::sph::h_rho(part_mass, o.rho * 0.1, Kernel::hfactd));
                    vec_cs.push_back(o.cs);
                }
 
                // reserve space to avoid allocating during copy
                pdat.reserve(vec_pos.size());
 
                PatchDataLayer tmp(sched.get_layout_ptr_old());
                tmp.resize(vec_pos.size());
                tmp.fields_raz();
 
                {
                    u32 len = vec_pos.size();
                    PatchDataField<Tvec> &f
                        = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("xyz"));
                    sycl::buffer<Tvec> buf(vec_pos.data(), len);
                    f.override(buf, len);
                }
 
                {
                    u32 len = vec_pos.size();
                    PatchDataField<Tscal> &f
                        = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("hpart"));
                    sycl::buffer<Tscal> buf(vec_h.data(), len);
                    f.override(buf, len);
                }
 
                {
                    u32 len = vec_pos.size();
                    PatchDataField<Tscal> &f
                        = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("uint"));
                    sycl::buffer<Tscal> buf(vec_u.data(), len);
                    f.override(buf, len);
                }
 
                if (solver.solver_config.is_eos_locally_isothermal()) {
                    u32 len = vec_pos.size();
                    PatchDataField<Tscal> &f
                        = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("soundspeed"));
                    sycl::buffer<Tscal> buf(vec_cs.data(), len);
                    f.override(buf, len);
                }
 
                {
                    u32 len = vec_pos.size();
                    PatchDataField<Tvec> &f
                        = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("vxyz"));
                    sycl::buffer<Tvec> buf(vec_vel.data(), len);
                    f.override(buf, len);
                }
 
                pdat.insert_elements(tmp);
            });
 
            max_loc_sum_ins_cnt = shamalgs::collective::allreduce_max(loc_sum_ins_cnt);
 
            if (shamcomm::world_rank() == 0) {
                logger::info_ln(
                    "Model",
                    "--> insertion loop : max loc insert count = ",
                    max_loc_sum_ins_cnt,
                    "sum =",
                    acc_count);
            }
        } while (!gen.is_done() && max_loc_sum_ins_cnt < sched.crit_patch_split * 8);
 
        sched.check_patchdata_locality_correctness();
 
        // if(logger::details::loglevel >= shamcomm::logs::log_info){
        //     std::string log_gathered = "";
        //     shamalgs::collective::gather_str(log, log_gathered);
        //
        //     if (shamcomm::world_rank() == 0) {
        //         shamlog_debug_ln("Model", "Push particles : ", log_gathered);
        //     }
        // }
        log = "";
 
        modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .update_load_balancing();
        post_insert_data<Tvec>(sched);
    }
 
    if (true) {
        modules::ParticleReordering<Tvec, u32, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .reorder_particles();
    }
 
    time_setup.stop();
    if (shamcomm::world_rank() == 0) {
        logger::info_ln("Model", "add_big_disc took :", time_setup.elapsed_sec(), "s");
    }
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::add_cube_fcc_3d(
    Tscal dr, std::pair<Tvec, Tvec> _box) {
    StackEntry stack_loc{};
 
    shammath::CoordRange<Tvec> box = _box;
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    std::string log = "";
 
    auto make_sliced = [&]() {
        std::vector<Tvec> vec_lst;
        generic::setup::generators::add_particles_fcc(
            dr,
            {box.lower, box.upper},
            [&](Tvec r) {
                return box.contain_pos(r);
            },
            [&](Tvec r, Tscal h) {
                vec_lst.push_back(r);
            });
 
        std::vector<std::vector<Tvec>> sliced_buf;
 
        u32 sz_buf = sched.crit_patch_split * 4;
 
        std::vector<Tvec> cur_buf;
        for (u32 i = 0; i < vec_lst.size(); i++) {
            cur_buf.push_back(vec_lst[i]);
 
            if (cur_buf.size() > sz_buf) {
                sliced_buf.push_back(std::exchange(cur_buf, std::vector<Tvec>{}));
            }
        }
 
        if (cur_buf.size() > 0) {
            sliced_buf.push_back(std::exchange(cur_buf, std::vector<Tvec>{}));
        }
 
        return sliced_buf;
    };
 
    std::vector<std::vector<Tvec>> sliced_buf = make_sliced();
 
    for (std::vector<Tvec> to_ins : sliced_buf) {
 
        sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
            PatchCoordTransform<Tvec> ptransf = sched.get_sim_box().get_patch_transform<Tvec>();
 
            shammath::CoordRange<Tvec> patch_coord = ptransf.to_obj_coord(p);
 
            std::vector<Tvec> vec_acc;
            for (Tvec r : to_ins) {
                if (patch_coord.contain_pos(r)) {
                    vec_acc.push_back(r);
                }
            }
 
            if (vec_acc.size() == 0) {
                return;
            }
 
            log += shambase::format(
                "\n  rank = {}  patch id={}, add N={} particles, coords = {} {}",
                shamcomm::world_rank(),
                p.id_patch,
                vec_acc.size(),
                patch_coord.lower,
                patch_coord.upper);
 
            PatchDataLayer tmp(sched.get_layout_ptr_old());
            tmp.resize(vec_acc.size());
            tmp.fields_raz();
 
            {
                u32 len = vec_acc.size();
                PatchDataField<Tvec> &f
                    = tmp.get_field<Tvec>(sched.pdl_old().get_field_idx<Tvec>("xyz"));
                sycl::buffer<Tvec> buf(vec_acc.data(), len);
                f.override(buf, len);
            }
 
            {
                PatchDataField<Tscal> &f
                    = tmp.get_field<Tscal>(sched.pdl_old().get_field_idx<Tscal>("hpart"));
                f.override(dr);
            }
 
            pdat.insert_elements(tmp);
        });
 
        sched.check_patchdata_locality_correctness();
 
        std::string log_gathered = "";
        shamalgs::collective::gather_str(log, log_gathered);
 
        if (shamcomm::world_rank() == 0) {
            logger::info_ln("Model", "Push particles : ", log_gathered);
        }
        log = "";
 
        modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .update_load_balancing();
        post_insert_data<Tvec>(sched);
    }
}
 
template<class Tvec, template<class> class SPHKernel>
auto shammodels::sph::Model<Tvec, SPHKernel>::gen_config_from_phantom_dump(
    PhantomDump &phdump, bool bypass_error) -> SolverConfig {
    StackEntry stack_loc{};
    SolverConfig conf{};
 
    auto massoftype = phdump.read_header_floats<Tscal>("massoftype");
 
    conf.gpart_mass           = massoftype[0];
    conf.cfl_config.cfl_cour  = phdump.read_header_float<Tscal>("C_cour");
    conf.cfl_config.cfl_force = phdump.read_header_float<Tscal>("C_force");
 
    conf.eos_config      = get_shamrock_eosconfig<Tvec>(phdump, bypass_error);
    conf.artif_viscosity = get_shamrock_avconfig<Tvec>(phdump);
 
    conf.set_units(get_shamrock_units<Tscal>(phdump));
 
    conf.boundary_config = get_shamrock_boundary_config<Tvec>(phdump);
 
    return conf;
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::init_from_phantom_dump(
    PhantomDump &phdump, Tscal hpart_fact_load) {
    StackEntry stack_loc{};
 
    bool has_coord_in_header = true;
 
    Tscal xmin, xmax, ymin, ymax, zmin, zmax;
    has_coord_in_header = phdump.has_header_entry("xmin");
 
    std::string log = "";
 
    std::vector<Tvec> xyz, vxyz;
    std::vector<Tscal> h, u, alpha;
 
    {
        std::vector<Tscal> x, y, z, vx, vy, vz;
 
        phdump.blocks[0].fill_vec("x", x);
        phdump.blocks[0].fill_vec("y", y);
        phdump.blocks[0].fill_vec("z", z);
 
        if (has_coord_in_header) {
            xmin = phdump.read_header_float<f64>("xmin");
            xmax = phdump.read_header_float<f64>("xmax");
            ymin = phdump.read_header_float<f64>("ymin");
            ymax = phdump.read_header_float<f64>("ymax");
            zmin = phdump.read_header_float<f64>("zmin");
            zmax = phdump.read_header_float<f64>("zmax");
 
            resize_simulation_box({{xmin, ymin, zmin}, {xmax, ymax, zmax}});
        } else {
            Tscal box_tolerance = 1.2;
 
            xmin = *std::min_element(x.begin(), x.end());
            xmax = *std::max_element(x.begin(), x.end());
            ymin = *std::min_element(y.begin(), y.end());
            ymax = *std::max_element(y.begin(), y.end());
            zmin = *std::min_element(z.begin(), z.end());
            zmax = *std::max_element(z.begin(), z.end());
 
            Tvec bm = {xmin, ymin, zmin};
            Tvec bM = {xmax, ymax, zmax};
 
            Tvec center = (bm + bM) * 0.5;
 
            Tvec d = (bM - bm) * 0.5;
 
            // expand the box
            d *= box_tolerance;
 
            resize_simulation_box({center - d, center + d});
        }
 
        phdump.blocks[0].fill_vec("h", h);
 
        phdump.blocks[0].fill_vec("vx", vx);
        phdump.blocks[0].fill_vec("vy", vy);
        phdump.blocks[0].fill_vec("vz", vz);
 
        phdump.blocks[0].fill_vec("u", u);
        phdump.blocks[0].fill_vec("alpha", alpha);
 
        for (u32 i = 0; i < x.size(); i++) {
            xyz.push_back({x[i], y[i], z[i]});
        }
        for (u32 i = 0; i < vx.size(); i++) {
            vxyz.push_back({vx[i], vy[i], vz[i]});
        }
    }
 
    // Load time infos
    f64 time_phdump = phdump.read_header_float<f64>("time");
    solver.set_time(time_phdump);
 
    using namespace shamrock::patch;
 
    PatchScheduler &sched = shambase::get_check_ref(ctx.sched);
 
    u32 sz_buf = sched.crit_patch_split * 4;
 
    u32 Ntot = xyz.size();
 
    std::vector<u64> insert_ranges;
    insert_ranges.push_back(0);
    for (u64 i = sz_buf; i < Ntot; i += sz_buf) {
        insert_ranges.push_back(i);
    }
    insert_ranges.push_back(Ntot);
 
    for (u64 krange = 0; krange < insert_ranges.size() - 1; krange++) {
        u64 start_id = insert_ranges[krange];
        u64 end_id   = insert_ranges[krange + 1];
 
        u64 Nloc = end_id - start_id;
 
        sched.for_each_local_patchdata([&](const Patch &p, PatchDataLayer &pdat) {
            PatchCoordTransform<Tvec> ptransf = sched.get_sim_box().get_patch_transform<Tvec>();
 
            shammath::CoordRange<Tvec> patch_coord = ptransf.to_obj_coord(p);
 
            std::vector<u64> sel_index;
            for (u64 i = start_id; i < end_id; i++) {
                Tvec r   = xyz[i];
                Tscal h_ = h[i];
                if (patch_coord.contain_pos(r) && (h_ >= 0)) {
                    sel_index.push_back(i);
                }
            }
 
            if (sel_index.size() == 0) {
                return;
            }
 
            log += shambase::format(
                "\n  rank = {}  patch id={}, add N={} particles, coords = {} {}",
                shamcomm::world_rank(),
                p.id_patch,
                sel_index.size(),
                patch_coord.lower,
                patch_coord.upper);
 
            std::vector<Tvec> ins_xyz, ins_vxyz;
            std::vector<Tscal> ins_h, ins_u, ins_alpha;
            for (u64 i : sel_index) {
                ins_xyz.push_back(xyz[i]);
            }
            for (u64 i : sel_index) {
                ins_vxyz.push_back(vxyz[i]);
            }
            for (u64 i : sel_index) {
                ins_h.push_back(h[i] * hpart_fact_load);
            }
            if (u.size() > 0) {
                for (u64 i : sel_index) {
                    ins_u.push_back(u[i]);
                }
            }
            if (alpha.size() > 0) {
                for (u64 i : sel_index) {
                    ins_alpha.push_back(alpha[i]);
                }
            }
 
            PatchDataLayer ptmp(sched.get_layout_ptr_old());
            ptmp.resize(sel_index.size());
            ptmp.fields_raz();
 
            ptmp.override_patch_field("xyz", ins_xyz);
            ptmp.override_patch_field("vxyz", ins_vxyz);
            ptmp.override_patch_field("hpart", ins_h);
 
            if (ins_alpha.size() > 0) {
                ptmp.override_patch_field("alpha_AV", ins_alpha);
            }
 
            if (ins_u.size() > 0) {
                ptmp.override_patch_field("uint", ins_u);
            }
 
            pdat.insert_elements(ptmp);
        });
 
        sched.check_patchdata_locality_correctness();
 
        std::string log_gathered = "";
        shamalgs::collective::gather_str(log, log_gathered);
 
        if (shamcomm::world_rank() == 0) {
            logger::info_ln("Model", "Push particles : ", log_gathered);
        }
        log = "";
 
        modules::ComputeLoadBalanceValue<Tvec, SPHKernel>(ctx, solver.solver_config, solver.storage)
            .update_load_balancing();
 
        post_insert_data<Tvec>(sched);
 
        // add sinks
 
        PhantomDumpBlock &sink_block = phdump.blocks[1];
        {
            std::vector<Tscal> xsink, ysink, zsink;
            std::vector<Tscal> vxsink, vysink, vzsink;
            std::vector<Tscal> mass;
            std::vector<Tscal> Racc;
 
            sink_block.fill_vec("x", xsink);
            sink_block.fill_vec("y", ysink);
            sink_block.fill_vec("z", zsink);
            sink_block.fill_vec("vx", vxsink);
            sink_block.fill_vec("vy", vysink);
            sink_block.fill_vec("vz", vzsink);
            sink_block.fill_vec("m", mass);
            sink_block.fill_vec("h", Racc);
 
            for (u32 i = 0; i < xsink.size(); i++) {
                add_sink(
                    mass[i],
                    {xsink[i], ysink[i], zsink[i]},
                    {vxsink[i], vysink[i], vzsink[i]},
                    Racc[i]);
            }
        }
    }
}
 
template<class Tvec, template<class> class SPHKernel>
void shammodels::sph::Model<Tvec, SPHKernel>::add_pdat_to_phantom_block(
    PhantomDumpBlock &block, shamrock::patch::PatchDataLayer &pdat) {
 
    std::vector<Tvec> xyz = pdat.fetch_data<Tvec>("xyz");
 
    u64 xid = block.get_ref_fort_real("x");
    u64 yid = block.get_ref_fort_real("y");
    u64 zid = block.get_ref_fort_real("z");
 
    for (auto vec : xyz) {
        block.blocks_fort_real[xid].vals.push_back(vec.x());
        block.blocks_fort_real[yid].vals.push_back(vec.y());
        block.blocks_fort_real[zid].vals.push_back(vec.z());
    }
 
    std::vector<Tscal> h = pdat.fetch_data<Tscal>("hpart");
    u64 hid              = block.get_ref_f32("h");
    for (auto h_ : h) {
        block.blocks_f32[hid].vals.push_back(h_);
    }
 
    if (solver.solver_config.has_field_alphaAV()) {
        std::vector<Tscal> alpha = pdat.fetch_data<Tscal>("alpha_AV");
        u64 aid                  = block.get_ref_f32("alpha");
        for (auto alp_ : alpha) {
            block.blocks_f32[aid].vals.push_back(alp_);
        }
    }
 
    if (solver.solver_config.has_field_divv()) {
        std::vector<Tscal> vecdivv = pdat.fetch_data<Tscal>("divv");
        u64 divvid                 = block.get_ref_f32("divv");
        for (auto d_ : vecdivv) {
            block.blocks_f32[divvid].vals.push_back(d_);
        }
    }
 
    std::vector<Tvec> vxyz = pdat.fetch_data<Tvec>("vxyz");
 
    u64 vxid = block.get_ref_fort_real("vx");
    u64 vyid = block.get_ref_fort_real("vy");
    u64 vzid = block.get_ref_fort_real("vz");
 
    for (auto vec : vxyz) {
        block.blocks_fort_real[vxid].vals.push_back(vec.x());
        block.blocks_fort_real[vyid].vals.push_back(vec.y());
        block.blocks_fort_real[vzid].vals.push_back(vec.z());
    }
 
    std::vector<Tscal> u = pdat.fetch_data<Tscal>("uint");
    u64 uid              = block.get_ref_fort_real("u");
    for (auto u_ : u) {
        block.blocks_fort_real[uid].vals.push_back(u_);
    }
 
    block.tot_count = block.blocks_fort_real[xid].vals.size();
}
 
template<class Tvec, template<class> class SPHKernel>
shammodels::sph::PhantomDump shammodels::sph::Model<Tvec, SPHKernel>::make_phantom_dump() {
    StackEntry stack_loc{};
 
    PhantomDump dump;
 
    bool bypass_error_check = false;
 
    auto get_sink_count = [&]() -> int {
        return int(get_sink_pos<Tvec>(shambase::get_check_ref(ctx.sched).synchronized_data).size());
    };
 
    dump.override_magic_number();
    dump.iversion = 1;
    dump.fileid   = shambase::format("{:100s}", "FT:Phantom Shamrock writer");
 
    u32 Ntot = get_total_part_count();
    dump.table_header_fort_int.add("nparttot", Ntot);
    dump.table_header_fort_int.add("ntypes", 8);
    dump.table_header_fort_int.add("npartoftype", Ntot);
    dump.table_header_fort_int.add("npartoftype", 0);
    dump.table_header_fort_int.add("npartoftype", 0);
    dump.table_header_fort_int.add("npartoftype", 0);
    dump.table_header_fort_int.add("npartoftype", 0);
    dump.table_header_fort_int.add("npartoftype", 0);
    dump.table_header_fort_int.add("npartoftype", 0);
    dump.table_header_fort_int.add("npartoftype", 0);
 
    dump.table_header_i64.add("nparttot", Ntot);
    dump.table_header_i64.add("ntypes", 8);
    dump.table_header_i64.add("npartoftype", Ntot);
    dump.table_header_i64.add("npartoftype", 0);
    dump.table_header_i64.add("npartoftype", 0);
    dump.table_header_i64.add("npartoftype", 0);
    dump.table_header_i64.add("npartoftype", 0);
    dump.table_header_i64.add("npartoftype", 0);
    dump.table_header_i64.add("npartoftype", 0);
    dump.table_header_i64.add("npartoftype", 0);
 
    dump.table_header_fort_int.add("nblocks", 1);
    dump.table_header_fort_int.add("nptmass", get_sink_count());
    dump.table_header_fort_int.add("ndustlarge", 0);
    dump.table_header_fort_int.add("ndustsmall", 0);
    dump.table_header_fort_int.add("idust", 7);
    dump.table_header_fort_int.add("idtmax_n", 1);
    dump.table_header_fort_int.add("idtmax_frac", 0);
    dump.table_header_fort_int.add("idumpfile", 0);
    dump.table_header_fort_int.add("majorv", 2023);
    dump.table_header_fort_int.add("minorv", 0);
    dump.table_header_fort_int.add("microv", 0);
    dump.table_header_fort_int.add("isink", 0);
 
    dump.table_header_i32.add("iexternalforce", 0);
 
    write_shamrock_eos_in_phantom_dump(solver.solver_config.eos_config, dump, bypass_error_check);
 
    dump.table_header_fort_real.add("time", solver.get_time());
    dump.table_header_fort_real.add("dtmax", solver.get_dt_sph());
 
    dump.table_header_fort_real.add("rhozero", 0);
    dump.table_header_fort_real.add("hfact", Kernel::hfactd);
    dump.table_header_fort_real.add("tolh", 0.0001);
    dump.table_header_fort_real.add("C_cour", solver.solver_config.cfl_config.cfl_cour);
    dump.table_header_fort_real.add("C_force", solver.solver_config.cfl_config.cfl_force);
    dump.table_header_fort_real.add("alpha", 0);
    dump.table_header_fort_real.add("alphau", 1);
    dump.table_header_fort_real.add("alphaB", 1);
 
    dump.table_header_fort_real.add("massoftype", solver.solver_config.gpart_mass);
    dump.table_header_fort_real.add("massoftype", 0);
    dump.table_header_fort_real.add("massoftype", 0);
    dump.table_header_fort_real.add("massoftype", 0);
    dump.table_header_fort_real.add("massoftype", 0);
    dump.table_header_fort_real.add("massoftype", 0);
    dump.table_header_fort_real.add("massoftype", 0);
    dump.table_header_fort_real.add("massoftype", 0);
 
    dump.table_header_fort_real.add("Bextx", 0);
    dump.table_header_fort_real.add("Bexty", 0);
    dump.table_header_fort_real.add("Bextz", 0);
    dump.table_header_fort_real.add("dum", 0);
 
    PatchScheduler &sched = shambase::get_check_ref(solver.context.sched);
 
    auto box_size = sched.get_box_volume<Tvec>();
 
    write_shamrock_boundaries_in_phantom_dump(
        solver.solver_config.boundary_config, box_size, dump, bypass_error_check);
 
    dump.table_header_fort_real.add("get_conserv", -1);
    dump.table_header_fort_real.add("etot_in", 0.59762);
    dump.table_header_fort_real.add("angtot_in", 0.0189694);
    dump.table_header_fort_real.add("totmom_in", 0.0306284);
 
    write_shamrock_units_in_phantom_dump(solver.solver_config.unit_sys, dump, bypass_error_check);
 
    PhantomDumpBlock block_part;
 
    {
        NamedStackEntry stack_loc{"gather data"};
        std::vector<std::unique_ptr<shamrock::patch::PatchDataLayer>> gathered
            = ctx.allgather_data();
 
        for (auto &dat : gathered) {
            add_pdat_to_phantom_block(block_part, shambase::get_check_ref(dat));
        }
    }
 
    dump.blocks.push_back(std::move(block_part));
 
    {
        auto &sync = shambase::get_check_ref(ctx.sched).synchronized_data;
        auto edges = get_sink_edges<Tvec>(sync);
        if (edges.has_sinks()) {
            auto sinks = to_sink_particles(edges);
            // add sinks to block 1
            PhantomDumpBlock sink_block;
 
            u64 xid  = sink_block.get_ref_fort_real("x");
            u64 yid  = sink_block.get_ref_fort_real("y");
            u64 zid  = sink_block.get_ref_fort_real("z");
            u64 mid  = sink_block.get_ref_fort_real("m");
            u64 hid  = sink_block.get_ref_fort_real("h");
            u64 vxid = sink_block.get_ref_fort_real("vx");
            u64 vyid = sink_block.get_ref_fort_real("vy");
            u64 vzid = sink_block.get_ref_fort_real("vz");
 
            for (SinkParticle<Tvec> s : sinks) {
                sink_block.blocks_fort_real[xid].vals.push_back(s.pos.x());
                sink_block.blocks_fort_real[yid].vals.push_back(s.pos.y());
                sink_block.blocks_fort_real[zid].vals.push_back(s.pos.z());
                sink_block.blocks_fort_real[mid].vals.push_back(s.mass);
                sink_block.blocks_fort_real[hid].vals.push_back(s.accretion_radius);
                sink_block.blocks_fort_real[vxid].vals.push_back(s.velocity.x());
                sink_block.blocks_fort_real[vyid].vals.push_back(s.velocity.y());
                sink_block.blocks_fort_real[vzid].vals.push_back(s.velocity.z());
            }
 
            sink_block.tot_count = sinks.size();
 
            dump.blocks.push_back(std::move(sink_block));
        }
    }
 
    return dump;
}
 
using namespace shammath;
 
template class shammodels::sph::Model<f64_3, M4>;
template class shammodels::sph::Model<f64_3, M6>;
template class shammodels::sph::Model<f64_3, M8>;
 
template class shammodels::sph::Model<f64_3, C2>;
template class shammodels::sph::Model<f64_3, C4>;
template class shammodels::sph::Model<f64_3, C6>;