U e{jC@sdZddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddl Z ddlmZddlmZddgZed d ZGd d d eZee j e jfZed ddZedddZGdddejZGdddeZeddZGdddeZGddde Z!GdddeZ"dS)z4 This module is to contain an improved bounding box N) namedtuple)AnyCallableDictListTupleQuantity) isiterableModelBoundingBoxCompoundBoundingBox _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=))lowerupperselfr\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/modeling/bounding_box.py__repr__1sz_Interval.__repr__cCs t|SN)copydeepcopyrrrrr4sz_Interval.copyc Csd}zt|}Wndtk rvz,t|dkr8|d}tdd|D}Wn"tttfk rpt|YnXYnX|dk}|st|dko|ddkotdd |D}t|r|st|d S) z0Validate the shape of an interval representationz5An interval must be some sort of sequence of length 2rcSsg|] }|qSr)Zto_value.0brrr Csz-_Interval._validate_shape..)))rr)rrrcss|]}t|tjVqdSr) isinstancenpndarrayrrrr Lsz,_Interval._validate_shape..N)r!shape TypeErrorlen ValueErrorAttributeErrorallr )intervalZMESSAGEr$Z valid_shaperrr_validate_shape7s&    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_boundsRs 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&tupler2)r1r*rrrvalidate^s   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! logical_orrr)rr5rrroutsideus z_Interval.outsidecCst|j|j||Sr)r!Zarangerr)r resolutionrrrdomainsz_Interval.domainN)__name__ __module__ __qualname____doc__rr staticmethodr+ classmethodr2r4r!r"r7r9rrrrrs   rreturncCst|tr<||jkr"|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 strinputsindexr'r!Z issubdtypetypeintegerr& IndexError)modelkeyrErrr get_indexs  rKrEcCs |j|Sz3Get the input name corresponding to the input indexrD)rIrErrrget_namesrOc@sdeZdZdZd9eeedddZeddZ eed d d Z eeed d d Z d:eedddZ ed ddZ edddZeeed ddZeeedddZddZejedddZejeeeefd d d!Zed"d#Zd$d%Zd&d'Zd(d)Zd*d+Zee d,d-d.Z!e"e d/d0d1Z#e"e d/d2d3Z$ed4d5Z%e"d6d7d8Z&dS);_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. NCignoredordercCs"||_|||_|||_dSr)_model_validate_ignored_ignored _get_order_order)rrIrSrTrrr__init__s z_BoundingDomain.__init__cCs|jSr)rUrrrrrIsz_BoundingDomain.modelr@cCs|jSr)rYrrrrrTsz_BoundingDomain.ordercCs|jSrrWrrrrrSsz_BoundingDomain.ignored)rTrAcCs*|dkr|j}|dkr&td|d|S)zc Get if bounding_box is C/python ordered or Fortran/mathematically ordered N)rQFzTorder must be either 'C' (C/python order) or 'F' (Fortran/mathematical order), got: r,)rYr')rrTrrrrXs z_BoundingDomain._get_ordercCs 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. )rKrUrrJrrr _get_indexsz_BoundingDomain._get_indexrLcCs t|j|SrM)rOrU)rrErrr _get_namesz_BoundingDomain._get_namecsfddjDS)Ncsg|]}|qSrr_)rrErrrrsz2_BoundingDomain.ignored_inputs..r[rrrrignored_inputssz_BoundingDomain.ignored_inputs)rSrAcs"|dkr gSfdd|DSdS)Ncsg|]}|qSrr^)rrJrrrrsz5_BoundingDomain._validate_ignored..r)rrSrrrrVsz!_BoundingDomain._validate_ignoredcOs 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.Nrc)rrIrirrr fix_inputss z_BoundingDomain.fix_inputscCs 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.Nrc)r input_shaperDrrrprepare_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)rm fill_valuerrr _base_output+sz_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)rq)r_rprmrrrrPsz3_BoundingDomain._all_out_output..N)rangerU n_outputs)rrmrprrsr_all_out_output?s  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)rqr$r!arrayisscalaritem)r valid_output valid_indexrmrpoutputrrr_modify_outputUs    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. )appendr})r valid_outputsr{rmrpoutputsrzrrr_prepare_outputsws z _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)rUrur)rrr{rmrprrrprepare_outputss 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 unitN)getattr)rrrrr_get_valid_outputs_unitsz'_BoundingDomain._get_valid_outputs_unit)evaluaterc 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 )rr) rr valid_inputsr{rmrprrvalid_outputs_unitrrr_evaluate_models $ 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)rnrvr) rrrDrmrprrr{all_outrrr _evaluates z_BoundingDomain._evaluatecCs|dk rt||dddS|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)rZsubokr)rrrrr_set_outputs_unitsz!_BoundingDomain._set_outputs_unit)rcCs6|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 )rUrmrZbbox_with_unitsr3r)rrrDrprmrrrrrr+s z_BoundingDomain.evaluate)NrQ)N)'r:r;r<r=rintrCrZpropertyrIrTrSrXr^r_ralistrVrgabcabstractmethoddictrkrrrnr>rqrvr}rrboolrrrrrrrrrrrPsH   " - ( rPcseZdZdZd?eeefeeedfdd Z d@ddZ e eeefd d d Z e eeefd d d Z ddZddZddZddZddZdAedddZddZddZdd Zed!d"d#Ze d$d%ZdBedd&d'Ze ed d(d)ZdCedd*d+ZdDedd,d-ZedEeee d/d0d1Z!dFed2d3d4Z"e d5d6Z#dGedd7d8Z$d9d:Z%d;d<Z&e'e(e(e(fd d=d>Z)Z*S)Hr 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. NrQ) intervalsrSrTcs8t|||i|_|dkr4|ikr4|j||ddS)NrrT)superrZ _intervals _validate)rrrIrSrT __class__rrrZ`szModelBoundingBox.__init__cCs:dd|jD}|dkr&|j}t||j||jdS)NcSsi|]\}}||qSr)r)rrEr*rrr nsz)ModelBoundingBox.copy..rR)ritemsrWrr rUrY)rrSrrrrrms zModelBoundingBox.copyr@cCs|jS)z1Return bounding_box labeled using input positions)rrrrrryszModelBoundingBox.intervalscsfddjDS)z-Return bounding_box labeled using input namescsi|]\}}||qSrr`)rrEbboxrrrrsz4ModelBoundingBox.named_intervals..)rrrrrrnamed_intervals~sz ModelBoundingBox.named_intervalscCsddg}|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='rB ) rrr~r&rWrarUrr:rDrYjoin)rpartsnamer*rrrrs  zModelBoundingBox.__repr__cCs t|jSr)r&rrrrr__len__szModelBoundingBox.__len__c Cs8z|||jkp|jWSttfk r2YdSXdSNF)r^rrWrHr'r]rrr __contains__szModelBoundingBox.__contains__cCs|||jkSr)r^rr]rrr has_intervalszModelBoundingBox.has_intervalcCs,||}||jkrtS|j||SdS)z.)r&rr3rvaluesrXrUrD)rrTrDrrrr 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 r3rr rrSrvaluerrr__eq__s     zModelBoundingBox.__eq__cCs4||}||jkr |j|t||j|<dS)zBValidate and store interval under key (input index or input name).N)r^rWremoverr4r)rrJrrErrr __setitem__s   zModelBoundingBox.__setitem__cCs<||}||jkr$td|d|j|=|j|dS)zDelete stored intervalzCannot delete ignored input: !N)r^rW RuntimeErrorrr~rrrr __delitem__s   zModelBoundingBox.__delitem__)rcCs|D]\}}|||<qdS)z:Validate passing dictionary of intervals and setting them.Nr)rrrJrrrr_validate_dictszModelBoundingBox._validate_dictcs(fddjjD}fdd|DS)Ncsg|]}|qSrrbrr5rrrrsz;ModelBoundingBox._available_input_index..csg|]}|jkr|qSrr[rrrrrs )rUrD)rZmodel_input_indexrrr_available_input_indexsz'ModelBoundingBox._available_input_indexcCsD||}|dkr |ddd}t|D]\}}|||j|<q(dS)z` Validate passing tuple of tuples representation (or related) and setting them. rQNr)rX enumerater)rrrTrErrrr_validate_sequences  z#ModelBoundingBox._validate_sequencecCs&|jjt|j}|dkr|SdSdS)Nr)rUn_inputsr&rW)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 rrrrrrTrrr_validate_iterables  z#ModelBoundingBox._validate_iterablecCs4|jdkr$t|ts$|||jd<n |||dS)z#Validate and set any representationrrN)rr rrrrrrrrszModelBoundingBox._validateF)rSrT_preserve_ignorecKs>t|tr |j}|r|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' rR)r r rTrSrr)r1rIrrSrTrrfnewrrrr4 s  zModelBoundingBox.validaterhcCsB|}|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) NrR)rkeysrSr r4rrY)rrIri _keep_ignoredrr5rSrrrrk.s zModelBoundingBox.fix_inputscCst|Sr)r&rrrr dimensionJszModelBoundingBox.dimensioncs<jj}|}|dkr(|ddd}fdd|DS)NrQrcsg|]}|qSr)r9rr8rrrrTsz+ModelBoundingBox.domain..)rUrDrX)rr8rTrDrrrr9Ns  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!rorrr- broadcast_tor7r))rrmrDr outside_indexrEr5r7rrr_outsideVs 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&)rrmrDrrr{rrr _valid_indexzs zModelBoundingBox._valid_indexcCst|||\}}g}|sf|D]H}|rZtt|||}t|rN|d}||q||qt|||fS)rlr)rr!rrrxryr~r3)rrmrDr{rrr5Z valid_inputrrrrns    zModelBoundingBox.prepare_inputs)NrQ)N)N)N)N)N)NrQF)F)N)+r:r;r<r=rrrrrCrZrrrrrrrrrrrrrrrrrrrrr?rrr4rkrr9rrrrrn __classcell__rrrrr Hs\      ! $_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__)r1rEignorerrrrrsz_SelectorArgument.__new__T)rScCs|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 )rK)r1rIargumentrSrrrr4sz_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)rEr r&r3rrD _selectorrrr get_selectors  z_SelectorArgument.get_selectorr@cCs 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. )rOrErrIrrrrs 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=rrrrrrr pretty_repr s z_SelectorArgument.pretty_reprrcCsL|j|kr||jS|||kr0|||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)rErrrrrIrrrrget_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 )rErK)rrIrrrr is_argument-s 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. rrrrr named_tuple<s z_SelectorArgument.named_tuple)T)r:r;r<r=rr?rr4rrCrrrrrrrrrrrrs   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||}|dkrg|_n||_|Sr)rr _kept_ignore)r1rrrrrrrcs 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)r~rr)rrIrrrrrrms  z_SelectorArguments.pretty_reprcCsdd|D}||j|S)zGet the list of ignored inputscSsg|]}|jr|jqSr)rrErrrrrrsz-_SelectorArguments.ignore..)extendr)rrrrrr}s z_SelectorArguments.ignorecCs|jS)z$The arguments to persist in ignoring)rrrrrrsz_SelectorArguments.kept_ignore)rcCsg}|D]J}tj|f|}|jdd|DkrHtdt||jd||qt|dkrhtdt|tr|dkr~g}| |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|] }|jqSrrL)rthisrrrrsz/_SelectorArguments.validate..zInput: 'z' has been repeated.rz-There must be at least one selector argument.N) rr4rEr'rOr~r&r rrrr3)r1rI argumentsrrDrr5rrrr4s    z_SelectorArguments.validatecstfdd|DS)z Get the selector corresponding to these inputs Parameters ---------- *inputs : All the processed model evaluation inputs. c3s|]}|jVqdSr)rrrNrrr#sz2_SelectorArguments.get_selector..r)rrDrrNrrs z_SelectorArguments.get_selectorcCst|tot|t|kS)z Determine if this is a reasonable selector Parameters ---------- _selector : tuple The selector to check )r r3r&rrrrr is_selectors 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. c3s|]}|VqdSr)rrrIrrrr#sz6_SelectorArguments.get_fixed_values..rrrrrget_fixed_valuess z#_SelectorArguments.get_fixed_valuesr@cCs |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)rrIr 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')rrIrrErrrrselector_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 ) rpoprrErrrr4r3)rrIrrrrrrreduces  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'rKrr4)rrIrrrrr add_ignore s  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. c3s|]}|VqdSr)r)rrrIrrr#,sz1_SelectorArguments.named_tuple..rrrrrr#s z_SelectorArguments.named_tuple)N)N)r:r;r<r=rrrrrrrrrr?r4rrrrrrrrrrrrrrrrIs$   #  rcsZeZdZdZd5eeefeee e e dfdd Z ddZ d d Zeeeefd d d Zeed ddZejddZeed ddZeddZeddZddZedddZddZed6eee edd d!Zd"d#Z d$d%Z!d&d'Z"ed d(d)Z#e$eeefd d*d+Z%eeefd d,d-Z&d.d/Z'd0d1Z(ed2d3d4Z)Z*S)7r 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. NrQ)bounding_boxes selector_argscreate_selectorrSrTcs8t|||||_t|||_i|_||dSr)rrZ_create_selectorrr4_selector_args_bounding_boxesr)rrrIrrrSrTrrrrZKs zCompoundBoundingBox.__init__cs8fddjD}t|jjtjjdS)Ncs i|]\}}||jjqSr)rrr)rselectorrrrrr]sz,CompoundBoundingBox.copy..)rrrT) rrr rUrrrrrY)rrrrrr\s  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) rrrsplitr~rrrrUr)rrrrZ bbox_reprpartZselector_args_reprrrrrjs  zCompoundBoundingBox.__repr__r@cCs|jSrrrrrrr}sz"CompoundBoundingBox.bounding_boxescCs|jSr)rrrrrrsz!CompoundBoundingBox.selector_argscCs t|j||_tdtdS)NzhOverriding selector_args may cause problems you should re-validate the compound bounding box before use!)rr4rUrr.r/r0rrrrrs cCs|j|jSr)rrrUrrrrnamed_selector_tuplesz(CompoundBoundingBox.named_selector_tuplecCs|jSr)rrrrrrsz#CompoundBoundingBox.create_selectorcCst|rt|S|fSdSr)r r3)rJrrr_get_selector_keysz%CompoundBoundingBox._get_selector_keycCsR||}|j|s$t|d|jj|j}tj|j|||j d|j |<dS)Nz is not a selector!r) rrrr'rrSr r4rUrYr)rrJrrrSrrrrs  zCompoundBoundingBox.__setitem__)rcCs|D]\}}|||<qdSrr)rrrrrrrrszCompoundBoundingBox._validatecCs6t|tr.|j|jko,|j|jko,|j|jkSdSdSr)r r rrrrrrrrs    zCompoundBoundingBox.__eq__F)rrSrTrc Ks`t|tr<|dkr|j}|dkr&|j}|j}|r6|j}|j}|dkrLtd|||||||dS)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' NzTSelector arguments must be provided (can be passed as part of bounding_box argument))rrSrT)r r rrrTrSrr') r1rIrrrrSrTrkwargrrrr4s* zCompoundBoundingBox.validatecCs ||jkSrrr]rrrrsz CompoundBoundingBox.__contains__cCs|j||jd||<||S)Nr)rrUrrrr_create_bounding_boxsz(CompoundBoundingBox._create_bounding_boxcCsD||}||kr|j|S|jdk r0||Std|ddS)Nz)No bounding box is defined for selector: r,)rrrrr)rrJrrrrrs     zCompoundBoundingBox.__getitem__cCs|jj|}||Sr)rrrrrr_select_bounding_boxs z(CompoundBoundingBox._select_bounding_boxcCs||}|||S)rl)rrn)rrmrDrrrrrns 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,) rrrUrrrrrrkrr3r&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 rUr)rrrZmatching_bounding_boxesrrr_fix_input_selector_arg3s z+CompoundBoundingBox._fix_input_selector_argcCsLi}|jD]"\}}|j|j||idd||<qt||j|j|j|S)NTr)rrrkrUr rr)rrrrrrrrr_fix_input_bbox_arg?s z'CompoundBoundingBox._fix_input_bbox_argrhc Cst|}|}||}|j|j|r:|||}n |||}t|dkrl| }||=| ||}t |t r|j }|} nt |trd}|j} |jj|| |j|dS)rjrN)rTr)rrrrrrUrrr&rrkr r rr rrr4rT) rrIriZfixed_input_keysrrrZnew_fixed_inputsrZ bbox_dictrrrrkLs,      zCompoundBoundingBox.fix_inputs)NNrQ)NNNrQF)+r:r;r<r=rrr rrrrrCrZrrrrrsetterr3rrr>rrrrrr?rrr4rrrrrrnrrrrkrrrrrr /sb       5   )#r=rrr. collectionsrtypingrrrrrnumpyr!Z astropy.unitsr Z astropy.utilsr __all__r rr4infrrrKrOABCrPr rrr3rr rrrrs6    p!u  g