#! python3
import os, json, traceback, collections, math
import Rhino
import Rhino.Geometry as rg
B = r"D:\★Pier 자료\프로그램\Pier_T\검토파일"
M = {"운곡": B + r"\철근 모델랑 운곡교P1_바이브코딩\운곡교p1_수정7_20261026.3dm",
     "산외": B + r"\검토_산외교P1\산외교P1 G5_수정5_20261027.3dm",
     "옥성": B + r"\검토_옥성교P1\옥성교P1_G5_수정6_20261026.3dm",
     "RampA": B + r"\Ramp-A교P1\ramp-A G5_수정6_20261026.3dm"}
res = {}
try:
    for k, p in M.items():
        doc = Rhino.RhinoDoc.OpenHeadless(p)
        st = Rhino.DocObjects.ObjectEnumeratorSettings(); st.HiddenObjects = True; st.LockedObjects = True; st.NormalObjects = True
        lay = {l.Index: l.FullPath for l in doc.Layers}
        d = collections.defaultdict(list)
        for o in doc.Objects.GetObjectList(st):
            g = o.Geometry
            if isinstance(g, rg.Curve) and "중심선" in lay.get(o.Attributes.LayerIndex, ""):
                m = o.Attributes.GetUserString("mark")
                if m in ("B1", "B2", "B3", "B4", "B5", "B6"):
                    d[m].append((o, g))
        out = {}
        for m, lst in sorted(d.items()):
            o, g = lst[len(lst)//2]
            segs = []
            if isinstance(g, rg.PolyCurve):
                for i in range(g.SegmentCount):
                    s = g.SegmentCurve(i)
                    segs.append(("L" if s.IsLinear() else "A", round(s.GetLength(), 1), [round(x) for x in (s.PointAtStart.X, s.PointAtStart.Y, s.PointAtStart.Z)], [round(x) for x in (s.PointAtEnd.X, s.PointAtEnd.Y, s.PointAtEnd.Z)]))
            else:
                segs.append((type(g).__name__, round(g.GetLength(), 1), [round(x) for x in (g.PointAtStart.X, g.PointAtStart.Y, g.PointAtStart.Z)], [round(x) for x in (g.PointAtEnd.X, g.PointAtEnd.Y, g.PointAtEnd.Z)]))
            out[m] = {"n": len(lst), "name": o.Attributes.Name, "dia": o.Attributes.GetUserString("dia"), "shape": o.Attributes.GetUserString("shape"), "design_length": o.Attributes.GetUserString("design_length"), "len": round(g.GetLength(), 1), "segs": segs}
        res[k] = out
        doc.Dispose()
except Exception:
    res["error"] = traceback.format_exc()
json.dump(res, open(r"C:\Users\User\AppData\Local\PierApp\jobs\c1_work\c1_dump_b1b4.json", "w", encoding="utf-8"), ensure_ascii=False, indent=1)