B îq\$ã@s˜dZddlmZddlZddlmZddlm Z ddd gZ ddde d œd d„Z ej ej ejejejejejejfZe fd d„Zdddde d œdd „ZdS)z+ Built-in distribution-creation functions. é)ÚwarnN)Úunitsé)Ú DistributionÚnormalÚpoissonÚuniform)ÚstdÚvarÚivarÚclsc Ks¾t |¡}|dk r2|dkr*t |¡d}ntdƒ‚|dk rZ|dkrRt |¡d}ntdƒ‚|dkrltdƒ‚n t |¡}t ||¡j|f}|dtjftjj|Ž|dtjf}||f|Ž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 : ``cls``, usually `Distribution` The sampled Gaussian distribution. Ngà?z#normal cannot take both std and vargà¿z/normal cannot take both ivar and and std or varz(normal requires one of std, var, or ivar.)ÚnpÚ asanyarrayÚ ValueErrorZ broadcastÚshapeÚnewaxisÚrandomZrandn) Úcenterr r r Ú n_samplesr ÚkwargsÚ randshapeÚsamples©rú@lib/python3.7/site-packages/astropy/uncertainty/distributions.pyrs!   (cKsšd}t|dƒr d}t |j¡}n t |¡}|j|f}tj |dtjf|¡}|rŽ|jt j krht dƒn|jt kr„t d  |jt ¡ƒ||j}||f|ŽS)u. 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 : ``cls``, usually `Distribution` The sampled poisson distribution. FÚunitT.z­ADUs 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 {} was provided to poisson, which is not one of {}, and therefore suspect as a "counting" unit. Ensure you mean to use Poisson statistics.)Úhasattrr rÚvaluerrrrrÚuZadurÚ COUNT_UNITSÚformat)rrr rZhas_unitZ poissonarrrrrrrrJs        )ÚlowerÚupperrÚwidthr c Ksø|dkr:|dkr:t |¡}t |¡}|j|jkr€tdƒ‚nF|dkrx|dkrxt |¡}t |¡}||d}||d}ntdƒ‚|j|f}|jtƒkr®|jtƒkr®||}n(||dd…tjf}|dd…tjf}||tjj|d}||f|ŽS)aç 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 : ``cls``, usually `Distribution` The sampled uniform distribution. Nz+lower and upper must have consistent shapeséz^either upper/lower or center/width must be given to uniform - other combinations are not valid)Úsize)r rrrÚtuplerrr) r r!rr"rr rZnewshaperrrrrys$#         )Ú__doc__ÚwarningsrZnumpyr ZastropyrrZcorerÚ__all__rÚcountZelectronZdimensionless_unscaledZchanÚbinZvoxÚbitZbyterrrrrrrÚs    6$ /