#pragma once

/*------------------------------------

 Iwa_GradientWarpFx

 参照画像の勾配方向にWarpするエフェクト

------------------------------------*/

#ifndef IWA_GRADIENT_WARP_H

#define IWA_GRADIENT_WARP_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_GradientWarpFx : public TStandardRasterFx

{

	FX_PLUGIN_DECLARATION(Iwa_GradientWarpFx)

protected:

	TRasterFxPort m_source; /*- 入力画像 -*/

	TRasterFxPort m_warper; /*- 参照画像 -*/

	TDoubleParamP m_h_maxlen;

	TDoubleParamP m_v_maxlen;

	TDoubleParamP m_scale; /*- ワープ量を増やすスカラー値。この値はマージン値には影響しない -*/

	void get_render_real_hv(const double frame,

							const TAffine affine,

							double &h_maxlen,

							double &v_maxlen);

	void get_render_enlarge(const double frame,

							const TAffine affine,

							TRectD &bBox);

	/*- タイルの画像を０〜１に正規化してホストメモリに読み込む -*/

	template <typename pixel="" raster,="" typename=""></typename>

	void setSourceRaster(const RASTER srcRas,

						 float4 *dstMem,

						 TDimensionI dim);

	/*- タイルの画像の輝度値を０〜１に正規化してホストメモリに読み込む -*/

	template <typename pixel="" raster,="" typename=""></typename>

	void setWarperRaster(const RASTER srcRas,

						 float *dstMem,

						 TDimensionI dim);

	/*- 出力結果をChannel値に変換して格納 -*/

	template <typename pixel="" raster,="" typename=""></typename>

	void setOutputRaster(float4 *srcMem,

						 const RASTER dstRas,

						 TDimensionI dim,

						 int2 margin);

	void doCompute_CPU(TTile &tile,

					   const double frame,

					   const TRenderSettings &settings,

					   double hLength,

					   double vLength,

					   int margin,

					   TDimensionI &enlargedDim,

					   float4 *source_host,

					   float *warper_host,

					   float4 *result_host);

	float4 getSourceVal_CPU(float4 *source_host,

							TDimensionI &enlargedDim,

							int pos_x, int pos_y);

	float4 interp_CPU(float4 val1, float4 val2, float ratio);

public:

	Iwa_GradientWarpFx();

	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);

};

#endif