#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 conerted 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 RASTER, typename PIXEL>
  void convertToBrightness(const RASTER srcRas, float* dst, float* alpha,
                           TDimensionI dim);
 
  template <typename RASTER, typename PIXEL>
  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,
                    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);
 
  void do_createBlurFilter(float* dst_p, int size, float radius);
 
  void do_applyFilter(float* depth_map_p, TDimensionI dim, float* distace_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,
                      const QList<QSize>& noise_base_resolution,
                      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&,
                                   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);
 
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