From a978fc339e4b11c97eb68ab6f7ad33f13f06ad6b Mon Sep 17 00:00:00 2001
From: rinkk <karsten.rink@ufz.de>
Date: Mon, 18 Jan 2016 17:56:20 +0100
Subject: [PATCH] moved content of AddTopLayer utility to MeshEditing
---
Applications/Utils/MeshEdit/AddTopLayer.cpp | 133 +---------------
MeshLib/MeshEditing/AddLayerToMesh.cpp | 163 ++++++++++++++++++++
MeshLib/MeshEditing/AddLayerToMesh.h | 32 ++++
3 files changed, 198 insertions(+), 130 deletions(-)
create mode 100644 MeshLib/MeshEditing/AddLayerToMesh.cpp
create mode 100644 MeshLib/MeshEditing/AddLayerToMesh.h
diff --git a/Applications/Utils/MeshEdit/AddTopLayer.cpp b/Applications/Utils/MeshEdit/AddTopLayer.cpp
index dea3d17411d..325353f9148 100644
--- a/Applications/Utils/MeshEdit/AddTopLayer.cpp
+++ b/Applications/Utils/MeshEdit/AddTopLayer.cpp
@@ -27,60 +27,8 @@
// MeshLib
#include "MeshLib/Mesh.h"
-#include "MeshLib/Node.h"
-#include "MeshLib/Elements/Elements.h"
-#include "MeshLib/MeshSurfaceExtraction.h"
+#include "MeshLib/MeshEditing/AddLayerToMesh.h"
-MeshLib::Prism* extrudeElement(std::vector<MeshLib::Node*> const& subsfc_nodes,
- MeshLib::Tri const*const sfc_elem,
- std::map<std::size_t, std::size_t> const& subsfc_sfc_id_map)
-{
- std::array<MeshLib::Node*, 6> prism_nodes;
- prism_nodes[0] = subsfc_nodes[sfc_elem->getNode(0)->getID()];
- prism_nodes[1] = subsfc_nodes[sfc_elem->getNode(1)->getID()];
- prism_nodes[2] = subsfc_nodes[sfc_elem->getNode(2)->getID()];
- prism_nodes[3] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(0)->getID())];
- prism_nodes[4] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(1)->getID())];
- prism_nodes[5] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(2)->getID())];
- return new MeshLib::Prism(prism_nodes);
-}
-
-MeshLib::Hex* extrudeElement(std::vector<MeshLib::Node*> const& subsfc_nodes,
- MeshLib::Quad const*const sfc_elem,
- std::map<std::size_t, std::size_t> const& subsfc_sfc_id_map)
-{
- std::array<MeshLib::Node*, 8> hex_nodes;
- hex_nodes[0] = subsfc_nodes[sfc_elem->getNode(0)->getID()];
- hex_nodes[1] = subsfc_nodes[sfc_elem->getNode(1)->getID()];
- hex_nodes[2] = subsfc_nodes[sfc_elem->getNode(2)->getID()];
- hex_nodes[3] = subsfc_nodes[sfc_elem->getNode(3)->getID()];
- hex_nodes[4] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(0)->getID())];
- hex_nodes[5] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(1)->getID())];
- hex_nodes[6] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(2)->getID())];
- hex_nodes[7] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(3)->getID())];
- return new MeshLib::Hex(hex_nodes);
-}
-
-MeshLib::Quad* extrudeElement(std::vector<MeshLib::Node*> const& subsfc_nodes,
- MeshLib::Line const*const sfc_elem,
- std::map<std::size_t, std::size_t> const& subsfc_sfc_id_map)
-{
- std::array<MeshLib::Node*, 4> quad_nodes;
- quad_nodes[0] = subsfc_nodes[sfc_elem->getNode(1)->getID()];
- quad_nodes[1] = subsfc_nodes[sfc_elem->getNode(0)->getID()];
- quad_nodes[2] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(0)->getID())];
- quad_nodes[3] = subsfc_nodes[
- subsfc_sfc_id_map.at(sfc_elem->getNode(1)->getID())];
- return new MeshLib::Quad(quad_nodes);
-}
int main (int argc, char* argv[])
{
@@ -114,85 +62,10 @@ int main (int argc, char* argv[])
MeshLib::Mesh * subsfc_mesh(FileIO::readMeshFromFile(mesh_arg.getValue()));
INFO("done.");
- INFO("Extracting top surface of mesh \"%s\" ... ",
- mesh_arg.getValue().c_str());
- const MathLib::Vector3 dir(0,0,-1);
- double const angle(90);
- std::unique_ptr<MeshLib::Mesh> sfc_mesh(
- MeshLib::MeshSurfaceExtraction::getMeshSurface(
- *subsfc_mesh, dir, angle, true
- )
- );
- INFO("done.");
+ MeshLib::Mesh* result (MeshLib::addTopLayerToMesh(*subsfc_mesh, layer_thickness_arg.getValue()));
- // *** add new top surface nodes
- std::vector<MeshLib::Node*> & subsfc_nodes(
- const_cast<std::vector<MeshLib::Node*> &>(subsfc_mesh->getNodes())
- );
- std::size_t const n_subsfc_nodes(subsfc_nodes.size());
-
- std::vector<MeshLib::Node*> const& sfc_nodes(sfc_mesh->getNodes());
- std::size_t const n_sfc_nodes(sfc_nodes.size());
-
- double const layer_thickness(layer_thickness_arg.getValue());
- // *** copy sfc nodes to subsfc mesh node
- std::map<std::size_t, std::size_t> subsfc_sfc_id_map;
- for (std::size_t k(0); k<n_sfc_nodes; ++k) {
- subsfc_nodes.push_back(new MeshLib::Node(*sfc_nodes[k]));
- std::size_t const subsfc_id(sfc_nodes[k]->getID());
- std::size_t const sfc_id(k+n_subsfc_nodes);
- subsfc_sfc_id_map.insert(std::make_pair(subsfc_id, sfc_id));
- (*(subsfc_nodes.back()))[2] += layer_thickness;
- }
- subsfc_mesh->resetNodeIDs();
-
- // *** insert new top layer elements into subsfc_mesh
- std::vector<MeshLib::Element*> & subsfc_elements(
- const_cast<std::vector<MeshLib::Element*> &>(subsfc_mesh->getElements())
- );
- std::size_t orig_size(subsfc_elements.size());
- std::vector<MeshLib::Element*> const& sfc_elements(sfc_mesh->getElements());
- std::size_t const n_sfc_elements(sfc_elements.size());
- for (std::size_t k(0); k<n_sfc_elements; ++k) {
- MeshLib::Element const*const sfc_elem(sfc_elements[k]);
- if (sfc_elem->getGeomType() == MeshLib::MeshElemType::TRIANGLE) {
- // add a prism
- subsfc_elements.push_back(extrudeElement(subsfc_nodes,
- static_cast<MeshLib::Tri const*const>(sfc_elem),
- subsfc_sfc_id_map));
- } else {
- if (sfc_elements[k]->getGeomType() == MeshLib::MeshElemType::QUAD) {
- // add a hexahedron
- subsfc_elements.push_back(extrudeElement(subsfc_nodes,
- static_cast<MeshLib::Quad const*const>(sfc_elem),
- subsfc_sfc_id_map));
- } else {
- if (sfc_elements[k]->getGeomType() == MeshLib::MeshElemType::LINE) {
- // add a quad
- subsfc_elements.push_back(extrudeElement(subsfc_nodes,
- static_cast<MeshLib::Line const*const>(sfc_elem),
- subsfc_sfc_id_map));
- }
- }
- }
- }
-
- boost::optional<MeshLib::PropertyVector<int> &> opt_materials(
- subsfc_mesh->getProperties().getPropertyVector<int>("MaterialIDs")
- );
- if (!opt_materials) {
- ERR("Can not set material properties for new layer");
- } else {
- MeshLib::PropertyVector<int> & materials(opt_materials.get());
- unsigned layer_id(*(std::max_element(
- materials.cbegin(), materials.cend()))+1);
- while (orig_size<subsfc_elements.size()) {
- materials.push_back(layer_id);
- orig_size++;
- }
- }
INFO("Writing mesh \"%s\" ... ", mesh_out_arg.getValue().c_str());
- FileIO::VtuInterface mesh_io(subsfc_mesh, vtkXMLWriter::Binary);
+ FileIO::VtuInterface mesh_io(result.get(), vtkXMLWriter::Binary);
mesh_io.writeToFile(mesh_out_arg.getValue());
INFO("done.");
diff --git a/MeshLib/MeshEditing/AddLayerToMesh.cpp b/MeshLib/MeshEditing/AddLayerToMesh.cpp
new file mode 100644
index 00000000000..d47537de2d3
--- /dev/null
+++ b/MeshLib/MeshEditing/AddLayerToMesh.cpp
@@ -0,0 +1,163 @@
+/**
+ * \file AddLayerToMesh.cpp
+ * \author Karsten Rink
+ * \date 2014-03-25
+ * \brief Implementation of AddLayerToMesh class.
+ *
+ * \copyright
+ * Copyright (c) 2012-2016, OpenGeoSys Community (http://www.opengeosys.org)
+ * Distributed under a Modified BSD License.
+ * See accompanying file LICENSE.txt or
+ * http://www.opengeosys.org/project/license
+ *
+ */
+
+#include "AddLayerToMesh.h"
+
+#include <vector>
+#include <map>
+#include <memory>
+
+#include "logog/include/logog.hpp"
+
+#include "MeshLib/Mesh.h"
+#include "MeshLib/Node.h"
+#include "MeshLib/Elements/Elements.h"
+#include "MeshLib/MeshSurfaceExtraction.h"
+
+namespace MeshLib
+{
+
+MeshLib::Prism* extrudeElement(std::vector<MeshLib::Node*> const& subsfc_nodes,
+ MeshLib::Tri const*const sfc_elem,
+ std::map<std::size_t, std::size_t> const& subsfc_sfc_id_map)
+{
+ std::array<MeshLib::Node*, 6> prism_nodes;
+ prism_nodes[0] = subsfc_nodes[sfc_elem->getNode(0)->getID()];
+ prism_nodes[1] = subsfc_nodes[sfc_elem->getNode(1)->getID()];
+ prism_nodes[2] = subsfc_nodes[sfc_elem->getNode(2)->getID()];
+ prism_nodes[3] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(0)->getID())];
+ prism_nodes[4] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(1)->getID())];
+ prism_nodes[5] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(2)->getID())];
+ return new MeshLib::Prism(prism_nodes);
+}
+
+MeshLib::Hex* extrudeElement(std::vector<MeshLib::Node*> const& subsfc_nodes,
+ MeshLib::Quad const*const sfc_elem,
+ std::map<std::size_t, std::size_t> const& subsfc_sfc_id_map)
+{
+ std::array<MeshLib::Node*, 8> hex_nodes;
+ hex_nodes[0] = subsfc_nodes[sfc_elem->getNode(0)->getID()];
+ hex_nodes[1] = subsfc_nodes[sfc_elem->getNode(1)->getID()];
+ hex_nodes[2] = subsfc_nodes[sfc_elem->getNode(2)->getID()];
+ hex_nodes[3] = subsfc_nodes[sfc_elem->getNode(3)->getID()];
+ hex_nodes[4] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(0)->getID())];
+ hex_nodes[5] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(1)->getID())];
+ hex_nodes[6] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(2)->getID())];
+ hex_nodes[7] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(3)->getID())];
+ return new MeshLib::Hex(hex_nodes);
+}
+
+MeshLib::Quad* extrudeElement(std::vector<MeshLib::Node*> const& subsfc_nodes,
+ MeshLib::Line const*const sfc_elem,
+ std::map<std::size_t, std::size_t> const& subsfc_sfc_id_map)
+{
+ std::array<MeshLib::Node*, 4> quad_nodes;
+ quad_nodes[0] = subsfc_nodes[sfc_elem->getNode(1)->getID()];
+ quad_nodes[1] = subsfc_nodes[sfc_elem->getNode(0)->getID()];
+ quad_nodes[2] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(0)->getID())];
+ quad_nodes[3] = subsfc_nodes[
+ subsfc_sfc_id_map.at(sfc_elem->getNode(1)->getID())];
+ return new MeshLib::Quad(quad_nodes);
+}
+
+MeshLib::Mesh* addTopLayerToMesh(MeshLib::Mesh const& mesh, double thickness)
+{
+ INFO("Extracting top surface of mesh \"%s\" ... ", mesh.getName().c_str());
+ const MathLib::Vector3 dir(0,0,-1);
+ double const angle(90);
+ std::unique_ptr<MeshLib::Mesh> sfc_mesh(
+ MeshLib::MeshSurfaceExtraction::getMeshSurface(mesh, dir, angle, true)
+ );
+ INFO("done.");
+
+ MeshLib::Mesh* subsfc_mesh = new MeshLib::Mesh (mesh);
+
+ // *** add new top surface nodes
+ std::vector<MeshLib::Node*> & subsfc_nodes(
+ const_cast<std::vector<MeshLib::Node*> &>(subsfc_mesh->getNodes())
+ );
+ std::size_t const n_subsfc_nodes(subsfc_nodes.size());
+
+ std::vector<MeshLib::Node*> const& sfc_nodes(sfc_mesh->getNodes());
+ std::size_t const n_sfc_nodes(sfc_nodes.size());
+
+ // *** copy sfc nodes to subsfc mesh node
+ std::map<std::size_t, std::size_t> subsfc_sfc_id_map;
+ for (std::size_t k(0); k<n_sfc_nodes; ++k) {
+ subsfc_nodes.push_back(new MeshLib::Node(*sfc_nodes[k]));
+ std::size_t const subsfc_id(sfc_nodes[k]->getID());
+ std::size_t const sfc_id(k+n_subsfc_nodes);
+ subsfc_sfc_id_map.insert(std::make_pair(subsfc_id, sfc_id));
+ (*(subsfc_nodes.back()))[2] += thickness;
+ }
+ subsfc_mesh->resetNodeIDs();
+
+ // *** insert new top layer elements into subsfc_mesh
+ std::vector<MeshLib::Element*> & subsfc_elements(
+ const_cast<std::vector<MeshLib::Element*> &>(subsfc_mesh->getElements())
+ );
+ std::size_t orig_size(subsfc_elements.size());
+ std::vector<MeshLib::Element*> const& sfc_elements(sfc_mesh->getElements());
+ std::size_t const n_sfc_elements(sfc_elements.size());
+ for (std::size_t k(0); k<n_sfc_elements; ++k) {
+ MeshLib::Element const*const sfc_elem(sfc_elements[k]);
+ if (sfc_elem->getGeomType() == MeshLib::MeshElemType::TRIANGLE) {
+ // add a prism
+ subsfc_elements.push_back(extrudeElement(subsfc_nodes,
+ static_cast<MeshLib::Tri const*const>(sfc_elem),
+ subsfc_sfc_id_map));
+ } else {
+ if (sfc_elements[k]->getGeomType() == MeshLib::MeshElemType::QUAD) {
+ // add a hexahedron
+ subsfc_elements.push_back(extrudeElement(subsfc_nodes,
+ static_cast<MeshLib::Quad const*const>(sfc_elem),
+ subsfc_sfc_id_map));
+ } else {
+ if (sfc_elements[k]->getGeomType() == MeshLib::MeshElemType::LINE) {
+ // add a quad
+ subsfc_elements.push_back(extrudeElement(subsfc_nodes,
+ static_cast<MeshLib::Line const*const>(sfc_elem),
+ subsfc_sfc_id_map));
+ }
+ }
+ }
+ }
+
+ boost::optional<MeshLib::PropertyVector<int> &> opt_materials(
+ subsfc_mesh->getProperties().getPropertyVector<int>("MaterialIDs")
+ );
+ if (!opt_materials) {
+ ERR("Can not set material properties for new layer");
+ } else {
+ MeshLib::PropertyVector<int> & materials(opt_materials.get());
+ unsigned layer_id(*(std::max_element(
+ materials.cbegin(), materials.cend()))+1);
+ while (orig_size<subsfc_elements.size()) {
+ materials.push_back(layer_id);
+ orig_size++;
+ }
+ }
+
+ return subsfc_mesh;
+}
+
+} // namespace MeshLib
diff --git a/MeshLib/MeshEditing/AddLayerToMesh.h b/MeshLib/MeshEditing/AddLayerToMesh.h
new file mode 100644
index 00000000000..1b552d12fca
--- /dev/null
+++ b/MeshLib/MeshEditing/AddLayerToMesh.h
@@ -0,0 +1,32 @@
+/**
+ * \file AddLayerToMesh.h
+ * \author Karsten Rink
+ * \date 2016-01-18
+ * \brief Definition of AddLayerToMesh class
+ *
+ * \copyright
+ * Copyright (c) 2012-2016, OpenGeoSys Community (http://www.opengeosys.org)
+ * Distributed under a Modified BSD License.
+ * See accompanying file LICENSE.txt or
+ * http://www.opengeosys.org/project/license
+ *
+ */
+
+#ifndef ADDLAYERTOMESH_H
+#define ADDLAYERTOMESH_H
+
+#include <vector>
+
+namespace MeshLib
+{
+
+class Mesh;
+class Node;
+class Element;
+
+MeshLib::Mesh* addTopLayerToMesh(MeshLib::Mesh const& mesh, double thickness);
+
+
+} // end namespace MeshLib
+
+#endif //ADDLAYERTOMESH_H
--
GitLab