^j.?^dZddlZddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?ddl@mAZAmBZBmCZCmDZDmEZEmFZFddlGmHZHmIZIddlJmKZKmLZLmMZMmNZNmOZOmPZPddlQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[e/eRe0eSe1eTe3eWe4eXe5eYe6eZiZ\dee]ze^zdeUfdZ_dee]ze^zdeRfd Z`eIjfdee]ze^zd!eIdeTfd"ZaeKfdee]ze^zd#ebdece]e]ffd$Zde d%eHjd&d'd(d)e]eze zd*efd+eHeceHeHfzdzd,efd-e d.e]zdeVeZeTzf d/Zge ed0d%eHjd&d'dd1d)e]eze zd*efd+eHeceHeHfzdzd,efd-e d.e]zd2efdzdeVeZeTzfd3Zgd%eHjd&d'dd1d)e]eze zd*efd+eHeceHeHfzdzd,efd-e d.e]zd2efdzdeVeZeTzfd4Zgy)5a build123d imports name: importers.py by: Gumyr date: March 1st, 2023 desc: This python module contains importers from multiple file formats. license: Copyright 2022 Gumyr Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. N)PathLikefsdecode)Path)LiteralTextIOoverload) deprecated)Bnd_Box) BRep_Builder) BRepBndLib) BRepTools)gp_Trsf)Quantity_ColorRGBA)RWStl)STEPCAFControl_Reader)TCollection_AsciiStringTCollection_ExtendedString) TDataStd_Name) TDF_LabelTDF_LabelSequence)TDocStd_Document) TopAbs_FACE)TopExp_Explorer)TopoDS_Compound TopoDS_Edge TopoDS_Face TopoDS_Shape TopoDS_Shell TopoDS_Solid TopoDS_Vertex TopoDS_Wire)XCAFDoc_ColorCurvXCAFDoc_ColorGenXCAFDoc_ColorSurfXCAFDoc_ColorToolXCAFDoc_DocumentTool) ColorAndLabelimport_svg_document)CMFTINMCMMM)AlignUnit) TOL_DIGITS TOLERANCEColorLocationVectorto_align_offset) CompoundEdgeFaceShape ShapeListShellSolidVertexWiredowncast file_namereturnct}t}t|}tj||||j rt d|tj|S)zImport shape from a BREP file Args: file_name (Union[PathLike, str, bytes]): brep file Raises: ValueError: file not found Returns: Shape: build123d object zCould not import ) rr rr Read_sIsNull ValueErrorr7cast)rAshapebuilder file_name_strs L/opt/ringagent/.cad-venv/lib/python3.12/site-packages/build123d/importers.py import_breprLmsY NEnGY'M UM73 ||~,]O<== == filenamecN dtdtfd dtdtdzffd dtdtdzf fd dtdtdzfd  dd tdzdtt f fd t jj|s t|itd }t|}tj|j tj|jt!}|j#d |j%d |j'd |j)t+||j-|t/}|_t3|j0dk(r|j0d}|S)aimport_step Extract shapes from a STEP file and return them as a Compound object. Args: file_name (Union[PathLike, str, bytes]): file path of STEP file to import Raises: ValueError: can't open file Returns: Compound: contents of STEP file labelrBc*d}t}|jtj|r't|j j }dj d|D}|jtjddS)zExtract name and formatc3XK|]"}tj|ddk7s|$yw)rCN) unicodedatacategory).0chs rK z0import_step..get_name..s(UB+2F2Fr2J12MQT2TRUs **z .()____) r FindAttributeGetID_srGet ToCStringjoin translatestr maketrans)rPnamestd_name clean_names rKget_namezimport_step..get_namesp ?   }446 A*8<<>:DDFDWWU$UU ##CMM&&$ABBrMobjNc t|j}|vr|St}j|t|xs0j|t |xsj|t |}|r|nd|<|S)z%Get the color of a shape from a cacheN)hashTShaperGetColorr"r#r$)rgkeycol has_color _color_cache color_tools rKget_shape_color_from_cachez/import_step..get_shape_color_from_caches3::<  , $ $ "   %6 < @""3(8#> @""3(93?  $-C$ SC  rMrHcP |}||Sd}d}t|t}|jrt|j} |}|9t }t j |||j}||kDr|}|}|j|jrt|S)z9Get the color - take that of the largest Face if multipleNg) rrMoreCurrentr r Add_s SquareExtentNext) rH shape_color max_extentwinnerexpfacermboxextentrqs rK get_colorzimport_step..get_colors07  "  e[1hhj;;=D,T2Ci  s+))+J&!'J F HHJhhj rMct}tj|t|s6tj|t|stj|t |r|Sy)z%Get color directly from an XDE label.N)rr% GetColor_sr"r#r$)rPrms rKget_label_colorz$import_step..get_label_colorsK "  ( (0A3 G ++E3CSI ++E3DcJJrMparent_tdf_labelct}|j|nj||g}t|j D]}|j |dz}j |rt}j||n|}tj|}j|rt} ||_ ntt||} |} |}  |} |xs| xs| |_||_|j%t'j)||j+||S)z+Recursively extract object into an assembly)r GetFreeShapesGetComponents_srangeLengthValue IsReference_srGetReferredShape_sr@ GetShape_s IsAssembly_sr7children topods_luttypecolorrPmover4 GetLocation_sappend)rsub_tdf_labels sub_shapesi sub_tdf_label ref_tdf_labelsub_topo_shape sub_shapeinstance_colorreferred_colorshape_fallback_colorbuild_assemblyrrrf shape_tools rKrz#import_step..build_assemblysL*,  #  $ $^ 4  & &'7 H"$ ~,,./ )A*00Q7M'' 6 ) --m]K - %j&;&;M&JKN&&}5$J %3M%B "&tN';<^L -];N,];N#,^#< ,VVBVIO&}5IO NN8J$<$<]$KL M   i (1 )2rMXCAFTrrN)rrarrlistr:ospathexistsFileNotFoundErrorrrr& ShapeTool_sMain ColorTool_sr SetNameMode SetColorMode SetLayerModeReadFilerTransferr7rlen) rNfmtdocreaderrootrorrprrrfrqrs @@@@@@@@rK import_steprsnC CcC ! !9Kd9R !*>( #)):tjt|}|tjk(rntj t tjttj t tjttjttjti} ||}t}|j|td|j!dzD]6}|j#|} | j%||j'|| 8t)} t+j-| |t/j0| S#t$r6}tdtDcgc]}|jncc}wc}|d}~wwxYw)aimport_stl Extract shape from an STL file and return it as a Face reference object. Note that importing with this method and creating a reference is very fast while creating an editable model (with Mesher) may take minutes depending on the size of the STL file. Args: file_name (Union[PathLike, str, bytes]): file path of STL file to import model_unit (Unit, optional): the default unit used when creating the model. For example, Blender defaults to Unit.M. Defaults to Unit.MM. Raises: ValueError: Could not import file ValueError: Invalid model_unit Returns: Face: STL model zmodel_scale must be one of: Nr)r ReadFile_srr0r-r,r)r.r+r*KeyErrorrFrcrSetScaleFactorrNbNodesNode TransformSetNoderr MakeFacer9rG) rArrconversion_factor scale_factorexcunittransformationrpr|s rK import_stlr sA,  hy1 2FTWW  GGR GGR GGR FFA GGR GGR   ,Z8L !%%l3q&..*Q./ !A AA KK ' NN1a  ! =DND&) 99T? .d/Kd /K/K.LM  s$.E F&F4F FF precisionc dtffd dtjdtffd }gdgdgdd}t|}tj|\}}t j j|jd d }|jr|nd }d d |dg}|D]}|D]} t| j} | dk(r|jd|| 8|| ddD cgc]} | j| } } djfd| D} |jd|| d d| ddj||fScc} w)a]translate_to_buildline_code Translate the contents of the given svg file into executable build123d/BuildLine code. Args: file_name (PathLike | str | bytes]): svg file name precision (int): # digits to round values to. Defaults to # digits in TOLERANCE Returns: tuple[str, str]: code, builder instance name rBct|tr'd|jddd|jdddS|ddS)N(0.g, )) isinstancecomplexrealimag)valuers rK fmt_valuez/import_svg_as_buildline_code..fmt_valueSsS eW %uzz"YKq1EJJr)Ao3NaP P9+Q'(rMcurvec|j}|j}|j}t|jj }t|jj }t|j}t|j}t|j}t||z tkrad|j d dd|j d dd|j d dd|j d dd|d dd|j d Sd|j d dd|j d dd|d dd|d dd |d dd |d dd |d ddS) Nz RadiusArc((rrrz), (z), rzEllipticalCenterArc((z, start_angle=z , arc_size=z , rotation=) centerstartendabsradiusrrfloatthetadeltarotationr2 large_arc) rrrrx_radiusy_radius start_anglearc_sizerrs rK arc_to_codez1import_svg_as_buildline_code..arc_to_codeXs iiu||(()u||(()EKK( %( x(" #y 0JJr)Ao.bByk?0KLHHR {!O,Bsxx9+Q.GsByk?+2&'q *   Byk?+2fkk"YKq-I9+Q'r9+Q'(&r)Ao67 I;a01 I;a0  3 rM)Linerr)Bezierrcontrol1control2r)rrcontrolr)r CubicBezierQuadraticBezier.rrIzfrom build123d import *zwith BuildLine() as :Arcz rN,c3.K|] }|ywr)rWvrs rKrYz/import_svg_as_buildline_code..s!?1)A,!?srr )ra svgpathtoolsrr svg2pathsrrbasenamesplit isidentifierr__name__r__dict__r_)rArr translatorpaths_path_attributes builder_namebuildline_coderr class_nameparmvalues values_strrs ` @rKimport_svg_as_buildline_coderCs )C)  <++  <)I@J #I*44Y?E 77##I.44S9!r?r@rrabytesrLrrinttuplerMINboolrrrMrKrJsD> !,,(!%#+4O&0("&   688-   X%%6  8c>E1 e 0D(S.50DXDNFJWW4(S.504d4QU4r I5#~%I5I5 38_I5X 05 #?C !Dj6!! ! 5& & - !  ! 56< !td{! !  @05 #?C%)!Dj6!! ! 5& & - !  ! 56< !d{!td{!  !05 #?C%)ADj6!A A 5& & - A  A 56< Ad{Atd{ArM