LocalBoundary.hpp

Go to the documentation of this file.

#pragma once
 
#include <array>
#include <vector>
#include <iostream>
#include <cassert>
 
namespace polyfem
{
    namespace mesh
    {
        enum class BoundaryType
        {
            TRI_LINE,  
            QUAD_LINE, 
            PRISM,
            TRI,
            QUAD,
            POLYGON,
            POLYHEDRON,
            INVALID
        };
 
        class LocalBoundary
        {
        public:
            LocalBoundary(const int global_element_id, BoundaryType type);
 
            LocalBoundary(const LocalBoundary &other);
 
            void add_boundary_primitive(const int global_index, const int local_index);
 
            int size() const { return local_primitive_id_.size(); }
 
            bool empty() const { return size() <= 0; }
 
            int element_id() const { return global_element_id_; }
 
            BoundaryType type() const { return type_; }
 
            int local_primitive_id(const int index) const { return local_primitive_id_[index]; }
 
            int global_primitive_id(const int index) const { return global_primitive_id_[index]; }
 
            int operator[](const int index) const { return local_primitive_id(index); }
 
            void remove_from(const LocalBoundary &other);
 
            void remove_tag_for_index(const int index);
 
            friend std::ostream &operator<<(std::ostream &os, const LocalBoundary &lb);
 
        private:
            std::vector<int> global_primitive_id_;
            std::vector<int> local_primitive_id_;
 
            const int global_element_id_;
            const BoundaryType type_;
        };
    } // namespace mesh
} // namespace polyfem