vj6ddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZdd lmZmZdd lmZdd lmZmZmZmZmZmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-de de.fdZ/de de)dzfdZ0de de(dzfdZ1dedede(dzfdZ2de'de.fd Z3de'de.fd!Z4de'de.fd"Z5d#ede'fd$Z6y)%)cast)Path) TopoDS_Shape)TCollection_ExtendedString)Quantity_ColorRGBA) TDF_LabelTDF_LabelSequence)IFSelect_RetDone)TDocStd_Document) TDataStd_NameTDataStd_TreeNode)STEPCAFControl_Reader)XCAFDoc_ColorSurfXCAFDoc_DocumentToolXCAFDoc_ColorToolXCAFDocXCAFDoc_ColorTypeXCAFDoc_Material)TDocStd_Application)XmlXCAFDrivers)BinXCAFDrivers)Interface_Static)PCDM_ReaderStatus)AssemblyProtocolColorMaterial)Location)Shapelabelreturncd}t}|jtjrK|jtj|t |j j }|S)z2 Helper to get the name of a given label. )r IsAttributeGetID_s FindAttributestrGet ToExtString)r rv name_attrs Q/DATA/.local/lib/python3.12/site-packages/cadquery/occ_impl/importers/assembly.py _get_namer-s^ BI ..01 M113Y? ,,. / INcd}tj}|j|rt}|j |||j j }t}|j tj||jj}|jj}|j}|jj} t|||| }|S)z7 Helper to get the material for a given label. N)name descriptiondensity densityUnit)rMaterialRefGUID_sr$r r&FatherLabelrr%GetName ToCStringGetDescription GetDensityGetDensValTyper) r r*material_ref_guidattrmaterial_label material_attrr0r1r2 density_units r, _get_materialrA.s B113 *+ " -t4,,.(* $$%5%=%=%?O$$&002#224>>@ **,$335??A #$   Ir.ctjtj}tjtj}t }|j |r|j|||jj}t}tj|||j}t|j|j!|j#|j%d}|S|j |r|j|||jj}t}tj|||j}t|j|j!|j#|j%d}|Sd}|S)z< Helper to get the instance color of a given label. FN)rColorRefGUID_srrXCAFDoc_ColorGenr r$r&r5r6rr GetColor_sGetRGBrRedGreenBlueAlpha)r color_ref_guidcolor_ref_guid_genericr= color_labelcolorrgbr*s r,_get_ref_colorrPNsW ++,=,O,OPN$334E4V4VW  D ( ND1kkm))+ "$$$[%8lln 3779ciik388:u{{}e L I   1 2 2D9kkm))+ "$$$[%8lln 3779ciik388:u{{}e L I Ir.s color_toolct}|j|t|rY|j}t |j |j |j|jd}|Sd}|S)z9 Helper to get the shape color of a given shape. FN) rGetColorrrFrrGrHrIrJ)rQrRrNrOr*s r,_get_shape_colorrUssi  E1/7lln 3779ciik388:u{{}e L I Ir.assypathc~t}|jd|jd|jd|j dt j dd|j|}|tk7rtd|ttd}|j|t||y)z Import a step file into an assembly. :param assy: An Assembly object that will be packed with the contents of the STEP file. :param path: Path and filename to the STEP file to read. :return: None Tzread.stepcaf.subshapes.namezError reading STEP file: XmXCAFN)r SetColorMode SetNameMode SetLayerMode SetSHUOModer SetIVal_sReadFiler ValueErrorr rTransfer _importDoc)rVrW step_readerstatusdocs r, importSteprgs()KT"D!T"D!._process_labels () ""3 4{))+,A$**1q51J!*-I**:6C&)Xc]xzF'' 3%K --j)D'z2(4**95(3H#~~8~rs#6+0)(94,--*&88 Y 3 x$@")" "J 2CPT "%SB$C:$C:C$C,<Cr.