U>~dZddlZddlZddlZddZdZdZdZddZ dd Z d Z dd Z ddZ dZdZddZddZdS)z6 util ==== This modules provides various utilities. Nc ddlm}t|dr3fd}|||d<|j||ddfdi|}|j|jfSt jjfd}|||d<|j ||fi|S) aAllows leastsq to take bounds if minimize function is missing. Args: errfunc0 (func): Function that should be used to compute errors. It should take an array of parameters as its first argument. x0 (list): Initial guess at parameter values. args (Optional[list]): Additional arguments that should be passed to the provided error function. (default = None) bounds (Optional[list]): (min,max) pairs placing bounds on what values are allowed for each parameter. (default = None) **exkw: Additional keywords arguments are passed to either `scipy.optimize.minimize` or `scipy.optimize.leastsq`. Returns: list: Best fit parameters. int: Error code for the fit. r)optimizeminimizec6t|g|RdzS)N)sum)xiargserrfunc0s J/var/www/html/ai-engine/env/lib/python3.11/site-packages/tesseract/util.pyerrfunczmyleastsq..errfunc&s(xx)5)))1,-- -Nargsboundscdkr{tt|D]^}|ddkr|||dkrcS|ddkr|||dkrcS_|g|RS)Nr)rangelen)r r idxrr lress r r zmyleastsq..errfunc-st|| Q==SSCc{1~t++#vc{1~0E0Ed{{{c{1~t++#vc{1~0E0Ed{{{8A%u%%% %r) scipyrhasattrrr successsys float_infomaxleastsq) r x0rrexkwrr resrs ` ` @r myleastsqr!s&x ##3 . . . . .  DT&\h111CCVCdCCuS[  ~! & & & & & & &  DT&\x22T222rctj|dddfdz|dddfdzz|dddfdzzS)zCalculates radius from position. Args: pos (np.ndarray): (N,3) Positions. Returns: r (np.ndarray): (N,) Radii. Nrrrnpsqrt)poss r pos2radr'8sN 73qqqs8Q;QQQqS1,s111Q3x{: ; ;;rc,dtjz|dzzS)zReturns spherical volume for provided radii. Args: r (np.ndarray): (N,) Radii. Returns: vol (np.ndarray): (N,) Volumes. gUUUUUU?@r$pi)rs r sphvolr-Cs 25=!R%  rc2d|zdtjzz dzS)zReturns spherical radii for provided volumes. Args: vol (np.ndarray): (N,) Volumes. Returns: r (np.ndarray): (N,) Radii. r)g@gUUUUUU?r*)vols r sphradr0Ns sFBruH  ''r?c>|d}tj||z|z S)a6Returns the circular velocity. Args: r (np.ndarray): (N,) Radii. menc (np.ndarray): (N,) Mass enclosed at each radius. G (Optional[float]): Gravitational constant in correct units. (default = 1) Returns: vcirc (np.ndarray): (N,) Circular velocities. Nr1r#)r,mencGs r calc_vcircr5Ys% yb! 71T6!8  rc|Itj|}tj|}tj|||}ntj|}|S)aCalculates the total mass enclosed by each particle. Args: m (np.ndarray): (N,) Masses of particles. sortby (Optional[np.ndarray]): Quantity that will be used to sort masses before summing. (The enclosed mass will be returned in the original particle order). If not provided, masses are summed in their current order. Returns: menc (np.ndarray): (N,) Mass enclosed. )r$argsortcumsum)msortbyidxsort idxsort_revr3s r calc_mencr=hsV*V$$j)) y7$$[1y|| Krcb||k}||t|z S)aReturns the density enclosed within rmax. Args: mass (np.ndarray): (N,) Masses of particles. r (np.ndarray): (N,) Radii. rmax (float): Radius which density is computed within. Returns: float: Average density enclosed within rmax. )rr-)massr,rmaxrs r calc_rhoencrAs* T6C 9==??6$<< ''rF?cjt|tjrt|t|krRt dt|dt|ztj|}nlt|tr5|rtdt|}n7tj|}n"td|ttj ||}t|tjrt|t|krRt dt|dt|z||zd|z |tj|zz}|r tj|}||}||fS|S|r||fS|dkr|Stj|}||S)aCalculate particle radii from volume information. Volumes are sorted then cummulatively summed to get enclosed volume at each particle. Radii are then calculated by assumming that each enclosed volume is a sphere. Returned radii are in the same particle order as the input volumes. Args: vol (np.ndarray): (N,) Volumes corresponding to each particle sortby (Optional[np.ndarray]): (N,) Array to sort volumes by before calculating radii. If not provided, vol is used. If set to True, volumes are assumed to be already sorted. outsort (Optional[bool]): If True, the array for sorting the particles is also output and the returned radii are already sorted. weightby (Optional[np.ndarray]): (N,) Values to weight volumes by after calculating radii. Sorting occurs before weighting. This is useful if you would like to weight by the real radius or another measure of radius. The returned radii are then the weighted arithmetic mean of the volume based radii (rvol) and the weightby array (i.e. weight*weightby + rvol*(1-weightby)) The resulting array is again sorted if outsort is True. (default = False) weight (Optional[float]): Value used to weight the weightby array by. Only used if weightby is not False. (default = 0.5) Returns: r (np.ndarray): (N,) Radii. Raises: Exception: If the `sortby` and `vol` arrays are not the same length. TypeError: If the `sortby` array is not an array or bool. Exception: If the `weightby` and `vol` arrays are not the same length. z(The sortby array (len={}) must have the z)same length as the volume array (len={}).rzInvalid sortby value: {}z*The weightby array (len={}) must have the r1T) isinstancer$ndarrayr Exceptionformatr7boolslice TypeErrorr0r8)r/r:outsortweightbyweightrradidxrevs r vol2radrPsF&$$ C v;;C FMMcRXkkZZGNNsSVxxXXYZZ Zj   F4 C  "#c((##CC*S//CC299&AABBB 3s8$$ % %C(2:&&  x==#c(( " "HOOPST\P]P]^^GNNsSVxxXXYZZ ZXoFC 3,@ @@  *S//Cc(Cs7NJ3w T>>JZ__Fv; rcd}|dkr||z}|dddfxx|zcc<|dkr||z}|dddfxx|zcc<||dzz}|S)aSqueezes 3D positions along y and z axes while preserving volume. Args: pos (np.ndarray): (N,3) Positions. yfact (Optional[float]): Factor that y direction is squeezed by. (0 < yfact < 1) zfact (Optional[float]): Factor that z direction is squeezed by. (0 < zfact < 1) Returns: pos (np.ndarray): (N,3) Transformed positions. r1rNrrgUUUUUUտ)r&yfactzfactsqufacts r squeeze_constvolrWs{G qyy AAAaC% qyy AAAaC%'F C JrcZd}|}tj|dz||dz|dzz zz }|S)a:Return y and z squeeze factors for given triaxiality and ellipticity. Args: t (float): triaxiality defined as (c**2 - b**2)/(c**2 - a**2) where a > b > c. e (float): Overall ellipticity defined as c/a. Returns: b (float): Factor to squeeze intermediate axis by. r1rr#)teacbs r triax2squeezer^s; A A 1q!Q$q!t)}$%%A Hrct|tr|}nt|tr$t|dd}nt|t r$t|dd}n\t|t r8t|dddd}nt|}|S)zWrites values to a string. Args: val (any): Any object that should be represented by a string. Returns: valstr (str): String representation of `val`. z, ,z: :)rDstrtuplereprreplacelistdict)valvalstrs r val2strrjs C  s C  tCyy'8'8c'B'B C  tCyy'8'8c'B'B C  tCyy'8'8c'B'B'J'J4PS'T'Tc& Mrc> |}t|trt|}t|ts5t |dkrd}n|dr.|dr|d}d}n|dkrd}n|dr+|d rd}n| s&|ddvrD|d d  r(t|t|krd }n!d}n|d r-|d r|d }d}n|d krd }n|ddvrd}n|dr&|drd|vrd}ncd|vrd}n\d}nY|dr|drd}n,|dr|drd}t|tr8|}|dksd|vr|d r,|d r|d }n$|dr,|dr|d}n|}n|dksd|vrt|}n|dksd |vrt|}n|d!ksd |vr2tt| dd"}nm|d#ksd|vr2tt| dd"}n1|d$ksd|vr%t| d%d"}n|d&ksd |vrd }n|d'ksd|vr|d(|dvrd)}nd*|dvrd+}ntd,|d|vrW|d d-d}g}|D]$}|t'|%t)|}nd|vrG|d d-d}g}|D]$}|t'|%nd|vrx|d d-d}i}i}g}|D]R}d.|vr3|d.\}} | ||<||9||d-xxd|zz cc<SnK d|vr$t| d%d"}nt|}n#t$r|}YnwxYw|S)/aRecovers variables from a string. Args: valstr (str): String to converted into an object. dtype (type): Type or string specifying type of the object `valstr` should be converted to. If not provided, we try to infer the type from the format of the string. (default = None) Returns: val (any): Object recreated from `valstr`. Raises: ValueError: If the data type is boolean, but 'true' or 'false' is not in `valstr`. rrb"nanfloatLlong)-+rNint'none.)truefalserH()r`rcjcomplex[]rf{}rgsfddoubleilr\ nr]rwTrxFz%Unrecognised string for bool type: {}rQra)striprDtyperbr startswithendswithlowerrstripisdigitrsrprnrer| ValueErrorrGsplitappendstr2valrc) ridtype valstr_striprhvallistivalstrval0key0ikeyivals r rrs"L%1s5zzu eC  |  a  u  $ $S ) ) l.C.CC.H.H %++C00LEE    ! !U * *'EE  " "3 ' ' L,?,?,D,D,L,L,N,N EE  ! ! # #  Q9(D(DVWVXVXIYIaIaIcIc(D<  D$6$666e>D  $ $S ) ) l.C.CC.H.H %++C00LEE    ! !V + +Vuu    $ $ * * , ,0@ @ @%%  $ $S ) ) l.C.CC.H.H f}}Geei%%  $ $S ) ) l.C.CC.H.H EE  $ $S ) ) l.C.CC.H.H E%6kkmm CZZ5E>>((-- #,2G2G2L2L # &&s++((-- #,2G2G2L2L # &&s++"## CZZ7--53F3Fss CZZ8--53F3Fss CZZ5--E,..s2667788CC CZZ6--U<//B778899CC CZZ9-- ,,S4455CC CZZ6--4ss CZZ6--L..s3399;;;;SSL..s3399;;;;SS !H!O!OPV!W!WXXX    2&,,S11GC" - - 77++,,,,c CC u__ 2&,,S11GC" - - 77++,,,, - u__ 2&,,S11GC"TBt" 0 0'>> ' c 2 2ID#DJKK%%%%bNNNCK/NNNN l""L00R8899,''   CCC  Js7X XXprTcddl}ddlm}|d|}|rt |t t fsd|dz z}|dkrt |}nht |dkr#t||krt |}n2t|t |z|krt |}t |t t frd}d|zd z|} |rd | vrd | vsd | vr_d | vr| d \} } d | vr| d \} } d | vr"td | n| | } d} | d| d | | z} d} d} ||z}|| |}| || S)a*Convert number to string, replacing punctuation. Args: numval (float,int): Numeric value to be converted. formstr (Optional[str]): Format string used for the conversion. If not provided, one is generated based on `nsig`. decrep (Optional[str]): String that '.'s are replaced with. (default = 'p') negrep (Optional[str]): String that '-'s are replaced with. (default = 'n') nsig (Optional[int]): Number of significan digits. (default = 3) trimzero (Optional[bool]): If true, remove trailing zeros in decimal. (default = True) Returns: str: String created from the number. Raises: Exception: If both 'e' and 'E' are in the generated string and `trimzero` is True. rN) maketransz.{}gg$@rrz{:rrvrZEz"Both e and E are in the string: {}r0z+ z.-) copystringrrGrDrsrpabsrrFdeepcopyr translate)numvalformstrdecrepnegrepnsigtrimzerorrprecvaldecstrdecbasdecexprtabitabftabtrantabs r num2strrs.KKK       d 3 3 ' 63t*55 'Qi  Q;;v;;FF6{{av;;((v;;FvF +,,w66v;;F&#d$$1cg 'M#  % %f - -F C6MM &==C6MMf}}FLL,=,=mfVf}}FLL,=,=mfVf}} D K KF S STTT%==((FF c cf} D D D Id4 G   GD ) ))r)NN)r1)N)FFFrB)rQrQ)NrrrT)__doc__osnumpyr$rr!r'r-r0r5r=rArPrWr^rjrrrSrr rs*  #3#3#3#3N < < < ! ! ! ( ( (2 ( ( (EEEER0   &*eeeeN>*>*>*>*>*>*r