Differential D12304 Diff 41477 source/blender/compositor/operations/COM_PosterizeOperation.cc

Changeset View

Standalone View

source/blender/compositor/operations/COM_PosterizeOperation.cc

This file was added.

				/*
				* This program is free software; you can redistribute it and/or
				* modify it under the terms of the GNU General Public License
				* as published by the Free Software Foundation; either version 2
				* of the License, or (at your option) any later version.
				*
				* This program is distributed in the hope that it will be useful,
				* but WITHOUT ANY WARRANTY; without even the implied warranty of
				* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
				* GNU General Public License for more details.
				*
				* You should have received a copy of the GNU General Public License
				* along with this program; if not, write to the Free Software Foundation,
				* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
				*
				* Copyright 2021, Blender Foundation.
				*/

				#include "COM_PosterizeOperation.h"

				namespace blender::compositor {

				PosterizeOperation::PosterizeOperation()
				{
				this->addInputSocket(DataType::Color);
				this->addInputSocket(DataType::Value);
				this->addOutputSocket(DataType::Color);
				this->m_inputProgram = nullptr;
				this->m_inputStepsProgram = nullptr;
				flags.can_be_constant = true;
				}

				void PosterizeOperation::initExecution()
				{
				this->m_inputProgram = this->getInputSocketReader(0);
				this->m_inputStepsProgram = this->getInputSocketReader(1);
				}

				void PosterizeOperation::executePixelSampled(float output[4],
				float x,
				float y,
				PixelSampler sampler)
				{
				float inputValue[4];
				float inputSteps[4];

				this->m_inputProgram->readSampled(inputValue, x, y, sampler);
				this->m_inputStepsProgram->readSampled(inputSteps, x, y, sampler);
				CLAMP(inputSteps[0], 2.0f, 1024.0f);
				const float steps_inv = 1.0f / inputSteps[0];

				output[0] = floor(inputValue[0] / steps_inv) * steps_inv;
				output[1] = floor(inputValue[1] / steps_inv) * steps_inv;
				output[2] = floor(inputValue[2] / steps_inv) * steps_inv;
				output[3] = inputValue[3];
				}

				void PosterizeOperation::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_value = it.in(0);
				const float *in_steps = it.in(1);
				float steps = in_steps[0];
				CLAMP(steps, 2.0f, 1024.0f);
				const float steps_inv = 1.0f / steps;

				it.out[0] = floor(in_value[0] / steps_inv) * steps_inv;
				it.out[1] = floor(in_value[1] / steps_inv) * steps_inv;
				it.out[2] = floor(in_value[2] / steps_inv) * steps_inv;
				it.out[3] = in_value[3];
				}
				}

				void PosterizeOperation::deinitExecution()
				{
				this->m_inputProgram = nullptr;
				this->m_inputStepsProgram = nullptr;
				}

				} // namespace blender::compositor