Autoexposure example restoration

26_Autoexposure/CMakeLists.txt

@@ -0,0 +1,33 @@
+
+include(common RESULT_VARIABLE RES)
+if(NOT RES)
+	message(FATAL_ERROR "common.cmake not found. Should be in {repo_root}/cmake directory")
+endif()
+
+set(NBL_INCLUDE_SERACH_DIRECTORIES
+	"${CMAKE_CURRENT_SOURCE_DIR}/include"
+)
+
+list(APPEND NBL_LIBRARIES
+	Nabla::ext::FullScreenTriangle
+)
+
+nbl_create_executable_project("" "" "${NBL_INCLUDE_SERACH_DIRECTORIES}" "${NBL_LIBRARIES}" "${NBL_EXECUTABLE_PROJECT_CREATION_PCH_TARGET}")
+
+if(NBL_EMBED_BUILTIN_RESOURCES)
+	set(_BR_TARGET_ ${EXECUTABLE_NAME}_builtinResourceData)
+	set(RESOURCE_DIR "app_resources")
+
+	get_filename_component(_SEARCH_DIRECTORIES_ "${CMAKE_CURRENT_SOURCE_DIR}" ABSOLUTE)
+	get_filename_component(_OUTPUT_DIRECTORY_SOURCE_ "${CMAKE_CURRENT_BINARY_DIR}/src" ABSOLUTE)
+	get_filename_component(_OUTPUT_DIRECTORY_HEADER_ "${CMAKE_CURRENT_BINARY_DIR}/include" ABSOLUTE)
+
+    file(GLOB_RECURSE BUILTIN_RESOURCE_FILES RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}/${RESOURCE_DIR}" CONFIGURE_DEPENDS "${CMAKE_CURRENT_SOURCE_DIR}/${RESOURCE_DIR}/*")
+    foreach(RES_FILE ${BUILTIN_RESOURCE_FILES})
+      LIST_BUILTIN_RESOURCE(RESOURCES_TO_EMBED "${RES_FILE}")
+    endforeach()
+
+	ADD_CUSTOM_BUILTIN_RESOURCES(${_BR_TARGET_} RESOURCES_TO_EMBED "${_SEARCH_DIRECTORIES_}" "${RESOURCE_DIR}" "nbl::this_example::builtin" "${_OUTPUT_DIRECTORY_HEADER_}" "${_OUTPUT_DIRECTORY_SOURCE_}")
+
+	LINK_BUILTIN_RESOURCES_TO_TARGET(${EXECUTABLE_NAME} ${_BR_TARGET_})
+endif()

26_Autoexposure/app_resources/avg_luma_meter.comp.hlsl

@@ -0,0 +1,71 @@
+// Copyright (C) 2024-2024 - DevSH Graphics Programming Sp. z O.O.
+// This file is part of the "Nabla Engine".
+// For conditions of distribution and use, see copyright notice in nabla.h
+
+#include "nbl/builtin/hlsl/luma_meter/geom_mean.hlsl"
+#include "nbl/builtin/hlsl/bda/bda_accessor.hlsl"
+#include "nbl/builtin/hlsl/colorspace/encodeCIEXYZ.hlsl"
+#include "app_resources/common.hlsl"
+
+[[vk::combinedImageSampler]] [[vk::binding(0, 0)]] Texture2D texture;
+[[vk::combinedImageSampler]] [[vk::binding(0, 0)]] SamplerState samplerState;
+
+using namespace nbl::hlsl;
+using Ptr = bda::__ptr < uint32_t >;
+using PtrAccessor = BdaAccessor < uint32_t >;
+
+[[vk::push_constant]] AutoexposurePushData pushData;
+
+groupshared float32_t sdata[WORKGROUP_SIZE];
+struct SharedAccessor
+{
+    using type = float32_t;
+    void get(const uint32_t index, NBL_REF_ARG(uint32_t) value)
+    {
+        value = sdata[index];
+    }
+
+    void set(const uint32_t index, const uint32_t value)
+    {
+        sdata[index] = value;
+    }
+
+    void workgroupExecutionAndMemoryBarrier()
+    {
+        glsl::barrier();
+    }
+};
+
+struct TexAccessor
+{
+    static float32_t3 toXYZ(float32_t3 srgbColor) {
+        return dot(colorspace::sRGBtoXYZ[1], srgbColor);
+    }
+
+    float32_t3 get(float32_t2 uv) {
+        return texture.SampleLevel(samplerState, uv, 0.f).rgb;
+    }
+};
+
+uint32_t3 glsl::gl_WorkGroupSize()
+{
+    return uint32_t3(SUBGROUP_SIZE, SUBGROUP_SIZE, 1);
+}
+
+[numthreads(SUBGROUP_SIZE, SUBGROUP_SIZE, 1)]
+[shader("compute")]
+void main(uint32_t3 ID : SV_GroupThreadID, uint32_t3 GroupID : SV_GroupID)
+{
+    const Ptr val_ptr = Ptr::create(pushData.lumaMeterBDA);
+    PtrAccessor val_accessor = PtrAccessor::create(val_ptr);
+
+    SharedAccessor sdata;
+    TexAccessor tex;
+
+    using LumaMeter = luma_meter::geom_meter< WORKGROUP_SIZE, SUBGROUP_SIZE, PtrAccessor, SharedAccessor, TexAccessor>;
+    LumaMeter meter = LumaMeter::create(pushData.lumaMin, pushData.lumaMax, pushData.sampleCount, pushData.rcpFirstPassWGCount);
+
+    uint32_t texWidth, texHeight;
+    texture.GetDimensions(texWidth, texHeight);
+    meter.sampleLuma(pushData.window, val_accessor, tex, sdata, (float32_t2)(glsl::gl_WorkGroupID() * glsl::gl_WorkGroupSize()), float32_t2(texWidth, texHeight));
+}

26_Autoexposure/app_resources/avg_luma_tonemap.comp.hlsl

@@ -0,0 +1,88 @@
+// Copyright (C) 2024-2024 - DevSH Graphics Programming Sp. z O.O.
+// This file is part of the "Nabla Engine".
+// For conditions of distribution and use, see copyright notice in nabla.h
+
+#include "nbl/builtin/hlsl/luma_meter/geom_mean.hlsl"
+#include "nbl/builtin/hlsl/bda/bda_accessor.hlsl"
+#include "nbl/builtin/hlsl/colorspace/EOTF.hlsl"
+#include "nbl/builtin/hlsl/colorspace/encodeCIEXYZ.hlsl"
+#include "nbl/builtin/hlsl/colorspace/decodeCIEXYZ.hlsl"
+#include "nbl/builtin/hlsl/colorspace/OETF.hlsl"
+#include "nbl/builtin/hlsl/tonemapper/operators/reinhard.hlsl"
+#include "app_resources/common.hlsl"
+
+[[vk::combinedImageSampler]] [[vk::binding(0, 0)]] Texture2D textureIn;
+[[vk::combinedImageSampler]] [[vk::binding(0, 0)]] SamplerState samplerStateIn;
+[[vk::binding(0, 3)]] RWTexture2D<float32_t4> textureOut;
+
+using namespace nbl::hlsl;
+using Ptr = bda::__ptr < uint32_t >;
+using PtrAccessor = BdaAccessor < uint32_t >;
+
+[[vk::push_constant]] AutoexposurePushData pushData;
+
+groupshared float32_t sdata[WORKGROUP_SIZE];
+struct SharedAccessor
+{
+    using type = float32_t;
+    void get(const uint32_t index, NBL_REF_ARG(uint32_t) value)
+    {
+        value = sdata[index];
+    }
+
+    void set(const uint32_t index, const uint32_t value)
+    {
+        sdata[index] = value;
+    }
+
+    void workgroupExecutionAndMemoryBarrier()
+    {
+        glsl::barrier();
+    }
+};
+
+struct TexAccessor
+{
+    static float32_t3 toXYZ(float32_t3 srgbColor) {
+        return dot(colorspace::sRGBtoXYZ[1], srgbColor);
+    }
+
+    float32_t3 get(float32_t2 uv) {
+        return textureIn.SampleLevel(samplerStateIn, uv, 0.f).rgb;
+    }
+};
+
+uint32_t3 glsl::gl_WorkGroupSize()
+{
+    return uint32_t3(SUBGROUP_SIZE, SUBGROUP_SIZE, 1);
+}
+
+[numthreads(SUBGROUP_SIZE, SUBGROUP_SIZE, 1)]
+[shader("compute")]
+void main(uint32_t3 ID : SV_GroupThreadID, uint32_t3 GroupID : SV_GroupID)
+{
+    const Ptr val_ptr = Ptr::create(pushData.lumaMeterBDA);
+    PtrAccessor val_accessor = PtrAccessor::create(val_ptr);
+
+    SharedAccessor sdata;
+    TexAccessor tex;
+
+    using LumaMeter = luma_meter::geom_meter< WORKGROUP_SIZE, SUBGROUP_SIZE, PtrAccessor, SharedAccessor, TexAccessor>;
+    LumaMeter meter = LumaMeter::create(pushData.lumaMin, pushData.lumaMax, pushData.sampleCount, pushData.rcpFirstPassWGCount);
+
+    float32_t EV = meter.gatherLuma(val_accessor);
+
+    uint32_t tid = workgroup::SubgroupContiguousIndex();
+    morton::code<false, 32, 2> mc;
+    mc.value = tid;
+    uint32_t2 coord = _static_cast<uint32_t2>(mc);
+
+    uint32_t2 pos = (glsl::gl_WorkGroupID() * glsl::gl_WorkGroupSize()).xy + coord;
+    float32_t2 uv = (float32_t2)(pos) / pushData.viewportSize;
+    float32_t3 color = colorspace::eotf::sRGB(tex.get(uv).rgb);
+    float32_t3 CIEColor = mul(colorspace::sRGBtoXYZ, color);
+    tonemapper::Reinhard<float32_t> reinhard = tonemapper::Reinhard<float32_t>::create(EV, 0.18f, 0.85f);
+    float32_t3 tonemappedColor = mul(colorspace::decode::XYZtoscRGB, reinhard(CIEColor));
+
+    textureOut[pos] = float32_t4(colorspace::oetf::sRGB(tonemappedColor), 1.0f);
+}