Files
tgcc/core/game_menu/train_selector.gd
2026-06-27 11:54:01 +00:00

58 lines
2.2 KiB
GDScript

extends VBoxContainer
@onready var color_pickers_1: Array[TrainColorPicker] = [$%ColorPicker1, $%ColorPicker2, $%ColorPicker3, $%ColorPicker4, $%ColorPicker5, $%ColorPicker6]
@onready var color_pickers_2: Array[TrainColorPicker] = [$%ColorPicker7, $%ColorPicker8, $%ColorPicker9, $%ColorPicker10, $%ColorPicker11, $%ColorPicker12]
@onready var treno: Node3D = $TrainViewportContainer/SubViewport/Treno
var selected_color: Color
func _ready() -> void:
for color_picker in color_pickers_1:
color_picker.pressed.connect(_on_color_selected.bind(color_picker, 0))
color_picker.deselect()
var init_idx_1 = GameState.save_data.train_color_1_index
_set_selected_color(init_idx_1, 0)
for color_picker in color_pickers_2:
color_picker.pressed.connect(_on_color_selected.bind(color_picker, 1))
color_picker.deselect()
var init_idx_2 = GameState.save_data.train_color_2_index
_set_selected_color(init_idx_2, 1)
func _process(delta: float) -> void:
if treno:
treno.rotation_degrees.y += 30.0 * delta
func _set_selected_color(index: int, material_index: int) -> void:
var color_pickers = color_pickers_1 if material_index == 0 else color_pickers_2
for i in range(color_pickers.size()):
if i == index:
color_pickers[i].select()
selected_color = color_pickers[i].color
_apply_train_color(selected_color, material_index, index)
else:
color_pickers[i].deselect()
func _on_color_selected(selected_color_picker: TrainColorPicker, material_index: int) -> void:
var color_pickers = color_pickers_1 if material_index == 0 else color_pickers_2
for i in range(color_pickers.size()):
if color_pickers[i] == selected_color_picker:
_set_selected_color(i, material_index)
break
func _apply_train_color(color: Color, material_index: int, color_index: int = 0) -> void:
var path = "res://tgcc/train/Color%s_train.tres" % ("1" if material_index == 0 else "2")
var mat: ShaderMaterial = load(path)
if mat:
mat.set_shader_parameter("albedo_color", color)
if material_index == 0:
GameState.save_data.train_color_1 = color.to_html()
GameState.save_data.train_color_1_index = color_index
else:
GameState.save_data.train_color_2 = color.to_html()
GameState.save_data.train_color_2_index = color_index
GameState.save_game()