B dd@sTddlmZddlmZddlZddlmZddlmZm Z ddZ dd Z d d Z d d Z ddZddZddZddZddZddZeeeeedZd>ddZddZd d!Zd?d#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Z d5d6Z!d7d8Z"d9d:Z#d;d<Z$eee e e!e!e ee e e e e e e#e"e!e!e$e$e$d=Z%dS)@)division)sqrtN)uniform)PIMTSUN_SIcCsddt|dS)z} A0 is a conversion factor between tau0 and mchirp, defined in eqn B3 of the appendix of arxiv.org/abs/0706.4437 g@gUUUUUU@)r)flowr g/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/lalinspiral/sbank/tau0tau3.pyA0sr cCstdt|dS)z} A3 is a conversion factor between tau3 and tau0*M, defined in eqn B3 of the appendix of arxiv.org/abs/0706.4437 g?)r)rr r r A3&sr cCsXddtt|||}t|||d}|t}|dd|d}|||fS)z= Convert tau0-tau3 coordinates to m1-m2 coordinates. g@ gg?g?)rr r)tau0tau3rmtotaletam1r r r tau0tau3_to_m1m2-s rc CsX|t}|t}||}||||}t|||d}t|||d}||fS)z= Convert m1-m2 coordinates to tau0-tau3 coordinates. ggUUUUUU)rr r ) rm2rZm1_sZm2_srrrrr r r m1m2_to_tau0tau3;src sTddddddddgfd d |D}t|rDtd d ||dd \}}|dksd|dkrltd|dd \dkr|dkr|f|d<|dd \}}|dks||kr|}|dks||kr|}|dd \}}|dks||krt|d}|dkr0td|dksRt|||krdt||}||f|d<|dd \}} |dkrtfdd||fD}| dkrtfdd||fD} || f|d<||d|d|dkr|d|d|d}|| d|d|dkrD| d|d|d}||f|d<|S)a Check that the constraints dict does not containt unknown constraints. Check that required constraints are given. To define a 2D parameter space, one needs at least three and possibly four points of intersection. This function preps the constraint dict so that it defines a bounded region (if possible) or crashes (if not). By convention, we require mass1 limits to be set. mass1mass2mratiormchirpspin1spin2durationcsg|]}|kr|qSr r ).0k)known_constraintsr r ^sz+set_default_constraints..zunknown constraints %sz, )NNNz,Must specify both minimum and maximum mass1.z)We use the convention that q = m1/m2 > 1.c3s&|]}fD]}t||VqqdS)N)m1m2_to_mchirp)rrr) mass2_max mass2_minr r sz*set_default_constraints..c3s&|]}fD]}t||VqqdS)N)r#)rrr)r$r%r r r&sg333333?)keyslen ValueErrorjoin setdefaultmaxmin) constraintsZunknown_constraints mass1_min mass1_max mtotal_min mtotal_maxqminqmaxmcminmcmaxr )r r$r%r set_default_constraintsRsN    "     r8cCs||S)Nr )rrr r r m1m2_to_mratiosr9cCs|S)Nr )rrr r r m1m2_to_m1sr:cCs|S)Nr )rrr r r m1m2_to_m2sr;cCs||S)Nr )rrr r r m1m2_to_mtotalsr<cCs$||||||||dS)Ng?r )rrr r r r#sr#)rrrrr绽|=c Csx|D]}||\}}|dkr,|dkr,q t|||}|dkrP|||k}n.|dkrf|||k}n|||k|||k}||}||}q W||fS)zt Return those values of m1,m2 from the input arrays that are consistent with the specified constraints. N)r( m1m2_mappings) rrZm1m2_constraintstolparamZmnZmxparamsincluder r r allowed_m1m2s  rCcCs@d|ddd|d}d|ddd|d}||fS)Ng?g?g@r )Zm_totalrrrr r r mtotal_eta_to_mass1_mass2srDcCs||d}t||S)Ng333333?)rD)Zm_chirprMr r r mchirp_eta_to_mass1_mass2s rFư>cs~|d |d}||}tdd||g}fdd|D}t|dkrTtd}n&t|dkrj|d}ntd||f|S) Nr"rcs*g|]"}t|jkr|jdkr|jqS)r)absimagreal)rr)r?r r r!sz"mchirpm1_to_m2..nanz.csg|] }|fqSr r )rr)m2maxr r r!scsg|] }|fqSr r )rr)m1minr r r!scsg|] }|fqSr r )rr)m1maxr r r!srcsg|]}||fqSr r )rr)valr r r!scsg|]}||fqSr r )rr)rZr r r!srcsg|]}||fqSr r )rr)rZr r r!scsg|]}||fqSr r )rr)rZr r r! sr)NNg?r'Ncsg|]}|t|fqSr )rU)rr)r7r r r!"scsg|]}t||fqSr )rU)rr)r7r r r!#scsg|]}|t|fqSr )rU)rr)r6r r r!0scsg|]}t||fqSr )rU)rr)r7r r r!1scSsg|] }|dqS)rr )rir r r r!4scSsg|] }|dqS)r"r )rr[r r r r!5srz'The requested parameter space is empty.) r8rOZlinspacer,rFarrayrCr)r*rr.r-)rr/ZnptsZ m1m2_pointsZmrminZmrmaxZetaminZetamaxZ mcmax_maxm1Z mcmax_minm2Z mcmax_minm1Z mcmax_maxm2Z mcmin_maxm1Z mcmin_minm2Z mcmin_minm1Z mcmin_maxm2rr lims_tau0 lims_tau3r )rYrXrWrVr7r6rZr tau0tau3_boundsj       $ $               r_ccsZ|d||dd}||dd||d}d}x|tdd||Vq:WdS)z This is a generator for random total mass values corresponding to a uniform distribution of mass pairs in (tau0, tau3) space. See also urand_eta_generator(), and see LIGO-T1300127 for details. r"g@g۶m۶mۿrN)r)r2r3alphabetanr r r urand_mtotal_generatorCs rcccsV|td||d}||dd||d}x|ttdd|Vq6WdS)z This is a generator for random eta (symmetric mass ratio) values corresponding to a uniform distribution of mass pairs in (tau0, tau3) space. See also urand_mtotal_generator(), and see LIGO-T1300127 for details. r"r'rN)rr)Zeta_minZeta_maxr`rar r r urand_eta_generatorPsrdcksXt|}t|f|\}}|\}}|\}}|d\}} |d\} } |d\} } |d\}}t|}t|}||}ddlm}d}x|||}|||||}||||}|dkrq|t}|d td|}||}| |kr| krnq||kr | krnq| |kr$| krnq|||krB|krnq||fVqWd S) a_ This is a generator for random (m1, m2) pairs that are uniformly distributed in (tau0, tau3) space, subject to the constraints given in the inputs. @param flow UNDOCUMENTED @param constraints: must specify a mass1 range; mtotal, q, and tau0 ranges are optional. The arguments for each of these keywords should be a tuple of (min, max). E.g., mtotal = (50, 100) constrains 50 <= mtotal < 100. Example: >>> urand_tau0tau3 = urand_tau0tau3_generator(40, mass1 = (1, 99), mtotal = (55, 100), mratio = (1, 10)) >>> urand_tau0tau3.next() (49.836271184652254, 6.6152675269639829) >>> urand_tau0tau3.next() (46.701700599815645, 14.07462196069671) Equivalently, use a dictionary with **: >>> urand_tau0tau3 = urand_tau0tau3_generator(40, **{"mass1": (1, 99), "mtotal": (55, 100), "mratio": (1, 10)}) rrrrr)rgg?g?N)r8r_r r numpy.random.mtrandrrr)rr/r]r^Ztau0_minZtau0_maxZtau3_minZtau3_maxr0r1r%r$r2r3r4r5Z_A0Z_A3ZA0_A3rZminus_five_thirdsrmtotrrrr r r urand_tau0tau3_generator]s6      rgcksV|dr|d|dr(|dx(t|f|D]\}}||||dVq6WdS)zb Wrapper for urand_tau0tau3_generator() to remove spin options for EOBNRv2 waveforms. rr)bankN)has_keypoprg)r tmplt_classrhr/rrr r r nonspin_param_generators     rlc ks|dd\}}|dtdtd|td|f}td|td|f}xt|f|D]\}} t|| | |} d | krd krnnt|} t|} n2d | krd krnnt|} t|} ntd ||| | | |d VqXWdS)a~ Specify the min and max mass of the bigger component, then the min and max mass of the total mass. This function includes restrictions on q and chi based on IMRPhenomB's range of believability. Ref: http://arxiv.org/pdf/0908.2090 @param flow: Lower frequency at which to generate waveform @param tmplt_class: Template generation class for this waveform @param bank: sbank bank object @param kwargs: must specify a component_mass range; mtotal, q, chi, and tau0 ranges are optional. If no chi is specified, the IMRPhenomB limits will be used. See urand_tau0tau3_generator for more usage help. r)gg?rzNPhenomB: spin2 limits not implemented. Using spin1 limits for both components.g333333g333333?gg? r"zmass ratio out of range)rhN)rjWarningr-r.rgrr*) rrkrhkwargsZsminZsmaxZchi_low_boundsZchi_high_boundsrrrSrrr r r IMRPhenomB_param_generators   rqc ks|dd\}}|d||f\}}td|td|}}td|td|}}xdt|f|D]T\}} t|| | |} | dkrt||} t||} ntd||| | | |dVq^Wd S) au Generate random parameters for the IMRPhenomC waveform model. Specify the min and max mass of the bigger component, then the min and max mass of the total mass. This function includes restrictions on q and chi based on IMRPhenomC's range of believability, namely q <=20 and |chi| <= 0.9. @param flow: low frequency cutoff @param tmplt_class: Template generation class for this waveform @param bank: sbank bank object @param kwargs: constraints on waveform parameters. See urand_tau0tau3_generator for more usage help. If no spin limits are specified, the IMRPhenomC limits will be used. r)gg?rgg?zmass ratio out of range)rhN)rjr-r.rgrr*) rrkrhrpZs1minZs1maxs2mins2maxrrrSrrr r r IMRPhenomC_param_generators  ruc+s|dd\}}d|krXd|krXd|krX|d|d|dfdd}n&|dd |d d fd d}xt|f|D]\}}|||\} } ||} || d || d | } || d || d | } t| | }t| d | ||| d |}t| d | ||| d |}t||}|| |||}||||||d}|dk rx|j|ks|dk r|j|krq|VqWdS)z Specify the min and max mass of the bigger component, the min and max mass of the total mass and the min and max values for the z-axis spin angular momentum. r)NNZns_bh_boundary_massZbh_spinZns_spincs |kr n|krnfS)Nr )rr)bh_spin_boundsns_bh_boundaryns_spin_boundsr r spin_boundssz1aligned_spin_param_generator..spin_boundsr)gg?rcsfS)Nr )rr)spin1bspin2br r ry srr")rhN)rjrgrr-r.Zdur)rrkrhrpZdur_minZdur_maxryrr spin1_bounds spin2_boundsrfZchis_minZchis_maxZchisrsrtrrtr )rvrwrxrzr{r aligned_spin_param_generators2      "" rcksl|dd}|dd}xLt|f|D]:\}}t|}t|} tdtj} tddtj} tdtj} tddtj} |t| }|t| t| }|t| t| }| t| }| t| t| }| t| t| }tdtj}tddtj}tddtj}tdtj}tddtj}|||||||||||||||dVq(WdS)zB Currently a stub to test precessing template generation. r)gg?rrr')rhN)rjrgrrOpicossin)rrkrhrpr|r}rrspin1magZspin2mag spin1ang1 spin1ang2Z spin2ang1Z spin2ang2spin1zspin1xspin1yZspin2zZspin2xZspin2ythetaphipsiiota orb_phaser r r &double_spin_precessing_param_generator$s,      rc ks|dd}|dd}xt|f|D]\}}t|}tdtj} tddtj} |t| } |t| t| } |t| t| } tdtj}tddtj}tddtj}tdtj}|||| | | |||||d Vq&WdS)zB Currently a stub to test precessing template generation. r)gg?rrr')rhN)rjrgrrOrrr)rrkrhrpr|r}rrrrrrrrrrrrr r r &single_spin_precessing_param_generatorBs     rcKstdS)N)NotImplementedError)rrkrhrpr r r SpinTaylorT4_param_generator[src ks|dr|d|dr(|dxt|f|D]t\}}tdtj}tddtj}tddtj}tdtj} tddtj} |||dddddd||| || |Vq6WdS)zb Wrapper for urand_tau0tau3_generator() to remove spin options for EOBNRv2 waveforms. rrrr'N)rirjrgrrOr) rrkrhr/rrrrrrrr r r nonspin_hom_param_generator`s      r)Z IMRPhenomBZ IMRPhenomCZ IMRPhenomDZTaylorF2Z IMRPhenomPZ IMRPhenomPv2ZTaylorF2RedSpinZEOBNRv2ZSEOBNRv1ZSEOBNRv2ZSEOBNRv2_ROM_DoubleSpinZSEOBNRv2_ROM_DoubleSpin_HIZSEOBNRv4Z SEOBNRv4_ROMZ SpinTaylorT4Z SpinTaylorF2ZSpinTaylorT5FourierZSEOBNRv3Z EOBNRv2HM_ROMZEOBNRv2HM_ROM_AmpMaxZEOBNRv2HM_ROM_PhaseMax)r=)rG)& __future__rmathrrOrerZlalrrr r rrr8r9r:r;r<r#r>rCrDrFrUr_rcrdrgrlrqrurrrrrZ proposalsr r r r sp   B  a  A )!/