tahoma2d/toonz/sources/stdfx/iwa_spectrumfx.h
2023-02-25 20:59:02 -05:00

82 lines
2.5 KiB
C++

#pragma once
/*------------------------------------
Iwa_SpectrumFx
参照画像を位相差として、干渉色を出力する
------------------------------------*/
#ifndef IWA_SPECTRUM_H
#define IWA_SPECTRUM_H
#include "stdfx.h"
#include "tfxparam.h"
struct float3 {
float x, y, z;
float3 operator*(const float &a) { return {x * a, y * a, z * a}; }
float3 operator+(const float3 &a) { return {x + a.x, y + a.y, z + a.z}; }
};
struct float4 {
float x, y, z, w;
};
class Iwa_SpectrumFx : public TStandardRasterFx {
FX_PLUGIN_DECLARATION(Iwa_SpectrumFx)
protected:
TRasterFxPort m_input; /*- 位相差マップの入力 -*/
TRasterFxPort m_light; /*- 光源用マップの入力 -*/
TDoubleParamP m_intensity;
TDoubleParamP m_refractiveIndex;
TDoubleParamP m_thickMax;
TDoubleParamP m_thickMin;
TDoubleParamP m_RGamma;
TDoubleParamP m_GGamma;
TDoubleParamP m_BGamma;
TDoubleParamP m_loopSpectrumFadeWidth;
TDoubleParamP m_spectrumShift;
TDoubleParamP m_lensFactor;
TDoubleParamP m_lightThres;
TDoubleParamP m_lightIntensity;
/*- シャボン色マップの生成 -*/
void calcBubbleMap(float3 *bubbleColor, double frame, bool isLinear,
double colorSpaceGamma, bool computeAngularAxis = false);
template <typename RASTER, typename PIXEL>
void convertRaster(const RASTER ras, TDimensionI dim, float3 *bubbleColor);
template <typename RASTER, typename PIXEL>
void convertRasterWithLight(const RASTER ras, TDimensionI dim,
float3 *bubbleColor, const RASTER lightRas,
float lightThres, float lightIntensity);
/*- 素材タイルを0〜1に正規化して格納 -*/
template <typename RASTER, typename PIXEL>
void setSourceRasters(const RASTER ras, float4 *in_out_tile_host,
const RASTER light_ras, float4 *light_host,
TDimensionI dim, bool useLight);
/*- 出力結果をChannel値に変換してタイルに格納 -*/
template <typename RASTER, typename PIXEL>
void outputRasters(const RASTER outRas, float4 *in_out_tile_host,
TDimensionI dim);
public:
Iwa_SpectrumFx();
void doCompute(TTile &tile, double frame,
const TRenderSettings &settings) override;
void doCompute_CUDA(TTile &tile, double frame,
const TRenderSettings &settings);
bool doGetBBox(double frame, TRectD &bBox,
const TRenderSettings &info) override;
bool canHandle(const TRenderSettings &info, double frame) override;
};
#endif