2a0a6f18cc
This adds a new "Quality" option for OIDN to switch between the existing "High" and "Balanced" modes and the new "Fast" mode introduced in OIDN 2.3. Pull Request: https://projects.blender.org/blender/blender/pulls/121374
92 lines
2.2 KiB
C++
92 lines
2.2 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#include "device/denoise.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
const char *denoiserTypeToHumanReadable(DenoiserType type)
|
|
{
|
|
switch (type) {
|
|
case DENOISER_OPTIX:
|
|
return "OptiX";
|
|
case DENOISER_OPENIMAGEDENOISE:
|
|
return "OpenImageDenoise";
|
|
|
|
case DENOISER_NUM:
|
|
case DENOISER_NONE:
|
|
case DENOISER_ALL:
|
|
return "UNKNOWN";
|
|
}
|
|
|
|
return "UNKNOWN";
|
|
}
|
|
|
|
const NodeEnum *DenoiseParams::get_type_enum()
|
|
{
|
|
static NodeEnum type_enum;
|
|
|
|
if (type_enum.empty()) {
|
|
type_enum.insert("optix", DENOISER_OPTIX);
|
|
type_enum.insert("openimageio", DENOISER_OPENIMAGEDENOISE);
|
|
}
|
|
|
|
return &type_enum;
|
|
}
|
|
|
|
const NodeEnum *DenoiseParams::get_prefilter_enum()
|
|
{
|
|
static NodeEnum prefilter_enum;
|
|
|
|
if (prefilter_enum.empty()) {
|
|
prefilter_enum.insert("none", DENOISER_PREFILTER_NONE);
|
|
prefilter_enum.insert("fast", DENOISER_PREFILTER_FAST);
|
|
prefilter_enum.insert("accurate", DENOISER_PREFILTER_ACCURATE);
|
|
}
|
|
|
|
return &prefilter_enum;
|
|
}
|
|
|
|
const NodeEnum *DenoiseParams::get_quality_enum()
|
|
{
|
|
static NodeEnum quality_enum;
|
|
|
|
if (quality_enum.empty()) {
|
|
quality_enum.insert("high", DENOISER_QUALITY_HIGH);
|
|
quality_enum.insert("balanced", DENOISER_QUALITY_BALANCED);
|
|
quality_enum.insert("fast", DENOISER_QUALITY_FAST);
|
|
}
|
|
|
|
return &quality_enum;
|
|
}
|
|
|
|
NODE_DEFINE(DenoiseParams)
|
|
{
|
|
NodeType *type = NodeType::add("denoise_params", create);
|
|
|
|
const NodeEnum *type_enum = get_type_enum();
|
|
const NodeEnum *prefilter_enum = get_prefilter_enum();
|
|
const NodeEnum *quality_enum = get_quality_enum();
|
|
|
|
SOCKET_BOOLEAN(use, "Use", false);
|
|
|
|
SOCKET_ENUM(type, "Type", *type_enum, DENOISER_OPENIMAGEDENOISE);
|
|
|
|
SOCKET_INT(start_sample, "Start Sample", 0);
|
|
|
|
SOCKET_BOOLEAN(use_pass_albedo, "Use Pass Albedo", true);
|
|
SOCKET_BOOLEAN(use_pass_normal, "Use Pass Normal", false);
|
|
|
|
SOCKET_BOOLEAN(temporally_stable, "Temporally Stable", false);
|
|
|
|
SOCKET_ENUM(prefilter, "Prefilter", *prefilter_enum, DENOISER_PREFILTER_FAST);
|
|
SOCKET_ENUM(quality, "Quality", *quality_enum, DENOISER_QUALITY_HIGH);
|
|
|
|
return type;
|
|
}
|
|
|
|
DenoiseParams::DenoiseParams() : Node(get_node_type()) {}
|
|
|
|
CCL_NAMESPACE_END
|