MeshAttributeReader.h

#pragma once
#include "Engine/Attribute/AttributeHandle.h"
#include "Engine/Core/Types.h"
#include <array>
#include <string_view>
#include <variant>
#include <vector>
 
namespace enzo::geo {
class Mesh;
}
 
namespace enzo::script {
 
using ScriptAttributeHandle = std::variant<
    std::monostate,
    attr::AttributeHandleRO<floatT>,
    attr::AttributeHandleRO<intT>,
    attr::AttributeHandleRO<Vector3>,
    attr::AttributeHandleRO<boolT>>;
 
struct ScriptAttribute
{
    String name;
    attr::AttributeType type;
    ScriptAttributeHandle handle;
};
 
class MeshAttributeReader
{
  public:
    explicit MeshAttributeReader(const geo::Mesh& mesh);
 
    bool hasElement(attr::AttributeOwner owner, intT index) const;
 
    const ScriptAttribute* getAttribute(attr::AttributeOwner owner, std::string_view name) const;
 
  private:
    static constexpr size_t ownerCount = 4;
 
    std::array<Offset, ownerCount> elementCounts_;
    std::array<std::vector<ScriptAttribute>, ownerCount> attributes_;
};
 
} // namespace enzo::script