:vjdZddlZddlmZmZddlmZmZ ddl m Z ddl m Z d Zd d Zd Zy#e $r4ZddlmZej eZ ej eZ YdZ[?dZ[wwxYw) zp nsphere.py -------------- Functions for fitting and minimizing nspheres: circles, spheres, hyperspheres, etc. N)convexutil)logtol)spatial)leastsq) exceptionsc tj|}|jd}tj|dj}||z |z }t |\}}}||z dzj djdz|z}||z|z}|dkr||fStj|d} t|jt|zd z}|d kDrttjj|j|d jd} | j%} tj&| | |z} |j| |z|z} | |kDr||fS| | fS#t$rotj d tj"|jD cgc](} || z dzj dj*ncc} wc} } YwxYw)aT Compute the minimum n- sphere for a mesh or a set of points. Uses the fact that the minimum n- sphere will be centered at one of the vertices of the furthest site voronoi diagram, which is n*log(n) but should be pretty fast due to using the scipy/qhull implementations of convex hulls and voronoi diagrams. Parameters ---------- obj : (n, d) float or trimesh.Trimesh Points or mesh to find minimum bounding nsphere Returns ---------- center : (d,) float Center of fitted n- sphere radius : float Radius of fitted n-sphere raxisrg?gư>T) furthest_site geA sqeuclidean)metricz*MemoryError: falling back to slower check!)r hull_pointsminnpptp fit_nspheresummaxrVoronoilenvertices MemoryErrordistancecdistrwarningarrayargminsqrt)objpoints points_origin points_scalefit_Cfit_Rfit_Evoronoimemory_estimateradii_2v radii_idxradius_vcenter_vs `/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/nsphere.pyminimum_nspherer3s2   $F JJAJ&M66&q)--/L}$ 4F &f-E5%~!# ( (a ( 0 4 4 6# = ME \ !] 2E t|e| oofD9G  g../#f+=A S  ""((   f]) #1#+  Iwwwy)*\9H  +l:mKH%e| X %   @A((.residualss:666F?q0$78<<>CM9::rr g:0yE>)xtol)rrzLeast square fit failed!) r asanyarrayfloat64r:shapemeanr ValueErrorrrow_normr) r%priorr;guess center_result return_coderadiiradiuserrorr:s ` @r2rrts*]]6 4F 776<<? #D; }  # e$!(E!EM;,&344 MM&=0 1E ZZ\F FF5ME &% ''r<cJt|\}}}|tjk}|S)z Check if a list of points is an nsphere. Parameters ----------- points : (n, dimension) float Points in space Returns ----------- check : bool True if input points are on an nsphere )rrmerge)r%_center_radiusrLchecks r2 is_nsphererRs(*&1GWe CII E Lr<)N)__doc__numpyrrr constantsrrscipyrscipy.optimizer BaseExceptionEr ExceptionWrapperr3rrRr<r2r]sl -&Wt.(bg-)j))!,G)j))!,G -s /A(*A##A(