Magpie/Runtime/EffectDrawer.h

#pragma once
#include "pch.h"
#include "EffectDesc.h"


class EffectDrawer {
public:
	EffectDrawer() = default;
	EffectDrawer(const EffectDrawer&) = delete;
	EffectDrawer(EffectDrawer&&) = delete;

	bool Initialize(
		const EffectDesc& desc,
		const EffectParams& params,
		ID3D11Texture2D* inputTex,
		ID3D11Texture2D** outputTex,
		RECT* outputRect = nullptr,
		RECT* virtualOutputRect = nullptr
	);

	void Draw(UINT& idx, bool noUpdate = false);

	bool IsUseDynamic() const noexcept {
		return _desc.isUseDynamic;
	}

	const EffectDesc& GetDesc() const noexcept {
		return _desc;
	}

private:
	void _DrawPass(UINT i);

	EffectDesc _desc;

	std::vector<ID3D11SamplerState*> _samplers;
	std::vector<winrt::com_ptr<ID3D11Texture2D>> _textures;
	std::vector<std::vector<ID3D11ShaderResourceView*>> _srvs;
	// 后半部分为空，用于解绑
	std::vector<std::vector<ID3D11UnorderedAccessView*>> _uavs;

	std::vector<EffectConstant32> _constants;
	winrt::com_ptr<ID3D11Buffer> _constantBuffer;

	std::vector<winrt::com_ptr<ID3D11ComputeShader>> _shaders;

	std::vector<std::pair<UINT, UINT>> _dispatches;
};