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

Iwa_MotionBlurCompFx

Motion blur fx considering the exposure of light and trajectory of the object.

Available to composite exposure value with background.

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

#include "iwa_motionblurfx.h"

#include "tfxattributes.h"

#include "toonz/tstageobject.h"

#include "trop.h"

/* Normalize the source image to 0 - 1 and read it into the host memory.

 * Check if the source image is premultiped or not here, if it is not specified

 * in the combo box. */

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

bool Iwa_MotionBlurCompFx::setSourceRaster(const RASTER srcRas, float4 *dstMem,

                                           TDimensionI dim,

                                           PremultiTypes type) {

  bool isPremultiplied = (type == SOURCE_IS_NOT_PREMUTIPLIED) ? false : true;

  float4 *chann_p = dstMem;

  float threshold = 100.0 / (float)TPixel64::maxChannelValue;

  int max = 0;

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

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

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

      (*chann_p).x = (float)pix->r / (float)PIXEL::maxChannelValue;

      (*chann_p).y = (float)pix->g / (float)PIXEL::maxChannelValue;

      (*chann_p).z = (float)pix->b / (float)PIXEL::maxChannelValue;

      (*chann_p).w = (float)pix->m / (float)PIXEL::maxChannelValue;

      /* If there are pixels whose RGB values ​​are larger than the alpha

       * channel, determine the source is not premutiplied */

      if (type == AUTO && isPremultiplied &&

          (((*chann_p).x > (*chann_p).w && (*chann_p).x > threshold) ||

           ((*chann_p).y > (*chann_p).w && (*chann_p).y > threshold) ||

           ((*chann_p).z > (*chann_p).w && (*chann_p).z > threshold)))

        isPremultiplied = false;

      pix++;

      chann_p++;

    }

  }

  if (isPremultiplied) {

    chann_p = dstMem;

    for (int i = 0; i < dim.lx * dim.ly; i++, chann_p++) {

      if ((*chann_p).x > (*chann_p).w) (*chann_p).x = (*chann_p).w;

      if ((*chann_p).y > (*chann_p).w) (*chann_p).y = (*chann_p).w;

      if ((*chann_p).z > (*chann_p).w) (*chann_p).z = (*chann_p).w;

    }

  }

  return isPremultiplied;

}

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

 Convert output result to Channel value and store it in the tile

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

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

void Iwa_MotionBlurCompFx::setOutputRaster(float4 *srcMem, const RASTER dstRas,

                                           TDimensionI dim, int2 margin) {

  int out_j = 0;

  for (int j = margin.y; j < dstRas->getLy() + margin.y; j++, out_j++) {

    PIXEL *pix     = dstRas->pixels(out_j);

    float4 *chan_p = srcMem;

    chan_p += j * dim.lx + margin.x;

    for (int i = 0; i < dstRas->getLx(); i++) {

      float val;

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

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

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

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

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

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

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

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

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

      val    = (*chan_p).w * (float)PIXEL::maxChannelValue + 0.5f;

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

                                             ? (float)PIXEL::maxChannelValue

                                             : val);

      pix++;

      chan_p++;

    }

  }

}

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

 Create and normalize filters

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

void Iwa_MotionBlurCompFx::makeMotionBlurFilter_CPU(

    float *filter_p, TDimensionI &filterDim, int marginLeft, int marginBottom,

    float4 *pointsTable, int pointAmount, float startValue, float startCurve,

    float endValue, float endCurve) {

  /* Variable for adding filter value*/

  float fil_val_sum = 0.0f;

  /* The current filter position to be looped in the 'for' statement */

  float *current_fil_p = filter_p;

  /* For each coordinate in the filter */

  for (int fily = 0; fily < filterDim.ly; fily++) {

    for (int filx = 0; filx < filterDim.lx; filx++, current_fil_p++) {

      /* Get filter coordinates */

      float2 pos = {static_cast<float>(filx - marginLeft),</float>

                    static_cast<float>(fily - marginBottom)};</float>

      /* Value to be updated */

      float nearestDist2         = 100.0f;

      int nearestIndex           = -1;

      float nearestFramePosRatio = 0.0f;

      /* Find the nearest point for each pair of sample points */

      for (int v = 0; v < pointAmount - 1; v++) {

        float4 p0 = pointsTable[v];

        float4 p1 = pointsTable[v + 1];

        /* If it is not within the range, continue */

        if (pos.x < std::min(p0.x, p1.x) - 1.0f ||

            pos.x > std::max(p0.x, p1.x) + 1.0f ||

            pos.y < std::min(p0.y, p1.y) - 1.0f ||

            pos.y > std::max(p0.y, p1.y) + 1.0f)

          continue;

        /* Since it is within the range, obtain the distance between the line

         * segment and the point. */

        /* Calculate the inner product of 'p0'->sampling point and 'p0'->'p1' */

        float2 vec_p0_sample = {static_cast<float>(pos.x - p0.x),</float>

                                static_cast<float>(pos.y - p0.y)};</float>

        float2 vec_p0_p1 = {static_cast<float>(p1.x - p0.x),</float>

                            static_cast<float>(p1.y - p0.y)};</float>

        float dot =

            vec_p0_sample.x * vec_p0_p1.x + vec_p0_sample.y * vec_p0_p1.y;

        /* Calculate the square of distance */

        float dist2;

        float framePosRatio;

        /* If it is before 'p0' */

        if (dot <= 0.0f) {

          dist2 = vec_p0_sample.x * vec_p0_sample.x +

                  vec_p0_sample.y * vec_p0_sample.y;

          framePosRatio = 0.0f;

        } else {

          /* Calculate the square of the length of the trajectory vector */

          float length2 = p0.z * p0.z;

          /* If it is between 'p0' and 'p1'

           * If the trajectory at p is a point,

           * 'length2' becomes 0, so it will never fall into this condition.

           * So, there should not be worry of becoming ZeroDivide. */

          if (dot < length2) {

            float p0_sample_dist2 = vec_p0_sample.x * vec_p0_sample.x +

                                    vec_p0_sample.y * vec_p0_sample.y;

            dist2         = p0_sample_dist2 - dot * dot / length2;

            framePosRatio = dot / length2;

          }

          /* If it is before 'p1' */

          else {

            float2 vec_p1_sample = {pos.x - p1.x, pos.y - p1.y};

            dist2                = vec_p1_sample.x * vec_p1_sample.x +

                    vec_p1_sample.y * vec_p1_sample.y;

            framePosRatio = 1.0f;

          }

        }

        /* If the distance is farther than (√ 2 + 1) / 2, continue

         * Because it is a comparison with dist2, the value is squared */

        if (dist2 > 1.4571f) continue;

        /* Update if distance is closer */

        if (dist2 < nearestDist2) {

          nearestDist2         = dist2;

          nearestIndex         = v;

          nearestFramePosRatio = framePosRatio;

        }

      }

      /* If neighborhood vector of the current pixel can not be found,

       * set the filter value to 0 and return */

      if (nearestIndex == -1) {

        *current_fil_p = 0.0f;

        continue;

      }

      /* Count how many subpixels (16 * 16) of the current pixel

       * are in the range 0.5 from the neighborhood vector.

       */

      int count  = 0;

      float4 np0 = pointsTable[nearestIndex];

      float4 np1 = pointsTable[nearestIndex + 1];

      for (int yy = 0; yy < 16; yy++) {

        /* Y coordinate of the subpixel */

        float subPosY = pos.y + ((float)yy - 7.5f) / 16.0f;

        for (int xx = 0; xx < 16; xx++) {

          /* X coordinate of the subpixel */

          float subPosX = pos.x + ((float)xx - 7.5f) / 16.0f;

          float2 vec_np0_sub = {subPosX - np0.x, subPosY - np0.y};

          float2 vec_np0_np1 = {np1.x - np0.x, np1.y - np0.y};

          float dot =

              vec_np0_sub.x * vec_np0_np1.x + vec_np0_sub.y * vec_np0_np1.y;

          /* Calculate the square of the distance */

          float dist2;

          /* If it is before 'p0' */

          if (dot <= 0.0f)

            dist2 =

                vec_np0_sub.x * vec_np0_sub.x + vec_np0_sub.y * vec_np0_sub.y;

          else {

            /* Compute the square of the length of the trajectory vector */

            float length2 = np0.z * np0.z;

            /* If it is between 'p0' and 'p1' */

            if (dot < length2) {

              float np0_sub_dist2 =

                  vec_np0_sub.x * vec_np0_sub.x + vec_np0_sub.y * vec_np0_sub.y;

              dist2 = np0_sub_dist2 - dot * dot / length2;

            }

            /* if it is before 'p1' */

            else {

              float2 vec_np1_sub = {subPosX - np1.x, subPosY - np1.y};

              dist2 =

                  vec_np1_sub.x * vec_np1_sub.x + vec_np1_sub.y * vec_np1_sub.y;

            }

          }

          /* Increment count if squared distance is less than 0.25 */

          if (dist2 <= 0.25f) count++;

        }

      }

      /* 'safeguard' - If the count is 0, set the field value to 0 and return.

       */

      if (count == 0) {

        *current_fil_p = 0.0f;

        continue;

      }

      /* Count is 256 at a maximum */

      float countRatio = (float)count / 256.0f;

      /* The brightness of the filter value is inversely proportional

       * to the area of ​​the line of width 1 made by the vector.

       *

       * Since there are semicircular caps with radius 0.5

       * before and after the vector, it will never be 0-divide

       * even if the length of vector is 0.

       */

      /* Area of the neighborhood vector when width = 1 */

      float vecMenseki = 0.25f * 3.14159265f + np0.z;

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

      /* Next, get the value of the gamma strength */

      /* Offset value of the frame of the neighbor point */

      float curveValue;

      float frameOffset =

          np0.w * (1.0f - nearestFramePosRatio) + np1.w * nearestFramePosRatio;

      /* If the frame is exactly at the frame origin,

       * or there is no attenuation value, set curveValue to 1 */

      if (frameOffset == 0.0f || (frameOffset < 0.0f && startValue == 1.0f) ||

          (frameOffset > 0.0f && endValue == 1.0f))

        curveValue = 1.0f;

      else {

        /* Change according to positive / negative of offset */

        float value, curve, ratio;

        if (frameOffset < 0.0f) /* Start side */

        {

          value = startValue;

          curve = startCurve;

          ratio = 1.0f - (frameOffset / pointsTable[0].w);

        } else {

          value = endValue;

          curve = endCurve;

          ratio = 1.0f - (frameOffset / pointsTable[pointAmount - 1].w);

        }

        curveValue = value + (1.0f - value) * powf(ratio, 1.0f / curve);

      }

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

      /* Store field value */

      *current_fil_p = curveValue * countRatio / vecMenseki;

      fil_val_sum += *current_fil_p;

    }

  }

  /* Normalization */

  current_fil_p = filter_p;

  for (int f = 0; f < filterDim.lx * filterDim.ly; f++, current_fil_p++) {

    *current_fil_p /= fil_val_sum;

  }

}

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

 Create afterimage filter and normalize

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

void Iwa_MotionBlurCompFx::makeZanzoFilter_CPU(

    float *filter_p, TDimensionI &filterDim, int marginLeft, int marginBottom,

    float4 *pointsTable, int pointAmount, float startValue, float startCurve,

    float endValue, float endCurve) {

  /* Variable for adding filter value */

  float fil_val_sum = 0.0f;

  /* The current filter position to be looped in the 'for' statement */

  float *current_fil_p = filter_p;

  /* For each coordinate in the filter */

  for (int fily = 0; fily < filterDim.ly; fily++) {

    for (int filx = 0; filx < filterDim.lx; filx++, current_fil_p++) {

      /* Get filter coordinates */

      float2 pos = {(float)(filx - marginLeft), (float)(fily - marginBottom)};

      /* Variable to be accumulated */

      float filter_sum = 0.0f;

      /* Measure the distance for each sample point and accumulate the density

       */

      for (int v = 0; v < pointAmount; v++) {

        float4 p0 = pointsTable[v];

        /* If it is not within the distance 1 around the coordinates of 'p0',

         * continue */

        if (pos.x < p0.x - 1.0f || pos.x > p0.x + 1.0f || pos.y < p0.y - 1.0f ||

            pos.y > p0.y + 1.0f)

          continue;

        /* Linear interpolation with 4 neighboring pixels */

        float xRatio = 1.0f - abs(pos.x - p0.x);

        float yRatio = 1.0f - abs(pos.y - p0.y);

        /* Next, get the value of the gamma strength */

        /* Offset value of the frame of the neighbor point */

        float curveValue;

        float frameOffset = p0.w;

        /* If the frame is exactly at the frame origin,

        * or there is no attenuation value, set curveValue to 1 */

        if (frameOffset == 0.0f || (frameOffset < 0.0f && startValue == 1.0f) ||

            (frameOffset > 0.0f && endValue == 1.0f))

          curveValue = 1.0f;

        else {

          /* Change according to positive / negative of offset */

          float value, curve, ratio;

          if (frameOffset < 0.0f) /* Start side */

          {

            value = startValue;

            curve = startCurve;

            ratio = 1.0f - (frameOffset / pointsTable[0].w);

          } else {

            value = endValue;

            curve = endCurve;

            ratio = 1.0f - (frameOffset / pointsTable[pointAmount - 1].w);

          }

          curveValue = value + (1.0f - value) * powf(ratio, 1.0f / curve);

        }

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

        /* Filter value integration */

        filter_sum += xRatio * yRatio * curveValue;

      }

      /* Store value */

      *current_fil_p = filter_sum;

      fil_val_sum += *current_fil_p;

    }

  }

  /* Normalization */

  current_fil_p = filter_p;

  for (int f = 0; f < filterDim.lx * filterDim.ly; f++, current_fil_p++) {

    *current_fil_p /= fil_val_sum;

  }

}

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

 Unpremultiply -> convert to exposure value -> premultiply

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

void Iwa_MotionBlurCompFx::convertRGBtoExposure_CPU(

    float4 *in_tile_p, TDimensionI &dim, float hardness,

    bool sourceIsPremultiplied) {

  float4 *cur_tile_p = in_tile_p;

  for (int i = 0; i < dim.lx * dim.ly; i++, cur_tile_p++) {

    /* if alpha is 0, return */

    if (cur_tile_p->w == 0.0f) {

      cur_tile_p->x = 0.0f;

      cur_tile_p->y = 0.0f;

      cur_tile_p->z = 0.0f;

      continue;

    }

    /* Unpremultiply on sources that are premultiplied, such as regular Level.

     * It is not done for 'digital overlay' (image with alpha mask added by

     * using Photoshop, as known as 'DigiBook' in Japanese animation industry)

     * etc. */

    if (sourceIsPremultiplied) {

      /* unpremultiply */

      cur_tile_p->x /= cur_tile_p->w;

      cur_tile_p->y /= cur_tile_p->w;

      cur_tile_p->z /= cur_tile_p->w;

    }

    /* convert RGB to Exposure */

    cur_tile_p->x = powf(10, (cur_tile_p->x - 0.5f) * hardness);

    cur_tile_p->y = powf(10, (cur_tile_p->y - 0.5f) * hardness);

    cur_tile_p->z = powf(10, (cur_tile_p->z - 0.5f) * hardness);

    /* Then multiply with the alpha channel */

    cur_tile_p->x *= cur_tile_p->w;

    cur_tile_p->y *= cur_tile_p->w;

    cur_tile_p->z *= cur_tile_p->w;

  }

}

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

 Filter and blur exposure values

 Loop for the range of 'outDim'.

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

void Iwa_MotionBlurCompFx::applyBlurFilter_CPU(

    float4 *in_tile_p, float4 *out_tile_p, TDimensionI &enlargedDim,

    float *filter_p, TDimensionI &filterDim, int marginLeft, int marginBottom,

    int marginRight, int marginTop, TDimensionI &outDim) {

  for (int i = 0; i < outDim.lx * outDim.ly; i++) {

    /* in_tile_dev and out_tile_dev contain data with dimensions lx * ly.

     * So, convert i to coordinates for output. */

    int2 outPos  = {i % outDim.lx + marginRight, i / outDim.lx + marginTop};

    int outIndex = outPos.y * enlargedDim.lx + outPos.x;

    /* put the result in out_tile_dev [outIndex] */

    /* Prepare a container to accumulate values ​​- */

    float4 value = {0.0f, 0.0f, 0.0f, 0.0f};

    /* Loop by filter size.

     * Note that the filter is used to 'collect' pixels at sample points

     * so flip the filter vertically and horizontally and sample it */

    int filterIndex = 0;

    for (int fily = -marginBottom; fily < filterDim.ly - marginBottom; fily++) {

      /* Sample coordinates and index of the end of this scan line */

      int2 samplePos  = {outPos.x + marginLeft, outPos.y - fily};

      int sampleIndex = samplePos.y * enlargedDim.lx + samplePos.x;

      for (int filx = -marginLeft; filx < filterDim.lx - marginLeft;

           filx++, filterIndex++, sampleIndex--) {

        /* If the filter value is 0 or the sample pixel is transparent, continue

         */

        if (filter_p[filterIndex] == 0.0f || in_tile_p[sampleIndex].w == 0.0f)

          continue;

        /* multiply the sample point value by the filter value and integrate */

        value.x += in_tile_p[sampleIndex].x * filter_p[filterIndex];

        value.y += in_tile_p[sampleIndex].y * filter_p[filterIndex];

        value.z += in_tile_p[sampleIndex].z * filter_p[filterIndex];

        value.w += in_tile_p[sampleIndex].w * filter_p[filterIndex];

      }

    }

    out_tile_p[outIndex].x = value.x;

    out_tile_p[outIndex].y = value.y;

    out_tile_p[outIndex].z = value.z;

    out_tile_p[outIndex].w = value.w;

  }

}

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

 Unpremultiply the exposure value

 -> convert back to RGB value (0 to 1)

 -> premultiply

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

void Iwa_MotionBlurCompFx::convertExposureToRGB_CPU(float4 *out_tile_p,

                                                    TDimensionI &dim,

                                                    float hardness) {

  float4 *cur_tile_p = out_tile_p;

  for (int i = 0; i < dim.lx * dim.ly; i++, cur_tile_p++) {

    /* if alpha is 0 return */

    if (cur_tile_p->w == 0.0f) {

      cur_tile_p->x = 0.0f;

      cur_tile_p->y = 0.0f;

      cur_tile_p->z = 0.0f;

      continue;

    }

    // unpremultiply

    cur_tile_p->x /= cur_tile_p->w;

    cur_tile_p->y /= cur_tile_p->w;

    cur_tile_p->z /= cur_tile_p->w;

    /* Convert Exposure to RGB value */

    cur_tile_p->x = log10f(cur_tile_p->x) / hardness + 0.5f;

    cur_tile_p->y = log10f(cur_tile_p->y) / hardness + 0.5f;

    cur_tile_p->z = log10f(cur_tile_p->z) / hardness + 0.5f;

    // multiply

    cur_tile_p->x *= cur_tile_p->w;

    cur_tile_p->y *= cur_tile_p->w;

    cur_tile_p->z *= cur_tile_p->w;

    /* Clamp */

    cur_tile_p->x = (cur_tile_p->x > 1.0f)

                        ? 1.0f

                        : ((cur_tile_p->x < 0.0f) ? 0.0f : cur_tile_p->x);

    cur_tile_p->y = (cur_tile_p->y > 1.0f)

                        ? 1.0f

                        : ((cur_tile_p->y < 0.0f) ? 0.0f : cur_tile_p->y);

    cur_tile_p->z = (cur_tile_p->z > 1.0f)

                        ? 1.0f

                        : ((cur_tile_p->z < 0.0f) ? 0.0f : cur_tile_p->z);

  }

}

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

 If there is a background, and the foreground does not move, simply over

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

void Iwa_MotionBlurCompFx::composeWithNoMotion(

    TTile &tile, double frame, const TRenderSettings &settings) {

  assert(m_background.isConnected());

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

  TTile fore_tile;

  m_input->allocateAndCompute(fore_tile, tile.m_pos,

                              tile.getRaster()->getSize(), tile.getRaster(),

                              frame, settings);

  TRasterP up(fore_tile.getRaster()), down(tile.getRaster());

  TRop::over(down, up);

}

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

  Convert background to exposure value and over it

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

void Iwa_MotionBlurCompFx::composeBackgroundExposure_CPU(

    float4 *out_tile_p, TDimensionI &enlargedDimIn, int marginRight,

    int marginTop, TTile &back_tile, TDimensionI &dimOut, float hardness) {

  /* Memory allocation of host */

  TRasterGR8P background_host_ras(sizeof(float4) * dimOut.lx, dimOut.ly);

  background_host_ras->lock();

  float4 *background_host = (float4 *)background_host_ras->getRawData();

  bool bgIsPremultiplied;

  /* normalize the background image to 0 - 1 and read it into the host memory

   */

  TRaster32P backRas32 = (TRaster32P)back_tile.getRaster();

  TRaster64P backRas64 = (TRaster64P)back_tile.getRaster();

  if (backRas32)

    bgIsPremultiplied = setSourceRaster<traster32p, tpixel32="">(</traster32p,>

        backRas32, background_host, dimOut);

  else if (backRas64)

    bgIsPremultiplied = setSourceRaster<traster64p, tpixel64="">(</traster64p,>

        backRas64, background_host, dimOut);

  float4 *bg_p = background_host;

  float4 *out_p;

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

    out_p = out_tile_p + ((marginTop + j) * enlargedDimIn.lx + marginRight);

    for (int i = 0; i < dimOut.lx; i++, bg_p++, out_p++) {

      /* if upper layer is completely opaque continue */

      if ((*out_p).w >= 1.0f) continue;

      /* even if the lower layer is completely transparent continue */

      if ((*bg_p).w < 0.0001f) continue;

      float3 bgExposure = {(*bg_p).x, (*bg_p).y, (*bg_p).z};

      /* Unpremultiply on sources that are premultiplied, such as regular Level.

       * It is not done for 'digital overlay' (image with alpha mask added by

       * using Photoshop, as known as 'DigiBook' in Japanese animation industry)

       * etc.

      */

      if (bgIsPremultiplied) {

        // Unpremultiply

        bgExposure.x /= (*bg_p).w;

        bgExposure.y /= (*bg_p).w;

        bgExposure.z /= (*bg_p).w;

      }

      /* Set Exposure to RGB value */

      bgExposure.x = powf(10, (bgExposure.x - 0.5f) * hardness);

      bgExposure.y = powf(10, (bgExposure.y - 0.5f) * hardness);

      bgExposure.z = powf(10, (bgExposure.z - 0.5f) * hardness);

      // multiply

      bgExposure.x *= (*bg_p).w;

      bgExposure.y *= (*bg_p).w;

      bgExposure.z *= (*bg_p).w;

      /* Over composte with front layers */

      (*out_p).x = (*out_p).x + bgExposure.x * (1.0f - (*out_p).w);

      (*out_p).y = (*out_p).y + bgExposure.y * (1.0f - (*out_p).w);

      (*out_p).z = (*out_p).z + bgExposure.z * (1.0f - (*out_p).w);

      /* Alpha value also over-composited */

      (*out_p).w = (*out_p).w + (*bg_p).w * (1.0f - (*out_p).w);

    }

  }

  background_host_ras->unlock();

}

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

Iwa_MotionBlurCompFx::Iwa_MotionBlurCompFx()

    : m_hardness(0.3)

    /* Parameters for blurring left and right */

    , m_startValue(1.0)

    , m_startCurve(1.0)

    , m_endValue(1.0)

    , m_endCurve(1.0)

    , m_zanzoMode(false)

    , m_premultiType(new TIntEnumParam(AUTO, "Auto")) {

  /* Bind common parameters */

  addInputPort("Source", m_input);

  addInputPort("Back", m_background);

  bindParam(this, "hardness", m_hardness);

  bindParam(this, "shutterStart", m_shutterStart);

  bindParam(this, "shutterEnd", m_shutterEnd);

  bindParam(this, "traceResolution", m_traceResolution);

  bindParam(this, "motionObjectType", m_motionObjectType);

  bindParam(this, "motionObjectIndex", m_motionObjectIndex);

  bindParam(this, "startValue", m_startValue);

  bindParam(this, "startCurve", m_startCurve);

  bindParam(this, "endValue", m_endValue);

  bindParam(this, "endCurve", m_endCurve);

  bindParam(this, "zanzoMode", m_zanzoMode);

  bindParam(this, "premultiType", m_premultiType);

  /* Common parameter range setting */

  m_hardness->setValueRange(0.05, 10.0);

  m_startValue->setValueRange(0.0, 1.0);

  m_startCurve->setValueRange(0.1, 10.0);

  m_endValue->setValueRange(0.0, 1.0);

  m_endCurve->setValueRange(0.1, 10.0);

  m_premultiType->addItem(SOURCE_IS_PREMULTIPLIED, "Source is premultiplied");

  m_premultiType->addItem(SOURCE_IS_NOT_PREMUTIPLIED,

                          "Source is NOT premultiplied");

  getAttributes()->setIsSpeedAware(true);

}

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

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

                                     const TRenderSettings &settings) {

  /* Do not process if not connected */

  if (!m_input.isConnected() && !m_background.isConnected()) {

    tile.getRaster()->clear();

    return;

  }

  /* For BG only connection */

  if (!m_input.isConnected()) {

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

    return;

  }

  /* Get parameters */

  QList<tpointd> points = getAttributes()->getMotionPoints();</tpointd>

  double hardness       = m_hardness->getValue(frame);

  double shutterStart   = m_shutterStart->getValue(frame);

  double shutterEnd     = m_shutterEnd->getValue(frame);

  int traceResolution   = m_traceResolution->getValue();

  float startValue      = (float)m_startValue->getValue(frame);

  float startCurve      = (float)m_startCurve->getValue(frame);

  float endValue        = (float)m_endValue->getValue(frame);

  float endCurve        = (float)m_endCurve->getValue(frame);

  /* Do not process if there are no more than two trajectory data */

  if (points.size() < 2) {

    if (!m_background.isConnected()) m_input->compute(tile, frame, settings);

    /* If there is a background and the foreground does not move, simply over */

    else

      composeWithNoMotion(tile, frame, settings);

    return;

  }

  /* Get display area */

  TRectD bBox =

      TRectD(tile.m_pos /* Render position on Pixel unit */

             ,

             TDimensionD(/* Size of Render image (Pixel unit) */

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

  /* Get upper, lower, left and right margin */

  double minX = 0.0;

  double maxX = 0.0;

  double minY = 0.0;

  double maxY = 0.0;

  for (int p = 0; p < points.size(); p++) {

    if (points.at(p).x > maxX) maxX = points.at(p).x;

    if (points.at(p).x < minX) minX = points.at(p).x;

    if (points.at(p).y > maxY) maxY = points.at(p).y;

    if (points.at(p).y < minY) minY = points.at(p).y;

  }

  int marginLeft   = (int)ceil(abs(minX));

  int marginRight  = (int)ceil(abs(maxX));

  int marginTop    = (int)ceil(abs(maxY));

  int marginBottom = (int)ceil(abs(minY));

  /* Return the input tile as-is if there is not movement

  * (= filter margins are all 0). */

  if (marginLeft == 0 && marginRight == 0 && marginTop == 0 &&

      marginBottom == 0) {

    if (!m_background.isConnected()) m_input->compute(tile, frame, settings);

    /* If there is a background, and the foreground does not move, simply over

     */

    else

      composeWithNoMotion(tile, frame, settings);

    return;

  }

  /* resize the bbox with the upper/lower/left/right inverted margins.

   */

  TRectD enlargedBBox(bBox.x0 - (double)marginRight,

                      bBox.y0 - (double)marginTop, bBox.x1 + (double)marginLeft,

                      bBox.y1 + (double)marginBottom);

  // std::cout<<"Margin Left:"<

  //	" Bottom:"<

  TDimensionI enlargedDimIn(/* Rounded to the nearest Pixel */

                            (int)(enlargedBBox.getLx() + 0.5),

                            (int)(enlargedBBox.getLy() + 0.5));

  TTile enlarge_tile;

  m_input->allocateAndCompute(enlarge_tile, enlargedBBox.getP00(),

                              enlargedDimIn, tile.getRaster(), frame, settings);

  /* If background is required */

  TTile back_Tile;

  if (m_background.isConnected()) {

    m_background->allocateAndCompute(back_Tile, tile.m_pos,

                                     tile.getRaster()->getSize(),

                                     tile.getRaster(), frame, settings);

  }

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

  /* Compute range */

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

  TDimensionI filterDim(marginLeft + marginRight + 1,

                        marginTop + marginBottom + 1);

  /* Release of pointsTable is done within each doCompute */

  int pointAmount     = points.size();

  float4 *pointsTable = new float4[pointAmount];

  float dt = (float)(shutterStart + shutterEnd) / (float)traceResolution;

  for (int p = 0; p < pointAmount; p++) {

    pointsTable[p].x = (float)points.at(p).x;

    pointsTable[p].y = (float)points.at(p).y;

    /* z stores the distance of p -> p + 1 vector */

    if (p < pointAmount - 1) {

      float2 vec = {(float)(points.at(p + 1).x - points.at(p).x),

                    (float)(points.at(p + 1).y - points.at(p).y)};

      pointsTable[p].z = sqrtf(vec.x * vec.x + vec.y * vec.y);

    }

    /* w stores shutter time offset */

    pointsTable[p].w = -(float)shutterStart + (float)p * dt;

  }

  doCompute_CPU(tile, frame, settings, pointsTable, pointAmount, hardness,

                shutterStart, shutterEnd, traceResolution, startValue,

                startCurve, endValue, endCurve, marginLeft, marginRight,

                marginTop, marginBottom, enlargedDimIn, enlarge_tile, dimOut,

                filterDim, back_Tile);

}

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

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

  /* Processing memory */

  float4 *in_tile_p;  /* With margin */

  float4 *out_tile_p; /* With margin */

  /* Filter */

  float *filter_p;

  /* Memory allocation */

  TRasterGR8P in_tile_ras(sizeof(float4) * enlargedDimIn.lx, enlargedDimIn.ly);

  in_tile_ras->lock();

  in_tile_p = (float4 *)in_tile_ras->getRawData();

  TRasterGR8P out_tile_ras(sizeof(float4) * enlargedDimIn.lx, enlargedDimIn.ly);

  out_tile_ras->lock();

  out_tile_p = (float4 *)out_tile_ras->getRawData();

  TRasterGR8P filter_ras(sizeof(float) * filterDim.lx, filterDim.ly);

  filter_ras->lock();

  filter_p = (float *)filter_ras->getRawData();

  bool sourceIsPremultiplied;

  /* normalize the source image to 0 - 1 and read it into memory */

  TRaster32P ras32 = (TRaster32P)enlarge_tile.getRaster();

  TRaster64P ras64 = (TRaster64P)enlarge_tile.getRaster();

  if (ras32)

    sourceIsPremultiplied = setSourceRaster<traster32p, tpixel32="">(</traster32p,>

        ras32, in_tile_p, enlargedDimIn,

        (PremultiTypes)m_premultiType->getValue());

  else if (ras64)

    sourceIsPremultiplied = setSourceRaster<traster64p, tpixel64="">(</traster64p,>

        ras64, in_tile_p, enlargedDimIn,

        (PremultiTypes)m_premultiType->getValue());

  /* When afterimage mode is off */

  if (!m_zanzoMode->getValue()) {

    /* Create and normalize filters */

    makeMotionBlurFilter_CPU(filter_p, filterDim, marginLeft, marginBottom,

                             pointsTable, pointAmount, startValue, startCurve,

                             endValue, endCurve);

  }

  /* When afterimage mode is ON */

  else {

    /* Create / normalize the afterimage filter */

    makeZanzoFilter_CPU(filter_p, filterDim, marginLeft, marginBottom,

                        pointsTable, pointAmount, startValue, startCurve,

                        endValue, endCurve);

  }

  delete[] pointsTable;

  /* Unpremultiply RGB value (0 to 1)

   * -> convert it to exposure value

   * -> premultiply again

   */

  convertRGBtoExposure_CPU(in_tile_p, enlargedDimIn, hardness,

                           sourceIsPremultiplied);

  /* Filter and blur exposure value */

  applyBlurFilter_CPU(in_tile_p, out_tile_p, enlargedDimIn, filter_p, filterDim,

                      marginLeft, marginBottom, marginRight, marginTop, dimOut);

  /* Memory release */

  in_tile_ras->unlock();

  filter_ras->unlock();

  /* If there is a background, do Exposure multiplication */

  if (m_background.isConnected()) {

    composeBackgroundExposure_CPU(out_tile_p, enlargedDimIn, marginRight,

                                  marginTop, back_tile, dimOut,

                                  (float)hardness);

  }

  /* Unpremultiply the exposure value

   * -> convert back to RGB value (0 to 1)

   * -> premultiply */

  convertExposureToRGB_CPU(out_tile_p, enlargedDimIn, hardness);

  /* Clear raster */

  tile.getRaster()->clear();

  TRaster32P outRas32 = (TRaster32P)tile.getRaster();

  TRaster64P outRas64 = (TRaster64P)tile.getRaster();

  int2 margin         = {marginRight, marginTop};

  if (outRas32)

    setOutputRaster<traster32p, tpixel32="">(out_tile_p, outRas32, enlargedDimIn,</traster32p,>

                                          margin);

  else if (outRas64)

    setOutputRaster<traster64p, tpixel64="">(out_tile_p, outRas64, enlargedDimIn,</traster64p,>

                                          margin);

  /* Memory release */

  out_tile_ras->unlock();

}

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

bool Iwa_MotionBlurCompFx::doGetBBox(double frame, TRectD &bBox,

                                     const TRenderSettings &info) {

  if (!m_input.isConnected() && !m_background.isConnected()) {

    bBox = TRectD();

    return false;

  }

  /* Rough implemetation - return infinite size if the background is connected

   */

  if (m_background.isConnected()) {

    bool _ret = m_background->doGetBBox(frame, bBox, info);

    bBox      = TConsts::infiniteRectD;

    return _ret;