Height Map Through Compute Shader, Need Some Pointers

Started by
2 comments, last by unbird 7 years, 8 months ago

Hi all,

For about 2 weeks now I've been struggling with an exercise I can't figure out so far.

Here's the goal, in short:

- render waves through a vertex and pixel shader

-- where the height of the waves vertices are read from a texture, containing floats (each float contains the height of a wave vertex)

- use a compute shader to update the wave heights

- update them, and render again etc.

My first goal is to achieve a situation where at least I get no errors/ warnings and the Compute shader works, stores the new texture, which then is read by the rendering effect, fetching these height map values for the wave vertices.

In my book it says to do this, I would need 4 textures in the compute shader:


// textures containing single floats
Texture2D	gPrevSolInput;
Texture2D	gCurrSolInput;

RWTexture2D<float> gCurrSolOutput;
RWTexture2D<float> gNextSolOutput;

Here are some relevant code snippers of my current (non-working) approach.
Let me know if you need other parts to be able to identify the issue.

The extra line in the existing rendering function, right before drawint the waves:


	Effects::mBasicFX->SetWavesHeightMap(mWavesCS.mOutputNextSolUAV);

The basic rendering effects (VS and PS), which should fetch the height map:


Texture2D gWavesHeightMap;

SamplerState samHeightMap
{
	Filter = MIN_MAG_MIP_POINT;

	AddressU = WRAP;
	AddressV = WRAP;
};

VertexOut VS(VertexIn vin)
{
	VertexOut vout;

	// Retrieve wave height map data
	if(gIsWaves)
	{
		float heightMapVal = gWavesHeightMap.SampleLevel(samHeightMap, vin.Tex, 0.0f).r;
		vin.PosL.y = heightMapVal;
	}

Calling the compute shader and the compute shader itself:


void CSWrapper::ExecuteCS(ID3D11DeviceContext *dc)
{
	ID3D11ShaderResourceView *nullSRV[2] = { 0, 0 };
	ID3D11UnorderedAccessView *nullUAV[2] = { 0, 0 };

	D3DX11_TECHNIQUE_DESC techDesc;
	Effects::mWavesCSFX->mWavesTech->GetDesc(&techDesc);

	for(UINT p=0;p<techDesc.Passes;++p)
	{
		Effects::mWavesCSFX->SetInputPrevSolSRV(mInputPrevSolSRV);
		Effects::mWavesCSFX->SetInputCurrSolSRV(mInputCurrSolSRV);

		Effects::mWavesCSFX->SetOutputCurrSolUAV(mOutputCurrSolUAV);
		Effects::mWavesCSFX->SetOutputNextSolUAV(mOutputNextSolUAV);

		Effects::mWavesCSFX->mWavesTech->GetPassByIndex(p)->Apply(0, dc);

		// grid 160x160 = 25.600 threads needed, /64 (group size 64,1,1) = 400
		dc->Dispatch(400, 1, 1);
	}

	// clean up
	dc->CSSetShaderResources(0, 2, nullSRV);
	dc->CSSetUnorderedAccessViews(0, 2, nullUAV, 0);

	// disable compute shader
	dc->CSSetShader(0, 0, 0);
}

[numthreads(64, 1, 1)]
void CS(int3 dispatchThreadID : SV_DispatchThreadID)
{
	gNextSolOutput[dispatchThreadID.xy] = gCurrSolInput[dispatchThreadID.xy] + 0.01f;

	gCurrSolOutput[dispatchThreadID.xy] = gNextSolOutput[dispatchThreadID.xy];
}

Creating the 4 textures and SRV/UAV's for (compute) shader:


void CSWrapper::Init(ID3D11Device *pDevice, ID3D11DeviceContext *dc, DirectX::XMFLOAT3* pPrevSolution, DirectX::XMFLOAT3* pCurrSolution, const int pM, const int pN)
{
	// There are 4 views

	// 2 input SRV's	-> prev and curr solution
	// 2 output UAV's	-> curr and next solution

	// The SRV's has to initially be filled and created: textures
	
	/***************************************/
	/**** 1: input SRV (prev solution)	****/
	/***************************************/

	// create texture
	D3D11_TEXTURE2D_DESC inputPrevSolTexDesc;
	inputPrevSolTexDesc.Width				= pM;		// grid size of the waves, rows
	inputPrevSolTexDesc.Height				= pN;		// grid size of the waves, colums
	inputPrevSolTexDesc.MipLevels			= 1;
	inputPrevSolTexDesc.ArraySize			= 1;
	inputPrevSolTexDesc.Format				= DXGI_FORMAT_R32_FLOAT;
	inputPrevSolTexDesc.SampleDesc.Count	= 1;
	inputPrevSolTexDesc.SampleDesc.Quality	= 0;
	inputPrevSolTexDesc.Usage				= D3D11_USAGE_DEFAULT;
	inputPrevSolTexDesc.BindFlags			= D3D11_BIND_SHADER_RESOURCE | D3D11_BIND_UNORDERED_ACCESS;
	inputPrevSolTexDesc.CPUAccessFlags		= 0;
	inputPrevSolTexDesc.MiscFlags			= 0;

	// initial texture data
	float *tempArrayA = new float[pM*pN];
	for(int i=0;i<pM*pN;++i) tempArrayA[i] = pPrevSolution[i].y;
	
	D3D11_SUBRESOURCE_DATA vinitDataA;
	vinitDataA.SysMemPitch = sizeof(float) * pM;
	vinitDataA.pSysMem = &tempArrayA[0];

	// create the texture
	ID3D11Texture2D *inputPrevSolTex = 0;
	HR(pDevice->CreateTexture2D(&inputPrevSolTexDesc, &vinitDataA, &inputPrevSolTex));
	
	delete[] tempArrayA;

	// create SRV
	D3D11_SHADER_RESOURCE_VIEW_DESC prevSolSrvDesc;
	prevSolSrvDesc.Format					= DXGI_FORMAT_R32_FLOAT;
	prevSolSrvDesc.ViewDimension			= D3D11_SRV_DIMENSION_TEXTURE2D;
	prevSolSrvDesc.BufferEx.FirstElement	= 0;
	prevSolSrvDesc.BufferEx.Flags			= 0;
	prevSolSrvDesc.BufferEx.NumElements		= pM * pN;

	prevSolSrvDesc.Texture2D.MipLevels			= 1;
	prevSolSrvDesc.Texture2D.MostDetailedMip	= 0;

	pDevice->CreateShaderResourceView(inputPrevSolTex, &prevSolSrvDesc, &mInputPrevSolSRV);
	ReleaseCOM(inputPrevSolTex);

	/***************************************/
	/**** 2: input SRV (curr solution)	****/
	/***************************************/

	// create texture
	D3D11_TEXTURE2D_DESC inputCurrSolTexDesc;
	inputCurrSolTexDesc.Width				= pM;		// grid size of the waves, rows
	inputCurrSolTexDesc.Height				= pN;		// grid size of the waves, colums
	inputCurrSolTexDesc.MipLevels			= 1;
	inputCurrSolTexDesc.ArraySize			= 1;
	inputCurrSolTexDesc.Format				= DXGI_FORMAT_R32_FLOAT;
	inputCurrSolTexDesc.SampleDesc.Count	= 1;
	inputCurrSolTexDesc.SampleDesc.Quality	= 0;
	inputCurrSolTexDesc.Usage				= D3D11_USAGE_DEFAULT;
	inputCurrSolTexDesc.BindFlags			= D3D11_BIND_SHADER_RESOURCE | D3D11_BIND_UNORDERED_ACCESS;
	inputCurrSolTexDesc.CPUAccessFlags		= 0;
	inputCurrSolTexDesc.MiscFlags			= 0;

	// initial texture data
	float *tempArrayB = new float[pM*pN];
	for(int i=0;i<pM*pN;++i) tempArrayB[i] = pCurrSolution[i].y;
	
	D3D11_SUBRESOURCE_DATA vinitDataB;
	vinitDataB.SysMemPitch = sizeof(float) * pM;
	vinitDataB.pSysMem = &tempArrayB[0];

	// create the texture
	ID3D11Texture2D *inputCurrSolTex = 0;
	HR(pDevice->CreateTexture2D(&inputCurrSolTexDesc, &vinitDataB, &inputCurrSolTex));
	
	delete[] tempArrayB;

	// create SRV
	D3D11_SHADER_RESOURCE_VIEW_DESC currSolSrvDesc;
	currSolSrvDesc.Format					= DXGI_FORMAT_R32_FLOAT;
	currSolSrvDesc.ViewDimension			= D3D11_SRV_DIMENSION_TEXTURE2D;
	currSolSrvDesc.BufferEx.FirstElement	= 0;
	currSolSrvDesc.BufferEx.Flags			= 0;
	currSolSrvDesc.BufferEx.NumElements		= pM * pN;

	currSolSrvDesc.Texture2D.MipLevels			= 1;
	currSolSrvDesc.Texture2D.MostDetailedMip	= 0;

	pDevice->CreateShaderResourceView(inputCurrSolTex, &currSolSrvDesc, &mInputCurrSolSRV);

	/***************************************/
	/**** 3: output UAV (curr solution)	****/
	/***************************************/

	D3D11_UNORDERED_ACCESS_VIEW_DESC uavCurrDesc;
	uavCurrDesc.Format					= DXGI_FORMAT_R32_FLOAT;
	uavCurrDesc.ViewDimension			= D3D11_UAV_DIMENSION_TEXTURE2D;
	uavCurrDesc.Buffer.FirstElement		= 0;
	uavCurrDesc.Buffer.Flags			= 0;
	uavCurrDesc.Buffer.NumElements		= pM * pN;

	pDevice->CreateUnorderedAccessView(inputCurrSolTex, &uavCurrDesc, &mOutputCurrSolUAV);

//	ReleaseCOM(inputCurrSolTex);		// release texture, both SRV and UAV created

	/***************************************/
	/**** 4: output UAV (next solution)	****/
	/***************************************/

	D3D11_UNORDERED_ACCESS_VIEW_DESC uavNextDesc;
	uavNextDesc.Format					= DXGI_FORMAT_R32_FLOAT;
	uavNextDesc.ViewDimension			= D3D11_UAV_DIMENSION_TEXTURE2D;
	uavNextDesc.Buffer.FirstElement		= 0;
	uavNextDesc.Buffer.Flags			= 0;
	uavNextDesc.Buffer.NumElements		= pM * pN;

	pDevice->CreateUnorderedAccessView(inputCurrSolTex, &uavNextDesc, &mOutputNextSolUAV);

	ReleaseCOM(inputCurrSolTex);		// release texture, both SRV and UAV created

}

I would really appreciate if someone could help me in the right direction, to finally get this case solved..

Thanks in advance.

Crealysm game & engine development: http://www.crealysm.com

Looking for a passionate, disciplined and structured producer? PM me

Advertisement
Could you specify which errors/warnings you currently receive?

Here's some additional info:

A. I don't understand why I would need both 2 SRV's and 2 UAV's to achieve the task/ goal

B. the following info (no error?) appears when trying to set the UAV as a Texture2D input in the rendering effect:


Effects11: ID3DX11EffectVariable::AsUnorderedAccessView: Invalid typecast

C. the CS gives a warning about a vector truncation


[numthreads(64, 1, 1)]
void CS(int3 dispatchThreadID : SV_DispatchThreadID)
{
	gNextSolOutput[dispatchThreadID.xy] = gCurrSolInput[dispatchThreadID.xy] + 0.01f;

	gCurrSolOutput[dispatchThreadID.xy] = gNextSolOutput[dispatchThreadID.xy];
}

// line 4 above gives this warning

1>E:\projects\3D Game programming DX11\Chapter 12 - Exercise 5\Chapter 12 - Exercise 5\FX\wavesCS.fx(25,2-81): warning X3206: implicit truncation of vector type

D. I'm not sure I'm creating the SRV's and UAV's the correct way

E. I'm not sure I'm fetching the UAV/ Texture2D Heights correctly, in the rendering effect.

The code uses .r to retrieve the value, but the texture only has float32's, so 1 float per pixel, is this automatically the .R component then?


	// Retrieve wave height map data
	if(gIsWaves)
	{
		float heightMapVal = gWavesHeightMap.SampleLevel(samHeightMap, vin.Tex, 0.0f).r;
		vin.PosL.y = heightMapVal;
	}

Crealysm game & engine development: http://www.crealysm.com

Looking for a passionate, disciplined and structured producer? PM me

Why not just peek at the solution (3rd source package, SelectedCodeSolutions\CSGpuWavesDemo) ? :wink:

This topic is closed to new replies.

Advertisement