B
â‹dŸ ã @ sž d Z ddlZddlZddlZddlZdejkr8e d¡ ddlZddlZddl Zddl
m Z dd„ Z dd d
d œZ
d d
„ Zdd„ Zdd„ Zddd„Zdd„ ZdS )zT This module provides functions to generate followup plots and trigger
time series.
é Nzmatplotlib.backendsZagg)Úsegmentc C s¶ | |¡} | |t||ƒ¡} i }x| D ]ˆ}t |jd¡}|d dd… }t ||k ||k¡} t | ¡d }
xB|D ]:} | |kr„g || < t || || dd… |
g¡|| < qpW q&W |S )aN Return columns of information stored in single detector trigger
files.
Parameters
----------
file_list_file : string
pickle file containing the list of single detector
triggers.
ifo : string
The ifo to return triggers for.
columns : list of strings
The list of columns to read from the trigger files.
start : int
The start time to get triggers from
end : int
The end time to get triggers from
Returns
-------
trigger_dict : dict
A dictionary of column vectors with column names as keys.
ÚrÚend_timeNr )
Zfind_output_with_ifoZfind_all_output_in_ranger Úh5pyÚFileZ storage_pathÚnumpyÚ logical_andÚwhereZ concatenate) Ú file_listÚcolumnsÚifoÚstartÚendZ trig_dictZ trig_fileÚfÚtimeÚpickZpick_locÚcol© r úc/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/results/followup.pyÚcolumns_from_file_list' s
.r ÚblueÚredÚgreen)ZH1ZL1ZV1c C s¾ t ¡ }t | d¡}|d }|d }|d }|d }|jd } |jd }
t j||| t| d t j|||
t|
d d } tj |tjj
| d
¡ t ¡ t d ¡ t
d ¡ t ¡ t |¡S )
Nr zforeground/stat1zforeground/stat2zforeground/time1zforeground/time2Z
detector_1Z
detector_2)ÚlabelÚcolorz.12g)ÚfmtzTime (s)ZNewSNR)ÚpylabÚfigurer r ÚattrsÚscatterÚ ifo_colorÚmpld3ÚpluginsÚconnectÚ
MousePositionÚlegendÚxlabelÚylabelÚgridÚ fig_to_html) Z
coinc_filer
r Úfigr Zstat1Zstat2Ztime1Ztime2Zifo1Zifo2r r r r Úcoinc_timeseries_plotR s"
r+ c C sš t ¡ }xb|D ]Z}t| ddg|||ƒ}t|ƒ t j|d |d |t| d d}tj |tjj |d¡ qW t
¡ t d¡ t d¡ t
¡ t |¡S )NZsnrr )r r z.12g)r zTime (s)ZSNR)r r r Úprintr r r! r" r# r$ r% r& r' r( r) )r
Zifosr
r r* r Ztrigsr r r r Útrigger_timeseries_ploth s
r- c C s d}t | |||dS )Nz/../followup/%s/%s/%s)Úbase)Útimes_to_links)ÚtimesÚwindowÚtagr. r r r Ú
times_to_urls{ s r3 c C sF |d kr d}g }x0| D ](}|| }|| }| ||||f ¡ qW |S )Nz<followup)Úappend)r0 r1 r2 r. Úurlsr r
r r r r r/ s
r/ c C s: dt | ¡d t | ¡d f }d |¡}d| d }|S )Nz %.0f+..+%.0fé z6https://gracedb.ligo.org/search/?query={}&query_type=Sz Search)r ÚfloorÚceilÚformat)r Zgdb_search_queryZgdb_search_urlZgdb_search_linkr r r Úget_gracedb_search_linkŠ s r: )N)Ú__doc__r r Z
matplotlibÚsysÚmodulesZuser r! Z
mpld3.pluginsZ
ligo.segmentsr r r r+ r- r3 r/ r: r r r r Ú s
)