Files
CMPlanning/3d-drawing/scripts/extract_pattern_stations.py

29 lines
1.1 KiB
Python

#!/usr/bin/env python3
"""Task 5: support-pattern (지보패턴) station ranges from drawings 03-003..008.
Collects every TEXT that names a station (STA. / N+mmm / N+mmm.mm) and every
pattern name (NP-...) with its layer and position, without interpreting ranges."""
import glob
import os
import re
import sys
import ezdxf
RAG = sys.argv[1]
pat_re = re.compile(r"NP-\d[A-Z0-9-]*|WL-\d")
sta_re = re.compile(r"(STA\.?\s*)?\d{1,2}\+\d{2,3}(\.\d{2})?")
files = sorted(glob.glob(os.path.join(RAG, "03_본선터널굴착", "C0060303-00[3-8]*.dxf")))
for f in files:
msp = ezdxf.readfile(f).modelspace()
stas, pats = [], set()
for e in msp.query("TEXT MTEXT"):
s = e.dxf.text if e.dxftype() == "TEXT" else e.plain_text()
for m in sta_re.finditer(s):
stas.append((round(e.dxf.insert.x), round(e.dxf.insert.y), m.group(0).strip()))
for m in pat_re.finditer(s):
pats.add(m.group(0))
print(os.path.basename(f)[:28], "| patterns:", sorted(pats))
for x, y, t in sorted(stas, key=lambda r: (r[0], r[1]))[:12]:
print(f" ({x},{y}) {t}")
print(f" station-like texts: {len(stas)}")