#pragma once

#include<hgl/graph/VKRenderResource.h>
#include<hgl/graph/VertexAttribDataAccess.h>
#include<hgl/graph/VKShaderModule.h>
#include<hgl/graph/VKIndexBuffer.h>
#include<hgl/graph/VKVertexAttribBuffer.h>

VK_NAMESPACE_BEGIN
class VertexDataManager;
struct PrimitiveData;

/**
 * 可绘制图元创建器
 */
class PrimitiveCreater
{
protected:

    GPUDevice *device;
    const GPUPhysicalDevice *phy_device;

    const VIL *         vil;

protected:

    PrimitiveData *     prim_data;

    void_pointer        vab_map_ptr[HGL_MAX_VERTEX_ATTRIB_COUNT];
    
    VkDeviceSize        vertices_number;
    IndexBuffer *       ibo;
    void *              ibo_map;

protected:

    template<typename T> friend class VABRawMap;
    template<typename T> friend class VABMap;

    VABAccess *AcquirePVB(const AnsiString &,const VkFormat &,const void *data);           ///<请求一个顶点属性数据区

    void ClearAllData();

public:

    PrimitiveCreater(GPUDevice *,const VIL *);
    //PrimitiveCreater(VertexDataManager *);
    virtual ~PrimitiveCreater();

    virtual bool                    Init(const VkDeviceSize vertices_count,const VkDeviceSize index_count,IndexType it=IndexType::AUTO);                 ///<初始化，参数为顶点数量
    
            const   VkDeviceSize    GetVertexCount()const{ return vertices_number; }                            ///<取得顶点数量

                    bool            WriteVAB(const AnsiString &name,const void *data,const uint32_t bytes);     ///<直接写入顶点属性数据

            const   IndexType       GetIndexType()const{return ibo?ibo->GetType():IndexType::ERR;}              ///<取得索引数据类型

            template<typename T> 
                    T *             AccessIBO()
            {
                if(!ibo)return(nullptr);
                if(ibo->GetStride()!=sizeof(T))return(nullptr);

                return (T *)ibo_map;
            }

            template<typename T> bool WriteIBO(const T *data)
            {
                if(!ibo)return(false);
                if(ibo->GetStride()!=sizeof(T))return(false);

                hgl_cpy<T>((T *)ibo_map,data,index_number);

                return(true);
            }

    virtual Primitive *             Finish(RenderResource *,const AnsiString &);                                                                   ///<结束并创建可渲染对象
};//class PrimitiveCreater

/**
* VAB原生数据访问映射
*/
template<typename T> class VABRawMap
{
    VABAccess *vaba;
    T *map_ptr;

public:

    VABRawMap(PrimitiveCreater *pc,const AnsiString &name)
    {
        vaba=pc->AcquirePVB(name,T::GetVulkanFormat(),nullptr);

        map_ptr=(T *)(vaba->vab->Map(vaba->start,vaba->count));
    }

    ~VABRawMap()
    {
        vaba->vab->Unmap();
    }

    T *operator->(){ return map_ptr; }
};//template<typename T> class VABRawMap

typedef VABRawMap<int8>   VABRawMapi8, VABRawMapByte;
typedef VABRawMap<int16>  VABRawMapi16,VABRawMapShort;
typedef VABRawMap<int32>  VABRawMapi32,VABRawMapInt;
typedef VABRawMap<uint8>  VABRawMapu8, VABRawMapUByte;
typedef VABRawMap<uint16> VABRawMapu16,VABRawMapUShort;
typedef VABRawMap<uint32> VABRawMapu32,VABRawMapUInt;
typedef VABRawMap<float>  VABRawMapFloat;
typedef VABRawMap<double> VABRawMapDouble;

/**
* VAB VertexAttribDataAccess数据访问映射
*/
template<typename T> class VABMap
{
    VABAccess *vaba;
    T *vb;

public:

    VABMap(PrimitiveCreater *pc,const AnsiString &name)
    {
        vaba=pc->AcquirePVB(name,T::GetVulkanFormat(),nullptr);

        void *map_ptr=vaba->vab->Map(vaba->start,vaba->count);

        vb=T::Create(pc->GetVertexCount(),map_ptr);

        vb->Begin();
    }

    ~VABMap()
    {
        vaba->vab->Unmap();
    }

    void Restart()
    {
        vb->Begin();
    }

    T *operator->(){ return vb; }
};//template<typename T> class VABMap

typedef VABMap<VB1i8>   VABMap1i8 ,VABMap1b;
typedef VABMap<VB1i16>  VABMap1i16,VABMap1s;
typedef VABMap<VB1i32>  VABMap1i32,VABMap1i;
typedef VABMap<VB1u8>   VABMap1u8 ,VABMap1ub;
typedef VABMap<VB1u16>  VABMap1u16,VABMap1us;
typedef VABMap<VB1u32>  VABMap1u32,VABMap1ui;
typedef VABMap<VB1f>    VABMap1f;
typedef VABMap<VB1d>    VABMap1d;

typedef VABMap<VB2i8>   VABMap2i8 ,VABMap2b;
typedef VABMap<VB2i16>  VABMap2i16,VABMap2s;
typedef VABMap<VB2i32>  VABMap2i32,VABMap2i;
typedef VABMap<VB2u8>   VABMap2u8 ,VABMap2ub;
typedef VABMap<VB2u16>  VABMap2u16,VABMap2us;
typedef VABMap<VB2u32>  VABMap2u32,VABMap2ui;
typedef VABMap<VB2f>    VABMap2f;
typedef VABMap<VB2d>    VABMap2d;

typedef VABMap<VB3i32>  VABMap3i32,VABMap3i;
typedef VABMap<VB3u32>  VABMap3u32,VABMap3ui;
typedef VABMap<VB3f>    VABMap3f;
typedef VABMap<VB3d>    VABMap3d;

typedef VABMap<VB4i8>   VABMap4i8 ,VABMap4b;
typedef VABMap<VB4i16>  VABMap4i16,VABMap4s;
typedef VABMap<VB4i32>  VABMap4i32,VABMap4i;
typedef VABMap<VB4u8>   VABMap4u8, VABMap4ub;
typedef VABMap<VB4u16>  VABMap4u16,VABMap4us;
typedef VABMap<VB4u32>  VABMap4u32,VABMap4ui;
typedef VABMap<VB4f>    VABMap4f;
typedef VABMap<VB4d>    VABMap4d;

VK_NAMESPACE_END