mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
5af4e56f30
This PR: - Introduces a new member of UnitType, hit_points, which is an abstract representation of the durability of a unit, and loads it in from the YAML files in the various subclasses (Ship, Ground etc). - Adds scripts for populating/updating the unit YAML files with hit point data from DCS. This script also gets the data for static objects, but I'll leave the plugging in of static object data into Liberation for another PR. - Updates the unit YAML files by running the above scripts. I did toy with the idea of adding this data to the unit definitions in pydcs via an export from DCS, but it would be a more involved change, since the current pydcs export script runs in the Hooks Lua environment in DCS and AFAICT the hit points (via Unit.getLife()) is run in the mission scripting environment.
126 lines
3.6 KiB
Python
126 lines
3.6 KiB
Python
import copy
|
|
import os
|
|
import typing
|
|
|
|
from dcs import (
|
|
action,
|
|
countries,
|
|
helicopters,
|
|
planes,
|
|
ships,
|
|
statics,
|
|
triggers,
|
|
unittype,
|
|
vehicles,
|
|
)
|
|
from dcs.country import Country
|
|
from dcs.mapping import Point, Vector2
|
|
from dcs.mission import Mission
|
|
from dcs.terrain.caucasus import Caucasus
|
|
from dcs.translation import String, Translation
|
|
|
|
|
|
def add_units(
|
|
unit_class: str,
|
|
unit_map: dict[str, typing.Any],
|
|
origin: Point,
|
|
spacing: Vector2 = Vector2(0, 1000),
|
|
country_name: str = "USA",
|
|
) -> typing.Generator[dict[str, typing.Any], None, None]:
|
|
position = copy.deepcopy(origin)
|
|
country = mission.country(country_name)
|
|
for unit_name in unit_map.keys():
|
|
unit_type = unit_map[unit_name]
|
|
position += spacing
|
|
yield {
|
|
"name": unit_class + "," + unit_name,
|
|
"_type": unit_type,
|
|
"position": position,
|
|
"country": country,
|
|
}
|
|
position = copy.deepcopy(position)
|
|
|
|
|
|
def handle_flying_unit(unit: dict[str, typing.Any], altitude: int) -> None:
|
|
unit["altitude"] = altitude
|
|
unit["aircraft_type"] = unit["_type"]
|
|
unit.pop("_type")
|
|
|
|
|
|
def setup_export_trigger(
|
|
mission: Mission, output_filename: str, script_file: str = "export.lua"
|
|
) -> None:
|
|
trigger_rule = triggers.TriggerStart(comment="Run export script")
|
|
script = f"""
|
|
local output_file = io.open('{output_filename}', 'w')
|
|
for i, group in pairs(coalition.getGroups(2)) do
|
|
for j, unit in pairs(group:getUnits()) do
|
|
output_file:write(group:getName(), ',', unit:getLife(), '\\n')
|
|
end
|
|
end
|
|
for i, static_object in pairs(coalition.getStaticObjects(2)) do
|
|
output_file:write(static_object:getName(), ',', static_object:getLife(), '\\n')
|
|
end
|
|
output_file:close()
|
|
"""
|
|
translation = Translation(mission)
|
|
script_string = String(_id=script, translation=translation)
|
|
script_string.set(script)
|
|
trigger_rule.add_action(action.DoScript(script_string))
|
|
mission.triggerrules.triggers.append(trigger_rule)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
output_path = os.path.dirname(__file__)
|
|
miz_output = os.path.join(output_path, "hit_points_generator.miz")
|
|
data_output = os.path.join(output_path, "hit_points_data.csv")
|
|
|
|
terrain = Caucasus()
|
|
|
|
mission = Mission(terrain=terrain)
|
|
mission.filename = miz_output
|
|
|
|
# Add ships
|
|
for unit in add_units(
|
|
"ship", ships.ship_map, origin=Point(-200000, -300000, terrain)
|
|
):
|
|
mission.ship_group(**unit)
|
|
|
|
# Add statics
|
|
for unit in add_units(
|
|
"static", statics.fortification_map, origin=Point(-500000, -300000, terrain)
|
|
):
|
|
mission.static_group(**unit)
|
|
for unit in add_units(
|
|
"static", statics.warehouse_map, origin=Point(-501000, -300000, terrain)
|
|
):
|
|
mission.static_group(**unit)
|
|
|
|
# Add vehicles
|
|
for unit in add_units(
|
|
"vehicle", vehicles.vehicle_map, origin=Point(-502000, -300000, terrain)
|
|
):
|
|
mission.vehicle_group(**unit)
|
|
|
|
# Add helicopters
|
|
for unit in add_units(
|
|
"helicopter",
|
|
helicopters.helicopter_map,
|
|
origin=Point(-210000, -300000, terrain),
|
|
):
|
|
handle_flying_unit(unit, altitude=5000)
|
|
mission.flight_group_inflight(**unit)
|
|
|
|
# Add planes
|
|
for unit in add_units(
|
|
"plane", planes.plane_map, origin=Point(-190000, -300000, terrain)
|
|
):
|
|
if unit["name"].split(",")[1] in ["Mirage-F1JA"]: # skip problematic units
|
|
continue
|
|
handle_flying_unit(unit, altitude=10000)
|
|
mission.flight_group_inflight(**unit)
|
|
|
|
setup_export_trigger(mission, output_filename=data_output.replace(os.sep, "//"))
|
|
|
|
mission.save()
|