public void CreateShaderResources()
{
// Create the output resource. The dimensions and format should match the swap-chain
ResourceDescription resDesc = new ResourceDescription();
resDesc.DepthOrArraySize = 1;
resDesc.Dimension = ResourceDimension.Texture2D;
resDesc.Format = Format.R8G8B8A8_UNorm; // The backbuffer is actually DXGI_FORMAT_R8G8B8A8_UNORM_SRGB, but sRGB formats can't be used with UAVs. We will convert to sRGB ourselves in the shader
resDesc.Flags = ResourceFlags.AllowUnorderedAccess;
resDesc.Height = mSwapChainRect.Height;
resDesc.Layout = TextureLayout.Unknown;
resDesc.MipLevels = 1;
resDesc.SampleDescription = new SampleDescription(1, 0);
resDesc.Width = mSwapChainRect.Width;
mpOutputResource = mpDevice.CreateCommittedResource(AccelerationStructures.kDefaultHeapProps, HeapFlags.None, resDesc, ResourceStates.CopySource, null); // Starting as copy-source to simplify onFrameRender()
// Create an SRV/UAV descriptor heap. Need 2 entries - 1 SRV for the scene and 1 UAV for the output
mpSrvUavHeap = this.context.CreateDescriptorHeap(mpDevice, 2, DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView, true);
// Create the UAV. Based on the root signature we created it should be the first entry
UnorderedAccessViewDescription uavDesc = new UnorderedAccessViewDescription();
uavDesc.ViewDimension = UnorderedAccessViewDimension.Texture2D;
mpDevice.CreateUnorderedAccessView(mpOutputResource, null, uavDesc, mpSrvUavHeap.GetCPUDescriptorHandleForHeapStart());
// Create the TLAS SRV right after the UAV. Note that we are using a different SRV desc here
ShaderResourceViewDescription srvDesc = new ShaderResourceViewDescription();
srvDesc.ViewDimension = ShaderResourceViewDimension.RaytracingAccelerationStructure;
srvDesc.Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping;
srvDesc.RaytracingAccelerationStructure = new RaytracingAccelerationStructureShaderResourceView();
srvDesc.RaytracingAccelerationStructure.Location = mpTopLevelAS.GPUVirtualAddress;
CpuDescriptorHandle srvHandle = mpSrvUavHeap.GetCPUDescriptorHandleForHeapStart();
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
mpDevice.CreateShaderResourceView(null, srvDesc, srvHandle);
}
public void CreateShaderResources()
{
// Create the output resource. The dimensions and format should match the swap-chain
ResourceDescription resDesc = new ResourceDescription();
resDesc.DepthOrArraySize = 1;
resDesc.Dimension = ResourceDimension.Texture2D;
resDesc.Format = Format.R8G8B8A8_UNorm; // The backbuffer is actually DXGI_FORMAT_R8G8B8A8_UNORM_SRGB, but sRGB formats can't be used with UAVs. We will convert to sRGB ourselves in the shader
resDesc.Flags = ResourceFlags.AllowUnorderedAccess;
resDesc.Height = mSwapChainRect.Height;
resDesc.Layout = TextureLayout.Unknown;
resDesc.MipLevels = 1;
resDesc.SampleDescription = new SampleDescription(1, 0);
resDesc.Width = mSwapChainRect.Width;
mpOutputResource = mpDevice.CreateCommittedResource(AccelerationStructures.kDefaultHeapProps, HeapFlags.None, resDesc, ResourceStates.CopySource, null); // Starting as copy-source to simplify onFrameRender()
// Create an SRV/UAV/VertexSRV/IndexSRV descriptor heap. Need 5 entries - 1 SRV for the scene, 1 UAV for the output, 1 SRV for VertexBuffer, 1 SRV for IndexBuffer, 1 SceneContantBuffer, 1 primitiveConstantBuffer
mpSrvUavHeap = this.context.CreateDescriptorHeap(mpDevice, 6, DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView, true);
// Create the UAV. Based on the root signature we created it should be the first entry
UnorderedAccessViewDescription uavDesc = new UnorderedAccessViewDescription();
uavDesc.ViewDimension = UnorderedAccessViewDimension.Texture2D;
mpDevice.CreateUnorderedAccessView(mpOutputResource, null, uavDesc, mpSrvUavHeap.GetCPUDescriptorHandleForHeapStart());
// Create the TLAS SRV right after the UAV. Note that we are using a different SRV desc here
ShaderResourceViewDescription srvDesc = new ShaderResourceViewDescription();
srvDesc.ViewDimension = ShaderResourceViewDimension.RaytracingAccelerationStructure;
srvDesc.Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping;
srvDesc.RaytracingAccelerationStructure = new RaytracingAccelerationStructureShaderResourceView();
srvDesc.RaytracingAccelerationStructure.Location = mpTopLevelAS.GPUVirtualAddress;
CpuDescriptorHandle srvHandle = mpSrvUavHeap.GetCPUDescriptorHandleForHeapStart();
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
mpDevice.CreateShaderResourceView(null, srvDesc, srvHandle);
// Index SRV
var indexSRVDesc = new ShaderResourceViewDescription()
{
ViewDimension = ShaderResourceViewDimension.Buffer,
Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping,
Format = Format.R32_Typeless,
Buffer =
{
NumElements = (int)(this.acs.IndexCount * 2 / 4),
Flags = BufferShaderResourceViewFlags.Raw,
StructureByteStride = 0,
}
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
indexSRVHandle = srvHandle;
mpDevice.CreateShaderResourceView(this.acs.IndexBuffer, indexSRVDesc, indexSRVHandle);
// Vertex SRV
var vertexSRVDesc = new ShaderResourceViewDescription()
{
ViewDimension = ShaderResourceViewDimension.Buffer,
Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping,
Format = Format.Unknown,
Buffer =
{
NumElements = (int)this.acs.VertexCount,
Flags = BufferShaderResourceViewFlags.None,
StructureByteStride = Unsafe.SizeOf <VertexPositionNormalTangentTexture>(),
}
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
vertexSRVHandle = srvHandle;
mpDevice.CreateShaderResourceView(this.acs.VertexBuffer, vertexSRVDesc, vertexSRVHandle);
// CB Scene
Vector3 cameraPosition = new Vector3(0, 1, -7);
Matrix4x4 view = Matrix4x4.CreateLookAt(cameraPosition, Vector3.Zero, Vector3.UnitY);
Matrix4x4 proj = Matrix4x4.CreatePerspectiveFieldOfView(MathHelper.PiOver4, (float)mSwapChainRect.Width / mSwapChainRect.Height, 0.1f, 1000f);
Matrix4x4 viewProj = Matrix4x4.Multiply(view, proj);
Matrix4x4.Invert(viewProj, out Matrix4x4 projectionToWorld);
SceneConstantBuffer sceneConstantBuffer = new SceneConstantBuffer()
{
projectionToWorld = Matrix4x4.Transpose(projectionToWorld),
cameraPosition = cameraPosition,
lightPosition = new Vector3(0.0f, 1.0f, -2.0f),
lightDiffuseColor = new Vector4(0.5f, 0.5f, 0.5f, 1.0f),
lightAmbientColor = new Vector4(0.1f, 0.1f, 0.1f, 1.0f),
backgroundColor = new Vector4(0.4f, 0.6f, 0.2f, 1.0f),
MaxRecursionDepth = 4,
};
sceneCB = this.acs.CreateBuffer(mpDevice, (uint)Unsafe.SizeOf <SceneConstantBuffer>(), ResourceFlags.None, ResourceStates.GenericRead, AccelerationStructures.kUploadHeapProps);
IntPtr pData;
pData = sceneCB.Map(0, null);
Helpers.MemCpy(pData, sceneConstantBuffer, (uint)Unsafe.SizeOf <SceneConstantBuffer>());
sceneCB.Unmap(0, null);
var sceneCBV = new ConstantBufferViewDescription()
{
BufferLocation = sceneCB.GPUVirtualAddress,
SizeInBytes = (Unsafe.SizeOf <SceneConstantBuffer>() + 255) & ~255,
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
sceneCBVHandle = srvHandle;
mpDevice.CreateConstantBufferView(sceneCBV, sceneCBVHandle);
// CB Primitive
PrimitiveConstantBuffer primitiveConstantBuffer = new PrimitiveConstantBuffer()
{
diffuseColor = new Vector4(0.8f, 0f, 0f, 1.0f),
inShadowRadiance = 0.35f,
diffuseCoef = 0.1f,
specularCoef = 0.7f,
specularPower = 50,
reflectanceCoef = 0.7f,
};
ID3D12Resource primitiveCB = this.acs.CreateBuffer(mpDevice, (uint)Unsafe.SizeOf <PrimitiveConstantBuffer>(), ResourceFlags.None, ResourceStates.GenericRead, AccelerationStructures.kUploadHeapProps);
IntPtr pData2;
pData2 = primitiveCB.Map(0, null);
Helpers.MemCpy(pData2, primitiveConstantBuffer, (uint)Unsafe.SizeOf <PrimitiveConstantBuffer>());
primitiveCB.Unmap(0, null);
var primitiveCBV = new ConstantBufferViewDescription()
{
BufferLocation = primitiveCB.GPUVirtualAddress,
SizeInBytes = (Unsafe.SizeOf <PrimitiveConstantBuffer>() + 255) & ~255,
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
var primitiveCBVHandle = srvHandle;
mpDevice.CreateConstantBufferView(primitiveCBV, primitiveCBVHandle);
}
public void CreateShaderResources()
{
// Create the output resource. The dimensions and format should match the swap-chain
ResourceDescription resDesc = new ResourceDescription();
resDesc.DepthOrArraySize = 1;
resDesc.Dimension = ResourceDimension.Texture2D;
resDesc.Format = Format.R8G8B8A8_UNorm; // The backbuffer is actually DXGI_FORMAT_R8G8B8A8_UNORM_SRGB, but sRGB formats can't be used with UAVs. We will convert to sRGB ourselves in the shader
resDesc.Flags = ResourceFlags.AllowUnorderedAccess;
resDesc.Height = mSwapChainRect.Height;
resDesc.Layout = TextureLayout.Unknown;
resDesc.MipLevels = 1;
resDesc.SampleDescription = new SampleDescription(1, 0);
resDesc.Width = mSwapChainRect.Width;
mpOutputResource = mpDevice.CreateCommittedResource(AccelerationStructures.kDefaultHeapProps, HeapFlags.None, resDesc, ResourceStates.CopySource, null); // Starting as copy-source to simplify onFrameRender()
// Create an SRV/UAV/VertexSRV/IndexSRV descriptor heap. Need 5 entries - 1 SRV for the scene, 1 UAV for the output, 1 SRV for VertexBuffer, 1 SRV for IndexBuffer, 1 SRV for texture
mpSrvUavHeap = this.context.CreateDescriptorHeap(mpDevice, 5, DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView, true);
// Create the UAV. Based on the root signature we created it should be the first entry
UnorderedAccessViewDescription uavDesc = new UnorderedAccessViewDescription();
uavDesc.ViewDimension = UnorderedAccessViewDimension.Texture2D;
mpDevice.CreateUnorderedAccessView(mpOutputResource, null, uavDesc, mpSrvUavHeap.GetCPUDescriptorHandleForHeapStart());
// Create the TLAS SRV right after the UAV. Note that we are using a different SRV desc here
ShaderResourceViewDescription srvDesc = new ShaderResourceViewDescription();
srvDesc.ViewDimension = ShaderResourceViewDimension.RaytracingAccelerationStructure;
srvDesc.Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping;
srvDesc.RaytracingAccelerationStructure = new RaytracingAccelerationStructureShaderResourceView();
srvDesc.RaytracingAccelerationStructure.Location = mpTopLevelAS.GPUVirtualAddress;
CpuDescriptorHandle srvHandle = mpSrvUavHeap.GetCPUDescriptorHandleForHeapStart();
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
mpDevice.CreateShaderResourceView(null, srvDesc, srvHandle);
// Index SRV
var indexSRVDesc = new ShaderResourceViewDescription()
{
ViewDimension = ShaderResourceViewDimension.Buffer,
Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping,
Format = Format.R32_Typeless,
Buffer =
{
NumElements = (int)(this.acs.IndexCount * 2 / 4),
Flags = BufferShaderResourceViewFlags.Raw,
StructureByteStride = 0,
}
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
indexSRVHandle = srvHandle;
mpDevice.CreateShaderResourceView(this.acs.IndexBuffer, indexSRVDesc, indexSRVHandle);
// Vertex SRV
var vertexSRVDesc = new ShaderResourceViewDescription()
{
ViewDimension = ShaderResourceViewDimension.Buffer,
Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping,
Format = Format.Unknown,
Buffer =
{
NumElements = (int)this.acs.VertexCount,
Flags = BufferShaderResourceViewFlags.None,
StructureByteStride = Unsafe.SizeOf <VertexPositionNormalTangentTexture>(),
}
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
vertexSRVHandle = srvHandle;
mpDevice.CreateShaderResourceView(this.acs.VertexBuffer, vertexSRVDesc, vertexSRVHandle);
// Texture
var image = TextureHelper.Load("GLTF/Default_albedo.jpg");
var textureDesc = ResourceDescription.Texture2D(image.Format, (int)image.Width, (int)image.Height);
this.texture = this.mpDevice.CreateCommittedResource(new HeapProperties(HeapType.Default), HeapFlags.None, textureDesc, ResourceStates.CopyDestination);
// Create the GPU upload buffer.
var textureUploadHeap = mpDevice.CreateCommittedResource(new HeapProperties(CpuPageProperty.WriteBack, MemoryPool.L0), HeapFlags.None, textureDesc, ResourceStates.GenericRead);
// Copy data to the intermediate upload heap and then schedule a copy
// from the upload heap to the Texture2D.
byte[] textureData = image.Data;
var handle = GCHandle.Alloc(textureData, GCHandleType.Pinned);
var ptr = Marshal.UnsafeAddrOfPinnedArrayElement(textureData, 0);
textureUploadHeap.WriteToSubresource(0, null, ptr, (int)(image.TexturePixelSize * image.Width), textureData.Length);
handle.Free();
mpCmdList.CopyTextureRegion(new TextureCopyLocation(texture, 0), 0, 0, 0, new TextureCopyLocation(textureUploadHeap, 0), null);
mpCmdList.ResourceBarrierTransition(this.texture, ResourceStates.CopyDestination, ResourceStates.PixelShaderResource);
// Describe and create a SRV for the texture.
var textureSRVDesc = new ShaderResourceViewDescription
{
Shader4ComponentMapping = D3D12DefaultShader4ComponentMapping,
Format = textureDesc.Format,
ViewDimension = ShaderResourceViewDimension.Texture2D,
Texture2D = { MipLevels = 1 },
};
srvHandle.Ptr += mpDevice.GetDescriptorHandleIncrementSize(DescriptorHeapType.ConstantBufferViewShaderResourceViewUnorderedAccessView);
textureSRVHandle = srvHandle;
this.mpDevice.CreateShaderResourceView(this.texture, textureSRVDesc, textureSRVHandle);
}
