#pragma once

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

Iwa_SoapBubbleFx

Generates thin film interference colors from two reference images;

one is for thickness and the other one is for shape or normal vector

distribution of the film.

Inherits Iwa_SpectrumFx.

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

#ifndef IWA_SOAPBUBBLE_H

#define IWA_SOAPBUBBLE_H

#include "iwa_spectrumfx.h"

class Iwa_SoapBubbleFx final : public Iwa_SpectrumFx {

  FX_PLUGIN_DECLARATION(Iwa_SoapBubbleFx)

protected:

  /* target shape, used to create a pseudo normal vector */

  TRasterFxPort m_shape;

  /* another option, to input a depth map directly */

  TRasterFxPort m_depth;

  // rendering mode

  TIntEnumParamP m_renderMode;

  // shape parameters

  TDoubleParamP m_binarize_threshold;

  TDoubleParamP m_shape_aspect_ratio;

  TDoubleParamP m_blur_radius;

  TDoubleParamP m_blur_power;

  TBoolParamP m_multi_source;

  TDoubleParamP m_center_opacity;

  TBoolParamP m_fit_thickness;

  // obsolete parameter. to be converted to m_center_opacity

  TBoolParamP m_mask_center;

  // noise parameters

  TIntParamP m_normal_sample_distance;

  TIntParamP m_noise_sub_depth;

  TDoubleParamP m_noise_resolution_s;

  TDoubleParamP m_noise_resolution_t;

  TDoubleParamP m_noise_sub_composite_ratio;

  TDoubleParamP m_noise_evolution;

  TDoubleParamP m_noise_depth_mix_ratio;

  TDoubleParamP m_noise_thickness_mix_ratio;

  enum { RENDER_MODE_BUBBLE, RENDER_MODE_THICKNESS, RENDER_MODE_DEPTH };

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

  void convertToBrightness(const RASTER srcRas, float* dst, float* alpha,

                           TDimensionI dim);

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

  void convertToRaster(const RASTER ras, float* thickness_map_p,

                       float* depth_map_p, float* alpha_map_p, TDimensionI dim,

                       float3* bubbleColor_p);

  void processShape(double frame, TTile& shape_tile, float* depth_map_p,

                    float* alpha_map_p, USHORT* regionIds_p,

                    QList<qrect>& regionBoundingRects, TDimensionI dim,</qrect>

                    const TRenderSettings& settings);

  int do_binarize(TRaster32P srcRas, USHORT* dst_p, float thres,

                  float* distance_p, float* alpha_map_p,

                  QList<qrect>& regionBoundingRects, TDimensionI dim);</qrect>

  void do_createBlurFilter(float* dst_p, int size, float radius);

  void do_applyFilter(float* depth_map_p, TDimensionI dim, float* distance_p,

                      USHORT* binarized_p, float* blur_filter_p,

                      int blur_filter_size, double frame,

                      const TRenderSettings& settings);

  void processNoise(float* thickness_map_p, float* depth_map_p, TDimensionI dim,

                    double frame, const TRenderSettings& settings);

  void calc_norm_angle(float* norm_angle_p, float* depth_map_p, TDimensionI dim,

                       int shrink);

  void make_noise_map(float* noise_map_p, float* depth_map_p,

                      float* norm_angle_p, TDimensionI dim,

                      const QList<int>& noise_amount,</int>

                      const QList<qsize>& noise_base_resolution,</qsize>

                      int noise_sub_depth, float* noise_base);

  float noise_interp(int left, int right, int bottom, int top, float ratio_s,

                     float ratio_t, float* noise_layer_base, int noise_dim_x);

  void add_noise(float* thickness_map_p, float* depth_map_p, TDimensionI dim,

                 float* noise_map_p, float noise_thickness_mix_ratio,

                 float noise_depth_mix_ratio);

  void do_distance_transform(float* dst_p, USHORT* binarized_p, int regionCount,

                             TDimensionI dim, double frame);

  // if the rendering process is canceled, release raster and return TRUE

  bool checkCancelAndReleaseRaster(const QList<trastergr8p>&, TTile&,</trastergr8p>

                                   const TRenderSettings&);

  void applyDistanceToAlpha(float* distance_p, float* alpha_map_p,

                            TDimensionI dim, float center_opacity);

  void fitThicknessPatches(TRasterP thickRas, TDimensionI thickDim,

                           float* thickness_map_p, TDimensionI dim,

                           USHORT* regionIds_p,

                           QList<qrect>& regionBoundingRects);</qrect>

public:

  Iwa_SoapBubbleFx();

  void doCompute(TTile& tile, double frame,

                 const TRenderSettings& settings) override;

  // This will be called in TFx::loadData when obsolete "mask center" value is

  // loaded

  void onObsoleteParamLoaded(const std::string& paramName) override;

};

#endif