73 lines
2.3 KiB
C++
73 lines
2.3 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#include "kernel/svm/node_types.h"
|
|
#include "kernel/svm/util.h"
|
|
|
|
#include "kernel/util/colorspace.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
/* Conversion Nodes */
|
|
|
|
template<typename FloatType, typename Float3Type>
|
|
ccl_device_noinline void svm_node_convert(KernelGlobals kg,
|
|
ccl_private float *ccl_restrict stack,
|
|
const ccl_global SVMNodeConvert &ccl_restrict node)
|
|
{
|
|
|
|
switch (node.convert_type) {
|
|
case NODE_CONVERT_FI: {
|
|
/* TODO(weizhen): should actually store 0 for int, but none of the nodes that we compute
|
|
* derivatives for has int inputs, so seems fine. */
|
|
const float f = stack_load_float(stack, node.from_offset);
|
|
stack_store_int(stack, node.to_offset, float_to_int(f));
|
|
break;
|
|
}
|
|
case NODE_CONVERT_FV: {
|
|
const FloatType f = stack_load<FloatType>(stack, node.from_offset);
|
|
stack_store(stack, node.to_offset, make_float3(f, f, f));
|
|
break;
|
|
}
|
|
case NODE_CONVERT_CF: {
|
|
const Float3Type f = stack_load<Float3Type>(stack, node.from_offset);
|
|
stack_store(stack, node.to_offset, linear_rgb_to_gray(kg, f));
|
|
break;
|
|
}
|
|
case NODE_CONVERT_CI: {
|
|
const float3 f = stack_load_float3(stack, node.from_offset);
|
|
const int i = (int)linear_rgb_to_gray(kg, f);
|
|
stack_store_int(stack, node.to_offset, i);
|
|
break;
|
|
}
|
|
case NODE_CONVERT_VF: {
|
|
const Float3Type f = stack_load<Float3Type>(stack, node.from_offset);
|
|
stack_store(stack, node.to_offset, average(f));
|
|
break;
|
|
}
|
|
case NODE_CONVERT_VI: {
|
|
const float3 f = stack_load_float3(stack, node.from_offset);
|
|
const int i = (int)average(f);
|
|
stack_store_int(stack, node.to_offset, i);
|
|
break;
|
|
}
|
|
case NODE_CONVERT_IF: {
|
|
const float f = (float)stack_load_int(stack, node.from_offset);
|
|
stack_store(stack, node.to_offset, FloatType(f));
|
|
break;
|
|
}
|
|
case NODE_CONVERT_IV: {
|
|
const float f = (float)stack_load_int(stack, node.from_offset);
|
|
stack_store(stack, node.to_offset, Float3Type(make_float3(f, f, f)));
|
|
break;
|
|
}
|
|
default:
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
CCL_NAMESPACE_END
|