57 lines
2.4 KiB
C++
57 lines
2.4 KiB
C++
/* SPDX-FileCopyrightText: 2014 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "COM_CornerPinNode.h"
|
|
|
|
#include "COM_PlaneCornerPinOperation.h"
|
|
#include "COM_SMAAOperation.h"
|
|
#include "COM_SetAlphaMultiplyOperation.h"
|
|
|
|
namespace blender::compositor {
|
|
|
|
CornerPinNode::CornerPinNode(bNode *editor_node) : Node(editor_node) {}
|
|
|
|
void CornerPinNode::convert_to_operations(NodeConverter &converter,
|
|
const CompositorContext & /*context*/) const
|
|
{
|
|
PlaneCornerPinMaskOperation *plane_mask_operation = new PlaneCornerPinMaskOperation();
|
|
converter.add_operation(plane_mask_operation);
|
|
|
|
SMAAOperation *smaa_operation = new SMAAOperation();
|
|
converter.add_operation(smaa_operation);
|
|
|
|
converter.add_link(plane_mask_operation->get_output_socket(),
|
|
smaa_operation->get_input_socket(0));
|
|
|
|
converter.map_output_socket(this->get_output_socket(1), smaa_operation->get_output_socket());
|
|
|
|
PlaneCornerPinWarpImageOperation *warp_image_operation = new PlaneCornerPinWarpImageOperation();
|
|
converter.add_operation(warp_image_operation);
|
|
converter.map_input_socket(this->get_input_socket(0), warp_image_operation->get_input_socket(0));
|
|
|
|
/* NOTE: socket order differs between UI node and operations:
|
|
* bNode uses intuitive order following top-down layout:
|
|
* upper-left, upper-right, lower-left, lower-right
|
|
* Operations use same order as the tracking blenkernel functions expect:
|
|
* lower-left, lower-right, upper-right, upper-left
|
|
*/
|
|
const int node_corner_index[4] = {3, 4, 2, 1};
|
|
for (int i = 0; i < 4; i++) {
|
|
NodeInput *corner_input = get_input_socket(node_corner_index[i]);
|
|
converter.map_input_socket(corner_input, warp_image_operation->get_input_socket(i + 1));
|
|
converter.map_input_socket(corner_input, plane_mask_operation->get_input_socket(i));
|
|
}
|
|
|
|
SetAlphaMultiplyOperation *set_alpha_operation = new SetAlphaMultiplyOperation();
|
|
converter.add_operation(set_alpha_operation);
|
|
converter.add_link(warp_image_operation->get_output_socket(),
|
|
set_alpha_operation->get_input_socket(0));
|
|
converter.add_link(smaa_operation->get_output_socket(),
|
|
set_alpha_operation->get_input_socket(1));
|
|
converter.map_output_socket(this->get_output_socket(0),
|
|
set_alpha_operation->get_output_socket());
|
|
}
|
|
|
|
} // namespace blender::compositor
|