o hs @s.ddlZddlmZmZd ddZddZdS) N) ellipkinc ellipeinc?rrFc Cs|dks |dks |dkrtdtjdtj|}ttj|||f |}ttj|||f|d}tdD]6}||||ddkrQ||d8<t||||||} d| vrq||t| | dk8<q;tj|d|d|d|d|d|d|d|d|df\} } } |s| t |d| t |d| t |ddk} | S| t |d| t |d| t |dd} | S)a Generates ellipsoid with semimajor axes aligned with grid dimensions on grid with specified `spacing`. Parameters ---------- a : float Length of semimajor axis aligned with x-axis. b : float Length of semimajor axis aligned with y-axis. c : float Length of semimajor axis aligned with z-axis. spacing : tuple of floats, length 3 Spacing in (x, y, z) spatial dimensions. levelset : bool If True, returns the level set for this ellipsoid (signed level set about zero, with positive denoting interior) as np.float64. False returns a binarized version of said level set. Returns ------- ellip : (N, M, P) array Ellipsoid centered in a correctly sized array for given `spacing`. Boolean dtype unless `levelset=True`, in which case a float array is returned with the level set above 0.0 representing the ellipsoid. r&Parameters a, b, and c must all be > 0)rrr) ValueErrornpr_ceilfloorrangearangemaxmgridfloat)abcspacinglevelsetoffsetlowhighdimnumxyzarrr"N/var/www/html/scripts/venv/lib/python3.10/site-packages/skimage/draw/draw3d.py ellipsoidsB  r$c Cs|dks |dks |dkrtd|||g}|jdd|d}|d}|d}dtj|||}td|d|d d }t|d|dd }|d|d|dt|d|d|d}t||}t||} dtj|d||d||||| } || fS) a Calculates analytical surface area and volume for ellipsoid with semimajor axes aligned with grid dimensions of specified `spacing`. Parameters ---------- a : float Length of semimajor axis aligned with x-axis. b : float Length of semimajor axis aligned with y-axis. c : float Length of semimajor axis aligned with z-axis. Returns ------- vol : float Calculated volume of ellipsoid. surf : float Calculated surface area of ellipsoid. rrT)reverserr gUUUUUU?rg@g?)r sortr piarcsinrellip_Fellip_E) rrrabcvolphidmFEsurfr"r"r#ellipsoid_statsBs*     r3)rF) numpyr scipy.specialrr)rr*r$r3r"r"r"r#s  =