import argparse

import json

import math

import sys

import time

from pathlib import Path

import cadsupport_path # noqa: F401

from cadsupport.occ_env import ensure_occ # noqa: E402

ensure_occ()

from OCC.Core.BRep import BRep_Tool # noqa: E402

from OCC.Core.BRepAdaptor import BRepAdaptor_Surface # noqa: E402

from OCC.Core.BRepAlgoAPI import BRepAlgoAPI_Cut, BRepAlgoAPI_Fuse # noqa: E402

from OCC.Core.BRepPrimAPI import (BRepPrimAPI_MakeBox, BRepPrimAPI_MakeCone, # noqa: E402

BRepPrimAPI_MakeCylinder,

BRepPrimAPI_MakeSphere, BRepPrimAPI_MakeTorus)

from OCC.Core.Geom import (Geom_RectangularTrimmedSurface, Geom_SurfaceOfLinearExtrusion, # noqa: E402

Geom_SurfaceOfRevolution)

from OCC.Core.GeomAbs import GeomAbs_BSplineSurface # noqa: E402

from OCC.Core.GeomAdaptor import GeomAdaptor_Curve # noqa: E402

from OCC.Core.ShapeAnalysis import ShapeAnalysis_CanonicalRecognition # noqa: E402

from OCC.Core.STEPCAFControl import STEPCAFControl_Reader # noqa: E402

from OCC.Core.IFSelect import IFSelect_RetDone # noqa: E402

from OCC.Core.TCollection import TCollection_AsciiString # noqa: E402

from OCC.Core.TDF import TDF_Label, TDF_LabelSequence, TDF_Tool # noqa: E402

from OCC.Core.TDocStd import TDocStd_Document # noqa: E402

from OCC.Core.TopAbs import TopAbs_FACE, TopAbs_REVERSED, TopAbs_SOLID # noqa: E402

from OCC.Core.TopExp import TopExp_Explorer, topexp # noqa: E402

from OCC.Core.TopTools import TopTools_IndexedMapOfShape # noqa: E402

from OCC.Core.TopoDS import topods # noqa: E402

from OCC.Core.XCAFDoc import XCAFDoc_DocumentTool # noqa: E402

from OCC.Core.gp import gp_Dir, gp_Vec # noqa: E402

from cadsupport.analytic import CURVE_TYPE_NAME, SURFACE_TYPE_NAME # noqa: E402

from cadsupport.census import _xyz, bounding_box, edge_census, halfspace_side, volume_of # noqa: E402

from cadsupport.primitives import _cross, _dot, _norm, _sub # noqa: E402

CENSUS_FORMAT_VERSION = 3

QUADRIC_TYPES = ("plane", "cylinder", "cone", "sphere", "torus")

TEMPLATE_REL_TOL = 1.0e-6

TEMPLATE_ANG_TOL = 1.0e-6

def _parallel(a, b, tol=TEMPLATE_ANG_TOL):

return _norm(_cross(a, b)) <= tol

def _antiparallel(a, b, tol=TEMPLATE_ANG_TOL):

return _parallel(a, b, tol) and _dot(a, b) < 0

def _perp(a, b, tol=TEMPLATE_ANG_TOL):

return abs(_dot(a, b)) <= tol

def _point_on_axis(p, loc, direction, tol):

d = _sub(p, loc)

perp = _sub(d, tuple(c * _dot(d, direction) for c in direction))

return _norm(perp) <= tol

def _basis_curve_type(surface, stype):

return "other"

def _canonical_recognition(face, tol):

return None, None

def classify_face(face, canonical_tol, do_canonical=True):

return info

def _carriers(faces):

return out

def _scale_of(carriers, bbox_diag):

return max(bbox_diag, 1.0)

def distinct_carriers(carriers, scale):

return len(uniq)

def carrier_clusters(carriers, scale):

"nAxisClusters": len(out), "nPlaneDirections": len(normals)}

def match_box(carriers, scale):

"dz": dims[2] / 2}}

def _coaxial(items, scale):

return d0, p0

def match_tube_or_cone(carriers, scale):

return {"template": name, "params": params}

def match_sphere(carriers, scale):

return {"template": "TGeoSphere", "params": {"r": sph[0]["r"], "cuts": len(planes)}}

def match_torus(carriers, scale):

"cuts": len(planes)}}

def match_revolution(carriers, scale, faces_info):

"params": {"nlateral": len(axial), "phiCut": nwedge == 2}}

def match_extrusion(carriers, scale, faces_info):

return None

def tier2_sketch(clusters):

return f"{na} axis clusters ({sizes}), {np_} plane directions"

def match_template(faces, faces_info, scale):

return {"template": "none", "params": {}}

def solid_faces(solid):

fmap = TopTools_IndexedMapOfShape()

topexp.MapShapes(solid, TopAbs_FACE, fmap)

return [topods.Face(fmap.FindKey(i)) for i in range(1, fmap.Size() + 1)]

def census_solid(solid, name, canonical_tol, do_canonical=True, carrier_face_cap=400):

return rec

def load_step_solids(path):

return out

def detect_unit_scale_to_cm(path):

return 0.1, "mm"

def census_model(path, canonical_tol, do_canonical=True, max_faces=None, progress=True):

}

def self_test(verbose=True):