mirror of
https://gitea.wildfiregames.com/0ad/0ad
synced 2026-06-16 05:13:58 -07:00
e36c6a31fe
In the ruff config file added in #6954 explicitly selecting the ruff rules to check was missed, resulting in ruff only checking a very small subset of its available rules. That hasn't been desired, so this is the first of a series of commits enabling more rules. In this PR all rules whose violations can be either automatically fixed by ruff or are trivial to fix manually get enabled. For the follow up PRs it's intended to focus on one area of rules per PR to gradually improve the Python code quality.
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
# This script provides an overview of the zero_ad wrapper for 0 AD
|
|
# First, we will define some helper functions we will use later.
|
|
import math
|
|
from os import path
|
|
|
|
import zero_ad
|
|
|
|
|
|
def dist(p1, p2):
|
|
return math.sqrt(sum(math.pow(x2 - x1, 2) for (x1, x2) in zip(p1, p2)))
|
|
|
|
|
|
def center(units):
|
|
sum_position = map(sum, zip(*(u.position() for u in units)))
|
|
return [x / len(units) for x in sum_position]
|
|
|
|
|
|
def closest(units, position):
|
|
dists = (dist(unit.position(), position) for unit in units)
|
|
index = 0
|
|
min_dist = next(dists)
|
|
for i, d in enumerate(dists):
|
|
if d < min_dist:
|
|
index = i
|
|
min_dist = d
|
|
|
|
return units[index]
|
|
|
|
|
|
# Connect to a 0 AD game server listening at localhost:6000
|
|
game = zero_ad.ZeroAD("http://localhost:6000")
|
|
|
|
# Load the Arcadia map
|
|
samples_dir = path.dirname(path.realpath(__file__))
|
|
scenario_config_path = path.join(samples_dir, "arcadia.json")
|
|
with open(scenario_config_path, encoding="utf8") as f:
|
|
arcadia_config = f.read()
|
|
|
|
state = game.reset(arcadia_config)
|
|
|
|
# The game is paused and will only progress upon calling "step"
|
|
state = game.step()
|
|
|
|
# Units can be queried from the game state
|
|
citizen_soldiers = state.units(owner=1, type="infantry")
|
|
# (including gaia units like trees or other resources)
|
|
nearby_tree = closest(state.units(owner=0, type="tree"), center(citizen_soldiers))
|
|
|
|
# Action commands can be created using zero_ad.actions
|
|
collect_wood = zero_ad.actions.gather(citizen_soldiers, nearby_tree)
|
|
|
|
female_citizens = state.units(owner=1, type="female_citizen")
|
|
house_tpl = "structures/spart/house"
|
|
x = 680
|
|
z = 640
|
|
build_house = zero_ad.actions.construct(female_citizens, house_tpl, x, z, autocontinue=True)
|
|
|
|
# These commands can then be applied to the game in a `step` command
|
|
state = game.step([collect_wood, build_house])
|
|
|
|
# We can also fetch units by id using the `unit` function on the game state
|
|
female_id = female_citizens[0].id()
|
|
female_citizen = state.unit(female_id)
|
|
|
|
# A variety of unit information can be queried from the unit:
|
|
print("female citizen's max health is", female_citizen.max_health())
|
|
|
|
# Raw data for units and game states are available via the data attribute
|
|
print(female_citizen.data)
|
|
|
|
# Units can be built using the "train action"
|
|
civic_center = state.units(owner=1, type="civil_centre")[0]
|
|
spearman_type = "units/spart/infantry_spearman_b"
|
|
train_spearmen = zero_ad.actions.train([civic_center], spearman_type)
|
|
|
|
state = game.step([train_spearmen])
|
|
|
|
|
|
# Let's step the engine until the house has been built
|
|
def is_unit_busy(state, unit_id):
|
|
return len(state.unit(unit_id).data["unitAIOrderData"]) > 0
|
|
|
|
|
|
while is_unit_busy(state, female_id):
|
|
state = game.step()
|
|
|
|
# The units for the other army can also be controlled
|
|
enemy_units = state.units(owner=2)
|
|
walk = zero_ad.actions.walk(enemy_units, *civic_center.position())
|
|
game.step([walk], player=[2])
|
|
|
|
# Step the game engine a bit to give them some time to walk
|
|
for _ in range(150):
|
|
state = game.step()
|
|
|
|
# Let's attack with our entire military
|
|
state = game.step([zero_ad.actions.chat("An attack is coming!")])
|
|
|
|
while len(state.units(owner=2, type="unit")) > 0:
|
|
attack_units = [
|
|
unit for unit in state.units(owner=1, type="unit") if "female" not in unit.type()
|
|
]
|
|
target = closest(state.units(owner=2, type="unit"), center(attack_units))
|
|
state = game.step([zero_ad.actions.attack(attack_units, target)])
|
|
|
|
while state.unit(target.id()):
|
|
state = game.step()
|
|
|
|
game.step([zero_ad.actions.chat("The enemies have been vanquished. Our home is safe again.")])
|