DiffuseProbeGridVisualizationAccelerationStructurePass.h 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. /*
  2. * Copyright (c) Contributors to the Open 3D Engine Project.
  3. * For complete copyright and license terms please see the LICENSE at the root of this distribution.
  4. *
  5. * SPDX-License-Identifier: Apache-2.0 OR MIT
  6. *
  7. */
  8. #pragma once
  9. #include <Atom/RHI/ScopeProducer.h>
  10. #include <Atom/RPI.Public/Pass/Pass.h>
  11. #include <Atom/RPI.Public/Buffer/Buffer.h>
  12. #include <Atom/RHI/RayTracingBufferPools.h>
  13. namespace AZ
  14. {
  15. namespace Render
  16. {
  17. class DiffuseProbeGrid;
  18. //! This pass builds the DiffuseProbeGrid visualization acceleration structure
  19. class DiffuseProbeGridVisualizationAccelerationStructurePass final
  20. : public RPI::Pass
  21. , public RHI::ScopeProducer
  22. {
  23. public:
  24. AZ_RPI_PASS(DiffuseProbeGridVisualizationAccelerationStructurePass);
  25. AZ_RTTI(DiffuseProbeGridVisualizationAccelerationStructurePass, "{103D8917-D4DC-4CA3-BFB4-CD62846D282A}", Pass);
  26. AZ_CLASS_ALLOCATOR(DiffuseProbeGridVisualizationAccelerationStructurePass, SystemAllocator);
  27. //! Creates a DiffuseProbeGridVisualizationAccelerationStructurePass
  28. static RPI::Ptr<DiffuseProbeGridVisualizationAccelerationStructurePass> Create(const RPI::PassDescriptor& descriptor);
  29. ~DiffuseProbeGridVisualizationAccelerationStructurePass() = default;
  30. private:
  31. explicit DiffuseProbeGridVisualizationAccelerationStructurePass(const RPI::PassDescriptor& descriptor);
  32. bool ShouldUpdate(const AZStd::shared_ptr<DiffuseProbeGrid>& diffuseProbeGrid) const;
  33. // Scope producer functions
  34. void SetupFrameGraphDependencies(RHI::FrameGraphInterface frameGraph) override;
  35. void BuildCommandList(const RHI::FrameGraphExecuteContext& context) override;
  36. // Pass overrides
  37. bool IsEnabled() const override;
  38. void BuildInternal() override;
  39. void FrameBeginInternal(FramePrepareParams params) override;
  40. void FrameEndInternal() override;
  41. bool m_visualizationBlasBuilt = false;
  42. };
  43. } // namespace RPI
  44. } // namespace AZ