B d.@spdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZGd d d e ZdS) z3 This modules provides spin distributions of CBCs. N) conversions)Uniform) UniformAngle)UniformPowerLaw) Arbitrary)get_param_bounds_from_config VARARGS_DELIM BoundedDistcsreZdZdZdZddddddd d gZdfd d ZddZddZdddZ ddZ dddZ e ddZ ZS)IndependentChiPChiEffaA distribution such that :math:`\chi_{\mathrm{eff}}` and :math:`\chi_p` are uniform and independent of each other. To ensure constraints are applied correctly, this distribution produces all three components of both spins as well as the component masses. Parameters ---------- mass1 : BoundedDist, Bounds, or tuple The distribution or bounds to use for mass1. Must be either a BoundedDist giving the distribution on mass1, or bounds (as either a Bounds instance or a tuple) giving the minimum and maximum values to use for mass1. If the latter, a Uniform distribution will be used. mass2 : BoundedDist, Bounds, or tuple The distribution or bounds to use for mass2. Syntax is the same as mass1. chi_eff : BoundedDist, Bounds, or tuple; optional The distribution or bounds to use for :math:`chi_eff`. Syntax is the same as mass1, except that None may also be passed. In that case, `(-1, 1)` will be used for the bounds. Default is None. chi_a : BoundedDist, Bounds, or tuple; optional The distribution or bounds to use for :math:`chi_a`. Syntax is the same as mass1, except that None may also be passed. In that case, `(-1, 1)` will be used for the bounds. Default is None. xi_bounds : Bounds or tuple, optional The bounds to use for :math:`\xi_1` and :math:`\xi_2`. Must be :math:`\in (0, 1)`. If None (the default), will be `(0, 1)`. nsamples : int, optional The number of samples to use for the internal kde. The larger the number of samples, the more accurate the pdf will be, but the longer it will take to evaluate. Default is 10000. seed : int, optional Seed value to use for the number generator for the kde. The current random state of numpy will be saved prior to setting the seed. After the samples are generated, the state will be set back to what it was. If None provided, will use 0. Zindependent_chip_chieffmass1mass2xi1xi2chi_effchi_aphi_aphi_sNc st|tr||_n t|d|_t|tr0||_n t|d|_t|trN||_n|dkrZd}t|d|_t|trx||_n|dkrd}t|d|_|dkrd}|ddks|dd ks|d dks|d d krtd td |d |_ td |d|_ t dd|_ t dd|_ |j|j|j |j |j|j|j |j d|_|dkr>d}tj}|dkrVd}tj||jt|d} tj|tdd|jD} tt|j| d| d| d| d| d| d| d| d| d dS) N)r )r )gg?)r)r)rg?rg?gzxi bounds must be in [0, 1)g?)dimr )rr)r)r)r)r r r rrrrrg@)sizecss"|]}|jD] }|VqqdS)N)boundsitems).0Zdistrbrf/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/distributions/spins.py ~sz1IndependentChiPChiEff.__init__..r r r rrrrr) r r r rrrrrr) isinstancer mass1_distrr mass2_distr chieff_distr chia_distr ValueErrorr xi1_distr xi2_distrr phia_distr phis_distr distributionsnumpyrandomZ get_stateseedrvsintZ set_statedictvaluessuperr __init__) selfr r rr xi_boundsnsamplesr+ZrstateZrvalsr) __class__rrr1Gs^                  zIndependentChiPChiEff.__init__c Cst|d|d|d|d|d|d|d|d\ }}}}}}}} } t|||} t||| ||} t|||} t||| ||}t||||}t||||}| d | d |d d k| d |d |d d k@}|S) a/Applies physical constraints to the given parameter values. Parameters ---------- values : {arr or dict} A dictionary or structured array giving the values. Returns ------- bool Whether or not the values satisfy physical r r rrrrr rg@g?)rZ ensurearrayZspin1x_from_xi1_phi_a_phi_sZ'spin2x_from_mass1_mass2_xi2_phi_a_phi_sZspin1y_from_xi1_phi_a_phi_sZ'spin2y_from_mass1_mass2_xi2_phi_a_phi_sZ%spin1z_from_mass1_mass2_chi_eff_chi_aZ%spin2z_from_mass1_mass2_chi_eff_chi_a)r2r/r r rrrrr r_Zs1xZs2xZs1yZs2yZs1zZs2ztestrrr _constraintss"  "  z"IndependentChiPChiEff._constraintscs.tfdd|jD}|s$dS|S)zpDetermines whether the given values are in each parameter's bounds and satisfy the constraints. csg|] }|kqSrr)rdist)paramsrr sz6IndependentChiPChiEff.__contains__..F)allr(r/r8)r2r:isinr)r:r __contains__sz"IndependentChiPChiEff.__contains__rc Ks y |d}Wn&tk r2|jj|dd}YnXy |d}Wn&tk rf|jj|dd}YnXy |d}Wn&tk r|jj|dd}YnXy |d}Wn&tk r|jj|dd}YnXy |d}Wn(tk r|jj|dd}YnXy |d}Wn(tk r:|jj|dd}YnXy |d} Wn(tk rp|jj|dd} YnXy |d } Wn(tk r|j j|dd } YnXd d |j D} t j || d } || d<|| d<|| d<|| d<|| d<|| d<| | d<| | d <| S) zCDraws random samples without applying physical constrains. r )rr rrrrr rcSsg|] }|tfqSr)float)rprrrr;sz/IndependentChiPChiEff._draw..)dtype) KeyErrorrr,r r&r'r!r"r$r%r:r)zeros) r2rkwargsr r rrrrr rrAarrrrr_drawsV        zIndependentChiPChiEff._drawcKs|S)Nr)r2rDrrrapply_boundary_conditionssz/IndependentChiPChiEff.apply_boundary_conditionsc Kst|}dd|jD}tj||d}|}d}xR|r|jfd|i|}||}|} |||||| <|| 7}||}q0W|S)z9Returns random values for all of the parameters. cSsg|] }|tfqSr)r?)rr@rrrr;sz-IndependentChiPChiEff.rvs..)rArr)r-r:r)rCrFr8sum) r2rrDrArE remainingZkeepidxZdrawsmaskZaddptsrrrr,s  zIndependentChiPChiEff.rvsc Cs|}|t}t|t|jks(tdt|||d}t|||d}t|||d}t|||d}t|||d} |d||gdrt| d||gd} nd } |||||| | d S) 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. Returns ------- IndependentChiPChiEff A distribution instance. zTNot all parameters used by this distribution included in tag portion of section namer r rrr3-r4N)r r rrr3r4) splitrset_paramsr#r has_optionjoinr-get) clscpsectionZ variable_argstagr r rrr3r4rrr from_configs  z!IndependentChiPChiEff.from_config)NNNNNNN)r)r)__name__ __module__ __qualname____doc__namerNr1r8r>rFrGr, classmethodrV __classcell__rr)r5rr s& >  3 r )rZr)ZpycbcrZpycbc.distributions.uniformrZpycbc.distributions.angularrZpycbc.distributions.power_lawrZpycbc.distributions.arbitraryrZpycbc.distributions.boundedrrr r rrrrs