IR-eLdZddlmZddlZddlmZddlm Z gdZ ddde dd Z ej ej ejejejejejejfZe fd Zdddde d d ZdS) z+ Built-in distribution-creation functions. )warnN)units) Distribution)normalpoissonuniform)stdvarivarclsc tj|}|)|tj|dz}ntd|)|tj|dz}ntd|tdtj|}tj||j|fz}|dtjftjj||dtjfzz}||fi|S)u Create a Gaussian/normal distribution. Parameters ---------- center : `~astropy.units.Quantity` The center of this distribution std : `~astropy.units.Quantity` or None The standard deviation/σ of this distribution. Shape must match and unit must be compatible with ``center``, or be `None` (if ``var`` or ``ivar`` are set). var : `~astropy.units.Quantity` or None The variance of this distribution. Shape must match and unit must be compatible with ``center``, or be `None` (if ``std`` or ``ivar`` are set). ivar : `~astropy.units.Quantity` or None The inverse variance of this distribution. Shape must match and unit must be compatible with ``center``, or be `None` (if ``std`` or ``var`` are set). n_samples : int The number of Monte Carlo samples to use with this distribution cls : class The class to use to create this distribution. Typically a `Distribution` subclass. Remaining keywords are passed into the constructor of the ``cls`` Returns ------- distr : `~astropy.uncertainty.Distribution` or object The sampled Gaussian distribution. The type will be the same as the parameter ``cls``. Ng?z#normal cannot take both std and vargz/normal cannot take both ivar and and std or varz(normal requires one of std, var, or ivar.)np asanyarray ValueError broadcastshapenewaxisrandomrandn) centerr r r n_samplesr kwargs randshapesampless Alib/python3.11/site-packages/astropy/uncertainty/distributions.pyrrsH]6 " "F  ;-$$+CCBCC C  ;-%%-CCNOO O {CDDDmC   S&))/9,>IsBJ")/9"=CO@T"TT  3w ! !& ! !!c d}t|drd}tj|j}ntj|}|j|fz}tj|dtjf|}|r]|jtj krtdn.|jtvr td|jdtd||jz}||fi|S) ux Create a Poisson distribution. Parameters ---------- center : `~astropy.units.Quantity` The center value of this distribution (i.e., λ). n_samples : int The number of Monte Carlo samples to use with this distribution cls : class The class to use to create this distribution. Typically a `Distribution` subclass. Remaining keywords are passed into the constructor of the ``cls`` Returns ------- distr : `~astropy.uncertainty.Distribution` or object The sampled Poisson distribution. The type will be the same as the parameter ``cls``. FunitT.zADUs were provided to poisson. ADUs are not strictly countunits because they need the gain to be applied. It is recommended you apply the gain to convert to e.g. electrons.zUnit z. was provided to poisson, which is not one of zY, and therefore suspect as a "counting" unit. Ensure you mean to use Poisson statistics.) hasattrrrvaluerrrrruadur COUNT_UNITS)rrr rhas_unit poissonarrrrs rrrXs.Hvv+]6<00 ]6**  I</Ii 3 ? ;YGGG( ;!%   O     [ + + 7 77777   FK' 3w ! !& ! !!r)lowerupperrwidthr c l|J|Htj|}tj|}|j|jkrtdnL|;|9tj|}tj|}||dz z }||dz z}ntd|j|fz}|jt kr|jt kr||z }n/||z ddtjf}|ddtjf}||tj|zz}||fi|S)a1 Create a Uniform distriution from the lower and upper bounds. Note that this function requires keywords to be explicit, and requires either ``lower``/``upper`` or ``center``/``width``. Parameters ---------- lower : array-like The lower edge of this distribution. If a `~astropy.units.Quantity`, the distribution will have the same units as ``lower``. upper : `~astropy.units.Quantity` The upper edge of this distribution. Must match shape and if a `~astropy.units.Quantity` must have compatible units with ``lower``. center : array-like The center value of the distribution. Cannot be provided at the same time as ``lower``/``upper``. width : array-like The width of the distribution. Must have the same shape and compatible units with ``center`` (if any). n_samples : int The number of Monte Carlo samples to use with this distribution cls : class The class to use to create this distribution. Typically a `Distribution` subclass. Remaining keywords are passed into the constructor of the ``cls`` Returns ------- distr : `~astropy.uncertainty.Distribution` or object The sampled uniform distribution. The type will be the same as the parameter ``cls``. Nz+lower and upper must have consistent shapesz^either upper/lower or center/width must be given to uniform - other combinations are not valid)size)rrrrtuplerrr ) r'r(rr)rr rnewshapers rr r sMX~%- e$$ e$$ ;%+ % %JKK K & 5=v&& e$$"" <   {i\)H {egg%+"8"8 2: .aaam$ebi//X/>>>>G 3w ! !& ! !!r)__doc__warningsrnumpyrastropyrr"corer__all__rcountelectrondimensionless_unscaledchanbinvoxbitbyter$rr rrr>s  * * *T\8"8"8"8"8"xGJFEEEF  $01"1"1"1"l    D"D"D"D"D"D"D"r