'j!ddlmZddlmZmZmZmZddlmZm Z m Z m Z ddl m Z mZmZerddlmZgdZgdZgdZgd Zgd ZeeeegZgd Zgd Zgd gdgdgdgdgdgZGddZd,dZd-dZ d.d/d'Z d0d1d*Zd+S)2) 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-cjeZdZdZ d#d$d Zd%dZeZed&dZ d'd(dZ d)d Z d*d"Z dS)+ 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 === =========================== r6r6?rrlocationr lengthfloatlevelintkindc6t|||||_dS)N)r8r9r;r=)_menger_spongecube_definitions)selfr8r9r;r=s T/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/addons/menger_sponge.py__init__zMengerSponge.__init__es*!/fE! ! ! returnIterator[list[Vec3]]c#hK|jD]#\}|\fdtDV$dS)z5Yields the cube vertices as list of (x, y, z) tuples.c ^g|])\}}}t|zz|zz|zz*Sr).0xfyfzfr9xyzs rB z)MengerSponge.vertices..tsSBBQf_a"v+oq2;GGrDN)r@_cube_vertices)rAr8r9rOrPrQs @@@@rBverticeszMengerSponge.verticesps $ 5   HfGAq!"0      rDlist[list[int]]ctS)zuReturns list of cube faces. All cube vertices have the same order, so one faces list fits them all. ) cube_facesrIrDrBfaceszMengerSponge.faces{s rDFNlayoutrmergeboolmatrixOptional[Matrix44]ucs Optional[UCS]Nonec|r/|}|||||dS|D]}|||||dS)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 ) dxfattribsr\r^)r\r^N)mesh render_meshcubes)rArYrZrbr\r^rccubes rBrenderzMengerSponge.renders,  M99;;D   :f#        M M  F LLLL M MrDIterator[MeshTransformer]c#K|}|D]=}t}|||tj|V>dS)zfYields all cubes of the menger sponge as individual :class:`MeshTransformer` objects. rTrXN)rXr add_meshr from_builder)rArXrTrcs rBrezMengerSponge.cubessi  5 5H#%%D MM85M 9 9 9!.t44 4 4 4 4 5 5rDr c|}t}|D]}|||t|S)z8Returns geometry as one :class:`MeshTransformer` object.rj)rXr rkremove_duplicate_inner_faces)rArXrcrTs rBrczMengerSponge.meshsP !! : :H MM85M 9 9 9 9+D111rDr5r7rr)r8r r9r:r;r<r=r<)rErF)rErU)FNNN) rYrrZr[r\r]r^r_rEr`)rErh)rEr ) __name__ __module__ __qualname____doc__rCrT__iter__ staticmethodrXrgrercrIrDrBr4r4Ss&)      H\%)! MMMMM>5555222222rDr4rcrrEr ct}|j|_tt|j|_|SN)r rTlistmanifold_facesrX)rcnew_meshs rBrnrns5  H H.4455HN OrDrXlist[Sequence[int]]Iterator[Sequence[int]]c#Ki}|D]P}tt|} |||;#t$r |g||<YMwxYw|D]}t |dkr |dV dS)Nrr)tuplesortedappendKeyErrorvalueslen)rXledgerfacekeys rBryrys9;F!!F4LL!! ! 3K  t $ $ $ $ ! ! !&F3KKK ! u::??(NNNsAAAr5r7r8r r9r:r=r<list[tuple[Vec3, float]]c|\t|t|z t|}dfd fd|DS)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) rErcV|\}}}t|zz|zz|zzSrwrJ)indicesrOrPrQinit_xinit_yinit_z step_sizes rB sub_locationz _subdivide..sub_locationsD1a Q] " Q] " Q] "   rDc*g|]}|fSrIrI)rKrrrs rBrRz_subdivide..s( N N N7\\' " "I . N N NrD)rEr)r: cube_sizesbuilding_schemas) r8r9r=remaining_cubesrrrrrs @@@@@rB _subdividers(&FFFf 4 00I&t,O          O N N N No N N NNrDrr;c Bt|}|dvrtdt|}|dkrtdt|||}t|dz D]3}g}|D]*\}}|t|||+|}4|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.)r=)r< ValueErrorrrangeextend)r8r9r;r=re_next_level_cubess rBr?r?s$ t99D <7888 JJE qyy7888 xd 3 3 3E 519  !! % M M Hf  # #Jxd$K$K$K L L L L  LrDN)rcrrEr )rXr{rEr|)r5r7r)r8r r9r:r=r<rErro) r8r r9r:r;r<r=r<rEr) __future__rtypingrrrr ezdxf.mathrr r r ezdxf.render.meshr r r ezdxf.eztypesrall_cubes_size_3_templateoriginal_menger_cubes menger_v1 menger_v2jerusalem_cuberrrSrWr4rnryrr?rIrDrBrs#""""">>>>>>>>>>>>000000000000LLLLLLLLLL0//////                     LLLLLLLLLLLL  _2_2_2_2_2_2_2_2D    HI O O O O OH% rD