MeshSubset.h

Go to the documentation of this file.

 
#pragma once
 
#include <cassert>
#include <vector>
 
#include "Mesh.h"
#include "Node.h"
#include "Elements/Element.h"
 
 
namespace MeshLib
{
class MeshSubset
{
public:
    MeshSubset(const Mesh& msh, std::vector<Node*> const& vec_items)
        : _msh(msh), _nodes(vec_items)
    {
        // If the mesh nodes and the given nodes point to the same vector, they
        // must be equal.
        if (&_msh.getNodes() == &_nodes)
        {
            return;
        }
 
        //
        // Testing if the given nodes belong to the mesh.
        //
        {
            // Need sorted version of the large vector.
            auto sorted_nodes = _msh.getNodes();  // full copy of pointers.
            sort(begin(sorted_nodes), end(sorted_nodes));
 
            // Then proceed with the search function.
            auto node_is_part_of_mesh = [& mesh_nodes = sorted_nodes](
                                            MeshLib::Node* const& n) {
                auto it = lower_bound(begin(mesh_nodes), end(mesh_nodes), n);
                if (it == end(mesh_nodes))
                {
                    ERR("A node {:d} ({:g}, {:g}, {:g}) in mesh subset is not "
                        "a part of the mesh.",
                        n->getID(), (*n)[0], (*n)[1], (*n)[2]);
                    return false;
                }
                return true;
            };
            if (!std::all_of(begin(_nodes), end(_nodes), node_is_part_of_mesh))
            {
                OGS_FATAL("The mesh subset construction failed.");
            }
        }
    }
 
    std::size_t getMeshID() const
    {
        return _msh.getID();
    }
 
    std::size_t getNumberOfNodes() const
    {
        return _nodes.size();
    }
 
    std::size_t getNodeID(std::size_t const i) const
    {
        assert(i < _nodes.size());
        return _nodes[i]->getID();
    }
 
    std::vector<Element*>::const_iterator elementsBegin() const
    {
        return _msh.getElements().cbegin();
    }
 
    std::vector<Element*>::const_iterator elementsEnd() const
    {
        return _msh.getElements().cend();
    }
 
    std::vector<Node*> const& getNodes() const
    {
        return _nodes;
    }
 
    Mesh const& getMesh() const
    {
        return _msh;
    }
 
private:
    Mesh const& _msh;
    std::vector<Node*> const& _nodes;
};
}   // namespace MeshLib