U ˆ±ËhPã@s†ddlmZddlZddlmZmZmZddlZ erTddl m Z ddl m Z mZmZddœd d „Zd dœd d „ZdAddddddœdd„ZdBdddddœdd„Zddœdddddddœdd „Zddœdddddddœd!d"„ZdCdddddd#œd$d%„ZdDdddd&d'œd(d)„ZdEd*ddd+œd,d-„Zdde jdfd&d.dd/dd&d0œd1d2„ZdFd3ddd4œd5d6„ZdGdd8d9d:ddd;œdœd?d@„ZdS)Ié)Ú annotationsN)Ú TYPE_CHECKINGÚAnyÚSequence)Ú DTypeLikeé)Ú FloatNumTypeÚ IntNumTypeÚNumTypeÚint)ÚreturncCs t dd¡S)Nrlÿÿ)Ú py_randomÚrandint©rrú?/tmp/pip-unpacked-wheel-e8onvpoz/albumentations/random_utils.pyÚget_random_seedsrznp.random.RandomStatecCstj tƒ¡S©N)ÚnpÚrandomZ RandomStaterrrrrÚget_random_statesrççð?r ztuple[int, ...] | int | Noneznp.random.RandomState | Noner)ÚlowÚhighÚsizeÚ random_stater cCs|dkrtƒ}| |||¡Sr)rÚuniform)rrrrrrrrsrçà?)ÚalphaÚbetarr cCs|dkrtƒ}| ||¡Sr)rr)rrrrrrr!sr)rrz np.ndarray)Úd0Úd1ÚmorerÚkwargsr cOs"|dkrtƒ}|j||f|ž|ŽSr)rÚrand©r r!rr"r#rrrr$+sr$cOs"|dkrtƒ}|j||f|ž|ŽSr)rÚrandnr%rrrr&7sr&)ÚlocÚscalerrr cCs|dkrtƒ}| |||¡Sr)rÚnormal)r'r(rrrrrr)Csr)r )Úlamrrr cCs|dkrtƒ}| ||¡Sr)rÚpoisson)r*rrrrrr+Nsr+z"int | Sequence[float] | np.ndarray)Úxrr cCs|dkrtƒ}| |¡Sr)rÚ permutation)r,rrrrr-Xsr-zIntNumType | Noner)rrrÚdtyperr cCs|dkrtƒ}| ||||¡Sr)rr)rrrr.rrrrrasrzNumType | None)rrr cCs|dkrtƒ}| |¡Sr)rr)rrrrrrmsrTztuple[int, int] | int | NoneÚboolz#Sequence[float] | np.ndarray | None)ÚarÚreplaceÚprr cCs|dkrtƒ}| ||||¡Sr)rÚchoice)r0rr1r2rrrrr3ssr3)r0rr cCs|dkrtƒ}| |¡|S)a•Shuffles an array in-place, using a specified random state or creating a new one if not provided. Args: a (np.ndarray): The array to be shuffled. random_state (Optional[np.random.RandomState], optional): The random state used for shuffling. Defaults to None. Returns: np.ndarray: The shuffled array (note: the shuffle is in-place, so the original array is modified). N)rÚshuffle)r0rrrrr4s  r4)rrNN)rrN)rrNN)rNN)N)NN)NTNN)N)Ú __future__rrr ÚtypingrrrZnumpyrZ numpy.typingrZ core.typesrr r rrrrr$r&r)r+r-Zint32rr3r4rrrrÚsZ  ü ýüü ü ý þ û ûþ