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

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.

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

#include "iwa_soapbubblefx.h"

#include "iwa_cie_d65.h"

#include "iwa_xyz.h"

#include "trop.h"

#include <qlist></qlist>

#include <qpoint></qpoint>

#include <qsize></qsize>

#include <qrect></qrect>

namespace {

const float PI = 3.14159265f;

#define INF 1e20 /* less than FLT_MAX */

/* dt of 1d function using squared distance */

static float* dt(float* f, int n, float a = 1.0f) {

  float* d = new float[n];

  int* v   = new int[n];

  float* z = new float[n + 1];

  /* index of rightmost parabola in lower envelope */

  int k = 0;

  /* locations of parabolas in lower envelope */

  v[0] = 0;

  /* locations of boundaries between parabolas */

  z[0] = -INF;

  z[1] = +INF;

  /* compute lower envelope */

  for (int q = 1; q <= n - 1; q++) {

    /* compute intersection */

    float s =

        ((f[q] / a + q * q) - (f[v[k]] / a + v[k] * v[k])) / (2 * q - 2 * v[k]);

    while (s <= z[k]) {

      k--;

      s = ((f[q] / a + q * q) - (f[v[k]] / a + v[k] * v[k])) /

          (2 * q - 2 * v[k]);

    }

    k++;

    v[k]     = q;

    z[k]     = s;

    z[k + 1] = +INF;

  }

  k = 0;

  /* fill in values of distance transform */

  for (int q = 0; q <= n - 1; q++) {

    while (z[k + 1] < q) k++;

    d[q] = a * (q - v[k]) * (q - v[k]) + f[v[k]];

  }

  delete[] v;

  delete[] z;

  return d;

}

}  // namespace

//------------------------------------

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

void Iwa_SoapBubbleFx::convertToRaster(const RASTER ras, float* thickness_map_p,

                                       float* depth_map_p, float* alpha_map_p,

                                       TDimensionI dim, float3* bubbleColor_p) {

  int renderMode     = m_renderMode->getValue();

  float* depth_p     = depth_map_p;

  float* thickness_p = thickness_map_p;

  float* alpha_p     = alpha_map_p;

  for (int j = 0; j < dim.ly; j++) {

    PIXEL* pix = ras->pixels(j);

    for (int i = 0; i < dim.lx;

         i++, depth_p++, thickness_p++, alpha_p++, pix++) {

      float alpha = (*alpha_p);

      if (!m_fit_thickness->getValue())

        alpha *= (float)pix->m / (float)PIXEL::maxChannelValue;

      if (alpha == 0.0f) { /* no change for the transparent pixels */

        pix->m = (typename PIXEL::Channel)0;

        continue;

      }

      // thickness and depth render mode

      if (renderMode != RENDER_MODE_BUBBLE) {

        float val = alpha * (float)PIXEL::maxChannelValue + 0.5f;

        pix->m = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue)

                                               ? (float)PIXEL::maxChannelValue

                                               : val);

        float mapVal =

            (renderMode == RENDER_MODE_THICKNESS) ? (*thickness_p) : (*depth_p);

        val = alpha * mapVal * (float)PIXEL::maxChannelValue + 0.5f;

        typename PIXEL::Channel chanVal =

            (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue)

                                          ? (float)PIXEL::maxChannelValue

                                          : val);

        pix->r = chanVal;

        pix->g = chanVal;

        pix->b = chanVal;

        continue;

      }

      float coordinate[2];

      coordinate[0] = 256.0f * std::min(1.0f, *depth_p);

      coordinate[1] = 256.0f * std::min(1.0f, *thickness_p);

      int neighbors[2][2];

      /* interpolate sampling */

      if (coordinate[0] <= 0.5f)

        neighbors[0][0] = 0;

      else

        neighbors[0][0] = (int)std::floor(coordinate[0] - 0.5f);

      if (coordinate[0] >= 255.5f)

        neighbors[0][1] = 255;

      else

        neighbors[0][1] = (int)std::floor(coordinate[0] + 0.5f);

      if (coordinate[1] <= 0.5f)

        neighbors[1][0] = 0;

      else

        neighbors[1][0] = (int)std::floor(coordinate[1] - 0.5f);

      if (coordinate[1] >= 255.5f)

        neighbors[1][1] = 255;

      else

        neighbors[1][1] = (int)std::floor(coordinate[1] + 0.5f);

      float interp_ratio[2];

      interp_ratio[0] = coordinate[0] - 0.5f - std::floor(coordinate[0] - 0.5f);

      interp_ratio[1] = coordinate[1] - 0.5f - std::floor(coordinate[1] - 0.5f);

      float3 nColors[4] = {

          bubbleColor_p[neighbors[0][0] * 256 + neighbors[1][0]],

          bubbleColor_p[neighbors[0][1] * 256 + neighbors[1][0]],

          bubbleColor_p[neighbors[0][0] * 256 + neighbors[1][1]],

          bubbleColor_p[neighbors[0][1] * 256 + neighbors[1][1]]};

      float3 color =

          nColors[0] * (1.0f - interp_ratio[0]) * (1.0f - interp_ratio[1]) +

          nColors[1] * interp_ratio[0] * (1.0f - interp_ratio[1]) +

          nColors[2] * (1.0f - interp_ratio[0]) * interp_ratio[1] +

          nColors[3] * interp_ratio[0] * interp_ratio[1];

      /* clamp */

      float val = alpha * (float)PIXEL::maxChannelValue + 0.5f;

      pix->m = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue)

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

      val    = alpha * color.x * (float)PIXEL::maxChannelValue + 0.5f;

      pix->r = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue)

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

      val    = alpha * color.y * (float)PIXEL::maxChannelValue + 0.5f;

      pix->g = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue)

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

      val    = alpha * color.z * (float)PIXEL::maxChannelValue + 0.5f;

      pix->b = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue)

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

    }

  }

}

// specialization for floating point

template <>

void Iwa_SoapBubbleFx::convertToRaster<trasterfp, tpixelf="">(</trasterfp,>

    const TRasterFP ras, float* thickness_map_p, float* depth_map_p,

    float* alpha_map_p, TDimensionI dim, float3* bubbleColor_p) {

  int renderMode     = m_renderMode->getValue();

  float* depth_p     = depth_map_p;

  float* thickness_p = thickness_map_p;

  float* alpha_p     = alpha_map_p;

  for (int j = 0; j < dim.ly; j++) {

    TPixelF* pix = ras->pixels(j);

    for (int i = 0; i < dim.lx;

         i++, depth_p++, thickness_p++, alpha_p++, pix++) {

      float alpha = (*alpha_p);

      if (!m_fit_thickness->getValue()) alpha *= pix->m;

      if (alpha == 0.f) { /* no change for the transparent pixels */

        pix->m = 0.f;

        continue;

      }

      // thickness and depth render mode

      if (renderMode != RENDER_MODE_BUBBLE) {

        pix->m = std::min(alpha, 1.f);

        float mapVal =

            (renderMode == RENDER_MODE_THICKNESS) ? (*thickness_p) : (*depth_p);

        float chanVal = alpha * mapVal;

        pix->r        = chanVal;

        pix->g        = chanVal;

        pix->b        = chanVal;

        continue;

      }

      float coordinate[2];

      coordinate[0] = 256.0f * std::min(1.0f, *depth_p);

      coordinate[1] = 256.0f * std::min(1.0f, *thickness_p);

      int neighbors[2][2];

      /* interpolate sampling */

      if (coordinate[0] <= 0.5f)

        neighbors[0][0] = 0;

      else

        neighbors[0][0] = (int)std::floor(coordinate[0] - 0.5f);

      if (coordinate[0] >= 255.5f)

        neighbors[0][1] = 255;

      else

        neighbors[0][1] = (int)std::floor(coordinate[0] + 0.5f);

      if (coordinate[1] <= 0.5f)

        neighbors[1][0] = 0;

      else

        neighbors[1][0] = (int)std::floor(coordinate[1] - 0.5f);

      if (coordinate[1] >= 255.5f)

        neighbors[1][1] = 255;

      else

        neighbors[1][1] = (int)std::floor(coordinate[1] + 0.5f);

      float interp_ratio[2];

      interp_ratio[0] = coordinate[0] - 0.5f - std::floor(coordinate[0] - 0.5f);

      interp_ratio[1] = coordinate[1] - 0.5f - std::floor(coordinate[1] - 0.5f);

      float3 nColors[4] = {

          bubbleColor_p[neighbors[0][0] * 256 + neighbors[1][0]],

          bubbleColor_p[neighbors[0][1] * 256 + neighbors[1][0]],

          bubbleColor_p[neighbors[0][0] * 256 + neighbors[1][1]],

          bubbleColor_p[neighbors[0][1] * 256 + neighbors[1][1]]};

      float3 color =

          nColors[0] * (1.0f - interp_ratio[0]) * (1.0f - interp_ratio[1]) +

          nColors[1] * interp_ratio[0] * (1.0f - interp_ratio[1]) +

          nColors[2] * (1.0f - interp_ratio[0]) * interp_ratio[1] +

          nColors[3] * interp_ratio[0] * interp_ratio[1];

      /* clamp */

      pix->m = std::min(alpha, 1.f);

      pix->r = alpha * color.x;

      pix->g = alpha * color.y;

      pix->b = alpha * color.z;

    }

  }

}

//------------------------------------

Iwa_SoapBubbleFx::Iwa_SoapBubbleFx()

    : Iwa_SpectrumFx()

    , m_renderMode(new TIntEnumParam(RENDER_MODE_BUBBLE, "Bubble"))

    , m_binarize_threshold(0.5)

    , m_shape_aspect_ratio(1.0)

    , m_blur_radius(5.0)

    , m_blur_power(0.5)

    , m_multi_source(false)

    , m_mask_center(false)  // obsolete

    , m_center_opacity(1.0)

    , m_fit_thickness(false)

    , m_normal_sample_distance(1)

    , m_noise_sub_depth(3)

    , m_noise_resolution_s(18.0)

    , m_noise_resolution_t(5.0)

    , m_noise_sub_composite_ratio(0.5)

    , m_noise_evolution(0.0)

    , m_noise_depth_mix_ratio(0.05)

    , m_noise_thickness_mix_ratio(0.05) {

  removeInputPort("Source");

  removeInputPort("Light"); /* not used */

  addInputPort("Thickness", m_input);

  addInputPort("Shape", m_shape);

  addInputPort("Depth", m_depth);

  bindParam(this, "renderMode", m_renderMode);

  m_renderMode->addItem(RENDER_MODE_THICKNESS, "Thickness");

  m_renderMode->addItem(RENDER_MODE_DEPTH, "Depth");

  bindParam(this, "binarizeThresold", m_binarize_threshold);

  bindParam(this, "shapeAspectRatio", m_shape_aspect_ratio);

  bindParam(this, "blurRadius", m_blur_radius);

  bindParam(this, "blurPower", m_blur_power);

  bindParam(this, "multiSource", m_multi_source);

  bindParam(this, "maskCenter", m_mask_center, false, true);  // obsolete

  bindParam(this, "centerOpacity", m_center_opacity);

  bindParam(this, "fitThickness", m_fit_thickness);

  bindParam(this, "normalSampleDistance", m_normal_sample_distance);

  bindParam(this, "noiseSubDepth", m_noise_sub_depth);

  bindParam(this, "noiseResolutionS", m_noise_resolution_s);

  bindParam(this, "noiseResolutionT", m_noise_resolution_t);

  bindParam(this, "noiseSubCompositeRatio", m_noise_sub_composite_ratio);

  bindParam(this, "noiseEvolution", m_noise_evolution);

  bindParam(this, "noiseDepthMixRatio", m_noise_depth_mix_ratio);

  bindParam(this, "noiseThicknessMixRatio", m_noise_thickness_mix_ratio);

  m_binarize_threshold->setValueRange(0.01, 0.99);

  m_shape_aspect_ratio->setValueRange(0.2, 5.0);

  m_blur_radius->setMeasureName("fxLength");

  m_blur_radius->setValueRange(0.0, 25.0);

  m_blur_power->setValueRange(0.01, 5.0);

  m_center_opacity->setValueRange(0.0, 1.0);

  m_normal_sample_distance->setValueRange(1, 20);

  m_noise_sub_depth->setValueRange(1, 5);

  m_noise_resolution_s->setValueRange(1.0, 40.0);

  m_noise_resolution_t->setValueRange(1.0, 20.0);

  m_noise_sub_composite_ratio->setValueRange(0.0, 5.0);

  m_noise_depth_mix_ratio->setValueRange(0.0, 1.0);

  m_noise_thickness_mix_ratio->setValueRange(0.0, 1.0);

}

//------------------------------------

void Iwa_SoapBubbleFx::doCompute(TTile& tile, double frame,

                                 const TRenderSettings& settings) {

  if (!m_input.isConnected()) return;

  if (!m_shape.isConnected() && !m_depth.isConnected()) return;

  TDimensionI dim(tile.getRaster()->getLx(), tile.getRaster()->getLy());

  TRectD bBox(tile.m_pos, TPointD(dim.lx, dim.ly));

  QList<trastergr8p> allocatedRasList;</trastergr8p>

  if (m_renderMode->getValue() == RENDER_MODE_DEPTH && m_depth.isConnected()) {

    m_depth->allocateAndCompute(tile, bBox.getP00(), dim, tile.getRaster(),

                                frame, settings);

    return;

  }

  /* soap bubble color map */

  TRasterGR8P bubbleColor_ras(sizeof(float3) * 256 * 256, 1);

  bubbleColor_ras->lock();

  allocatedRasList.append(bubbleColor_ras);

  float3* bubbleColor_p = (float3*)bubbleColor_ras->getRawData();

  if (m_renderMode->getValue() == RENDER_MODE_BUBBLE)

    calcBubbleMap(bubbleColor_p, frame, tile.getRaster()->isLinear(), true);

  if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

  /* depth map */

  TRasterGR8P depth_map_ras(sizeof(float) * dim.lx * dim.ly, 1);

  depth_map_ras->lock();

  allocatedRasList.append(depth_map_ras);

  float* depth_map_p = (float*)depth_map_ras->getRawData();

  /* alpha map */

  TRasterGR8P alpha_map_ras(sizeof(float) * dim.lx * dim.ly, 1);

  alpha_map_ras->lock();

  allocatedRasList.append(alpha_map_ras);

  float* alpha_map_p = (float*)alpha_map_ras->getRawData();

  /* region indices */

  TRasterGR8P regionIds_ras(sizeof(USHORT) * dim.lx * dim.ly, 1);

  regionIds_ras->lock();

  regionIds_ras->clear();

  allocatedRasList.append(regionIds_ras);

  USHORT* regionIds_p = (USHORT*)regionIds_ras->getRawData();

  QList<qrect> regionBoundingRects;</qrect>

  /* if the depth image is connected, use it */

  if (m_depth.isConnected()) {

    TTile depth_tile;

    m_depth->allocateAndCompute(depth_tile, bBox.getP00(), dim,

                                tile.getRaster(), frame, settings);

    if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

    TRasterP depthRas = depth_tile.getRaster();

    depthRas->lock();

    TRaster32P depthRas32 = (TRaster32P)depthRas;

    TRaster64P depthRas64 = (TRaster64P)depthRas;

    TRasterFP depthRasF   = (TRasterFP)depthRas;

    {

      if (depthRas32)

        convertToBrightness<traster32p, tpixel32="">(depthRas32, depth_map_p,</traster32p,>

                                                  alpha_map_p, dim);

      else if (depthRas64)

        convertToBrightness<traster64p, tpixel64="">(depthRas64, depth_map_p,</traster64p,>

                                                  alpha_map_p, dim);

      else if (depthRasF)

        convertToBrightness<trasterfp, tpixelf="">(depthRasF, depth_map_p,</trasterfp,>

                                                alpha_map_p, dim);

    }

    depthRas->unlock();

    // set one region covering whole camera rect

    regionBoundingRects.append(QRect(0, 0, dim.lx, dim.ly));

  }

  /* or, use the shape image to obtain pseudo depth */

  else { /* m_shape.isConnected */

    /* obtain shape image */

    TTile shape_tile;

    {

      TRenderSettings settings_aux(settings);

      settings_aux.m_bpp = 32;

      m_shape->allocateAndCompute(shape_tile, bBox.getP00(), dim, nullptr,

                                  frame, settings_aux);

    }

    if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

    processShape(frame, shape_tile, depth_map_p, alpha_map_p, regionIds_p,

                 regionBoundingRects, dim, settings);

  }

  if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

  // conpute thickness

  TRasterGR8P thickness_map_ras(sizeof(float) * dim.lx * dim.ly, 1);

  thickness_map_ras->lock();

  allocatedRasList.append(thickness_map_ras);

  float* thickness_map_p = (float*)thickness_map_ras->getRawData();

  TRasterP tileRas = tile.getRaster();

  TRaster32P ras32 = (TRaster32P)tileRas;

  TRaster64P ras64 = (TRaster64P)tileRas;

  TRasterFP rasF   = (TRasterFP)tileRas;

  if (m_fit_thickness->getValue()) {

    // Get the original bbox of thickness image

    TRectD thickBBox;

    m_input->getBBox(frame, thickBBox, settings);

    if (thickBBox == TConsts::infiniteRectD)

      thickBBox = TRectD(tile.m_pos, TDimensionD(tile.getRaster()->getLx(),

                                                 tile.getRaster()->getLy()));

    // Compute the thickenss tile.

    TTile thicknessTile;

    TDimension thickDim(static_cast<int>(thickBBox.getLx() + 0.5),</int>

                        static_cast<int>(thickBBox.getLy() + 0.5));</int>

    m_input->allocateAndCompute(thicknessTile, thickBBox.getP00(), thickDim,

                                tile.getRaster(), frame, settings);

    if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

    TRasterP thickRas = thicknessTile.getRaster();

    fitThicknessPatches(thickRas, thickDim, thickness_map_p, dim, regionIds_p,

                        regionBoundingRects);

  } else {

    /* compute the thickness input and temporarily store to the tile */

    m_input->compute(tile, frame, settings);

    if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

    if (ras32)

      convertToBrightness<traster32p, tpixel32="">(ras32, thickness_map_p, nullptr,</traster32p,>

                                                dim);

    else if (ras64)

      convertToBrightness<traster64p, tpixel64="">(ras64, thickness_map_p, nullptr,</traster64p,>

                                                dim);

    else if (rasF)

      convertToBrightness<trasterfp, tpixelf="">(rasF, thickness_map_p, nullptr,</trasterfp,>

                                              dim);

  }

  if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

  /* process noise */

  processNoise(thickness_map_p, depth_map_p, dim, frame, settings);

  if (checkCancelAndReleaseRaster(allocatedRasList, tile, settings)) return;

  if (ras32)

    convertToRaster<traster32p, tpixel32="">(ras32, thickness_map_p, depth_map_p,</traster32p,>

                                          alpha_map_p, dim, bubbleColor_p);

  else if (ras64)

    convertToRaster<traster64p, tpixel64="">(ras64, thickness_map_p, depth_map_p,</traster64p,>

                                          alpha_map_p, dim, bubbleColor_p);

  else if (rasF)

    convertToRaster<trasterfp, tpixelf="">(rasF, thickness_map_p, depth_map_p,</trasterfp,>

                                        alpha_map_p, dim, bubbleColor_p);

  for (int i = 0; i < allocatedRasList.size(); i++)

    allocatedRasList.at(i)->unlock();

}

//------------------------------------

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

void Iwa_SoapBubbleFx::convertToBrightness(const RASTER srcRas, float* dst,

                                           float* alpha, TDimensionI dim) {

  float* dst_p   = dst;

  float* alpha_p = alpha;

  for (int j = 0; j < dim.ly; j++) {

    PIXEL* pix = srcRas->pixels(j);

    for (int i = 0; i < dim.lx; i++, dst_p++, pix++) {

      float r = (float)pix->r / (float)PIXEL::maxChannelValue;

      float g = (float)pix->g / (float)PIXEL::maxChannelValue;

      float b = (float)pix->b / (float)PIXEL::maxChannelValue;

      /* brightness */

      *dst_p = 0.298912f * r + 0.586611f * g + 0.114478f * b;

      // clamp 0.f to 1.f in case computing HDR

      *dst_p = std::min(1.f, std::max(0.f, *dst_p));

      if (alpha) {

        *alpha_p = (float)pix->m / (float)PIXEL::maxChannelValue;

        alpha_p++;

      }

    }

  }

}

//------------------------------------

void Iwa_SoapBubbleFx::processShape(double frame, TTile& shape_tile,

                                    float* depth_map_p, float* alpha_map_p,

                                    USHORT* regionIds_p,

                                    QList<qrect>& regionBoundingRects,</qrect>

                                    TDimensionI dim,

                                    const TRenderSettings& settings) {

  TRaster32P shapeRas = shape_tile.getRaster();

  shapeRas->lock();

  TRasterGR8P distance_ras(sizeof(float) * dim.lx * dim.ly, 1);

  distance_ras->lock();

  float* distance_p = (float*)distance_ras->getRawData();

  float binarize_thres = (float)m_binarize_threshold->getValue(frame);

  int regionCount =

      do_binarize(shapeRas, regionIds_p, binarize_thres, distance_p,

                  alpha_map_p, regionBoundingRects, dim);

  shapeRas->unlock();

  if (settings.m_isCanceled && *settings.m_isCanceled) {

    distance_ras->unlock();

    return;

  }

  do_distance_transform(distance_p, regionIds_p, regionCount, dim, frame);

  if (settings.m_isCanceled && *settings.m_isCanceled) {

    distance_ras->unlock();

    return;

  }

  float center_opacity = (float)m_center_opacity->getValue(frame);

  if (center_opacity != 1.0f)

    applyDistanceToAlpha(distance_p, alpha_map_p, dim, center_opacity);

  /* create blur filter */

  float blur_radius = (float)m_blur_radius->getValue(frame) *

                      std::sqrt(std::abs((float)settings.m_affine.det()));

  /* if blur radius is 0, set the distance image to the depth image as-is */

  if (blur_radius == 0.0f) {

    float power      = (float)m_blur_power->getValue(frame);

    float* tmp_depth = depth_map_p;

    float* tmp_dist  = distance_p;

    USHORT* rid_p    = regionIds_p;

    for (int i = 0; i < dim.lx * dim.ly;

         i++, tmp_depth++, tmp_dist++, rid_p++) {

      if (*rid_p == 0)

        *tmp_depth = 0.0f;

      else

        *tmp_depth = 1.0f - std::pow(*tmp_dist, power);

    }

    distance_ras->unlock();

    return;

  }

  int blur_filter_size = (int)std::floor(blur_radius) * 2 + 1;

  TRasterGR8P blur_filter_ras(

      sizeof(float) * blur_filter_size * blur_filter_size, 1);

  blur_filter_ras->lock();

  float* blur_filter_p = (float*)blur_filter_ras->getRawData();

  do_createBlurFilter(blur_filter_p, blur_filter_size, blur_radius);

  if (settings.m_isCanceled && *settings.m_isCanceled) {

    blur_filter_ras->unlock();

    distance_ras->unlock();

    return;

  }

  /* blur filtering, normarize & power */

  do_applyFilter(depth_map_p, dim, distance_p, regionIds_p, blur_filter_p,

                 blur_filter_size, frame, settings);

  distance_ras->unlock();

  blur_filter_ras->unlock();

}

//------------------------------------

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

                                  float* distance_p, float* alpha_map_p,

                                  QList<qrect>& regionBoundingRects,</qrect>

                                  TDimensionI dim) {

  TPixel32::Channel channelThres =

      (TPixel32::Channel)(thres * (float)TPixel32::maxChannelValue);

  USHORT* tmp_p   = dst_p;

  float* tmp_dist = distance_p;

  float* alpha_p  = alpha_map_p;

  for (int j = 0; j < dim.ly; j++) {

    TPixel32* pix = srcRas->pixels(j);

    for (int i = 0; i < dim.lx; i++, pix++, tmp_p++, tmp_dist++, alpha_p++) {

      (*tmp_p)    = (pix->m > channelThres) ? 1 : 0;

      (*tmp_dist) = (*tmp_p == 1) ? INF : 0.0f;

      *alpha_p    = (float)pix->m / (float)TPixel32::maxChannelValue;

    }

  }

  // label regions when multi bubble option is on

  if (!m_multi_source->getValue()) {

    if (m_fit_thickness->getValue()) {

      regionBoundingRects.append(QRect());

      // calc boundingRect of the bubble

      QPoint topLeft(dim.lx, dim.ly);

      QPoint bottomRight(0, 0);

      USHORT* tmp_p = dst_p;

      for (int j = 0; j < dim.ly; j++) {

        for (int i = 0; i < dim.lx; i++, tmp_p++) {

          if ((*tmp_p) == 0) continue;

          if (topLeft.x() > i) topLeft.setX(i);

          if (bottomRight.x() < i) bottomRight.setX(i);

          if (topLeft.y() > j) topLeft.setY(j);

          if (bottomRight.y() < j) bottomRight.setY(j);

        }

      }

      regionBoundingRects.append(QRect(topLeft, bottomRight));

    }

    return 1;

  }

  QList<int> lut;</int>

  for (int i = 0; i < 65536; i++) lut.append(i);

  tmp_p           = dst_p;

  int regionCount = 0;

  for (int j = 0; j < dim.ly; j++) {

    for (int i = 0; i < dim.lx; i++, tmp_p++) {

      if ((*tmp_p) == 1) {

        int up   = (j == 0) ? 0 : *(tmp_p - dim.lx);

        int left = (i == 0) ? 0 : *(tmp_p - 1);

        assert(up >= 0 && left >= 0);

        if (!up && !left) {

          if (regionCount < 65535) regionCount++;

          (*tmp_p) = regionCount;

        } else if (up && !left)

          (*tmp_p) = up;

        else if (!up && left)

          (*tmp_p) = left;

        else if (up == left)

          (*tmp_p) = up;

        else if (up > left) {

          (*tmp_p) = left;

          lut[up]  = left;

        } else {

          (*tmp_p)  = up;

          lut[left] = up;

        }

      }

    }

  }

  // organize lut

  QList<int> convIndex;</int>

  int currentIndex = 0;

  for (int i = 0; i < 65536; i++) {

    if (lut.at(i) == i) {

      lut[i] = currentIndex;

      currentIndex++;

    } else

      convIndex.append(i);

  }

  for (int i = 0; i < convIndex.count(); i++)

    lut[convIndex.at(i)] = lut.at(lut.at(convIndex.at(i)));

  // apply lut

  int maxRegionIndex = 0;

  tmp_p              = dst_p;

  for (int j = 0; j < dim.ly; j++) {

    for (int i = 0; i < dim.lx; i++, tmp_p++) {

      (*tmp_p) = lut[*tmp_p];

      if (maxRegionIndex < (*tmp_p)) maxRegionIndex = (*tmp_p);

    }

  }

  // compute bounding boxes of each bubble

  if (m_fit_thickness->getValue()) {

    regionBoundingRects.append(QRect());

    USHORT* tmp_p = dst_p;

    for (int j = 0; j < dim.ly; j++) {

      for (int i = 0; i < dim.lx; i++, tmp_p++) {

        int rId = (*tmp_p);

        if (rId == 0) continue;

        while (regionBoundingRects.size() <= rId)

          regionBoundingRects.append(QRect());

        if (regionBoundingRects.at(rId).isNull())

          regionBoundingRects[rId].setRect(i, j, 1, 1);

        else {

          if (regionBoundingRects[rId].left() > i)

            regionBoundingRects[rId].setLeft(i);

          if (regionBoundingRects[rId].right() < i)

            regionBoundingRects[rId].setRight(i);

          if (regionBoundingRects[rId].top() > j)

            regionBoundingRects[rId].setTop(j);

          if (regionBoundingRects[rId].bottom() < j)

            regionBoundingRects[rId].setBottom(j);

        }

      }

    }

  }

  return maxRegionIndex;

}

//------------------------------------

void Iwa_SoapBubbleFx::do_createBlurFilter(float* dst_p, int size,

                                           float radius) {

  float radius2 = radius * radius;

  float* tmp_p  = dst_p;

  float sum     = 0.0f;

  int rad       = (size - 1) / 2;

  for (int j = -rad; j <= rad; j++) {

    for (int i = -rad; i <= rad; i++, tmp_p++) {

      float length2 = (float)i * (float)i + (float)j * (float)j;

      /* out of range */

      if (length2 >= radius2)

        *tmp_p = 0.0f;

      else {

        /* normalize distance from the filter center, to 0-1 */

        *tmp_p = 1.0f - std::sqrt(length2) / radius;

        sum += *tmp_p;

      }

    }

  }

  /* normalize */

  tmp_p = dst_p;

  for (int i = 0; i < size * size; i++, tmp_p++) {

    *tmp_p /= sum;

  }

}

//------------------------------------

void Iwa_SoapBubbleFx::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) {

  float power = (float)m_blur_power->getValue(frame);

  memset(depth_map_p, 0, sizeof(float) * dim.lx * dim.ly);

  int fil_margin = (blur_filter_size - 1) / 2;

  float* dst_p   = depth_map_p;

  USHORT* bin_p  = binarized_p;

  for (int j = 0; j < dim.ly; j++) {

    for (int i = 0; i < dim.lx; i++, dst_p++, bin_p++) {

      if (*bin_p == 0) continue;

      float* fil_p = blur_filter_p;

      for (int fy = j - fil_margin; fy <= j + fil_margin; fy++) {

        if (fy < 0 || fy >= dim.ly) {

          fil_p += blur_filter_size;

          continue;

        }

        for (int fx = i - fil_margin; fx <= i + fil_margin; fx++, fil_p++) {

          if (fx < 0 || fx >= dim.lx) continue;

          *dst_p += *fil_p * distance_p[fy * dim.lx + fx];

        }

      }

      /* power the value */

      *dst_p = 1.0f - std::pow(*dst_p, power);

    }

    if (settings.m_isCanceled && *settings.m_isCanceled) return;

  }

}

//------------------------------------

void Iwa_SoapBubbleFx::processNoise(float* thickness_map_p, float* depth_map_p,

                                    TDimensionI dim, double frame,

                                    const TRenderSettings& settings) {

  float noise_depth_mix_ratio = (float)m_noise_depth_mix_ratio->getValue(frame);

  float noise_thickness_mix_ratio =

      (float)m_noise_thickness_mix_ratio->getValue(frame);

  /* If the noise ratio is 0, do nothing and return */

  if (noise_depth_mix_ratio == 0.0f && noise_thickness_mix_ratio == 0.0f)

    return;

  int noise_sub_depth    = m_noise_sub_depth->getValue();

  int noise_resolution_s = (int)m_noise_resolution_s->getValue(frame);

  int noise_resolution_t = (int)m_noise_resolution_t->getValue(frame);

  float noise_composite_ratio =

      (float)m_noise_sub_composite_ratio->getValue(frame);

  float noise_evolution = (float)m_noise_evolution->getValue(frame);

  /* initialize the phase map */

  QList<int> noise_amount;</int>

  QList<qsize> noise_base_resolution;</qsize>

  int whole_noise_amount = 0;

  for (int layer = 0; layer < noise_sub_depth; layer++) {

    /* noise resolution */

    /* width: circumferential direction   height:distal direction */

    QSize size;

    size.setWidth(std::pow(2, layer) * noise_resolution_s);

    size.setHeight(std::pow(2, layer) * noise_resolution_t + 1);

    noise_base_resolution.append(size);

    int amount = size.width() * size.height();

    noise_amount.append(amount);

    whole_noise_amount += amount;

  }

  float* noise_phases = new float[whole_noise_amount];

  float* ph_p         = noise_phases;

  srand(0);

  /* Set the phase differences (0-2) */

  for (int i = 0; i < whole_noise_amount; i++, ph_p++) {

    *ph_p = (float)rand() / (float)RAND_MAX * 2.0f * PI;

  }

  /* make noise base */

  /* compute composite ratio of each layer */

  QList<float> comp_ratios;</float>

  comp_ratios.append(10.0f);

  float ratio_sum = 10.0f;

  for (int i = 1; i < noise_sub_depth; i++) {

    comp_ratios.append(comp_ratios.last() * noise_composite_ratio);

    ratio_sum += comp_ratios.last();

  }

  /* normalize */

  for (int i = 0; i < noise_sub_depth; i++) comp_ratios[i] /= ratio_sum;

  float* noise_base = new float[whole_noise_amount];

  float* nb_p = noise_base;

  ph_p        = noise_phases;

  /* for each sub-noise layer */

  for (int layer = 0; layer < noise_sub_depth; layer++) {

    float tmp_evolution = noise_evolution * (float)(layer + 1);

    for (int i = 0; i < noise_amount[layer]; i++, nb_p++, ph_p++) {

      *nb_p = comp_ratios[layer] * (cosf(tmp_evolution + *ph_p) / 2.0f + 0.5f);

    }

  }

  delete[] noise_phases;

  TRasterGR8P norm_angle_ras(sizeof(float) * dim.lx * dim.ly, 1);

  norm_angle_ras->lock();

  float* norm_angle_p = (float*)norm_angle_ras->getRawData();

  calc_norm_angle(norm_angle_p, depth_map_p, dim, settings.m_shrinkX);

  TRasterGR8P noise_map_ras(sizeof(float) * dim.lx * dim.ly, 1);

  noise_map_ras->lock();

  float* noise_map_p = (float*)noise_map_ras->getRawData();

  make_noise_map(noise_map_p, depth_map_p, norm_angle_p, dim, noise_amount,

                 noise_base_resolution, noise_sub_depth, noise_base);

  norm_angle_ras->unlock();

  delete[] noise_base;

  /* composite with perlin noise */

  add_noise(thickness_map_p, depth_map_p, dim, noise_map_p,

            noise_thickness_mix_ratio, noise_depth_mix_ratio);

  noise_map_ras->unlock();

}

//------------------------------------

void Iwa_SoapBubbleFx::calc_norm_angle(float* norm_angle_p, float* depth_map_p,

                                       TDimensionI dim, int shrink) {

  struct Locals {

    TDimensionI _dim;

    const float* _depth_p;

    float data(int x, int y) {

      if (x < 0 || _dim.lx <= x || y < 0 || _dim.ly <= y) return 0.0f;

      return _depth_p[y * _dim.lx + x];

    }