B zdF@srdZddlmZmZmZmZdddgZdZdddd ej fd dZ dd dZ dddddd dej fd dZ d dZ dS)a  `gwosc.locate` provides functions to determine the file URLs containing data for a specific dataset. You can search for remote data URLS based on the event name: >>> from gwosc.locate import get_event_urls >>> get_event_urls('GW150914') ['https://gwosc.org/eventapi/json/GWTC-1-confident/GW150914/v3/H-H1_GWOSC_4KHZ_R1-1126259447-32.hdf5', 'https://gwosc.org/eventapi/json/GWTC-1-confident/GW150914/v3/H-H1_GWOSC_4KHZ_R1-1126257415-4096.hdf5', 'https://gwosc.org/eventapi/json/GWTC-1-confident/GW150914/v3/L-L1_GWOSC_4KHZ_R1-1126259447-32.hdf5', 'https://gwosc.org/eventapi/json/GWTC-1-confident/GW150914/v3/L-L1_GWOSC_4KHZ_R1-1126257415-4096.hdf5'] You can down-select the URLs using keyword arguments: >>> get_event_urls('GW150914', detector='L1', duration=32) ['https://gwosc.org/eventapi/json/GWTC-1-confident/GW150914/v3/L-L1_GWOSC_4KHZ_R1-1126259447-32.hdf5'] You can search for remote data URLs based on the GPS time interval as follows: >>> from gwosc.locate import get_urls >>> get_urls('L1', 968650000, 968660000) ['https://gwosc.org/archive/data/S6/967835648/L-L1_LOSC_4_V1-968646656-4096.hdf5', 'https://gwosc.org/archive/data/S6/967835648/L-L1_LOSC_4_V1-968650752-4096.hdf5', 'https://gwosc.org/archive/data/S6/967835648/L-L1_LOSC_4_V1-968654848-4096.hdf5', 'https://gwosc.org/archive/data/S6/967835648/L-L1_LOSC_4_V1-968658944-4096.hdf5'] By default, this method will return the paths to HDF5 files for the 4 kHz sample-rate data, these can be specified as keyword arguments. For full information, see :func:`get_urls`. )apidatasetsurlsutilsget_urls get_run_urlsget_event_urlsz(Duncan Macleod Nhdf5c Cst|}t|}xdD]x}tj|||||f||d} xX| D]P} |dkr^t| ||||||d} nt| |||||||d} t| ||fr:| Sq:WqWtd|||fdS)aFetch the URLs from GWOSC regarding a given GPS interval Parameters ---------- detector : `str` the prefix of the relevant GW detector start : `int` the GPS start time of your query end : `int` the GPS end time of your query dataset : `str`, `None`, optional the name of the dataset to query, e.g. ``'GW150914'`` version : `int`, `None`, optional the data-release version for the selected datasets sample_rate : `int`, optional, default : ``4096`` the sampling rate (Hz) of files you want to find format : `str`, optional, default: ``'hdf5'`` the file format (extension) you want to find host : `str`, optional the URL of the remote GWOSC server Returns ------- urls : `list` of `str` the list of remote file URLs that contain data matching the relevant parameters )eventrun)matchtypedetectorsegmentversionhostr )r sample_rateformat)rstartendrrrrz4Cannot find a GWOSC dataset for %s covering [%d, %d)N)intrZ_iter_datasetsrrrZ full_coverage ValueError) rrrZdatasetrrrrZdstypeZdsetsdstrrY/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/gwosc/locate.pyr1sB*    c Ks>tj|||||dtjd}ttj|df||d|S)a\Fetch the URLs from GWOSC regarding a given event Parameters ---------- event : `str` the ID of the event detector : `str`, optional the detector for files you want to find start : `int` the GPS start time of your query end : `int` the GPS end time of your query format : `str`, optional, default: ``'hdf5'`` the file format (extension) you want to find sample_rate : `int`, optional, default : ``4096`` the sampling rate (Hz) of files you want to find host : `str`, optional the URL of the remote GWOSC server version : `int`, `None`, optional the data-release version for the selected datasets Returns ------- urls : `list` of `str` the list of remote file URLs that contain data matching the relevant parameters r)ZgpsstartZgpsendrstrain)r sampling_rate)rZfetch_run_jsonpop DEFAULT_URL_urlslurlssieve)r rrrrrr metarrrrs,c Ksttj|||d|ddd} | d} |dkrB|dkrBd} n,|dkrTtd n||dkrhtd n|f} ttj| f|||| d| S) a\Fetch the URLs from GWOSC regarding a given event Parameters ---------- event : `str` the ID of the event detector : `str`, optional the detector for files you want to find format : `str`, optional, default: ``'hdf5'`` the file format (extension) you want to find sample_rate : `int`, optional, default : ``4096`` the sampling rate (Hz) of files you want to find host : `str`, optional the URL of the remote GWOSC server start : `int` the GPS start time of your query end : `int` the GPS end time of your query version : `int`, `None`, optional the data-release version for the selected datasets Returns ------- urls : `list` of `str` the list of remote file URLs that contain data matching the relevant parameters T)catalogrfullreventsrNinf)rrrr)listrZ_fetch_allevents_event_jsonvaluesfloatr r!r") r r$rrrrrrrr r#rrrrrrs&/cCsdd|DS)NcSsg|] }|dqS)urlr).0urrr sz_urls..r)rrrrr sr )r r )__doc__rrrr!r__all__ __author__rrrrr rrrr$s& V 9C