Differential D10978 Diff 37170 source/blender/blenkernel/intern/attribute_math.cc

Changeset View

Standalone View

source/blender/blenkernel/intern/attribute_math.cc

	Show All 12 Lines
	* along with this program; if not, write to the Free Software Foundation,			* along with this program; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.			* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
	*/			*/

	#include "BKE_attribute_math.hh"			#include "BKE_attribute_math.hh"

	namespace blender::attribute_math {			namespace blender::attribute_math {

	Color4fMixer::Color4fMixer(MutableSpan<Color4f> output_buffer, Color4f default_color)			ColorGeometryMixer::ColorGeometryMixer(MutableSpan<ColorGeometry4f> output_buffer,
				ColorGeometry4f default_color)
	: buffer_(output_buffer),			: buffer_(output_buffer),
	default_color_(default_color),			default_color_(default_color),
	total_weights_(output_buffer.size(), 0.0f)			total_weights_(output_buffer.size(), 0.0f)
	{			{
	buffer_.fill(Color4f(0, 0, 0, 0));			buffer_.fill(ColorGeometry4f(0.0f, 0.0f, 0.0f, 0.0f));
	}			}

	void Color4fMixer::mix_in(const int64_t index, const Color4f &color, const float weight)			void ColorGeometryMixer::mix_in(const int64_t index,
				const ColorGeometry4f &color,
				const float weight)
	{			{
	BLI_assert(weight >= 0.0f);			BLI_assert(weight >= 0.0f);
	Color4f &output_color = buffer_[index];			ColorGeometry4f &output_color = buffer_[index];
	output_color.r += color.r * weight;			output_color.r += color.r * weight;
	output_color.g += color.g * weight;			output_color.g += color.g * weight;
	output_color.b += color.b * weight;			output_color.b += color.b * weight;
	output_color.a += color.a * weight;			output_color.a += color.a * weight;
	total_weights_[index] += weight;			total_weights_[index] += weight;
	}			}

	void Color4fMixer::finalize()			void ColorGeometryMixer::finalize()
	{			{
	for (const int64_t i : buffer_.index_range()) {			for (const int64_t i : buffer_.index_range()) {
	const float weight = total_weights_[i];			const float weight = total_weights_[i];
	Color4f &output_color = buffer_[i];			ColorGeometry4f &output_color = buffer_[i];
	if (weight > 0.0f) {			if (weight > 0.0f) {
	const float weight_inv = 1.0f / weight;			const float weight_inv = 1.0f / weight;
	output_color.r *= weight_inv;			output_color.r *= weight_inv;
	output_color.g *= weight_inv;			output_color.g *= weight_inv;
	output_color.b *= weight_inv;			output_color.b *= weight_inv;
	output_color.a *= weight_inv;			output_color.a *= weight_inv;
	}			}
	else {			else {
	output_color = default_color_;			output_color = default_color_;
	}			}
	}			}
	}			}

	} // namespace blender::attribute_math			} // namespace blender::attribute_math