B d7M@sddlmZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z d Zdd lmZdd lmZGd d d ejZejddddZddZGddde jZGddde jZdddeddfddZGdddeZdS))print_functionN)ligolw) lsctables) offsetvector) snglcoinc)_thincaz"Kipp Cannon )date)versionc@sdeZdZdZddZddZddZdd Zd d Zd d Z e ddZ e ddZ e ddZ dS) SnglInspiralcCs |j|kS)N)end)selfotherr r _/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/lalinspiral/thinca.py__lt__FszSnglInspiral.__lt__cCs |j|kS)N)r )rrr r r__le__IszSnglInspiral.__le__cCs |j|kS)N)r )rrr r r__eq__LszSnglInspiral.__eq__cCs |j|kS)N)r )rrr r r__ne__OszSnglInspiral.__ne__cCs |j|kS)N)r )rrr r r__ge__RszSnglInspiral.__ge__cCs |j|kS)N)r )rrr r r__gt__UszSnglInspiral.__gt__cCs |j|jS)N)mass1mass2)rr r rmtotal_szSnglInspiral.mtotalcCs|j|j|jdS)Ng@)rrr)rr r retacszSnglInspiral.etacCs|j|jdS)Ng333333?)rr)rr r rmchirpgszSnglInspiral.mchirpN)__name__ __module__ __qualname__ __slots__rrrrrrpropertyrrrr r r rr =s  r Zinspiralz+sngl_inspiral<-->sngl_inspiral coincidences)searchsearch_coinc_type descriptioncCs t|ddd}|j||jS)a Compute the end time of an inspiral coincidence. events is an iterable of sngl_inspiral triggers, offset_vector is a dictionary mapping instrument to offset. This function returns the time shifted end time of the trigger with the highest SNR. The end time reported by this function gets used for things like plot titles, alert messages, and so on. It is not meant to be an accurate estimate of the time at which the gravitational wave passed through the geocentre, or any other such thing. This end time is also used to parallelize thinca by allowing a single lock stretch to be split across several jobs without missing or double counting any coincs. This is achieved by using a definition that is guaranteed to return a bit-identical "end time" for a given set of triggers. Guaranteeing that allows thinca to clip coincs to a sequence of contiguous segments and know that no coinc will missed or double counted. cSs|jS)N)snr)eventr r rz)coinc_inspiral_end_time..)key)maxr ifo)events offset_vectorr%r r rcoinc_inspiral_end_timesr-cs4eZdZfddZfddZfddZZS)InspiralCoincTablescs^tt|||ytj||_Wn4tk rXttj|_|j d |jYnXdS)Nr) superr.__init__rCoincInspiralTable get_tablecoinc_inspiral_table ValueErrorNewZ childNodes appendChild)rxmldoccoinc_definer_row) __class__r rr0s zInspiralCoincTables.__init__c stt||||d\}}|j|t|ddd}|j|jtdd|D}|jj |j |j |j |j ttdd|Dddd|d } tfd d|D|B|_||| fS) NZ sngl_inspiralcSs|jS)N)r$)r%r r rr&r'z0InspiralCoincTables.coinc_rows..)r(css|] }|jVqdS)N)r*).0r%r r r sz1InspiralCoincTables.coinc_rows..css|]}|jdVqdS)g@N)r$)r:r%r r rr;s) coinc_event_idZmassrr$Zfalse_alarm_rateZ combined_farZminimum_durationr Z instrumentsc3s*|]"\}}t||kr|VqdS)N)float)r: instrumentZsegs)r rr rr;s)r/r. coinc_rowstime_slide_indexr)r r* frozensetr3ZRowTyper<rrrmathsqrtsumsetitemsZinsts) r process_id time_slide_idr+seglistsZcoincZ coincmapsZrefeventZparticipating_instrumentscoinc_inspiral)r9)r rrr?s"   "zInspiralCoincTables.coinc_rowscs*tt|||}|j|_|j||S)N)r/r. append_coincr<r3append)r coinc_eventZcoinc_event_mapsrJ)r9r rrKs z InspiralCoincTables.append_coinc)rrrr0r?rK __classcell__r r )r9rr.s  7r.c@s&eZdZGdddejjZejZdS)coincgen_doublesc@seZdZedZdS)zcoincgen_doubles.singlesqueuer N)rrroperator attrgetterZ event_timer r r r singlesqueuesrRN)rrrrrOrRrZ get_coincsr r r rrOsrOTFc  s`| rtdtjdt||} tjt| j||| d} tj |} dk r`fdd| D} |}|sxt t | dddd dD]\} }| | t|tqWnxt t | d ddd dD]`\} }t|}| | |td d|Drx2| j|d D]"\}}| j| j||j||dqWqWx6| j|ddD]$\}}| j| j||j||dq4W|S)Nz indexing ...)file)min_instrumentsverbosec3s,|]$}|jks |j|jkr|VqdS)N)r*r )r:r%) veto_segmentsr rr;(sz ligolw_thinca..cSs|jS)N)r*)rowr r rr&4r'zligolw_thinca..)r(cSs|jS)N)r*)r%r r rr&4r'cSs |j|jfS)N)r r*)rXr r rr&;r'cSs|jS)N)r*)r%r r rr&;r'css|] }|jVqdS)N)r )r:r%r r rr;=s) coinc_sieve)rIT)rYflush)printsysstderrr.rZTimeSlideGraphrOr@rSnglInspiralTabler2 itertoolsgroupbysortedpushtupleZ PosInfinityr)pullrKr?rH)r7rGZdelta_trIZntuple_comparefuncrWrUr8 incrementalrVZ coinc_tablesZtime_slide_graphsngl_inspiral_tabler>r+noder )rWr ligolw_thincas&  &&" rhc@sheZdZdZddZeddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZeZdS)sngl_inspiral_coincsa5 Dictionary-like device to extract XML document trees containing individual sngl_inspiral coincs from a source XML document tree containing several. An instance of the class is initialized with an XML document tree. The coinc event ID of a sngl_inspiral<-->sngl_inspiral coinc in the document can then be used like a dictionary key to retrieve a newly-constructed XML document containing that coinc by itself. The output document trees are complete, self-describing, documents with all metadata about the event from the source document preserved. Example: >>> coincs = sngl_inspiral_coincs(xmldoc) >>> print(coincs.coinc_def_id) 0 >>> coincs.keys() [0] >>> coinc_id = coincs.keys()[0] >>> print(coinc_id) 83763 >>> coincs[coinc_id].write() ... The XML documents returned from this class share references to the row objects in the original document. Modifications to the row objects in the tables returned by this class will affect both the original document and all other documents returned by this class. However, each retrieval constructs a new document from scratch, they are not cached nor re-used, therefore this operation can be time consuming if it needs to be performed repeatedly but the table objects and document trees can be edited without affecting each other. If the source document is modified after this class has been instantiated, the behaviour is undefined. To assist with memory clean-up, it is helpful to invoke the .unlink() method on the XML trees returned by this class when they are no longer needed. c stj|_tj|_tj|_tj|_ tj |_ tj |_ tj|_tj|_tddjD_i_x$jD]}j|jg|qWtddjD_i_x$jD]}j|jg|qWtddjD_ddj D\_tddjDtfddj D_ tfddj D_!tj tj!kst"td dj D_#xBjD]8}yj#|j$|Wnt%k rwYnXqWd S) z Initialize an instance of the class. xmldoc is the source XML document tree from which the sngl_inspiral<-->sngl_inspiral coincs will be extracted. css|]}|j|fVqdS)N)rG)r:rXr r rr;sz0sngl_inspiral_coincs.__init__..css|]}|j|fVqdS)N)event_id)r:rXr r rr;scss"|]\}}t|s|VqdS)N)anyvalues)r:rHr,r r rr;scss*|]"}|jtjkr|jtjkr|VqdS)N)r!InspiralCoincDefr")r:rXr r rr;scss|] }|jVqdS)N)r<)r:rXr r rr;sc3s0|](}|jjjkr|jkr|j|fVqdS)N) coinc_def_id coinc_defr<)r:rX)coinc_event_map_idsrr rr;sc3s"|]}|jkr|j|fVqdS)N)r<)r:rX)rpr rr;scss|]}|gfVqdS)Nr )r:r<r r rr;sN)&r ProcessTabler2 process_tableProcessParamsTableprocess_params_tabler^rf CoincDefTablecoinc_def_table CoincTablecoinc_event_tabler1r3 CoincMapTablecoinc_event_map_tableTimeSlideTabletime_slide_tabledict process_indexprocess_params_index setdefaultrGrLsngl_inspiral_indexr@rHrAas_dictrFzero_lag_time_slide_idsrocoinc_event_indexcoinc_inspiral_indexAssertionErrorcoinc_event_map_indexr<KeyError)rr7rXr )rprrr0s8    zsngl_inspiral_coincs.__init__cCs|jjS)za The coinc_def_id of the sngl_inspiral<-->sngl_inspiral coincs in the source XML document. )rorn)rr r rrnsz!sngl_inspiral_coincs.coinc_def_idcsfddj|DS)zm Return a list of the sngl_inspiral rows that participated in the coincidence given by coinc_event_id. csg|]}j|jqSr )rrj)r:rX)rr r sz7sngl_inspiral_coincs.sngl_inspirals..)r)rr<r )rrsngl_inspiralssz#sngl_inspiral_coincs.sngl_inspiralsccsLxF|jD]8\}}|j|jkr |jdkr |j|\}|j|jVq WdS)a Generator returns a sequence of the sngl_inspiral table rows that formed zero-lag single-instrument "coincs". This is only meaningful if the coincidence engine was run with min_instruments = 1, otherwise this sequence will be empty by construction. Also, if there was no zero-lag time slide included in the time slide graph then this sequence will be empty. This method is used by codes that want lists of non-coincident triggers for background models even if min_instruments has been set below 2. The constraint that they be "zero-lag" singles might at first seem nonsensical but is included to exclude triggers that form genuine coincidences at zero-lag but are present only as single-detector candidates in one or more time slides. rSN)rrFrHrZneventsrrrj)rr<rMrXr r rsingle_sngl_inspiralss z*sngl_inspiral_coincs.single_sngl_inspiralscCstdd|j|DS)z5 Return the offsetvector given by time_slide_id. css|]}|j|jfVqdS)N)r>offset)r:rXr r rr;sz5sngl_inspiral_coincs.offset_vector..)rr@)rrHr r rr,sz"sngl_inspiral_coincs.offset_vectorc Cst}|t}|ttj|jj}|ttj |j j}|ttj |j j}|ttj |jj}|ttj|jj}|ttj|jj} |ttj|jj} |ttj|jj} ||j|j|} || | |j|t| j|j|t| j|j| jx | D]} ||j| j q2Wxrt!|"dt!|"dBt!| "dBD]D}||j#|yt|j|j$|Wnt%k rYnXqzW|S)z Construct and return an XML document containing the sngl_inspiral<-->sngl_inspiral coinc carrying the given coinc_event_id. rG)&rZDocumentr6ZLIGO_LWrr5rqrrZ columnnamesrsrtr^rfrurvrwrxr1r3ryrzr{r|rLrorrmaprr@rHrrjrEZgetColumnByNamer~rr)rr<Z newxmldocZ ligolw_elemZnew_process_tableZnew_process_params_tableZnew_sngl_inspiral_tableZnew_coinc_def_tableZnew_coinc_event_tableZnew_coinc_inspiral_tableZnew_coinc_event_map_tableZnew_time_slide_tablerMrXrGr r r __getitem__s2    0 z sngl_inspiral_coincs.__getitem__cCs t|jS)z? Iterate over the coinc_event_id's in the source document. )iterr)rr r r__iter__szsngl_inspiral_coincs.__iter__cCs t|jS)N)boolr)rr r r __nonzero__sz sngl_inspiral_coincs.__nonzero__cCs |jS)z{ A list of the coinc_event_id's of the sngl_inspiral<-->sngl_inspiral coincs available in the source XML document. )rkeys)rr r rr"szsngl_inspiral_coincs.keysccs x|D]}|||fVqWdS)aG Yield a sequence of (coinc_event_id, XML tree) tuples, one for each sngl_inspiral<-->sngl_inspiral coinc in the source document. NOTE: to allow this to work more easily with very large documents, instead of returning the complete sequence as a pre-constructed list this method is implemented as a generator. Nr )rr<r r rrF*s zsngl_inspiral_coincs.itemsN)rrr__doc__r0r rnrrr,rrrrrF iteritemsr r r rri]s12 )ri) __future__rr_rBrPr\Zligo.lwrrZlalburstrrr __author__Z git_versionr __date__r __version__r ZCoincDefrmr-Z CoincTablesr.rOrhobjectrir r r rs0        =V L