nastran_io.h

Go to the documentation of this file.

/*
 * MAST: Multidisciplinary-design Adaptation and Sensitivity Toolkit
 * Copyright (C) 2013-2020  Manav Bhatia and MAST authors
 *
 * 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_nastran_io_h__
#define __mast_nastran_io_h__

// C++ includes.
#include <iostream>
#include <map>
#include <list>

// libMesh includes.
#include <libmesh/libmesh_common.h>
#include <libmesh/mesh_input.h>
#include <libmesh/elem.h>

#include "base/mast_config.h"

// Python includes.
#include <Python.h>

// MAST includes.
#include "mesh/nastran_io.h"
#include "mesh/pynastran_io.h"


namespace MAST {

class NastranIO : public libMesh::MeshInput<libMesh::MeshBase> {
    public:

        explicit NastranIO(libMesh::MeshBase& mesh, const bool python_preinit=false);

        virtual ~NastranIO();

        virtual void read(const std::string & filename) override;

        virtual void read(BDFModel* model);

        std::map<uint64_t, libMesh::Node*> get_nastran_to_libmesh_node_map();

        std::map<const libMesh::Node*, uint64_t> get_libmesh_to_nastran_node_map();

        std::map<uint64_t, libMesh::Elem*> get_nastran_to_libmesh_elem_map();

        std::map<libMesh::Elem*, uint64_t> get_libmesh_to_nastran_elem_map();

        std::map<std::pair<int,int>, int> get_nastran_pid_elemtype_to_libmesh_subdomain_map();

        std::map<int, std::set<int>> get_nastran_property_to_libmesh_subdomains_map();

        void print_pid_to_subdomain_id_map();

    private:

        const bool  python_preinitialized = false;
        bool        python_initialized =     false;

        std::map<uint64_t, libMesh::Node*> nastran_to_libmesh_node_map;
        std::map<const libMesh::Node*, uint64_t> libmesh_to_nastran_node_map;
        std::map<uint64_t, libMesh::Elem*> nastran_to_libmesh_elem_map;
        std::map<libMesh::Elem*, uint64_t> libmesh_to_nastran_elem_map;

        std::map<std::pair<int, int>, int> nastran_pid_elemtype_to_libmesh_subdomain_map = {};

        void read_nodes(BDFModel* model, libMesh::MeshBase& the_mesh);
        void read_elements(BDFModel* model, libMesh::MeshBase& the_mesh);
        void read_node_boundaries(BDFModel* model, libMesh::MeshBase& the_mesh);

        std::map<std::string, libMesh::ElemType> nastran_to_libmesh_elem_type_map = {

            // 0D Elements (i.e. Ground Springs)
            {"CELAS1_1", libMesh::NODEELEM},    {"CELAS2_1",  libMesh::NODEELEM},
            {"CELAS3_1", libMesh::NODEELEM},    {"CELAS4_1",  libMesh::NODEELEM},
            {"CBUSH_1",  libMesh::NODEELEM},    {"CBUSH1D_1", libMesh::NODEELEM},

            // 1D Elements
            {"CELAS1_2", libMesh::EDGE2},       {"CELAS2_2",  libMesh::EDGE2},
            {"CELAS3_2", libMesh::EDGE2},       {"CELAS4_2",  libMesh::EDGE2},
            {"CBUSH_2",  libMesh::EDGE2},       {"CBUSH1D_2", libMesh::EDGE2},
            {"CBUSH2D_2",libMesh::EDGE2},
            {"CROD_2",   libMesh::EDGE2},       {"CBAR_2",    libMesh::EDGE2},
            {"CBEAM_2",  libMesh::EDGE2},       {"CBEAM3_3",  libMesh::EDGE3},

            // 2D Elements
            {"CTRIA3_3", libMesh::TRI3},        {"CTRIA6_6",  libMesh::TRI6},
            {"CTRIAR_3", libMesh::TRI3},
            {"CQUAD4_4", libMesh::QUAD4},       {"CQUAD8_8",  libMesh::QUAD8},
            {"CQUADR_4", libMesh::QUADSHELL4},  {"CQUAD_4",   libMesh::QUAD4},
            {"CQUAD_8",  libMesh::QUAD8},       {"CQUAD_9",   libMesh::QUAD9},
            {"CQUADX_4", libMesh::QUAD4},       {"CQUADX_8",  libMesh::QUAD8},
            {"CQUADX_9", libMesh::QUAD9},

            // 3D Elements
            {"CTETRA_4", libMesh::TET4},        {"CTETRA_10", libMesh::TET10},
            {"CPENTA_6", libMesh::PRISM6},      {"CPENTA_15", libMesh::PRISM15},
            {"CPYRAM_5", libMesh::PYRAMID5},    {"CPYRAM_13", libMesh::PYRAMID13},
            {"CHEXA_8",  libMesh::HEX8},        {"CHEXA_20",  libMesh::HEX20}
        };

        void initialize_python();
        void finalize_python();
    };

    // Utility functions.
    void printElementMap(std::map<std::string, std::vector<std::vector<int>>> elementMap);
    void printNodeCoords(std::vector<std::vector<double>> nodes);
}

#endif // __mast_nastran_io_h__