Hans Goudey
91d9f46aec
Push the const usage a bit further for compositor nodes, so that they are more explicit about not modifying original nodes from the editor. Differential Revision: https://developer.blender.org/D15822
99 lines
2.4 KiB
C++
99 lines
2.4 KiB
C++
/* SPDX-License-Identifier: GPL-2.0-or-later
|
|
* Copyright 2019 Blender Foundation. */
|
|
|
|
#pragma once
|
|
|
|
#include "COM_SingleThreadedOperation.h"
|
|
#include "DNA_node_types.h"
|
|
|
|
namespace blender::compositor {
|
|
|
|
bool COM_is_denoise_supported();
|
|
|
|
class DenoiseBaseOperation : public SingleThreadedOperation {
|
|
protected:
|
|
bool output_rendered_;
|
|
|
|
protected:
|
|
DenoiseBaseOperation();
|
|
|
|
public:
|
|
bool determine_depending_area_of_interest(rcti *input,
|
|
ReadBufferOperation *read_operation,
|
|
rcti *output) override;
|
|
|
|
void get_area_of_interest(int input_idx, const rcti &output_area, rcti &r_input_area) override;
|
|
};
|
|
|
|
class DenoiseOperation : public DenoiseBaseOperation {
|
|
private:
|
|
/**
|
|
* \brief Cached reference to the input programs
|
|
*/
|
|
SocketReader *input_program_color_;
|
|
SocketReader *input_program_albedo_;
|
|
SocketReader *input_program_normal_;
|
|
|
|
/**
|
|
* \brief settings of the denoise node.
|
|
*/
|
|
const NodeDenoise *settings_;
|
|
|
|
public:
|
|
DenoiseOperation();
|
|
/**
|
|
* Initialize the execution
|
|
*/
|
|
void init_execution() override;
|
|
|
|
/**
|
|
* Deinitialize the execution
|
|
*/
|
|
void deinit_execution() override;
|
|
|
|
void set_denoise_settings(const NodeDenoise *settings)
|
|
{
|
|
settings_ = settings;
|
|
}
|
|
|
|
void update_memory_buffer(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs) override;
|
|
|
|
protected:
|
|
void hash_output_params() override;
|
|
void generate_denoise(MemoryBuffer *output,
|
|
MemoryBuffer *input_color,
|
|
MemoryBuffer *input_normal,
|
|
MemoryBuffer *input_albedo,
|
|
const NodeDenoise *settings);
|
|
|
|
MemoryBuffer *create_memory_buffer(rcti *rect) override;
|
|
};
|
|
|
|
class DenoisePrefilterOperation : public DenoiseBaseOperation {
|
|
private:
|
|
std::string image_name_;
|
|
|
|
public:
|
|
DenoisePrefilterOperation(DataType data_type);
|
|
|
|
void set_image_name(StringRef name)
|
|
{
|
|
image_name_ = name;
|
|
}
|
|
|
|
void update_memory_buffer(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs) override;
|
|
|
|
protected:
|
|
void hash_output_params() override;
|
|
MemoryBuffer *create_memory_buffer(rcti *rect) override;
|
|
|
|
private:
|
|
void generate_denoise(MemoryBuffer *output, MemoryBuffer *input);
|
|
};
|
|
|
|
} // namespace blender::compositor
|