WhiteBalancePass.cpp 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  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. #include <PostProcessing/WhiteBalancePass.h>
  9. #include <PostProcess/PostProcessFeatureProcessor.h>
  10. #include <Atom/RPI.Public/RenderPipeline.h>
  11. #include <Atom/RPI.Public/Scene.h>
  12. namespace AZ
  13. {
  14. namespace Render
  15. {
  16. RPI::Ptr<WhiteBalancePass> WhiteBalancePass::Create(const RPI::PassDescriptor& descriptor)
  17. {
  18. RPI::Ptr<WhiteBalancePass> pass = aznew WhiteBalancePass(descriptor);
  19. return AZStd::move(pass);
  20. }
  21. WhiteBalancePass::WhiteBalancePass(const RPI::PassDescriptor& descriptor)
  22. : RPI::ComputePass(descriptor)
  23. {
  24. }
  25. bool WhiteBalancePass::IsEnabled() const
  26. {
  27. if (!ComputePass::IsEnabled())
  28. {
  29. return false;
  30. }
  31. const RPI::Scene* scene = GetScene();
  32. if (!scene)
  33. {
  34. return false;
  35. }
  36. PostProcessFeatureProcessor* fp = scene->GetFeatureProcessor<PostProcessFeatureProcessor>();
  37. const RPI::ViewPtr view = GetRenderPipeline()->GetDefaultView();
  38. if (!fp)
  39. {
  40. return false;
  41. }
  42. PostProcessSettings* postProcessSettings = fp->GetLevelSettingsFromView(view);
  43. if (!postProcessSettings)
  44. {
  45. return false;
  46. }
  47. const WhiteBalanceSettings* whiteBalanceSettings = postProcessSettings->GetWhiteBalanceSettings();
  48. if (!whiteBalanceSettings)
  49. {
  50. return false;
  51. }
  52. return whiteBalanceSettings->GetEnabled();
  53. }
  54. void WhiteBalancePass::FrameBeginInternal(FramePrepareParams params)
  55. {
  56. // Must match the struct in WhiteBalance.azsl
  57. struct Constants
  58. {
  59. AZStd::array<u32, 2> m_outputSize;
  60. AZStd::array<float, 2> m_outputCenter;
  61. float m_temperature = WhiteBalance::DefaultTemperature;
  62. float m_tint = WhiteBalance::DefaultTint;
  63. AZStd::array<float, 2> m_pad;
  64. } constants{};
  65. RPI::Scene* scene = GetScene();
  66. PostProcessFeatureProcessor* fp = scene->GetFeatureProcessor<PostProcessFeatureProcessor>();
  67. if (fp)
  68. {
  69. RPI::ViewPtr view = scene->GetDefaultRenderPipeline()->GetDefaultView();
  70. PostProcessSettings* postProcessSettings = fp->GetLevelSettingsFromView(view);
  71. if (postProcessSettings)
  72. {
  73. WhiteBalanceSettings* whiteBalanceSettings = postProcessSettings->GetWhiteBalanceSettings();
  74. if (whiteBalanceSettings)
  75. {
  76. constants.m_temperature = whiteBalanceSettings->GetTemperature();
  77. constants.m_tint = whiteBalanceSettings->GetTint();
  78. }
  79. }
  80. }
  81. AZ_Assert(GetOutputCount() > 0, "WhiteBalancePass: No output bindings!");
  82. RPI::PassAttachment* outputAttachment = GetOutputBinding(0).GetAttachment().get();
  83. AZ_Assert(outputAttachment != nullptr, "WhiteBalancePass: Output binding has no attachment!");
  84. RHI::Size size = outputAttachment->m_descriptor.m_image.m_size;
  85. constants.m_outputSize[0] = size.m_width;
  86. constants.m_outputSize[1] = size.m_height;
  87. constants.m_outputCenter[0] = (size.m_width - 1) * 0.5f;
  88. constants.m_outputCenter[1] = (size.m_height -1) * 0.5f;
  89. m_shaderResourceGroup->SetConstant(m_constantsIndex, constants);
  90. RPI::ComputePass::FrameBeginInternal(params);
  91. }
  92. } // namespace Render
  93. } // namespace AZ