hddlmZddlmZmZmZmZmZddlZddl m Z m Z m Z m Z ddlmZmZerddlmZdej&dz Zej&d d z Zej,ej.d z dz Zd ej2ej.d z z ZGd d Z d ddZ d ddZ d ddZ d ddZy)) annotations) TYPE_CHECKINGIterableSequenceIteratorOptionalN)Vec3UVecMatrix44UCS)MeshVertexMergerMeshTransformer)GenericLayoutType?@g@g@?ceZdZdZ d d dZd dZeZ d dZddZ d ddZ ddZ dd Z y)SierpinskyPyramidz Args: location: location of base center as (x, y, z) tuple length: side length level: subdivide level sides: sides of base geometry c:||_t|||||_y)N)locationlengthlevelsides)rsierpinsky_pyramidpyramid_definitions)selfrrrrs ?lib/python3.12/site-packages/ezdxf/addons/sierpinski_pyramid.py__init__zSierpinskyPyramid.__init__s! #5fE$  c#\K|jD]\}}|j||yw)z@Yields the pyramid vertices as list of :class:`ezdxf.math.Vec3`.N)r_calc_vertices)rrrs rverticeszSierpinskyPyramid.vertices*s. $ 8 8 Hf%%h7 7!9s*,c |dz }|\}}}|jdk(r`t||z ||z |t||z||z |t||z||z|t||z ||z|t||||tzzgS|jdk(r]|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` rsides has to be 3 or 4.)rr HEIGHT4 DY1_FACTOR DY2_FACTORHEIGHT3 ValueError) rrrlen2xyzdy1dy2s rr!z SierpinskyPyramid._calc_vertices1s|1a ::?QXq4x+QXq4x+QXq4x+QXq4x+Q1v//0  ZZ1_:%C:%CQXq3w*QXq3w*QC#Q1v//0  67 7rcd|jdk(rgdS|jdk(rgdStd)z{Returns list of pyramid faces. All pyramid vertices have the same order, so one faces list fits them all. r$))r%r)rr4r$)r4r3r$)r3r%r$)r%rr$r%))r3r4r)rr4r%)r4r3r%)r3rr%r&)rr+)rs rfaceszSierpinskyPyramid.facesTs2 ::?M M ZZ1_? ?67 7rNc|r&|j}|j||||y|jD]}|j||||y)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 ) dxfattribsmatrixucs)r8r9N)mesh render_meshpyramids)rlayoutmerger7r8r9r:pyramids rrenderzSierpinskyPyramid.render`sW* 99;D   :f#   ==?##FJv3#O+rc#|K|j}|D]#}t}|j|||%yw)znYields all pyramids of the sierpinsky pyramid as individual :class:`MeshTransformer` objects. r"r5N)r5radd_mesh)rr5r"r:s rr<zSierpinskyPyramid.pyramids~s: H"$D MM85M 9Js:<c|j}t}|D]}|j||tj|S)z8Returns geometry as one :class:`MeshTransformer` object.rB)r5r rCr from_builder)rr5r:r"s rr:zSierpinskyPyramid.meshs@ !H MM85M 9++D11r)rGrGrr4r$)rr rfloatrintrrI)returnzIterator[list[Vec3]])rr rrHrJz list[Vec3])rJzlist[Sequence[int]])FNNN) r=rr>boolr8zOptional[Matrix44]r9z Optional[UCS]rJNone)rJzIterable[MeshTransformer])rJr) __name__ __module__ __qualname____doc__rr"__iter__r!r5r@r<r:rrrrs)          8 H!8!8&+!8 !8F 8%)! P!PP # P  P P<2rrc t|}t|}|dkr tdt|||}t |dz D],}g}|D]!\}}|j t|||#|}.|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 r4zlevel has to be 1 or bigger.)r rIr+_sierpinsky_pyramidrangeextend)rrrrr<_next_level_pyramidss rrrs"H~H JE qy788"8VU;H 519   ( Hf  & &#Hfe< !)'  Orc\|dk(r t||S|dk(r t||Std)Nr%r$r&)sierpinsky_pyramid_3sierpinsky_pyramid_4r+)rrrs rrTrTs8 z#Hf55 !#Hf55233rc |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 r3r$)r r')rrr,len4r-r.r/s rr[r[s A:D A:DGAq! a$hD! $d+ a$hD! $d+ a$hD! $d+ a$hD! $d+ aAw& '.  rc |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 rr3r$)r(r)r r*) rrr0r1r,r]r-r.r/s rrZrZs :  #C :  #C A:D A:DGAq! a$hC #T* a$hC #T* aS! d# aAw& '.  rrF)rrHrrIrrIrJlist[tuple[Vec3, float]])rr$)rr rrHrrIrJr_)r)rr rrHrJr_) __future__rtypingrrrrrmath ezdxf.mathr r r r ezdxf.render.meshr r ezdxf.eztypesrsqrtr'r*tanpir(cosr)rrrTr[rZrRrrrjs9#HH 00?/  #  $))C.3  TXXdggm $s * 8488DGGcM* * z2z2|       B7844!40344%(!4%(!r