You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
142 lines
6.9 KiB
142 lines
6.9 KiB
#include <mshio/mshio.h>
|
|
|
|
#include <nanobind/nanobind.h>
|
|
#include <nanobind/stl/string.h>
|
|
#include <nanobind/stl/vector.h>
|
|
|
|
namespace nb = nanobind;
|
|
|
|
NB_MODULE(pymshio, m)
|
|
{
|
|
nb::class_<mshio::MeshFormat>(m, "MeshFormat")
|
|
.def(nb::init<>())
|
|
.def_readwrite("version", &mshio::MeshFormat::version)
|
|
.def_readwrite("file_type", &mshio::MeshFormat::file_type)
|
|
.def_readwrite("data_size", &mshio::MeshFormat::data_size);
|
|
|
|
nb::class_<mshio::NodeBlock>(m, "NodeBlock")
|
|
.def(nb::init<>())
|
|
.def_readwrite("entity_dim", &mshio::NodeBlock::entity_dim)
|
|
.def_readwrite("entity_tag", &mshio::NodeBlock::entity_tag)
|
|
.def_readwrite("parametric", &mshio::NodeBlock::parametric)
|
|
.def_readwrite("num_nodes_in_block", &mshio::NodeBlock::num_nodes_in_block)
|
|
.def_readwrite("tags", &mshio::NodeBlock::tags)
|
|
.def_readwrite("data", &mshio::NodeBlock::data);
|
|
|
|
nb::class_<mshio::Nodes>(m, "Nodes")
|
|
.def(nb::init<>())
|
|
.def_readwrite("num_entity_blocks", &mshio::Nodes::num_entity_blocks)
|
|
.def_readwrite("num_nodes", &mshio::Nodes::num_nodes)
|
|
.def_readwrite("min_node_tag", &mshio::Nodes::min_node_tag)
|
|
.def_readwrite("max_node_tag", &mshio::Nodes::max_node_tag)
|
|
.def_readwrite("entity_blocks", &mshio::Nodes::entity_blocks);
|
|
|
|
nb::class_<mshio::ElementBlock>(m, "ElementBlock")
|
|
.def(nb::init<>())
|
|
.def_readwrite("entity_dim", &mshio::ElementBlock::entity_dim)
|
|
.def_readwrite("entity_tag", &mshio::ElementBlock::entity_tag)
|
|
.def_readwrite("element_type", &mshio::ElementBlock::element_type)
|
|
.def_readwrite("num_elements_in_block", &mshio::ElementBlock::num_elements_in_block)
|
|
.def_readwrite("data", &mshio::ElementBlock::data);
|
|
|
|
nb::class_<mshio::Elements>(m, "Elements")
|
|
.def(nb::init<>())
|
|
.def_readwrite("num_entity_blocks", &mshio::Elements::num_entity_blocks)
|
|
.def_readwrite("num_elements", &mshio::Elements::num_elements)
|
|
.def_readwrite("min_element_tag", &mshio::Elements::min_element_tag)
|
|
.def_readwrite("max_element_tag", &mshio::Elements::max_element_tag)
|
|
.def_readwrite("entity_blocks", &mshio::Elements::entity_blocks);
|
|
|
|
nb::class_<mshio::DataHeader>(m, "DataHeader")
|
|
.def(nb::init<>())
|
|
.def_readwrite("string_tags", &mshio::DataHeader::string_tags)
|
|
.def_readwrite("real_tags", &mshio::DataHeader::real_tags)
|
|
.def_readwrite("int_tags", &mshio::DataHeader::int_tags);
|
|
|
|
nb::class_<mshio::DataEntry>(m, "DataEntry")
|
|
.def(nb::init<>())
|
|
.def_readwrite("tag", &mshio::DataEntry::tag)
|
|
.def_readwrite("num_nodes_per_elements", &mshio::DataEntry::num_nodes_per_element)
|
|
.def_readwrite("data", &mshio::DataEntry::data);
|
|
|
|
nb::class_<mshio::Data>(m, "Data")
|
|
.def(nb::init<>())
|
|
.def_readwrite("header", &mshio::Data::header)
|
|
.def_readwrite("entries", &mshio::Data::entries);
|
|
|
|
nb::class_<mshio::PointEntity>(m, "PointEntity")
|
|
.def(nb::init<>())
|
|
.def_readwrite("tag", &mshio::PointEntity::tag)
|
|
.def_readwrite("x", &mshio::PointEntity::x)
|
|
.def_readwrite("y", &mshio::PointEntity::y)
|
|
.def_readwrite("z", &mshio::PointEntity::z)
|
|
.def_readwrite("physical_group_tags", &mshio::PointEntity::physical_group_tags);
|
|
|
|
nb::class_<mshio::CurveEntity>(m, "CurveEntity")
|
|
.def(nb::init<>())
|
|
.def_readwrite("tag", &mshio::CurveEntity::tag)
|
|
.def_readwrite("min_x", &mshio::CurveEntity::min_x)
|
|
.def_readwrite("min_y", &mshio::CurveEntity::min_y)
|
|
.def_readwrite("min_z", &mshio::CurveEntity::min_z)
|
|
.def_readwrite("max_x", &mshio::CurveEntity::max_x)
|
|
.def_readwrite("max_y", &mshio::CurveEntity::max_y)
|
|
.def_readwrite("max_z", &mshio::CurveEntity::max_z)
|
|
.def_readwrite("physical_group_tags", &mshio::CurveEntity::physical_group_tags)
|
|
.def_readwrite("boundary_point_tags", &mshio::CurveEntity::boundary_point_tags);
|
|
|
|
nb::class_<mshio::SurfaceEntity>(m, "SurfaceEntity")
|
|
.def(nb::init<>())
|
|
.def_readwrite("tag", &mshio::SurfaceEntity::tag)
|
|
.def_readwrite("min_x", &mshio::SurfaceEntity::min_x)
|
|
.def_readwrite("min_y", &mshio::SurfaceEntity::min_y)
|
|
.def_readwrite("min_z", &mshio::SurfaceEntity::min_z)
|
|
.def_readwrite("max_x", &mshio::SurfaceEntity::max_x)
|
|
.def_readwrite("max_y", &mshio::SurfaceEntity::max_y)
|
|
.def_readwrite("max_z", &mshio::SurfaceEntity::max_z)
|
|
.def_readwrite("physical_group_tags", &mshio::SurfaceEntity::physical_group_tags)
|
|
.def_readwrite("boundary_curve_tags", &mshio::SurfaceEntity::boundary_curve_tags);
|
|
|
|
nb::class_<mshio::VolumeEntity>(m, "VolumeEntity")
|
|
.def(nb::init<>())
|
|
.def_readwrite("tag", &mshio::VolumeEntity::tag)
|
|
.def_readwrite("min_x", &mshio::VolumeEntity::min_x)
|
|
.def_readwrite("min_y", &mshio::VolumeEntity::min_y)
|
|
.def_readwrite("min_z", &mshio::VolumeEntity::min_z)
|
|
.def_readwrite("max_x", &mshio::VolumeEntity::max_x)
|
|
.def_readwrite("max_y", &mshio::VolumeEntity::max_y)
|
|
.def_readwrite("max_z", &mshio::VolumeEntity::max_z)
|
|
.def_readwrite("physical_group_tags", &mshio::VolumeEntity::physical_group_tags)
|
|
.def_readwrite("boundary_surface_tags", &mshio::VolumeEntity::boundary_surface_tags);
|
|
|
|
nb::class_<mshio::Entities>(m, "Entities")
|
|
.def(nb::init<>())
|
|
.def("empty", &mshio::Entities::empty)
|
|
.def_readwrite("points", &mshio::Entities::points)
|
|
.def_readwrite("curves", &mshio::Entities::curves)
|
|
.def_readwrite("surfaces", &mshio::Entities::surfaces)
|
|
.def_readwrite("volumes", &mshio::Entities::volumes);
|
|
|
|
nb::class_<mshio::PhysicalGroup>(m, "PhysicalGroup")
|
|
.def(nb::init<>())
|
|
.def_readwrite("dim", &mshio::PhysicalGroup::dim)
|
|
.def_readwrite("tag", &mshio::PhysicalGroup::tag)
|
|
.def_readwrite("name", &mshio::PhysicalGroup::name);
|
|
|
|
nb::class_<mshio::MshSpec>(m, "MshSpec")
|
|
.def(nb::init<>())
|
|
.def_readwrite("mesh_format", &mshio::MshSpec::mesh_format)
|
|
.def_readwrite("nodes", &mshio::MshSpec::nodes)
|
|
.def_readwrite("elements", &mshio::MshSpec::elements)
|
|
.def_readwrite("entities", &mshio::MshSpec::entities)
|
|
.def_readwrite("physical_groups", &mshio::MshSpec::physical_groups)
|
|
.def_readwrite("node_data", &mshio::MshSpec::node_data)
|
|
.def_readwrite("element_data", &mshio::MshSpec::element_data)
|
|
.def_readwrite("element_node_data", &mshio::MshSpec::element_node_data);
|
|
|
|
m.def("load_msh", nb::overload_cast<const std::string&>(&mshio::load_msh));
|
|
m.def(
|
|
"save_msh", nb::overload_cast<const std::string&, const mshio::MshSpec&>(&mshio::save_msh));
|
|
m.def("validate_spec", &mshio::validate_spec);
|
|
m.def("nodes_per_element", &mshio::nodes_per_element);
|
|
m.def("get_element_dim", &mshio::get_element_dim);
|
|
}
|
|
|