Differential D12341 Diff 41207 source/blender/compositor/operations/COM_InpaintOperation.cc

Changeset View

Standalone View

source/blender/compositor/operations/COM_InpaintOperation.cc

Show All 33 Lines	InpaintSimpleOperation::InpaintSimpleOperation()
this->addInputSocket(DataType::Color);		this->addInputSocket(DataType::Color);
this->addOutputSocket(DataType::Color);		this->addOutputSocket(DataType::Color);
this->flags.complex = true;		this->flags.complex = true;
this->m_inputImageProgram = nullptr;		this->m_inputImageProgram = nullptr;
this->m_pixelorder = nullptr;		this->m_pixelorder = nullptr;
this->m_manhattan_distance = nullptr;		this->m_manhattan_distance = nullptr;
this->m_cached_buffer = nullptr;		this->m_cached_buffer = nullptr;
this->m_cached_buffer_ready = false;		this->m_cached_buffer_ready = false;
		flags.is_fullframe_operation = true;
}		}
void InpaintSimpleOperation::initExecution()		void InpaintSimpleOperation::initExecution()
{		{
this->m_inputImageProgram = this->getInputSocketReader(0);		this->m_inputImageProgram = this->getInputSocketReader(0);

this->m_pixelorder = nullptr;		this->m_pixelorder = nullptr;
this->m_manhattan_distance = nullptr;		this->m_manhattan_distance = nullptr;
this->m_cached_buffer = nullptr;		this->m_cached_buffer = nullptr;
▲ Show 20 Lines • Show All 231 Lines • ▼ Show 20 Lines	bool InpaintSimpleOperation::determineDependingAreaOfInterest(rcti * /input/,
newInput.xmax = getWidth();		newInput.xmax = getWidth();
newInput.xmin = 0;		newInput.xmin = 0;
newInput.ymax = getHeight();		newInput.ymax = getHeight();
newInput.ymin = 0;		newInput.ymin = 0;

return NodeOperation::determineDependingAreaOfInterest(&newInput, readOperation, output);		return NodeOperation::determineDependingAreaOfInterest(&newInput, readOperation, output);
}		}

		void InpaintSimpleOperation::get_area_of_interest(const int input_idx,
		const rcti &UNUSED(output_area),
		rcti &r_input_area)
		{
		BLI_assert(input_idx == 0);
		UNUSED_VARS_NDEBUG(input_idx);
		r_input_area.xmin = 0;
		r_input_area.xmax = this->getWidth();
		r_input_area.ymin = 0;
		r_input_area.ymax = this->getHeight();
		}

		void InpaintSimpleOperation::update_memory_buffer(MemoryBuffer *output,
		const rcti &area,
		Span<MemoryBuffer *> inputs)
		{
		/* TODO(manzanilla): once tiled implementation is removed, run multi-threaded where possible. */
		MemoryBuffer *input = inputs[0];
		if (!m_cached_buffer_ready) {
		if (input->is_a_single_elem()) {
		MemoryBuffer *tmp = input->inflate();
		m_cached_buffer = tmp->release_ownership_buffer();
		delete tmp;
		}
		else {
		m_cached_buffer = (float *)MEM_dupallocN(input->getBuffer());
		}

		this->calc_manhattan_distance();

		int curr = 0;
		int x, y;
		while (this->next_pixel(x, y, curr, this->m_iterations)) {
		this->pix_step(x, y);
		}
		m_cached_buffer_ready = true;
		}

		const int num_channels = COM_data_type_num_channels(getOutputSocket()->getDataType());
		MemoryBuffer buf(m_cached_buffer, num_channels, input->getWidth(), input->getHeight());
		output->copy_from(&buf, area);
		}

} // namespace blender::compositor		} // namespace blender::compositor