mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
9a00a60cd4
* Initial support of Germany Cold War terrain * Updated with latest beacon import * Updated changelog and ran black * Fixed the GermanyCW landmap and naming scheme * Temporarily changed the requirements for pydcs to my branch (will switch back when merged) * Updated landmap for germanycw with even fewer exclusions * Removed unnecessary changes * Updated requirements to lastest pydcs * Fixed naming issues for Germany * Update pydcs to latest commit for GCW support * Add landmap to game.theater.__init__ * Update requirements.txt * Resolve PyCharm type-waring * Updated pydcs and exclusion of germany * Updated land and sea shapefiles for Germany Cold War with new map limits --------- Co-authored-by: Raffson <Raffson@users.noreply.github.com>
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Union
|
|
|
|
from dcs import Point
|
|
from dcs.mapping import LatLng
|
|
from pydantic import BaseModel
|
|
from shapely.geometry import LineString, MultiLineString, MultiPolygon, Polygon
|
|
|
|
from game.theater import ConflictTheater
|
|
|
|
|
|
class LeafletPoint(BaseModel):
|
|
lat: float
|
|
lng: float
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
title = "LatLng"
|
|
|
|
@staticmethod
|
|
def from_latlng(latlng: LatLng) -> LeafletPoint:
|
|
return LeafletPoint(lat=latlng.lat, lng=latlng.lng)
|
|
|
|
|
|
LeafletLine = list[LeafletPoint]
|
|
|
|
|
|
# Leaflet allows either a single array of latlng to define the boundary, or an array of
|
|
# arrays of latlngs, with the first array being the boundary and the rest defining
|
|
# holes. To avoid the UI needing to test for which type of polygon we send, we always
|
|
# use the array of arrays type, even for geometries without holes.
|
|
# https://leafletjs.com/reference.html#polygon
|
|
LeafletPoly = list[LeafletLine]
|
|
|
|
|
|
class ShapelyUtil:
|
|
@staticmethod
|
|
def latlng_to_leaflet(latlng: LatLng) -> LeafletPoint:
|
|
return LeafletPoint(lat=latlng.lat, lng=latlng.lng)
|
|
|
|
@classmethod
|
|
def poly_to_leaflet(cls, poly: Polygon, theater: ConflictTheater) -> LeafletPoly:
|
|
if poly.is_empty:
|
|
return []
|
|
try:
|
|
lines = poly.boundary.geoms
|
|
except AttributeError:
|
|
lines = [poly.boundary]
|
|
return cls.lines_to_leaflet(MultiLineString(lines), theater)
|
|
|
|
@classmethod
|
|
def polys_to_leaflet(
|
|
cls, poly: Union[Polygon, MultiPolygon], theater: ConflictTheater
|
|
) -> list[LeafletPoly]:
|
|
if isinstance(poly, MultiPolygon):
|
|
polys = poly.geoms
|
|
else:
|
|
polys = [poly]
|
|
return [cls.poly_to_leaflet(poly, theater) for poly in polys]
|
|
|
|
@classmethod
|
|
def line_to_leaflet(cls, line: LineString, theater: ConflictTheater) -> LeafletLine:
|
|
return [
|
|
cls.latlng_to_leaflet(Point(x, y, theater.terrain).latlng())
|
|
for x, y in line.coords
|
|
]
|
|
|
|
@classmethod
|
|
def lines_to_leaflet(
|
|
cls, line_string: MultiLineString | LineString, theater: ConflictTheater
|
|
) -> list[LeafletLine]:
|
|
if isinstance(line_string, MultiLineString):
|
|
lines = line_string.geoms
|
|
else:
|
|
lines = [line_string]
|
|
return [cls.line_to_leaflet(line, theater) for line in lines]
|