#pragma once

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

 Iwa_SpectrumFx

 参照画像を位相差として、干渉色を出力する

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

#ifndef IWA_SPECTRUM_H

#define IWA_SPECTRUM_H

#include "stdfx.h"

#include "tfxparam.h"

struct float3 {

  float x, y, z;

  float3 operator*(const float &a) { return {x * a, y * a, z * a}; }

  float3 operator+(const float3 &a) { return {x + a.x, y + a.y, z + a.z}; }

};

struct float4 {

  float x, y, z, w;

};

class Iwa_SpectrumFx : public TStandardRasterFx {

  FX_PLUGIN_DECLARATION(Iwa_SpectrumFx)

protected:

  TRasterFxPort m_input; /*- 位相差マップの入力 -*/

  TRasterFxPort m_light; /*- 光源用マップの入力 -*/

  TDoubleParamP m_intensity;

  TDoubleParamP m_refractiveIndex;

  TDoubleParamP m_thickMax;

  TDoubleParamP m_thickMin;

  TDoubleParamP m_RGamma;

  TDoubleParamP m_GGamma;

  TDoubleParamP m_BGamma;

  TDoubleParamP m_loopSpectrumFadeWidth;

  TDoubleParamP m_spectrumShift;

  TDoubleParamP m_lensFactor;

  TDoubleParamP m_lightThres;

  TDoubleParamP m_lightIntensity;

  /*- シャボン色マップの生成 -*/

  void calcBubbleMap(float3 *bubbleColor, double frame, bool isLinear,

                     double colorSpaceGamma, bool computeAngularAxis = false);

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

  void convertRaster(const RASTER ras, TDimensionI dim, float3 *bubbleColor);

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

  void convertRasterWithLight(const RASTER ras, TDimensionI dim,

                              float3 *bubbleColor, const RASTER lightRas,

                              float lightThres, float lightIntensity);

  /*- 素材タイルを０〜１に正規化して格納 -*/

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

  void setSourceRasters(const RASTER ras, float4 *in_out_tile_host,

                        const RASTER light_ras, float4 *light_host,

                        TDimensionI dim, bool useLight);

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

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

  void outputRasters(const RASTER outRas, float4 *in_out_tile_host,

                     TDimensionI dim);

public:

  Iwa_SpectrumFx();

  void doCompute(TTile &tile, double frame,

                 const TRenderSettings &settings) override;

  void doCompute_CUDA(TTile &tile, double frame,

                      const TRenderSettings &settings);

  bool doGetBBox(double frame, TRectD &bBox,

                 const TRenderSettings &info) override;

  bool canHandle(const TRenderSettings &info, double frame) override;

};

#endif