Code/EditorPlugins/ProcGen/EditorPluginProcGen/ProcGenGraphAsset/ProcGenNodes.h - dev - redgpu/ezEngine

Code/EditorPlugins/ProcGen/EditorPluginProcGen/ProcGenGraphAsset/ProcGenNodes.h
317 строк10 KB
Jan Krassnigg
Visual graph infrastructure cleanup (#1821)
19 фев 2026, 09:02
19 фев 2026, 09:027f7f12f
#pragma once

#include <EditorPluginProcGen/ProcGenGraphAsset/ProcGenNodePins.h>
#include <Foundation/CodeUtils/Expression/ExpressionAST.h>
#include <Foundation/Tracks/CurveEditData.h>
#include <Foundation/Types/TagSet.h>
#include <ProcGenPlugin/Resources/ProcGenGraphSharedData.h>

class ezProcGenNodeBase : public ezReflectedClass
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGenNodeBase, ezReflectedClass);

public:
  struct GraphContext
  {
    enum OutputType
    {
      Unknown,
      Placement,
      Color,
    };

    ezProcGenInternal::GraphSharedData m_SharedData;
    ezSmallArray<ezUInt8, 4> m_VolumeTagSetIndices;
    ezSmallArray<ezUInt8, 4> m_CurveIndices;
    OutputType m_OutputType = OutputType::Unknown;
  };

  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) = 0;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGenOutput : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGenOutput, ezProcGenNodeBase);

public:
  bool m_bActive = true;

  void Save(ezStreamWriter& inout_stream);
  void CopyValuesFromContext(const GraphContext& context);

  ezString m_sName;

  ezSmallArray<ezUInt8, 4> m_VolumeTagSetIndices;
  ezSmallArray<ezUInt8, 4> m_CurveIndices;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_PlacementOutput : public ezProcGenOutput
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_PlacementOutput, ezProcGenOutput);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  void Save(ezStreamWriter& inout_stream);

  ezHybridArray<ezString, 4> m_ObjectsToPlace;

  float m_fFootprint = 1.0f;

  ezVec3 m_vMinOffset = ezVec3(0);
  ezVec3 m_vMaxOffset = ezVec3(0);

  ezAngle m_YawRotationSnap = ezAngle::MakeFromRadian(0.0f);
  float m_fAlignToNormal = 1.0f;

  ezVec3 m_vMinScale = ezVec3(1);
  ezVec3 m_vMaxScale = ezVec3(1);

  float m_fCullDistance = 30.0f;

  ezUInt32 m_uiCollisionLayer = 0;

  ezString m_sSurface;

  ezString m_sColorGradient;

  ezEnum<ezProcPlacementMode> m_PlacementMode;
  ezUInt8 m_uiNumAdditionalRays = 4;
  float m_fRaySpread = 1.0f;

  ezEnum<ezProcPlacementPattern> m_PlacementPattern;

  ezProcGenNodeInputPin m_DensityPin;
  ezProcGenNodeInputPin m_ScalePin;
  ezProcGenNodeInputPin m_ColorIndexPin;
  ezProcGenNodeInputPin m_ObjectIndexPin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_VertexColorOutput : public ezProcGenOutput
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_VertexColorOutput, ezProcGenOutput);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  void Save(ezStreamWriter& inout_stream);

  ezProcGenNodeInputPin m_RPin;
  ezProcGenNodeInputPin m_GPin;
  ezProcGenNodeInputPin m_BPin;
  ezProcGenNodeInputPin m_APin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Random : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Random, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezInt32 m_iSeed = -1;

  float m_fOutputMin = 0.0f;
  float m_fOutputMax = 1.0f;

  ezProcGenNodeOutputPin m_OutputValuePin;

private:
  void OnObjectCreated(const ezAbstractObjectNode& node);

  ezUInt32 m_uiAutoSeed;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_PerlinNoise : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_PerlinNoise, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezVec3 m_Scale = ezVec3(10);
  ezVec3 m_Offset = ezVec3::MakeZero();
  ezUInt32 m_uiNumOctaves = 3;

  float m_fOutputMin = 0.0f;
  float m_fOutputMax = 1.0f;

  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Blend : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Blend, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezEnum<ezProcGenBinaryOperator> m_Operator;
  float m_fInputValueA = 1.0f;
  float m_fInputValueB = 1.0f;
  bool m_bClampOutput = false;

  ezProcGenNodeInputPin m_InputValueAPin;
  ezProcGenNodeInputPin m_InputValueBPin;
  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Remap : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Remap, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  float m_fInputMin = 0.0f;
  float m_fInputMax = 1.0f;
  float m_fOutputMin = 0.0f;
  float m_fOutputMax = 1.0f;
  bool m_bClampIntermediate = false;

  ezProcGenNodeInputPin m_InputValuePin;
  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Curve : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Curve, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezSingleCurveData m_CurveData;
  ezUInt32 m_uiNumSamples = 32;

  ezProcGenNodeInputPin m_InputValuePin;
  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Contrast : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Contrast, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  float m_fInputValue = 0.5f;
  float m_fContrast = 0.0f;

  ezProcGenNodeInputPin m_InputValuePin;
  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Height : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Height, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  float m_fMinHeight = 0.0f;
  float m_fMaxHeight = 1000.0f;
  float m_fLowerFade = 0.2f;
  float m_fUpperFade = 0.2f;

  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Slope : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Slope, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezAngle m_MinSlope = ezAngle::MakeFromDegree(0.0f);
  ezAngle m_MaxSlope = ezAngle::MakeFromDegree(30.0f);
  float m_fLowerFade = 0.0f;
  float m_fUpperFade = 0.2f;

  ezProcGenNodeOutputPin m_OutputValuePin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Position : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Position, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezProcGenNodeOutputPin m_XPin;
  ezProcGenNodeOutputPin m_YPin;
  ezProcGenNodeOutputPin m_ZPin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_Normal : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_Normal, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezProcGenNodeOutputPin m_XPin;
  ezProcGenNodeOutputPin m_YPin;
  ezProcGenNodeOutputPin m_ZPin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_MeshVertexColor : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_MeshVertexColor, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  ezProcGenNodeOutputPin m_RPin;
  ezProcGenNodeOutputPin m_GPin;
  ezProcGenNodeOutputPin m_BPin;
  ezProcGenNodeOutputPin m_APin;
};

//////////////////////////////////////////////////////////////////////////

class ezProcGen_ApplyVolumes : public ezProcGenNodeBase
{
  EZ_ADD_DYNAMIC_REFLECTION(ezProcGen_ApplyVolumes, ezProcGenNodeBase);

public:
  virtual ezExpressionAST::Node* GenerateExpressionASTNode(ezTempHashedString sOutputName, ezArrayPtr<ezExpressionAST::Node*> inputs, ezExpressionAST& out_ast, GraphContext& ref_context) override;

  float m_fInputValue = 0.0f;

  ezTagSet m_IncludeTags;

  ezEnum<ezProcVolumeImageMode> m_ImageVolumeMode;
  ezColorGammaUB m_RefColor;

  ezProcGenNodeInputPin m_InputValuePin;
  ezProcGenNodeOutputPin m_OutputValuePin;
};