mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
fa8c0d9660
Split the oversized file into one per plan type. This also moves the layout responsibility out of the oversized FlightPlanBuilder and into each flight plan type file.
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from shapely.geometry import LineString, Point as ShapelyPoint
|
|
|
|
from game import Game
|
|
from game.ato.flightplans.patrolling import PatrollingFlightPlan
|
|
from game.ato.flightplans.cas import CasFlightPlan
|
|
from game.server import GameContext
|
|
from game.server.flights.models import FlightJs
|
|
from game.server.leaflet import LeafletPoly, ShapelyUtil
|
|
|
|
router: APIRouter = APIRouter(prefix="/flights")
|
|
|
|
|
|
@router.get("/", operation_id="list_flights", response_model=list[FlightJs])
|
|
def list_flights(
|
|
with_waypoints: bool = False, game: Game = Depends(GameContext.require)
|
|
) -> list[FlightJs]:
|
|
return FlightJs.all_in_game(game, with_waypoints)
|
|
|
|
|
|
@router.get("/{flight_id}", operation_id="get_flight_by_id", response_model=FlightJs)
|
|
def get_flight(
|
|
flight_id: UUID,
|
|
with_waypoints: bool = False,
|
|
game: Game = Depends(GameContext.require),
|
|
) -> FlightJs:
|
|
flight = game.db.flights.get(flight_id)
|
|
return FlightJs.for_flight(flight, with_waypoints)
|
|
|
|
|
|
@router.get(
|
|
"/{flight_id}/commit-boundary",
|
|
operation_id="get_commit_boundary_for_flight",
|
|
response_model=LeafletPoly,
|
|
)
|
|
def commit_boundary(
|
|
flight_id: UUID, game: Game = Depends(GameContext.require)
|
|
) -> LeafletPoly:
|
|
flight = game.db.flights.get(flight_id)
|
|
if not isinstance(flight.flight_plan, PatrollingFlightPlan):
|
|
return []
|
|
start = flight.flight_plan.patrol_start
|
|
end = flight.flight_plan.patrol_end
|
|
if isinstance(flight.flight_plan, CasFlightPlan):
|
|
center = flight.flight_plan.target.position
|
|
commit_center = ShapelyPoint(center.x, center.y)
|
|
else:
|
|
commit_center = LineString(
|
|
[
|
|
ShapelyPoint(start.x, start.y),
|
|
ShapelyPoint(end.x, end.y),
|
|
]
|
|
)
|
|
bubble = commit_center.buffer(flight.flight_plan.engagement_distance.meters)
|
|
return ShapelyUtil.poly_to_leaflet(bubble, game.theater)
|