244 lines
8.1 KiB
C++
244 lines
8.1 KiB
C++
|
#if defined _WIN32 && !defined NOMINMAX
|
||
|
#define NOMINMAX
|
||
|
#endif
|
||
|
|
||
|
#include <sstream> /* std::ostringstream */
|
||
|
#include "tfxparam.h"
|
||
|
#include "stdfx.h"
|
||
|
|
||
|
#include "ino_common.h"
|
||
|
//------------------------------------------------------------
|
||
|
class ino_hsv_adjust : public TStandardRasterFx
|
||
|
{
|
||
|
FX_PLUGIN_DECLARATION(ino_hsv_adjust)
|
||
|
TRasterFxPort m_input;
|
||
|
TRasterFxPort m_refer;
|
||
|
|
||
|
TDoubleParamP m_hue_pivot;
|
||
|
TDoubleParamP m_hue_scale;
|
||
|
TDoubleParamP m_hue_shift;
|
||
|
TDoubleParamP m_sat_pivot;
|
||
|
TDoubleParamP m_sat_scale;
|
||
|
TDoubleParamP m_sat_shift;
|
||
|
TDoubleParamP m_val_pivot;
|
||
|
TDoubleParamP m_val_scale;
|
||
|
TDoubleParamP m_val_shift;
|
||
|
|
||
|
TBoolParamP m_anti_alias;
|
||
|
TIntEnumParamP m_ref_mode;
|
||
|
|
||
|
public:
|
||
|
ino_hsv_adjust()
|
||
|
: m_hue_pivot(0.0), m_hue_scale(1.0 * ino::param_range()), m_hue_shift(0.0), m_sat_pivot(0.0 * ino::param_range()), m_sat_scale(1.0 * ino::param_range()), m_sat_shift(0.0 * ino::param_range()), m_val_pivot(0.0 * ino::param_range()), m_val_scale(1.0 * ino::param_range()), m_val_shift(0.0 * ino::param_range())
|
||
|
|
||
|
,
|
||
|
m_anti_alias(true), m_ref_mode(new TIntEnumParam(0, "Red"))
|
||
|
{
|
||
|
addInputPort("Source", this->m_input);
|
||
|
addInputPort("Reference", this->m_refer);
|
||
|
|
||
|
bindParam(this, "pivot_hue", this->m_hue_pivot);
|
||
|
bindParam(this, "pivot_saturation", this->m_sat_pivot);
|
||
|
bindParam(this, "pivot_value", this->m_val_pivot);
|
||
|
bindParam(this, "scale_hue", this->m_hue_scale);
|
||
|
bindParam(this, "scale_saturation", this->m_sat_scale);
|
||
|
bindParam(this, "scale_value", this->m_val_scale);
|
||
|
bindParam(this, "shift_hue", this->m_hue_shift);
|
||
|
bindParam(this, "shift_saturation", this->m_sat_shift);
|
||
|
bindParam(this, "shift_value", this->m_val_shift);
|
||
|
|
||
|
bindParam(this, "anti_alias", this->m_anti_alias);
|
||
|
bindParam(this, "reference", this->m_ref_mode);
|
||
|
|
||
|
this->m_hue_pivot->setValueRange(0.0, 360.0);
|
||
|
this->m_hue_scale->setValueRange(
|
||
|
0.0 * ino::param_range(), std::numeric_limits<double>::max());
|
||
|
this->m_sat_pivot->setValueRange(
|
||
|
0.0 * ino::param_range(), 1.0 * ino::param_range());
|
||
|
this->m_sat_scale->setValueRange(
|
||
|
0.0 * ino::param_range(), std::numeric_limits<double>::max());
|
||
|
this->m_val_pivot->setValueRange(
|
||
|
0.0 * ino::param_range(), 1.0 * ino::param_range());
|
||
|
this->m_val_scale->setValueRange(
|
||
|
0.0 * ino::param_range(), std::numeric_limits<double>::max());
|
||
|
|
||
|
this->m_ref_mode->addItem(1, "Green");
|
||
|
this->m_ref_mode->addItem(2, "Blue");
|
||
|
this->m_ref_mode->addItem(3, "Alpha");
|
||
|
this->m_ref_mode->addItem(4, "Luminance");
|
||
|
this->m_ref_mode->addItem(-1, "Nothing");
|
||
|
}
|
||
|
bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info)
|
||
|
{
|
||
|
if (this->m_input.isConnected()) {
|
||
|
return this->m_input->doGetBBox(frame, bBox, info);
|
||
|
} else {
|
||
|
bBox = TRectD();
|
||
|
return false;
|
||
|
}
|
||
|
}
|
||
|
bool canHandle(const TRenderSettings &rend_sets, double frame)
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
void doCompute(
|
||
|
TTile &tile, double frame, const TRenderSettings &rend_sets);
|
||
|
};
|
||
|
FX_PLUGIN_IDENTIFIER(ino_hsv_adjust, "inohsvAdjustFx");
|
||
|
//------------------------------------------------------------
|
||
|
#include "igs_hsv_adjust.h"
|
||
|
namespace
|
||
|
{
|
||
|
void fx_(
|
||
|
TRasterP in_ras, const TRasterP refer_ras, const int ref_mode, const double hue_pivot, const double hue_scale, const double hue_shift, const double sat_pivot, const double sat_scale, const double sat_shift, const double val_pivot, const double val_scale, const double val_shift, const bool anti_alias_sw)
|
||
|
{
|
||
|
/***std::vector<unsigned char> in_vec;
|
||
|
ino::ras_to_vec( in_ras, ino::channels(), in_vec );***/
|
||
|
|
||
|
TRasterGR8P in_gr8(
|
||
|
in_ras->getLy(), in_ras->getLx() * ino::channels() *
|
||
|
((TRaster64P)in_ras ? sizeof(unsigned short) : sizeof(unsigned char)));
|
||
|
in_gr8->lock();
|
||
|
ino::ras_to_arr(in_ras, ino::channels(), in_gr8->getRawData());
|
||
|
|
||
|
igs::hsv_adjust::change(
|
||
|
//in_ras->getRawData() // BGRA
|
||
|
//&in_vec.at(0) // RGBA
|
||
|
in_gr8->getRawData()
|
||
|
|
||
|
,
|
||
|
in_ras->getLy(), in_ras->getLx() // Not use in_ras->getWrap()
|
||
|
,
|
||
|
ino::channels(), ino::bits(in_ras)
|
||
|
|
||
|
,
|
||
|
(((0 <= ref_mode) && (0 != refer_ras)) ? refer_ras->getRawData() : 0) //BGRA
|
||
|
,
|
||
|
(((0 <= ref_mode) && (0 != refer_ras)) ? ino::bits(refer_ras) : 0), ref_mode
|
||
|
|
||
|
,
|
||
|
hue_pivot, hue_scale, hue_shift, sat_pivot, sat_scale, sat_shift, val_pivot, val_scale, val_shift
|
||
|
|
||
|
//,true /* add_blend_sw */
|
||
|
,
|
||
|
anti_alias_sw);
|
||
|
|
||
|
/***ino::vec_to_ras( in_vec, ino::channels(), in_ras, 0 );***/
|
||
|
|
||
|
ino::arr_to_ras(in_gr8->getRawData(), ino::channels(), in_ras, 0);
|
||
|
in_gr8->unlock();
|
||
|
}
|
||
|
}
|
||
|
//------------------------------------------------------------
|
||
|
void ino_hsv_adjust::doCompute(
|
||
|
TTile &tile, double frame, const TRenderSettings &rend_sets)
|
||
|
{
|
||
|
/* ------ 接続していなければ処理しない -------------------- */
|
||
|
if (!this->m_input.isConnected()) {
|
||
|
tile.getRaster()->clear(); /* 塗りつぶしクリア */
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
/* ------ サポートしていないPixelタイプはエラーを投げる --- */
|
||
|
if (!((TRaster32P)tile.getRaster()) &&
|
||
|
!((TRaster64P)tile.getRaster())) {
|
||
|
throw TRopException("unsupported input pixel type");
|
||
|
}
|
||
|
|
||
|
/* ------ 動作パラメータを得る ---------------------------- */
|
||
|
const double hue_pivot = this->m_hue_pivot->getValue(frame);
|
||
|
const double hue_scale = this->m_hue_scale->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
const double hue_shift = this->m_hue_shift->getValue(frame);
|
||
|
const double sat_pivot = this->m_sat_pivot->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
const double sat_scale = this->m_sat_scale->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
const double sat_shift = this->m_sat_shift->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
const double val_pivot = this->m_val_pivot->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
const double val_scale = this->m_val_scale->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
const double val_shift = this->m_val_shift->getValue(frame) /
|
||
|
ino::param_range();
|
||
|
|
||
|
const bool anti_alias_sw = this->m_anti_alias->getValue();
|
||
|
const int ref_mode = this->m_ref_mode->getValue();
|
||
|
|
||
|
/* ------ 画像生成 ---------------------------------------- */
|
||
|
this->m_input->compute(tile, frame, rend_sets);
|
||
|
|
||
|
/*------ 参照画像生成 --------------------------------------*/
|
||
|
TTile reference_tile;
|
||
|
bool reference_sw = false;
|
||
|
if (this->m_refer.isConnected()) {
|
||
|
reference_sw = true;
|
||
|
this->m_refer->allocateAndCompute(
|
||
|
reference_tile, tile.m_pos, TDimensionI(/* Pixel単位 */
|
||
|
tile.getRaster()->getLx(), tile.getRaster()->getLy()),
|
||
|
tile.getRaster(), frame, rend_sets);
|
||
|
}
|
||
|
|
||
|
/* ------ (app_begin)log記憶 ------------------------------ */
|
||
|
const bool log_sw = ino::log_enable_sw();
|
||
|
|
||
|
if (log_sw) {
|
||
|
std::ostringstream os;
|
||
|
os << "params"
|
||
|
<< " h_pvt " << hue_pivot
|
||
|
<< " h_scl " << hue_scale
|
||
|
<< " h_sft " << hue_shift
|
||
|
<< " s_pvt " << sat_pivot
|
||
|
<< " s_scl " << sat_scale
|
||
|
<< " s_sft " << sat_shift
|
||
|
<< " v_pvt " << val_pivot
|
||
|
<< " v_scl " << val_scale
|
||
|
<< " v_sft " << val_shift
|
||
|
<< " anti_alias " << anti_alias_sw
|
||
|
<< " reference " << ref_mode
|
||
|
<< " tile w " << tile.getRaster()->getLx()
|
||
|
<< " h " << tile.getRaster()->getLy()
|
||
|
<< " pixbits " << ino::pixel_bits(tile.getRaster())
|
||
|
<< " frame " << frame;
|
||
|
if (reference_sw) {
|
||
|
os
|
||
|
<< " reference_tile.m_pos " << reference_tile.m_pos
|
||
|
<< " reference_tile_getLx " << reference_tile.getRaster()->getLx()
|
||
|
<< " y " << reference_tile.getRaster()->getLy();
|
||
|
}
|
||
|
}
|
||
|
/* ------ fx処理 ------------------------------------------ */
|
||
|
try {
|
||
|
tile.getRaster()->lock();
|
||
|
fx_(
|
||
|
tile.getRaster(), (reference_sw ? reference_tile.getRaster() : nullptr), ref_mode, hue_pivot, hue_scale, hue_shift, sat_pivot, sat_scale, sat_shift, val_pivot, val_scale, val_shift, anti_alias_sw // --> add_blend_sw, default is true
|
||
|
);
|
||
|
tile.getRaster()->unlock();
|
||
|
}
|
||
|
/* ------ error処理 --------------------------------------- */
|
||
|
catch (std::bad_alloc &e) {
|
||
|
tile.getRaster()->unlock();
|
||
|
if (log_sw) {
|
||
|
std::string str("std::bad_alloc <");
|
||
|
str += e.what();
|
||
|
str += '>';
|
||
|
}
|
||
|
throw;
|
||
|
} catch (std::exception &e) {
|
||
|
tile.getRaster()->unlock();
|
||
|
if (log_sw) {
|
||
|
std::string str("exception <");
|
||
|
str += e.what();
|
||
|
str += '>';
|
||
|
}
|
||
|
throw;
|
||
|
} catch (...) {
|
||
|
tile.getRaster()->unlock();
|
||
|
if (log_sw) {
|
||
|
std::string str("other exception");
|
||
|
}
|
||
|
throw;
|
||
|
}
|
||
|
}
|