HR-e dZddlmZddlZddlZddlZddlmZddlm Z m Z ddl Z ddl mZddlmZdd gZed d ZGd d eZee j e jfZddZddZGddejZGddeZeddZGddeZGddeZGdd eZ dS)z5 This module is to contain an improved bounding box. ) annotationsN) namedtuple)AnyCallableQuantity) isiterableModelBoundingBoxCompoundBoundingBox _BaseIntervalz lower uppercneZdZdZdZdZedZedZ edZ d d Z d Z d S) _Intervala A single input's bounding box interval. Parameters ---------- lower : float The lower bound of the interval upper : float The upper bound of the interval Methods ------- validate : Constructs a valid interval outside : Determine which parts of an input array are outside the interval. domain : Constructs a discretization of the points inside the interval. c(d|jd|jdS)NzInterval(lower=z, upper=))lowerupperselfs =lib/python3.11/site-packages/astropy/modeling/bounding_box.py__repr__z_Interval.__repr__2sBBBTZBBBBc*tj|SN)copydeepcopyrs rrz_Interval.copy5s}T"""rcd} tj|}nu#t$rh t|dkr|d}tjd|D}n*#ttt f$rt |wxYwYnwxYw|dv}|s8t|dko$|ddkot d|D}t|r|st |dS) z1Validate the shape of an interval representation.z5An interval must be some sort of sequence of length 2rc6g|]}|S)to_value.0bs r z-_Interval._validate_shape..Ds !A!A!A1!**,,!A!A!Ar)))rr%)r%rr%c3JK|]}t|tjVdSr) isinstancenpndarrayr!s r z,_Interval._validate_shape..Ms.DDa 1bj11DDDDDDrN)r(shape TypeErrorlen ValueErrorAttributeErrorallr )intervalMESSAGEr+ valid_shapes r_validate_shapez_Interval._validate_shape8s:N *HX&&EE * * * *x==A%%'{H!A!A!A!A!ABB >: * * * ))) *  *55  E QE!HMEDD8DDDDD  (## &; &W%% % & &s& B 9AB 'BB  B ctj|tj|kr!tjd|d|dt |||S)zGValidate the bounds are reasonable and construct an interval from them.zInvalid interval: upper bound z# is strictly less than lower bound .)r( asanyarrayr0warningswarnRuntimeWarning)clsrrs r_validate_boundsz_Interval._validate_boundsSs M% 2=#7#7 7 < < > >  M>>>5:>>>    s5%   rc||t|dkrt|d}nt|}||d|dS)z Construct and validate an interval. Parameters ---------- interval : iterable A representation of the interval. Returns ------- A validated interval. rr)r4r-tupler<)r;r1s rvalidatez_Interval.validate_se H%%% x==A  Xa[))HHXH##HQK!===r_input np.ndarraycPtj||jk||jkS)aU Parameters ---------- _input : np.ndarray The evaluation input in the form of an array. Returns ------- Boolean array indicating which parts of _input are outside the interval: True -> position outside interval False -> position inside interval )r( logical_orrr)rr@s routsidez_Interval.outsidevs$}Vdj0&4:2EFFFrcHtj|j|j|z|Sr)r(arangerr)r resolutions rdomainz_Interval.domainsyTZ*%>[>, G G G GJJJJJrrreturnintct|tr?||jvr|j|}nt d|d|jdt jt|t jrMd|cxkrt|jkrnn|}n;td|dt|jdt d|d|S) z Get the input index corresponding to the given key. Can pass in either: the string name of the input or the input index itself. 'z' is not one of the inputs: r6rz Integer key: z must be non-negative and < z Key value: z must be string or integer.) r'strinputsindexr.r( issubdtypetypeintegerr- IndexError)modelkeyrUs r get_indexr\s#s I %,  L&&s++EEQQQ%,QQQRR R tCyy"* - -I  ' ' ' 'c%,'' ' ' ' ' 'EEUUUU\ARARUUU GsGGGHHH LrrUc|j|Sz4Get the input name corresponding to the input index.)rT)rZrUs rget_namer_s < rcZeZdZdZd*d+dZed Zed,d Zed-d Zd.d/d Z d0dZ d1dZ ed2dZ d3dZ dZejd4dZejd5dZedZdZdZdZd Zed6d#Zd7d&Zd7d'Zed(Zd8d)ZdS)9_BoundingDomaina Base class for ModelBoundingBox and CompoundBoundingBox. This is where all the `~astropy.modeling.core.Model` evaluation code for evaluating with a bounding box is because it is common to both types of bounding box. Parameters ---------- model : `~astropy.modeling.Model` The Model this bounding domain is for. prepare_inputs : Generates the necessary input information so that model can be evaluated only for input points entirely inside bounding_box. This needs to be implemented by a subclass. Note that most of the implementation is in ModelBoundingBox. prepare_outputs : Fills the output values in for any input points outside the bounding_box. evaluate : Performs a complete model evaluation while enforcing the bounds on the inputs and returns a complete output. NCignored list[int]orderrSc|||_|||_|||_dSr)_model_validate_ignored_ignored _get_order_order)rrZrcres r__init__z_BoundingDomain.__init__s5 ..w77 ooe,, rc|jSr)rgrs rrZz_BoundingDomain.model {rrOc|jSr)rkrs rrez_BoundingDomain.orderrnrc|jSrrirs rrcz_BoundingDomain.ignoreds }rcF||j}|dvrtd|d|S)zd Get if bounding_box is C/python ordered or Fortran/mathematically ordered. N)rbFzTorder must be either 'C' (C/python order) or 'F' (Fortran/mathematical order), got: r6)rkr.)rres rrjz_BoundingDomain._get_ordersO =KE  " "C:?CCC   rrPc,t|j|S)z Get the input index corresponding to the given key. Can pass in either: the string name of the input or the input index itself. )r\rgrr[s r _get_indexz_BoundingDomain._get_indexsc***rrUc,t|j|Sr^)r_rg)rrUs r _get_namez_BoundingDomain._get_names U+++r list[str]c*fdjDS)Nc:g|]}|Srrx)r"rUrs rr$z2_BoundingDomain.ignored_inputs..s%AAA%u%%AAArrqrs`rignored_inputsz_BoundingDomain.ignored_inputssAAAA4=AAAArlistc(|gSfd|DS)Nc:g|]}|Srrv)r"r[rs rr$z5_BoundingDomain._validate_ignored..s%<<.Ns9      k: 6 6   rN)rangerg n_outputs)rrrs```r_all_out_outputz_BoundingDomain._all_out_output>sQ       4;011     rc|||}|jstj|}n|||<tj|r|d}|S)a  For a single output fill in all the parts corresponding to inputs outside the bounding box. Parameters ---------- valid_output : numpy array The output from the model corresponding to inputs inside the bounding box valid_index : numpy array array of all indices of inputs inside the bounding box input_shape : tuple The shape that all inputs have be reshaped/broadcasted into fill_value : float The value which will be assigned to inputs which are outside the bounding box Returns ------- An output array with all the indices corresponding to inputs outside the bounding box filled in by fill_value r)rr+r(arrayisscalaritem)r valid_output valid_indexrroutputs r_modify_outputz_BoundingDomain._modify_outputSsd.""; ;;| /Xl++FF".F;  ;| $ $ $[[^^F rc jg}|D]-}||||||.|S)a Fill in all the outputs of the model corresponding to inputs outside the bounding_box. Parameters ---------- valid_outputs : list of numpy array The list of outputs from the model corresponding to inputs inside the bounding box valid_index : numpy array array of all indices of inputs inside the bounding box input_shape : tuple The shape that all inputs have be reshaped/broadcasted into fill_value : float The value which will be assigned to inputs which are outside the bounding box Returns ------- List of filled in output arrays. )appendr)r valid_outputsrrroutputsrs r_prepare_outputsz _BoundingDomain._prepare_outputsusQ,)  L NN##L+{JWW    rcX|jjdkr|g}|||||S)a Fill in all the outputs of the model corresponding to inputs outside the bounding_box, adjusting any single output model so that its output becomes a list of containing that output. Parameters ---------- valid_outputs : list The list of outputs from the model corresponding to inputs inside the bounding box valid_index : array_like array of all indices of inputs inside the bounding box input_shape : tuple The shape that all inputs have be reshaped/broadcasted into fill_value : float The value which will be assigned to inputs which are outside the bounding box r)rgrr)rrrrrs rprepare_outputsz_BoundingDomain.prepare_outputss<& ; A % %*OM$$ ; Z   r with_unitsboolc,|rt|ddSdS)aD Get the unit for outputs if one is required. Parameters ---------- valid_outputs : list of numpy array The list of outputs from the model corresponding to inputs inside the bounding box with_units : bool whether or not a unit is required unitN)getattr)rrs r_get_valid_outputs_unitz'_BoundingDomain._get_valid_outputs_units(  8=&$77 7 8 8revaluatercx||}|||}||||||fS)a Evaluate the model using the given evaluate routine. Parameters ---------- evaluate : Callable callable which takes in the valid inputs to evaluate model valid_inputs : list of numpy arrays The inputs reduced to just those inputs which are all inside their respective bounding box intervals valid_index : numpy array array of all indices inside the bounding box input_shape : tuple The shape that all inputs have be reshaped/broadcasted into fill_value : float The value which will be assigned to inputs which are outside the bounding box with_units : bool whether or not a unit is required Returns ------- outputs : list containing filled in output values valid_outputs_unit : the unit that will be attached to the outputs )rr) rr valid_inputsrrrrrvalid_outputs_units r_evaluate_modelz_BoundingDomain._evaluate_modelsPH!.. !99-TT   [* U U   rc|||\}}}|r|||S|||||||S)aEvaluate model with steps: prepare_inputs -> evaluate -> prepare_outputs. Parameters ---------- evaluate : Callable callable which takes in the valid inputs to evaluate model valid_inputs : list of numpy arrays The inputs reduced to just those inputs which are all inside their respective bounding box intervals valid_index : numpy array array of all indices inside the bounding box input_shape : tuple The shape that all inputs have be reshaped/broadcasted into fill_value : float The value which will be assigned to inputs which are outside the bounding box with_units : bool whether or not a unit is required Returns ------- outputs : list containing filled in output values valid_outputs_unit : the unit that will be attached to the outputs )rrr) rrrTrrrrrall_outs r _evaluatez_BoundingDomain._evaluatesc:.2-@-@f-U-U* k7  '' Z@@ @'', [*j rc0|t||ddS|S)a Set the units on the outputs prepare_inputs -> evaluate -> prepare_outputs -> set output units. Parameters ---------- outputs : list containing filled in output values valid_outputs_unit : the unit that will be attached to the outputs Returns ------- List containing filled in output values and units NFT)rsubokr)rrs r_set_outputs_unitz!_BoundingDomain._set_outputs_units'"  )G%7e4PPP Prc|j|}||||||jj\}}t |||S)a Perform full model evaluation steps: prepare_inputs -> evaluate -> prepare_outputs -> set output units. Parameters ---------- evaluate : callable callable which takes in the valid inputs to evaluate model valid_inputs : list The inputs reduced to just those inputs which are all inside their respective bounding box intervals valid_index : array_like array of all indices inside the bounding box fill_value : float The value which will be assigned to inputs which are outside the bounding box )rgrrbbox_with_unitsr>r)rrrTrrrrs rrz_BoundingDomain.evaluate%se$k--f55 '+nn fk:t{7R' ' ##T++G5GHHIIIrNrb)rcrdrerSrOrS)rOrdr)rerSrOrSrOrPrUrP)rOry)rcr~rOrdrrrOr)rr)rrrr)rr)rIrJrKrLrlpropertyrZrercrjrvrxr}rhrabcabstractmethodrrrMrrrrrrrrrrrrrrarasA4----- XXX ++++,,,,BBBXB====      R R R R WWWW.22\2&*   D<   4 8 8 8\ 8* * * * X$$$$L\*JJJJJJrracNeZdZdZ d/d0fd Zd1d Zed2d Zed3dZdZ dZ dZ dZ dZ d1d4dZdZdZdZd5dZedZd1d4dZed6dZd1d4d Zd1d4d!Ze d7d8d&Zd9d:d(Zed)Zd1d4d*Zd+Zd,Zd;d.ZxZ S) input index values -> interval for that index model : `~astropy.modeling.Model` The Model this bounding_box is for. ignored : list A list containing all the inputs (index) which will not be checked for whether or not their elements are in/out of an interval. order : optional, str The ordering that is assumed for the tuple representation of this bounding_box. Options: 'C': C/Python order, e.g. z, y, x. (default), 'F': Fortran/mathematical notation order, e.g. x, y, z. Nrb intervalsdict[int, _Interval]rcrdrerSct|||i|_|dkr|ikr|||dSdSdS)Nrre)superrl _intervals _validate)rrrZrcre __class__s rrlzModelBoundingBox.__init__Zs` /// ??yB NN9EN 2 2 2 2 2 ?rcd|jD}||j}t ||j||jS)Nc>i|]\}}||Sr)r)r"rUr1s r z)ModelBoundingBox.copy..hs3   '6uhE8==??   rrcre)ritemsrirr rgrk)rrcrs rrzModelBoundingBox.copygsl  :>/:O:O:Q:Q    ?m((**G t{G4;    rrOc|jS)z2Return bounding_box labeled using input positions.)rrs rrzModelBoundingBox.intervalsss rdict[str, _Interval]cNfdjDS)z.Return bounding_box labeled using input names.cBi|]\}}||Srr|)r"rUbboxrs rrz4ModelBoundingBox.named_intervals..{s+WWW tu%%tWWWr)rrrs`rnamed_intervalsz ModelBoundingBox.named_intervalsxs.XWWWt?T?T?V?VWWWWrc ddg}|jD] \}}|d|d|!|dt|jdkr|d|j|d|jjjd |jj d |d |j d |d d |S)NzModelBoundingBox(z intervals={ z:  }rz ignored=z model=z(inputs=rz order='rR ) rrrr-rir}rgrrIrTrkjoin)rpartsnamer1s rrzModelBoundingBox.__repr__}s$&78"288:: 8 8ND( LL6D66H66 7 7 7 7 W t}   ! ! LL=(;== > > > V.7 V VAS V V V     14;111222 Syyrc*t|jSr)r-rrs r__len__zModelBoundingBox.__len__s4?###rcz |||jvp|jS#ttf$rYdSwxYwNF)rvrrirYr.rus r __contains__zModelBoundingBox.__contains__sN ??3''4?:Kdm KJ'   55 s "%::c:|||jvSr)rvrrus r has_intervalzModelBoundingBox.has_intervalss##t66rc||}||jvrtS|j||S)z=Get bounding_box entries by either input name or input index.)rvri_ignored_intervalrrr[rUs r __getitem__zModelBoundingBox.__getitem__s?$$ DM ! !$ $?4??3#7#78 8rcztjdkr9ttjdS|}jj}|dkr |ddd}tfd|D}t|dkr|d}|S)z Return the old tuple of tuples representation of the bounding_box order='C' corresponds to the old bounding_box ordering order='F' corresponds to the gwcs bounding_box ordering. rrrbNc3BK|]}t|VdSrr>)r" input_namers rr*z0ModelBoundingBox.bounding_box..s0JJZtJ/00JJJJJJr)r-rr>r~valuesrjrgrT)rrerTrs` r bounding_boxzModelBoundingBox.bounding_boxs t  1 $ $do446677:;; ;OOE**E['F||"JJJJ6JJJJJD4yyA~~AwKrct|tr||kSt|tr |j|jko|j|jkSdS)z?Note equality can be either with old representation or new one.F)r'r>rr rrcrvalues r__eq__zModelBoundingBox.__eq__sg eU # # $$&&%/ / / 0 0 Neo5  - 5rc||}||jvr|j|t||j|<dS)zBValidate and store interval under key (input index or input name).N)rvriremoverr?r)rr[rrUs r __setitem__zModelBoundingBox.__setitem__sV$$ DM ! ! M  ' ' '!*!3!3E!:!:rc||}||jvrtd|d|j|=|j|dS)zDelete stored interval.zCannot delete ignored input: !N)rvri RuntimeErrorrrrs r __delitem__zModelBoundingBox.__delitem__s`$$ DM ! !EsEEEFF F OE " U#####rrrcD|D] \}}|||< dS)z:Validate passing dictionary of intervals and setting them.Nr)rrr[rs r_validate_dictzModelBoundingBox._validate_dicts6&,,..  JCDII  rcPfdjjD}fd|DS)Nc:g|]}|Srrr"r@rs rr$z;ModelBoundingBox._available_input_index..s%VVVT__V44VVVrc&g|] }|jv |Srrqrs rr$z;ModelBoundingBox._available_input_index..s%VVV6& :U:U:U:U:Ur)rgrT)rmodel_input_indexs` r_available_input_indexz'ModelBoundingBox._available_input_indexs=VVVV4;CUVVVVVVV%6VVVVrc||}|dkr |ddd}t|D]\}}|||j|<dS)z` Validate passing tuple of tuples representation (or related) and setting them. rbNr)rj enumerater)rrrerUrs r_validate_sequencez#ModelBoundingBox._validate_sequencesj&& C<<("-L%l33 = =LE57Q  z,'E'E 3?D,Q/ 0 0 0  # #L% 8 8 8 8 8rFr~_preserve_ignorerc t|tr|j}|r|j}|j}|i|||}|||S)a Construct a valid bounding box for a model. Parameters ---------- model : `~astropy.modeling.Model` The model for which this will be a bounding_box bounding_box : dict, tuple A possible representation of the bounding box order : optional, str The order that a tuple representation will be assumed to be Default: 'C' r)r'r rercrr)r;rZrrcrerrnews rr?zModelBoundingBox.validatesf. l$4 5 5 8 &E /&.'7Lc"eWE::: l### rrc|}|D]}||=|r|j}nd}t||j||jS)a Fix the bounding_box for a `fix_inputs` compound model. Parameters ---------- model : `~astropy.modeling.Model` The new model for which this will be a bounding_box fixed_inputs : dict Dictionary of inputs which have been fixed by this bounding box. keep_ignored : bool Keep the ignored inputs of the bounding box (internal argument only) Nr)rkeysrcr r?rrk)rrZr _keep_ignoredrr@rcs rrzModelBoundingBox.fix_inputs(sviikk"''))  FF  kGGG(( 3&sz)   rc t|Sr)r-rs r dimensionzModelBoundingBox.dimensionCs4yyrcjj}|}|dkr |ddd}fd|DS)NrbrcFg|]}|Sr)rH)r"rrGrs rr$z+ModelBoundingBox.domain..Ms,MMM Z '' 33MMMr)rgrTrj)rrGrerTs`` rrHzModelBoundingBox.domainGsT#&& C<<DDbD\FMMMMMfMMMMrc,d}tj|t}t|D]d\}}tj|}tj||||}d||<|rd}ne||fS)a Get all the input positions which are outside the bounding_box, so that the corresponding outputs can be filled with the fill value (default NaN). Parameters ---------- input_shape : tuple The shape that all inputs have be reshaped/broadcasted into inputs : list List of all the model inputs Returns ------- outside_index : bool-numpy array True -> position outside bounding_box False -> position inside bounding_box all_out : bool if all of the inputs are outside the bounding_box F)dtypeT)r(rrr r7 broadcast_torDr0)rrrTr outside_indexrUr@rDs r_outsidezModelBoundingBox._outsideOs*D999 &v..  ME6]6**Fod5k&9&9&&A&A;OOG%)M' "  ""  g%%rc|||\}}tjtj|}t |ddkrd}||fS)a Get the indices of all the inputs inside the bounding_box. Parameters ---------- input_shape : tuple The shape that all inputs have be reshaped/broadcasted into inputs : list List of all the model inputs Returns ------- valid_index : numpy array array of all indices inside the bounding box all_out : bool if all of the inputs are outside the bounding_box rT)r r( atleast_1d logical_notnonzeror-)rrrTrrrs r _valid_indexzModelBoundingBox._valid_indexssh$"&{F!C!C wmBN=$A$ABBJJLL {1~  ! # #GG##rrcp|||\}}g}|s|D]}|rltjtj|||}tj|r|d}||p||t|||fS)rr)r%r(rr"rrrr>)rrrTrrrr@ valid_inputs rrzModelBoundingBox.prepare_inputss* $00fEE W  0 0 00"$/"-2G2G"U"U##K{6**:&1&6&6q&9&9  '' 4444 ''////\""K88rr)rrrcrdrerSr)rOr)rOr)rerS)rrr)NrbF)rcr~rerSrr)Frr)!rIrJrKrLrlrrrrrrrrrrrrrrrr r rrrNr?rrrHr r%r __classcell__rs@rr r Bs6" 3 3 3 3 3 3 3     XXXXXX   $$$$ 777999(   ;;;$$$ WWXW = = = = =X 9 9 9 9 999999 !& [B     6XNNNNN"&"&"&H$$$4$9$9$9$9$9$9$9$9r_BaseSelectorArgumentz index ignorecfeZdZdZfdZedddZdZdd Zd Z ddZ ddZ dZ xZ S)_SelectorArgumenta# Contains a single CompoundBoundingBox slicing input. Parameters ---------- index : int The index of the input in the input list ignore : bool Whether or not this input will be ignored by the bounding box. Methods ------- validate : Returns a valid SelectorArgument for a given model. get_selector : Returns the value of the input for use in finding the correct bounding_box. get_fixed_value : Gets the slicing value from a fix_inputs set of values. cNt|||}|Sr)r__new__)r;rUignorerrs rr.z_SelectorArgument.__new__s!wwsE622 rTrcrc6|t|||S)a Construct a valid selector argument for a CompoundBoundingBox. Parameters ---------- model : `~astropy.modeling.Model` The model for which this will be an argument for. argument : int or str A representation of which evaluation input to use ignored : optional, bool Whether or not to ignore this argument in the ModelBoundingBox. Returns ------- Validated selector_argument )r\)r;rZargumentrcs rr?z_SelectorArgument.validates $s9UH--w777rc||j}t|r*t|dkr|dSt|S|S)z Get the selector value corresponding to this argument. Parameters ---------- *inputs : All the processed model evaluation inputs. rr)rUr r-r>rrT _selectors r get_selectorz_SelectorArgument.get_selectorsO4:& i  (9~~"" |#Y'''rrOrSc,t||jS)z Get the name of the input described by this selector argument. Parameters ---------- model : `~astropy.modeling.Model` The Model this selector argument is for. )r_rUrrZs rrz_SelectorArgument.namestz***rcDd||d|jdS)z Get a pretty-print representation of this object. Parameters ---------- model : `~astropy.modeling.Model` The Model this selector argument is for. zArgument(name='z ', ignore=rrr/r7s r pretty_reprz_SelectorArgument.pretty_reprs,L5!1!1KKT[KKKKrrrc|j|vr ||jS|||vr|||St||d|)a Gets the value fixed input corresponding to this argument. Parameters ---------- model : `~astropy.modeling.Model` The Model this selector argument is for. values : dict Dictionary of fixed inputs. z was not found in )rUrrr:rrZrs rget_fixed_valuez!_SelectorArgument.get_fixed_values} :  $*% %yy6))dii..//"''..JJ&JJrc4|jt||kS)a@ Determine if passed argument is described by this selector argument. Parameters ---------- model : `~astropy.modeling.Model` The Model this selector argument is for. argument : int or str A representation of which evaluation input is being used )rUr\rrZr1s r is_argumentz_SelectorArgument.is_argument&szYuh7777rc:|||jfS)z Get a tuple representation of this argument using the input name from the model. Parameters ---------- model : `~astropy.modeling.Model` The Model this selector argument is for. r9r7s r named_tuplez_SelectorArgument.named_tuple4s %  $+..r)T)rcrrrrrOr)rIrJrKrLr.rNr?r5rr:r=r@rBr(r)s@rr,r,s0 8888[8&" + + + + L L L, 8 8 8 8 / / / / / / /rr,ceZdZdZdZddfd ZdZed Zed Z e ddd Z d Z d Z ddZddZdZdZdZdZxZS)_SelectorArgumentsa Contains the CompoundBoundingBox slicing description. Parameters ---------- input_ : The SelectorArgument values Methods ------- validate : Returns a valid SelectorArguments for its model. get_selector : Returns the selector a set of inputs corresponds to. is_selector : Determines if a selector is correctly formatted for this CompoundBoundingBox. get_fixed_value : Gets the selector from a fix_inputs set of values. Ninput_tuple[_SelectorArgument] kept_ignorer~cnt||}|g|_n||_|Sr)rr. _kept_ignore)r;rGrIrrs rr.z_SelectorArguments.__new__[s8wwsF++   "D   +D  rcdg}|D]-}|d||.|dd|S)z Get a pretty-print representation of this object. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. zSelectorArguments(z rr)rr:r)rrZrr1s rr:z_SelectorArguments.pretty_represk&& ? ?H LL= 4 4U ; ;== > > > > SyyrcRd|D}||j|S)zGet the list of ignored inputs.c*g|]}|j |jSr)r/rU)r"r1s rr$z-_SelectorArguments.ignore..xs!IIIXI(.IIIr)extendrK)rr/s rr/z_SelectorArguments.ignoreus2JIIII d'((( rc|jS)z%The arguments to persist in ignoring.)rKrs rrIz_SelectorArguments.kept_ignore}s   rcg}|D]b}tj|g|R}|jd|Dvr&tdt ||jd||ct |dkrtdt|tr|g}| |j |t||S)ad Construct a valid Selector description for a CompoundBoundingBox. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. arguments : The individual argument information kept_ignore : Arguments to persist as ignored cg|] }|j Sr)rU)r"thiss rr$z/_SelectorArguments.validate..s>>>t >>>rzInput: 'z' has been repeated.rz-There must be at least one selector argument.) r,r?rUr.r_rr-r'rFrOrIr>)r;rZ argumentsrIrTr1r@s rr?z_SelectorArguments.validates ! " "H&/AAAAF|>>v>>>>> Rxv|<<RRR MM& ! ! ! ! v;;!  LMM M i!3 4 4 6"   y4 5 5 5s5==+...rc:tfd|DS)z Get the selector corresponding to these inputs. Parameters ---------- *inputs : All the processed model evaluation inputs. c3,K|]}|jVdSr)r5)r"r1rTs rr*z2_SelectorArguments.get_selector..s.II*X*F3IIIIIIrr)rrTs `rr5z_SelectorArguments.get_selectors(IIIIDIIIIIIrclt|tot|t|kS)z Determine if this is a reasonable selector. Parameters ---------- _selector : tuple The selector to check )r'r>r-rr4s r is_selectorz_SelectorArguments.is_selectors*)U++KI#d))0KKrrrc>tfd|DS)a Gets the value fixed input corresponding to this argument. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. values : dict Dictionary of fixed inputs. c3DK|]}|VdSr)r=)r"r1rZrs rr*z6_SelectorArguments.get_fixed_values..s3RRX--eV<<RRRRRRrrr<s ``rget_fixed_valuesz#_SelectorArguments.get_fixed_valuess,RRRRRTRRRRRRrrOrc>tfd|DS)a= Determine if passed argument is one of the selector arguments. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. argument : int or str A representation of which evaluation input is being used c3DK|]}|VdSr)r@)r" selector_argr1rZs rr*z1_SelectorArguments.is_argument..s3VV<++E8<<VVVVVVr)anyr?s ``rr@z_SelectorArguments.is_arguments-VVVVVQUVVVVVVrct|D]\}}|||r|cS t|d)a3 Get the index of the argument passed in the selector tuples. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. argument : int or str A representation of which argument is being used z. does not correspond to any selector argument.)r r@r.)rrZr1rUr_s rselector_indexz!_SelectorArguments.selector_indexsd$-T??   E<''x88   KKK rct|}||||jg}||jt |t||S)a+ Reduce the selector arguments by the argument given. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. argument : int or str A representation of which argument is being used ) r~poprbrUrOrKrFr?r>)rrZr1rTrIs rreducez_SelectorArguments.reducesmJJ  }}T%8%8%I%IJJPQ 4,---!**5% 2B2BKPPPrc|||rt|dt||g}t|||S)a Add argument to the kept_ignore list. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. argument : int or str A representation of which argument is being used z/: is a selector argument and cannot be ignored.)r@r.r\rFr?)rrZr1rIs r add_ignorez_SelectorArguments.add_ignoresf   E8 , , LLL !112 !**5$ DDDrc:tfd|DS)z Get a tuple of selector argument tuples using input names. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. c3BK|]}|VdSr)rB)r"r_rZs rr*z1_SelectorArguments.named_tuple..s1NN\--e44NNNNNNrrr7s `rrBz_SelectorArguments.named_tuples(NNNNNNNNNNrr)rGrHrIr~)rIr~rCrD)rIrJrKrLrKr.r:rr/rIrNr?r5rYr\r@rbrergrBr(r)s@rrFrFAs[.L   X!!X!!/!/!/!/[!/F J J J L L L S S S S W W W W(QQQ$EEE* O O O O O O OrrFcBeZdZdZ d/d0fd ZdZdZed1dZed2dZ e j dZ ed3dZ edZ e dZdZd4dZdZe d5d6d"Zd#Zd$Zd%Zd7d'Zd8d)Zd1d*Zd+Zd,Zd9d.ZxZS):r a A model's compound bounding box. Parameters ---------- bounding_boxes : dict A dictionary containing all the ModelBoundingBoxes that are possible keys -> _selector (extracted from model inputs) values -> ModelBoundingBox model : `~astropy.modeling.Model` The Model this compound bounding_box is for. selector_args : _SelectorArguments A description of how to extract the selectors from model inputs. create_selector : optional A method which takes in the selector and the model to return a valid bounding corresponding to that selector. This can be used to construct new bounding_boxes for previously undefined selectors. These new boxes are then stored for future lookups. order : optional, str The ordering that is assumed for the tuple representation of the bounding_boxes. Nrbbounding_boxesdict[Any, ModelBoundingBox] selector_argsrFcreate_selectorrrcrdrerSct|||||_t|||_i|_||dSr)rrl_create_selectorrFr?_selector_args_bounding_boxesr)rrkrZrmrnrcrers rrlzCompoundBoundingBox.__init__;sa /// /099%OO! ~&&&&&rcfdjD}t|jjt jjjS)NcVi|]%\}}||jj&Sr)rrmr/)r"selectorrrs rrz,CompoundBoundingBox.copy..MsA   $ dii 2 9::   r)rmrnre) rrrr rgrqrrrprk)rrks` rrzCompoundBoundingBox.copyLsv    "&"6"<"<">">    #  K- M$*?@@+     rc ddg}|jD]w\}}|d}|d|d|d|D]}|d|x|d|j|jd}|d |d|D]}|d||d d |S) NzCompoundBoundingBox(z bounding_boxes={rrz = rz rz selector_args = r) rrrrsplitrrdrmr:rgr)rrr4r bbox_reprpartselector_args_reprs rrzCompoundBoundingBox.__repr__Zsa')?@#399;; 4 4OIt --d33I LLDIDD)--2B2BDD E E E! 4 4 2D223333 4 W"/;;DKHHNNtTT G,>,B,B1,E,EGGHHH& , ,D LL*D** + + + + SyyrrOc|jSrrrrs rrkz"CompoundBoundingBox.bounding_boxesms ##rc|jSr)rqrs rrmz!CompoundBoundingBox.selector_argsqs ""rct|j||_t jdt dS)NzhOverriding selector_args may cause problems you should re-validate the compound bounding box before use!)rFr?rgrqr8r9r:rs rrmz!CompoundBoundingBox.selector_argsusB099$+uMM  4      rr>c@|j|jSr)rqrBrgrs rnamed_selector_tuplez(CompoundBoundingBox.named_selector_tuples"..t{;;;rc|jSr)rprs rrnz#CompoundBoundingBox.create_selectors $$rcDt|rt|S|fSr)r r>)r[s r_get_selector_keyz%CompoundBoundingBox._get_selector_keys# c?? :: 6Mrc||}|j|st|d|jj|jz}t |j|||j |j |<dS)Nz is not a selector!r) rrmrYr.r/rcr r?rgrkrr)rr[rr4rcs rrzCompoundBoundingBox.__setitem__s**3// !--i88 @ >>>?? ?$+dl:*:*C*C Kt{+D+ + Y'''rrcD|D] \}}|||< dSrr)rrkr4rs rrzCompoundBoundingBox._validates7'5';';'='= + + #I|*DOO + +rct|tr0|j|jko|j|jko|j|jkSdSr)r'r rkrmrnrs rrzCompoundBoundingBox.__eq__sW e0 1 1 #u';;B&%*==B(E,AA  5rFrr~rrc t|tr)||j}||j}|j}|r|j}|j}|td|||||||S)ab Construct a valid compound bounding box for a model. Parameters ---------- model : `~astropy.modeling.Model` The model for which this will be a bounding_box bounding_box : dict Dictionary of possible bounding_box representations selector_args : optional Description of the selector arguments create_selector : optional, callable Method for generating new selectors order : optional, str The order that a tuple representation will be assumed to be Default: 'C' NzTSelector arguments must be provided (can be passed as part of bounding_box argument))rnrcre)r'r rmrnrercrkr.) r;rZrrmrnrcrerkwargs rr?zCompoundBoundingBox.validates: l$7 8 8 7$ , : &"."> &E /&.'6L  C  s   +     rc||jvSrr|rus rrz CompoundBoundingBox.__contains__sd***rcP|||j||<||S)N)rZ)rprgrXs r_create_bounding_boxz(CompoundBoundingBox._create_bounding_boxs*// /MMYIrc||}||vr |j|S|j||St d|d)Nz)No bounding box is defined for selector: r6)rrrrprr)rr[r4s rrzCompoundBoundingBox.__getitem__se**3//   ' 2 2  " .,,Y77 7W9WWWXX Xrr c0|jj|}||Sr)rmr5r3s r_select_bounding_boxz(CompoundBoundingBox._select_bounding_boxs3D&3V< IrrcX||}|||S)r)rr)rrrTrs rrz"CompoundBoundingBox.prepare_inputss-*0088 **;???rc|j|j|}i}|jD]\}}|||krt |}||||r ||j||id}n| }||t|<t|dkrtd|d|d|S)NTrrzAttempting to fix input z5, but there are no bounding boxes for argument value r6) rmrbrgrrrr~rdrrrr>r-r.) rr1rrbmatching selector_keyrnew_selector_keynew_bboxs r_matching_bounding_boxesz,CompoundBoundingBox._matching_bounding_boxes s$+::4;QQ"&"6"<"<">"> = = L$N+u44#' #5#5  $$^444$$X..+# h%6d /  HH $yy{{H4</001 x==A  >8>>5:>>>  rc|||}t|jdkr|dSt||j|j|j|S)Nrr)rr-rmr rgre)rr1rmatching_bounding_boxess r_fix_input_selector_argz+CompoundBoundingBox._fix_input_selector_arg#sk"&"?"?%"P"P t! " "a ' '*2. .&' "))$+x@@ rci}|jD]'\}}||j||id||<(t ||j|j|j|S)NTr)rrrrrgr rmrg)rr1rrkrrs r_fix_input_bbox_argz'CompoundBoundingBox._fix_input_bbox_arg/s"&"6"<"<">">   L$+/?? h.d,;,,N< ( (#  K   ) )$+x @ @   rrcVt|}|}||}|j|j|r|||}n|||}t|dkr-| }||=| ||}t|tr |j }|} nt|tr d}|j} |j|| |j|S)rrN)rerm)r~rrdrmr@rgrrr-rrr'r rr rrr?re) rrZrfixed_input_keysr1rrnew_fixed_inputsrm bbox_dicts rrzCompoundBoundingBox.fix_inputs<s0 1 1 3 344#''))X&   ) )$+x @ @ =//%@@DD++He<)rkr)NNNrbF)rrrcr~rerSrr)rOr rr)rIrJrKrLrlrrrrkrmsetterrrnrMrrrrrNr?rrrrrrrrrr(r)s@rr r s6@%)!'''''''"       &$$$X$###X#   <<<X<%%X%\    ++++ !&3 3 3 3 [3 j+++ YYY @@@@02       # # # # # # # # rrr)!rL __future__rrrr8 collectionsrtypingrrnumpyr( astropy.unitsr astropy.utilsr __all__r rr?infrr\r_ABCrar r*r,r>rFr rrrrsl#""""" """""" """"""$$$$$$ 4 5 ?M:: lJlJlJlJlJ lJlJlJ`&&'8992 TJTJTJTJTJcgTJTJTJn o9o9o9o9o9o9o9o9d # #:NKKG/G/G/G/G/-G/G/G/T[O[O[O[O[O[O[O[O|@ @ @ @ @ /@ @ @ @ @ r