1
0
mirror of https://github.com/DigvijaysinhGohil/Godot-Shader-Lib.git synced 2025-01-08 10:23:24 +08:00
2024-09-28 00:45:16 +05:30

102 lines
2.4 KiB
GDScript

@tool
class_name VisualShaderNodeMathsSquareWave extends ShaderLib
func _get_name() -> String:
return "SquareWave"
func _get_category() -> String:
return "Maths/Wave"
func _get_description() -> String:
return "Returns a square wave from the value of input in. Resulting output values will be between -1 and 1."
func _get_return_icon_type() -> VisualShaderNode.PortType:
var vector_index: int = get_option_index(0)
match vector_index:
0:
return PORT_TYPE_SCALAR
1:
return PORT_TYPE_VECTOR_2D
2:
return PORT_TYPE_VECTOR_3D
_:
return PORT_TYPE_VECTOR_4D
func _get_input_port_count() -> int:
return 1
func _get_input_port_name(port: int) -> String:
return "input"
func _get_input_port_type(port: int) -> VisualShaderNode.PortType:
var vector_index: int = get_option_index(0)
match vector_index:
0:
return PORT_TYPE_SCALAR
1:
return PORT_TYPE_VECTOR_2D
2:
return PORT_TYPE_VECTOR_3D
_:
return PORT_TYPE_VECTOR_4D
func _get_output_port_count() -> int:
return 1
func _get_output_port_name(port: int) -> String:
return "out"
func _get_output_port_type(port: int) -> VisualShaderNode.PortType:
var vector_index: int = get_option_index(0)
match vector_index:
0:
return PORT_TYPE_SCALAR
1:
return PORT_TYPE_VECTOR_2D
2:
return PORT_TYPE_VECTOR_3D
_:
return PORT_TYPE_VECTOR_4D
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 ["Vector1", "Vector2", "Vector3", "Vector4"]
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 input: String
var vector_index: int = get_option_index(0)
match vector_index:
0:
input = "0.0"
1:
input = "vec2(0.0)"
2:
input = "vec3(0.0)"
_:
input = "vec4(0.0)"
if input_vars[0]:
input = input_vars[0]
match vector_index:
0:
return output_vars[0] + " = square_wave(vec4(%s)).x;" % [input]
1:
return output_vars[0] + " = square_wave(vec4(%s, 0.0, 0.0)).xy;" % [input]
2:
return output_vars[0] + " = square_wave(vec4(%s, 0.0)).xyz;" % [input]
_:
return output_vars[0] + " = square_wave(%s);" % [input]