h3UddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZmZmZmZer dd lmZdd lmZeeZd e d<GddeZ!e!ddde"Z#e#ddf d'dZ$d(dZ%d)dZ&GddeZ'GddeZ(Gddee!Z)d*dZ*d+dZ+d,dZ,d-d Z-d-d!Z.d.d"Z/ d/d#Z0d0d$Z1Gd%d&eZ2y)1) annotations)AnySequenceIteratorUnionList TYPE_CHECKINGOptional) TypeAliasN)datetime)const) ParsingErrorInvalidLinkStructure) AcisHeader)AbstractEntityAbstractBuilder DataLoader DataExporterEntityExporter) AcisEntity)Vec3r SatRecordc4eZdZdZ d ddZdZy) SatEntityz2Low level representation of an ACIS entity (node).NcR||_||_||_||ng|_d|_yN)nameattr_ptriddata attributes)selfrrr r!s .lib/python3.12/site-packages/ezdxf/acis/sat.py__init__zSatEntity.__init__&s.   '+'7tR '+c:|jd|jdS)N())rr r#s r$__str__zSatEntity.__str__3s))AdggYa((r&)$-1N)rstrrr.r intr!Optional[list[Any]])__name__ __module__ __qualname____doc__r%r+r&r$rr#s@< $( , , ,  , " ,)r&rznull-ptrr,r-c0t|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_entityr8:s  $r4(AAL Hr&c2t|dkDxr|ddk(S)z@Returns ``True`` if the string `s` represents an entity pointer.r$)len)ss r$is_ptrr=Ns q6A: %!A$#+%r&cdfd }D]n}||j|_d|_g}|jD]6}t|r|j ||&|j |8||_pS)Nc@t|dd}|dk(rtS|S)Nr r-)r/NULL_PTR)r<numentitiess r$ptrz!resolve_str_pointers..ptrTs(!AB%j "9O}r&r,)r<r.returnr)rr"r!r=append)rBrCentityr!tokens` r$resolve_str_pointersrHSsr 0[[Ee} CJ' E" !   Or&c^eZdZd dZd dZdddZddZddZddZddZ dd Z dd Z dd Z y) SatDataLoaderc.||_||_d|_y)Nr)versionr!index)r#r!rLs r$r%zSatDataLoader.__init__hs   r&cF|jt|jkSr)rMr;r!r*s r$has_datazSatDataLoader.has_datamszzS^++r&Nc||S|j|j} t|}|xjdz c_|S#t$rt d|wxYw)Nzexpected integer, got r )r!rMr/ ValueErrorr)r#skip_satentryvalues r$read_intzSatDataLoader.read_intpsh  O $**% AJE a   A!7w?@ @ As AAc|j|j} t|}|xjdz c_|S#t$rt d|wxYw)Nzexpected double, got r )r!rMfloatrQr)r#rSrTs r$ read_doublezSatDataLoader.read_double|s] $**% @%LE a   @!6ug>? ? @s =Acj|jdd}|r|jStjS)NFI) read_boolrXmathinf)r#finites r$ read_intervalzSatDataLoader.read_intervals-S) ##% %xxr&cl|j}|j}|j}|||fSr)rX)r#xyzs r$ read_vec3zSatDataLoader.read_vec3s7            !Qwr&c|j|j}||k(r|xjdz c_y||k(r|xjdz c_ytd|d|d|)Nr TFzexpected bool string 'z' or 'z', got )r!rMr)r#truefalserTs r$r\zSatDataLoader.read_boolse $**% D= JJ!OJ e^ JJ!OJ$TF&wug F  r&c"|j|j}|jtjj ks|j dr1|xjdz c_|j|jdz Std|)N@r zexpected string, got )r!rMrLrFeaturesAT startswithrr#rTs r$read_strzSatDataLoader.read_strsn $**% <<%..++ +u/?/?/D JJ!OJ99TZZ!^, ,25':;;r&c|j|j}t|tr|xjdz c_|St dt |)Nr zexpected pointer, got )r!rM isinstancerrtype)r#rFs r$read_ptrzSatDataLoader.read_ptrsG4::& fn - JJ!OJM3DL>BCCr&cZtdDcgc]}|jc}Scc}w)N )rangerX)r#_s r$read_transformzSatDataLoader.read_transforms) -2"I6Iq  "I666s()r! list[Any]rLr/)rDboolr)rRz Optional[int]rDr/)rDrW)rDztuple[float, float, float])rgr.rhr.rDr{)rDr.)rDr)rDz list[float]) r1r2r3r%rOrUrXr`rer\rprtryr5r&r$rJrJgs5 ,    <D7r&rJc(eZdZdZddZddZddZy) SatBuilderz7Low level data structure to manage ACIS SAT data files.cLt|_g|_g|_i|_yr)rheaderbodiesrB_export_mappingr*s r$r%zSatBuilder.__init__s! l ') )+ 57r&cD|jt|jt|jj z|j_|jjdk(rd|j_n$t|j|j_|jj}|jt|j|jj|j|jj|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_entitiesrL n_recordsrBdumpsextendbuild_str_recordsrEsat_end_markerr#r!s r$dump_satzSatBuilder.dump_sats !$T[[!1C KK & &5 "   ;;  # %$%DKK !$' $6DKK !{{  " %dmmT[[5H5HIJ DKK..01 r&cf|Dcgc]}|jdk(s|c}|_||_ycc}w)z.Reset entities and bodies list. (internal API)bodyN)rrrB)r#rBr7s r$ set_entitieszSatBuilder.set_entitiess0"*?(Qaff.>q(?   @s..N)rDNonerD list[str])rBlist[SatEntity]rDr)r1r2r3r4r%rrr5r&r$r}r}sA8 ,!r&r}c$eZdZddZddZddZy) SatExporterc^t|j|j}t|_|S)N)r )rrsr r@r"r#rFrecords r$ make_recordzSatExporter.make_records#6;;6995$ r&c.t||jSr)SatDataExporterr!)r#rs r$make_data_exporterzSatExporter.make_data_exporterstV[[11r&ct}|j|_|j|j|j Sr)r}rrexport_recordsr)r#builders r$rzSatExporter.dump_sats:,T0023!!r&N)rFrrDr)rrrDrr)r1r2r3rrrr5r&r$rrs 2"r&rc#Kdfd }D]}|jg}|j||j|dk\r$|jt|j|j D]D}t |tr|j||+|jt|F|jddj|yw)Nc|tury dj|S#t$rtdt |dwxYw)Nr,r:zentity z not in record storage)r@rMrQrr.)r7rBs r$ptr_strz"build_str_records..ptr_strsT = Qx~~a()* * Q&Q8N'OP P Qs  "Ar# )r7rrDr.) rrEr"r.r r!rrrjoin)rBrLrrFtokensr!s` r$rrsQ++ gf//01 c> MM#fii. )KKD$ * gdm, c$i(   chhvsCC c#Kd}d}d}|jD]D}|dkDr||z }|dz}|dk(s|d}|dk(r%|dvr||z }/|dk(s5|s8t|}d}Fyw)Nrr rj 0123456789r)rstripr/)r<rAcollectrGcs r$parse_header_strrs CG E XXZ Q; QJE qLG!|  #X  ,  1HC #X##hGCs.A!A!A!A!ct}|dj}t|d|_ t|d|_t|d|_t|d|_tt|d} |d|_ |d|_ t|dkDr- tj|dt j"|_|dj} t'|d|_||ddfS#ttf$rYwxYw#t$rYwxYw#t$rY]wxYw#ttf$rYKwxYw)Nrr rk)rsplitr/rLrrflags IndexErrorrQlistr product_id acis_versionr;r strptimerDATE_FMT creation_daterW units_in_mm)r!rrs r$ parse_headerrsN \F !W]]_F^FN vay>q N6!9~ "47+ ,F "1I$Qi 6{Q #+#4#4VAY#OF !W]]_F "6!9- 48 '  #           #   sG9DD&,D$&D3DD D! D!$ D0/D03EEc#K|D]F}|jtjs|jtjry|Hywr)rnrEND_OF_ACIS_DATA_SATBEGIN_OF_ACIS_HISTORY_DATA)r!lines r$_filter_recordsr1s@ ??555 6$//  , ,;   sA Ac#Kdjt|}|jdD]}|j}|s|yw)Nrr)rrrstrip)r! merged_datars r$merge_record_stringsr:sA((?401K##C( L)s AA A cd}g}t|D]y}|j}|dj}|jdr-t | }||k7r t d|j d|j||dz }{|S)Nr-z-non-continuous sequence numbers not supportedr )rrrrnr/rpoprE)r!expected_seq_numrecordsrr first_tokenrAs r$ parse_recordsrBs!G$T* JJLQioo'  ! !# &{##C&&"C JJqMvA+ Nr&c g}|D]I}|d}|d}d}|dk\rt|d}|dd}n|dd}|jt||||K|S)Nrr r-rrkr)r/rEr)rrLrBrrattrid_r!s r$build_entitiesrTsr!#Hayay c>fQi.C!":D!":D $c489 Or&c6t|tr|j}n|}t|ts t dt }t |\}}||_t|}t||j}|jt||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) rrr. splitlinesr TypeErrorr}rrrrrLrrH)r<r!rrrrBs r$ parse_satres!S||~ dH %EFFlG%LFDGND!Ggv~~6H -h78 Nr&cfeZdZd dZdddZddZddZddZddZddZ ddZ dd Z dd Z dd Z y )rcB|j|_||_||_yr)rLexporterr!)r#rr!s r$r%zSatDataExporter.__init__~s''    r&cR|s%|jjt|yy)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!rEr.)r#rTrRs r$ write_intzSatDataExporter.write_ints" II  SZ (r&c>|jj|dy)Ngr!rEros r$ write_doublezSatDataExporter.write_doubles E!9&r&ctj|r|jjdy|jjd|j |y)Nr[rZ)r]isinfr!rErros r$write_intervalzSatDataExporter.write_intervals@ ::e  II  S ! II  S !   e $r&c|j|j|j|j|j|jyrrrbrcrdros r$write_loc_vec3zSatDataExporter.write_loc_vec38 %''" %''" %''"r&c|j|j|j|j|j|jyrrros r$write_dir_vec3zSatDataExporter.write_dir_vec3rr&cL|jj|r|y|yrr)r#rTrgrhs r$ write_boolzSatDataExporter.write_bools 2E2r&c|jjdt||jjt|y)Nrj)r!rEr;r.ros r$ write_strzSatDataExporter.write_strs5 1SZL)* U$r&c&|j|yr)rros r$write_literal_strz!SatDataExporter.write_literal_strs ur&ct}|js|jj|}|jj |yr)r@is_noner get_recordr!rErs r$ write_ptrzSatDataExporter.write_ptrs4~~]]--f5F  r&c:|jj|yr)r!rrs r$write_transformzSatDataExporter.write_transforms r&N)rrr!rz)F)rTr/rDr)rTrWrDr)rTrrDr)rTr{rgr.rhr.rDr)rTr.rDr)rFrrDr)r!rrDr)r1r2r3r%rrrrrrrrrrr5r&r$rr}s9 )'%# # 3%! r&r)rr.r!r0rDr)r<r.rDr{)rBrrDr)rBrrLr/rD Iterator[str])r<r.rDr)r! Sequence[str]rDz tuple[AcisHeader, Sequence[str]])r!rrDr)r!rrDzlist[SatRecord])rzSequence[SatRecord]rLr/rDr)r<zUnion[str, Sequence[str]]rDr})3 __future__rtypingrrrrrr r typing_extensionsr r]r rrrrhdrrabstractrrrrrrBr ezdxf.mathrr.r__annotations__rtupler@r8r=rHrJr}rrrrrrrrrrr5r&r$rs#( 5$Cy 9 ))( ZEG 4  $        (& (I7JI7X"!"!J ".+ " .(<$ +."02l2r&