base_mesh.h

Go to the documentation of this file.

/* -----------------------------------------------------------------------------*
 *                               SPHinXsys                                      *
 * -----------------------------------------------------------------------------*
 * SPHinXsys (pronunciation: s'finksis) is an acronym from Smoothed Particle    *
 * Hydrodynamics for industrial compleX systems. It provides C++ APIs for       *
 * physical accurate simulation and aims to model coupled industrial dynamic    *
 * systems including fluid, solid, multi-body dynamics and beyond with SPH      *
 * (smoothed particle hydrodynamics), a meshless computational method using     *
 * particle discretization.                                                     *
 *                                                                              *
 * SPHinXsys is partially funded by German Research Foundation                  *
 * (Deutsche Forschungsgemeinschaft) DFG HU1527/6-1, HU1527/10-1,               *
 * HU1527/12-1 and HU1527/12-4.                                                 *
 *                                                                              *
 * Portions copyright (c) 2017-2022 Technical University of Munich and          *
 * the authors' affiliations.                                                   *
 *                                                                              *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may      *
 * not use this file except in compliance with the License. You may obtain a    *
 * copy of the License at http://www.apache.org/licenses/LICENSE-2.0.           *
 *                                                                              *
 * -----------------------------------------------------------------------------*/
#ifndef BASE_MESH_H
#define BASE_MESH_H

#include "base_data_package.h"
#include "sph_data_containers.h"
#include "my_memory_pool.h"

#include <fstream>
#include <algorithm>
#include <mutex>
#include <functional>
using namespace std::placeholders;

namespace SPH
{
    typedef std::function<void(const Vecu &, Real)> MeshFunctor;
    void MeshIterator(const Vecu &index_begin, const Vecu &index_end, MeshFunctor &mesh_functor, Real dt = 0.0);
    void MeshIterator_parallel(const Vecu &index_begin, const Vecu &index_end, MeshFunctor &mesh_functor, Real dt = 0.0);

    class BaseMesh
    {
    protected:
        Vecd mesh_lower_bound_;      
        Real grid_spacing_;          
        Vecu number_of_grid_points_; 
    public:
        BaseMesh();
        explicit BaseMesh(Vecu number_of_grid_points);
        BaseMesh(Vecd mesh_lower_bound, Real grid_spacing, Vecu number_of_grid_points);
        BaseMesh(BoundingBox tentative_bounds, Real grid_spacing, size_t buffer_width);
        virtual ~BaseMesh(){};

        Vecd MeshLowerBound() { return mesh_lower_bound_; };
        Real GridSpacing() { return grid_spacing_; };
        Vecu NumberOfGridPoints() { return number_of_grid_points_; };
        Vecu NumberOfGridPointsFromNumberOfCells(const Vecu &number_of_cells) { return number_of_cells + Vecu(1); };
        Vecu NumberOfCellsFromNumberOfGridPoints(const Vecu &number_of_grid_points) { return number_of_grid_points - Vecu(1); };
        Vecd GridPositionFromCellPosition(const Vecd &cell_position) { return cell_position - Vecd(0.5 * grid_spacing_); };

        Vecu CellIndexFromPosition(const Vecd &position);
        Vecd CellPositionFromIndex(const Vecu &cell_index);
        Vecd GridPositionFromIndex(const Vecu &grid_index);
        Vecu transfer1DtoMeshIndex(const Vecu &number_of_mesh_indexes, size_t i);
        size_t transferMeshIndexTo1D(const Vecu &number_of_mesh_indexes, const Vecu &mesh_index);
        size_t MortonCode(const size_t &i);
        size_t transferMeshIndexToMortonOrder(const Vecu &mesh_index);
    };

    class Mesh : public BaseMesh
    {
    protected:
        size_t buffer_width_;  
        Vecu number_of_cells_; 
        void copyMeshProperties(Mesh *another_mesh);

    public:
        Mesh(BoundingBox tentative_bounds, Real grid_spacing, size_t buffer_width);
        Mesh(Vecd mesh_lower_bound, Vecu number_of_cells, Real grid_spacing);
        virtual ~Mesh(){};

        Vecu NumberOfCells() { return number_of_cells_; };
        size_t MeshBufferSize() { return buffer_width_; };
        virtual Real DataSpacing() { return grid_spacing_; };
    };

    class BaseMeshField
    {
    protected:
        std::string name_;

    public:
        explicit BaseMeshField(const std::string &name) : name_(name){};
        virtual ~BaseMeshField(){};

        std::string Name() { return name_; };
        virtual void writeMeshFieldToPlt(std::ofstream &output_file) = 0;
    };

    template <class CoarseMeshType>
    class RefinedMesh : public CoarseMeshType
    {
    public:
        template <typename... Args>
        RefinedMesh(BoundingBox tentative_bounds, CoarseMeshType &coarse_mesh, Args &&...args)
            : CoarseMeshType(tentative_bounds, 0.5 * coarse_mesh.DataSpacing(), std::forward<Args>(args)...),
              coarse_mesh_(coarse_mesh){};
        virtual ~RefinedMesh(){};

    protected:
        CoarseMeshType &coarse_mesh_;
    };

    template <class MeshFieldType, class CoarsestMeshType, class RefinedMeshType>
    class MultilevelMesh : public MeshFieldType
    {
    private:
        UniquePtrKeepers<CoarsestMeshType> mesh_level_ptr_vector_keeper_;

    protected:
        size_t total_levels_; 
        StdVec<CoarsestMeshType *> mesh_levels_;

    public:
        template <typename... Args>
        MultilevelMesh(BoundingBox tentative_bounds, Real reference_spacing, size_t total_levels, Args &&...args)
            : MeshFieldType(std::forward<Args>(args)...), total_levels_(total_levels)
        {
            Real coarsest_spacing = reference_spacing;
            mesh_levels_.push_back(
                mesh_level_ptr_vector_keeper_
                    .template createPtr<CoarsestMeshType>(tentative_bounds, coarsest_spacing, std::forward<Args>(args)...));

            for (size_t level = 1; level != total_levels_; ++level)
            {
                mesh_levels_.push_back(
                    mesh_level_ptr_vector_keeper_
                        .template createPtr<RefinedMeshType>(tentative_bounds, *mesh_levels_.back(), std::forward<Args>(args)...));
            }
        };

        virtual ~MultilevelMesh(){};

        StdVec<CoarsestMeshType *> getMeshLevels() { return mesh_levels_; };

        void writeMeshFieldToPlt(std::ofstream &output_file) override
        {
            for (size_t l = 0; l != total_levels_; ++l)
            {
                mesh_levels_[l]->writeMeshFieldToPlt(output_file);
            }
        }
    };
}
#endif //BASE_MESH_H