189 lines
4.1 KiB
C++
189 lines
4.1 KiB
C++
// SceneTree
|
||
// 用于测试树形排列的场景中，每一级节点对变换矩阵的处理是否正确
|
||
|
||
#include"VulkanAppFramework.h"
|
||
#include<hgl/filesystem/FileSystem.h>
|
||
#include<hgl/graph/InlineGeometry.h>
|
||
#include<hgl/graph/RenderList.h>
|
||
#include<hgl/Time.h>
|
||
|
||
using namespace hgl;
|
||
using namespace hgl::graph;
|
||
|
||
constexpr uint32_t SCREEN_WIDTH=1280;
|
||
constexpr uint32_t SCREEN_HEIGHT=720;
|
||
|
||
class TestApp:public CameraAppFramework
|
||
{
|
||
struct
|
||
{
|
||
Color4f diffuse; //虽然shader中写vec3，但这里依然用Color4f
|
||
Color4f abiment;
|
||
}color_material;
|
||
|
||
Vector3f sun_direction;
|
||
|
||
private:
|
||
|
||
double start_time;
|
||
|
||
SceneNode render_root;
|
||
RenderList *render_list;
|
||
|
||
Material * material =nullptr;
|
||
MaterialInstance * material_instance =nullptr;
|
||
|
||
GPUBuffer * ubo_color =nullptr;
|
||
GPUBuffer * ubo_sun =nullptr;
|
||
|
||
Primitive * renderable_object =nullptr;
|
||
|
||
Pipeline * pipeline =nullptr;
|
||
|
||
public:
|
||
|
||
TestApp()
|
||
{
|
||
start_time=GetDoubleTime();
|
||
}
|
||
|
||
~TestApp()
|
||
{
|
||
SAFE_CLEAR(render_list);
|
||
}
|
||
|
||
private:
|
||
|
||
bool InitMaterial()
|
||
{
|
||
material=db->CreateMaterial(OS_TEXT("res/material/PhongFullyRough"));
|
||
if(!material)
|
||
return(false);
|
||
|
||
material_instance=db->CreateMaterialInstance(material);
|
||
|
||
return(true);
|
||
}
|
||
|
||
void CreateRenderObject()
|
||
{
|
||
renderable_object=CreateRenderableSphere(db,material_instance->GetVAB(),40);
|
||
}
|
||
|
||
bool InitUBO()
|
||
{
|
||
color_material.diffuse.Set(1,1,1);
|
||
color_material.abiment.Set(0.25,0.25,0.25);
|
||
|
||
ubo_color=db->CreateUBO(sizeof(color_material),&color_material);
|
||
if(!ubo_color)return(false);
|
||
|
||
sun_direction=normalized(Vector3f(rand(),rand(),rand()));
|
||
|
||
ubo_sun=db->CreateUBO(sizeof(sun_direction),&sun_direction);
|
||
if(!ubo_sun)return(false);
|
||
|
||
{
|
||
MaterialParameters *mp=material_instance->GetMP(DescriptorSetsType::Value);
|
||
|
||
if(!mp)return(false);
|
||
|
||
mp->BindUBO("material",ubo_color);
|
||
mp->BindUBO("sun",ubo_sun);
|
||
|
||
mp->Update();
|
||
}
|
||
|
||
BindCameraUBO(material_instance);
|
||
return(true);
|
||
}
|
||
|
||
bool InitPipeline()
|
||
{
|
||
pipeline=CreatePipeline(material_instance,InlinePipeline::Solid3D,Prim::Triangles);
|
||
|
||
return pipeline;
|
||
}
|
||
|
||
bool InitScene()
|
||
{
|
||
SceneNode *cur_node;
|
||
|
||
uint count;
|
||
float size;
|
||
|
||
Renderable *ri=db->CreateRenderable(renderable_object,material_instance,pipeline);
|
||
|
||
for(uint i=0;i<360;i++)
|
||
{
|
||
size=(i+1)/100.0f;
|
||
|
||
cur_node=render_root.CreateSubNode( rotate(i/5.0f,camera->world_up)*
|
||
translate(i/4.0f,0,0)*
|
||
scale(size));
|
||
|
||
count=(rand()%16)+1;
|
||
|
||
for(uint n=0;n<count;n++)
|
||
cur_node->CreateSubNode(translate(0,0,size*n*1.1),ri);
|
||
}
|
||
|
||
render_root.RefreshMatrix();
|
||
render_list->Expend(camera->info,&render_root);
|
||
return(true);
|
||
}
|
||
|
||
public:
|
||
|
||
bool Init()
|
||
{
|
||
if(!CameraAppFramework::Init(SCREEN_WIDTH,SCREEN_HEIGHT))
|
||
return(false);
|
||
|
||
render_list=new RenderList(device);
|
||
|
||
if(!InitMaterial())
|
||
return(false);
|
||
|
||
CreateRenderObject();
|
||
|
||
if(!InitUBO())
|
||
return(false);
|
||
|
||
if(!InitPipeline())
|
||
return(false);
|
||
|
||
if(!InitScene())
|
||
return(false);
|
||
|
||
return(true);
|
||
}
|
||
|
||
void Draw() override
|
||
{
|
||
CameraAppFramework::Draw();
|
||
|
||
Matrix4f rot=rotate(GetDoubleTime()-start_time,camera->world_up);
|
||
|
||
render_root.RefreshMatrix(&rot);
|
||
render_list->Expend(GetCameraInfo(),&render_root);
|
||
}
|
||
|
||
void BuildCommandBuffer(uint32 index)
|
||
{
|
||
VulkanApplicationFramework::BuildCommandBuffer(index,render_list);
|
||
}
|
||
};//class TestApp:public CameraAppFramework
|
||
|
||
int main(int,char **)
|
||
{
|
||
TestApp app;
|
||
|
||
if(!app.Init())
|
||
return(-1);
|
||
|
||
while(app.Run());
|
||
|
||
return 0;
|
||
}
|