Extra2D/include/renderer/material.h

#pragma once

#include <types/ptr/ref_counted.h>
#include <types/ptr/intrusive_ptr.h>
#include <types/math/vec2.h>
#include <types/math/color.h>
#include <renderer/render_types.h>
#include <renderer/shader.h>
#include <renderer/uniform_buffer.h>
#include <string>
#include <unordered_map>
#include <vector>

namespace extra2d {

// 前向声明
class Material;

/**
 * @brief 材质参数信息
 */
struct MaterialParamInfo {
    MaterialParamType type;
    uint32_t offset;
    uint32_t size;
};

/**
 * @brief 材质布局类
 * 
 * 定义材质参数的布局，可被多个材质共享
 */
class MaterialLayout : public RefCounted {
public:
    /**
     * @brief 默认构造函数
     */
    MaterialLayout();
    
    /**
     * @brief 添加参数
     * @param name 参数名称
     * @param type 参数类型
     */
    void addParam(const std::string& name, MaterialParamType type);
    
    /**
     * @brief 完成布局定义，计算偏移和总大小
     */
    void finalize();
    
    /**
     * @brief 获取参数信息
     * @param name 参数名称
     * @return 参数信息指针，不存在返回 nullptr
     */
    const MaterialParamInfo* getParam(const std::string& name) const;
    
    /**
     * @brief 获取缓冲区大小
     * @return 缓冲区大小
     */
    uint32_t getBufferSize() const { return bufferSize_; }
    
    /**
     * @brief 检查是否已最终化
     * @return 是否已最终化
     */
    bool isFinalized() const { return finalized_; }

private:
    std::unordered_map<std::string, MaterialParamInfo> params_;
    uint32_t bufferSize_ = 0;
    bool finalized_ = false;
};

/**
 * @brief 材质类
 * 
 * 管理着色器和材质参数
 * 支持通过 UBO 高效上传材质数据到 GPU
 */
class Material : public RefCounted {
public:
    /**
     * @brief 默认构造函数
     */
    Material();
    
    /**
     * @brief 设置材质布局
     * @param layout 材质布局
     */
    void setLayout(Ptr<MaterialLayout> layout);
    
    /**
     * @brief 设置着色器
     * @param shader 着色器
     */
    void setShader(Ptr<Shader> shader);
    
    /**
     * @brief 设置 float 参数
     * @param name 参数名称
     * @param value 值
     */
    void setFloat(const std::string& name, float value);
    
    /**
     * @brief 设置 vec2 参数
     * @param name 参数名称
     * @param value 值
     */
    void setVec2(const std::string& name, const Vec2& value);
    
    /**
     * @brief 设置 vec4 参数
     * @param name 参数名称
     * @param value 值
     */
    void setVec4(const std::string& name, float x, float y, float z, float w);
    
    /**
     * @brief 设置颜色参数
     * @param name 参数名称
     * @param value 颜色值
     */
    void setColor(const std::string& name, const Color& value);
    
    /**
     * @brief 设置 mat4 参数
     * @param name 参数名称
     * @param value 矩阵数据指针
     */
    void setMat4(const std::string& name, const float* value);
    
    /**
     * @brief 设置纹理
     * @param name 参数名称
     * @param texture 纹理句柄
     * @param slot 纹理槽位
     */
    void setTexture(const std::string& name, TextureHandle texture, uint32_t slot);
    
    /**
     * @brief 应用材质
     * 
     * 绑定着色器、上传 UBO 数据、绑定纹理
     * @param uboManager UBO 管理器
     */
    void apply(UniformBufferManager& uboManager);
    
    /**
     * @brief 获取着色器
     * @return 着色器
     */
    Ptr<Shader> getShader() const { return shader_; }
    
    /**
     * @brief 获取材质数据指针
     * @return 数据指针
     */
    const void* getData() const { return data_.data(); }
    
    /**
     * @brief 获取材质数据大小
     * @return 数据大小
     */
    uint32_t getDataSize() const { return static_cast<uint32_t>(data_.size()); }

private:
    Ptr<MaterialLayout> layout_;    // 材质布局
    Ptr<Shader> shader_;            // 着色器
    std::vector<uint8_t> data_;     // 材质数据
    std::vector<std::pair<TextureHandle, uint32_t>> textures_;  // 纹理列表
};

} // namespace extra2d