B d&5@s^dZddlZddlmZddlZddlmZddlmZddZd d d Z Gd d d e Z dS)zI This modules provides classes for evaluating distributions with bounds. N)Error) boundaries) VARARGS_DELIMc CsRyt||d||}Wntk r2d}YnXyt||d||}Wntk rfd}YnX|dkr~|dkr~d}n|dks|dkrtd|dn||d}y||d||}Wntk rd}YnXy||d ||} Wntk rd } YnX||| d ||d ||} |d | itjf|}|S)aGets bounds for the given parameter from a section in a config file. Minimum and maximum values for bounds are specified by adding `min-{param}` and `max-{param}` options, where `{param}` is the name of the parameter. The types of boundary (open, closed, or reflected) to create may also be specified by adding options `btype-min-{param}` and `btype-max-{param}`. Cyclic conditions can be adding option `cyclic-{param}`. If no `btype` arguments are provided, the left bound will be closed and the right open. For example, the following will create right-open bounds for parameter `foo`: .. code-block:: ini [{section}-{tag}] min-foo = -1 max-foo = 1 This would make the boundaries cyclic: .. code-block:: ini [{section}-{tag}] min-foo = -1 max-foo = 1 cyclic-foo = For more details on boundary types and their meaning, see `boundaries.Bounds`. If the parameter is not found in the section will just return None (in this case, all `btype` and `cyclic` arguments are ignored for that parameter). If bounds are specified, both a minimum and maximum must be provided, else a Value or Type Error will be raised. Parameters ---------- cp : ConfigParser instance The config file. section : str The name of the section. tag : str Any tag in the section name. The full section name searched for in the config file is `{section}(-{tag})`. param : str The name of the parameter to retrieve bounds for. Returns ------- bounds : {Bounds instance | None} If bounds were provided, a `boundaries.Bounds` instance representing the bounds. Otherwise, `None`. zmin-Nzmax-zif specifying bounds for %s, z-you must provide both a minimum and a maximum)Z min_boundZ max_boundz btype-min-{}closedz btype-max-{}open)Z btype_minZ btype_maxz cyclic-{}cyclic) float get_opt_tagr ValueErrorformatupdateZhas_option_tagrBounds) cpsectiontagparamZminbndZmaxbndbndsZbndargsZminbtypeZmaxbtyperrh/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/distributions/bounded.pyget_param_bounds_from_configs:7         rFc Cs.|}|t}|dkri}dgdd|Ddd|Ddd|Ddd|Ddd|Dt|}i}x:|D]2} t|||| } |r| dkrtd | | || <q|Wxf|d ||gD]N} | |krq||| |} y t | } Wntk rYnX| | | iqW| ||f|S) aReturns a bounded distribution based on a configuration file. The parameters for the distribution are retrieved from the section titled "[`section`-`variable_args`]" in the config file. Parameters ---------- cls : pycbc.prior class The class to initialize with. cp : pycbc.workflow.WorkflowConfigParser A parsed configuration file that contains the distribution options. section : str Name of the section in the configuration file. variable_args : str The names of the parameters for this distribution, separated by `prior.VARARGS_DELIM`. These must appear in the "tag" part of the section header. bounds_required : {False, bool} If True, raise a ValueError if a min and max are not provided for every parameter. Otherwise, the prior will be initialized with the parameter set to None. Even if bounds are not required, a ValueError will be raised if only one bound is provided; i.e., either both bounds need to provided or no bounds. additional_opts : {None, dict} Provide additional options to be passed to the distribution class; should be a dictionary specifying option -> value. If an option is provided that also exists in the config file, the value provided will be used instead of being read from the file. Returns ------- cls An instance of the given class. NnamecSsg|]}d|qS)zmin-{})r ).0argrrr sz'bounded_from_config..cSsg|]}d|qS)zmax-{})r )rrrrrrscSsg|]}d|qS)z btype-min-{})r )rrrrrrscSsg|]}d|qS)z btype-max-{})r )rrrrrrscSsg|]}d|qS)z cyclic-{})r )rrrrrrsz'min and/or max missing for parameter %s-) splitrlistkeysrr optionsjoinr rr ) clsrr variable_argsbounds_requiredZadditional_optsrZ special_args dist_argsrboundskeyvalrrrbounded_from_configus0$ J     r'c@seZdZdZddZeddZeddZdd Zd d Z d d Z ddZ ddZ ddZ e ZddZddZdddZed ddZdS)! BoundedDista] A generic class for storing common properties of distributions in which each parameter has a minimum and maximum value. Parameters ---------- \**params : The keyword arguments should provide the names of parameters and their corresponding bounds, as either tuples or a `boundaries.Bounds` instance. cKsx|D]z\}}|dkr(t||<n$t|tjsLt|d|d||<t|tjr |jjdksp|jjdkr td |dq W||_ t t | |_dS)NrZ reflectedzParam {} has one or more zQreflected boundaries. Reflected boundaries can cause issues when used in an MCMC.)itemsrr isinstanceminrmaxwarningswarnr _boundssortedrr_params)selfparamsrrrrr__init__s      zBoundedDist.__init__cCs|jS)z-list of strings: The list of parameter names.)r2)r3rrrr4szBoundedDist.paramscCs|jS)z;dict: A dictionary of the parameter names and their bounds.)r0)r3rrrr$szBoundedDist.boundscsHytfddjDStk rBtddjYnXdS)Nc3s"|]}j||VqdS)N)r0Zcontains_conditioned)rp)r4r3rr sz+BoundedDist.__contains__..z must provide all parameters [%s]z, )allr2KeyErrorr r)r3r4r)r4r3r __contains__s  zBoundedDist.__contains__c stfdd|DS)aApplies any boundary conditions to the given values (e.g., applying cyclic conditions, and/or reflecting values off of boundaries). This is done by running `apply_conditions` of each bounds in self on the corresponding value. See `boundaries.Bounds.apply_conditions` for details. Parameters ---------- \**kwargs : The keyword args should be the name of a parameter and value to apply its boundary conditions to. The arguments need not include all of the parameters in self. Any unrecognized arguments are ignored. Returns ------- dict A dictionary of the parameter names and the conditioned values. cs.g|]&\}}|jkr|j||gqSr)r0Zapply_conditions)rr6r&)r3rrr sz9BoundedDist.apply_boundary_conditions..)dictr*)r3kwargsr)r3rapply_boundary_conditionss z%BoundedDist.apply_boundary_conditionscKs|jf|jf|S)zReturns the pdf at the given values. The keyword arguments must contain all of parameters in self's params. Unrecognized arguments are ignored. Any boundary conditions are applied to the values before the pdf is evaluated. )_pdfr=)r3r<rrrpdfszBoundedDist.pdfcKs tddS)zThe underlying pdf function called by `self.pdf`. This must be set by any class that inherits from this class. Otherwise, a `NotImplementedError` is raised. zpdf function not setN)NotImplementedError)r3r<rrrr>szBoundedDist._pdfcKs|jf|jf|S)a Returns the log of the pdf at the given values. The keyword arguments must contain all of parameters in self's params. Unrecognized arguments are ignored. Any boundary conditions are applied to the values before the pdf is evaluated. )_logpdfr=)r3r<rrrlogpdfszBoundedDist.logpdfcKs tddS)zThe underlying log pdf function called by `self.logpdf`. This must be set by any class that inherits from this class. Otherwise, a `NotImplementedError` is raised. zpdf function not setN)r@)r3r<rrrrA%szBoundedDist._logpdfcCs tddS)z/Return the cdfinv for a single given parameter zinverse cdf not setN)r@)r3rvaluerrr _cdfinv_param.szBoundedDist._cdfinv_paramcKs,i}x"|jD]}||||||<q W|S)zReturn the inverse cdf to map the unit interval to parameter bounds. You must provide a keyword for every parameter. )r4rD)r3kwdsupdatedrrrrcdfinv2s zBoundedDist.cdfinvr)cKstdd|jD}tj||d}i}x$|jD]}tjjdd|d||<q*W|jf|}x|jD]}||||<q\W|S)zDraw random valuecSsg|] }|tfqSr)r)rr6rrrr=sz#BoundedDist.rvs..)dtyperr))size)r4numpyZzerosrandomuniformrG)r3rIrErHZarrZdrawrexprrrrvs;s   zBoundedDist.rvsFcCst|||||dS)aReturns a distribution based on a configuration file. The parameters for the distribution are retrieved from the section titled "[`section`-`variable_args`]" in the config file. Parameters ---------- cp : pycbc.workflow.WorkflowConfigParser A parsed configuration file that contains the distribution options. section : str Name of the section in the configuration file. variable_args : str The names of the parameters for this distribution, separated by `prior.VARARGS_DELIM`. These must appear in the "tag" part of the section header. bounds_required : {False, bool} If True, raise a ValueError if a min and max are not provided for every parameter. Otherwise, the prior will be initialized with the parameter set to None. Even if bounds are not required, a ValueError will be raised if only one bound is provided; i.e., either both bounds need to provided or no bounds. Returns ------- BoundedDist A distribution instance from the pycbc.distribution subpackage. )r")r')r rrr!r"rrr from_configGs zBoundedDist.from_configN)r))F)__name__ __module__ __qualname____doc__r5propertyr4r$r:r=r?r>rBrA__call__rDrGrN classmethodrOrrrrr(s     r()FN) rSr. configparserrrJZpycbcrrrr'objectr(rrrrs   Y R