ó šßÈ[c@`sŒdZddlmZmZmZmZddlmZmZddl Z ddl m Z ddl mZmZddlmZmZdd lmZdd lmZdd lmZd d ddddgZdefd„ƒYZdefd„ƒYZdefd„ƒYZd„Z d„Z!d„Z"d„Z#d„Z$d„Z%d„Z&d„Z'd „Z(ddd!„Z*d"„Z+ddd#„Z,dS($u? This module provides utility functions for the models package i(tabsolute_importtunicode_literalstdivisiontprint_function(tdequetMutableMappingNi(tsix(trangetzip(t isiterabletcheck_broadcast(t NUMPY_LT_1_14(t signature(tunitsuExpressionTreeu AliasDictucheck_broadcastupoly_map_domainucombuellipse_extenttExpressionTreecB`s‰eZdddgZddd„Zd„Zd„Zed„ƒZd„Z d„Z d „Z dd dd „Z d „Z dd „ZRS(ulefturightuvaluecC`sF||_||_|dk r9||kr9|jƒ}n||_dS(N(tvaluetlefttNonetcopytright(tselfRRR((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt__init__s   c`st‡fd†ˆjDƒƒS(Nc3`s$|]}|tˆ|ƒfVqdS(N(tgetattr(t.0tslot(R(s5lib/python2.7/site-packages/astropy/modeling/utils.pys ,s(tdictt __slots__(R((Rs5lib/python2.7/site-packages/astropy/modeling/utils.pyt __getstate__)scC`s1x*|jƒD]\}}t|||ƒq WdS(N(titemstsetattr(RtstateRR((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt __setstate__.scC`s|jdko|jdkS(N(RRR(R((s5lib/python2.7/site-packages/astropy/modeling/utils.pytisleaf2scc`sut|gƒ}x_|rp|jƒ}|V|jdk rK|j|jƒn|jdk r|j|jƒqqWdS(N(RtpopRRtappendR(Rtstacktnode((s5lib/python2.7/site-packages/astropy/modeling/utils.pyttraverse_preorder6s  cc`sktƒ}|}xU|s$|dk rf|dk rI|j|ƒ|j}q|jƒ}|V|j}qWdS(N(RRR"RR!R(RR#R$((s5lib/python2.7/site-packages/astropy/modeling/utils.pyttraverse_inorderAs     cc`sãt|gƒ}d}xÇ|rÞ|d}|dksR||jksR||jkr™|jdk rt|j|jƒqÕ|jdk rÕ|j|jƒqÕn<|j|krÊ|jdk rÊ|j|jƒn |jƒV|}qWdS(Niÿÿÿÿ(RRRRR"R!(RR#tlastR$((s5lib/python2.7/site-packages/astropy/modeling/utils.pyttraverse_postorderMs  * icC`sÊtƒ}|dkr!d„}n|dkr6d}nd}x}|jƒD]o}|jr„|j|||jƒ|fƒ|d7}qI||j}t|ƒdkr©qIn|jƒ} |jƒ} g} xz| | fD]l} | ddkrú| j| ƒqÔ| \} } || krÔ|dks*| |krÔ| j| | fƒqÔqÔWt| ƒdkr‰| \} } |j|| d| dƒdfƒqIt| ƒdkr«|j| ƒqI|j| ƒqIW|jƒdS(uFEvaluate the expression represented by this tree. ``Operators`` should be a dictionary mapping operator names ('tensor', 'product', etc.) to a function that implements that operator for the correct number of operands. If given, ``getter`` is a function evaluated on each *leaf* node's value before applying the operator between them. This could be used, for example, to operate on an attribute of the node values rather than directly on the node values. The ``getter`` is passed both the index of the leaf (a count starting at 0 that is incremented after each leaf is found) and the leaf node itself. The ``start`` and ``stop`` arguments allow evaluating a sub-expression within the expression tree. TODO: Document this better. cS`s|S(N((tidxR((s5lib/python2.7/site-packages/astropy/modeling/utils.pyttsiiiiÿÿÿÿN( RRR(R R"RtlenR!textend(Rt operatorstgettertstarttstopR#tleaf_idxR$toperatorRRtoperandstoperandR)((s5lib/python2.7/site-packages/astropy/modeling/utils.pytevaluate]s:           $ ' cC`syg}xL|j|jfD]8}t|tƒrD|j|jƒƒq|j|ƒqW|j|jd|dd|dƒS(NRiRi(RRt isinstanceRR"Rt __class__R(Rtchildrentchild((s5lib/python2.7/site-packages/astropy/modeling/utils.pyR±s c C`sId}tƒ}|dkr'd„}nx|jƒD]}|jri|j|||ƒƒ|d7}q4n||j}|jƒ}|jƒ}|jdk rÒ|jj rÒ||jj|krÒdj|ƒ}n|j dk r|j j r||j j|krdj|ƒ}n|jdj ||j|fƒƒq4Wdj |ƒS(NicS`s dj|ƒS(Nu[{0}](tformat(titl((s5lib/python2.7/site-packages/astropy/modeling/utils.pyR*Äsiu({0})u u( RRR(R R"RR!RR:Rtjoin( Rtoperator_precedencet format_leafR1R3R$t oper_orderRR((s5lib/python2.7/site-packages/astropy/modeling/utils.pytformat_expression¿s(        &N(t__name__t __module__RRRRRtpropertyR R%R&R(R5RRA(((s5lib/python2.7/site-packages/astropy/modeling/utils.pyRs   T t AliasDictcB`sSeZdZeZd„Zd„Zd„Zd„Zd„Z d„Z d„Z RS(uA 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 cC`s+||_|jƒ|_t|ƒ|_dS(N(t_parentt _store_typet_storeRt_aliases(Rtparenttaliases((s5lib/python2.7/site-packages/astropy/modeling/utils.pyR's cC`sS||jkrHy|j|j|SWqHtk rDt|ƒ‚qHXn|j|S(N(RIRFtKeyErrorRH(Rtkey((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt __getitem__,s  cC`s7||jkr&||j|j|§sN(i(ii( tn_inputsR tnptshapet ValueErrorR:tnameR+Rf(R`tmodelt bounding_boxtnd((s5lib/python2.7/site-packages/astropy/modeling/utils.pytvalidate€s     N( RBRCR[RR_R^Ret classmethodRp(((s5lib/python2.7/site-packages/astropy/modeling/utils.pyR\cs   c`s‡‡‡fd†S(u 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,) c`s5t‡fd†tˆ||ƒˆ||ƒƒDƒƒS(Nc3`s$|]\}}ˆ||ƒVqdS(N((Rtxty(toper(s5lib/python2.7/site-packages/astropy/modeling/utils.pys Ás(RfR(tinputstparams(tftgRt(s5lib/python2.7/site-packages/astropy/modeling/utils.pyR*Às((RtRwRx((RwRxRts5lib/python2.7/site-packages/astropy/modeling/utils.pytmake_binary_operator_evalªscC`stj|dtjƒ}tj|dtjƒ}|d|d|d|d}|d|d|d|d|d|d}|||S(u Map domain into window by shifting and scaling. Parameters ---------- oldx : array original coordinates domain : list or tuple of length 2 function domain window : list or tuple of length 2 range into which to map the domain tdtypeii(Ritarraytfloat64(toldxtdomaintwindowtscltoff((s5lib/python2.7/site-packages/astropy/modeling/utils.pytpoly_map_domainÅs "2cC`sl||ks$|dks$|dkr(dSd}x7tt|||ƒƒD]}||||d}qHW|S(uÅ The number of combinations of N things taken k at a time. Parameters ---------- N : int, array Number of things. k : int, array Number of elements taken. ii(Rtmin(tNtktvaltj((s5lib/python2.7/site-packages/astropy/modeling/utils.pytcombÙs $ cC`sJtr dnd}tj|d|dtƒ}djd„|jƒDƒƒS(uR Represents a multi-dimensional Numpy array flattened onto a single line. u,u, t separatortsuppress_smallu cs`s|]}|jƒVqdS(N(tstrip(RR<((s5lib/python2.7/site-packages/astropy/modeling/utils.pys ós(R Rit array2stringtTrueR=t splitlines(R{tseptr((s5lib/python2.7/site-packages/astropy/modeling/utils.pytarray_repr_onelineíscC`sLt|ƒj|ƒrDtd„|Dƒƒ}td„|Dƒƒ}n||S(uR 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). cs`s|]}|dVqdS(u0N((RR<((s5lib/python2.7/site-packages/astropy/modeling/utils.pys scs`s|]}|dVqdS(u1N((RR((s5lib/python2.7/site-packages/astropy/modeling/utils.pys s(tsett intersectionRf(RR((s5lib/python2.7/site-packages/astropy/modeling/utils.pytcombine_labelsös cC`stj| tj|ƒ|ƒ}|tj|ƒtj|ƒ|tj|ƒtj|ƒ}tj||tj|ƒƒ}|tj|ƒtj|ƒ|tj|ƒtj|ƒ}t|tjƒsßt|tjƒrûtjtj||gƒƒStj||gƒSdS(u  Calculates the extent of a box encapsulating a rotated 2D ellipse. Parameters ---------- a : float or `~astropy.units.Quantity` Major axis. b : float or `~astropy.units.Quantity` Minor axis. theta : float or `~astropy.units.Quantity` Rotation angle. If given as a floating-point value, it is assumed to be in radians. 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() N( Ritarctan2ttantcostsinR6tutQuantitytabs(tatbtthetatttdxtdy((s5lib/python2.7/site-packages/astropy/modeling/utils.pytellipse_extents4 >>$cC`s•t|ƒ}g}g}xp|jjƒD]_}|j|j|jfkrXtdƒ‚n|j|jkrz|j |ƒq(|j |ƒq(W||fS(uò 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 u)Signature must not have *args or **kwargs( R t parameterstvaluestkindtVAR_POSITIONALt VAR_KEYWORDRktdefaulttemptyR"(tfunctsigRuRvtparam((s5lib/python2.7/site-packages/astropy/modeling/utils.pytget_inputs_and_paramsGs cC`s.|jdkr|j|S|jj|ƒSdS(N(tunitRRtquantitytto(t parameterR®((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt_parameter_with_unitds cC`s%|dkr|S||j|ƒSdS(N(RR°(Rtold_unittnew_unit((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt_parameter_without_unitks cC`s…i}xx|D]p}g||<|dk rM||krM||j||ƒn|dk r ||kr ||j||ƒq q W|S(N(RR,(tkeysteq1teq2teqRM((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt_combine_equivalency_dictrs  cC`s3t|tjƒr"|jtjƒStj|ƒSdS(u Convert ``value`` to radian. N(R6R™RšR°tradRitdeg2rad(R((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt _to_radian€scC`s.|dk r||j|ƒStj|ƒSdS(u3 Convert value with ``raw_unit`` to ``orig_unit``. N(RR°Ritrad2deg(Rtraw_unitt orig_unit((s5lib/python2.7/site-packages/astropy/modeling/utils.pyt _to_orig_unitˆs (-R[t __future__RRRRt collectionsRRtnumpyRitexternRtextern.six.movesRRtutilsR R t utils.compatR tutils.compat.funcsigsR tR R™t__all__tobjectRRERfR\RyR‚RˆR‘R”R¢R­R²RµRRºR½RÁ(((s5lib/python2.7/site-packages/astropy/modeling/utils.pyts4"  Á‡G     @