| #include <sstream> /* std::ostringstream */ |
| #include "tfxparam.h" |
| #include "stdfx.h" |
| |
| #include "ino_common.h" |
| |
| class ino_hls_adjust : public TStandardRasterFx |
| { |
| FX_PLUGIN_DECLARATION(ino_hls_adjust) |
| TRasterFxPort m_input; |
| TRasterFxPort m_refer; |
| |
| TDoubleParamP m_hue_pivot; |
| TDoubleParamP m_hue_scale; |
| TDoubleParamP m_hue_shift; |
| TDoubleParamP m_lig_pivot; |
| TDoubleParamP m_lig_scale; |
| TDoubleParamP m_lig_shift; |
| TDoubleParamP m_sat_pivot; |
| TDoubleParamP m_sat_scale; |
| TDoubleParamP m_sat_shift; |
| |
| TBoolParamP m_anti_alias; |
| TIntEnumParamP m_ref_mode; |
| |
| public: |
| ino_hls_adjust() |
| : m_hue_pivot(0.0), m_hue_scale(1.0 * ino::param_range()), m_hue_shift(0.0), m_lig_pivot(0.0 * ino::param_range()), m_lig_scale(1.0 * ino::param_range()), m_lig_shift(0.0 * ino::param_range()), 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_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_lightness", this->m_lig_pivot); |
| bindParam(this, "pivot_saturation", this->m_sat_pivot); |
| bindParam(this, "scale_hue", this->m_hue_scale); |
| bindParam(this, "scale_lightness", this->m_lig_scale); |
| bindParam(this, "scale_saturation", this->m_sat_scale); |
| bindParam(this, "shift_hue", this->m_hue_shift); |
| bindParam(this, "shift_lightness", this->m_lig_shift); |
| bindParam(this, "shift_saturation", this->m_sat_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_lig_pivot->setValueRange( |
| 0.0 * ino::param_range(), 1.0 * ino::param_range()); |
| this->m_lig_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_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_hls_adjust, "inohlsAdjustFx"); |
| |
| #include "igs_hls_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 lig_pivot, const double lig_scale, const double lig_shift, const double sat_pivot, const double sat_scale, const double sat_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::hls_adjust::change( |
| |
| |
| in_gr8->getRawData() |
| |
| , |
| in_ras->getLy(), in_ras->getLx() |
| , |
| ino::channels(), ino::bits(in_ras) |
| |
| , |
| (((0 <= ref_mode) && (0 != refer_ras)) ? refer_ras->getRawData() : 0) |
| , |
| (((0 <= ref_mode) && (0 != refer_ras)) ? ino::bits(refer_ras) : 0), ref_mode |
| |
| , |
| hue_pivot, hue_scale, hue_shift, lig_pivot, lig_scale, lig_shift, sat_pivot, sat_scale, sat_shift |
| |
| |
| , |
| anti_alias_sw); |
| |
| |
| |
| ino::arr_to_ras(in_gr8->getRawData(), ino::channels(), in_ras, 0); |
| in_gr8->unlock(); |
| } |
| } |
| |
| void ino_hls_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 lig_pivot = this->m_lig_pivot->getValue(frame) / |
| ino::param_range(); |
| const double lig_scale = this->m_lig_scale->getValue(frame) / |
| ino::param_range(); |
| const double lig_shift = this->m_lig_shift->getValue(frame) / |
| ino::param_range(); |
| 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 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( |
| 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 |
| << " l_pvt " << lig_pivot |
| << " l_scl " << lig_scale |
| << " l_sft " << lig_shift |
| << " s_pvt " << sat_pivot |
| << " s_scl " << sat_scale |
| << " s_sft " << sat_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(); |
| } |
| } |
| |
| try { |
| tile.getRaster()->lock(); |
| fx_( |
| tile.getRaster(), (reference_sw ? reference_tile.getRaster() : nullptr), ref_mode, hue_pivot, hue_scale, hue_shift, lig_pivot, lig_scale, lig_shift, sat_pivot, sat_scale, sat_shift, anti_alias_sw |
| ); |
| tile.getRaster()->unlock(); |
| } |
| |
| 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; |
| } |
| } |