Cast selected Nodes to BrushStroke Nodes in typed array

mbrlabs · Sep 11, 2024 · 112ff31 · 112ff31
1 parent 1419bb2
commit 112ff31
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lorien/InfiniteCanvas/Tools/SelectionTool.gd b/lorien/InfiniteCanvas/Tools/SelectionTool.gd
@@ -239,8 +239,14 @@ func is_selecting() -> bool:
 	return _state == State.SELECTING
 
 # ------------------------------------------------------------------------------------------------
-func get_selected_strokes() -> Array:
-	return get_tree().get_nodes_in_group(GROUP_SELECTED_STROKES)
+func get_selected_strokes() -> Array[BrushStroke]:
+	# Can't cast from Array[Node] to Array[BrushStroke] directly (godot bug/missing feature?)
+	# so let's do it per item
+	var strokes: Array[BrushStroke]
+	for stroke in get_tree().get_nodes_in_group(GROUP_SELECTED_STROKES):
+		strokes.append(stroke as BrushStroke)
+
+	return strokes
 
 # ------------------------------------------------------------------------------------------------
 func _on_brush_color_changed(color: Color) -> void: