91 lines
2.2 KiB
C++
91 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_INT(passes, "Passes", DENOISER_PASS_ALBEDO | DENOISER_PASS_NORMAL);
|
|
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);
|
|
SOCKET_FLOAT(upscale_factor, "Upscale Factor", 1.0f);
|
|
|
|
return type;
|
|
}
|
|
|
|
DenoiseParams::DenoiseParams() : Node(get_node_type()) {}
|
|
|
|
CCL_NAMESPACE_END
|