// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once

#include "CoreMinimal.h"
#include "RenderMaterial.h"
#include "SamplerStates_FX.h"

class FxMaterial;
typedef std::shared_ptr<FxMaterial>	FxMaterialPtr;

class TEXTUREGRAPHENGINE_API RenderMaterial_FX : public RenderMaterial
{
protected:
	mutable FxMaterialPtr			FXMaterial;			/// The FX material from the shader
	CHashPtr						HashValue;			/// The hash for the material
	
public:
									RenderMaterial_FX(FString InName, FxMaterialPtr InFXMaterial);
	virtual							~RenderMaterial_FX() override;

	//////////////////////////////////////////////////////////////////////////
	/// BlobTransform implementation
	//////////////////////////////////////////////////////////////////////////
	virtual void					Bind(int32 Value, const ResourceBindInfo& BindInfo) override;
	virtual void					Bind(float Value, const ResourceBindInfo& BindInfo) override;
	virtual void					Bind(const FLinearColor& Value, const ResourceBindInfo& BindInfo) override;
	virtual void					Bind(const FIntVector4& Value, const ResourceBindInfo& BindInfo) override;
	virtual void					Bind(const FMatrix& Value, const ResourceBindInfo& BindInfo) override;
	virtual void					BindStruct(const char* ValueAddress, size_t StructSize, const ResourceBindInfo& BindInfo) override;
	virtual void					BindScalarArray(const char* StartingAddress, size_t TypeSize, size_t ArraySize, const ResourceBindInfo& BindInfo) override;

	virtual CHashPtr				Hash() const override;
	virtual std::shared_ptr<BlobTransform> DuplicateInstance(FString NewName) override;

	//////////////////////////////////////////////////////////////////////////
	/// RenderMaterial Implementation
	//////////////////////////////////////////////////////////////////////////
	virtual void					BlitTo(FRHICommandListImmediate& RHI, UTextureRenderTarget2D* RenderTarget, const RenderMesh* MeshObj, int32 TargetId) const override;
	virtual void					SetTexture(FName InName, const UTexture* Texture) const override;
	virtual void					SetArrayTexture(FName InName, const std::vector<const UTexture*>& Textures) const override;
	virtual void					SetInt(FName InName, int32 Value) const override;
	virtual void					SetFloat(FName InName, float Value) const override;
	virtual void					SetColor(FName InName, const FLinearColor& Value) const override;
	virtual void					SetIntVector4(FName InName, const FIntVector4& Value) const override;
	virtual void					SetMatrix(FName InName, const FMatrix& Value) const override;

	//////////////////////////////////////////////////////////////////////////
	/// Inline functions
	//////////////////////////////////////////////////////////////////////////
	FORCEINLINE FxMaterialPtr		FxMaterial() const { return FXMaterial; }
};



typedef std::shared_ptr<RenderMaterial_FX> RenderMaterial_FXPtr;