griefith/test2
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
286cbc8317bdbe9f2eaebb5df8ecccb53bc830f4
test2/source/blender/compositor/operations/COM_MixOperation.cc
Brecht Van Lommel 7a395e2e7f Revert changes from main commits that were merged into blender-v4.1-release 
The last good commit was f57e4c5b98.

After this one more fix was committed, this one is preserved as well:
67bd678887.
2024-03-18 15:04:12 +01:00

660 lines
18 KiB
C++
Raw Blame History

/* SPDX-FileCopyrightText: 2011 Blender Authors
*
* SPDX-License-Identifier: GPL-2.0-or-later */
#include "COM_MixOperation.h"
#include "BLI_math_color.h"
namespace blender::compositor {
/* ******** Mix Base Operation ******** */
MixBaseOperation::MixBaseOperation()
{
this->add_input_socket(DataType::Value);
this->add_input_socket(DataType::Color);
this->add_input_socket(DataType::Color);
this->add_output_socket(DataType::Color);
this->set_use_value_alpha_multiply(false);
this->set_use_clamp(false);
flags_.can_be_constant = true;
}
void MixBaseOperation::determine_canvas(const rcti &preferred_area, rcti &r_area)
{
NodeOperationInput *socket;
rcti temp_area = COM_AREA_NONE;
socket = this->get_input_socket(1);
bool determined = socket->determine_canvas(COM_AREA_NONE, temp_area);
if (determined) {
this->set_canvas_input_index(1);
}
else {
socket = this->get_input_socket(2);
determined = socket->determine_canvas(COM_AREA_NONE, temp_area);
if (determined) {
this->set_canvas_input_index(2);
}
else {
this->set_canvas_input_index(0);
}
}
NodeOperation::determine_canvas(preferred_area, r_area);
}
void MixBaseOperation::update_memory_buffer_partial(MemoryBuffer *output,
const rcti &area,
Span<MemoryBuffer *> inputs)
{
const MemoryBuffer *input_value = inputs[0];
const MemoryBuffer *input_color1 = inputs[1];
const MemoryBuffer *input_color2 = inputs[2];
const int width = BLI_rcti_size_x(&area);
PixelCursor p;
p.out_stride = output->elem_stride;
p.value_stride = input_value->elem_stride;
p.color1_stride = input_color1->elem_stride;
p.color2_stride = input_color2->elem_stride;
for (int y = area.ymin; y < area.ymax; y++) {
p.out = output->get_elem(area.xmin, y);
p.row_end = p.out + width * output->elem_stride;
p.value = input_value->get_elem(area.xmin, y);
p.color1 = input_color1->get_elem(area.xmin, y);
p.color2 = input_color2->get_elem(area.xmin, y);
update_memory_buffer_row(p);
}
}
void MixBaseOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
p.out[0] = value_m * p.color1[0] + value * p.color2[0];
p.out[1] = value_m * p.color1[1] + value * p.color2[1];
p.out[2] = value_m * p.color1[2] + value * p.color2[2];
p.out[3] = p.color1[3];
p.next();
}
}
/* ******** Mix Add Operation ******** */
void MixAddOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
p.out[0] = p.color1[0] + value * p.color2[0];
p.out[1] = p.color1[1] + value * p.color2[1];
p.out[2] = p.color1[2] + value * p.color2[2];
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Blend Operation ******** */
void MixBlendOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
float value_m = 1.0f - value;
p.out[0] = value_m * p.color1[0] + value * p.color2[0];
p.out[1] = value_m * p.color1[1] + value * p.color2[1];
p.out[2] = value_m * p.color1[2] + value * p.color2[2];
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Burn Operation ******** */
void MixColorBurnOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
float tmp = value_m + value * p.color2[0];
if (tmp <= 0.0f) {
p.out[0] = 0.0f;
}
else {
tmp = 1.0f - (1.0f - p.color1[0]) / tmp;
p.out[0] = std::clamp(tmp, 0.0f, 1.0f);
}
tmp = value_m + value * p.color2[1];
if (tmp <= 0.0f) {
p.out[1] = 0.0f;
}
else {
tmp = 1.0f - (1.0f - p.color1[1]) / tmp;
p.out[1] = std::clamp(tmp, 0.0f, 1.0f);
}
tmp = value_m + value * p.color2[2];
if (tmp <= 0.0f) {
p.out[2] = 0.0f;
}
else {
tmp = 1.0f - (1.0f - p.color1[2]) / tmp;
p.out[2] = std::clamp(tmp, 0.0f, 1.0f);
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Color Operation ******** */
void MixColorOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
float colH, colS, colV;
rgb_to_hsv(p.color2[0], p.color2[1], p.color2[2], &colH, &colS, &colV);
if (colS != 0.0f) {
float rH, rS, rV;
float tmpr, tmpg, tmpb;
rgb_to_hsv(p.color1[0], p.color1[1], p.color1[2], &rH, &rS, &rV);
hsv_to_rgb(colH, colS, rV, &tmpr, &tmpg, &tmpb);
p.out[0] = (value_m * p.color1[0]) + (value * tmpr);
p.out[1] = (value_m * p.color1[1]) + (value * tmpg);
p.out[2] = (value_m * p.color1[2]) + (value * tmpb);
}
else {
copy_v3_v3(p.out, p.color1);
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Darken Operation ******** */
void MixDarkenOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
float value_m = 1.0f - value;
p.out[0] = min_ff(p.color1[0], p.color2[0]) * value + p.color1[0] * value_m;
p.out[1] = min_ff(p.color1[1], p.color2[1]) * value + p.color1[1] * value_m;
p.out[2] = min_ff(p.color1[2], p.color2[2]) * value + p.color1[2] * value_m;
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Difference Operation ******** */
void MixDifferenceOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
p.out[0] = value_m * p.color1[0] + value * fabsf(p.color1[0] - p.color2[0]);
p.out[1] = value_m * p.color1[1] + value * fabsf(p.color1[1] - p.color2[1]);
p.out[2] = value_m * p.color1[2] + value * fabsf(p.color1[2] - p.color2[2]);
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Exclusion Operation ******** */
void MixExclusionOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
p.out[0] = max_ff(value_m * p.color1[0] +
value * (p.color1[0] + p.color2[0] - 2.0f * p.color1[0] * p.color2[0]),
0.0f);
p.out[1] = max_ff(value_m * p.color1[1] +
value * (p.color1[1] + p.color2[1] - 2.0f * p.color1[1] * p.color2[1]),
0.0f);
p.out[2] = max_ff(value_m * p.color1[2] +
value * (p.color1[2] + p.color2[2] - 2.0f * p.color1[2] * p.color2[2]),
0.0f);
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Divide Operation ******** */
void MixDivideOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
if (p.color2[0] != 0.0f) {
p.out[0] = value_m * (p.color1[0]) + value * (p.color1[0]) / p.color2[0];
}
else {
p.out[0] = 0.0f;
}
if (p.color2[1] != 0.0f) {
p.out[1] = value_m * (p.color1[1]) + value * (p.color1[1]) / p.color2[1];
}
else {
p.out[1] = 0.0f;
}
if (p.color2[2] != 0.0f) {
p.out[2] = value_m * (p.color1[2]) + value * (p.color1[2]) / p.color2[2];
}
else {
p.out[2] = 0.0f;
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Dodge Operation ******** */
void MixDodgeOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
float tmp;
if (p.color1[0] != 0.0f) {
tmp = 1.0f - value * p.color2[0];
if (tmp <= 0.0f) {
p.out[0] = 1.0f;
}
else {
p.out[0] = p.color1[0] / tmp;
CLAMP_MAX(p.out[0], 1.0f);
}
}
else {
p.out[0] = 0.0f;
}
if (p.color1[1] != 0.0f) {
tmp = 1.0f - value * p.color2[1];
if (tmp <= 0.0f) {
p.out[1] = 1.0f;
}
else {
p.out[1] = p.color1[1] / tmp;
CLAMP_MAX(p.out[1], 1.0f);
}
}
else {
p.out[1] = 0.0f;
}
if (p.color1[2] != 0.0f) {
tmp = 1.0f - value * p.color2[2];
if (tmp <= 0.0f) {
p.out[2] = 1.0f;
}
else {
p.out[2] = p.color1[2] / tmp;
CLAMP_MAX(p.out[2], 1.0f);
}
}
else {
p.out[2] = 0.0f;
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Glare Operation ******** */
void MixGlareOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
const float value = p.value[0];
/* Linear interpolation between 3 cases:
* value=-1:output=input value=0:output=input+glare value=1:output=glare
*/
float input_weight;
float glare_weight;
if (value < 0.0f) {
input_weight = 1.0f;
glare_weight = 1.0f + value;
}
else {
input_weight = 1.0f - value;
glare_weight = 1.0f;
}
p.out[0] = input_weight * std::max(p.color1[0], 0.0f) + glare_weight * p.color2[0];
p.out[1] = input_weight * std::max(p.color1[1], 0.0f) + glare_weight * p.color2[1];
p.out[2] = input_weight * std::max(p.color1[2], 0.0f) + glare_weight * p.color2[2];
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Hue Operation ******** */
void MixHueOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
float colH, colS, colV;
rgb_to_hsv(p.color2[0], p.color2[1], p.color2[2], &colH, &colS, &colV);
if (colS != 0.0f) {
float rH, rS, rV;
float tmpr, tmpg, tmpb;
rgb_to_hsv(p.color1[0], p.color1[1], p.color1[2], &rH, &rS, &rV);
hsv_to_rgb(colH, rS, rV, &tmpr, &tmpg, &tmpb);
p.out[0] = value_m * p.color1[0] + value * tmpr;
p.out[1] = value_m * p.color1[1] + value * tmpg;
p.out[2] = value_m * p.color1[2] + value * tmpb;
}
else {
copy_v3_v3(p.out, p.color1);
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Lighten Operation ******** */
void MixLightenOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
float value_m = 1.0f - value;
p.out[0] = max_ff(p.color1[0], p.color2[0]) * value + p.color1[0] * value_m;
p.out[1] = max_ff(p.color1[1], p.color2[1]) * value + p.color1[1] * value_m;
p.out[2] = max_ff(p.color1[2], p.color2[2]) * value + p.color1[2] * value_m;
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Linear Light Operation ******** */
void MixLinearLightOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
if (p.color2[0] > 0.5f) {
p.out[0] = p.color1[0] + value * (2.0f * (p.color2[0] - 0.5f));
}
else {
p.out[0] = p.color1[0] + value * (2.0f * (p.color2[0]) - 1.0f);
}
if (p.color2[1] > 0.5f) {
p.out[1] = p.color1[1] + value * (2.0f * (p.color2[1] - 0.5f));
}
else {
p.out[1] = p.color1[1] + value * (2.0f * (p.color2[1]) - 1.0f);
}
if (p.color2[2] > 0.5f) {
p.out[2] = p.color1[2] + value * (2.0f * (p.color2[2] - 0.5f));
}
else {
p.out[2] = p.color1[2] + value * (2.0f * (p.color2[2]) - 1.0f);
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Multiply Operation ******** */
void MixMultiplyOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
p.out[0] = p.color1[0] * (value_m + value * p.color2[0]);
p.out[1] = p.color1[1] * (value_m + value * p.color2[1]);
p.out[2] = p.color1[2] * (value_m + value * p.color2[2]);
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Overlay Operation ******** */
void MixOverlayOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
if (p.color1[0] < 0.5f) {
p.out[0] = p.color1[0] * (value_m + 2.0f * value * p.color2[0]);
}
else {
p.out[0] = 1.0f - (value_m + 2.0f * value * (1.0f - p.color2[0])) * (1.0f - p.color1[0]);
}
if (p.color1[1] < 0.5f) {
p.out[1] = p.color1[1] * (value_m + 2.0f * value * p.color2[1]);
}
else {
p.out[1] = 1.0f - (value_m + 2.0f * value * (1.0f - p.color2[1])) * (1.0f - p.color1[1]);
}
if (p.color1[2] < 0.5f) {
p.out[2] = p.color1[2] * (value_m + 2.0f * value * p.color2[2]);
}
else {
p.out[2] = 1.0f - (value_m + 2.0f * value * (1.0f - p.color2[2])) * (1.0f - p.color1[2]);
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Saturation Operation ******** */
void MixSaturationOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
float rH, rS, rV;
rgb_to_hsv(p.color1[0], p.color1[1], p.color1[2], &rH, &rS, &rV);
if (rS != 0.0f) {
float colH, colS, colV;
rgb_to_hsv(p.color2[0], p.color2[1], p.color2[2], &colH, &colS, &colV);
hsv_to_rgb(rH, (value_m * rS + value * colS), rV, &p.out[0], &p.out[1], &p.out[2]);
}
else {
copy_v3_v3(p.out, p.color1);
}
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Screen Operation ******** */
void MixScreenOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
p.out[0] = 1.0f - (value_m + value * (1.0f - p.color2[0])) * (1.0f - p.color1[0]);
p.out[1] = 1.0f - (value_m + value * (1.0f - p.color2[1])) * (1.0f - p.color1[1]);
p.out[2] = 1.0f - (value_m + value * (1.0f - p.color2[2])) * (1.0f - p.color1[2]);
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Soft Light Operation ******** */
void MixSoftLightOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
const float value_m = 1.0f - value;
float scr, scg, scb;
/* First calculate non-fac based Screen mix. */
scr = 1.0f - (1.0f - p.color2[0]) * (1.0f - p.color1[0]);
scg = 1.0f - (1.0f - p.color2[1]) * (1.0f - p.color1[1]);
scb = 1.0f - (1.0f - p.color2[2]) * (1.0f - p.color1[2]);
p.out[0] = value_m * p.color1[0] +
value * ((1.0f - p.color1[0]) * p.color2[0] * p.color1[0] + p.color1[0] * scr);
p.out[1] = value_m * p.color1[1] +
value * ((1.0f - p.color1[1]) * p.color2[1] * p.color1[1] + p.color1[1] * scg);
p.out[2] = value_m * p.color1[2] +
value * ((1.0f - p.color1[2]) * p.color2[2] * p.color1[2] + p.color1[2] * scb);
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Subtract Operation ******** */
void MixSubtractOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
p.out[0] = p.color1[0] - value * p.color2[0];
p.out[1] = p.color1[1] - value * p.color2[1];
p.out[2] = p.color1[2] - value * p.color2[2];
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
/* ******** Mix Value Operation ******** */
void MixValueOperation::update_memory_buffer_row(PixelCursor &p)
{
while (p.out < p.row_end) {
float value = p.value[0];
if (this->use_value_alpha_multiply()) {
value *= p.color2[3];
}
float value_m = 1.0f - value;
float rH, rS, rV;
float colH, colS, colV;
rgb_to_hsv(p.color1[0], p.color1[1], p.color1[2], &rH, &rS, &rV);
rgb_to_hsv(p.color2[0], p.color2[1], p.color2[2], &colH, &colS, &colV);
hsv_to_rgb(rH, rS, (value_m * rV + value * colV), &p.out[0], &p.out[1], &p.out[2]);
p.out[3] = p.color1[3];
clamp_if_needed(p.out);
p.next();
}
}
} // namespace blender::compositor
Reference in New Issue View Git Blame Copy Permalink