ComputeShader.cpp

////////////////////////////////////////////////////////////////////////////
//	Created		: 21.05.2009
//	Author		: Mykhailo Parfeniuk
//	Copyright (C) GSC Game World - 2009
////////////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "ComputeShader.h"

void ComputeShader::Construct(
	ID3D11ComputeShader* cs,
	ref_ctable ctable,
	xr_vector<ID3D11SamplerState*>& Samplers,
	xr_vector<ID3D11ShaderResourceView*>& Textures,
	xr_vector<ID3D11UnorderedAccessView*>& Outputs
)
{
	m_cs = cs;
	m_ctable = ctable;
	m_Textures.swap(Textures);
	m_Outputs.swap(Outputs);
	m_Samplers.swap(Samplers);
}

ComputeShader::~ComputeShader()
{
	for (size_t i = 0; i < m_Textures.size(); ++i)
		m_Textures[i]->Release();

	for (size_t i = 0; i < m_Outputs.size(); ++i)
		m_Outputs[i]->Release();

	for (size_t i = 0; i < m_Samplers.size(); ++i)
		m_Samplers[i]->Release();
}

u32 GetCB(R_constant* C)
{
	return (C->destination & RC_dest_pixel_cb_index_mask) >> RC_dest_pixel_cb_index_shift;
}


ComputeShader& ComputeShader::set_c(shared_str name, const Fvector4& value)
{
	R_constant* c = m_ctable->get(name);
	m_ctable->m_CBTable[GetCB(c)].second->set(c, c->ps, value);
	return *this;
}

ComputeShader& ComputeShader::set_c(shared_str name, float x, float y, float z, float w)
{
	Fvector4 vec;
	vec.set(x, y, z, w);
	return set_c(name, vec);
}

void ComputeShader::Dispatch(u32 dimx, u32 dimy, u32 dimz)
{
	u32 count = m_ctable->m_CBTable.size();

	for (u32 i = 0; i < count; ++i)
	{
		m_ctable->m_CBTable[i].second->Flush();
	}

	ID3DBuffer* tempBuffer[CBackend::MaxCBuffers];

	for (u32 i = 0; i < count; ++i)
	{
		tempBuffer[i] = m_ctable->m_CBTable[i].second->GetBuffer();
	}

	// process constant-loaders
	R_constant_table::c_table::iterator it = m_ctable->table.begin();
	R_constant_table::c_table::iterator end = m_ctable->table.end();
	for (; it != end; it++)
	{
		R_constant* Cs = &**it;
		if (Cs->handler) Cs->handler->setup(Cs);
	}

	HW.pContext->CSSetConstantBuffers(0, count, tempBuffer);

	if (!m_Textures.empty())
		HW.pContext->CSSetShaderResources(0, m_Textures.size(), &m_Textures[0]);

	if (!m_Samplers.empty())
		HW.pContext->CSSetSamplers(0, m_Samplers.size(), &m_Samplers[0]);

	if (!m_Outputs.empty())
	{
		UINT num = 0;
		HW.pContext->CSSetUnorderedAccessViews(0, m_Outputs.size(), &m_Outputs[0], &num);
	}

	HW.pContext->Dispatch(dimx, dimy, dimz);
}