^jOdZddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZddlZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0m1Z1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>ddl?m@Z@mAZAmBZBddlCmDZDddlEmFZFddlGmHZHmIZIddlJmKZKdd lLmMZMmNZNmOZOmPZPmQZQeIjd!fd"ePd#eId$eSd%e:fd&ZTd"ePd'eeUzeVzeze zd%eSfd(ZWeIjd!d)d*fd"ePd'eeUzeVzd#eId+eSd,eXd-eXd%eSfd.ZYeIjd/eHjfdd0d"ePd'eeUzeVzeze zd#eId1eSd2eHd3eUezdzd%eSfd4Z[ d:d"ePd'eeUzeVzd5eXd6eXd7eSd%eSf d8Z\eIjd/eHjfd"ePd#eId1eSd2eHd%eUf d9Z]y);a build123d 3D Exporters name: exporters3d.py by: Gumyr date: March 6th, 2024 desc: Exporters for 3D models such as STEP. license: Copyright 2024 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)datetime)BytesIO)PathLikefsdecode)Path)BinaryIOcast) PreOrderIter)APIHeaderSection_MakeHeader)BRepMesh_IncrementalMesh) BRepTools)IFSelect_ReturnStatus)IGESControl_Controller)Interface_Static)MessageMessage_GravityMessage_ProgressRange)RWGltf_CafWriter)STEPCAFControl_ControllerSTEPCAFControl_Writer)STEPControl_ControllerSTEPControl_StepModelType) StlAPI_Writer)TCollection_AsciiStringTCollection_ExtendedStringTCollection_HAsciiString)$TColStd_IndexedDataMapOfStringString) TDataStd_Name) TDF_Label)TDocStd_Document)TopExp_Explorer)XCAFApp_Application)XCAFDoc_ColorTypeXCAFDoc_DocumentToolXCAFDoc_ShapeTool)XSControl_WorkSession)UNITS_PER_METER) PrecisionModeUnit)Location)CompoundCurvePartShapeSketchF to_exportunit auto_namingreturnc ttd}tj}|j td||j |t j|dt|z t j|j t j|j j|i}dtdtfd dtdtdd f fd }t|D]}|j t#|d d }| j%|j d } nO|j'|t}  | } | j)ro j+| |j } | j)r| ||<|j,s |j.|||  j1|S) acreate_xde An OpenCASCADE Technology (OCCT) XDE (eXtended Data Exchange) document is a sophisticated framework designed for comprehensive CAD data management, facilitating not just the geometric modeling of shapes but also the handling of associated metadata, such as material properties, color, layering, and annotations. This integrated environment supports the organization and manipulation of complex assemblies and parts, enabling users to navigate, edit, and query both the structure and attributes of CAD models. XDE's architecture is particularly beneficial for applications requiring detailed interoperability between different CAD systems, offering robust tools for data exchange, visualization, and modification while preserving the integrity and attributes of the CAD data across various stages of product development. Args: to_export (Shape): object or assembly unit (Unit, optional): shape units. Defaults to Unit.MM. auto_naming (bool, optional): whether to use ShapeTool AutoNaming. Defaults to False. Returns: TDocStd_Document: XDE document XmlOcafz MDTV-XCAFlabelr3c|jr|Stj|r2t}tj||r|js|S|S)zReturn a label suitable for assembly operations. If `label` is a reference/component label, this resolves and returns the referred shape label. Otherwise returns `label` unchanged. Null labels are returned as-is. )IsNullr% IsReference_srGetReferredShape_s)r7referreds N/opt/ringagent/.cad-venv/lib/python3.12/site-packages/build123d/exporters3d.pyresolve_component_parent_labelz3_create_xde..resolve_component_parent_label}sK <<>L  * *5 1 {H!44UHE ) node node_labelNc|jry|jrtj|t |jt j |rYt}t j||r9|js)tj|t |j|j@tj} j||jj|t j |rWt}t j||r7|js' j||jj|t|tr|j s}t|t"r5t%|jt&j(}tj*}nt|t,r5t%|jt&j.}tj*}n_t|t0r5t%|jt&j2}tj4}nt%}tj} |}|j7r{|j9} j;||}|js' j||jj||j=|j7rzyyyyy)aAssign label/color metadata for one XDE node label. Behavior: - Sets `TDataStd_Name` on the instance label and, when applicable, also on the referred shape label so STEP PRODUCT names persist. - Sets generic color on the instance label and referred label. - For leaf `Compound` wrappers (`Part`, `Sketch`, `Curve`), propagates color to relevant sub-shapes (solid/face/edge) using appropriate XCAF color channels. N)r9r7rSet_srr%r:rr;colorr#XCAFDoc_ColorGenSetColorwrapped isinstancer+childrenr-r!ta TopAbs_SOLIDXCAFDoc_ColorSurfr/ TopAbs_FACEr, TopAbs_EDGEXCAFDoc_ColorCurvMoreCurrent AddSubShapeNext) r@rAr<node_color_typeexplorersub_color_typeshape_label_for_subsub_node sub_label color_toolr> shape_tools r=set_name_and_colorz'_create_xde..set_name_and_colors#      ::    ,Ftzz,R S ..z:$;%88XN$OO-!'' ",> P ..z:$;%88XN$OO-''$**2D2DoV$)$--dD).t||R__MH%6%H%HNf-.t||R^^LH%6%H%HNe,.t||R^^LH%6%H%HN.0H%6%G%GN&DZ&P#mmo'//1H * 6 67JH UI$++-"++%tzz'9'9>MMOmmo3@) "r?parentF)r rr"GetApplication_s NewDocument InitDocumentr$SetLengthUnit_sr' ShapeTool_sMain ColorTool_sSetAutoNaming_srr.r rGgetattrAddShapegetr9 AddComponentr7rDUpdateAssemblies)r0r1r2doc application label_mapr\r@r]rA parent_labelrZr>r[s @@@r= _create_xderoKs6 5i@ AC&668K6{CSIS!((a/$2G.GH&11#((*=J%11#((*=J {+)+IiI$=$=$I=$$=$BY'1 <<  x. >#,,T\\5AJ$===L9,GL""$#00t||LJ     $ $ ::/ tZ 0'1*! Jr? file_pathct|ttztzr t |}nt t |}tj|j|}|dS|S)zExport this shape to a BREP file Args: to_export (Shape): object or assembly file_path: Union[PathLike, str, bytes, BytesIO]: brep file path or memory buffer Returns: bool: write status T) rHrstrbytesrr rr Write_srG)r0rp return_values r= export_breprvsU)hnu46Y' ), $$Y%6%6 BL'49\9r?MbP?皙?binarylinear_deflectionangular_deflectionc|j td|j}|xjtdddzc_t|D]!}|j|j ||#t ||d}ttt||} | jdt} t} tj} | jD]!} | j!t"j$#| j'|| | }t)j*|j||_|S) aexport_gltf The glTF (GL Transmission Format) specification primarily focuses on the efficient transmission and loading of 3D models as a compact, binary format that is directly renderable by graphics APIs like WebGL, OpenGL, and Vulkan. It's designed to store detailed 3D model data, including meshes (vertices, normals, textures, etc.), animations, materials, and scene hierarchy, among other aspects. Args: to_export (Shape): object or assembly file_path (Union[PathLike, str, bytes]): glTF file path unit (Unit, optional): shape units. Defaults to Unit.MM. binary (bool, optional): output format. Defaults to False. linear_deflection (float, optional): A linear deflection setting which limits the distance between a curve and its tessellation. Setting this value too low will result in large meshes that can consume computing resources. Setting the value too high can result in meshes with a level of detail that is too low. The default is a good starting point for a range of cases. Defaults to 1e-3. angular_deflection (float, optional): Angular deflection setting which limits the angle between subsequent segments in a polyline. Defaults to 0.1. Raises: RuntimeError: Failed to write glTF file Returns: bool: write status z,Shape must have a location to export to glTF)rrr)r6rriFr2)theFile theIsBinaryT)location ValueErrorr*r rGmeshrorrr SetParallelrrrDefaultMessenger_sPrinters SetTraceLevelr Message_FailPerformr Clean_s)r0rpr1ryrzr{original_locationr@rkwriter index_mapprogress messengerprinterstatuss r= export_gltfrs3P!GHH!** (9i==Y'= << # II'); <= ( 4UKC'(;<&F t46I$&H**,I%%'<o::;<^^CH 5Fi''(+I Mr?T) timestamp write_pcurvesprecision_moderct||d}tj}|jD]!}|j t j #t} t| d} | jd| jd| jdt| jj} | js8td} | j!| jj|j"r$| j%t'|j"|St)|t*r)| j-t'|j/n| j-t'|| j1t'dt3j4t7j4t9j4t;j<dt?|t;j<d|j@| jC|tDjFt)|tHtJtLfr| jOtQ|} n| jStUtV|} | tXjZk(} | s t]d| S) aexport_step Export a build123d Shape or assembly with color and label attributes. Note that if the color of a node in an assembly isn't set, it will be assigned the color of its nearest ancestor. Args: to_export (Shape): object or assembly file_path (Union[PathLike, str, bytes, BytesIO]): step file path unit (Unit, optional): shape units. Defaults to Unit.MM. write_pcurves (bool, optional): write parametric curves to the STEP file. Defaults to True. precision_mode (PrecisionMode, optional): geometric data precision. Defaults to PrecisionMode.AVERAGE. Raises: RuntimeError: Unknown Compound type Returns: bool: success Tr}Fr build123dzwrite.surfacecurve.modezwrite.precision.modezFailed to write STEP file)/rorrrrrrr&r SetColorMode SetLayerMode SetNameModer WriterModelIsDoneApplyr7SetNamerrHr SetTimeStamp isoformatSetOriginatingSystemrInit_srrr SetIVal_sintvalueTransferrSTEPControl_AsIsrrrrsWriter WriteStreamr rrIFSelect_RetDone RuntimeError)r0rpr1rrrrkrrsessionrheaderrsuccesss r= export_steprRs@ i4 8C**,I%%'<o::;<$%G "7E 2F   t ()>)>)@ AF ==?,Q/ V]]_**,-/ @A i *    89L9L9N O P    8 C D  8 EF$$&!!#!!#8#m:LM5~7K7KL OOC2CCD)hU34hy12##D)$<=->>>G 677 Nr? toleranceangular_tolerance ascii_formatct|j|d|d}|jt}||_|j |jt |S)aExport STL Exports a shape to a specified STL file. Args: to_export (Shape): object or assembly file_path (Union[PathLike, str, bytes]): The path and file name to write the STL output to. tolerance (float, optional): A linear deflection setting which limits the distance between a curve and its tessellation. Setting this value too low will result in large meshes that can consume computing resources. Setting the value too high can result in meshes with a level of detail that is too low. The default is a good starting point for a range of cases. Defaults to 1e-3. angular_tolerance (float, optional): Angular deflection setting which limits the angle between subsequent segments in a polyline. Defaults to 0.1. ascii_format (bool, optional): Export the file as ASCII (True) or binary (False) STL format. Defaults to False (binary). Returns: bool: Success T)r rGrr ASCIIModerr)r0rprrrrrs r= export_stlrsV6 $9d,=t D LLN _F#F << ))8I+> ??r?c(d}d}d} tjdd5}t|j}dddt |||||t |d5}t j|d |j|d fid } ddd j | j} | jd}| jddk7st|tstd|  ||j!d t#j |dst%j&d||S#1swYxYw#1swYxYw#t$r!} td | jdd| d} ~ wwxYw#||j!dwwxYw)aExport a shape to PCBWay for quoting. This function writes ``to_export`` to a temporary STEP file, uploads that file to PCBWay's external web service, opens the returned pricing page in the default browser, and returns the pricing page URL. Args: to_export (Shape): object or assembly unit (Unit, optional): shape units. Defaults to Unit.MM. write_pcurves (bool, optional): write parametric curves to the STEP file. Defaults to True. precision_mode (PrecisionMode, optional): geometric data precision. Defaults to PrecisionMode.AVERAGE. Returns: str: URL of the pricing page z-https://www.pcbway.com/common/Build123dUpFileNz.stepF)suffixdelete)r1rrrbfilezapplication/step) x)filestimeoutz%PCBWay returned a non-JSON response: iredirectstateSUCCESSz.PCBWay upload failed or returned no redirect: T) missing_ok)newz*webbrowser failed to open on pricing page )tempfileNamedTemporaryFilernameropenrequestspostraise_for_statusjsonrrtextrhrHrrunlink webbrowserwarningswarn) r0r1rr upload_url redirect_url step_filetmpfresponseresultexcs r=export_to_pcbwayrs0AJLI%.  ( ( F '#SXXI '   ')  )T " a}} 3EFG!H  !!# ]]_F zz*- ::g ) +:lC3P@I 4Q      - ??9E:E E: E7E22E77E::F)rwrxF)^__doc__rrrriorosrrpathlibrtypingrr OCP.TopAbsTopAbsrJranytreer OCP.APIHeaderSectionr OCP.BRepMeshr OCP.BRepToolsr OCP.IFSelectrOCP.IGESControlr OCP.Interfacer OCP.Messagerrr OCP.RWGltfrOCP.STEPCAFControlrrOCP.STEPControlrr OCP.StlAPIrOCP.TCollectionrrr OCP.TColStdr OCP.TDataStdrOCP.TDFr OCP.TDocStdr OCP.TopExpr! OCP.XCAFAppr" OCP.XCAFDocr#r$r% OCP.XSControlr&build123d.build_commonr'build123d.build_enumsr(r)build123d.geometryr*build123d.topologyr+r,r-r.r/MMboolrorrrsrvfloatrAVERAGErrrr?r=rs>!! <1#.2*GG'OM$ =&(&+RR/25'CC$(77^^ ^9=^^B::#~%/(:: :2$ # MM#~%M M  M  M  M Mf$1$9$9 R(,RR#~%/(:R R R " RX~$R Rp" #@#@#~%#@#@ #@  #@  #@P$1$9$9 GG GG" G  Gr?