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
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.
|
|
*/
|
|
NodeDenoise *settings_;
|
|
|
|
public:
|
|
DenoiseOperation();
|
|
/**
|
|
* Initialize the execution
|
|
*/
|
|
void init_execution() override;
|
|
|
|
/**
|
|
* Deinitialize the execution
|
|
*/
|
|
void deinit_execution() override;
|
|
|
|
void set_denoise_settings(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,
|
|
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
|