NewAttributeStrategy.hpp

Go to the documentation of this file.

#pragma once
#include <functional>
#include <wmtk/PrimitiveType.hpp>
#include <wmtk/Types.hpp>
#include <wmtk/operations/AttributeTransferEdge.hpp>
#include <wmtk/simplex/Simplex.hpp>
 
#include <wmtk/attribute/MeshAttributeHandle.hpp>
 
#include <bitset>
 
namespace wmtk {
class Mesh;
class Tuple;
} // namespace wmtk
 
namespace wmtk::operations {
 
enum class BasicSimplexPredicate { Default, IsInterior, None };
 
class NewAttributeStrategy : public AttributeTransferEdge
{
public:
    using SimplexPredicateType = std::function<bool(const simplex::Simplex&)>;
 
    virtual ~NewAttributeStrategy();
 
    virtual bool matches_attribute(const attribute::MeshAttributeHandle&) const = 0;
    virtual void update_handle_mesh(Mesh&) = 0;
 
    virtual PrimitiveType primitive_type() const = 0;
 
    virtual Mesh& mesh() = 0;
    const Mesh& mesh() const;
 
    void set_simplex_predicate(SimplexPredicateType&& f);
    void set_simplex_predicate(BasicSimplexPredicate f);
 
    std::bitset<2> evaluate_predicate(PrimitiveType pt, const std::array<Tuple, 2>& simplices)
        const;
 
    std::vector<wmtk::attribute::MeshAttributeHandle> sources() const final override { return {}; }
 
    virtual bool invalid_state() const = 0;
    virtual std::string name() const = 0;
 
protected:
private:
    SimplexPredicateType m_simplex_predicate = nullptr;
};
 
} // namespace wmtk::operations