vector_transform.h

Go to the documentation of this file.

/* 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);
  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