64 lines
1.5 KiB
GDScript
64 lines
1.5 KiB
GDScript
tool
|
|
|
|
extends Area2D
|
|
|
|
class_name ElectricSocket
|
|
|
|
enum Direction { LEFT, RIGHT, UP, DOWN }
|
|
enum Flow { SOURCE, SINK, BIDIRECTIONAL }
|
|
|
|
export(Direction) var direction = Direction.DOWN setget set_direction
|
|
|
|
export(Array, NodePath) var connectionPaths = []
|
|
var connections = []
|
|
|
|
export(Color) var source_color
|
|
export(Color) var sink_color
|
|
export(Color) var bidirectional_color
|
|
|
|
export(Flow) var flow = Flow.SINK setget set_flow
|
|
|
|
onready var socket = $socket
|
|
|
|
var network = null
|
|
|
|
func _ready() -> void:
|
|
socket.material = socket.material.duplicate()
|
|
refresh_sprite()
|
|
if Engine.editor_hint:
|
|
return
|
|
for connection in connectionPaths:
|
|
add_connection(get_node(connection))
|
|
|
|
func add_connection(node: Node) -> void:
|
|
var manager = node.get_node("PowerManager") as PowerManager
|
|
manager.socket = self
|
|
connections.append(manager)
|
|
|
|
func set_direction(dir) -> void:
|
|
direction = dir
|
|
refresh_sprite()
|
|
|
|
func set_flow(val) -> void:
|
|
flow = val
|
|
refresh_sprite()
|
|
|
|
func refresh_sprite() -> void:
|
|
if socket == null:
|
|
return
|
|
match direction:
|
|
Direction.DOWN:
|
|
socket.region_rect.position = Vector2(0, 0)
|
|
Direction.UP:
|
|
socket.region_rect.position = Vector2(32, 0)
|
|
Direction.LEFT:
|
|
socket.region_rect.position = Vector2(32, 32)
|
|
Direction.RIGHT:
|
|
socket.region_rect.position = Vector2(0, 32)
|
|
match flow:
|
|
Flow.SOURCE:
|
|
socket.material.set_shader_param("cable_color", source_color)
|
|
Flow.SINK:
|
|
socket.material.set_shader_param("cable_color", sink_color)
|
|
Flow.BIDIRECTIONAL:
|
|
socket.material.set_shader_param("cable_color", bidirectional_color)
|