90 lines
3.2 KiB
C++
90 lines
3.2 KiB
C++
#ifndef HGL_GRAPH_SHADER_NODE_INCLUDE
|
||
#define HGL_GRAPH_SHADER_NODE_INCLUDE
|
||
|
||
#include<hgl/type/StringList.h>
|
||
#include<hgl/type/List.h>
|
||
#include<hgl/type/Map.h>
|
||
#include<hgl/graph/shader/node/type.h>
|
||
#include<hgl/graph/shader/param/in.h>
|
||
#include<hgl/graph/shader/param/out.h>
|
||
|
||
BEGIN_SHADER_NODE_NAMESPACE
|
||
|
||
#define SHADER_INPUT_PARAM(mj,name,type) AddInput(mj,#name,SHADER_PARAM_NAMESPACE::ParamType::type);
|
||
#define SHADER_OUTPUT_PARAM(name,type) AddOutput(#name,SHADER_PARAM_NAMESPACE::ParamType::type);
|
||
|
||
using InputParamList=ObjectList<param::InputParam>;
|
||
using OutputParamList=ObjectList<param::OutputParam>;
|
||
using InputParamMapByName=Map<UTF8String,param::InputParam *>;
|
||
using OutputParamMapByName=Map<UTF8String,param::OutputParam *>;
|
||
|
||
/**
|
||
* Shader 节点是所有Shader的基础，它可以是一个数据转接(纹理或流)，也可以是一个纯粹的计算公式
|
||
*/
|
||
class Node
|
||
{
|
||
NodeType node_type;
|
||
|
||
protected:
|
||
|
||
UTF8String node_name; ///<节点用户自定义名称
|
||
|
||
InputParamList input_params;
|
||
OutputParamList output_params;
|
||
|
||
InputParamMapByName input_params_by_name;
|
||
OutputParamMapByName output_params_by_name;
|
||
|
||
protected:
|
||
|
||
virtual bool GenInputParamCode(UTF8StringList &);
|
||
virtual bool GenOutputParamCode(UTF8StringList &){return true;}
|
||
|
||
protected:
|
||
|
||
param::InputParam * AddInput (bool mj,const UTF8String &n,const param::ParamType &pt);
|
||
param::OutputParam *AddOutput (const UTF8String &n,const param::ParamType &pt);
|
||
|
||
public:
|
||
|
||
param::InputParam * GetInput (const UTF8String &n) ///<根据名称获取输入参数
|
||
{return GetListObject(input_params_by_name,n);}
|
||
|
||
param::OutputParam *GetOutput (const UTF8String &n) ///<根据名称获取输出参数
|
||
{return GetListObject(output_params_by_name,n);}
|
||
|
||
public:
|
||
|
||
Node(const NodeType &nt){node_type=nt;}
|
||
Node(const NodeType &nt,const UTF8String &n){node_type=nt;node_name=n;}
|
||
virtual ~Node()=default;
|
||
|
||
const NodeType GetNodeType ()const{return node_type;}
|
||
|
||
const UTF8String & GetNodeName ()const{return node_name;}
|
||
void SetNodeName (const UTF8String &n){node_name=n;}
|
||
|
||
public: //参数相关
|
||
|
||
InputParamList & GetInputParamList (){return input_params;}
|
||
OutputParamList & GetOutputParamList (){return output_params;}
|
||
|
||
virtual bool JoinInput (const UTF8String &,node::Node *,param::OutputParam *);
|
||
|
||
public: //参数相关
|
||
|
||
virtual bool IsOutput(const param::OutputParam *) const;
|
||
|
||
virtual bool Check(); ///<检测当前节点是否可用
|
||
|
||
virtual bool GetInputParamName(UTF8String &result,const param::InputParam *);
|
||
virtual bool GetOutputParamName(UTF8String &result,const param::OutputParam *);
|
||
|
||
public: //产生代码相关
|
||
|
||
virtual bool GenTempValueDefine(UTF8StringList &); ///<产生临时变量定义
|
||
virtual bool GenCode(UTF8StringList &);
|
||
};//class Node
|
||
END_SHADER_NODE_NAMESPACE
|
||
#endif//HGL_GRAPH_SHADER_NODE_INCLUDE
|