k j UddlmZddlmZddlZddlmZmZm Z ddl m Z m Z m Z mZ d(d)dZggdgdgdgdgdgdgddZded<d*d"Zd+d%ZGd&d'ZdS),) annotations)IteratorN)Cell MinimalCell build_tree)FuncPoint ValuedPointbinary_search_zero'fnr pminr pmax min_depthint max_cellstolnp.ndarray | Nonec ~tj|}tj|}| ||z dz }ntj|}td||||||}tt ||}g}|D]*} t | ||} | || +||fS)uReturns the surface representing fn([x,y,z])=0 on pmin[0] ≤ x ≤ pmax[0] ∩ pmin[1] ≤ y ≤ pmax[1] ∩ pmin[2] ≤ z ≤ pmax[2]Ni)npasarrayrlistSimplexGenerator get_simplices march_simplexextend) rrrrrrocttree simplicesfacessimplex face_lists Q/home/agentuser/manim-venv/lib/python3.11/site-packages/isosurfaces/isosurface.pyplot_isosurfacer& s :d  D :d  D {d{d"jooBdIy#FFG%gr22@@BBCCI E$$!'2s33  LL # # # e ))rrrrr)rr*rr*rr*)rrr*r)rr))rrr*rrr)r+r/r(r1)r.r(r-r0)r.r,r)r1)rrr*rr z dict[int, list[tuple[int, int]]]TETRAHEDRON_TABLEr#list[ValuedPoint]returnlist[tuple[int, int]]c~d}|D]}d|z|jdkz}|tvr t|Std|z S)zFAssumes the simplex is a tetrahedron, so this is marching tetrahedronsrr*)valr5)r#idps r% march_indicesr>1sR B "" Vquqy !  $$ "--r' np.ndarray3list[list[Point]] | tuple[list[Point], list[Point]]cBt|}|rg}|D]D\}}t||||||\}}|sJ||jEt |dkr|gS|d|d|dg|d|d|dgfSdS)Nrrrr*)r>r appendposlen) r#rrindicespointsij intersectionis_zeros r%rr=sG$$G X  , ,DAq$6wqz71:rSV$W$W !L'NN7 MM,* + + + + v;;!  8O1Ivay&)4vay&)VTUY6WW W X Xr'c6eZdZddZdd Zdd ZddZddZdS)rrootrrr r7Nonec"||_||_dSN)rLr)selfrLrs r%__init__zSimplexGenerator.__init__Ps r'Iterator[list[ValuedPoint]]c6||jSrO)get_simplices_withinrL)rPs r%rzSimplexGenerator.get_simplicesTs((333r'octc #RK|jr'|jD]}||Ed{VdSdD]s}dD]n}|||}|D]S}|1|||||Ed{V5|||||Ed{VTotdS)N)rrr*r.)childrenrTwalk_leaves_in_directionget_simplices_between_face get_subcellget_simplices_between)rPrUchildaxisdiradjleafs r%rTz%SimplexGenerator.get_simplices_withinWs' < X < <44U;;;;;;;;;; < <" X X!XXC66tSAAC #XX<'+'F'FsCOO\`beLfLf'g'ggggggggg'+'A'A#tTSV'W'WWWWWWWWW XX X Xr'abr]rr^c#K|j|jkr ||g\}}d|z }||d|z }||fD]}|||Ed{VdS)u{ Parameters axis and dir are same as Cell.get_leaves_in_direction. They denote the direction a→b rN)depthrZrY)rPrarbr]r^facevolumes r%r[z&SimplexGenerator.get_simplices_betweenfs 7QW  VFQc'C}}T1s7++!f E EF66vtDD D D D D D D D D E Er'rfrerc#KtdD]x}||dz|dz}|jD]R}||j||j||j|gVSydS)Nr2r*)rangerZverticesget_dualr)rPrfrerGedgevs r%rYz+SimplexGenerator.get_simplices_between_facess q  A##AE1622D]  OODG,,MM$'**MM$'**    r'N)rLrrr r7rM)r7rR)rUrr7rR) rarrbrr]rr^rr7rR)rfrrerr7rR)__name__ __module__ __qualname__rQrrTr[rYr'r%rrOs4444 X X X X E E E Er'r)r rN) rr rr rr rrrrrr)r#r6r7r8)r#r6rr rr?r7r@) __future__rtypingrnumpyrcellrrrpointr r r r r&r5__annotations__r>rrrpr'r%rwsa"""""""//////////????????????! 6  $ $ $ $ $ $ $ $ $ $ $ $ , , , , , , , , , 7 7     . . . .XXXX$2222222222r'