#pragma once
#include<hgl/graph/VKVertexAttribBuffer.h>

VK_NAMESPACE_BEGIN
// ubo_range大致分为三档:
//
//  16k: Mali-T系列或更早、Mali-G71为16k
// 
//  64k: 大部分手机与PC均为64k
// 
// >64k: Intel 核显与 PowerVR 为128MB，AMD显卡为4GB，可视为随显存无上限。
// 
// 我们使用uint8类型在vertex input中保存MaterialInstance ID，表示范围0-255。
// 所以MaterialInstance结构容量按16k/64k分为两个档次，64字节和256字节

// 如果一定要使用超过16K/64K硬件限制的容量，有两种办法
// 一、分多次渲染，使用UBO Offset偏移UBO数据区。
// 二、使用SSBO，但这样会导致性能下降，所以不推荐使用。

// 但我们不解决这个问题
// 我们天然要求将材质实例数据分为两个等级，同时要求一次渲染不能超过256种材质实例。
// 所以 UBO Range为16k时，实例数据不能超过64字节。UBO Range为64k时，实例数据不能超过256字节。

/*
* 渲染节点额外提供的数据
*/
struct RenderExtraBuffer
{
    uint node_count;                            ///<渲染节点数量
//    uint mi_count;                              ///<材质实例数量

    //uint mi_size;                               ///<单个材质实例数量长度
    //DeviceBuffer *mi_data_buffer;               ///<材质实例数据UBO/SSBO

    //VBO *mi_id;
    //VkBuffer mi_id_buffer;

//    VBO *bone_id,*bone_weight;
//    VkBuffer bone_id_buffer,bone_weight_buffer;

    VBO *l2w_vbo[4];
    VkBuffer l2w_buffer[4];

//------------------------------------------------------------

    //Assign UBO
    DeviceBuffer *assigns_l2w;
    DeviceBuffer *assigns_mi;
    
    //Assign VBO
    VBO *assigns_vbo;           ///<RG16UI格式的VertexInputStream,,,,X:L2W ID,Y:MI ID

    const uint assigns_vbo_strip=2;     ///<Assign VBO的每个节点的字节数

public:

    RenderExtraBuffer()
    {
        hgl_zero(*this);
    }

    ~RenderExtraBuffer()
    {
        Clear();
    }

    void ClearNode()
    {
        SAFE_CLEAR(assigns_l2w);
        SAFE_CLEAR(assigns_mi);
        SAFE_CLEAR(assigns_vbo);

        SAFE_CLEAR(l2w_vbo[0])
        SAFE_CLEAR(l2w_vbo[1])
        SAFE_CLEAR(l2w_vbo[2])
        SAFE_CLEAR(l2w_vbo[3])
        node_count=0;
    }

    //void ClearMI()
    //{
    //    SAFE_CLEAR(mi_id)
    //    SAFE_CLEAR(mi_data_buffer);
    //    mi_count=0;
    //    mi_size=0;
    //}

    void Clear()
    {
        ClearNode();
//        ClearMI();

//        SAFE_CLEAR(bone_id)
//        SAFE_CLEAR(bone_weight)
    }

    void NodeAlloc(GPUDevice *dev,const uint c)
    {
        ClearNode();
        node_count=power_to_2(c);

        for(uint i=0;i<4;i++)
        {
            l2w_vbo[i]=dev->CreateVBO(VF_V4F,node_count);
            l2w_buffer[i]=l2w_vbo[i]->GetBuffer();
        }

        assigns_l2w=dev->CreateUBO(node_count*sizeof(Matrix4f));
        //assigns_mi=dev->CreateUBO(node_count*sizeof(uint8));
        assigns_vbo=dev->CreateVBO(VF_V1U16,node_count);
    }

    //void MIAlloc(GPUDevice *dev,const uint c,const uint mis)
    //{
    //    ClearMI();
    //    if(c<=0||mi_size<=0)return;
    //
    //    mi_count=power_to_2(c);
    //    mi_size=mis;

    //    mi_id=dev->CreateVBO(VF_V1U8,mi_count);
    //    mi_id_buffer=mi_id->GetBuffer();

    //    mi_data_buffer=dev->CreateUBO(mi_count*mi_size);
    //}

    void WriteLocalToWorld(RenderNode *render_node,const uint count)
    {
        RenderNode *rn;

        //old l2w in vertex input stream
        {
            glm::vec4 *tp;

            for(uint col=0;col<4;col++)
            {
                tp=(glm::vec4 *)(l2w_vbo[col]->Map());

                rn=render_node;

                for(uint i=0;i<count;i++)
                {
                    *tp=rn->local_to_world[col];
                    ++tp;
                    ++rn;
                }

                l2w_vbo[col]->Unmap();
            }
        }

        //new l2w array in ubo
        {
            Matrix4f *tp=(hgl::Matrix4f *)(assigns_l2w->Map());
            uint16 *idp=(uint16 *)(assigns_vbo->Map());

            rn=render_node;

            for(uint i=0;i<count;i++)
            {
                *tp=rn->local_to_world;
                ++tp;

                *idp=i;
                ++idp;

                ++rn;
            }

            assigns_vbo->Unmap();
            assigns_l2w->Unmap();
        }
    }

    //void WriteMaterialInstance(RenderNode *render_node,const uint count,const MaterialInstanceSets &mi_set)
    //{
    //    //MaterialInstance ID
    //    {
    //        uint8 *tp=(uint8 *)(mi_id->Map());

    //        for(uint i=0;i<count;i++)
    //        {
    //            *tp=mi_set.Find(render_node->ri->GetMaterialInstance());
    //            ++tp;
    //            ++render_node;
    //        }
    //        mi_id->Unmap();
    //    }

    //    //MaterialInstance Data
    //    {
    //        //const uint count=mi_set.GetCount();

    //        //uint8 *tp=(uint8 *)(mi_data_buffer->Map());
    //        //const MaterialInstance **mi=mi_set.GetData();

    //        //for(uint i=0;i<count;i++)
    //        //{
    //        //    memcpy(tp,(*mi)->GetData(),mi_size);
    //
    //        //    ++mi;
    //        //    tp+=mi_size;
    //        //}

    //        //mi_data_buffer->Unmap();
    //    }
    //}
};//struct RenderExtraBuffer
VK_NAMESPACE_END