AttributeTable.h

浏览该文件的文档.

#pragma once
 
#include "common/Result.h"
 
#include <cstdint>
#include <optional>
#include <string>
#include <string_view>
#include <unordered_map>
#include <variant>
#include <vector>
 
namespace eve::procgen {
 
struct ProcgenAttributeVector {
    float x = 0.f;
    float y = 0.f;
    float z = 0.f;
};
 
enum class ProcgenAttributeType : std::uint8_t { Float, Int, Bool, Vector, String };
 
class EVENGINE_API_DOMAINS AttributeTable {
public:
    [[nodiscard]] std::size_t rowCount() const noexcept;
    void resize(std::size_t rows);
    [[nodiscard]] std::size_t appendRow();
    [[nodiscard]] Result<std::size_t> appendRowFrom(const AttributeTable& source, std::size_t sourceRow);
    [[nodiscard]] Result<void> clearRow(std::size_t row);
    void clear() noexcept;
 
    [[nodiscard]] std::size_t columnCount() const noexcept;
    [[nodiscard]] std::string_view columnName(std::size_t index) const noexcept;
    [[nodiscard]] std::optional<ProcgenAttributeType> typeOf(std::string_view name) const;
    [[nodiscard]] bool has(std::size_t row, std::string_view name) const;
 
    [[nodiscard]] Result<void> renameColumn(std::string_view from, std::string_view to);
    [[nodiscard]] Result<void> removeColumn(std::string_view name);
    [[nodiscard]] Result<void> copyColumn(std::string_view from, std::string_view to);
 
    [[nodiscard]] Result<void> setFloat(std::size_t row, std::string_view name, float value);
    [[nodiscard]] Result<void> setInt(std::size_t row, std::string_view name, std::int64_t value);
    [[nodiscard]] Result<void> setBool(std::size_t row, std::string_view name, bool value);
    [[nodiscard]] Result<void> setVector(std::size_t row, std::string_view name, ProcgenAttributeVector value);
    [[nodiscard]] Result<void> setString(std::size_t row, std::string_view name, std::string value);
 
    [[nodiscard]] std::optional<float> getFloat(std::size_t row, std::string_view name) const;
    [[nodiscard]] std::optional<std::int64_t> getInt(std::size_t row, std::string_view name) const;
    [[nodiscard]] std::optional<bool> getBool(std::size_t row, std::string_view name) const;
    [[nodiscard]] std::optional<ProcgenAttributeVector> getVector(std::size_t row, std::string_view name) const;
    [[nodiscard]] std::optional<std::string_view> getString(std::size_t row, std::string_view name) const;
 
private:
    using FloatColumn  = std::vector<std::optional<float>>;
    using IntColumn    = std::vector<std::optional<std::int64_t>>;
    using BoolColumn   = std::vector<std::optional<bool>>;
    using VectorColumn = std::vector<std::optional<ProcgenAttributeVector>>;
    using StringColumn = std::vector<std::optional<std::string>>;
    using Storage      = std::variant<FloatColumn, IntColumn, BoolColumn, VectorColumn, StringColumn>;
    struct Column {
        ProcgenAttributeType type    = ProcgenAttributeType::Float;
        Storage              storage = FloatColumn{};
    };
 
    template <class T>
    [[nodiscard]] Result<void> set(std::size_t row, std::string_view name, ProcgenAttributeType type, T value);
    template <class T>
    [[nodiscard]] std::optional<T> get(std::size_t row, std::string_view name, ProcgenAttributeType type) const;
    [[nodiscard]] Result<Column*>  ensureColumn(std::string_view name, ProcgenAttributeType type);
 
    std::size_t                             rows_ = 0;
    std::unordered_map<std::string, Column> columns_;
    std::vector<std::string>                order_;
};
 
[[nodiscard]] std::string_view procgenAttributeTypeName(ProcgenAttributeType type) noexcept;
 
}  // namespace eve::procgen