B \ @s.ddlZddlmZmZd ddZddZdS) N) ellipkinc ellipeincg?g?g?Fc Cs|dks|dks|dkr tdtjdtj|}ttj|||f |}ttj|||f|d}xxtdD]l}||||ddkr||d8<t||||||} d| krx||t| | dk8<qxWtj|d|d|d|d|d|d|d|d|df\} } } |sp| t |d| t |d| t |ddk} n4| 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. rz&Parameters a, b, and c must all be > 0)rrr) ValueErrornpZr_ZceilZfloorrangeZarangemaxZmgridfloat)abcZspacingZlevelsetoffsetZlowZhighZdimZnumxyzZarrr2lib/python3.7/site-packages/skimage/draw/draw3d.py ellipsoids(" rc Cs|dks|dks|dkr td|||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. rz&Parameters a, b, and c must all be > 0T)reverserrgUUUUUU?g?g@g?)rsortr ZpiZarcsinr ellip_Fellip_E) r rrabcZvolZphidmFEZsurfrrrellipsoid_statsCs$    r )rF) Znumpyr Z scipy.specialrrrrrr rrrrs =