426 lines
9.7 KiB
C++
426 lines
9.7 KiB
C++
#include<hgl/graph/MaterialRenderList.h>
|
||
#include<hgl/graph/Mesh.h>
|
||
#include<hgl/graph/VKDevice.h>
|
||
#include<hgl/graph/VKCommandBuffer.h>
|
||
#include<hgl/graph/VKVertexInput.h>
|
||
#include<hgl/graph/VKRenderAssign.h>
|
||
#include<hgl/util/sort/Sort.h>
|
||
#include"RenderAssignBuffer.h"
|
||
#include<hgl/graph/SceneNode.h>
|
||
#include<hgl/graph/CameraInfo.h>
|
||
#include<hgl/component/StaticMeshComponent.h>
|
||
|
||
VK_NAMESPACE_BEGIN
|
||
MaterialRenderList::MaterialRenderList(VulkanDevice *d,bool l2w,const RenderPipelineIndex &rpi)
|
||
{
|
||
device=d;
|
||
cmd_buf=nullptr;
|
||
rp_index=rpi;
|
||
|
||
camera_info=nullptr;
|
||
|
||
assign_buffer=new RenderAssignBuffer(device,rp_index.material);
|
||
|
||
icb_draw=nullptr;
|
||
icb_draw_indexed=nullptr;
|
||
|
||
ri_count=0;
|
||
|
||
vab_list=new VABList(rp_index.material->GetVertexInput()->GetCount());
|
||
|
||
last_data_buffer=nullptr;
|
||
last_vdm=nullptr;
|
||
last_render_data=nullptr;
|
||
|
||
first_indirect_draw_index=-1;
|
||
indirect_draw_count=0;
|
||
}
|
||
|
||
MaterialRenderList::~MaterialRenderList()
|
||
{
|
||
SAFE_CLEAR(icb_draw_indexed)
|
||
SAFE_CLEAR(icb_draw)
|
||
|
||
SAFE_CLEAR(vab_list);
|
||
SAFE_CLEAR(assign_buffer);
|
||
}
|
||
|
||
void MaterialRenderList::Add(StaticMeshComponent *smc)
|
||
{
|
||
if(!smc)
|
||
return;
|
||
|
||
RenderNode rn;
|
||
|
||
rn.index =rn_list.GetCount();
|
||
|
||
rn.sm_component =smc;
|
||
|
||
rn.l2w_version =smc->GetLocalToWorldMatrixVersion();
|
||
rn.l2w_index =0;
|
||
|
||
rn.world_position =smc->GetWorldPosition();
|
||
|
||
if(camera_info)
|
||
rn.to_camera_distance=length(camera_info->pos,rn.world_position);
|
||
else
|
||
rn.to_camera_distance=0;
|
||
|
||
rn_list.Add(rn);
|
||
}
|
||
|
||
void MaterialRenderList::End()
|
||
{
|
||
//排序
|
||
Sort(rn_list.GetArray());
|
||
|
||
const uint node_count=rn_list.GetCount();
|
||
|
||
if(node_count<=0)return;
|
||
|
||
Stat();
|
||
|
||
if(assign_buffer)
|
||
assign_buffer->WriteNode(rn_list);
|
||
}
|
||
|
||
void MaterialRenderList::UpdateLocalToWorld()
|
||
{
|
||
if(!assign_buffer)
|
||
return;
|
||
|
||
rn_update_l2w_list.Clear();
|
||
|
||
const int node_count=rn_list.GetCount();
|
||
|
||
if(node_count<=0)return;
|
||
|
||
int first=-1,last=-1;
|
||
int update_count=0;
|
||
uint32 l2w_version=0;
|
||
RenderNode *rn=rn_list.GetData();
|
||
|
||
for(int i=0;i<node_count;i++)
|
||
{
|
||
l2w_version=rn->sm_component->GetLocalToWorldMatrixVersion();
|
||
|
||
if(rn->l2w_version!=l2w_version) //版本不对，需要更新
|
||
{
|
||
if(first==-1)
|
||
{
|
||
first=rn->l2w_index;
|
||
}
|
||
|
||
last=rn->l2w_index;
|
||
|
||
rn->l2w_version=l2w_version;
|
||
|
||
rn_update_l2w_list.Add(rn);
|
||
|
||
++update_count;
|
||
}
|
||
|
||
++rn;
|
||
}
|
||
|
||
if(update_count>0)
|
||
{
|
||
assign_buffer->UpdateLocalToWorld(rn_update_l2w_list,first,last);
|
||
rn_update_l2w_list.Clear();
|
||
}
|
||
}
|
||
|
||
void MaterialRenderList::UpdateMaterialInstance(StaticMeshComponent *smc)
|
||
{
|
||
if(!smc)return;
|
||
|
||
if(!assign_buffer)
|
||
return;
|
||
|
||
const int node_count=rn_list.GetCount();
|
||
|
||
if(node_count<=0)return;
|
||
RenderNode *rn=rn_list.GetData();
|
||
|
||
for(int i=0;i<node_count;i++)
|
||
{
|
||
if(rn->sm_component==smc)
|
||
{
|
||
assign_buffer->UpdateMaterialInstance(rn);
|
||
return;
|
||
}
|
||
|
||
++rn;
|
||
}
|
||
}
|
||
|
||
void MaterialRenderList::RenderItem::Set(Mesh *ri)
|
||
{
|
||
mi =ri->GetMaterialInstance();
|
||
pdb =ri->GetDataBuffer();
|
||
prd =ri->GetRenderData();
|
||
}
|
||
|
||
void MaterialRenderList::ReallocICB()
|
||
{
|
||
const uint32_t icb_new_count=power_to_2(rn_list.GetCount());
|
||
|
||
if(icb_draw)
|
||
{
|
||
if(icb_new_count<=icb_draw->GetMaxCount())
|
||
return;
|
||
|
||
delete icb_draw;
|
||
icb_draw=nullptr;
|
||
|
||
delete icb_draw_indexed;
|
||
icb_draw_indexed=nullptr;
|
||
}
|
||
|
||
icb_draw=device->CreateIndirectDrawBuffer(icb_new_count);
|
||
icb_draw_indexed=device->CreateIndirectDrawIndexedBuffer(icb_new_count);
|
||
}
|
||
|
||
void MaterialRenderList::WriteICB(VkDrawIndirectCommand *dicp,RenderItem *ri)
|
||
{
|
||
dicp->vertexCount =ri->prd->vertex_count;
|
||
dicp->instanceCount =ri->instance_count;
|
||
dicp->firstVertex =ri->prd->vertex_offset;
|
||
dicp->firstInstance =ri->first_instance;
|
||
}
|
||
|
||
void MaterialRenderList::WriteICB(VkDrawIndexedIndirectCommand *diicp,RenderItem *ri)
|
||
{
|
||
diicp->indexCount =ri->prd->index_count;
|
||
diicp->instanceCount=ri->instance_count;
|
||
diicp->firstIndex =ri->prd->first_index;
|
||
diicp->vertexOffset =ri->prd->vertex_offset;
|
||
diicp->firstInstance=ri->first_instance;
|
||
}
|
||
|
||
void MaterialRenderList::Stat()
|
||
{
|
||
const uint count=rn_list.GetCount();
|
||
RenderNode *rn=rn_list.GetData();
|
||
|
||
ReallocICB();
|
||
|
||
VkDrawIndirectCommand *dicp=icb_draw->MapCmd();
|
||
VkDrawIndexedIndirectCommand *diicp=icb_draw_indexed->MapCmd();
|
||
|
||
ri_array.Clear();
|
||
ri_array.Alloc(count);
|
||
|
||
RenderItem *ri=ri_array.GetData();
|
||
Mesh *ro=rn->sm_component->GetMesh();
|
||
|
||
ri_count=1;
|
||
|
||
ri->first_instance=0;
|
||
ri->instance_count=1;
|
||
ri->Set(ro);
|
||
|
||
last_data_buffer=ri->pdb;
|
||
last_vdm =ri->pdb->vdm;
|
||
last_render_data=ri->prd;
|
||
|
||
++rn;
|
||
|
||
for(uint i=1;i<count;i++)
|
||
{
|
||
ro=rn->sm_component->GetMesh();
|
||
|
||
if(*last_data_buffer==*ro->GetDataBuffer())
|
||
if(*last_render_data==*ro->GetRenderData())
|
||
{
|
||
++ri->instance_count;
|
||
++rn;
|
||
continue;
|
||
}
|
||
|
||
if(ri->pdb->vdm)
|
||
{
|
||
if(ri->pdb->ibo)
|
||
WriteICB(diicp,ri);
|
||
else
|
||
WriteICB(dicp,ri);
|
||
|
||
++dicp;
|
||
++diicp;
|
||
}
|
||
|
||
++ri_count;
|
||
++ri;
|
||
|
||
ri->first_instance=i;
|
||
ri->instance_count=1;
|
||
ri->Set(ro);
|
||
|
||
last_data_buffer=ri->pdb;
|
||
last_vdm =ri->pdb->vdm;
|
||
last_render_data=ri->prd;
|
||
|
||
++rn;
|
||
}
|
||
|
||
if(ri->pdb->vdm)
|
||
{
|
||
if(ri->pdb->ibo)
|
||
WriteICB(diicp,ri);
|
||
else
|
||
WriteICB(dicp,ri);
|
||
}
|
||
|
||
icb_draw->Unmap();
|
||
icb_draw_indexed->Unmap();
|
||
}
|
||
|
||
bool MaterialRenderList::BindVAB(const MeshDataBuffer *pdb,const uint ri_index)
|
||
{
|
||
//binding号都是在VertexInput::CreateVIL时连续紧密排列生成的，所以bind时first_binding写0就行了。
|
||
|
||
//const VIL *vil=last_vil;
|
||
|
||
//if(vil->GetCount(VertexInputGroup::Basic)!=prb->vab_count)
|
||
// return(false); //这里基本不太可能，因为CreateMesh时就会检查值是否一样
|
||
|
||
vab_list->Restart();
|
||
|
||
//Basic组，它所有的VAB信息均来自于Primitive，由vid参数传递进来
|
||
{
|
||
if(!vab_list->Add(pdb->vab_list,pdb->vab_offset,pdb->vab_count))
|
||
{
|
||
//这个情况很严重哦！
|
||
return(false);
|
||
}
|
||
}
|
||
|
||
if (assign_buffer) //L2W/MI分发组
|
||
{
|
||
if(!vab_list->Add(assign_buffer->GetVAB(),0))//ASSIGN_VAB_STRIDE_BYTES*ri_index);
|
||
{
|
||
//一般出现这个情况是因为材质中没有配置需要L2W
|
||
return(false);
|
||
}
|
||
}
|
||
|
||
//if(!vab_list.IsFull()) //Joint组，暂未支持
|
||
//{
|
||
// const uint joint_id_binding_count=vil->GetCount(VertexInputGroup::JointID);
|
||
|
||
// if(joint_id_binding_count>0) //有矩阵信息
|
||
// {
|
||
// count+=joint_id_binding_count;
|
||
|
||
// if(count<vab_count) //JointWeight组
|
||
// {
|
||
// const uint joing_weight_binding_count=vil->GetCount(VertexInputGroup::JointWeight);
|
||
|
||
// if(joing_weight_binding_count!=1)
|
||
// {
|
||
// ++count;
|
||
// }
|
||
// else //JointWieght不为1是个bug，除非未来支持8权重
|
||
// {
|
||
// return(false);
|
||
// }
|
||
// }
|
||
// else //有JointID没有JointWeight? 这是个BUG
|
||
// {
|
||
// return(false);
|
||
// }
|
||
// }
|
||
//}
|
||
|
||
//if(count!=vab_count)
|
||
//{
|
||
// //还有没支持的绑定组？？？？
|
||
|
||
// return(false);
|
||
//}
|
||
|
||
cmd_buf->BindVAB(vab_list);
|
||
|
||
return(true);
|
||
}
|
||
|
||
void MaterialRenderList::ProcIndirectRender()
|
||
{
|
||
if(last_data_buffer->ibo)
|
||
icb_draw_indexed->DrawIndexed(*cmd_buf,first_indirect_draw_index,indirect_draw_count);
|
||
else
|
||
icb_draw->Draw(*cmd_buf,first_indirect_draw_index,indirect_draw_count);
|
||
|
||
first_indirect_draw_index=-1;
|
||
indirect_draw_count=0;
|
||
}
|
||
|
||
bool MaterialRenderList::Render(RenderItem *ri)
|
||
{
|
||
if(!last_data_buffer||*(ri->pdb)!=*last_data_buffer) //换buf了
|
||
{
|
||
if(indirect_draw_count) //如果有间接绘制的数据，赶紧给画了
|
||
ProcIndirectRender();
|
||
|
||
last_data_buffer=ri->pdb;
|
||
last_render_data=nullptr;
|
||
|
||
if(!BindVAB(ri->pdb,ri->first_instance))
|
||
{
|
||
//这个问题很严重哦
|
||
return(false);
|
||
}
|
||
|
||
if(ri->pdb->ibo)
|
||
cmd_buf->BindIBO(ri->pdb->ibo);
|
||
}
|
||
|
||
if(ri->pdb->vdm)
|
||
{
|
||
if(indirect_draw_count==0)
|
||
first_indirect_draw_index=ri->first_instance;
|
||
|
||
++indirect_draw_count;
|
||
}
|
||
else
|
||
{
|
||
cmd_buf->Draw(ri->pdb,ri->prd,ri->instance_count,ri->first_instance);
|
||
}
|
||
|
||
return(true);
|
||
}
|
||
|
||
void MaterialRenderList::Render(RenderCmdBuffer *rcb)
|
||
{
|
||
if(!rcb)return;
|
||
const uint count=rn_list.GetCount();
|
||
|
||
if(count<=0)return;
|
||
|
||
if(ri_count<=0)return;
|
||
|
||
cmd_buf=rcb;
|
||
|
||
cmd_buf->BindPipeline(rp_index.pipeline);
|
||
|
||
last_data_buffer=nullptr;
|
||
last_vdm =nullptr;
|
||
last_render_data=nullptr;
|
||
|
||
if(assign_buffer)
|
||
assign_buffer->Bind(rp_index.material);
|
||
|
||
cmd_buf->BindDescriptorSets(rp_index.material);
|
||
|
||
RenderItem *ri=ri_array.GetData();
|
||
for(uint i=0;i<ri_count;i++)
|
||
{
|
||
Render(ri);
|
||
++ri;
|
||
}
|
||
|
||
if(indirect_draw_count) //如果有间接绘制的数据，赶紧给画了
|
||
ProcIndirectRender();
|
||
}
|
||
VK_NAMESPACE_END
|