'j1UddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZmZmZmZer dd lmZdd lmZeeZd e d<GddeZ!e!ddde"Z#e#ddfd9dZ$d:dZ%d;dZ&Gd d!eZ'Gd"d#eZ(Gd$d%ee!Z)dd-Z,d?d.Z-d?d/Z.d@d1Z/dAd4Z0dBd6Z1Gd7d8eZ2dS)C) annotations)AnySequenceIteratorUnionList TYPE_CHECKINGOptional) TypeAliasN)datetime)const) ParsingErrorInvalidLinkStructure) AcisHeader)AbstractEntityAbstractBuilder DataLoader DataExporterEntityExporter) AcisEntity)Vec3r SatRecordc(eZdZdZ ddd Zd ZdS) SatEntityz2Low level representation of an ACIS entity (node).$-1Nnamestrattr_ptridintdataOptional[list[Any]]cT||_||_||_||ng|_d|_dSN)rr r!r# attributes)selfrr r!r#s H/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/acis/sat.py__init__zSatEntity.__init__&s4   '+'7ttR '+c&|jd|jdS)N())rr!r(s r)__str__zSatEntity.__str__3s)((dg((((r+)rrN)rrr rr!r"r#r$)__name__ __module__ __qualname____doc__r*r0r+r)rr#sM<< $( , , , , ,)))))r+rznull-ptrrrrrr#r$returnc8t|d||}||_|S)zFactory to create new ACIS entities. Args: name: entity type attributes: reference to the entity attributes or :attr:`NULL_PTR`. id: unique entity id as integer or -1 data: generic data container as list r)rr')rr'r!r#es r) new_entityr9:s# $r4((AAL Hr+sboolc@t|dko |ddkS)z@Returns ``True`` if the string `s` represents an entity pointer.r$)len)r:s r)is_ptrr?Ns q66A: %!A$#+%r+entitieslist[SatEntity]cdfd }D]t}||j|_d|_g}|jD]E}t|r|||0||F||_uS)Nr:rr6rc\t|dd}|dkrtS|S)Nr r)r"NULL_PTR)r:numr@s r)ptrz!resolve_str_pointers..ptrTs.!ABB%jj "99O}r+r)r:rr6r)r r'r#r?append)r@rFentityr#tokens` r)resolve_str_pointersrJSs   C00[ # #Ee}} # CCJJ'''' E"""" Or+c`eZdZddZddZddd ZddZddZd dZd!dZ d"dZ d#dZ d$dZ d S)% SatDataLoaderr# list[Any]versionr"c0||_||_d|_dS)Nr)rNr#index)r(r#rNs r)r*zSatDataLoader.__init__hs   r+r6r;c<|jt|jkSr&)rPr>r#r/s r)has_datazSatDataLoader.has_datamszS^^++r+Nskip_sat Optional[int]c||S|j|j} t|}n #t$rt d|wxYw|xjdz c_|S)Nzexpected integer, got r )r#rPr" ValueErrorr)r(rSentryvalues r)read_intzSatDataLoader.read_intpsx  O $*% AJJEE A A A???@@ @ A a  s (Afloatc|j|j} t|}n #t$rt d|wxYw|xjdz c_|S)Nzexpected double, got r )r#rPrZrVr)r(rWrXs r) read_doublezSatDataLoader.read_double|sm $*% @%LLEE @ @ @>u>>?? ? @ a  s $Acr|dd}|r|StjS)NFI) read_boolr\mathinf)r(finites r) read_intervalzSatDataLoader.read_intervals6S))  &##%% %xr+tuple[float, float, float]c|}|}|}|||fSr&r\)r(xyzs r) read_vec3zSatDataLoader.read_vec3s@               !Qwr+truerfalsec|j|j}||kr|xjdz c_dS||kr|xjdz c_dStd|d|d|)Nr TFzexpected bool string 'z' or 'z', got )r#rPr)r(rlrmrXs r)r`zSatDataLoader.read_boolsw $*% D== JJ!OJJ4 e^^ JJ!OJJ5 FT F F F Fu F F   r+c|j|j}|jtjjks|dr%|xjdz c_|j|jdz Std|)N@r zexpected string, got )r#rPrNrFeaturesAT startswithrr(rXs r)read_strzSatDataLoader.read_strsn $*% <%.+ + +u/?/?/D/D + JJ!OJJ9TZ!^, ,:5::;;;r+rc|j|j}t|tr|xjdz c_|St dt |)Nr zexpected pointer, got )r#rP isinstancerrtype)r(rHs r)read_ptrzSatDataLoader.read_ptrsS4:& fn - -  JJ!OJJMBDLLBBCCCr+ list[float]c:fdtdDS)Nc8g|]}Sr5rg).0_r(s r) z0SatDataLoader.read_transform..s%666q  ""666r+ )ranger/s`r)read_transformzSatDataLoader.read_transforms% 7666E"II6666r+)r#rMrNr")r6r;r&)rSrTr6r")r6rZ)r6re)rlrrmrr6r;)r6r)r6r)r6r{) r1r2r3r*rRrYr\rdrkr`rvrzrr5r+r)rLrLgs ,,,,           <<<<DDDD777777r+rLc*eZdZdZd dZd dZd d Zd S) SatBuilderz7Low level data structure to manage ACIS SAT data files.r6NonecVt|_g|_g|_i|_dSr&)rheaderbodiesr@_export_mappingr/s r)r*zSatBuilder.__init__s) ll ') )+ 57r+ list[str]c|t|jt|jjz|j_|jjdkr d|j_nt|j |j_|j }| t|j |jj| |j|S)zReturns the text representation of the ACIS file as list of strings without line endings. Raise: InvalidLinkStructure: referenced ACIS entity is not stored in the :attr:`entities` storage r)reorder_recordsr>rr"rhas_asm_header n_entitiesrN n_recordsr@dumpsextendbuild_str_recordsrGsat_end_markerr(r#s r)dump_satzSatBuilder.dump_sats !$T[!1!1C K &5 5 "   ; # % %$%DK ! !$' $6$6DK !{  "" %dmT[5HIIJJJ DK..00111 r+r@rAc6d|D|_||_dS)z.Reset entities and bodies list. (internal API)c(g|]}|jdk |S)body)r)r~r8s r)rz+SatBuilder.set_entities..s$???Qaf.>.>q.>.>.>r+N)rr@)r(r@s r) set_entitieszSatBuilder.set_entitiess"??(???   r+N)r6rr6r)r@rAr6r)r1r2r3r4r*rrr5r+r)rrsVAA8888 ,!!!!!!r+rc&eZdZd dZd dZdd Zd S) SatExporterrHrr6rcTt|j|j}t|_|S)N)r!)rryr!rDr'r(rHrecords r) make_recordzSatExporter.make_records%6;69555$ r+rrc,t||jSr&)SatDataExporterr#)r(rs r)make_data_exporterzSatExporter.make_data_exporterstV[111r+rct}|j|_|||Sr&)rrrexport_recordsr)r(builders r)rzSatExporter.dump_satsF,,T0022333!!!r+N)rHrr6r)rrr6rr)r1r2r3rrrr5r+r)rrsP 2222""""""r+rrNr" Iterator[str]c#Kd fd }D]}|jg}|||j|dkr'|t|j|jD]X}t |tr|||6|t|Y|dd|VdS) Nr8rr6rc|turdS d|S#t$r!tdt |dwxYw)Nrr=zentity z not in record storage)rDrPrVrr)r8r@s r)ptr_strz"build_str_records..ptr_strsp ==5 Q*x~~a((** * Q Q Q&'OQ'O'O'OPP P Qs &+Ar# )r8rr6r) rrGr'rr!r#rxrjoin)r@rNrrHtokensr#s` r)rrsQQQQQQ  + ggf/00111 c>> MM#fi.. ) ) )K ) )D$ ** ) ggdmm,,,, c$ii(((( chhv  r+c#Kd}d}d}|D]I}|dkr||z }|dz}|dkr|Vd}|dkr&|dvr||z }0|dkr|rt|}d}JdS)Nrr rp 0123456789r)rstripr")r:rEcollectrIcs r)parse_header_strrs CG E XXZZ   Q;; QJE qLG!||  #XX  ,   1HCC #XX#X#hhGC  r+ Sequence[str] tuple[AcisHeader, Sequence[str]]c@t}|d}t|d|_ t|d|_t|d|_t|d|_n#ttf$rYnwxYwtt|d} |d|_ |d|_ n#t$rYnwxYwt|dkr< tj|dt j|_n#t$rYnwxYw|d} t'|d|_n#ttf$rYnwxYw||ddfS)Nrr rq)rsplitr"rNrrflags IndexErrorrVlistr product_id acis_versionr>r strptimerDATE_FMT creation_daterZ units_in_mm)r#rrs r) parse_headerrs \\F !W]]__F^^FN vay>>q NN6!9~~  #     "47++ , ,F "1I$Qi       6{{Q #+#4VAY#O#OF     D  !W]]__F "6!9--  #     48 sIABB'&B' C(( C54C5 *D77 EE"E==FFc#K|D]G}|tjs|tjrdS|VHdSr&)rtrEND_OF_ACIS_DATA_SATBEGIN_OF_ACIS_HISTORY_DATA)r#lines r)_filter_recordsr1se ??55 6 6 $//  ,; ;   FF r+c#Kdt|}|dD]}|}|r|VdS)Nrr)rrrstrip)r# merged_datars r)merge_record_stringsr:se((?40011K##C((  LLLr+list[SatRecord]c`d}g}t|D]}|}|d}|dr:t | }||krt d|d|||dz }|S)Nr-z-non-continuous sequence numbers not supportedr )rrrrtr"rpoprG)r#expected_seq_numrecordsrr first_tokenrEs r) parse_recordsrBs!G$T**   JJLLQioo''  ! !# & & {###C&&&"C JJqMMMvA Nr+rSequence[SatRecord]c g}|D]i}|d}|d}d}|dkr t|d}|dd}n |dd}|t||||j|S)Nrr rrrqr)r"rGr)rrNr@rrattrid_r#s r)build_entitiesrTs!#H : :ayay c>>fQi..C!"":DD!"":D $c4889999 Or+Union[str, Sequence[str]]ct|tr|}n|}t|tst dt }t |\}}||_t|}t||j }| t||S)zReturns the :class:`SatBuilder` for the ACIS :term:`SAT` file content given as string or list of strings. Raises: ParsingError: invalid or unsupported ACIS data structure z+expected as string or a sequence of strings) rxr splitlinesr TypeErrorrrrrrrNrrJ)r:r#rrrr@s r) parse_satres!S||~~ dH % %GEFFFllG%%LFDGND!!Ggv~66H -h77888 Nr+cheZdZddZd d!d Zd"d Zd"dZd#dZd#dZd$dZ d%dZ d%dZ d&dZ d'dZ dS)(rexporterrr#rMc:|j|_||_||_dSr&)rNrr#)r(rr#s r)r*zSatDataExporter.__init__~s'    r+FrXr"r6rc\|s)|jt|dSdS)zThere are sometimes additional int values in SAB files which are not present in SAT files, maybe reference counters e.g. vertex, coedge. N)r#rGr)r(rXrSs r) write_intzSatDataExporter.write_ints8 ) I  SZZ ( ( ( ( ( ) )r+rZc>|j|ddS)Ngr#rGrus r) write_doublezSatDataExporter.write_doubles" E&&&&&r+ctj|r|jddS|jd||dS)Nr_r^)raisinfr#rGrrus r)write_intervalzSatDataExporter.write_intervals` :e   % I  S ! ! ! ! ! I  S ! ! !   e $ $ $ $ $r+rc||j||j||jdSr&rrhrirjrus r)write_loc_vec3zSatDataExporter.write_loc_vec3J %'""" %'""" %'"""""r+c||j||j||jdSr&rrus r)write_dir_vec3zSatDataExporter.write_dir_vec3rr+r;rlrrmcB|j|r|n|dSr&r)r(rXrlrms r) write_boolzSatDataExporter.write_bools' 1E22222r+c|jdt||jt|dS)Nrp)r#rGr>rrus r) write_strzSatDataExporter.write_strsI )SZZ))*** U$$$$$r+c0||dSr&)rrus r)write_literal_strz!SatDataExporter.write_literal_strs ur+rHrct}|js|j|}|j|dSr&)rDis_noner get_recordr#rGrs r) write_ptrzSatDataExporter.write_ptrsB~ 6]--f55F      r+rc:|j|dSr&)r#rrs r)write_transformzSatDataExporter.write_transforms r+N)rrr#rM)F)rXr"r6r)rXrZr6r)rXrr6r)rXr;rlrrmrr6r)rXrr6r)rHrr6r)r#rr6r)r1r2r3r*rrrrrrrrrrr5r+r)rr}s )))))''''%%%%#### #### 3333%%%%!!!! r+r)rrr#r$r6r)r:rr6r;)r@rAr6rA)r@rArNr"r6r)r:rr6r)r#rr6r)r#rr6r)r#rr6r)rrrNr"r6rA)r:rr6r)3 __future__rtypingrrrrrr r typing_extensionsr rar rrrrhdrrabstractrrrrrr@r ezdxf.mathrrr__annotations__rtuplerDr9r?rJrLrrrrrrrrrrrr5r+r)rs^#""""""(''''' 55555555 $$$$$$Cy ))))))))( 9ZEEGG 4 4  $      (&&&& (I7I7I7I7I7JI7I7I7X"!"!"!"!"!"!"!"!J " " " " ".+ " " " .(<$"022222l22222r+