a ߙfbS*@s\dZddlmZmZddlmZgdZGdddeZGdd d eZGd d d eZ d S) z 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. ) FittableModelModel)Quantity)MappingIdentity UnitsMappingcsfeZdZdZdZdfdd ZeddZedd Zed d Z d d Z ddZ eddZ Z S)ra 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:`astropy: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) TNcsd|_d|_|dur$t|d|_n||_t||_tj||dtddt |jD|_ tddt |jD|_ ||_ dd|jD|_ d d|jD|_dS) Nrnamemetacss|]}dt|VqdSxNstr.0idxr r 8lib/python3.9/site-packages/astropy/modeling/mappings.py >z#Mapping.__init__..css|]}dt|VqdSr rrr r rr?rcSsi|] }|dqSFr rkeyr r r Brz$Mapping.__init__..cSsi|] }|dqSrr rr r rrCr)_inputs_outputsmax _n_inputslen _n_outputssuper__init__tuplerangeinputsoutputs_mapping_input_units_strict _input_units_allow_dimensionless)selfmappingn_inputsr r  __class__r rr"3s zMapping.__init__cCs|jSN)rr*r r rr,EszMapping.n_inputscCs|jSr/)r r0r r r n_outputsIszMapping.n_outputscCs|jS)z,Integers representing indices of the inputs.r'r0r r rr+MszMapping.mappingcCs.|jdurd|jdSd|jd|jdS)Nz , name=r r+r0r r r__repr__Rs zMapping.__repr__csnt|jkr@|jdur|jnd}t|d|jdttfdd|jD}|jdkrj|dS|S)Nrz expects z inputs; got c3s|]}|VqdSr/r rargsr rr]rz#Mapping.evaluate..rr)rr,r TypeErrorr#r'r1)r*r8r resultr r7revaluateWs zMapping.evaluatecszz tfddtjD}Wn"tyBtdjYn0|}j|_ j |_t |j |_ t |j|_ |S)a) 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|]}j|VqdSr/)r+indexrr0r rrqsz"Mapping.inverse..zZMappings such as {} that drop one or more of their inputs are not invertible at this time.) r#r$r, ValueErrorNotImplementedErrorformatr+r.rrrrr )r*r+invr r0rinverseds        zMapping.inverse)NNN)__name__ __module__ __qualname____doc__linearr"propertyr,r1r+r6r;rA __classcell__r r r-rrs"    rcs:eZdZdZdZd fdd ZddZedd ZZ S) ra 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) TNcs"tt|}tj|||ddS)Nr )r#r$r!r")r*r,r r r+r-r rr"s zIdentity.__init__cCs.|jdurd|jdSd|jd|jdS)Nz >> from astropy.modeling.models import Polynomial1D, UnitsMapping >>> from astropy import units as u >>> poly = Polynomial1D(1, c0=1, c1=2) >>> model = UnitsMapping(((u.m, None),)) | poly >>> model = model | UnitsMapping(((None, u.s),)) >>> model(u.Quantity(10, u.m)) # doctest: +FLOAT_CMP >>> model(u.Quantity(1000, u.cm)) # doctest: +FLOAT_CMP >>> model(u.Quantity(10, u.cm)) # doctest: +FLOAT_CMP Wrapping a unitless model but still permitting unitless input: >>> from astropy.modeling.models import Polynomial1D, UnitsMapping >>> from astropy import units as u >>> poly = Polynomial1D(1, c0=1, c1=2) >>> model = UnitsMapping(((u.m, None),), input_units_allow_dimensionless=True) | poly >>> model = model | UnitsMapping(((None, u.s),)) >>> model(u.Quantity(10, u.m)) # doctest: +FLOAT_CMP >>> model(10) # doctest: +FLOAT_CMP NFcsb||_tdd|D}|dkr4|t|kr4tdd|_||_||_tj||d|dS)NcSsg|]}|ddur|qS)Nr )rmr r r rz)UnitsMapping.__init__..rz1If one return unit is None, then all must be NoneTr ) r'rr=r(input_units_equivalenciesr)r!r"_rebuild_units)r*r+rLZinput_units_allow_dimensionlessr r Znone_mapping_countr-r rr"szUnitsMapping.__init__cCsddt|j|jD|_dS)NcSsi|]\}\}}||qSr r )rZ input_nameZ input_unit_r r rr rz/UnitsMapping._rebuild_units..)zipr%r+ _input_unitsr0r r rrM szUnitsMapping._rebuild_unitscCs t|jSr/rr'r0r r rr,szUnitsMapping.n_inputscCs t|jSr/rQr0r r rr1szUnitsMapping.n_outputscstjSr/)r!r%r0r-r rr%szUnitsMapping.inputscs"tt|jj|||dSr/)r!rr.r%fsetrMr*valuer-r rr%scstjSr/)r!r&r0r-r rr& szUnitsMapping.outputscs"tt|jj|||dSr/)r!rr.r&rRrMrSr-r rr&$scCs|jSr/)rPr0r r r input_units)szUnitsMapping.input_unitscCs|jSr/r2r0r r rr+-szUnitsMapping.mappingcGszg}t||jD]J\}\}}t|tr.|j}n|}|durF||q|t||ddq|jdkrn|dSt|SdS)NT)Zsubokrr)rOr+ isinstancerrTappendr1r#)r*r8r:argrNZ return_unitrTr r rr;1s   zUnitsMapping.evaluatecCs2|jdurd|jdSd|jd|jdSdS)Nzs  r5