IR-e]*dZddlmZddlmZmZgdZGddeZGdd eZGd 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. )Quantity) FittableModelModel)MappingIdentity UnitsMappingceZdZdZdZd fd ZedZedZedZ dZ d Z ed Z xZ 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) TNcd|_d|_|t|dz|_n||_t ||_t ||tdt|jD|_ tdt|jD|_ ||_ d|jD|_ d|jD|_dS)Nrnamemetac3:K|]}dt|zVdSxNstr.0idxs 9lib/python3.11/site-packages/astropy/modeling/mappings.py z#Mapping.__init__..>s,LLsC#c((NLLLLLLc3:K|]}dt|zVdSrrrs rrz#Mapping.__init__..?s,NNS3s88^NNNNNNrci|]}|dSFr rkeys r z$Mapping.__init__..Bs#G#G#G3C#G#G#Grci|]}|dSrr rs rr z$Mapping.__init__..Cs0T0T0Te0T0T0Tr)_inputs_outputsmax _n_inputslen _n_outputssuper__init__tuplerangeinputsoutputs_mapping_input_units_strict _input_units_allow_dimensionless)selfmappingn_inputsrr __class__s rr)zMapping.__init__3s    \\A-DNN%DNg,, d...LLeDN6K6KLLLLL NNuT_7M7MNNNNN  #G#G$,#G#G#G 0T0Tt|0T0T0T---rc|jSN)r%r1s rr3zMapping.n_inputsEs ~rc|jSr6)r'r7s r n_outputszMapping.n_outputsIs rc|jS)z,Integers representing indices of the inputs.r.r7s rr2zMapping.mappingMs }rcL|j d|jdSd|jd|jdS)Nz , name=rr2r7s r__repr__zMapping.__repr__Rs: 9 /t|/// /?4<?? ????rc t|jkr9|j|jnd}t|d|jdtt fd|jD}|jdkr|dS|S)Nrz expects z inputs; got c3(K|] }|V dSr6r )rrargss rrz#Mapping.evaluate..]s'::StCy::::::rrr)r&r3r TypeErrorr*r.r9)r1rCrresults ` revaluatezMapping.evaluateWs t99 % % $ 54999DtUUdmUU#d))UUVV V::::DM::::: >Q  !9  rcp tfdtjD}n&#t$rt djdwxYw|}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). c3LK|]}j|VdSr6)r2index)rrr1s rrz"Mapping.inverse..ps3TTDL..s33TTTTTTrzMappings such as zG that drop one or more of their inputs are not invertible at this time.) r*r+r3 ValueErrorNotImplementedErrorr2r4r#r"r&r%r')r1r2invs` rinversezMapping.inverseds TTTTuT]?S?STTTTTGG   %4DL444   nnW%%m | CK(( S\** s -1#A)NNN)__name__ __module__ __qualname____doc__linearr)propertyr3r9r2r@rFrM __classcell__r4s@rrrs  DFUUUUUU$XXX@@@   XrrcDeZdZdZdZdfd ZdZedZxZ 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) TNctt|}t|||dS)Nr )r*r+r(r))r1r3rrr2r4s rr)zIdentity.__init__s;h(( t$77777rcL|j d|jdSd|jd|jdS)Nz )rr3r7s rr@zIdentity.__repr__s: 9 1 111 1ADMAA$)AAAArc|S)zl The inverse transformation. In this case of `Identity`, ``self.inverse is self``. r r7s rrMzIdentity.inverses  r)NN) rNrOrPrQrRr)r@rSrMrTrUs@rrrs{<F888888BBB XrrceZdZdZ dfd ZdZedZedZefdZ e j fd Z efd Z e j fd Z ed Z ed Z dZdZxZS)r a Mapper that operates on the units of the input, first converting to canonical units, then assigning new units without further conversion. Used by Model.coerce_units to support units on otherwise unitless models such as Polynomial1D. Parameters ---------- mapping : tuple A tuple of (input_unit, output_unit) pairs, one per input, matched to the inputs by position. The first element of the each pair is the unit that the model will accept (specify ``dimensionless_unscaled`` to accept dimensionless input). The second element is the unit that the model will return. Specify ``dimensionless_unscaled`` to return dimensionless Quantity, and `None` to return raw values without Quantity. input_units_equivalencies : dict, optional Default equivalencies to apply to input values. If set, this should be a dictionary where each key is a string that corresponds to one of the model inputs. input_units_allow_dimensionless : dict or bool, optional Allow dimensionless input. If this is True, input values to evaluate will gain the units specified in input_units. If this is a dictionary then it should map input name to a bool to allow dimensionless numbers for that input. 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, optional Free-form metadata to associate with this model. Examples -------- Wrapping a unitless model to require and convert units: >>> 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 NFc0||_td|D}|dkr"|t|krtdd|_||_||_t |||dS)Nc"g|] }|d | S)r )rms r z)UnitsMapping.__init__..s!F!F!F" ! rrz1If one return unit is None, then all must be NoneTr ) r.r&rJr/input_units_equivalenciesr0r(r)_rebuild_units)r1r2r`input_units_allow_dimensionlessrrnone_mapping_countr4s rr)zUnitsMapping.__init__s   !F!FW!F!F!FGG  ! !&8CLL&H&HPQQ Q$( )B&0O- d... rcXdt|j|jD|_dS)Nc i|] \}\}}|| Sr r )r input_name input_unit_s rr z/UnitsMapping._rebuild_units.. s2   + OZ     r)zipr,r2 _input_unitsr7s rrazUnitsMapping._rebuild_units s7  /24; /M/M   rc*t|jSr6r&r.r7s rr3zUnitsMapping.n_inputs4=!!!rc*t|jSr6rlr7s rr9zUnitsMapping.n_outputsrmrc*tjSr6)r(r,r1r4s rr,zUnitsMapping.inputssww~rctt|jj|||dSr6)r(r r4r,fsetrar1valuer4s rr,zUnitsMapping.inputss@ lDN++277eDDD rc*tjSr6)r(r-rps rr-zUnitsMapping.outputs!swwrctt|jj|||dSr6)r(r r4r-rrrarss rr-zUnitsMapping.outputs%s@ lDN++388uEEE rc|jSr6)rjr7s r input_unitszUnitsMapping.input_units*s   rc|jSr6r;r7s rr2zUnitsMapping.mapping.s }rc>g}t||jD]d\}\}}t|tr|j}n|}|||?|t||de|jdkr|dSt|S)NT)subokrr)rir2 isinstancerrtappendr9r*)r1rCrEargrh return_unitrts rrFzUnitsMapping.evaluate2s%(t|%<%< H H !C!![#x((  " e$$$$ hukFFFGGGG >Q  !9 == rcL|j d|jdSd|jd|jdS)Nzr?r7s rr@zUnitsMapping.__repr__Cs: 9 4DL444 4HDLHHHHH Hr)NFNN)rNrOrPrQr)rarSr3r9r,setterr-rxr2rFr@rTrUs@rr r s;;@#'(-   2   ""X"""X"X ]]X ^^!!X!X!!!"IIIIIIIrr N) rQ astropy.unitsrcorerr__all__rrr r rrrs #"""""&&&&&&&& 1 1 1nnnnnmnnnb11111w111hTITITITITI5TITITITITIr