#pragma once
 
/*------------------------------------
 Iwa_DirectionalBlurFx
 ボケ足の伸ばし方を選択でき、参照画像を追加した DirectionalBlur
//------------------------------------*/
 
#ifndef IWA_DIRECTIONAL_BLUR_H
#define IWA_DIRECTIONAL_BLUR_H
 
#include "tfxparam.h"
#include "stdfx.h"
 
struct float2 {
	float x, y;
};
struct float4 {
	float x, y, z, w;
};
struct int2 {
	int x, y;
};
 
class Iwa_DirectionalBlurFx : public TStandardRasterFx
{
 
	FX_PLUGIN_DECLARATION(Iwa_DirectionalBlurFx)
 
	TRasterFxPort m_input;
	TRasterFxPort m_reference;
 
	TDoubleParamP m_angle;
	TDoubleParamP m_intensity;
	TBoolParamP m_bidirectional;
	/*- リニア／ガウシアン／平均化 -*/
	TIntEnumParamP m_filterType;
 
	/*- 参照画像の輝度を０〜１に正規化してホストメモリに読み込む -*/
	template <typename RASTER, typename PIXEL>
	void setReferenceRaster(const RASTER srcRas,
							float *dstMem,
							TDimensionI dim);
 
	/*- ソース画像を０〜１に正規化してホストメモリに読み込む -*/
	template <typename RASTER, typename PIXEL>
	void setSourceRaster(const RASTER srcRas,
						 float4 *dstMem,
						 TDimensionI dim);
 
	/*- 出力結果をChannel値に変換して格納 -*/
	template <typename RASTER, typename PIXEL>
	void setOutputRaster(float4 *srcMem,
						 const RASTER dstRas,
						 TDimensionI dim,
						 int2 margin);
 
	void doCompute_CPU(TTile &tile,
					   double frame,
					   const TRenderSettings &settings,
					   TPointD &blur,
					   bool bidirectional,
					   int marginLeft, int marginRight,
					   int marginTop, int marginBottom,
					   TDimensionI &enlargedDimIn,
					   TTile &enlarge_tile,
					   TDimensionI &dimOut,
					   TDimensionI &filterDim,
					   float *reference_host);
 
	void makeDirectionalBlurFilter_CPU(float *filter,
									   TPointD &blur,
									   bool bidirectional,
									   int marginLeft, int marginRight,
									   int marginTop, int marginBottom,
									   TDimensionI &filterDim);
 
public:
	Iwa_DirectionalBlurFx();
 
	void doCompute(TTile &tile,
				   double frame,
				   const TRenderSettings &settings);
 
	bool doGetBBox(double frame,
				   TRectD &bBox,
				   const TRenderSettings &info);
 
	bool canHandle(const TRenderSettings &info,
				   double frame);
 
	void getParamUIs(TParamUIConcept *&concepts,
					 int &length);
};
 
#endif