U e{jb*@s^dZddlmZddlmZmZdddgZGdddeZGd ddeZGd ddeZ 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. )Quantity) FittableModelModelMappingIdentity 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|_|dkr$t|d|_n||_t||_tj||dtddt |jD|_ tddt |jD|_ ||_ dd|jD|_ d d|jD|_dS) Nrnamemetacss|]}dt|VqdSxNstr.0idxr r X/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/modeling/mappings.py ?sz#Mapping.__init__..css|]}dt|VqdSr rrr r rr@scSsi|] }|dqSFr rkeyr r r Csz$Mapping.__init__..cSsi|] }|dqSrr rr r rrDs)_inputs_outputsmax _n_inputslen _n_outputssuper__init__tuplerangeinputsoutputs_mapping_input_units_strict _input_units_allow_dimensionless)selfmappingn_inputsr r  __class__r rr!4s zMapping.__init__cCs|jSN)rr)r r rr+FszMapping.n_inputscCs|jSr.)rr/r r r n_outputsJszMapping.n_outputscCs|jS)z,Integers representing indices of the inputs.r&r/r r rr*NszMapping.mappingcCs.|jdkrd|jdSd|jd|jdS)Nz , name=r r*r/r r r__repr__Ss zMapping.__repr__csnt|jkr@|jdk r|jnd}t|d|jdttfdd|jD}|jdkrj|dS|S)Nrz expects z inputs; got c3s|]}|VqdSr.r rargsr rr^sz#Mapping.evaluate..rr)rr+r TypeErrorr"r&r0)r)r7r resultr r6revaluateXs zMapping.evaluatecs~z tfddtjD}Wn&tk rFtdjdYnX|}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*indexrr/r rrrsz"Mapping.inverse..zMappings such as zG that drop one or more of their inputs are not invertible at this time.) r"r#r+ ValueErrorNotImplementedErrorr*r-rrrrr)r)r*invr r/rinversees      zMapping.inverse)NNN)__name__ __module__ __qualname____doc__linearr!propertyr+r0r*r5r:r? __classcell__r r r,rrs#    cs: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.|jdkrd|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|]}|ddkr|qS)Nr )rmr r r s z)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*rJZinput_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 rrs z/UnitsMapping._rebuild_units..)zipr$r* _input_unitsr/r r rrKs zUnitsMapping._rebuild_unitscCs t|jSr.rr&r/r r rr+szUnitsMapping.n_inputscCs t|jSr.rOr/r r rr0szUnitsMapping.n_outputscstjSr.)r r$r/r,r rr$szUnitsMapping.inputscs"tt|jj|||dSr.)r rr-r$fsetrKr)valuer,r rr$!scstjSr.)r r%r/r,r rr%&szUnitsMapping.outputscs"tt|jj|||dSr.)r rr-r%rPrKrQr,r rr%*scCs|jSr.)rNr/r r r input_units/szUnitsMapping.input_unitscCs|jSr.r1r/r r rr*3szUnitsMapping.mappingcGszg}t||jD]J\}\}}t|tr.|j}n|}|dkrF||q|t||ddq|jdkrn|dSt|SdS)NT)Zsubokrr)rMr* isinstancerrRappendr0r")r)r7r9argrLZ return_unitrRr r rr:7s   zUnitsMapping.evaluatecCs2|jdkrd|jdSd|jd|jdSdS)Nzs   s6