B d@s^ddlmZddlmZddlmZdZddlmZddlm Z dd Z d d Z dddZ d S))tqdm) iterutils)segmentsz"Kipp Cannon )date)versioncCs(tt|d|dt|d|dS)z; Return the smallest segment that contains both a and b. rr)rsegmentminmax)abr a/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/lalburst/snglcluster.pysmallest_enclosing_seg0src CsTt|d|t|d|d|||d|t|d|d||S)z Return the segment whose start and ends are the weighted arithmetic means of the start and ends of the two input segments, using the two weights given. The arithmetic means are computed in a way that is safe for LIGOTimeGPS objects. rr)rrfloat)Zseg1Zweight1Zseg2Zweight2r r rweighted_average_seg7s rNFc Cs6d}x*tdt|t|| d}|dk r:|j|dd}d} x| t|kr|| dkrp| d7} |qDd} xtt|| dd| dD]V\} } | dk r||| | s||| | || <d|| <d} q|dk r||| | rPqW| rd}qD| d7} |qDWWdQRX|sPtd d |d}qW|S) aO Cluster the events in an event list. testfunc will be passed a pair of events in random order, and must return 0 (or False) if they should be clustered. clusterfunc will be passed a pair of events in random order, and must return an event that is the "cluster" of the two. clusterfunc is free to return a new events, or modify one or the other of its parameters in place and return it. If sortkeyfunc and bailoutfunc are both not None (if one is provided the other must be as well), the events will be sorted into "increasing" order using sortkeyfunc as a sort key operator, and then only pairs of events for which bailoutfunc returns 0 (or False) will be considered for clustering. The return value is True if the events in the event list were modified, and False if they were not (although their order might have changed). Fzclustering %d events)desctotaldisableN)keyrrTcSs|dk S)Nr )eventr r rz cluster_events..)rlensortupdate enumeraterZinplace_filter) eventsZtestfuncZ clusterfuncZ sortkeyfuncZ bailoutfuncverbosechangedprogressZouter_did_clusteriZinner_did_clusterjZevent_jr r rcluster_eventsLs<  $r#)NNF)rZlalrZligor __author__Z git_versionr__date__r __version__rrr#r r r rs