tornavis/source/blender/compositor/nodes/COM_HueSaturationValueCorre...

52 lines
1.9 KiB
C++

/* SPDX-FileCopyrightText: 2011 Blender Authors
*
* SPDX-License-Identifier: GPL-2.0-or-later */
#include "COM_HueSaturationValueCorrectNode.h"
#include "COM_ConvertOperation.h"
#include "COM_HueSaturationValueCorrectOperation.h"
#include "COM_MixOperation.h"
namespace blender::compositor {
HueSaturationValueCorrectNode::HueSaturationValueCorrectNode(bNode *editor_node)
: Node(editor_node)
{
/* pass */
}
void HueSaturationValueCorrectNode::convert_to_operations(
NodeConverter &converter, const CompositorContext & /*context*/) const
{
NodeInput *value_socket = this->get_input_socket(0);
NodeInput *color_socket = this->get_input_socket(1);
NodeOutput *output_socket = this->get_output_socket(0);
const bNode *editorsnode = get_bnode();
CurveMapping *storage = (CurveMapping *)editorsnode->storage;
ConvertRGBToHSVOperation *rgbToHSV = new ConvertRGBToHSVOperation();
converter.add_operation(rgbToHSV);
ConvertHSVToRGBOperation *hsvToRGB = new ConvertHSVToRGBOperation();
converter.add_operation(hsvToRGB);
HueSaturationValueCorrectOperation *changeHSV = new HueSaturationValueCorrectOperation();
changeHSV->set_curve_mapping(storage);
converter.add_operation(changeHSV);
MixBlendOperation *blend = new MixBlendOperation();
blend->set_canvas_input_index(1);
converter.add_operation(blend);
converter.map_input_socket(color_socket, rgbToHSV->get_input_socket(0));
converter.add_link(rgbToHSV->get_output_socket(), changeHSV->get_input_socket(0));
converter.add_link(changeHSV->get_output_socket(), hsvToRGB->get_input_socket(0));
converter.add_link(hsvToRGB->get_output_socket(), blend->get_input_socket(2));
converter.map_input_socket(color_socket, blend->get_input_socket(1));
converter.map_input_socket(value_socket, blend->get_input_socket(0));
converter.map_output_socket(output_socket, blend->get_output_socket());
}
} // namespace blender::compositor