cl_FEM_DofMgr_BlockData.hpp

Go to the documentation of this file.

/*
 * BELFEM -- The Berkeley Lab Finite Element Framework
 * Copyright (c) 2026, The Regents of the University of California,
 * through Lawrence Berkeley National Laboratory (subject to receipt of any required
 * approvals from the U.S. Dept. of Energy).  All rights reserved.
 *
 * Developers: Christian Messe, Gregory Giard
 *
 * See the top-level LICENSE file for the complete license and disclaimer.
 */
#ifndef BELFEM_CL_FEM_DOFMGR_BLOCKDATA_HPP
#define BELFEM_CL_FEM_DOFMGR_BLOCKDATA_HPP
 
#include "typedefs.hpp"
#include "cl_Cell.hpp"
#include "cl_Map.hpp"
#include "cl_Vector.hpp"
#include "cl_IWG.hpp"
#include "cl_FEM_Block.hpp"
namespace belfem
{
    class Mesh ;
 
    namespace fem
    {
        class Kernel ;
        class DofManager ;
 
        namespace dofmgr
        {
            class BlockData
            {
                DofManager * mParent ;

                Kernel * mKernel ;

                Mesh * mMesh ;
 
                // my rank
                const proc_t mCommRank ;
                const proc_t mCommSize ;
 
                // map for fem blocks
                Map< id_t, Block * > mBlockMap ;
 
                // container for blocks
                Cell< Block * > mBlocks ;
 
                // the empty block is a dummy that is exposed if
                // a set is accessed that doesn't exist on this proc
                Block * mEmptyBlock = nullptr ;
 
//------------------------------------------------------------------------------
            public:
//------------------------------------------------------------------------------
 
                BlockData( DofManager * aParent ) ;
 
//------------------------------------------------------------------------------
 
                ~BlockData();
//------------------------------------------------------------------------------
 
                void
                create_blocks();
 
//------------------------------------------------------------------------------

                 Block *
                 block( const id_t aID );
 
//------------------------------------------------------------------------------

                Cell< Block * > &
                blocks();
 
//------------------------------------------------------------------------------

                bool
                block_exists( const id_t aID ) const ;
 
//------------------------------------------------------------------------------
 
                void
                reset();
 
//------------------------------------------------------------------------------

                static void
                selected_thin_shell_facet_ids(
                          Mesh                 * aMesh,
                    const Map< id_t, index_t > & aBlockIndexMap,
                          Vector< id_t >       & aFacetIDs );
 
//------------------------------------------------------------------------------
            private:
//------------------------------------------------------------------------------
 
                void
                collect_thin_shell_facet_ids(
                    const Map< id_t, index_t > & aBlockIndexMap,
                                Vector< id_t > & aAllSelectedThinShellFacets );
 
                void
                collect_element_indices(
                    const Map< id_t, index_t > & aBlockIndexMap,
                    const Map< id_t, index_t > & aSideSetIndexMap,
                    const Vector< id_t >       & aAllSelectedThinShellFacets,
                    Cell< Vector< index_t > >  & aOwnedElementIndices,
                    Cell< Vector< index_t > >  & aAuraElementIndices );
 
                void
                link_thin_shell_facets();
 
                void
                link_thin_shell_facets_serial();
 
                void
                link_thin_shell_facets_parallel();
 
//------------------------------------------------------------------------------
            };
 
//------------------------------------------------------------------------------
 
            inline Block *
            BlockData::block( const id_t aID )
            {
                if ( mBlockMap.key_exists( aID ) )
                {
                    return mBlockMap( aID );
                }
                else
                {
                    mEmptyBlock->set_mesh_id( aID );
                    return mEmptyBlock ;
                }
            }
 
//------------------------------------------------------------------------------
 
            inline Cell< Block * > &
            BlockData::blocks()
            {
                return mBlocks ;
            }
 
//------------------------------------------------------------------------------
 
            inline bool
            BlockData::block_exists( const id_t aID ) const
            {
                return mBlockMap.key_exists( aID ) ;
            }
 
//------------------------------------------------------------------------------
        } /* end namespace dofmgr */
    } /* end namespace fem */
} /* end namespace belfem */
 
#endif //BELFEM_CL_FEM_DOFMGR_BLOCKDATA_HPP