Merge branch 'shader_pipeline_irrlichtbaw' into irrlichtbaw-merge

Author: Hazardu

examples_tests/22.RaytracedAO/InstanceDataPerCamera.glsl

@@ -3,13 +3,35 @@
 
 #define MAX_ACCUMULATED_SAMPLES (1024*1024)
 
+#define WORKGROUP_SIZE 256
 
 #ifdef __cplusplus
-	#define mat4 nbl::core::matrix4SIMD
-	#define mat4x3 nbl::core::matrix3x4SIMD
+	#define uint uint32_t
+	struct uvec2
+	{
+		uint32_t x,y;
+	};
+	struct vec2
+	{
+		float x,y;
+	};
+	struct vec3
+	{
+		float x,y,z;
+	};
+	#define mat4 irr::core::matrix4SIMD
+	#define mat4x3 irr::core::matrix3x4SIMD
 #endif
 
 
+struct RaytraceShaderCommonData_t
+{
+	uvec2   imageDimensions;
+	uint    samplesPerPixelPerDispatch;
+	uint    samplesPerRowPerDispatch;
+};
+
+
 struct SLight
 {
 	#ifdef __cplusplus

examples_tests/22.RaytracedAO/common.glsl

@@ -0,0 +1,72 @@
+#version 430 core
+
+
+#include "drawCommon.glsl"
+layout(local_size_x = WORKGROUP_SIZE) in;
+
+
+#include <irr/builtin/glsl/utils/indirect_commands.glsl>
+layout(set=1, binding=0, std430, row_major) restrict readonly buffer PerInstanceStatic
+{
+    ObjectStaticData_t staticData[];
+};
+layout(set=1, binding=1, row_major) writeonly restrict buffer PerInstancePerCamera
+{
+    DrawData_t data[];
+} instanceDataPerCamera;
+layout(set=1, binding=2, std430, row_major) restrict readonly buffer PerInstanceCull
+{
+    CullData_t cullData[];
+};
+layout(set=1, binding=3, std430) restrict coherent buffer IndirectDraws
+{
+    irr_glsl_DrawElementsIndirectCommand_t draws[];
+} commandBuff[2];
+
+
+
+layout(push_constant, row_major) uniform PushConstants
+{
+    CullShaderData_t data;
+} pc;
+
+
+
+#include <irr/builtin/glsl/utils/culling.glsl>
+#include <irr/builtin/glsl/utils/transform.glsl>
+
+
+// base instance remains unchanged
+// we just do atomic add on the instance count
+void main()
+{
+    uint globalObjectID = gl_GlobalInvocationID.x;
+    if (globalObjectID>=pc.data.maxObjectCount)
+        return;
+
+    const mat4x3 worldMatrix = cullData[globalObjectID].worldMatrix;
+    const uint drawID = cullData[globalObjectID].drawID;
+
+    // clear drawcount for next buffer
+    commandBuff[pc.data.currentCommandBufferIx^0x1u].draws[drawID].instanceCount = 0u;
+
+    // cull
+    const mat4 MVP = irr_glsl_pseudoMul4x4with4x3(pc.data.viewProjMatrix,worldMatrix);
+    bool notCulled = true;
+    if (false)
+    {
+        mat2x3 bbox;
+        bbox[0] = cullData[globalObjectID].aabbMinEdge;
+        bbox[1] = cullData[globalObjectID].aabbMaxEdge;
+        notCulled = irr_glsl_couldBeVisible(MVP,bbox);
+    }
+
+    if (notCulled)
+    {        
+        const uint instanceID = atomicAdd(commandBuff[pc.data.currentCommandBufferIx].draws[drawID].instanceCount,1u)+cullData[globalObjectID].baseInstance;
+        
+        instanceDataPerCamera.data[instanceID].MVP = MVP;
+        instanceDataPerCamera.data[instanceID].detMVP = pc.data.viewProjDeterminant*staticData[globalObjectID].detWorldMatrix;
+        instanceDataPerCamera.data[instanceID].objectID = globalObjectID;
+    }
+}