| #include <sstream> /* std::ostringstream */ |
| #include "tfxparam.h" |
| #include "stdfx.h" |
| |
| #include "ino_common.h" |
| |
| class ino_hsv_adjust final : 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) override { |
| 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) override { |
| return true; |
| } |
| void doCompute(TTile &tile, double frame, |
| const TRenderSettings &rend_sets) override; |
| }; |
| FX_PLUGIN_IDENTIFIER(ino_hsv_adjust, "inohsvAdjustFx"); |
| |
| #include "igs_hsv_adjust.h" |
| namespace { |
| void fx_(TRasterP in_ras, const TRasterP refer_ras, const int refer_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) { |
| |
| |
| |
| 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_gr8->getRawData() |
| |
| , |
| in_ras->getLy(), in_ras->getLx() |
| , |
| ino::channels(), ino::bits(in_ras) |
| |
| , |
| (((refer_ras != nullptr) && (0 <= refer_mode)) ? refer_ras->getRawData() |
| : nullptr) |
| , |
| (((refer_ras != nullptr) && (0 <= refer_mode)) ? ino::bits(refer_ras) |
| : 0), |
| refer_mode |
| |
| , |
| hue_pivot, hue_scale, hue_shift, sat_pivot, sat_scale, sat_shift, |
| val_pivot, val_scale, val_shift |
| |
| |
| , |
| anti_alias_sw); |
| |
| |
| |
| 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; |
| } |
| |
| |
| 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 refer_mode = this->m_ref_mode->getValue(); |
| |
| |
| this->m_input->compute(tile, frame, rend_sets); |
| |
| |
| TTile refer_tile; |
| bool refer_sw = false; |
| if (this->m_refer.isConnected()) { |
| refer_sw = true; |
| this->m_refer->allocateAndCompute( |
| refer_tile, tile.m_pos, |
| TDimensionI( |
| tile.getRaster()->getLx(), tile.getRaster()->getLy()), |
| tile.getRaster(), frame, rend_sets); |
| } |
| |
| |
| 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 " << refer_mode << " tile w " |
| << tile.getRaster()->getLx() << " h " << tile.getRaster()->getLy() |
| << " pixbits " << ino::pixel_bits(tile.getRaster()) << " frame " |
| << frame; |
| if (refer_sw) { |
| os << " refer_tile.m_pos " << refer_tile.m_pos << " refer_tile_getLx " |
| << refer_tile.getRaster()->getLx() << " y " |
| << refer_tile.getRaster()->getLy(); |
| } |
| } |
| |
| try { |
| tile.getRaster()->lock(); |
| if (refer_tile.getRaster() != nullptr) { |
| refer_tile.getRaster()->lock(); |
| } |
| fx_(tile.getRaster(), refer_tile.getRaster(), refer_mode, hue_pivot, |
| hue_scale, hue_shift, sat_pivot, sat_scale, sat_shift, val_pivot, |
| val_scale, val_shift, |
| anti_alias_sw |
| ); |
| if (refer_tile.getRaster() != nullptr) { |
| refer_tile.getRaster()->unlock(); |
| } |
| tile.getRaster()->unlock(); |
| } |
| |
| catch (std::bad_alloc &e) { |
| if (refer_tile.getRaster() != nullptr) { |
| refer_tile.getRaster()->unlock(); |
| } |
| tile.getRaster()->unlock(); |
| if (log_sw) { |
| std::string str("std::bad_alloc <"); |
| str += e.what(); |
| str += '>'; |
| } |
| throw; |
| } catch (std::exception &e) { |
| if (refer_tile.getRaster() != nullptr) { |
| refer_tile.getRaster()->unlock(); |
| } |
| tile.getRaster()->unlock(); |
| if (log_sw) { |
| std::string str("exception <"); |
| str += e.what(); |
| str += '>'; |
| } |
| throw; |
| } catch (...) { |
| if (refer_tile.getRaster() != nullptr) { |
| refer_tile.getRaster()->unlock(); |
| } |
| tile.getRaster()->unlock(); |
| if (log_sw) { |
| std::string str("other exception"); |
| } |
| throw; |
| } |
| } |