Files
Aislo/B05_Profile/B05_Profile_Plus_Stations.py
T
eomsangdonandClaude Opus 5.5 39adebdaf5 refactor(472-4 ① 추가 측점 따라감): 서버 추가 측점 따라감도 TS 한 벌 — followPlusRows · Node 입구 facility.plus_follow
- common_util_route_follow.ts followPlusRows(땅 위 자리 → 새 노선 정수 m · 같은 m 하나) · followPlusStations 가 그것을 씀
- Plus_Stations.follow_plus_stations = run_calc 겉(옮긴 수 · 쓰기 그대로) · follow_structures 끌림 걷음
- 등록부 node:facility.plus_follow · NODE_IDS · 시험 test_472_4_follow_node(옛 코드 빨강) · 368_2 · 459 통과

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-19 09:37:00 +09:00

96 lines
3.7 KiB
Python

"""추가 측점 정본(`B05_Profile/route/plus_stations.json`) — 구조물 정본 곁(368-2 · 요구 240-21).
사용자가 종단 원지반선에서 넣은 일반 측점(정수 m) · 이름 없음. 자리 정본 = 좌표(x · y) — 노선이
바뀌면 관 · 구조물처럼 새 노선에 투영해 누가거리를 다시 매김(TS `followPlusRows` · Node 입구).
측점 계획에는 `resolve_extra_stations` → `plan_stations(plus=)` 로 들어감(TS `planStations` 짝).
"""
from __future__ import annotations
import json
import os
from pathlib import Path
from typing import Any, Sequence
from common_util.common_util_calc_runner import run_calc
from common_util.common_util_json import atomic_write_json
PLUS_STATIONS_FILE_NAME = "plus_stations.json"
def plus_stations_path(project_root: str | Path) -> Path:
return Path(project_root) / "B05_Profile" / "route" / PLUS_STATIONS_FILE_NAME
def load_plus_stations(project_root: str | Path) -> list[dict[str, Any]]:
"""[{chainage_m, x, y}] — 없거나 깨졌으면 빈 목록."""
path = plus_stations_path(project_root)
if not path.is_file():
return []
try:
rows = json.loads(path.read_text(encoding="utf-8")).get("stations") or []
except (OSError, ValueError, AttributeError):
return []
return [
row
for row in rows
if isinstance(row, dict) and isinstance(row.get("chainage_m"), (int, float))
]
def plus_chainages(project_root: str | Path) -> list[float]:
return sorted({float(row["chainage_m"]) for row in load_plus_stations(project_root)})
def save_plus_stations(
project_root: str | Path, chainages: Sequence[float], xy: Sequence[Sequence[float]] | None
) -> list[float]:
"""정수 m 로 맞춰(0 ~ 노선 끝 · 같은 m 하나) 좌표와 함께 씀 → 쓴 누가거리."""
from shapely.geometry import LineString
line = LineString([(float(p[0]), float(p[1])) for p in xy]) if xy and len(xy) > 1 else None
end = float(line.length) if line is not None else float("inf")
values = sorted({float(round(c)) for c in chainages if 0 <= round(c) <= end})
rows = []
for value in values:
point = line.interpolate(value) if line is not None else None
rows.append(
{
"chainage_m": value,
"x": round(float(point.x), 4) if point is not None else None,
"y": round(float(point.y), 4) if point is not None else None,
}
)
path = plus_stations_path(project_root)
if not rows and not path.exists():
return []
os.makedirs(path.parent, exist_ok=True)
atomic_write_json(str(path), {"stations": rows})
return values
def follow_plus_stations(project_root: str | Path, new_xy: Sequence[Sequence[float]]) -> int:
"""노선이 바뀔 때 좌표를 새 노선에 투영해 누가거리를 다시 매김 → 옮긴 수(같은 m 는 하나).
셈 = TS `followPlusRows`(Node 입구 `facility.plus_follow` · 472-4 ①)."""
rows = load_plus_stations(project_root)
if not rows:
return 0
results = run_calc(
["facility.plus_follow"],
{
"plus": {
"rows": [
{"chainage_m": float(r["chainage_m"]), "x": r.get("x"), "y": r.get("y")}
for r in rows
],
"line": [[float(p[0]), float(p[1])] for p in new_xy],
}
},
)
if results is None:
raise RuntimeError("추가 측점 따라감 실패 — 계산 Node 입구(facility.plus_follow)")
followed = results["facility.plus_follow"]
if followed["moved"]:
atomic_write_json(str(plus_stations_path(project_root)), {"stations": followed["rows"]})
return int(followed["moved"])