B 1agiB@sddlZddlZddlmZddlmZddlmZddl Z ddl Z ddl Tddlm ZddlZeedddlmZWdQRXdZed Zd Zed d d ZeeZeeZeefZeefZeedZeefZdeiZ edZ!de!Z"e!e d<e!e d<e"e d<e"e d<de d<ddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@(Z#dAdBZ$dCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddAZ%e&e%'Z(dde(DZ)e*e+e)e(Z,ddZ-ddZ.dde%'DZ/d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?gZ0ddddddddd'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d&ddddd!dddddddddg0Z1ej23ej24e5Z6e7ej28e6ddZ9e :e9Z;WdQRXe7ej28e6ddZ9e :e9ZD]"\Z?Z@e?e=kr|e@Ae=e?q|Wx(e=>D]\Z?Z@e?eD]"\Z?Z@e?eBkre@AeBe?qWx(eB>D]\Z?Z@e?eDZCdddddddddddddddZDx>eDE>D].\Z?Z@de@krde@kre@deDe?d<qWdddddddddddddddddddddМZFGdd҄deGZHGddԄdeGZIddքZJddefddلZKddd݄ZLddd߄ZMdddZNdddZOdddZPdddZQdddZRdddZSddlmZTGdddeTZUdddZVdS)N) gaussian_kde)interp1d)cumtrapz)*ignore)bounded_1d_kdegn@gRhV?g?g@gR@zfigure.figsize g?zaxes.labelsizezlegend.fontsizezxtick.labelsizezytick.labelsizetruez text.usetexZ GW200316IZ GW200311LZ GW200225BZ GW200224HZ GW200219DZ GW200208FZ GW200202FZ GW200129DZ GW200115A GW191222AZ GW191216GZ GW191215GZ GW191204GZ GW191129F GW191109A GW190408A GW190412A GW190421A GW190425A GW190503A GW190512A GW190513A GW190517A GW190519A GW190521A GW190521B GW190602A GW190630A GW190706A GW190707A GW190708A GW190720A GW190727A GW190728A GW190814A GW190828A GW190828B GW190910A GW190915A GW190924A)( GW200316A GW200311B GW200225A GW200224B GW200219A GW200208A GW200202A GW200129AGW200115r GW191216A GW191215A GW191204B GW191129Ar rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&cCst|S)N) o3_name_dict)Zcat_namer5C/work/yifan.wang/ringdown/LVKringdownfile/gwtc3/rin/pyring/utils.py cat_to_snamepsr7Z S200316bjZ S200311bgZS200225qZ S200224caZ S200219acZS200208qZ S200202acZS200129mZS200115jZS191222nZ S191216apZS191215wZS191204rZS191129uZS191109dZ S190408anZS190412mZS190413iZ S190413acZ S190421arZ S190424aoZS190425zZS190426cZ S190503bfZ S190512atZ S190513bmZS190514nZS190517hZ S190519bjZS190521gZS190521rZS190527wZ S190602aqZS190620eZ S190630agZ S190701ahZ S190706aiZS190707qZ S190708apZ S190719anZS190720aZS190727hZS190728qZ S190731aaZS190803eZ S190814bvZS190828jZS190828lZS190909wZS190910sZ S190915akZS190924hZS190929dZS190930sZS150914ZS151012ZS151226ZS170104ZS170608ZS170729ZS170809ZS170814ZS170817ZS170818ZS170823)Ar'r(r)r*r+r,r-r.r/r r0r1r2r3r rrZ GW190413AZ GW190413BrZ GW190424ArZ GW190426ArrrZ GW190514ArrrrZ GW190527ArZ GW190620ArZ GW190701ArrrZ GW190719Arrr Z GW190731AZ GW190803Ar!r"r#Z GW190909Ar$r%r&Z GW190929AZ GW190930A GW150914AZ GW151012A GW151226A GW170104A GW170608AZ GW170729A GW170809A GW170814AZ GW170817A GW170818A GW170823AcCsg|] }t|qSr5) catalog_id_to_superevent_id_dict).0vr5r5r6 srCcCst|S)z0 Get GraceDB superevent ID from catalog ID. )r@)Z catalog_idr5r5r6 cid_to_sidsrDcCst|S)z0 Get catalog ID from GraceDB superevent ID. ) superevent_id_to_catalog_id_dict)Z superevent_idr5r5r6 sid_to_cidsrFcCs g|]}d|ksd|kr|qS)ZGW15ZGW17r5)rAkr5r5r6rCsr8r9r:r;r<r=r>r?r3r2r1r0r/r.r-r,r+r*r)r(r'z colors.pklrbzparameters_o3a.pklzparameters_o1o2.pklzparameters_o3b.pklcCsi|]\}}||qSr5r5)rArGrBr5r5r6 srINAME chirp_mass total_mass mass_ratiomass_1mass_2 final_mass final_spinspin_1spin_2chi_effchi_predshiftluminosity_distancenetworkmatchedfiltersnr)MCMQM1M2ZMFZCHIFZCHI1ZCHI2ZCHIEFFZCHIPZDZSNRmassratio_sourceZ_SRCz M/M_\odotz\mathcal{{M}}/M_\odotz(1+z)M/M_\odotz(1+z)\mathcal{{M}}/M_\odotz m_1/M_\odotz m_2/M_\odotz(1+z)m_1/M_\odotz(1+z)m_2/M_\odotz\chi_1z\chi_2z\chi_{{\rm eff}}z\chi_pz D_L/{\rm Gpc}zqzM_{\rm f}/M_\odotz(1+z)M_{\rm f}/M_\odotzM_{\rm f}^{\rm det}/M_\odotz \chi_{\rm f}z {\rm SNR})Ztotal_mass_sourceZchirp_mass_sourcerLrKZ mass_1_sourceZ mass_2_sourcerNrOrRrSrTrUrWrVrMfinal_mass_sourcerPrerQrXc@seZdZddZdS) ParametercCsb||_t|j|_|jdd|_d|jkrRd|jkrRd|jkrR|jd7_t|j|_dS)N_r`rasourcedet)uppername_pesummary_namesZpe_namereplace macro_keycolumn_name2tex_namelatex)selfrlr5r5r6__init__3s   zParameter.__init__N)__name__ __module__ __qualname__rsr5r5r5r6rf2srfc@s4eZdZddZddZeddZeddZd S) EventcsfddtD}d_t|rNddtt|Dd_tj_d_ntkrj_tj_nx t kr _tj_nT dd}|tkr|_tj_n,t krt _tj_n t d jtk_jdd jd }jd krBtd |_td |_td|_nd_d_d_jdkrtj_fddtD_nfddtD_jd_jd_dS)Ncs g|]}|ddkqS)GWA)strip)rArG)keyr5r6rC>sz"Event.__init__..ZO3cSsg|]\}}|r|qSr5r5)rArGmr5r5r6rCBsrZO1O2rxSzunrecognized event %rZO3acolors linestyles linewidthsrh-csi|]\}}|j|qSr5)getrl)rArGrB)rrr5r6rIgsz"Event.__init__..csi|]\}}|j|qSr5)rcid)rArGrB)rrr5r6rIisZOBSERVINGINSTRUMENTSrJ) o1o2_eventsrunanyziprrDZsidsuperevent_idsrFrk catalog_idsrn _NAME_DICT ValueError all_eventsZin_paperlower _STYLE_DICTcolorlslwr7rl_PE_DICTitemspeZifos)rrr{Z match_o1o2new_keyZalt_cidr5)r{rrr6rs=sF           zEvent.__init__cCs|jt|jS)N)rrfro)rrr{r5r5r6 get_parammszEvent.get_paramcCs8g}x.tD]"}t|j|jkr||qW|S)N)rmkeysrfrorappendr)rrrrGr5r5r6 parametersps zEvent.parameterscCsd|jkrd|jS|jSdS)NZGW19z \NAME{%s})rrl)rrr5r5r6 name_macroxs  zEvent.name_macroN)rtrurvrsrpropertyrrr5r5r5r6rw<s0 rwcCsJ|d|d}t|}x*|D]"}|||9}|t||}q W|S)a: Multiply a set of KDEs, evaluated over array. Arguments --------- all_kdes: list list of KDE functions to be combined. x: array array of parameter values over which to evaluate KDEs. Returns ------- joint_like: array combined KDE evaluated over `x` grid. rr)np ones_likesum)all_kdesxdx joint_likekder5r5r6multiply_likelihood_kdess    rc Ks|dkr.tdd|D}tdd|D}n|\}}t|||}g}x,|D]$} || f|} |t|| |qNWt||} | |fS)a Multiply PDFs starting from samples. Arguments --------- all_samples: list list of arrays of samples drawn from the PDFs to be combined. range: tuple, None range of x over which to combine PDF(x), defaults to the min and max over all sets of samples provided. nbins: int number of bins used to produce grid overwhich to multiply PDFs (def. 1000). kde: function function to prodce KDE from samples (def. scipy.stats.gaussian_kde). kwargs: additional arguments passed to KDE function. Returns ------- joint_like: array combined PDF evaluated over `x` grid. x: array parameter grid over which PDF was evaluated. NcSsg|] }t|qSr5)min)rAsr5r5r6rCsz(multiply_likelihoods..cSsg|] }t|qSr5)max)rArr5r5r6rCs)rrrlinspacerrr) Z all_samplesrangenbinsrkwargsxminxmaxrrsamplespdfrr5r5r6multiply_likelihoodss   r?Tc Csdd|}d|}t||dd}|r2||}|t|krTtdt|}d}nt||}||}|t|krtdt|} n|pt||}||} || fS)a Compute symmetric credible interval from PDF evaluated over grid. Arguments --------- y: array array of PDF values x: array array of parameter values corresponding to `y`. p: float credible level (def. 0.9). normalize: bool normalize PDF before computing CI (def. True). Returns ------- left: float parameter value corresponding to lower interval edge. right: float parameter value corresponding to upper interval edge. g?rg)initialzWARNING: value below range.NzWARNING: value above range.)rrrprintr) yrp normalizeZcdf_leftZ cdf_rightcdfleft cdf_interprightr5r5r6get_sym_interval_from_pdfs"      rcCsRt||dd}|r||}|t|kr.)rrrr i)sizec s(g|] }tt|tqSr5)introundr)rAff)rrr5r6rCSsz&kdeplot_2d_clevels..cg?ij)indexingaxlevelsN)rrr TypeErrorrrrrrchoiceargsortrmeshgridflattenreshaperrgcacontour)xsysrrfkde_kwsrGrclZDxZDyrrZXSYSZZSrr5)rrrr6kdeplot_2d_clevels.s(,,"r)rT)rT)r)r)rT)rF)rT)r)r)Wosnumpyr scipy.statsrscipy.interpolaterscipy.integraterpicklepkljsonpylabstatssswarningscatch_warnings simplefilterZ#pesummary.core.plots.bounded_1d_kderZBounded_1d_kde scale_factor fig_width_pt inches_per_ptsqrt fig_ratio fig_width fig_heightfigsize_columnfigsize_squarefig_width_page figsize_pagercParamsfsfs_labelr4r7r@sortedrrrdictrrErDrFrZ o3a_eventsrpathdirnamerealpath__file__ base_pathopenjoinrloadrrZ _PE_DICT_O1O2rrGrBupdateZ _PE_DICT_O3Brrmcopyrpobjectrfrwrrrrrrrrrrrrrr5r5r5r6s                              G ' ,     & * i