Code/EnginePlugins/ProcGenPlugin/Components/ProcVolumeComponent.h - dev - redgpu/ezEngine

Code/EnginePlugins/ProcGenPlugin/Components/ProcVolumeComponent.h
136 строк5 KB
C-Core
Procedural spline volumes (#1686)
14 окт 2025, 23:32
14 окт 2025, 23:320208453
#pragma once

#include <Core/ResourceManager/ResourceHandle.h>
#include <Core/World/World.h>
#include <ProcGenPlugin/Declarations.h>

struct ezMsgTransformChanged;
struct ezMsgUpdateLocalBounds;
struct ezMsgExtractVolumes;

using ezImageDataResourceHandle = ezTypedResourceHandle<class ezImageDataResource>;

class EZ_PROCGENPLUGIN_DLL ezProcVolumeComponent : public ezComponent
{
  EZ_DECLARE_ABSTRACT_COMPONENT_TYPE(ezProcVolumeComponent, ezComponent);

public:
  ezProcVolumeComponent();
  ~ezProcVolumeComponent();

  virtual void OnActivated() override;
  virtual void OnDeactivated() override;

  void SetValue(float fValue);
  float GetValue() const { return m_fValue; }

  void SetSortOrder(float fOrder);
  float GetSortOrder() const { return m_fSortOrder; }

  void SetBlendMode(ezEnum<ezProcGenBlendMode> blendMode);
  ezEnum<ezProcGenBlendMode> GetBlendMode() const { return m_BlendMode; }

  virtual void SerializeComponent(ezWorldWriter& inout_stream) const override;
  virtual void DeserializeComponent(ezWorldReader& inout_stream) override;

  void OnTransformChanged(ezMsgTransformChanged& ref_msg);

  using AreaInvalidatedEvent = ezEvent<const ezProcGenInternal::InvalidatedArea&, ezMutex>;
  static const AreaInvalidatedEvent& GetAreaInvalidatedEvent() { return s_AreaInvalidatedEvent; }

protected:
  float m_fValue = 1.0f;
  float m_fSortOrder = 0.0f;
  ezEnum<ezProcGenBlendMode> m_BlendMode;

  void InvalidateArea();
  void InvalidateArea(const ezBoundingBox& area);

  static AreaInvalidatedEvent s_AreaInvalidatedEvent;
  static ezSpatialData::Category s_SpatialCategory;
};

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

using ezProcVolumeSphereComponentManager = ezComponentManager<class ezProcVolumeSphereComponent, ezBlockStorageType::Compact>;

class EZ_PROCGENPLUGIN_DLL ezProcVolumeSphereComponent : public ezProcVolumeComponent
{
  EZ_DECLARE_COMPONENT_TYPE(ezProcVolumeSphereComponent, ezProcVolumeComponent, ezProcVolumeSphereComponentManager);

public:
  ezProcVolumeSphereComponent();
  ~ezProcVolumeSphereComponent();

  float GetRadius() const { return m_fRadius; }
  void SetRadius(float fRadius);

  float GetFalloff() const { return m_fFalloff; }
  void SetFalloff(float fFalloff);

  virtual void SerializeComponent(ezWorldWriter& inout_stream) const override;
  virtual void DeserializeComponent(ezWorldReader& inout_stream) override;

  void OnUpdateLocalBounds(ezMsgUpdateLocalBounds& ref_msg) const;
  void OnExtractVolumes(ezMsgExtractVolumes& ref_msg) const;

protected:
  float m_fRadius = 5.0f;
  float m_fFalloff = 0.5f;
};

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

using ezProcVolumeBoxComponentManager = ezComponentManager<class ezProcVolumeBoxComponent, ezBlockStorageType::Compact>;

class EZ_PROCGENPLUGIN_DLL ezProcVolumeBoxComponent : public ezProcVolumeComponent
{
  EZ_DECLARE_COMPONENT_TYPE(ezProcVolumeBoxComponent, ezProcVolumeComponent, ezProcVolumeBoxComponentManager);

public:
  ezProcVolumeBoxComponent();
  ~ezProcVolumeBoxComponent();

  const ezVec3& GetExtents() const { return m_vExtents; }
  void SetExtents(const ezVec3& vExtents);

  const ezVec3& GetPositiveFalloff() const { return m_vPositiveFalloff; }
  void SetPositiveFalloff(const ezVec3& vFalloff);
  const ezVec3& GetNegativeFalloff() const { return m_vNegativeFalloff; }
  void SetNegativeFalloff(const ezVec3& vFalloff);

  virtual void SerializeComponent(ezWorldWriter& inout_stream) const override;
  virtual void DeserializeComponent(ezWorldReader& inout_stream) override;

  void OnUpdateLocalBounds(ezMsgUpdateLocalBounds& ref_msg) const;
  void OnExtractVolumes(ezMsgExtractVolumes& ref_msg) const;

protected:
  ezVec3 m_vExtents = ezVec3(10.0f);
  ezVec3 m_vPositiveFalloff = ezVec3(0.5f);
  ezVec3 m_vNegativeFalloff = ezVec3(0.5f);
};

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

using ezProcVolumeImageComponentManager = ezComponentManager<class ezProcVolumeImageComponent, ezBlockStorageType::Compact>;

class EZ_PROCGENPLUGIN_DLL ezProcVolumeImageComponent : public ezProcVolumeBoxComponent
{
  EZ_DECLARE_COMPONENT_TYPE(ezProcVolumeImageComponent, ezProcVolumeBoxComponent, ezProcVolumeImageComponentManager);

public:
  ezProcVolumeImageComponent();
  ~ezProcVolumeImageComponent();

  virtual void SerializeComponent(ezWorldWriter& inout_stream) const override;
  virtual void DeserializeComponent(ezWorldReader& inout_stream) override;

  void OnExtractVolumes(ezMsgExtractVolumes& ref_msg) const;

  void SetImage(const ezImageDataResourceHandle& hResource);
  ezImageDataResourceHandle GetImage() const { return m_hImage; }

protected:
  ezImageDataResourceHandle m_hImage; // [ property ]
};