h"2ddlmZddlmZmZmZmZddlmZm Z m Z m Z ddl m Z mZmZerddlmZgdZgdZgdZgd Zgd ZeeeegZgd Zgd Zgd gdgdgdgdgdgZGddZddZddZ d ddZ d ddZy) ) annotations) TYPE_CHECKINGIteratorSequenceOptional)Vec3UVecMatrix44UCS)MeshVertexMergerMeshTransformer MeshBuilder)GenericLayoutType)rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr)rrrrrrrrrrr!r#r$r%r&r'r)r*r+r,)rrrrrrrr r"r$r&r(r*r,) rrrrrrrr!r#r%r'r)r+)Lrrr)rr)rrrr)r-rr)r.rrr)r.rr)rr-r)rr-r)r-r-r)r.r-r)rr.r)rr.r)rr.r)r-r.r)r.r.rrr)r-rr)r.rrrr)r-rr)r.rr)rr-r)rr-r)r-r-r)r.r-r)rr.r)rr.r)r-r.r)r.r.rr$)r.rr)rr.r)r.r.r)rrr-)rrr-)r-rr-)r.rr-)rrr-)rrr-)r-rr-)r.rr-)rr-r-)rr-r-)r-r-r-)r.r-r-)rr.r-)rr.r-)r-r.r-)r.r.r-)rrr.)rrr.)rrr.)r-rr.)r.rr.)rrr.)rrr.)r-rr.)r.rr.)rrr.)r.rr.)rr-r.)rr-r.)r-r-r.)r.r-r.)rr.r.)rr.r.)rr.r.)r-r.r.)r.r.r.)@r/r/g@)rrrrrrrr)rr-rr)r.)rrr0r.)rrr1r0)r-r2r1r)rr.r2r-ceZdZdZ d d dZd dZeZed dZ d ddZ ddZ ddZ y) MengerSpongeaZ Args: location: location of lower left corner as (x, y, z) tuple length: side length level: subdivide level kind: type of menger sponge === =========================== 0 Original Menger Sponge 1 Variant XOX 2 Variant OXO 3 Jerusalem Cube === =========================== c,t|||||_y)N)locationlengthlevelkind)_menger_spongecube_definitions)selfr6r7r8r9s :lib/python3.12/site-packages/ezdxf/addons/menger_sponge.py__init__zMengerSponge.__init__es!/fE! c #K|jD]F\}}|\}}}tDcgc]%\}}}t|||zz|||zz|||zz'c}}}Hycc}}}ww)z5Yields the cube vertices as list of (x, y, z) tuples.N)r;_cube_verticesr) r<r6r7xyzxfyfzfs r=verticeszMengerSponge.verticespst $ 5 5 HfGAq!#1"0JBBQf_a"v+oq2;G"0 !6s#A *A A ctS)zuReturns list of cube faces. All cube vertices have the same order, so one faces list fits them all. ) cube_facesr?r=faceszMengerSponge.faces{s r?Nc|r&|j}|j||||y|jD]}|j||||y)aVRenders the menger sponge into layout, set `merge` to ``True`` for rendering the whole menger sponge into one MESH entity, set `merge` to ``False`` for rendering the individual cubes of the menger sponge as MESH entities. Args: layout: DXF target layout merge: ``True`` for one MESH entity, ``False`` for individual MESH entities per cube dxfattribs: DXF attributes for the MESH entities matrix: apply transformation matrix at rendering ucs: apply UCS transformation at rendering ) dxfattribsmatrixucs)rOrPN)mesh render_meshcubes)r<layoutmergerNrOrPrQcubes r=renderzMengerSponge.rendersW, 99;D   :f#     F L%r?c#K|j}|D]6}t}|j||tj|8yw)zfYields all cubes of the menger sponge as individual :class:`MeshTransformer` objects. rHrLN)rLr add_meshr from_builder)r<rLrHrQs r=rSzMengerSponge.cubessF H#%D MM85M 9!..t4 4sA Ac|j}t}|D]}|j||t|S)z8Returns geometry as one :class:`MeshTransformer` object.rY)rLr rZremove_duplicate_inner_faces)r<rLrQrHs r=rQzMengerSponge.meshs: !H MM85M 9+D11r?r`r`?rr)r6r r7floatr8intr9rc)returnzIterator[list[Vec3]])rdzlist[list[int]])FNNN) rTrrUboolrOzOptional[Matrix44]rPz Optional[UCS]rdNone)rdzIterator[MeshTransformer])rdr ) __name__ __module__ __qualname____doc__r>rH__iter__ staticmethodrLrWrSrQrKr?r=r4r4Ss&)          H%)! M!MM # M  M M>52r?r4ct}|j|_tt|j|_|SN)r rHlistmanifold_facesrL)rQnew_meshs r=r]r]s0 H H.45HN Or?c#Ki}|D]+}tt|} ||j|-|j D]}t |dk(s|dy#t$r |g||<YkwxYww)Nrr)tuplesortedappendKeyErrorvalueslen)rLledgerfacekeys r=rprps{9;FF4L! ! 3K  t $  u:?(N! !&F3K !s-A6A!#A6 A6!A30A62A33A6c |\t|t|z t|}d fd }|Dcgc] }|| fc}Scc}w)agDivides a cube in sub-cubes and keeps only cubes determined by the building schema. All sides are parallel to x-, y- and z-axis, location is a (x, y, z) tuple and represents the coordinates of the lower left corner (nearest to the axis origin) of the cube, length is the side-length of the cube Args: location: (x, y, z) tuple, coordinates of the lower left corner of the cube length: side length of the cube kind: int for 0: original menger sponge; 1: Variant XOX; 2: Variant OXO; 3: Jerusalem Cube; Returns: list of sub-cubes (location, length) cN|\}}}t|zz|zz|zzSrn)r)indicesrBrCrDinit_xinit_yinit_z step_sizes r= sub_locationz _subdivide..sub_locations?1a Q] " Q] " Q] "  r?)rdr)rb cube_sizesbuilding_schemas) r6r7r9remaining_cubesrr~rrrrs @@@@r= _subdividers](&FFFf 4 00I&t,O  ?N No7\' "I .o NN NsA c t|}|dvr tdt|}|dkr tdt|||}t|dz D]-}g}|D]"\}}|j t|||$|}/|S)aBuilds a menger sponge for given level. Args: location: (x, y, z) tuple, coordinates of the lower left corner of the cube length: side length of the cube level: level of menger sponge, has to be 1 or bigger kind: int for 0: original menger sponge; 1: Variant XOX; 2: Variant OXO; 3: Jerusalem Cube; Returns: list of sub-cubes (location, length) )rrrr-zkind has to be 0, 1, 2 or 3.rzlevel has to be 1 or bigger.)r9)rc ValueErrorrrangeextend)r6r7r8r9rS_next_level_cubess r=r:r:s$ t9D <788 JE qy788 xd 3E 519  % Hf  # #Jxd$K L!&   Lr?N)rQrrdr )rLzlist[Sequence[int]]rdzIterator[Sequence[int]])r_rar)r6r r7rbr9rcrdlist[tuple[Vec3, float]]r^) r6r r7rbr8rcr9rcrdr) __future__rtypingrrrr ezdxf.mathrr r r ezdxf.render.meshr r r ezdxf.eztypesrall_cubes_size_3_templateoriginal_menger_cubes menger_v1 menger_v2jerusalem_cuberrrArJr4r]rprr:rKr?r=rs#>>00LL/             _2_2D HI O O.3 OAD O OH%       r?