7a395e2e7f
The last good commit wasf57e4c5b98. After this one more fix was committed, this one is preserved as well:67bd678887.
84 lines
2.9 KiB
C++
84 lines
2.9 KiB
C++
/* SPDX-FileCopyrightText: 2011 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "COM_BoxMaskOperation.h"
|
|
|
|
namespace blender::compositor {
|
|
|
|
BoxMaskOperation::BoxMaskOperation()
|
|
{
|
|
this->add_input_socket(DataType::Value);
|
|
this->add_input_socket(DataType::Value);
|
|
this->add_output_socket(DataType::Value);
|
|
cosine_ = 0.0f;
|
|
sine_ = 0.0f;
|
|
}
|
|
void BoxMaskOperation::init_execution()
|
|
{
|
|
const double rad = double(data_->rotation);
|
|
cosine_ = cos(rad);
|
|
sine_ = sin(rad);
|
|
aspect_ratio_ = float(this->get_width()) / this->get_height();
|
|
}
|
|
|
|
void BoxMaskOperation::update_memory_buffer_partial(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs)
|
|
{
|
|
MaskFunc mask_func;
|
|
switch (mask_type_) {
|
|
case CMP_NODE_MASKTYPE_ADD:
|
|
mask_func = [](const bool is_inside, const float *mask, const float *value) {
|
|
return is_inside ? std::max(mask[0], value[0]) : mask[0];
|
|
};
|
|
break;
|
|
case CMP_NODE_MASKTYPE_SUBTRACT:
|
|
mask_func = [](const bool is_inside, const float *mask, const float *value) {
|
|
return is_inside ? std::clamp(mask[0] - value[0], 0.0f, 1.0f) : mask[0];
|
|
};
|
|
break;
|
|
case CMP_NODE_MASKTYPE_MULTIPLY:
|
|
mask_func = [](const bool is_inside, const float *mask, const float *value) {
|
|
return is_inside ? mask[0] * value[0] : 0;
|
|
};
|
|
break;
|
|
case CMP_NODE_MASKTYPE_NOT:
|
|
mask_func = [](const bool is_inside, const float *mask, const float *value) {
|
|
if (is_inside) {
|
|
return mask[0] > 0.0f ? 0.0f : value[0];
|
|
}
|
|
return mask[0];
|
|
};
|
|
break;
|
|
}
|
|
apply_mask(output, area, inputs, mask_func);
|
|
}
|
|
|
|
void BoxMaskOperation::apply_mask(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs,
|
|
MaskFunc mask_func)
|
|
{
|
|
const float op_last_x = std::max(this->get_width() - 1.0f, FLT_EPSILON);
|
|
const float op_last_y = std::max(this->get_height() - 1.0f, FLT_EPSILON);
|
|
const float half_w = data_->width / 2.0f + FLT_EPSILON;
|
|
const float half_h = data_->height / 2.0f + FLT_EPSILON;
|
|
for (BuffersIterator<float> it = output->iterate_with(inputs, area); !it.is_end(); ++it) {
|
|
const float op_ry = it.y / op_last_y;
|
|
const float dy = (op_ry - data_->y) / aspect_ratio_;
|
|
const float op_rx = it.x / op_last_x;
|
|
const float dx = op_rx - data_->x;
|
|
const float rx = data_->x + (cosine_ * dx + sine_ * dy);
|
|
const float ry = data_->y + (-sine_ * dx + cosine_ * dy);
|
|
|
|
const bool inside = (rx >= data_->x - half_w && rx <= data_->x + half_w &&
|
|
ry >= data_->y - half_h && ry <= data_->y + half_h);
|
|
const float *mask = it.in(0);
|
|
const float *value = it.in(1);
|
|
*it.out = mask_func(inside, mask, value);
|
|
}
|
|
}
|
|
|
|
} // namespace blender::compositor
|