#pragma once

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

 Iwa_MotionBlurCompFx

 露光量／オブジェクトの軌跡を考慮したモーションブラー

 背景との露光値合成を可能にする

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

#ifndef IWA_MOTION_BLUR_COMP_H

#define IWA_MOTION_BLUR_COMP_H

#include "tfxparam.h"

#include "stdfx.h"

#include "motionawarebasefx.h"

class TStageObjectId;

struct float2 {

	float x, y;

};

struct float3 {

	float x, y, z;

};

struct float4 {

	float x, y, z, w;

};

struct int2 {

	int x, y;

};

/*- m_premultiTypeの取る値 -*/

enum PremultiTypes {

	AUTO = 0,

	SOURCE_IS_PREMULTIPLIED,

	SOURCE_IS_NOT_PREMUTIPLIED

};

class Iwa_MotionBlurCompFx : public MotionAwareBaseFx

{

	FX_PLUGIN_DECLARATION(Iwa_MotionBlurCompFx)

protected:

	TRasterFxPort m_input;

	TRasterFxPort m_background;

	TDoubleParamP m_hardness; /*- フィルムのガンマ値 -*/

	/*-- 左右をぼかすためのパラメータ --*/

	TDoubleParamP m_startValue; /*- シャッター開け時のフィルタ値 -*/

	/*- シャッター開け時〜現時点までのフィルタ値のガンマ補正値 -*/

	TDoubleParamP m_startCurve;

	TDoubleParamP m_endValue; /*- シャッター閉じ時のフィルタ値 -*/

	/*- 現時点〜シャッター閉じ時までのフィルタ値のガンマ補正値 -*/

	TDoubleParamP m_endCurve;

	/*- 残像モードのトグル -*/

	TBoolParamP m_zanzoMode;

	TIntEnumParamP m_premultiType;

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

		ソース画像がPremultipyされているか、コンボボックスで指定されていない場合は

		ここで判定する -*/

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

	bool setSourceRaster(const RASTER srcRas,

						 float4 *dstMem,

						 TDimensionI dim,

						 PremultiTypes type = AUTO);

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

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

	void setOutputRaster(float4 *srcMem,

						 const RASTER dstRas,

						 TDimensionI dim,

						 int2 margin);

	/*-  フィルタをつくり、正規化する -*/

	void makeMotionBlurFilter_CPU(float *filter_p,

								  TDimensionI &filterDim,

								  int marginLeft, int marginBottom,

								  float4 *pointsTable,

								  int pointAmount,

								  float startValue, float startCurve,

								  float endValue, float endCurve);

	/*- 残像フィルタをつくり、正規化する -*/

	void makeZanzoFilter_CPU(float *filter_p,

							 TDimensionI &filterDim,

							 int marginLeft, int marginBottom,

							 float4 *pointsTable,

							 int pointAmount,

							 float startValue, float startCurve,

							 float endValue, float endCurve);

	/*- RGB値(０〜１)を露光値に変換 -*/

	void convertRGBtoExposure_CPU(float4 *in_tile_p,

								  TDimensionI &dim, float hardness,

								  bool sourceIsPremultiplied);

	/*- 露光値をフィルタリングしてぼかす -*/

	void applyBlurFilter_CPU(float4 *in_tile_p,

							 float4 *out_tile_p,

							 TDimensionI &dim,

							 float *filter_p,

							 TDimensionI &filterDim,

							 int marginLeft, int marginBottom,

							 int marginRight, int marginTop,

							 TDimensionI &outDim);

	/*- 露光値をdepremultipy→RGB値(０〜１)に戻す→premultiply -*/

	void convertExposureToRGB_CPU(float4 *out_tile_p,

								  TDimensionI &dim, float hardness);

	/*- 背景があり、前景が動かない場合、単純にOverする -*/

	void composeWithNoMotion(TTile &tile,

							 double frame,

							 const TRenderSettings &settings);

	/*- 背景を露光値にして通常合成 -*/

	void composeBackgroundExposure_CPU(float4 *out_tile_p,

									   TDimensionI &enlargedDimIn,

									   int marginRight, int marginTop,

									   TTile &back_tile,

									   TDimensionI &dimOut,

									   float hardness);

public:

	Iwa_MotionBlurCompFx();

	void doCompute(TTile &tile,

				   double frame,

				   const TRenderSettings &settings);

	void doCompute_CPU(TTile &tile,

					   double frame,

					   const TRenderSettings &settings,

					   float4 *pointsTable,

					   int pointAmount,

					   double hardness,

					   double shutterStart, double shutterEnd,

					   int traceResolution,

					   float startValue, float startCurve,

					   float endValue, float endCurve,

					   int marginLeft, int marginRight,

					   int marginTop, int marginBottom,

					   TDimensionI &enlargedDimIn,

					   TTile &enlarge_tile,

					   TDimensionI &dimOut,

					   TDimensionI &filterDim,

					   TTile &back_tile);

	bool doGetBBox(double frame,

				   TRectD &bBox,

				   const TRenderSettings &info);

	bool canHandle(const TRenderSettings &info,

				   double frame);

	/*- 参考にしているオブジェクトが動いている可能性があるので、

		エイリアスは毎フレーム変える -*/

	std::string getAlias(double frame, const TRenderSettings &info) const;

};

#endif