SolverStorage.hpp

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
//
// -------------------------------------------------------//
 
#pragma once

 
#include "shambase/StorageComponent.hpp"
#include "shambase/stacktrace.hpp"
#include "shambackends/vec.hpp"
#include "shammodels/gsph/modules/GSPHGhostHandler.hpp"
#include "shammodels/gsph/solvergraph/GhostHandlerEdge.hpp"
#include "shammodels/sph/solvergraph/NeighCache.hpp"
#include "shamrock/scheduler/SerialPatchTree.hpp"
#include "shamrock/scheduler/ShamrockCtx.hpp"
#include "shamrock/solvergraph/Field.hpp"
#include "shamrock/solvergraph/FieldRefs.hpp"
#include "shamrock/solvergraph/Indexes.hpp"
#include "shamrock/solvergraph/RankGetter.hpp"
#include "shamrock/solvergraph/ScalarsEdge.hpp"
#include "shamrock/solvergraph/SolverGraph.hpp"
#include "shamsys/legacy/log.hpp"
#include "shamtree/CompressedLeafBVH.hpp"
#include "shamtree/KarrasRadixTreeField.hpp"
#include "shamtree/RadixTree.hpp"
#include "shamtree/TreeTraversalCache.hpp"
#include <memory>
 
namespace shammodels::gsph {
 
    template<class T>
    using Component = shambase::StorageComponent<T>;

    template<class Tvec, class Tmorton>
    struct SolverStorage {
        using Tscal              = shambase::VecComponent<Tvec>;
        static constexpr u32 dim = shambase::VectorProperties<Tvec>::dimension;
 
        // Use GSPH ghost handler with Newtonian field names
        using GhostHandle      = gsph::GSPHGhostHandler<Tvec>;
        using GhostHandleCache = typename GhostHandle::CacheMap;
 
        using RTree = shamtree::CompressedLeafBVH<Tmorton, Tvec, 3>;
 
        shamrock::solvergraph::SolverGraph solver_graph;

        std::shared_ptr<shamrock::solvergraph::Indexes<u32>> part_counts;
        std::shared_ptr<shamrock::solvergraph::Indexes<u32>> part_counts_with_ghost;

        std::shared_ptr<shamrock::solvergraph::FieldRefs<Tvec>> positions_with_ghosts;
        std::shared_ptr<shamrock::solvergraph::FieldRefs<Tscal>> hpart_with_ghosts;

        std::shared_ptr<shammodels::sph::solvergraph::NeighCache> neigh_cache;

        std::shared_ptr<shamrock::solvergraph::RankGetter> patch_rank_owner;

        Component<SerialPatchTree<Tvec>> serial_patch_tree;

        std::shared_ptr<solvergraph::GhostHandlerEdge<Tvec>> ghost_handler;
        Component<GhostHandleCache> ghost_patch_cache;

        Component<shambase::DistributedData<shamrock::patch::PatchDataLayer>> merged_xyzh;

        Component<shambase::DistributedData<RTree>> merged_pos_trees;
        Component<shambase::DistributedData<shamtree::KarrasRadixTreeField<Tscal>>>
            rtree_rint_field;

        std::shared_ptr<shamrock::solvergraph::Field<Tscal>> omega;

        std::shared_ptr<shamrock::patch::PatchDataLayerLayout> xyzh_ghost_layout;
        std::shared_ptr<shamrock::patch::PatchDataLayerLayout> ghost_layout;
        Component<shambase::DistributedData<shamrock::patch::PatchDataLayer>>
            merged_patchdata_ghost;

        std::shared_ptr<shamrock::solvergraph::Field<Tscal>> density;

        std::shared_ptr<shamrock::solvergraph::Field<Tscal>> pressure;
        std::shared_ptr<shamrock::solvergraph::Field<Tscal>> soundspeed;

        std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_density;  
        std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_pressure; 
        std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_vx;       
        std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_vy;       
        std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_vz;       

        Tscal h_per_cs_min = std::numeric_limits<Tscal>::max();

        Component<shamrock::ComputeField<Tvec>> old_axyz;
        Component<shamrock::ComputeField<Tscal>> old_duint;

        struct Timings {
            f64 interface = 0;
            f64 neighbors = 0;
            f64 io        = 0;
            f64 riemann   = 0; 
 
            void reset() { *this = {}; }
        } timings_details;
    };
 
} // namespace shammodels::gsph