120 lines
No EOL
4.5 KiB
C++
120 lines
No EOL
4.5 KiB
C++
#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 converted 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* distance_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 |