Campbell Barton
c434782e3a
Use a shorter/simpler license convention, stops the header taking so much space. Follow the SPDX license specification: https://spdx.org/licenses - C/C++/objc/objc++ - Python - Shell Scripts - CMake, GNUmakefile While most of the source tree has been included - `./extern/` was left out. - `./intern/cycles` & `./intern/atomic` are also excluded because they use different header conventions. doc/license/SPDX-license-identifiers.txt has been added to list SPDX all used identifiers. See P2788 for the script that automated these edits. Reviewed By: brecht, mont29, sergey Ref D14069
48 lines
1.4 KiB
C++
48 lines
1.4 KiB
C++
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "BKE_attribute_math.hh"
|
|
|
|
namespace blender::attribute_math {
|
|
|
|
ColorGeometryMixer::ColorGeometryMixer(MutableSpan<ColorGeometry4f> output_buffer,
|
|
ColorGeometry4f default_color)
|
|
: buffer_(output_buffer),
|
|
default_color_(default_color),
|
|
total_weights_(output_buffer.size(), 0.0f)
|
|
{
|
|
buffer_.fill(ColorGeometry4f(0.0f, 0.0f, 0.0f, 0.0f));
|
|
}
|
|
|
|
void ColorGeometryMixer::mix_in(const int64_t index,
|
|
const ColorGeometry4f &color,
|
|
const float weight)
|
|
{
|
|
BLI_assert(weight >= 0.0f);
|
|
ColorGeometry4f &output_color = buffer_[index];
|
|
output_color.r += color.r * weight;
|
|
output_color.g += color.g * weight;
|
|
output_color.b += color.b * weight;
|
|
output_color.a += color.a * weight;
|
|
total_weights_[index] += weight;
|
|
}
|
|
|
|
void ColorGeometryMixer::finalize()
|
|
{
|
|
for (const int64_t i : buffer_.index_range()) {
|
|
const float weight = total_weights_[i];
|
|
ColorGeometry4f &output_color = buffer_[i];
|
|
if (weight > 0.0f) {
|
|
const float weight_inv = 1.0f / weight;
|
|
output_color.r *= weight_inv;
|
|
output_color.g *= weight_inv;
|
|
output_color.b *= weight_inv;
|
|
output_color.a *= weight_inv;
|
|
}
|
|
else {
|
|
output_color = default_color_;
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace blender::attribute_math
|