B dO@syddlmZWnek r,edZYnXddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl mZddl mZddl mZdd lmZddlZdd lmZddlmZd d lmZd d lmZdZd dlmZ d dlm!Z"ddZ#Gdddej$Z%ej&j'(e%j)e%e%e%j)iGdddej*Z*Gdddej+Z,d&ddZ-d'ddZ.d(ddZ/d d!Z0d"d#Z1d$d%Z2dS)))NegInfz-infN)ligolw)array)param) lsctables)utils)process)rate) offsetvector) snglcoincz"Kipp Cannon )date)versioncsnt}i}xXtdt|dD]B}x$\Delta t$ pairs. The return value is a dictionary of (instrument tuple)->TOATrangulator mappings. The instrument names in each tuple are sorted in alphabetical order, and the triangulators are constructed with the instruments in that order (the the documentation for snglcoinc.TOATriangulator for more information). Example: >>> x = triangulators({"H1": 0.005, "L1": 0.005, "V1": 0.005}) constructs a dictionary of triangulators for every combination of two or more instruments that can be constructed from those three. The program lalapps_string_plot_binj can be used to measure the timing uncertainties for the instruments in a search. r cSsg|]}tj|jqS)lalZcached_detector_by_prefixlocation).0 instrumentrra/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/lalburst/stringutils.py `sz!triangulators..csg|] }|qSrr)rr)timing_uncertaintiesrrr`s)sortedkeysrangelen itertools combinationsr ZTOATriangulator)rZallinstruments triangulatorsn instrumentsr)rrrFs  0rcs8eZdZdZdZfddZddZddZd ZZ S) InstrumentBinszy Example: >>> x = InstrumentBins() >>> x[frozenset(("H1", "L1"))] 55 >>> x.centres()[55] frozenset(['H1', 'L1']) ) ZE0ZE1ZE2ZE3ZG1H1H2zH1H2+zH1H2-ZL1ZV1cs.tt|fddttdDdS)Nc3s(|] }t|D]}t|VqqdS)N)rr frozenset)rrZcombo)namesrr sz*InstrumentBins.__init__..r )superr!__init__rr)selfr%) __class__)r%rr(szInstrumentBins.__init__cCstjdt|ddS)Nrdouble)dtype)numpyaranger)r)rrrlowerszInstrumentBins.lowercCstjdt|dddS)Nr r+)r,)r-r.r)r)rrrupperszInstrumentBins.upperZinstrumentbins) __name__ __module__ __qualname____doc__r%r(r/r0 xml_bins_name __classcell__rr)r*rr!vs   r!csbeZdZddZddZddZddd Zd d Zd d ZddZ fddZ e ddZ Z S) LnLRDensityc Csi|_x>|D]6}tttdddtdddf|jd|<q Wxtt|dD]}dtj |}ttt | | df|jd |<ttt d d df|jd |<ttt d ddf|jd|<qXWttt |t dddf|jd<dS)N gcAi!g?g@z %s_snr2_chi2rg{Gzt?z%s_%s_dtgg?z%s_%s_dAgɿg?iz%s_%s_dfg{Gzg{Gz?iz#instrumentgroup,rss_timing_residual) densitiesr BinnedLnPDFNDBinsZATanLogarithmicBinsrrrr Zlight_travel_timeZATanBinsr!)r)r rpairdtrrrr(s 6*&*zLnLRDensity.__init__c sPy |jWn"tk r,||jYnX|rLtfdd|DStS)Nc3s|]\}}||VqdS)Nr)rrvalue)interpsrrr&sz'LnLRDensity.__call__..)r?AttributeError mkinterpssumitemsr)r)paramsr)r?r__call__s   zLnLRDensity.__call__cCst|t|ks$t|jt|jkrrrr incrementszLnLRDensity.incrementcCs6t|g}x$|jD]\}}||j|<qW|S)N)rFr9rCcopy)r)newrJrKrrrrQs zLnLRDensity.copycCstdd|jD|_dS)Ncss&|]\}}d|kr||fVqdS)rss_timing_residualN)Zmkinterp)rrJrKrrrr&sz(LnLRDensity.mkinterps..)dictr9rCr?)r)rrrrAszLnLRDensity.mkinterpsc Csx|jD]~\}}|dr:t|jtjddddnH|dsX|dsX|drrt|jtjdddn|dr~nt| q W| dS) N _snr2_chi2 )sigmaZ_dtZ_dAZ_dfZinstrumentgroup) r9rCendswithr Z filter_arrayrZgaussian_window startswith Exception normalizerA)r)rJrKrrrfinishs   zLnLRDensity.finishcsltt||}tdd|jD}|tjdt j |x&|j D]\}}|||qLW|S)Ncss(|] }|dr|dddVqdS)rU_r rN)rYsplit)rrJrrrr&sz%LnLRDensity.to_xml..r ) r'r7to_xmlrGr9 appendChild ligolw_paramParamZ from_pyvaluerinstrumentspropertyrC)r)namexmlr rJrK)r*rrr`s zLnLRDensity.to_xmlcCsL|||}|tjt|d}x"|jD]}tj |||j|<q,W|S)Nr ) get_xml_rootrrdgetrbZ get_pyvaluer9r r:from_xml)clsrfrer)rJrrrris   zLnLRDensity.from_xml)rM)r1r2r3r(rErLrPrQrAr]r` classmethodrir6rr)r*rr7s  r7cseZdZdZejejejGdddej Z ddZ ddZ fdd Z d d Z d d ZddZddZeddZeddZddZZS)StringCoincParamsDistributionsZ#stringcusp_coincparamsdistributionsc@s eZdZdS)z3StringCoincParamsDistributions.LIGOLWContentHandlerN)r1r2r3rrrrLIGOLWContentHandlersrmcCs4tt|d|_t||_t||_t||_dS)Ngh㈵?)rrTfromkeysr7 numerator denominator candidates)r)r rrrr(s  z'StringCoincParamsDistributions.__init__cCspt|t|krt|t|jt|jkr.)r"r#)rGrr'rlrE)r)kwargsr )r*rrrE s z'StringCoincParamsDistributions.__call__cCs<t|g}|j|_|j|_|j|_|j|_|S)N)rFrrorQrprq)r)rRrrrrQs     z#StringCoincParamsDistributions.copycsi}t|dkr|Stt|ddd}tdd|D}|j|tfdd|D\}}}}d}x,|D]$}|jd |j|jf|d |j<qlWxt |dD]\}} |j| jkst d |j| jf} t |j |j| j | j} | f|d | <t t|j| j} | f|d | <|j|jd} | j| jd}t t | t |t | t |}|f|d| <qW|S)NrcSs|jS)N)ifo)eventrrr1z=StringCoincParamsDistributions.coinc_params..)rJcss|] }|jVqdS)N)rt)rrurrrr&2sz>StringCoincParamsDistributions.coinc_params..c3s|]}|j|jVqdS)N)peakrt)rru)r rrr&8sgg@z %s_snr2_chi2z%s_%s_z%sdtz%sdAz%sdf)rtuplerrZsnrZchisqZ chisq_dofrtrrAssertionErrorfloatrxmathlog10absZ amplitudeZ central_freq bandwidth)r)eventsr rDr ZignoredrSruZevent1Zevent2prefixr=ZdAZf_cut1Zf_cut2dfr)r r coinc_paramss*  (  $$,z+StringCoincParamsDistributions.coinc_paramscCs|f|||S)N)r)r)rr rrrln_lr_from_triggersisz2StringCoincParamsDistributions.ln_lr_from_triggerscCs"|j|j|j|S)N)ror]rprq)r)rrrr]ls   z%StringCoincParamsDistributions.finishcsRd|jffdd|tjjD}t|dkrJtdtjjf|dS)Nz%s:%scs$g|]}|dr|jkr|qS)Name)Z hasAttributer)relem)rerrrusz?StringCoincParamsDistributions.get_xml_root..r z5XML tree must contain exactly one %s element named %sr)ligo_lw_name_suffixZgetElementsByTagNamerLIGO_LWZtagNamerrr)rjrfrer)rerrgrs  z+StringCoincParamsDistributions.get_xml_rootcCsj|||}|g}t|d|_t|d|_t|d|_dd|jjD}tt |d|_|S)NrorprqcSs g|]}d|kr|ddqS)rU)replace)rrJrrrrsz;StringCoincParamsDistributions.from_xml..gh㈵?) rgr7rirorprqr9rrrTrn)rjrfrer)r rrrrizs z'StringCoincParamsDistributions.from_xmlcCsRtdd||jfi}||jd||jd||jd|S)Nrz%s:%srorprq)rrrraror`rprq)r)rerfrrrr`s z%StringCoincParamsDistributions.to_xml)r1r2r3r ligolw_arrayZuse_inrbrrrmr(rLrErQrrr]rkrgrir`r6rr)r*rrls J  rlFc Csd}d}xt|dD]\}}|roffset dictionaries, each vector of offsets in the sequence is applied to the segmentlists and the total time during which at least min_instruments were on is summed and returned. If clip is not None, after each offset vector is applied to seglists the result is intersected with clip before computing the livetime. If verbose is True then progress reports are printed to stderr. gz+computing the live time for %d time slides:)rz %.1f%% gY@r)rrNz 100.0%) rQrrrrroffsetsupdater{r~ segmentsUtilsZvotevalues) r time_slidesZmin_instrumentsrcliplivetimeNr time_sliderrrtime_slides_livetimes  $rc Csd}|j|d}|jt|t|d}t|}|rPtdd||ftjdxt|D]\} } |rtdd| |dtjd |j | |j | |d kr|t t | | || 7}qZ|t t ||@ | || 7}qZW|rtd tjd|S) z like time_slides_livetime() except computes the time for which exactly the instruments given by the sequence instruments were on (and nothing else). g)rz1computing the live time for %s in %d time slides:z, )rz %.1f%% gY@r)rrNz 100.0%)rQrGrrjoinrrrrrr{r~ intersectionrunion) rrr rrrZ onseglistsZ offseglistsrrrrrr)time_slides_livetime_for_instrument_combos"   *0rcCs,|}|d|d|f|dS)a Create a temporary table named "recovered_ln_likelihood_ratio" containing two columns: "simulation_id", the simulation_id of an injection, and "ln_likelihood_ratio", the highest log likelihood ratio at which that injection was recovered by a coincidence of type coinc_def_id. zr CREATE TEMPORARY TABLE recovered_ln_likelihood_ratio (simulation_id TEXT PRIMARY KEY, ln_likelihood_ratio REAL) a INSERT OR REPLACE INTO recovered_ln_likelihood_ratio SELECT sim_burst.simulation_id AS simulation_id, MAX(coinc_event.likelihood) AS ln_likelihood_ratio FROM sim_burst JOIN coinc_event_map AS a ON ( a.table_name == "sim_burst" AND a.event_id == sim_burst.simulation_id ) JOIN coinc_event_map AS b ON ( b.coinc_event_id == a.coinc_event_id ) JOIN coinc_event ON ( b.table_name == "coinc_event" AND b.event_id == coinc_event.coinc_event_id ) WHERE coinc_event.coinc_def_id == ? GROUP BY sim_burst.simulation_id N)cursorexecuteclose) connection coinc_def_idrrrr*create_recovered_ln_likelihood_ratio_tables  rcCs|d|fdS)z@ Construct a sim_burst --> best matching coinc_event mapping. a CREATE TEMPORARY TABLE sim_burst_best_string_sngl_map AS SELECT sim_burst.simulation_id AS simulation_id, ( SELECT sngl_burst.event_id FROM coinc_event_map AS a JOIN coinc_event_map AS b ON ( b.coinc_event_id == a.coinc_event_id ) JOIN coinc_event ON ( coinc_event.coinc_event_id == a.coinc_event_id ) JOIN sngl_burst ON ( b.table_name == 'sngl_burst' AND b.event_id == sngl_burst.event_id ) WHERE a.table_name == 'sim_burst' AND a.event_id == sim_burst.simulation_id AND coinc_event.coinc_def_id == ? ORDER BY (sngl_burst.chisq / sngl_burst.chisq_dof) / (sngl_burst.snr * sngl_burst.snr) LIMIT 1 ) AS event_id FROM sim_burst WHERE event_id IS NOT NULL N)rr)rrrrr%create_sim_burst_best_string_sngl_maps!rcCs|d|fdS)zg Construct a sim_burst --> best matching coinc_event mapping for string cusp injections and coincs. a CREATE TEMPORARY TABLE sim_burst_best_string_coinc_map AS SELECT sim_burst.simulation_id AS simulation_id, ( SELECT coinc_inspiral.coinc_event_id FROM coinc_event_map AS a JOIN coinc_event_map AS b ON ( b.coinc_event_id == a.coinc_event_id ) JOIN coinc_inspiral ON ( b.table_name == 'coinc_event' AND b.event_id == coinc_inspiral.coinc_event_id ) WHERE a.table_name == 'sim_burst' AND a.event_id == sim_burst.simulation_id AND coinc_event.coinc_def_id == ? ORDER BY (sngl_burst.chisq / sngl_burst.chisq_dof) / (sngl_burst.snr * sngl_burst.snr) LIMIT 1 ) AS coinc_event_id FROM sim_burst WHERE coinc_event_id IS NOT NULL N)rr)rrrrr&create_sim_burst_best_string_coinc_mapCsr)F)FN)FN)3Zfpconstr ImportErrorr{rr|r-Z scipy.statsZscipyrZligo.lwrrrrrbrrrZ ligo.lw.utilsrZligolw_processrr Z ligo.segmentsrr rr __author__Z git_versionr __date__r __version__rZ HashableBinsr!r;Zxml_bins_name_mappingrr5r7ZLnLikelihoodRatioMixinrlrrrrrrrrrrsH            0 T#   %'(