Trình lặp tùy chỉnh

Python khiến một lớp hoạt động như một danh sách bằng các phương thức đặc biệt như __len__, __getitem__, __contains__ và __iter__. Godot chỉ có các phương thức ảo cho một trong những chức năng này, đó là vòng lặp for. Ba phương thức giúp for item in my_object: hoạt động:

• _iter_init(iter) chạy một lần, trước vòng lặp đầu tiên. Phương thức này đặt vị trí bắt đầu và trả về true nếu có phần tử đầu tiên.
• _iter_get(position) trả về phần tử ở vị trí hiện tại, và biến vòng lặp nhận giá trị đó.
• _iter_next(iter) chạy trước mỗi vòng lặp tiếp theo. Phương thức này chuyển đến vị trí kế tiếp và trả về false khi không còn phần tử nào, khiến vòng lặp kết thúc.

# team.gd
class_name Team
extends RefCounted

var _members: Array[String] = []

func add(member: String) -> void:
	_members.append(member)

func _iter_init(iter: Array) -> bool:
	iter[0] = 0
	return iter[0] < _members.size()

func _iter_next(iter: Array) -> bool:
	iter[0] += 1
	return iter[0] < _members.size()

func _iter_get(position):
	return _members[position]

iter là một mảng có một ô mà Godot tạo ra cho mỗi vòng lặp. iter[0] lưu vị trí của vòng lặp đó, và Godot truyền giá trị này cho _iter_get(). Giờ đây, Team có thể hoạt động trong vòng lặp for:

var team := Team.new()
team.add("Ada")
team.add("Bo")
for member in team:
	print(member) # Ada, rồi đến Bo

Hãy giữ vị trí trong iter[0], không phải trong một biến thành viên như _position. Mỗi vòng lặp có iter riêng, nên hai vòng lặp trên cùng một đội, kể cả khi vòng này nằm bên trong vòng kia, sẽ không ảnh hưởng đến nhau. Nếu dùng biến thành viên, vòng lặp bên trong sẽ làm thay đổi vị trí của vòng lặp bên ngoài.

Các thao tác khác của danh sách không có phương thức ảo: len(team) và team[0] sẽ gây lỗi. Hãy viết các phương thức thông thường có tên giống như của Array, chẳng hạn size(), get_at(index) và has(member). Đừng kiểm tra phần tử có nằm trong danh sách hay không bằng in: "Ada" in team kiểm tra xem Team có thuộc tính hoặc phương thức tên là Ada hay không, vì vậy kết quả là false.