"""Build the direction-adjusted reference depth field.""" from .nodes import math_node def reference_plane_depth(group, direction, reference_depth): nodes = group.nodes links = group.links normal = nodes.new("ShaderNodeVectorMath") normal.operation = "NORMALIZE" links.new(direction, normal.inputs[0]) normal_components = nodes.new("ShaderNodeSeparateXYZ") links.new(normal.outputs["Vector"], normal_components.inputs["Vector"]) position = nodes.new("GeometryNodeInputPosition") position_components = nodes.new("ShaderNodeSeparateXYZ") links.new(position.outputs["Position"], position_components.inputs["Vector"]) x_term = math_node(nodes, "MULTIPLY") links.new(position_components.outputs["X"], x_term.inputs[0]) links.new(normal_components.outputs["X"], x_term.inputs[1]) y_term = math_node(nodes, "MULTIPLY") links.new(position_components.outputs["Y"], y_term.inputs[0]) links.new(normal_components.outputs["Y"], y_term.inputs[1]) point_dot = math_node(nodes, "ADD") links.new(x_term.outputs[0], point_dot.inputs[0]) links.new(y_term.outputs[0], point_dot.inputs[1]) divide = math_node(nodes, "DIVIDE") links.new(point_dot.outputs[0], divide.inputs[0]) links.new(normal_components.outputs["Z"], divide.inputs[1]) plane = math_node(nodes, "ADD") links.new(reference_depth, plane.inputs[0]) links.new(divide.outputs[0], plane.inputs[1]) return plane.outputs[0]