131 lines
3.8 KiB
C
131 lines
3.8 KiB
C
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#include "kernel/geom/object.h"
|
|
#include "kernel/svm/util.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
/* Vector Transform */
|
|
|
|
ccl_device_noinline void svm_node_vector_transform(KernelGlobals kg,
|
|
ccl_private ShaderData *sd,
|
|
ccl_private float *stack,
|
|
const uint4 node)
|
|
{
|
|
uint itype;
|
|
uint ifrom;
|
|
uint ito;
|
|
uint vector_in;
|
|
uint vector_out;
|
|
|
|
svm_unpack_node_uchar3(node.y, &itype, &ifrom, &ito);
|
|
svm_unpack_node_uchar2(node.z, &vector_in, &vector_out);
|
|
|
|
float3 in = stack_load_float3(stack, vector_in);
|
|
|
|
const NodeVectorTransformType type = (NodeVectorTransformType)itype;
|
|
const NodeVectorTransformConvertSpace from = (NodeVectorTransformConvertSpace)ifrom;
|
|
const NodeVectorTransformConvertSpace to = (NodeVectorTransformConvertSpace)ito;
|
|
|
|
Transform tfm;
|
|
const bool is_object = (sd->object != OBJECT_NONE) || (sd->type == PRIMITIVE_LAMP);
|
|
const bool is_normal = (type == NODE_VECTOR_TRANSFORM_TYPE_NORMAL);
|
|
const bool is_direction = (type == NODE_VECTOR_TRANSFORM_TYPE_VECTOR);
|
|
|
|
/* From world */
|
|
if (from == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_WORLD) {
|
|
if (to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_CAMERA) {
|
|
if (is_normal) {
|
|
tfm = kernel_data.cam.cameratoworld;
|
|
in = normalize(transform_direction_transposed(&tfm, in));
|
|
}
|
|
else {
|
|
tfm = kernel_data.cam.worldtocamera;
|
|
in = is_direction ? transform_direction(&tfm, in) : transform_point(&tfm, in);
|
|
}
|
|
}
|
|
else if (to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_OBJECT && is_object) {
|
|
if (is_normal) {
|
|
object_inverse_normal_transform(kg, sd, &in);
|
|
}
|
|
else if (is_direction) {
|
|
object_inverse_dir_transform(kg, sd, &in);
|
|
}
|
|
else {
|
|
object_inverse_position_transform(kg, sd, &in);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* From camera */
|
|
else if (from == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_CAMERA) {
|
|
if (to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_WORLD ||
|
|
to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_OBJECT)
|
|
{
|
|
if (is_normal) {
|
|
tfm = kernel_data.cam.worldtocamera;
|
|
in = normalize(transform_direction_transposed(&tfm, in));
|
|
}
|
|
else {
|
|
tfm = kernel_data.cam.cameratoworld;
|
|
in = is_direction ? transform_direction(&tfm, in) : transform_point(&tfm, in);
|
|
}
|
|
}
|
|
if (to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_OBJECT && is_object) {
|
|
if (is_normal) {
|
|
object_inverse_normal_transform(kg, sd, &in);
|
|
}
|
|
else if (is_direction) {
|
|
object_inverse_dir_transform(kg, sd, &in);
|
|
}
|
|
else {
|
|
object_inverse_position_transform(kg, sd, &in);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* From object */
|
|
else if (from == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_OBJECT) {
|
|
if ((to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_WORLD ||
|
|
to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_CAMERA) &&
|
|
is_object)
|
|
{
|
|
if (is_normal) {
|
|
object_normal_transform(kg, sd, &in);
|
|
}
|
|
else if (is_direction) {
|
|
object_dir_transform(kg, sd, &in);
|
|
}
|
|
else {
|
|
object_position_transform(kg, sd, &in);
|
|
}
|
|
}
|
|
if (to == NODE_VECTOR_TRANSFORM_CONVERT_SPACE_CAMERA) {
|
|
if (is_normal) {
|
|
tfm = kernel_data.cam.cameratoworld;
|
|
in = normalize(transform_direction_transposed(&tfm, in));
|
|
}
|
|
else {
|
|
tfm = kernel_data.cam.worldtocamera;
|
|
if (is_direction) {
|
|
in = transform_direction(&tfm, in);
|
|
}
|
|
else {
|
|
in = transform_point(&tfm, in);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Output */
|
|
if (stack_valid(vector_out)) {
|
|
stack_store_float3(stack, vector_out, in);
|
|
}
|
|
}
|
|
|
|
CCL_NAMESPACE_END
|