σ šίΘ[c@sbdZddlmZddlmZddgZdefd„ƒYZdefd„ƒYZd S( s  Special models useful for complex compound models where control is needed over which outputs from a source model are mapped to which inputs of a target model. i(t FittableModeli(trangetMappingtIdentitycBsteZdZeZdddd„Zed„ƒZed„ƒZ ed„ƒZ d„Z d„Z ed„ƒZ RS( s¬ Allows inputs to be reordered, duplicated or dropped. Parameters ---------- mapping : tuple A tuple of integers representing indices of the inputs to this model to return and in what order to return them. See :ref:`compound-model-mappings` for more details. n_inputs : int Number of inputs; if `None` (default) then ``max(mapping) + 1`` is used (i.e. the highest input index used in the mapping). name : str, optional A human-friendly name associated with this model instance (particularly useful for identifying the individual components of a compound model). meta : dict-like Free-form metadata to associate with this model. Raises ------ TypeError Raised when number of inputs is less that ``max(mapping)``. Examples -------- >>> from astropy.modeling.models import Polynomial2D, Shift, Mapping >>> poly1 = Polynomial2D(1, c0_0=1, c1_0=2, c0_1=3) >>> poly2 = Polynomial2D(1, c0_0=1, c1_0=2.4, c0_1=2.1) >>> model = (Shift(1) & Shift(2)) | Mapping((0, 1, 0, 1)) | (poly1 & poly2) >>> model(1, 2) # doctest: +FLOAT_CMP (17.0, 14.2) cCs¨|dkr8td„tt|ƒdƒDƒƒ|_ntd„t|ƒDƒƒ|_td„tt|ƒƒDƒƒ|_||_tt |ƒj d|d|ƒdS(Ncss|]}dt|ƒVqdS(txN(tstr(t.0tidx((s8lib/python2.7/site-packages/astropy/modeling/mappings.pys 4sicss|]}dt|ƒVqdS(RN(R(RR((s8lib/python2.7/site-packages/astropy/modeling/mappings.pys 7scss|]}dt|ƒVqdS(RN(R(RR((s8lib/python2.7/site-packages/astropy/modeling/mappings.pys 9stnametmeta( tNonettupleRtmaxt_inputstlent_outputst_mappingtsuperRt__init__(tselftmappingtn_inputsRR ((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyR2s  # % cCs|jS(sU The name(s) of the input variable(s) on which a model is evaluated. (R (R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pytinputs=scCs|jS(s*The name(s) of the output(s) of the model.(R(R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pytoutputsEscCs|jS(s,Integers representing indices of the inputs.(R(R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyRKscCs9|jdkrdj|jƒSdj|j|jƒSdS(Nss(RR tformatR(R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyt__repr__Qscs”tˆƒ|jkrZ|jdk r-|jnd}tdj||jtˆƒƒƒ‚nt‡fd†|jDƒƒ}|jdkr|dS|S(NRs{0} expects {1} inputs; got {2}c3s|]}ˆ|VqdS(N((RR(targs(s8lib/python2.7/site-packages/astropy/modeling/mappings.pys ^sii( RRRR t TypeErrorRR Rt n_outputs(RRRtresult((Rs8lib/python2.7/site-packages/astropy/modeling/mappings.pytevaluateWs cs€y)t‡fd†tˆjƒDƒƒ}Wn)tk rTtdjˆjƒƒ‚nXˆj|ƒ}ˆj|_ ˆj |_|S(s) A `Mapping` representing the inverse of the current mapping. Raises ------ `NotImplementedError` An inverse does no exist on mappings that drop some of its inputs (there is then no way to reconstruct the inputs that were dropped). c3s!|]}ˆjj|ƒVqdS(N(Rtindex(RR(R(s8lib/python2.7/site-packages/astropy/modeling/mappings.pys rss[Mappings such as {0} that drop one or more of their inputs are not invertible at this time.( R RRt ValueErrortNotImplementedErrorRRt __class__RR (RRtinv((Rs8lib/python2.7/site-packages/astropy/modeling/mappings.pytinversees    N(t__name__t __module__t__doc__tTruetlinearR RtpropertyRRRRRR$(((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyR s"   cBs;eZdZeZddd„Zd„Zed„ƒZ RS(sα Returns inputs unchanged. This class is useful in compound models when some of the inputs must be passed unchanged to the next model. Parameters ---------- n_inputs : int Specifies the number of inputs this identity model accepts. name : str, optional A human-friendly name associated with this model instance (particularly useful for identifying the individual components of a compound model). meta : dict-like Free-form metadata to associate with this model. Examples -------- Transform ``(x, y)`` by a shift in x, followed by scaling the two inputs:: >>> from astropy.modeling.models import (Polynomial1D, Shift, Scale, ... Identity) >>> model = (Shift(1) & Identity(1)) | Scale(1.2) & Scale(2) >>> model(1,1) # doctest: +FLOAT_CMP (2.4, 2.0) >>> model.inverse(2.4, 2) # doctest: +FLOAT_CMP (1.0, 1.0) cCs8tt|ƒƒ}tt|ƒj|d|d|ƒdS(NRR (R RRRR(RRRR R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyR scCs9|jdkrdj|jƒSdj|j|jƒSdS(Nss(RR RR(R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyR€scCs|S(sl The inverse transformation. In this case of `Identity`, ``self.inverse is self``. ((R((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyR$ͺsN( R%R&R'R(R)R RRR*R$(((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyRs  N(R'tcoreRtextern.six.movesRt__all__RR(((s8lib/python2.7/site-packages/astropy/modeling/mappings.pyts  r