26c93c8359
This enables the new lazy module loading behavior introduced in OIDN 2.3, without breaking compatibility with older versions of OIDN (using separate code paths). Also, the detection of OIDN support for devices is now much cleaner, and devices do not need to be matched by PCI address or device name anymore. Pull Request: https://projects.blender.org/blender/blender/pulls/121362
89 lines
2.7 KiB
C++
89 lines
2.7 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#if defined(WITH_OPENIMAGEDENOISE)
|
|
|
|
# include "integrator/denoiser_gpu.h"
|
|
# include "util/openimagedenoise.h"
|
|
# include "util/thread.h"
|
|
# include "util/unique_ptr.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
/* Implementation of a GPU denoiser which uses OpenImageDenoise library. */
|
|
class OIDNDenoiserGPU : public DenoiserGPU {
|
|
friend class OIDNDenoiseContext;
|
|
|
|
public:
|
|
/* Forwardly declared state which might be using compile-flag specific fields, such as
|
|
* OpenImageDenoise device and filter handles. */
|
|
class State;
|
|
|
|
OIDNDenoiserGPU(Device *path_trace_device, const DenoiseParams ¶ms);
|
|
~OIDNDenoiserGPU();
|
|
|
|
virtual bool denoise_buffer(const BufferParams &buffer_params,
|
|
RenderBuffers *render_buffers,
|
|
const int num_samples,
|
|
bool allow_inplace_modification) override;
|
|
|
|
static bool is_device_supported(const DeviceInfo &device);
|
|
|
|
protected:
|
|
enum class ExecMode {
|
|
SYNC,
|
|
ASYNC,
|
|
};
|
|
|
|
virtual uint get_device_type_mask() const override;
|
|
|
|
/* Create OIDN denoiser descriptor if needed.
|
|
* Will do nothing if the current OIDN descriptor is usable for the given parameters.
|
|
* If the OIDN denoiser descriptor did re-allocate here it is left unconfigured. */
|
|
virtual bool denoise_create_if_needed(DenoiseContext &context) override;
|
|
|
|
/* Configure existing OIDN denoiser descriptor for the use for the given task. */
|
|
virtual bool denoise_configure_if_needed(DenoiseContext &context) override;
|
|
|
|
/* Run configured denoiser. */
|
|
virtual bool denoise_run(const DenoiseContext &context, const DenoisePass &pass) override;
|
|
|
|
OIDNFilter create_filter();
|
|
bool commit_and_execute_filter(OIDNFilter filter, ExecMode mode = ExecMode::SYNC);
|
|
|
|
void set_filter_pass(OIDNFilter filter,
|
|
const char *name,
|
|
device_ptr ptr,
|
|
int format,
|
|
int width,
|
|
int height,
|
|
size_t offset_in_bytes,
|
|
size_t pixel_stride_in_bytes,
|
|
size_t row_stride_in_bytes);
|
|
|
|
/* Delete all allocated OIDN objects. */
|
|
void release_all_resources();
|
|
|
|
OIDNDevice oidn_device_ = nullptr;
|
|
OIDNFilter oidn_filter_ = nullptr;
|
|
OIDNFilter albedo_filter_ = nullptr;
|
|
OIDNFilter normal_filter_ = nullptr;
|
|
|
|
bool is_configured_ = false;
|
|
int2 configured_size_ = make_int2(0, 0);
|
|
|
|
bool use_pass_albedo_ = false;
|
|
bool use_pass_normal_ = false;
|
|
DenoiserQuality quality_ = DENOISER_QUALITY_HIGH;
|
|
|
|
/* Filter memory usage limit if we ran out of memory with OIDN's default limit. */
|
|
int max_mem_ = 768;
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif
|