89 lines
2.2 KiB
C++
89 lines
2.2 KiB
C++
|
/*
|
||
|
|
* Copyright 2011-2021 Blender Foundation
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "device/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;
|
||
|
|
}
|
||
|
|
|
||
|
|
NODE_DEFINE(DenoiseParams)
|
||
|
|
{
|
||
|
|
NodeType *type = NodeType::add("denoise_params", create);
|
||
|
|
|
||
|
|
const NodeEnum *type_enum = get_type_enum();
|
||
|
|
const NodeEnum *prefilter_enum = get_prefilter_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_ENUM(prefilter, "Prefilter", *prefilter_enum, DENOISER_PREFILTER_FAST);
|
||
|
|
|
||
|
|
return type;
|
||
|
|
}
|
||
|
|
|
||
|
|
DenoiseParams::DenoiseParams() : Node(get_node_type())
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
CCL_NAMESPACE_END
|