Tool to add a top layer with specified thickness to a mesh #649
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,202 @@ | ||
| /* | ||
| * \date 2015-04-14 | ||
| * \brief Adds a top layer to an existing mesh. | ||
| * | ||
| * \copyright | ||
| * Copyright (c) 2012-2015, OpenGeoSys Community (http://www.opengeosys.org) | ||
| * Distributed under a Modified BSD License. | ||
| * See accompanying file LICENSE.txt or | ||
| * http://www.opengeosys.org/project/license | ||
| */ | ||
|
|
||
| // TCLAP | ||
| #include "tclap/CmdLine.h" | ||
|
|
||
| // ThirdParty/logog | ||
| #include "logog/include/logog.hpp" | ||
|
|
||
| // BaseLib | ||
| #include "BaseLib/LogogSimpleFormatter.h" | ||
| #include "BaseLib/FileTools.h" | ||
|
|
||
| // FileIO | ||
| #include "FileIO/readMeshFromFile.h" | ||
| #include "FileIO/VtkIO/VtuInterface.h" | ||
|
|
||
| // MeshLib | ||
| #include "MeshLib/Mesh.h" | ||
| #include "MeshLib/Node.h" | ||
| #include "MeshLib/Elements/Element.h" | ||
| #include "MeshLib/Elements/Prism.h" | ||
| #include "MeshLib/Elements/Hex.h" | ||
| #include "MeshLib/Elements/Quad.h" | ||
|
There was a problem hiding this comment. technically, several includes are missing (tri, line, etc.) ✅ |
||
| #include "MeshLib/Elements/Tri.h" | ||
| #include "MeshLib/Elements/Line.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(2)->getID()]; | ||
| prism_nodes[1] = subsfc_nodes[sfc_elem->getNode(1)->getID()]; | ||
| prism_nodes[2] = subsfc_nodes[sfc_elem->getNode(0)->getID()]; | ||
| prism_nodes[3] = subsfc_nodes[ | ||
| subsfc_sfc_id_map.at(sfc_elem->getNode(2)->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(0)->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[]) | ||
| { | ||
| LOGOG_INITIALIZE(); | ||
| logog::Cout* logog_cout (new logog::Cout); | ||
| BaseLib::LogogSimpleFormatter *custom_format (new BaseLib::LogogSimpleFormatter); | ||
| logog_cout->SetFormatter(*custom_format); | ||
|
|
||
| TCLAP::CmdLine cmd( | ||
| "Adds a top layer to an existing mesh", | ||
| ' ', | ||
| "0.1"); | ||
|
|
||
| TCLAP::ValueArg<std::string> mesh_arg("i", "input-mesh-file", | ||
| "the name of the file containing the mesh", true, | ||
| "", "file name"); | ||
| cmd.add(mesh_arg); | ||
|
|
||
| TCLAP::ValueArg<std::string> mesh_out_arg("o", "output-mesh-file", | ||
| "the name of the file the mesh should be written to (vtu format)", true, | ||
| "", "file name"); | ||
| cmd.add(mesh_out_arg); | ||
|
|
||
| TCLAP::ValueArg<double> layer_thickness_arg("t", "layer-tickness", | ||
| "the thickness of the new layer", false, 10, "floating point value"); | ||
| cmd.add(layer_thickness_arg); | ||
|
|
||
| cmd.parse(argc, argv); | ||
|
|
||
| INFO("Reading mesh \"%s\" ... ", mesh_arg.getValue().c_str()); | ||
| 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."); | ||
|
There was a problem hiding this comment. you are not testing if the input mesh is actually 3D (not that it matters, the method will work just as well with a 2D mesh ... it is just not consistent with the documentation in the beginning) There was a problem hiding this comment. Generalized the documentation from "3d mesh" to "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()); | ||
|
|
||
| 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) { | ||
|
There was a problem hiding this comment. for a deep-copy of nodes/elements you can use the methods from There was a problem hiding this comment. In this loop simultaneously the |
||
| 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() == MeshElemType::TRIANGLE) { | ||
| // add a prism | ||
|
There was a problem hiding this comment. I'd extract the content of the innermost if into own function. Is an element of POINT type not implemented or is it just missing here? There was a problem hiding this comment.
I'll try to extract the instructions of the innermost if into separate methods. ✅
At the moment we have not an element type POINT. |
||
| 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() == 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() == 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); | ||
| mesh_io.writeToFile(mesh_out_arg.getValue()); | ||
| INFO("done."); | ||
|
|
||
| return 0; | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
copyright header missing. [TF] fixed.