a859837cde
Cycles already treats denoising fairly separate in its code, with a dedicated `Denoiser` base class used to describe denoising behavior. That class has been fully implemented for OIDN (`denoiser_oidn.cpp`), but for OptiX was mostly empty (`denoiser_optix.cpp`) and denoising was instead implemented in the OptiX device. That meant denoising code was split over various files and directories, making it a bit awkward to work with. This patch moves the OptiX denoising implementation into the existing `OptiXDenoiser` class, so that everything is in one place. There are no functional changes, code has been mostly moved as-is. To retain support for potential other denoiser implementations based on a GPU device in the future, the `DeviceDenoiser` base class was kept and slightly extended (and its file renamed to `denoiser_gpu.cpp` to follow similar naming rules as `path_trace_work_*.cpp`). Differential Revision: https://developer.blender.org/D16502
87 lines
3.2 KiB
C++
87 lines
3.2 KiB
C++
/* SPDX-License-Identifier: Apache-2.0
|
|
* Copyright 2011-2022 Blender Foundation */
|
|
|
|
#pragma once
|
|
|
|
#ifdef WITH_OPTIX
|
|
|
|
# include "integrator/denoiser_gpu.h"
|
|
|
|
# include "device/optix/util.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
/* Implementation of denoising API which uses the OptiX denoiser. */
|
|
class OptiXDenoiser : public DenoiserGPU {
|
|
public:
|
|
OptiXDenoiser(Device *path_trace_device, const DenoiseParams ¶ms);
|
|
~OptiXDenoiser();
|
|
|
|
protected:
|
|
virtual uint get_device_type_mask() const override;
|
|
|
|
private:
|
|
class DenoiseContext;
|
|
class DenoisePass;
|
|
|
|
virtual bool denoise_buffer(const DenoiseTask &task) override;
|
|
|
|
/* Read guiding passes from the render buffers, preprocess them in a way which is expected by
|
|
* OptiX and store in the guiding passes memory within the given context.
|
|
*
|
|
* Pre-processing of the guiding passes is to only happen once per context lifetime. DO not
|
|
* preprocess them for every pass which is being denoised. */
|
|
bool denoise_filter_guiding_preprocess(const DenoiseContext &context);
|
|
|
|
/* Set fake albedo pixels in the albedo guiding pass storage.
|
|
* After this point only passes which do not need albedo for denoising can be processed. */
|
|
bool denoise_filter_guiding_set_fake_albedo(const DenoiseContext &context);
|
|
|
|
void denoise_pass(DenoiseContext &context, PassType pass_type);
|
|
|
|
/* Read input color pass from the render buffer into the memory which corresponds to the noisy
|
|
* input within the given context. Pixels are scaled to the number of samples, but are not
|
|
* preprocessed yet. */
|
|
void denoise_color_read(const DenoiseContext &context, const DenoisePass &pass);
|
|
|
|
/* Run corresponding filter kernels, preparing data for the denoiser or copying data from the
|
|
* denoiser result to the render buffer. */
|
|
bool denoise_filter_color_preprocess(const DenoiseContext &context, const DenoisePass &pass);
|
|
bool denoise_filter_color_postprocess(const DenoiseContext &context, const DenoisePass &pass);
|
|
|
|
/* Make sure the OptiX denoiser is created and configured. */
|
|
bool denoise_ensure(DenoiseContext &context);
|
|
|
|
/* Create OptiX denoiser descriptor if needed.
|
|
* Will do nothing if the current OptiX descriptor is usable for the given parameters.
|
|
* If the OptiX denoiser descriptor did re-allocate here it is left unconfigured. */
|
|
bool denoise_create_if_needed(DenoiseContext &context);
|
|
|
|
/* Configure existing OptiX denoiser descriptor for the use for the given task. */
|
|
bool denoise_configure_if_needed(DenoiseContext &context);
|
|
|
|
/* Run configured denoiser. */
|
|
bool denoise_run(const DenoiseContext &context, const DenoisePass &pass);
|
|
|
|
OptixDenoiser optix_denoiser_ = nullptr;
|
|
|
|
/* Configuration size, as provided to `optixDenoiserSetup`.
|
|
* If the `optixDenoiserSetup()` was never used on the current `optix_denoiser` the
|
|
* `is_configured` will be false. */
|
|
bool is_configured_ = false;
|
|
int2 configured_size_ = make_int2(0, 0);
|
|
|
|
/* OptiX denoiser state and scratch buffers, stored in a single memory buffer.
|
|
* The memory layout goes as following: [denoiser state][scratch buffer]. */
|
|
device_only_memory<unsigned char> state_;
|
|
OptixDenoiserSizes sizes_ = {};
|
|
|
|
bool use_pass_albedo_ = false;
|
|
bool use_pass_normal_ = false;
|
|
bool use_pass_motion_ = false;
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif
|