a ߙfb9@sdZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddlZddlZddgZed d ZGd d d eZeej ejfZGd ddejZedddZedddZGdddeZeddZGdddeZGddde Z!GdddeZ"dS)z4 This module is to contain an improved bounding box N) namedtuple)DictListTupleCallableAny) isiterableQuantityModelBoundingBoxCompoundBoundingBox _BaseIntervalz lower upperc@s\eZdZdZddZddZeddZedd Z ed d Z e j d d dZ ddZdS) _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 : Contructs a valid interval outside : Determine which parts of an input array are outside the interval. domain : Contructs a discretization of the points inside the interval. cCsd|jd|jdS)NzInterval(lower=z, upper=))lowerupperselfrEz-_Interval._validate_shape..)))rr )r rr css|]}t|tjVqdSr) isinstancenpndarrayrrrr Lrz,_Interval._validate_shape..N)r"shape TypeErrorlen ValueErrorAttributeErrorallr)intervalZMESSAGEr%Z valid_shaperrr_validate_shape9s"   z_Interval._validate_shapecCs<t|t|kr2td|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" asanyarrayr*warningswarnRuntimeWarning)clsrrrrr_validate_boundsQs z_Interval._validate_boundscCs@||t|dkr$t|d}nt|}||d|dS)z Construct and validate an interval Parameters ---------- interval : iterable A representation of the interval. Returns ------- A validated interval. rr)r,r'tupler3)r2r+rrrvalidateZs   z_Interval.validate)_inputcCst||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"Z logical_orrr)rr6rrroutsideqs z_Interval.outsidecCst|j|j||Sr)r"Zarangerr)r resolutionrrrdomainsz_Interval.domainN)__name__ __module__ __qualname____doc__rr staticmethodr, classmethodr3r5r"r#r7r9rrrrrs   rc@seZdZdZddZddZejedddZ eje e e e fd d d Z e d d ZddZddZddZddZe edddZeedddZeedddZe ddZed d!d"Zd#S)$_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. cCs ||_dSr)_modelrmodelrrr__init__sz_BoundingDomain.__init__cOs tddS)NzPThis bounding box is fixed by the model and does not have adjustable parameters.NotImplementedError)rargskwargsrrr__call__sz_BoundingDomain.__call__ fixed_inputscCs tddS)C 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. z,This should be implemented by a child class.NrE)rrCrKrrr fix_inputss z_BoundingDomain.fix_inputsreturncCs tddS)| Get prepare the inputs with respect to 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_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 all_out: bool if all of the inputs are outside the bounding_box z1This has not been implemented for BoundingDomain.NrE)r input_shapeinputsrrrprepare_inputssz_BoundingDomain.prepare_inputscCst||S)a Create a baseline output, assuming that the entire input is outside the bounding box Parameters ---------- 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 array of the correct shape containing all fill_value )r"zeros)rQ fill_valuerrr _base_outputsz_BoundingDomain._base_outputcs"fddtjjDdfS)a Create output if all inputs are outside the domain Parameters ---------- 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 ------- A full set of outputs for case that all inputs are outside domain. csg|]}qSr)rV)r_rUrQrrrrsz3_BoundingDomain._all_out_output..N)rangerA n_outputs)rrQrUrrXr_all_out_outputs  z_BoundingDomain._all_out_outputcCs>|||}|jst|}n|||<t|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)rVr%r"Zarrayisscalaritem)r valid_output valid_indexrQrUoutputrrr_modify_outputs    z_BoundingDomain._modify_outputc Cs(g}|D]}||||||q|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. )appendra)r valid_outputsr_rQrUoutputsr^rrr_prepare_outputssz _BoundingDomain._prepare_outputscCs"|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)rArZre)rrcr_rQrUrrrprepare_outputs8s z_BoundingDomain.prepare_outputs) with_unitscCs|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 ZunitN)getattr)rcrgrrr_get_valid_outputs_unitPsz'_BoundingDomain._get_valid_outputs_unit)evaluatergc Cs(||}|||}||||||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 )rirf) rrj valid_inputsr_rQrUrgrcvalid_outputs_unitrrr_evaluate_modelas z_BoundingDomain._evaluate_modelc Cs:|||\}}}|r"|||S|||||||SdS)a Perform model evaluation 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 N)rSr[rm) rrjrRrQrUrgrkr_all_outrrr _evaluates   z_BoundingDomain._evaluatecCs|durt||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 NFrr )rdrlrrr_set_outputs_unitsz!_BoundingDomain._set_outputs_unit)rjcCs6|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 )rArQroZbbox_with_unitsr4rq)rrjrRrUrQrdrlrrrrjs   z_BoundingDomain.evaluateN)r:r;r<r=rDrIabcabstractmethoddictrMrrrSr>rVr[rarerfboolrirrmrorqrjrrrrr@s0 " # & r@indexcCs |j|Sz3Get the input name corresponding to the input indexrR)rCrwrrrget_namesrzrNcCst|tr<||jvr"|j|}qtd|d|jdndtt|tjrd|krht |jkrrnn|}qt d|dt |jdntd|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: r-rz Integer key: z must be non-negative and < z Key value: z must be string or integer.) r!strrRrwr(r"Z issubdtypetypeZintegerr' IndexError)rCkeyrwrrr get_indexs  rcseZdZdZdNeeefeeedfdd Z dOddZ e eeefd d d Z e ed d d Z e eed ddZedddZe eeefd ddZe eed ddZddZed ddZeeedddZddZd d!Zd"d#Zd$d%ZdPeed&d'd(ZdQed)d*d+Zd,d-Zd.d/Zd0d1Zed2d3d4Z dRed)d5d6Z!e ed d7d8Z"dSed)d9d:Z#dTed)d;d<Z$e%dUeed=d>d?Z&dVedAdBdCZ'e dDdEZ(dWed)dFdGZ)dHdIZ*dJdKZ+e,e-e-e-fd dLdMZ.Z/S)Xr a A model's bounding box Parameters ---------- intervals : dict A dictionary containing all the intervals for each model input keys -> 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. NC) intervalsignoredordercsFt|||_|||_i|_|dkrB|ikrB|j||ddS)Nrr)superrD_order_validate_ignored_ignored _intervals _validate)rrrCrr __class__rrrDs   zModelBoundingBox.__init__cCs:dd|jD}|dur&|j}t||j||jdS)NcSsi|]\}}||qSrrp)rrwr+rrr sz)ModelBoundingBox.copy..rr)ritemsrrr rAr)rrrrrrrs zModelBoundingBox.copyrNcCs|jS)z1Return bounding_box labeled using input positions)rrrrrr'szModelBoundingBox.intervalscCs|jSrrrrrrr,szModelBoundingBox.ordercCs|jSrrrrrrr0szModelBoundingBox.ignoredrvcCs t|j|Srx)rzrA)rrwrrr _get_name4szModelBoundingBox._get_namecsfddjDS)z-Return bounding_box labeled using input namescsi|]\}}||qSrr)rrwbboxrrrr;rz4ModelBoundingBox.named_intervals..)rrrrrrnamed_intervals8sz ModelBoundingBox.named_intervalscsfddjDS)Ncsg|]}|qSrr)rrwrrrr?rz3ModelBoundingBox.ignored_inputs..rrrrrignored_inputs=szModelBoundingBox.ignored_inputscCsddg}|jD]\}}|d|d|q|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='r{ ) rrrbr'rrrArr:rRrjoin)rpartsnamer+rrrrAs " zModelBoundingBox.__repr__cCs 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. )rrArrrrr _get_indexTszModelBoundingBox._get_index)rrOcs"|dur gSfdd|DSdS)Ncsg|]}|qSr)r)rrrrrrbrz6ModelBoundingBox._validate_ignored..r)rrrrrr^sz"ModelBoundingBox._validate_ignoredcCs t|jSr)r'rrrrr__len__dszModelBoundingBox.__len__c Cs6z|||jvp|jWSttfy0YdS0dSNF)rrrr~r(rrrr __contains__gszModelBoundingBox.__contains__cCs|||jvSr)rrrrrr has_intervalmszModelBoundingBox.has_intervalcCs,||}||jvrtS|j||SdS)z.)r'rr4listvaluesrrArR)rrrRrrrr bounding_boxs  zModelBoundingBox.bounding_boxcCs@t|tr||kSt|tr8|j|jko6|j|jkSdSdS)z?Note equality can be either with old representation or new one.FN)r!r4rr rrrvaluerrr__eq__s    zModelBoundingBox.__eq__cCs4||}||jvr |j|t||j|<dS)zBValidate and store interval under key (input index or input name).N)rrremoverr5r)rrrrwrrr __setitem__s   zModelBoundingBox.__setitem__cCs<||}||jvr$td|d|j|=|j|dS)zDelete stored intervalzCannot delete ignored input: !N)rr RuntimeErrorrrbrrrr __delitem__s   zModelBoundingBox.__delitem__)rcCs|D]\}}|||<qdS)z:Validate passing dictionary of intervals and setting them.Nr)rrrrrrr_validate_dictszModelBoundingBox._validate_dictcCs>||}|dkr |ddd}t|D]\}}|||<q(dS)zNValidate passing tuple of tuples representation (or related) and setting them.rNr)r enumerate)rrrrwrrrr_validate_sequences  z#ModelBoundingBox._validate_sequencecCs&|jjt|j}|dkr|SdSdS)Nr)rAn_inputsr'r)rrrrr _n_inputsszModelBoundingBox._n_inputscCsPt||jkr*tdt|d|jdt|tr@||n |||dS)z,Validate and set any iterable representationzFound z" intervals, but must have exactly r-N)r'rr(r!rtrrrrrrrr_validate_iterables   z#ModelBoundingBox._validate_iterablecCs.|jdkrt|ts||d<n |||dS)z#Validate and set any representationrrN)rr!rtrrrrrrs zModelBoundingBox._validatercKs4t|tr|j}|j}|i|||d}|||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 rrr)r2rCrrrrHnewrrrr5s   zModelBoundingBox.validateFrJcCsB|}|D] }||=q|r(|j}nd}tj||j||jdS)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)rkeysrr r5rr)rrCrK _keep_ignoredrr6rrrrrMs  zModelBoundingBox.fix_inputscCst|Sr)r'rrrr dimensionszModelBoundingBox.dimensioncs<jj}|}|dkr(|ddd}fdd|DS)Nrrcsg|]}|qSr)r9rr8rrrrrz+ModelBoundingBox.domain..)rArRr)rr8rrRrrrr9s  zModelBoundingBox.domaincCsdd}tj|td}t|D]@\}}t|}t||||}d||<|rd}q\q||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)ZdtypeT)r"rTrurr. broadcast_tor7r*)rrQrRrn outside_indexrwr6r7rrr_outsides zModelBoundingBox._outsidecCs@|||\}}tt|}t|ddkr8d}||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)rr" atleast_1dZ logical_notZnonzeror')rrQrRrrnr_rrr _valid_indexAs zModelBoundingBox._valid_indexcCst|||\}}g}|sf|D]H}|rZtt|||}t|rN|d}||q||qt|||fS)rPr)rr"rrr\r]rbr4)rrQrRr_rnrkr6Z valid_inputrrrrS[s    zModelBoundingBox.prepare_inputs)Nr)N)N)N)N)N)N)Nr)F)N)0r:r;r<r=rintrrr|rDrpropertyrrrrrrrrrrrrrrrrrrrrtrrrrrr?r5rMrr9rrrrrS __classcell__rrrrr s^       $_BaseSelectorArgumentz index ignorecsveZdZdZfddZededddZdd Ze d d d Z d dZ e dddZ ed ddZddZZS)_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. cst|||}|Sr)r__new__)r2rwignorerrrrrsz_SelectorArgument.__new__T)rcCs|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)r2rCargumentrrrrr5sz_SelectorArgument.validatecGs2||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)rwrr'r4rrR _selectorrrr get_selectors  z_SelectorArgument.get_selectorrNcCs 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. )rzrwrBrrrrs z_SelectorArgument.namecCsd||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=rrrrBrrr pretty_reprs z_SelectorArgument.pretty_reprrcCsL|j|vr||jS|||vr0|||St||d|dS)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 N)rwrrrrrCrrrrget_fixed_values  z!_SelectorArgument.get_fixed_valuecCs|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 )rwr)rrCrrrr is_arguments z_SelectorArgument.is_argumentcCs|||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. rrBrrr named_tuples z_SelectorArgument.named_tuple)T)r:r;r<r=rr?rur5rr|rrrtrrrrrrrrrs   rcseZdZdZdZd!eeedfdd ZddZ e dd Z e d d Z e d"ed d dZddZddZedddZedddZddZddZddZdd ZZS)#_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. N)input_ kept_ignorecs(t||}|durg|_n||_|Sr)rr _kept_ignore)r2rrrrrrr&s z_SelectorArguments.__new__cCs:dg}|D]}|d||q |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)rbrr)rrCrrrrrr0s  z_SelectorArguments.pretty_reprcCsdd|D}||j|S)zGet the list of ignored inputscSsg|]}|jr|jqSr)rrwrrrrrrErz-_SelectorArguments.ignore..)extendr)rrrrrrBs z_SelectorArguments.ignorecCs|jS)z$The arguments to persist in ignoring)rrrrrrJsz_SelectorArguments.kept_ignore)rcCsg}|D]L}tj|g|R}|jdd|DvrJtdt||jd||qt|dkrjtdt|tr|durg}| |j |t ||S)ae Construct a valid Selector description for a CompoundBoundingBox. Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. arguments : The individual argument informations kept_ignore : Arguments to persist as ignored cSsg|] }|jqSrrv)rthisrrrrbrz/_SelectorArguments.validate..zInput: 'z' has been repeated.rz-There must be at least one selector argument.N) rr5rwr(rzrbr'r!rrrr4)r2rC argumentsrrRrr6rrrr5Os    z_SelectorArguments.validatecstfdd|DS)z Get the selector corresponding to these inputs Parameters ---------- *inputs : All the processed model evaluation inputs. csg|]}|jqSr)rrryrrrzrz3_SelectorArguments.get_selector..r)rrRrryrrqs z_SelectorArguments.get_selectorcCst|tot|t|kS)z Determine if this is a reasonable selector Parameters ---------- _selector : tuple The selector to check )r!r4r'rrrrr is_selector|s z_SelectorArguments.is_selectorrcstfdd|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. csg|]}|qSr)rrrCrrrrrz7_SelectorArguments.get_fixed_values..rrrrrget_fixed_valuess z#_SelectorArguments.get_fixed_valuesrNcCs |D]}|||rdSqdS)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 TFN)r)rrCr selector_argrrrrs  z_SelectorArguments.is_argumentcCs8t|D]\}}|||r|Sqt|ddS)a2 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.N)rrr()rrCrrwrrrrselector_indexs   z!_SelectorArguments.selector_indexcCs<t|}||||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 ) rpoprrwrrrr5r4)rrCrrrrrrreduces  z_SelectorArguments.reducecCs4|||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.)rr(rrr5)rrCrrrrr add_ignores  z_SelectorArguments.add_ignorecstfdd|DS)z Get a tuple of selector argument tuples using input names Parameters ---------- model : `~astropy.modeling.Model` The Model these selector arguments are for. csg|]}|qSr)r)rrrCrrrrz2_SelectorArguments.named_tuple..rrBrrrrs z_SelectorArguments.named_tuple)N)N)r:r;r<r=rrrrrrrrrr?r5rrrtrrurrrrrrrrrrr s$   !  rcsbeZdZdZd6eeefeee dfdd Z ddZ d d Z e eeefd d d Ze ed ddZejddZe ed ddZe ddZe e d ddZeddZddZedddZddZed7ee d d!d"Zd#d$Zd%d&Zd'd(Zed d)d*Z e!eeefd d+d,Z"eeefd d-d.Z#d/d0Z$d1d2Z%ed3d4d5Z&Z'S)8r 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. Nr)bounding_boxes selector_argscreate_selectorrcs:t|||_||_t|||_i|_||dSr) rrDr_create_selectorrr5_selector_args_bounding_boxesr)rrrCrrrrrrrD s  zCompoundBoundingBox.__init__cs8fddjD}t|jjtjjdS)Ncs i|]\}}||jjqSr)rrr)rZselectorrrrrrsz,CompoundBoundingBox.copy..)rrr) rrr rArrrrr)rrrrrrs  zCompoundBoundingBox.copyc Csddg}|jD]L\}}|d}|d|d|d|D]}|d|qHq|d|j|jd}|d |d|D]}|d|q|d d |S) NzCompoundBoundingBox(z bounding_boxes={rrz = rz rz selector_args = r) rrrsplitrbrrrrAr)rrrrZ bbox_reprpartZselector_args_reprrrrrs  zCompoundBoundingBox.__repr__rNcCs|jSrrrrrrr1sz"CompoundBoundingBox.bounding_boxescCs|jSr)rrrrrr5sz!CompoundBoundingBox.selector_argscCs t|j||_tdtdS)NzhOverriding selector_args may cause problems you should re-validate the compound bounding box before use!)rr5rArr/r0r1rrrrr9scCs|j|jSr)rrrArrrrnamed_selector_tuple@sz(CompoundBoundingBox.named_selector_tuplecCs|jSr)rrrrrrDsz#CompoundBoundingBox.create_selectorcCs|jSrrrrrrrHszCompoundBoundingBox.ordercCst|rt|S|fSdSr)rr4)rrrr_get_selector_keyLsz%CompoundBoundingBox._get_selector_keycCsH||}|j|s$t|dtj|j||jj|jd|j |<dS)Nz is not a selector!r) rrrr(r r5rArrr)rrrrrrrrSs   zCompoundBoundingBox.__setitem__)rcCs|D]\}}|||<qdSrr)rrrrrrrr\szCompoundBoundingBox._validatecCs6t|tr.|j|jko,|j|jko,|j|jkSdSdSr)r!r rrrrrrrr`s    zCompoundBoundingBox.__eq__)rrcCsRt|tr2|dur|j}|dur&|j}|j}|j}|durBtd||||||S)ac 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 respresentations 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' NzUSelector arguments must be provided (can be passed as part of bounding_box argument)!)r!r rrrrr()r2rCrrrrrrrr5hs zCompoundBoundingBox.validatecCs ||jvSrrrrrrrsz CompoundBoundingBox.__contains__cCs|j||jd||<||S)Nr)rrArrrr_create_bounding_boxsz(CompoundBoundingBox._create_bounding_boxcCsD||}||vr|j|S|jdur0||Std|ddS)Nz)No bounding box is defined for selector: r-)rrrrr)rrrrrrrs     zCompoundBoundingBox.__getitem__cCs|jj|}||Sr)rrrrrr_select_bounding_boxs z(CompoundBoundingBox._select_bounding_boxcCs||}|||S)rP)rrS)rrQrRrrrrrSs z"CompoundBoundingBox.prepare_inputsc Cs|j|j|}i}|jD]\\}}|||krt|}||||rf|j|j||idd}n| }||t |<qt |dkrt d|d|d|S)NTrrzAttempting to fix input z5, but there are no bounding boxes for argument value r-) rrrArrrrrrMrr4r'r() rrrrZmatching selector_keyrZnew_selector_keyZnew_bboxrrr_matching_bounding_boxess"      z,CompoundBoundingBox._matching_bounding_boxescCs@|||}t|jdkr"|dSt||j|j|j|SdS)Nrr)rr'rr rAr)rrrZmatching_bounding_boxesrrr_fix_input_selector_args  z+CompoundBoundingBox._fix_input_selector_argcCsLi}|jD]"\}}|j|j||idd||<qt||j|j|j|S)NTr)rrrMrAr rr)rrrrrrrrr_fix_input_bbox_args z'CompoundBoundingBox._fix_input_bbox_argrJc Cst|}|}||}|j|j|r:|||}n |||}t|dkrl| }||=| ||}t |t r|j }|} nt |trd}|j} |jj|| |j|dS)rLrN)rr)rrrrrrArrr'rrMr!r rr rrr5r) rrCrKZfixed_input_keysrrrZnew_fixed_inputsrZ bbox_dictrrrrMs&       zCompoundBoundingBox.fix_inputs)Nr)NNr)(r:r;r<r=rrr rrr|rDrrrrrsetterr4rrrr>rrrtrrr?r5rrrrrrSrrrrMrrrrrr sN           )#r=rrr collectionsrtypingrrrrrZ astropy.utilsrZ astropy.unitsr r/Znumpyr"__all__r rr5infrABCr@rrzrr rrr4rr rrrrs6    jW  c