B d{@sdZddlZddlZddlmZddlmZddlmZ ddlm Z ddlm Z ddlZddlmZd d lmZd d lmZd Zd d lmZd dlmZGdddejZeej_e_GdddejZeej_e_ejdd ddZ ejddddZ!ejddddZ"ejddddZ#ejddddZ$ejddddZ%ejd d ddZ&ejd dddZ'ejd dddZ(ejd dddZ)ejd dddZ*ejd dddZ+ejd!dd"dZ,ejd!d ddZ-ejd!dddZ.ejd!dddZ/ejd!dddZ0ejd!dddZ1ejd!dddZ2ejd#dd"dZ3ejd#d ddZ4ejd#dddZ5ejd#dddZ6ejd#dddZ7ejd#dddZ8ejd#dddZ9Gd$d%d%e:Z;d&Zd+d,Z?dGd.d/Z@dHd0d1ZAd2d3ZBd4d5ZCd6d7ZDd8d9ZEd:d;ZFdd?ZHd@dAZIdBdCZJdIdEdFZKdS)Ja/ Burst injection identification library. Contains code providing the capacity to search a list of sngl_burst burst candidates for events matching entries in a sim_burst list of software injections, recording the matches as burst <--> injection coincidences using the standard coincidence infrastructure. Also, any pre-recorded burst <--> burst coincidences are checked for cases where all the burst events in a coincidence match an injection, and these are recorded as coinc <--> injection coincidences, again using the standard coincidence infrastructure. N) lsctables)coincs)process)search_summary) time_slide)segments)burca) SimBurstUtilsz"Kipp Cannon )date)versionc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) SnglBurstcCs |j|kS)N)end)selfotherrr^/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/lalburst/binjfind.py__lt__SszSnglBurst.__lt__cCs |j|kS)N)r)rrrrr__le__VszSnglBurst.__le__cCs |j|kS)N)r)rrrrr__eq__YszSnglBurst.__eq__cCs |j|kS)N)r)rrrrr__ne__\szSnglBurst.__ne__cCs |j|kS)N)r)rrrrr__ge___szSnglBurst.__ge__cCs |j|kS)N)r)rrrrr__gt__bszSnglBurst.__gt__N) __name__ __module__ __qualname__ __slots__rrrrrrrrrrr Jsr c@s eZdZdS) SimInspiralN)rrrrrrrrqsr excesspowerz$sim_burst<-->sngl_burst coincidences)searchsearch_coinc_type descriptionz'sim_inspiral<-->sngl_burst coincidencesz-sim_burst<-->coinc_event coincidences (exact)z.sim_burst<-->coinc_event coincidences (nearby)z0sim_inspiral<-->coinc_event coincidences (exact)z1sim_inspiral<-->coinc_event coincidences (nearby) StringCuspomegaz%sngl_burst<-->sngl_burst coincidences waveburstc@s0eZdZdZddZddZddZdd Zd S) DocContentsz, A wrapper interface to the XML document. c  s(| _tj|_ytj|_Wntk r@d_YnXytj|_ Wntk rnd_ YnXytj |} Wntk rd} YnX| dk r| _ ni_ jrt jd_nd_t|| _j dk r(tj|ijd| ddd} xj D] } | | _qWjdk rRtj||j|jd|jd_nd_j dk rtj||j|jd|jd_nd_ytj||j|jddWn2tk rdd_ d_!d_"d_#YnXjdk rtj||j|jd|jd_ tj||j|jd|jd_!n d_ d_!j dk rntj||j|jd|jd_"tj||j|jd|jd_#n d_"d_#ytj$|_%Wn6tk rt&tj$_%|j'd(j%YnXj%)ytj*|_+Wn6tk rt&tj*_+|j'd(j+YnXt,d d jD}t,fd d j%D_-xDj+D]:}yj-|j./||j0Wntk rYnXqNW~x6j-1D](\}}|j2d d dt3|j-|<qWj-1_-t,fdd j%D_4t,fdd t5jdD_j-j2dd ddS)NdurationgTF) create_newZ superset_okZ nonunique_ok)r,r!)r,rcss|]}|j|fVqdS)N)event_id).0rowrrr (sz'DocContents.__init__..c3s"|]}|jkr|jgfVqdS)N) coinc_def_idcoinc_event_id)r.r/)b_b_coinc_def_idrrr0*scSs|jS)N)peak)eventrrr5z&DocContents.__init__..)keyc3s*|]"}|jkr|jj|jfVqdS)N)r1r2 offsetvectors time_slide_id)r.r/)r3rrrr0;sc3s2|]*tfddjDdddfVqdS)c3s|]}|jkr|VqdS)N)ifo)r.r5) instrumentrrr0Fsz1DocContents.__init__...cSs|jS)N)r4)r5rrrr6Fr7z0DocContents.__init__...)r8N)sortedsnglbursttable)r.)r)r<rr0Fsr;cSs|ddjS)Nrr)r4)Zcoinc_id_eventsrrrr6Gr7)6rrSnglBurstTableZ get_tabler> SimBurstTable simbursttable ValueErrorSimInspiralTablesiminspiraltableZTimeSlideTableas_dictr9maxZgetColumnByNamelongestdurationligolw_search_summaryZ!segmentlistdict_fromsearchsummaryZcoalesceseglistsligolw_time_slideZget_time_slide_idfromkeysr: ligolw_coincsZget_coinc_def_idrr r!sb_b_coinc_def_idsi_b_coinc_def_idKeyErrorsb_c_e_coinc_def_idsb_c_n_coinc_def_idsi_c_e_coinc_def_idsi_c_n_coinc_def_idZ CoincTable coinctableZNewZ childNodesZ appendChildZ sync_next_idZ CoincMapTable coincmaptabledictrr2appendr-itemssorttuple coincoffsetsset)rxmldocb_b_defsb_b_defsi_b_def sb_c_e_def sb_c_n_def si_c_e_def si_c_n_defrlivetime_programZtimeslidetabler:simindexr/r2eventsr)r3rr__init__s                  $zDocContents.__init__cs"tfdd|jDgS)z Return a list of the burst events whose peak times (with offsetvector added) are within window seconds of t. This is not used to define any coincidences, only to provide a short list of burst events for use in more costly comparison tests. c 3sF|]>\}}|t||t||VqdS)N)bisect bisect_left bisect_right)r.r<rh) offsetvectortwindowrrr0Tsz3DocContents.bursts_near_peaktime..)sumr>rX)rrnrormr)rmrnrorbursts_near_peaktimeIs z DocContents.bursts_near_peaktimecs*t||fddjDS)a Return a list of the (coinc_event_id, event list) tuples in which at least one burst event's peak time is within window seconds of t. This is not used to define any coincidences, only to provide a short list of coinc events for use in more costly comparison tests. cs4g|],\}}t|@rj|r||fqSr)r\containsr[)r.r2rh) near_eventsrmrrr jsz4DocContents.coincs_near_peaktime..)r\rqr)rrnrormr)rsrmrrcoincs_near_peaktimeVs z DocContents.coincs_near_peaktimecCsR|j}|jj|_||_|j|_||_|dd|_ d|_ |j ||S)z Construct a new coinc_event row attached to the given process, and belonging to the set of coincidences defined by the given coinc_def_id. Nr) rTRowTyperZ process_idr1Z get_next_idr2r:set_instrumentsneventsZ likelihoodrW)rr1r:coincrrr new_coincls     zDocContents.new_coincN)rrr__doc__rirqrurzrrrrr*s $ r*Zlalapps_binjfindc Cstj|td|itdt|dS)zD Convenience wrapper for adding process metadata to the document. match_algorithmZlscsoft)programZ paramdictr Zcvs_repositoryZcvs_entry_timecomment)ligolw_processZregister_to_xmldocprocess_program_name __version____date__)r]r|r~rrrappend_processsrcCs4||j|}tjd}t|||||jS)z Return False (injection matches event) if an autocorrelation-width window centred on the injection is continuous with the time interval of the burst. r#)time_at_instrumentr;r stringcusp_autocorrelation_widthrsegmentdisjointperiod)rfburstrmtinjrorrrStringCuspSnglCompares rcCs ||j||jkp|j|jkS)z Return False (injection matches event) if the peak time and centre frequency of sim lie within the time-frequency tile of burst. )rr;rZ frequencyZband)rfrrmrrrExcessPowerSnglComparesr$@cCs tt||j||j|kS)z Return False (injection matches event) if the time of the sim and the peak time of the burst event differ by less than or equal to delta_t seconds. )absfloatrr;r4)rfrrmdelta_trrrOmegaSnglComparesrcCs tt||j||j|kS)z Return False (injection matches event) if the time of the sim and the peak time of the burst event differ by less than or equal to delta_t seconds. )rrrr;r4)rfrrmrrrrCWBSnglComparesrcCs:||j|}tjdtj}t|||||jS)zb Return False (injection matches coinc) if the peak time of the sim is "near" the burst event. r#) rr;r rburst_is_near_injection_windowrrrr)rfrrmrrorrrStringCuspNearCoincComparesrcCs0||j|}tj}t|||||jS)zb Return False (injection matches coinc) if the peak time of the sim is "near" the burst event. )rr;r rrrrr)rfrrmrrorrrExcessPowerNearCoincComparesrcCst|||d|jddS)zb Return False (injection matches coinc) if the peak time of the sim is "near" the burst event. g4@r#)r)rr+)rfrrmrrrOmegaNearCoincComparesrcCst|||d|jddS)zb Return False (injection matches coinc) if the peak time of the sim is "near" the burst event. g4@r#)r)rr+)rfrrmrrrCWBNearCoincComparesrcs.|jjfdd|j|DS)a Scan the burst table for triggers matching sim. sieve_window is used in a bisection search to quickly identify burst events within that many seconds of the injection's peak time at the geocentre; it should be larger than the greatest time difference that can separate a burst event's peak time from an injection's peak time at the geocentre and the two still be considered a match. csg|]}|s|qSrr)r.r) comparefuncrmrfrrrtsz+find_sngl_burst_matches..)r9r:rq time_geocent)contentsrfrZ sieve_windowr)rrmrfrfind_sngl_burst_matchess rcCs|||j}|tdd|Dt||_|j}|j|_|j j |_ |j |_ |j |x<|D]4}|j}|j|_|j j |_ |j |_ |j |qfW|S)z Create a coinc_event in the coinc table, and add arcs in the coinc_event_map table linking the sim_burst row and the list of sngl_burst rows to the new coinc_event row. css|] }|jVqdS)N)r;)r.r5rrrr0 sz&add_sim_burst_coinc..) rzr:rwr\lenrxrUrvr2 simulation_id table_namer-rW)rrfrhr1rycoincmapr5rrradd_sim_burst_coincs       rcs*t|tfdd|DS)z Return a set of the coinc_event_ids of the burst<-->burst coincs in which all burst events match sim and to which all instruments on at the time of the sim contributed events. c3sH|]@\}}tdd|Drtfdd|Ds|VqdS)css|] }|jVqdS)N)r;)r.r5rrrr05sz5find_exact_coinc_matches...c3s|]}|VqdS)Nr)r.r5)rrmrfrrr05sN)issubsetr\any)r.r2rh)rrmon_instrumentsrfrrr05sz+find_exact_coinc_matches..)r rr\)rrfrrIrmr)rrmrrfrfind_exact_coinc_matches&srcstfdd|DS)zt Return a set of the coinc_event_ids of the burst<-->burst coincs in which at least one burst event matches sim. c3s0|](\}}tfdd|Ds|VqdS)c3s|]}|VqdS)Nr)r.r5)rrmrfrrr0Esz4find_near_coinc_matches...N)all)r.r2rh)rrmrfrrr0Esz*find_near_coinc_matches..)r\)rrfrrmr)rrmrfrfind_near_coinc_matches8s rcCs|||j}t||_|j}|j|_|jj|_|j|_ |j |x8|D]0}|j}|j|_|j|_||_ |j |qNW|S)z Create a coinc_event in the coinc table, and add arcs in the coinc_event_map table linking the sim_burst row and the list of coinc_event rows to the new coinc_event row. ) rzr:rrxrUrvr2rrr-rW)rrfZcoinc_event_idsr1ryrr2rrradd_sim_coinc_coincHs      rFcCsV|rtdtjdtjtjttd|}tt t t d|}t t ttd|}ttttd|} ttttd|} ttttd|} ttttd|} t ||||| | | | |ddddd|d } t!j"t!j#d}|| j$7}|t%j&d d d d d|7}|t%j'}| j(dk rt)| j(}|r.td |j*tjdx^t+| j(D]P\}}|rdtd d||dtjdt,| |||}|r:t-| ||| j.q:W|rtdtjd|rtd| j*| j*ftjdxt+| j(D]\}}|rtd d||dtjd| j/|j0}| 1|j2||}t3|||| j4|}t5||||}|6|sDt7|rZt8| ||| j9|rt8| ||| j:qW|rtdtjdn|rtdt;jdk r6t)| j>}|rtd |j*tjdx^t+| j>D]P\}}|rtd d||dtjdt,| |||}|rt-| ||| j?qW|rLtdtjd|rltd| j*| j*ftjdxt+| j>D]\}}|rtd d||dtjd| j/|j0}| 1|j2||}t3|||| j4|}t5||||}|6|st7|rt8| ||| j@|rxt8| ||| jAqxW|rRtdtjdn|rRtdt;jBj=tjd|S)Nz indexing ...)file)r'rr)r(Z StringSearchZ lalapps_power)r'rr(r)) r]r^r_r`rarbrcrdrreg?r#gg>@zconstructing %s:z %.1f%% gY@ )rrz 100.0%zconstructing %s and %s:z!no %s table in document, skipping)Cprintsysstderrr ZStringCuspBBCoincDefZExcessPowerBBCoincDef CWBBBCoincDefOmegaBBCoincDefStringCuspSBBCoincDefExcessPowerSBBCoincDefCWBSBBCoincDefOmegaSBBCoincDefStringCuspSIBCoincDefExcessPowerSIBCoincDefCWBSIBCoincDefOmegaSIBCoincDefStringCuspSBCCoincDefExcessPowerSBCCoincDefCWBSBCCoincDefOmegaSBCCoincDefStringCuspSBCNearCoincDefExcessPowerSBCNearCoincDefCWBSBCNearCoincDefOmegaSBCNearCoincDefStringCuspSICCoincDefExcessPowerSICCoincDefCWBSICCoincDefOmegaSICCoincDefStringCuspSICNearCoincDefExcessPowerSICNearCoincDefCWBSICNearCoincDefOmegaSICNearCoincDefr*lalZ REARTH_SIZC_SIrGr rrrArr! enumeraterrrMr9r:rurrrIrrAssertionErrorrrPrQrr@Z tableNamerDrNrRrSrC)r]rrZsnglcomparefuncZnearcoinccomparefuncverboser^r_r`rarbrcrdrZburst_peak_time_windowZcoinc_peak_time_windowNnrfrhrmrZexact_coinc_event_idsZnear_coinc_event_idsrrrbinjfindks         r)r)r)F)Lr{rjrZligo.lwrZ ligo.lw.utilsrrLrrrrHrrJrZligorr r __author__Z git_versionr rr rr r?rvrrCZCoincDefrrrrrrrrrrrrrrrrrrrrrrrrrrobjectr*rrrrrrrrrrrrrrrrrrrr$s|            f      $#