Blob Blame Raw
#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>

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 RASTER, typename PIXEL>
  void setSourceTileToMat(const RASTER ras, cv::Mat &imgMat,
                          const double gamma);

  template <typename RASTER, typename PIXEL>
  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 &paramName) override;
  void onFxVersionSet() override;
  bool toBeComputedInLinearColorSpace(bool settingsIsLinear,
                                      bool tileIsLinear) const override;
};

#endif