#! python3
import 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"}
def classify(g):
    if isinstance(g, rg.PolyCurve):
        segs = [g.SegmentCurve(i) for i in range(g.SegmentCount)]
    elif isinstance(g, rg.PolylineCurve):
        ok, pl = g.TryGetPolyline()
        segs = [rg.LineCurve(pl[i], pl[i+1]) for i in range(pl.Count-1)] if ok else [g]
    else:
        segs = [g]
    nl = sum(1 for s in segs if s.IsLinear()); na = len(segs) - nl
    return segs, nl, na
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, ""):
                at = o.Attributes
                nm = at.Name
                base = nm
                d[(at.GetUserString("host") or "", at.GetUserString("mark") or "")].append((o, g))
        out = []
        for (host, m), lst in sorted(d.items()):
            lst.sort(key=lambda t: t[0].Attributes.Name)
            o, g = lst[len(lst)//2]
            segs, nl, na = classify(g)
            bb = g.GetBoundingBox(True)
            dims = sorted([bb.Max.X-bb.Min.X, bb.Max.Y-bb.Min.Y, bb.Max.Z-bb.Min.Z], reverse=True)
            closed = g.IsClosed
            lens = [round(s.GetLength()) for s in segs if s.IsLinear()]
            at = o.Attributes
            names = set(x.Attributes.Name.rsplit("-", 1)[0] if x.Attributes.Name.endswith(("뚜껑","바닥","좌","우")) else x.Attributes.Name for x, _ in lst)
            pts = []
            for sg in segs:
                if sg.IsLinear():
                    pts.append([sg.PointAtStart.X, sg.PointAtStart.Y, sg.PointAtStart.Z]); pts.append([sg.PointAtEnd.X, sg.PointAtEnd.Y, sg.PointAtEnd.Z])
                else:
                    for i in range(9):
                        q = sg.PointAt(sg.Domain.ParameterAt(i / 8.0)); pts.append([q.X, q.Y, q.Z])
            out.append({"pts": [[round(v, 1) for v in p] for p in pts], "host": host, "mark": m, "n_obj": len(lst), "shape": at.GetUserString("shape"), "dia": at.GetUserString("dia"), "design_len": at.GetUserString("design_length"),
                        "len": round(g.GetLength(), 1), "nlin": nl, "narc": na, "closed": closed, "dims": [round(x) for x in dims], "lin_lens": lens,
                        "pieces": sum(1 for x, _ in lst if x.Attributes.Name.endswith(("뚜껑","바닥","좌","우")))})
        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_shape_audit2.json", "w", encoding="utf-8"), ensure_ascii=False, indent=1)