'jvddlmZddlmZmZmZmZmZddlZddl m Z m Z m Z m Z ddlmZmZerddlmZdejdz Zejd d z Zejejd z dz Zd ejejd z z ZGd d Z dddZ d d!dZ d"d#dZ d"d#dZdS)$) annotations) TYPE_CHECKINGIterableSequenceIteratorOptionalN)Vec3UVecMatrix44UCS)MeshVertexMergerMeshTransformer)GenericLayoutType?@g@g@?cbeZdZdZ d%d&d Zd'dZeZd(dZd)dZ d*d+d Z d,d"Z d-d$Z dS).SierpinskyPyramidz Args: location: location of base center as (x, y, z) tuple length: side length level: subdivide level sides: sides of base geometry rrrlocationr lengthfloatlevelintsidescD||_t|||||_dS)N)rrrr)rsierpinsky_pyramidpyramid_definitions)selfrrrrs Y/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/addons/sierpinski_pyramid.py__init__zSierpinskyPyramid.__init__s1 #5fE$ $ $    returnIterator[list[Vec3]]c#TK|jD]\}}|||VdS)z@Yields the pyramid vertices as list of :class:`ezdxf.math.Vec3`.N)r!_calc_vertices)r"rrs r#verticeszSierpinskyPyramid.vertices*sF $ 8 8 8 Hf%%h77 7 7 7 7 8 8r% list[Vec3]c ,|dz }|\}}}|jdkrut||z ||z |t||z||z |t||z||z|t||z ||z|t||||tzzgS|jdkrp|tz}|tz}t||z ||z |t||z||z |t|||z|t||||t zzgSt d)z Calculates the pyramid vertices. Args: location: location of the pyramid as center point of the base length: pyramid side length Returns: list of :class:`ezdxf.math.Vec3` rrsides has to be 3 or 4.)rr HEIGHT4 DY1_FACTOR DY2_FACTORHEIGHT3 ValueError) r"rrlen2xyzdy1dy2s r#r)z SierpinskyPyramid._calc_vertices1s8|1a :??QXq4x++QXq4x++QXq4x++QXq4x++Q1v//00  Z1__:%C:%CQXq3w**QXq3w**QC##Q1v//00  677 7r%list[Sequence[int]]c\|jdkrgdS|jdkrgdStd)z{Returns list of pyramid faces. All pyramid vertices have the same order, so one faces list fits them all. r))r-rr)rrr)rr<r)r<r-r)r-rrr-))r<rr)rrr-)rr<r-)r<rr-r.)rr3)r"s r#faceszSierpinskyPyramid.facesTsA :??MMM M Z1__??? ?677 7r%FNlayoutrmergeboolmatrixOptional[Matrix44]ucs Optional[UCS]Nonec|r/|}|||||dS|D]}|||||dS)a,Renders the sierpinsky pyramid into layout, set `merge` to ``True`` for rendering the whole sierpinsky pyramid into one MESH entity, set `merge` to ``False`` for individual pyramids as MESH entities. Args: layout: DXF target layout merge: ``True`` for one MESH entity, ``False`` for individual MESH entities per pyramid dxfattribs: DXF attributes for the MESH entities matrix: apply transformation matrix at rendering ucs: apply UCS at rendering ) dxfattribsrArC)rArCN)mesh render_meshpyramids)r"r>r?rGrArCrHpyramids r#renderzSierpinskyPyramid.render`s*  P99;;D   :f#       ==?? P P##FJv3#OOOO P Pr%Iterable[MeshTransformer]c#K|}|D]+}t}||||V,dS)znYields all pyramids of the sierpinsky pyramid as individual :class:`MeshTransformer` objects. r*r=N)r=radd_mesh)r"r=r*rHs r#rJzSierpinskyPyramid.pyramids~sZ   H"$$D MM85M 9 9 9JJJJ  r%rc|}t}|D]}|||tj|S)z8Returns geometry as one :class:`MeshTransformer` object.rO)r=r rPr from_builder)r"r=rHr*s r#rHzSierpinskyPyramid.meshsS !! : :H MM85M 9 9 9 9+D111r%rrrr)rr rrrrrr)r&r')rr rrr&r+)r&r:)FNNN) r>rr?r@rArBrCrDr&rE)r&rM)r&r) __name__ __module__ __qualname____doc__r$r*__iter__r)r=rLrJrHr%r#rrs)      8888 H!8!8!8!8F 8 8 8 8%)! PPPPP<222222r%rrrrrrrrrr&list[tuple[Vec3, float]]c t|}t|}|dkrtdt|||}t |dz D]2}g}|D])\}}|t|||*|}3|S)aBuild a Sierpinski pyramid. Args: location: base center point of the pyramid length: base length of the pyramid level: recursive building levels, has to 1 or bigger sides: 3 or 4 sided pyramids supported Returns: list of pyramid vertices rzlevel has to be 1 or bigger.)r rr3_sierpinsky_pyramidrangeextend)rrrrrJ_next_level_pyramidss r#r r s"H~~H JJE qyy7888"8VU;;H 519  ''  (   Hf  & &#Hfe<<    ' Or%rr cx|dkrt||S|dkrt||Std)Nr-rr.)sierpinsky_pyramid_3sierpinsky_pyramid_4r3)rrrs r#r\r\sE zz#Hf555 !#Hf5552333r%c  |dz }|dz }|\}}}t||z ||z ||ft||z||z ||ft||z ||z||ft||z||z||ft||||tzz|fgS)aBuild a 4-sided Sierpinski pyramid. Pyramid height = length of the base square! Args: location: base center point of the pyramid length: base length of the pyramid Returns: list of (location, length) tuples, representing the sierpinski pyramid r<r)r r/)rrr4len4r5r6r7s r#rcrcs A:D A:DGAq! a$hD! $ $d+ a$hD! $ $d+ a$hD! $ $d+ a$hD! $ $d+ aAw& ' '.  r%c |tzdz}|tzdz}|dz }|dz }|\}}}t||z ||z ||ft||z||z ||ft|||z||ft||||tzz|fgS)zBuild a 3-sided Sierpinski pyramid (tetraeder). Args: location: base center point of the pyramid length: base length of the pyramid Returns: list of (location, length) tuples, representing the sierpinski pyramid rr<r)r0r1r r2) rrr8r9r4rer5r6r7s r#rbrbs :  #C :  #C A:D A:DGAq! a$hC # #T* a$hC # #T* aS!  d# aAw& ' '.  r%rS)rrrrrrr&rZ)rr)rr rrrrr&rZ)r)rr rrr&rZ) __future__rtypingrrrrrmath ezdxf.mathr r r r ezdxf.render.meshr r ezdxf.eztypesrsqrtr/r2tanpir0cosr1rr r\rcrbrYr%r#rqs#"""""HHHHHHHHHHHHHH 000000000000????????0//////  #  $)C..3  TXdgm $ $s * 848DGcM** * z2z2z2z2z2z2z2z2| B7844444%(4%(r%