#pragma once

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

 Iwa_BloomFx

 Based on the LightBloom plugin fx by Dwango, with modifications as follows:

 - parameters are reduced & simplified to gamma, gain and size

 - reduced the block noise appeared especially when the gamma is with higher

value

 - enabled to render alpha gradation

 - enabled to adjust size with the viewer gadget

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

#ifndef IWA_BLOOMFX_H

#define IWA_BLOOMFX_H

#include "stdfx.h"

#include "tfxparam.h"

#include <opencv2 opencv.hpp=""></opencv2>

class Iwa_BloomFx : public TStandardRasterFx {

  FX_PLUGIN_DECLARATION(Iwa_BloomFx)

  enum AlphaMode { NoAlpha = 0, Light, LightAndSource };

protected:

  TRasterFxPort m_source;

  TDoubleParamP m_gamma;

  TDoubleParamP m_gammaAdjust;  // Gamma offset from the current color space

                                // gamma (Version 3)

  TBoolParamP m_auto_gain;

  TDoubleParamP m_gain_adjust;

  TDoubleParamP m_gain;

  TDoubleParamP m_decay;

  TDoubleParamP m_size;

  TIntEnumParamP m_alpha_mode;    // no alpha / light / light and source

  TBoolParamP m_alpha_rendering;  // obsolete

  double getSizePixelAmount(const double val, const TAffine affine);

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

  void setSourceTileToMat(const RASTER ras, cv::Mat &imgMat,

                          const double gamma);

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

  void setMatToOutput(const RASTER ras, const RASTER srcRas, cv::Mat &imgMat,

                      const double gamma, const double gain,

                      const AlphaMode alphaMode, const int margin);

  double computeAutoGain(cv::Mat &imgMat);

public:

  Iwa_BloomFx();

  void doCompute(TTile &tile, double frame,

                 const TRenderSettings &settings) override;

  bool doGetBBox(double frame, TRectD &bBox,

                 const TRenderSettings &info) override;

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

  void getParamUIs(TParamUIConcept *&concepts, int &length) override;

  void onObsoleteParamLoaded(const std::string ¶mName) override;

  void onFxVersionSet() override;

  bool toBeComputedInLinearColorSpace(bool settingsIsLinear,

                                      bool tileIsLinear) const override;

};

#endif