a ߙfb4@sdZddlmZddlmZddlmZddlZddl Z ddl m Z ddl mZdgZgd Zd Zed eGd ddeZd dZddZddZddZddZddZddZddZd'ddZdd Zd(d!d"ZGd#d$d$eZGd%d&d&eZdS))z@ This module provides utility functions for the models package. )UserDict)MutableMapping) signatureN)units) deprecated AliasDict)rpoly_map_domaincombellipse_extentzY AliasDict is deprecated because it no longer serves a function anywhere inside astropy. z5.0c@sLeZdZdZeZddZddZddZdd Z d d Z d d Z ddZ dS)ra= Creates a `dict` like object that wraps an existing `dict` or other `MutableMapping`, along with a `dict` of *key aliases* that translate between specific keys in this dict to different keys in the underlying dict. In other words, keys that do not have an associated alias are accessed and stored like a normal `dict`. However, a key that has an alias is accessed and stored to the "parent" dict via the alias. Parameters ---------- parent : dict-like The parent `dict` that aliased keys and accessed from and stored to. aliases : dict-like Maps keys in this dict to their associated keys in the parent dict. Examples -------- >>> parent = {'a': 1, 'b': 2, 'c': 3} >>> aliases = {'foo': 'a', 'bar': 'c'} >>> alias_dict = AliasDict(parent, aliases) >>> alias_dict['foo'] 1 >>> alias_dict['bar'] 3 Keys in the original parent dict are not visible if they were not aliased: >>> alias_dict['b'] Traceback (most recent call last): ... KeyError: 'b' Likewise, updates to aliased keys are reflected back in the parent dict: >>> alias_dict['foo'] = 42 >>> alias_dict['foo'] 42 >>> parent['a'] 42 However, updates/insertions to keys that are *not* aliased are not reflected in the parent dict: >>> alias_dict['qux'] = 99 >>> alias_dict['qux'] 99 >>> 'qux' in parent False In particular, updates on the `AliasDict` to a key that is equal to one of the aliased keys in the parent dict does *not* update the parent dict. For example, ``alias_dict`` aliases ``'foo'`` to ``'a'``. But assigning to a key ``'a'`` on the `AliasDict` does not impact the parent: >>> alias_dict['a'] = 'nope' >>> alias_dict['a'] 'nope' >>> parent['a'] 42 cCs||_||_t||_dSN)_parent _store_type_storedict_aliases)selfparentaliasesr5lib/python3.9/site-packages/astropy/modeling/utils.py__init__gs zAliasDict.__init__cCsB||jvr8z|j|j|WSty6t|Yn0|j|Sr rr KeyErrorrrkeyrrr __getitem__ls   zAliasDict.__getitem__cCs*||jvr||j|j|<n ||j|<dSr )rr r)rrvaluerrr __setitem__us zAliasDict.__setitem__cCsF||jvr:z|j|j|=WqBty6t|YqB0n|j|=dSr rrrrr __delitem__{s   zAliasDict.__delitem__ccs:|jD]\}}||jvr |Vq |jD] }|Vq*dS)z First iterates over keys from the parent dict (if the aliased keys are present in the parent), followed by any keys in the local store. N)ritemsr r)rraliasrrr__iter__s   zAliasDict.__iter__cCsttt|Sr )lenlistiterrrrr__len__szAliasDict.__len__cCsH|}|jD] \}}||jvr|j|||<q||jt|Sr )r rrr updaterrepr)rZ store_copyrr rrr__repr__s   zAliasDict.__repr__N) __name__ __module__ __qualname____doc__rr rrrrr!r&r)rrrrrsC   csfddS)a Given a binary operator (as a callable of two arguments) ``oper`` and two callables ``f`` and ``g`` which accept the same arguments, returns a *new* function that takes the same arguments as ``f`` and ``g``, but passes the outputs of ``f`` and ``g`` in the given ``oper``. ``f`` and ``g`` are assumed to return tuples (which may be 1-tuples). The given operator is applied element-wise to tuple outputs). Example ------- >>> from operator import add >>> def prod(x, y): ... return (x * y,) ... >>> sum_of_prod = make_binary_operator_eval(add, prod, prod) >>> sum_of_prod(3, 5) (30,) cs(tfddt||||DS)Nc3s|]\}}||VqdSr r).0xy)operrr z>make_binary_operator_eval....)tuplezip)inputsparamsfgr1rrsz+make_binary_operator_eval..r)r1r9r:rr8rmake_binary_operator_evalsr<cCstj|tjd}tj|tjd}|jdks4|jdkr.)r@Z array2stringjoin splitlines)rArrrrarray_repr_onelinesrScCs:t||r2tdd|D}tdd|D}||S)aR For use with the join operator &: Combine left input/output labels with right input/output labels. If none of the labels conflict then this just returns a sum of tuples. However if *any* of the labels conflict, this appends '0' to the left-hand labels and '1' to the right-hand labels so there is no ambiguity). css|]}|dVqdS)0NrrNrrrr2r3z!combine_labels..css|]}|dVqdS)1Nr)r.rRrrrr2r3)set intersectionr4)leftrightrrrcombine_labelss rZcCst| t||}|t|t||t|t|}t||t|}|t|t||t|t|}t|tjst|tjrtt||gSt||gS)a Calculates the half size of a box encapsulating a rotated 2D ellipse. Parameters ---------- a : float or `~astropy.units.Quantity` The ellipse semimajor axis. b : float or `~astropy.units.Quantity` The ellipse semiminor axis. theta : float or `~astropy.units.Quantity` ['angle'] The rotation angle as an angular quantity (`~astropy.units.Quantity` or `~astropy.coordinates.Angle`) or a value in radians (as a float). The rotation angle increases counterclockwise. Returns ------- offsets : tuple The absolute value of the offset distances from the ellipse center that define its bounding box region, ``(dx, dy)``. Examples -------- .. plot:: :include-source: import numpy as np import matplotlib.pyplot as plt from astropy.modeling.models import Ellipse2D from astropy.modeling.utils import ellipse_extent, render_model amplitude = 1 x0 = 50 y0 = 50 a = 30 b = 10 theta = np.pi / 4 model = Ellipse2D(amplitude, x0, y0, a, b, theta) dx, dy = ellipse_extent(a, b, theta) limits = [x0 - dx, x0 + dx, y0 - dy, y0 + dy] model.bounding_box = limits image = render_model(model) plt.imshow(image, cmap='binary', interpolation='nearest', alpha=.5, extent = limits) plt.show() ) r@Zarctan2ZtanZcosZsin isinstanceuQuantityabs)abZthetatZdxZdyrrrr s400r cCsdt|}g}g}|jD]@}|j|j|jfvr8td|j|jkrP| |q| |q||fS)z Given a callable, determine the input variables and the parameters. Parameters ---------- func : callable Returns ------- inputs, params : tuple Each entry is a list of inspect.Parameter objects z)Signature must not have *args or **kwargs) r parametersvalueskindZVAR_POSITIONALZ VAR_KEYWORDrCdefaultemptyappend)funcZsigr6r7paramrrrget_inputs_and_paramsEs   rjcCs^i}|D]P}g||<|dur6||vr6|||||dur||vr||||q|Sr )extend)keysZeq1Zeq2eqrrrr_combine_equivalency_dictbsrncCs"t|tjr|tjSt|S)z Convert ``value`` to radian. )r[r\r]toZradr@Zdeg2radrDrrr _to_radianps  rpcCs |dur|||St|S)z3 Convert value with ``raw_unit`` to ``orig_unit``. N)ror@Zrad2deg)rZraw_unitZ orig_unitrrr _to_orig_unitwsrqcs,eZdZdZfddZfddZZS)_ConstraintsDictzv Wrapper around UserDict to allow updating the constraints on a Parameter when the dictionary is updated. csD||_||_i}|jD]}t||}t||||<qt|dSr )_modelconstraint_typeZ param_namesgetattrsuperr)rZmodelrtcnameri __class__rrrs  z_ConstraintsDict.__init__cs,t||t|j|}t||j|dSr )rvrrurssetattrrt)rrrJriryrrrs z_ConstraintsDict.__setitem__)r*r+r,r-rr __classcell__rrryrrr~s rrcsJeZdZdZdiffdd ZfddZddZd d Zd d ZZ S) _SpecialOperatorsDicta Wrapper around UserDict to allow for better tracking of the Special Operators for CompoundModels. This dictionary is structured so that one cannot inadvertently overwrite an existing special operator. Parameters ---------- unique_id: int the last used unique_id for a SPECIAL OPERATOR special_operators: dict a dictionary containing the special_operators Notes ----- Direct setting of operators (`dict[key] = value`) into the dictionary has been deprecated in favor of the `.add(name, value)` method, so that unique dictionary keys can be generated and tracked consistently. rcst|||_dSr )rvr _unique_id)rZ unique_idZspecial_operatorsryrrrs z_SpecialOperatorsDict.__init__cs,||vrtd|dnt||dS)NzSpecial operator "z" already exists)rCrvrrrrJryrr _set_valuesz _SpecialOperatorsDict._set_valuecCs|||ttddS)Nz Special operator dictionary assignment has been deprecated. Please use `.add` instead, so that you can capture a unique key for your operator. )rwarningswarnDeprecationWarningrrrrrs z!_SpecialOperatorsDict.__setitem__cCs|jd7_|jS)Nr?)r~r%rrr_get_unique_idsz$_SpecialOperatorsDict._get_unique_idcCs||f}||||S)a Adds a special operator to the dictionary, and then returns the unique key that the operator is stored under for later reference. Parameters ---------- operator_name: str the name for the operator operator: function the actual operator function which will be used Returns ------- the unique operator key for the dictionary `(operator_name, unique_id)` )rr)rZ operator_nameoperatorrrrradds  z_SpecialOperatorsDict.add) r*r+r,r-rrrrrr|rrryrr}s   r})NN)NN) r- collectionsrcollections.abcrinspectrZnumpyr@rZastropyrr\Zastropy.utils.decoratorsrZ__doctest_skip____all__Zdeprecation_msgrr<rrEr rSrZr rjrnrprqrrr}rrrrs4     ?