Code refactored to incorporate #6

addons/ShaderLib_v2_2_4/Maths/Vector/Project.gd

@@ -0,0 +1,77 @@
+@tool
+class_name VisualShaderNodeVectorProject extends ShaderLib
+
+func _get_name() -> String:
+	return "Project"
+
+func _get_category() -> String:
+	return "Maths/Vector"
+
+func _get_description() -> String:
+	return "Projects vector A onto vector B."
+
+func _get_return_icon_type() -> PortType:
+	var vector_index: int = get_option_index(0)
+	match vector_index:
+		0:
+			return PORT_TYPE_VECTOR_2D
+		_:
+			return PORT_TYPE_VECTOR_3D
+
+func _get_input_port_count() -> int:
+	return 2
+
+func _get_input_port_name(port: int) -> String:
+	match port:
+		0:
+			return "vector A"
+		_:
+			return "vector B"
+
+func _get_input_port_type(port: int) -> PortType:
+	var vector_index: int = get_option_index(0)
+	match vector_index:
+		0:
+			return PORT_TYPE_VECTOR_2D
+		_:
+			return PORT_TYPE_VECTOR_3D
+
+func _get_output_port_count() -> int:
+	return 1
+
+func _get_output_port_name(port: int) -> String:
+	return "vector"
+
+func _get_output_port_type(port: int) -> PortType:
+	var vector_index: int = get_option_index(0)
+	match vector_index:
+		0:
+			return PORT_TYPE_VECTOR_2D
+		_:
+			return PORT_TYPE_VECTOR_3D
+
+func _get_property_count() -> int:
+	return 1
+
+func _get_property_default_index(index: int) -> int:
+	return 0
+
+func _get_property_name(index: int) -> String:
+	return ""
+
+func _get_property_options(index: int) -> PackedStringArray:
+	return ["Vector2", "Vector3"]
+
+func _get_global_code(mode: Shader.Mode) -> String:
+	return "#include \"res://addons/ShaderLib_%s/Maths/Maths.gdshaderinc\"" % [version]
+
+func _get_code(input_vars: Array[String], output_vars: Array[String], mode: Shader.Mode, type: VisualShader.Type) -> String:
+	var vector_a: String = input_vars[0]
+	var vector_b: String = input_vars[1]
+	var vector_index: int = get_option_index(0)
+
+	match vector_index:
+		0:
+			return output_vars[0] + " = project_2d(%s, %s);" % [vector_a, vector_b]
+		_:
+			return output_vars[0] + " = project_3d(%s, %s);" % [vector_a, vector_b]