049b0e6539
This patch rewrites and optimizes the Double Edge Mask node to be orders of magnitude faster. For a 1k complex mask, it is 650x faster, while for a 1k simple mask, it is only 50x faster. This improvement is attributed to the use of a new Jump Flooding algorithm as well as multi-threading, matching the GPU implementation. The result of the new implementation differs in that the boundaries of the masks are now identified as the last pixels inside the mask. Pull Request: https://projects.blender.org/blender/blender/pulls/117545
84 lines
2.3 KiB
C++
84 lines
2.3 KiB
C++
/* SPDX-FileCopyrightText: 2011 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
#include "BLI_span.hh"
|
|
|
|
#include "COM_MultiThreadedOperation.h"
|
|
|
|
namespace blender::compositor {
|
|
|
|
class DoubleEdgeMaskOperation : public NodeOperation {
|
|
private:
|
|
/**
|
|
* Cached reference to the input_program
|
|
*/
|
|
SocketReader *input_outer_mask_;
|
|
SocketReader *input_inner_mask_;
|
|
bool include_all_inner_edges_;
|
|
bool include_edges_of_image_;
|
|
|
|
/* TODO(manzanilla): To be removed with tiled implementation. */
|
|
float *cached_instance_;
|
|
|
|
bool is_output_rendered_;
|
|
|
|
public:
|
|
DoubleEdgeMaskOperation();
|
|
|
|
void compute_boundary(const float *inner_mask,
|
|
const float *outer_mask,
|
|
MutableSpan<int2> inner_boundary,
|
|
MutableSpan<int2> outer_boundary);
|
|
|
|
void compute_gradient(const float *inner_mask_buffer,
|
|
const float *outer_mask_buffer,
|
|
MutableSpan<int2> flooded_inner_boundary,
|
|
MutableSpan<int2> flooded_outer_boundary,
|
|
float *output_mask);
|
|
|
|
void compute_double_edge_mask(const float *inner_mask,
|
|
const float *outer_mask,
|
|
float *output_mask);
|
|
|
|
/**
|
|
* The inner loop of this operation.
|
|
*/
|
|
void execute_pixel(float output[4], int x, int y, void *data) override;
|
|
|
|
/**
|
|
* Initialize the execution
|
|
*/
|
|
void init_execution() override;
|
|
|
|
/**
|
|
* Deinitialize the execution
|
|
*/
|
|
void deinit_execution() override;
|
|
|
|
void *initialize_tile_data(rcti *rect) override;
|
|
|
|
bool determine_depending_area_of_interest(rcti *input,
|
|
ReadBufferOperation *read_operation,
|
|
rcti *output) override;
|
|
|
|
void set_include_all_inner_edges(bool include_all_inner_edges)
|
|
{
|
|
include_all_inner_edges_ = include_all_inner_edges;
|
|
}
|
|
void set_include_edges_of_image(bool include_edges_of_image)
|
|
{
|
|
include_edges_of_image_ = include_edges_of_image;
|
|
}
|
|
|
|
void get_area_of_interest(int input_idx, const rcti &output_area, rcti &r_input_area) override;
|
|
|
|
void update_memory_buffer(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs) override;
|
|
};
|
|
|
|
} // namespace blender::compositor
|