interface_dof_handler.h

Go to the documentation of this file.

/*
 * MAST: Multidisciplinary-design Adaptation and Sensitivity Toolkit
 * Copyright (C) 2013-2019  Manav Bhatia
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#ifndef __mast_level_set_interface_dof_handler_h__
#define __mast_level_set_interface_dof_handler_h__


// MAST includes
#include "base/mast_data_types.h"


// libMesh includes
#include "libmesh/elem.h"
#include "libmesh/numeric_vector.h"


namespace MAST {
    
    // Forward declerations
    class SystemInitialization;
    template <typename ValType> class FieldFunction;
    class LevelSetIntersection;
    class NonlinearImplicitAssemblyElemOperations;
    
    class LevelSetInterfaceDofHandler {
        
    public:
        
        LevelSetInterfaceDofHandler();
        
        virtual ~LevelSetInterfaceDofHandler();
        
        void init(const MAST::SystemInitialization& sys_init,
                  MAST::LevelSetIntersection& intersection,
                  MAST::FieldFunction<Real>& phi);

        MAST::FieldFunction<Real>&
        get_level_set_function() {
            libmesh_assert(_phi);
            return *_phi;
        }
        
        bool if_factor_element(const libMesh::Elem& elem) const;
        
        void partition_local_elem_rows(const libMesh::Elem& elem,
                                       std::vector<libMesh::dof_id_type>& material_dofs,
                                       std::vector<libMesh::dof_id_type>& void_dofs);

        
        void partition_global_elem_rows(const libMesh::Elem& elem,
                                        std::vector<libMesh::dof_id_type>& material_dofs,
                                        std::vector<libMesh::dof_id_type>& void_dofs);


        void solution_of_factored_element(const libMesh::Elem& elem,
                                          RealVectorX& elem_sol);
        

        void element_factored_jacobian(const libMesh::Elem& elem,
                                       const RealMatrixX& jac,
                                       std::vector<libMesh::dof_id_type>& material_dof_ids,
                                       RealMatrixX& jac_factored_uu);

        void
        element_factored_residual_and_jacobian(const libMesh::Elem& elem,
                                               const RealMatrixX& jac,
                                               const RealVectorX& res,
                                               std::vector<libMesh::dof_id_type>& material_dof_ids,
                                               RealMatrixX& jac_factored_uu,
                                               RealVectorX& res_factored_u);

        void element_factored_jacobian(const libMesh::Elem& elem,
                                       const RealMatrixX& jac,
                                       std::vector<libMesh::dof_id_type>& material_dof_ids,
                                       std::vector<libMesh::dof_id_type>& void_dof_ids,
                                       RealMatrixX& jac_uu,
                                       RealMatrixX& jac_uf,
                                       RealMatrixX& jac_fu,
                                       RealMatrixX& jac_ff,
                                       RealMatrixX& jac_factored_uu);

        void update_factored_element_solution(const libMesh::Elem& elem,
                                              const RealMatrixX& res,
                                              const RealMatrixX& jac,
                                              const RealMatrixX& sol,
                                              const RealMatrixX& dsol,
                                              RealVectorX& updated_sol);

        
    protected:

        void _compute_matrix_inverse(const RealMatrixX& mat,
                                     RealMatrixX& mat_inv);
        
        const MAST::SystemInitialization*           _sys_init;
        MAST::FieldFunction<Real>*                  _phi;
        
        std::map<const libMesh::Elem*, RealVectorX> _elem_sol;
        
        std::map<const libMesh::Elem*, std::set<const libMesh::Node*>> _elem_void_nodes;

        std::map<const libMesh::Node*, std::set<const libMesh::Elem*>> _void_node_elems;
        
        std::map<const libMesh::Elem*, std::map<libMesh::dof_id_type, libMesh::dof_id_type>> _dof_ids;
    };
}


#endif // __mast_level_set_interface_dof_handler_h__