119 lines
4.5 KiB
C++
119 lines
4.5 KiB
C++
#pragma once
|
||
|
||
#include<hgl/graph/VK.h>
|
||
#include<hgl/graph/VKDevice.h>
|
||
#include<hgl/type/IDName.h>
|
||
|
||
VK_NAMESPACE_BEGIN
|
||
|
||
class RenderCmdBuffer;
|
||
class GraphModule;
|
||
|
||
class GraphModuleManager
|
||
{
|
||
GPUDevice *device;
|
||
|
||
Map<AnsiIDName,GraphModule *> graph_module_map;
|
||
|
||
public:
|
||
|
||
GraphModuleManager(GPUDevice *dev){device=dev;}
|
||
~GraphModuleManager();
|
||
|
||
GPUDevice * GetDevice ()noexcept{return device;} ///<取得GPU设备
|
||
VkDevice GetVkDevice (){return device->GetDevice();}
|
||
const GPUPhysicalDevice * GetPhysicalDevice (){return device->GetPhysicalDevice();} ///<取得物理设备
|
||
const GPUDeviceAttribute *GetDeviceAttribute (){return device->GetDeviceAttribute();} ///<取得设备属性
|
||
|
||
/**
|
||
* 获取指定名称的模块
|
||
* @param create 如果不存在，是否创建新的
|
||
*/
|
||
GraphModule *GetModule(const AnsiIDName &name,bool create); ///<获取指定名称的模块
|
||
|
||
/**
|
||
* 获取指定类型的模块
|
||
* @param create 如果不存在，是否创建新的
|
||
*/
|
||
template<typename T>
|
||
T *GetModule(bool create=false){return (T *)GetModule(T::GetModuleName(),create);} ///<获取指定类型的模块
|
||
|
||
void ReleaseModule(GraphModule *); ///<释放指定模块
|
||
|
||
const bool IsLoaded(const AnsiIDName &name){return graph_module_map.ContainsKey(name);} ///<是否已经加载了指定类型的模块
|
||
|
||
template<typename T>
|
||
const bool IsLoaded(){return graph_module_map.ContainsKey(T::GetModuleName());} ///<是否已经加载了指定类型的模块
|
||
|
||
public: //事件
|
||
|
||
void OnResize(const VkExtent2D &);
|
||
};//class GraphModuleManager
|
||
|
||
GraphModuleManager *GetGraphModuleManager(GPUDevice *);
|
||
|
||
class GraphModule
|
||
{
|
||
GraphModuleManager *module_manager;
|
||
|
||
AnsiIDName module_name;
|
||
|
||
bool module_enable;
|
||
bool module_ready;
|
||
|
||
protected:
|
||
|
||
virtual void SetModuleEnable(bool e){module_enable=e;}
|
||
virtual void SetModuleReady(bool r){module_ready=r;}
|
||
|
||
public:
|
||
|
||
virtual const bool IsRender(){return false;} ///<是否为渲染模块
|
||
|
||
GraphModuleManager *GetManager (){return module_manager;} ///<取得模块管理器
|
||
GPUDevice * GetDevice (){return module_manager->GetDevice();} ///<取得GPU设备
|
||
VkDevice GetVkDevice (){return module_manager->GetVkDevice();} ///<取得VkDevice
|
||
const GPUPhysicalDevice * GetPhysicalDevice (){return module_manager->GetPhysicalDevice();} ///<取得物理设备
|
||
const GPUDeviceAttribute *GetDeviceAttribute (){return module_manager->GetDeviceAttribute();}///<取得设备属性
|
||
|
||
static const AnsiIDName *GetModuleName(){return nullptr;} ///<取得模块名称(标准通用的名称，比如Upscale，供通用模块使用)
|
||
virtual const AnsiIDName *GetName()const{return &module_name;} ///<取得名称(完整的私有名称，比如FSR3Upscale,DLSS3Upscale)
|
||
|
||
const bool IsEnable ()const noexcept{return module_enable;} ///<当前模块是否启用
|
||
const bool IsReady ()const noexcept{return module_ready;} ///<当前模块是否准备好
|
||
|
||
public:
|
||
|
||
NO_COPY_NO_MOVE(GraphModule)
|
||
|
||
GraphModule(GraphModuleManager *gmm,const AnsiIDName &name)
|
||
{
|
||
module_manager=gmm;
|
||
module_name=name;
|
||
}
|
||
virtual ~GraphModule()=default;
|
||
|
||
virtual bool Init(){return true;} ///<初始化当前模块
|
||
|
||
public: //回调事件
|
||
|
||
virtual void OnRenderTarget(RenderTarget *){} ///<设置渲染目标
|
||
virtual void OnResize(const VkExtent2D &){} ///<窗口大小改变
|
||
|
||
virtual void OnPreFrame(){} ///<帧绘制前回调
|
||
virtual void OnExecute(const double,RenderCmdBuffer *){}
|
||
virtual void OnPostFrame(){} ///<帧绘制后回调
|
||
};//class GraphModule
|
||
|
||
#define GRAPH_MODULE_CONSTRUCT(name) public:\
|
||
NO_COPY_NO_MOVE(name##Module) \
|
||
static const AnsiIDName &GetModuleName() \
|
||
{ \
|
||
static const AnsiIDName id_name(#name); \
|
||
return id_name; \
|
||
} \
|
||
\
|
||
name##Module(GraphModuleManager *gmm):GraphModule(gmm,GetModuleName()){}
|
||
|
||
VK_NAMESPACE_END
|