e955c94ed3
Listing the "Blender Foundation" as copyright holder implied the Blender Foundation holds copyright to files which may include work from many developers. While keeping copyright on headers makes sense for isolated libraries, Blender's own code may be refactored or moved between files in a way that makes the per file copyright holders less meaningful. Copyright references to the "Blender Foundation" have been replaced with "Blender Authors", with the exception of `./extern/` since these this contains libraries which are more isolated, any changed to license headers there can be handled on a case-by-case basis. Some directories in `./intern/` have also been excluded: - `./intern/cycles/` it's own `AUTHORS` file is planned. - `./intern/opensubdiv/`. An "AUTHORS" file has been added, using the chromium projects authors file as a template. Design task: #110784 Ref !110783.
122 lines
3.4 KiB
C++
122 lines
3.4 KiB
C++
/* SPDX-FileCopyrightText: 2011 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "COM_DistanceRGBMatteOperation.h"
|
|
|
|
namespace blender::compositor {
|
|
|
|
DistanceRGBMatteOperation::DistanceRGBMatteOperation()
|
|
{
|
|
this->add_input_socket(DataType::Color);
|
|
this->add_input_socket(DataType::Color);
|
|
this->add_output_socket(DataType::Value);
|
|
|
|
input_image_program_ = nullptr;
|
|
input_key_program_ = nullptr;
|
|
flags_.can_be_constant = true;
|
|
}
|
|
|
|
void DistanceRGBMatteOperation::init_execution()
|
|
{
|
|
input_image_program_ = this->get_input_socket_reader(0);
|
|
input_key_program_ = this->get_input_socket_reader(1);
|
|
}
|
|
|
|
void DistanceRGBMatteOperation::deinit_execution()
|
|
{
|
|
input_image_program_ = nullptr;
|
|
input_key_program_ = nullptr;
|
|
}
|
|
|
|
float DistanceRGBMatteOperation::calculate_distance(const float key[4], const float image[4])
|
|
{
|
|
return len_v3v3(key, image);
|
|
}
|
|
|
|
void DistanceRGBMatteOperation::execute_pixel_sampled(float output[4],
|
|
float x,
|
|
float y,
|
|
PixelSampler sampler)
|
|
{
|
|
float in_key[4];
|
|
float in_image[4];
|
|
|
|
const float tolerance = settings_->t1;
|
|
const float falloff = settings_->t2;
|
|
|
|
float distance;
|
|
float alpha;
|
|
|
|
input_key_program_->read_sampled(in_key, x, y, sampler);
|
|
input_image_program_->read_sampled(in_image, x, y, sampler);
|
|
|
|
distance = this->calculate_distance(in_key, in_image);
|
|
|
|
/* Store matte(alpha) value in [0] to go with
|
|
* COM_SetAlphaMultiplyOperation and the Value output.
|
|
*/
|
|
|
|
/* Make 100% transparent. */
|
|
if (distance < tolerance) {
|
|
output[0] = 0.0f;
|
|
}
|
|
/* In the falloff region, make partially transparent. */
|
|
else if (distance < falloff + tolerance) {
|
|
distance = distance - tolerance;
|
|
alpha = distance / falloff;
|
|
/* Only change if more transparent than before. */
|
|
if (alpha < in_image[3]) {
|
|
output[0] = alpha;
|
|
}
|
|
else { /* leave as before */
|
|
output[0] = in_image[3];
|
|
}
|
|
}
|
|
else {
|
|
/* leave as before */
|
|
output[0] = in_image[3];
|
|
}
|
|
}
|
|
|
|
void DistanceRGBMatteOperation::update_memory_buffer_partial(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs)
|
|
{
|
|
for (BuffersIterator<float> it = output->iterate_with(inputs, area); !it.is_end(); ++it) {
|
|
const float *in_image = it.in(0);
|
|
const float *in_key = it.in(1);
|
|
|
|
float distance = this->calculate_distance(in_key, in_image);
|
|
const float tolerance = settings_->t1;
|
|
const float falloff = settings_->t2;
|
|
|
|
/* Store matte(alpha) value in [0] to go with
|
|
* COM_SetAlphaMultiplyOperation and the Value output.
|
|
*/
|
|
|
|
/* Make 100% transparent. */
|
|
if (distance < tolerance) {
|
|
it.out[0] = 0.0f;
|
|
}
|
|
/* In the falloff region, make partially transparent. */
|
|
else if (distance < falloff + tolerance) {
|
|
distance = distance - tolerance;
|
|
const float alpha = distance / falloff;
|
|
/* Only change if more transparent than before. */
|
|
if (alpha < in_image[3]) {
|
|
it.out[0] = alpha;
|
|
}
|
|
else { /* Leave as before. */
|
|
it.out[0] = in_image[3];
|
|
}
|
|
}
|
|
else {
|
|
/* Leave as before. */
|
|
it.out[0] = in_image[3];
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace blender::compositor
|