B d@sdZddlZddlZddlZddlZddlZddlmZm Z ddl Z ddl m Z m Z mZddlmZddlmZmZmZddlmZddlmZdd lmZmZmZmZdd lmZdd lm Z dd l!m"Z"ddl#Z dd l$m%Z%ddl&mZ'm(Z(m)Z)e*dZ+e,edde,eddiZ-ddddZ.ddZ/d1ddZ0ddZ1Gddde2Z3Gd d!d!ed"Z4Gd#d$d$e4Z5Gd%d&d&e4Z6Gd'd(d(e4Z7e5j8e5e6j8e6e7j8e7iZ9d)d*Z:d+d,Z;Gd-d.d.e2ZThis module provides utilities for injecting signals into dataN)ABCMetaabstractmethod)waveformframelibutils)LimitedSizeDict)get_td_waveformfd_detget_td_det_waveform_from_fd_det)utils)ringdown_td_approximants)float64float32 TimeSeriesload_timeseries)Detector)tau0_from_mass1_mass2)resample_to_delta_t)LIGOLWContentHandler)r ligolw lsctablesZ lalsimulationcGs tj|S)N)simZSimAddInjectionREAL4TimeSeries)argsr`/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/inject/inject.py4rcGs tj|S)N)rZSimAddInjectionREAL8TimeSeries)rrrrr5r longitudelatituder)radec approximantcCsvy t|}Wntk r$|}YnX|dkrJt||_td|d|_|dkrft||dd|gn t|||dS)z$Sets data of a SimInspiral instance.tcgeA)Zspin1Zspin2rN)sim_inspiral_mapKeyErrorintZgeocent_end_timeZgeocent_end_time_nssetattr)injfielddataZ sim_fieldrrr set_sim_data@s   r+r#c Cst|}||}||}y|j}|j}|j}Wn|j}|j}|j}YnX|j|7_|j|7_y t ||j } t ||j } Wnt k r|} |} YnXd} t |dr|j } td|| |j| | |||j| |d} | S)zT Use the injection row to project the polarizations into the detector frame laldetector_projection_methodzInjecting at %s, method is %s)methodZreference_time)rr"rr time_geocentrr start_timewfutilstaper_timeseriestaperAttributeErrorhasattrr-logginginfoZ project_wave polarization) detector_namer(hphcdistance_scaledetectorr"rr Z hp_taperedZ hc_taperedZprojection_methodsignalrrr projectorQs8    r?cCsHt|}yt|}Wntdd}YnXtt|}||fS)zConvert the old style xml approximant name to a name and phase_order. Alex: I hate this function. Please delete this when we use Collin's new tables. z>Warning: Could not read phase order from string, using default)strrZGetOrderFromStringprintZGetStringFromApproximantZGetApproximantFromString)Zapxordernamerrrlegacy_approximant_name{s rEc@sBeZdZdZddZdddZddd Zd d Zedd d Z dS)_XMLInjectionSeta:Manages sets of injections: reads injections from LIGOLW XML files and injects them into time series. Parameters ---------- sim_file : string Path to a LIGOLW XML file containing a SimInspiralTable with injection definitions. Attributes ---------- indoc table cKs,tj|dtd|_tj|j|_||_dS)NF)contenthandler) ligolw_utils load_filenamerindocrSimInspiralTable get_tabletable extra_args)selfsim_filekwdsrrr__init__sz_XMLInjectionSet.__init__Nr#cs|jttfkr tdt|j|}tjtj} t|j | } t|j | } t |j} |j } |dk rpd|} |j }rfdd|D}g}x|D]}|dkr|jn|}|jd}t|j|j|}|jd|d}|| ks|| krq|j|| |||d}t||j d d }t|j | kr q||j}|}| ||d|||dk r|j}|||qW|jjdd|jdd<t|}t|_ |j |7_ |dk r||_|S) aAdd injections (as seen by a particular detector) to a time series. Parameters ---------- strain : TimeSeries Time series to inject signals into, of type float32 or float64. detector_name : string Name of the detector used for projecting injections. f_lower : {None, float}, optional Low-frequency cutoff for injected signals. If None, use value provided by each injection. distance_scale: {1, float}, optional Factor to scale the distance of an injection with. The default is no scaling. simulation_ids: iterable, optional If given, only inject signals with the given simulation IDs. inj_filter_rejector: InjFilterRejector instance; optional, default=None If given send each injected waveform to the InjFilterRejector instance so that it can store a reduced representation of that injection if necessary. injection_sample_rate: float, optional The sample rate to generate the signal before injection Returns ------- None Raises ------ TypeError For invalid types of `strain`. z-Strain dtype must be float32 or float64, not Ng?csg|]}|jkr|qSr) simulation_id).0r()simulation_idsrr sz*_XMLInjectionSet.apply..r#)f_lowerr<ldas)r.)dtyperr TypeErrorrAr, REARTH_SIC_SIfloatr0end_timeinjection_func_mapdelta_trMrXr/rmass1mass2make_strain_from_inj_objectrastypeappendrSgenerate_short_inj_from_injr*copyrrKinjection_params)rOstrainr9rXr<rUinj_filter_rejectorinjection_sample_rate lalstrainearth_travel_timet0t1 add_injectionra injectionsZinjection_parametersr(f_lr_ inj_lengthr0r> signal_lalZsidinjectedr)rUrapplysR$           z_XMLInjectionSet.applyc CsV|dkr|jn|}t|j\}}t|f|||||jd|j\} } t||| | |dS)a;Make a h(t) strain time-series from an injection object as read from a sim_inspiral table, for example. Parameters ----------- inj : injection object The injection object to turn into a strain h(t). delta_t : float Sample rate to make injection at. detector_name : string Name of the detector used for projecting injections. f_lower : {None, float}, optional Low-frequency cutoff for injected signals. If None, use value provided by each injection. distance_scale: {1, float}, optional Factor to scale the distance of an injection with. The default is no scaling. Returns -------- signal : float h(t) corresponding to the injection. N)r!ra phase_orderrXdistance)r<)rXrErrryrNr?) rOr(rar9rXr<rsrDrxr:r;rrrrds z,_XMLInjectionSet.make_strain_from_inj_objectcCsdd|jDS)z&Return the end times of all injectionscSsg|] }|jqSr)r/)rTr(rrrrV$sz._XMLInjectionSet.end_times..)rM)rOrrr end_times"sz_XMLInjectionSet.end_timesc Cst}|tttj}|jd||dkr>i}|dkrL|j}xt |j D]z}t }x|j D]}t ||dqlWx$|D]} ||| } t|| | qWx |D]\} } t|| | qW||qXWtj||dddS)aCWrites the injection samples to the given xml. Parameters ---------- filename : str The name of the file to write to. samples : io.FieldArray FieldArray of parameters. write_params : list, optional Only write the given parameter names. All given names must be keys in ``samples``. Default is to write all parameters in ``samples``. static_args : dict, optional Dictionary mapping static parameter names to values. These are written to the ``attrs``. rNauto)compress)rZDocumentZ appendChildZLIGO_LWrZNewrKZ childNodes fieldnamesrangesizeZ SimInspiral __slots__r'r+itemsrfrHZwrite_filename) filenamesamples write_params static_argsZxmldocZsimtableiircolr)r*valuerrrwrite&s&    z_XMLInjectionSet.write)Nr#NNN)Nr#)NN) __name__ __module__ __qualname____doc__rRrwrdrz staticmethodrrrrrrFs Y %rFc@sleZdZdZejjZdZdZ dddZ e dddZ e dd d Z e d d Ze d dZedddZdS)_HDFInjectionSetaFManages sets of injections: reads injections from hdf files and injects them into time series. Parameters ---------- sim_file : string Path to an hdf file containing injections. \**kwds : The rest of the keyword arguments are passed to the waveform generation function when generating injections. Attributes ---------- filehandler table static_args extra_args required_params : tuple Parameter names that must exist in the injection HDF file in order to create an injection of that type. Nrc st|d}|dkr|n||||_t}fdd|D}x.|D]&}||jjdkrJ||d||<qJWt|dkrd}nt | dj }yj dd|_ Wntk rg|_ YnX||j x|j D]z} j | } t| tjtt fr,tj|td } x(t|D]} | | | <qWn t| |} | jjdkrP| d} | || <qWt|jt|} | rtd d | |jjf||_||_ dS) Nrcsi|]}|d|qS)rr)rTparam)grouprr tsz-_HDFInjectionSet.__init__..SUrr#r)rZz>required parameter(s) {} not found in the given injection filez, )!h5pyFileZ filehandlerlistkeysrZkindrelentuplevaluesrattrsrr%extend isinstancenpZndarrayemptyobjectr~repeatcharsetrequired_params ValueErrorformatjoin _tableclassZ from_kwargsrMrN)rOrPZ hdf_grouprQfp parametersZinjvalskZnuminjrvalZarrrmissingr)rrrRls@           z_HDFInjectionSet.__init__r#cKsdS)z,Adds injections to a detector's time series.Nr)rOrjr9r<rUrkkwargsrrrrwsz_HDFInjectionSet.applycKsdS)zAMake a h(t) strain time-series from an injection object. Nr)rOr(rar9r<rrrrrdsz,_HDFInjectionSet.make_strain_from_inj_objectcCsdS)z&Return the end times of all injectionsNr)rOrrrrzsz_HDFInjectionSet.end_timescCsdS)z,Return a list of the supported approximants.Nr)rOrrrsupported_approximantssz'_HDFInjectionSet.supported_approximantsc Ks2t|d}|dkri}ttt||jd<|j|jd<x|D]\}}||j|<qJW|dkrn|j }xF|D]:\} }y||j| <Wqxt k rt||j| <YqxXqxWxl|D]d} y|| || <Wqt k r} z.|| j j dkr ||  d|| <n| Wdd} ~ XYqXqWWdQRXdS)aWrites the injection samples to the given hdf file. Parameters ---------- filename : str The name of the file to write to. samples : io.FieldArray FieldArray of parameters. write_params : list, optional Only write the given parameter names. All given names must be keys in ``samples``. Default is to write all parameters in ``samples``. static_args : dict, optional Dictionary mapping static parameter names to values. These are written to the ``attrs``. \**metadata : All other keyword arguments will be written to the file's attrs. wNrinjtyperr)rrrmaprArrrrr}r[rZrre) clsrrrrmetadatarkeyrargr)errrrs*  z_HDFInjectionSet.write)N)r#NN)r#)NN)rrrrpycbcioZ FieldArrayrrrrRrrwrdrzr classmethodrrrrrrQs 6    r) metaclassc@sHeZdZdZejjZdZdZ dddZ ddd Z d d Z e d d ZdS)CBCHDFInjectionSetzManages CBC injections. Zcbc)r"Nr#cCs|jttfkr tdt|j|}|jddtkrPt|j } t|j } n(tj tj } t|j | } t|j | } t |j} |j} |dk rd|} |j}|r|t|}g}xt|D]\}}|dkr|jn|}|jd}t|j|j|}|jd|d}|| ks|| krq|j|| |||d}t||jd d }t|j | krLq||j}|}| ||d|||dk r|||qW|jjdd|jdd<t|}|t|t|_|dk rt |d r|j!}|j"}t#||g}t#||g}||_!||_"|S) aAdd injections (as seen by a particular detector) to a time series. Parameters ---------- strain : TimeSeries Time series to inject signals into, of type float32 or float64. detector_name : string Name of the detector used for projecting injections. f_lower : {None, float}, optional Low-frequency cutoff for injected signals. If None, use value provided by each injection. distance_scale: {1, float}, optional Factor to scale the distance of an injection with. The default is no scaling. simulation_ids: iterable, optional If given, only inject signals with the given simulation IDs. inj_filter_rejector: InjFilterRejector instance; optional, default=None If given send each injected waveform to the InjFilterRejector instance so that it can store a reduced representation of that injection if necessary. injection_sample_rate: float, optional The sample rate to generate the signal before injection Returns ------- None Raises ------ TypeError For invalid types of `strain`. z-Strain dtype must be float32 or float64, not rr!Ng?rWr#)rXr<rY)r.rv)$rZrr r[rAr,rMr r^r0r_r\r]r`rar enumeraterXr"rrbrcrdrrerfrgr*rhrarrayr&r5riZ injection_idsZ concatenate)rOrjr9rXr<rUrkrlrmrorprnrqrarrZ injected_idsrr(rsr_rtr0r>rurvZprev_pZprev_idrrrrwsb$            zCBCHDFInjectionSet.applyc Csz|dkr|j}n|}|dtkrHt|f|||d|j|}||}n.t|f||d|j\}} t|||| |d}|S)aMake a h(t) strain time-series from an injection object. Parameters ----------- inj : injection object The injection object to turn into a strain h(t). Can be any object which has waveform parameters as attributes, such as an element in a ``WaveformArray``. delta_t : float Sample rate to make injection at. detector_name : string Name of the detector used for projecting injections. f_lower : {None, float}, optional Low-frequency cutoff for injected signals. If None, use value provided by each injection. distance_scale: {1, float}, optional Factor to scale the distance of an injection with. The default is no scaling. Returns -------- signal : float h(t) corresponding to the injection. Nr!)rarXZifos)rarX)r<)rXr r rNrr?) rOr(rar9rXr<rsrjr:r;rrrrdWs  z.CBCHDFInjectionSet.make_strain_from_inj_objectcCs|jjS)z&Return the end times of all injections)rMr")rOrrrrzszCBCHDFInjectionSet.end_timescCsTg}x4tjjtjjgD] }x|D]}|||q WqW|tjjtt|S)N)rZtd_wavZfd_wavrr rr)Z all_apprxsdrrrrrs  z)CBCHDFInjectionSet.supported_approximants)Nr#NNN)Nr#)rrrrrrZ WaveformArrayrrrrwrdrzrrrrrrrs c +rc@s@eZdZdZdZdZdddZddd Zd d Ze d d Z dS)RingdownHDFInjectionSetzeManages a ringdown injection: reads injection from hdf file and injects it into time series. Zringdown)r"r#NcCs|dk rtd|jttfkr0tdt|j|}t|j}|j} |dk rXd|} |j } |rn| t |} xvt | j D]h} | | } |j | | ||d} t| |jdd} | |j} | }|||d|jjdd|jdd<qzWdS)a7Add injection (as seen by a particular detector) to a time series. Parameters ---------- strain : TimeSeries Time series to inject signals into, of type float32 or float64. detector_name : string Name of the detector used for projecting injections. distance_scale: float, optional Factor to scale the distance of an injection with. The default (=1) is no scaling. simulation_ids: iterable, optional If given, only inject signals with the given simulation IDs. inj_filter_rejector: InjFilterRejector instance, optional Not implemented. If not ``None``, a ``NotImplementedError`` will be raised. injection_sample_rate: float, optional The sample rate to generate the signal before injection Returns ------- None Raises ------ NotImplementedError If an ``inj_filter_rejector`` is provided. TypeError For invalid types of `strain`. Nz6Ringdown injections do not support inj_filter_rejectorz-Strain dtype must be float32 or float64, not g?)r<rY)r.)NotImplementedErrorrZrr r[rAr,r`rarMrr~rrdrrer*)rOrjr9r<rUrkrlrmrqrarrrZ injectionr>rurrrrws.!    zRingdownHDFInjectionSet.applycCs4t|d|fd|i|j\}}t|||||dS)aMake a h(t) strain time-series from an injection object as read from an hdf file. Parameters ----------- inj : injection object The injection object to turn into a strain h(t). delta_t : float Sample rate to make injection at. detector_name : string Name of the detector used for projecting injections. distance_scale: float, optional Factor to scale the distance of an injection with. The default (=1) is no scaling. Returns -------- signal : float h(t) corresponding to the injection. r!ra)r<)r rNr?)rOr(rar9r<r:r;rrrrds z3RingdownHDFInjectionSet.make_strain_from_inj_objectcCs|jj}|dS)zReturn the approximate end times of all injections. Currently, this just assumes all ringdowns are 2 seconds long. rW)rMr")rOZtcsrrrrzsz!RingdownHDFInjectionSet.end_timescCsttjS)N)rrr rrrrrrsz.RingdownHDFInjectionSet.supported_approximants)r#NNN)r#) rrrrrrrwrdrzrrrrrrrs ?  rc@sheZdZdZdZdZdZdZdZddZ e dd Z d d Z d d Z e ddZdddZdddZdS)!IncoherentFromFileHDFInjectionSeta Manages injecting an arbitrary time series loaded from a file. The injections must have the following attributes set: * ``filename``: (str) the name of the file to load containing the time series. The file type and format can be a frame file or anything understood by :py:func:`pycbc.types.timeseries.load_timeseries`. If a frame file (ends in ``.gwf``) is specified, a ``channel`` attribute must also be set. * ``DETECTOR_gps_time``: (float) The GPS time at which the time series should be added to the ``DETECTOR`` data, where ``DETECTOR`` is the name of the instrument to inject into (e.g., ``h1_gps_time``). **The time series will only be injected into a detector if a GPS time is given for that detector.** Set to -inf, nan, or do not provide a GPS time for a particular detector if you do not want to inject into that detector. * ``ref_point``: (str or float) What to use as the reference time of the injected time series. The time series will be injected into the detector such that the ``ref_point`` in the time series occurs at the specifed ``DETECTOR_gps_time``. Options are: ``'start'``, ``'end'``, ``'center'``, ``'absmax'``, or a float giving the number of seconds into the time series. In addition, the following attributes may optionally be provided: * ``channel``: (str): If the filename points to a frame file, the channel to load in that file. Must be provided for frame files. * ``DETECTOR_phase_shift``: (float) Apply a phase shift to the time series before adding it to detector ``DETECTOR``. * ``DETECTOR_amp_scale``: (float) Scale the amplitude by the given amount before adding it to detector ``DETECTOR``. * ``slice_start``: (float) Slice the time series starting at ``ref_point + slice_start`` before injecting into the data. Measured in seconds. * ``slice_end``: (float) Slice the time series ending at ``ref_point + slice_end`` before injecting into the data. Measured in seconds. * ``left_taper_width``: (float) Taper the start of the time series (after slicing) using half a kaiser window over the given number of seconds. See `:py:func:waveform.utils.td_taper` for more details. * ``right_taper_width``: (float) Taper the end of the time series (after slicing) using half a kaiser window over the given number of seconds. See `:py:func:waveform.utils.td_taper` for more details. The signal will be resampled to the same sample rate as the data it is being injected into. In order to use with ``pycbc_create_injections``, set the ``approximant`` name to ``'incoherent_from_file'``. incoherent_from_file)r ref_point NcCs tddS)NzIIncoherentFromFile times cannot be determined without loading time series)r)rOrrrrzCsz+IncoherentFromFileHDFInjectionSet.end_timescCsdgS)NrrrrrrrGsz8IncoherentFromFileHDFInjectionSet.supported_approximantsc Cs|jdkrt|jd|_y |j|jStk r8YnX|jdry |j}Wn*tk rz}z tdWdd}~XYnXt |j|}n t |j}||j|j<|S)zLoads an injection time series. After the first time a time series is loaded it will be added to an internal buffer for faster in case another injection uses the same series. N) size_limitz.gwfz&Must provide a channel for frame files) _bufferr _buffersizerr%endswithchannelr4rrZ read_framer)rOr(r_errtsrrrloadtsKs      z(IncoherentFromFileHDFInjectionSet.loadtsc Cs$y |j}Wn2tk r<}ztd|jWdd}~XYnX|jdkrVt|jd|_y|j|j|f}Wnt k r|dkrd}nz|dkrt | |j }n`|dkrt | |j d}nB|d kr| |j }n(t |ttfrt| }ntd |||j|j|f<YnX||_dS) zhSets t=0 of the given time series based on what the given injection's ``ref_point`` is. z*Must provide a ref_point for {} injectionsN)rstartgendcenterg@Zabsmaxz"Unrecognized ref_point {} provided)rr4rrr _rtbufferrrrr%rraZ abs_arg_maxrr^r&_epoch)rOr(rrrZreftimerrr set_ref_timehs0   z.IncoherentFromFileHDFInjectionSet.set_ref_timecCsy |j}Wntk r$|j}YnXy |j}Wntk rJ|j}YnX|||}y |j}Wntk r~d}YnX|rtj ||j|j|dd}y |j }Wntk rd}YnX|rtj ||j||jdd}|S)aSlices and tapers a timeseries based on the injection settings. This assumes that ``set_ref_time`` has been applied to the timeseries first. A copy of the time series will be returned even if no slicing or tapering is done. rleft)Zsideright) Z slice_startr4r0Z slice_endr_Z time_slicerhZleft_taper_widthr1Ztd_taperZright_taper_width)r(rZtstarttendZtwidthrrrslice_and_tapers0        z1IncoherentFromFileHDFInjectionSet.slice_and_taperr#c Cs|dk rtd|dk r"d|}n|j}|j}x|D]}||} ||| y|d|} Wntk rtj } YnXt | rtj } | | j } | | j } | |j ko| |j k} | r4|j ||||| d} | j|jkrt| |jdd} |j| ddq4WdS) Nz@IncoherentFromFile injections do not support inj_filter_rejectorg?z {}_gps_time)r<rrY)r.F)rh)rrarMrrrlowerrrinfisnanr0r_rdrinject)rOrjr9r<rlrkrarrr(rZinjtimer0r_rrrrrws4           z'IncoherentFromFileHDFInjectionSet.applyc Cs|dkrt|j}||||||}|j|d|7_t||dd}y|d|}Wntk rd}YnX|r| }|t d|9}| }y|d|}Wntk rd}YnX||}||9}|S) Nz {}_gps_timerY)r.z{}_phase_shiftry?z {}_amp_scaleg?) rrrrrrrrrZto_frequencyseriesrexpZ to_timeseries) rOr(rar9r<rZ phase_shiftfsZ amp_scalerrrrds.     z=IncoherentFromFileHDFInjectionSet.make_strain_from_inj_object)r#NN)r#N)rrrrrrrrrrzrrrrrrwrdrrrrrs9  " (rc Cst|d0}y|jd}Wntk r6tj}YnXWdQRXyt|Stk ryt|}Wnt k r~YnXt|SXdS)aGets the HDFInjectionSet class to use with the given file. This looks for the ``injtype`` in the given file's top level ``attrs``. If that attribute isn't set, will default to :py:class:`CBCHDFInjectionSet`. Parameters ---------- sim_file : str Name of the file. The file must already exist. Returns ------- HDFInjectionSet : The type of HDFInjectionSet to use. rrN) rrrr%rr hdfinjtypesrAdecoder4)rPrZftyperrrget_hdf_injtypesrcCs@d}x tD]}||kr|}qW|dkr.c3s|]}|dkVqdS)rNr)rTc)injclsrr }sz%InjectionSet.write..z'injections must all be of the same typerzCould not find an approximant in the static args or samples to determine the injection type. Please specify an injtype instead.) rrrrrFrrr}runiqueallrr)rrrrrrrZapprxsr)rrrVs$       zInjectionSet.writecCsH|jdkrdSi}|jdk r&|j|d<|jdk r:|j|d<t|jf|S)z`Return an instance of InjectionSet configured as specified on the command line. NZf_refZf_final)Zinjection_fileZinjection_f_refZinjection_f_finalr)optZkwarrrfrom_clis     zInjectionSet.from_cli)NNN)rrrrrRrrrrrrrr4s  3rc@s"eZdZdZddZdddZdS) SGBurstInjectionSetaKManages sets of sine-Gaussian burst injections: reads injections from LIGOLW XML files and injects them into time series. Parameters ---------- sim_file : string Path to a LIGOLW XML file containing a SimBurstTable with injection definitions. Attributes ---------- indoc table cKs,tj|dtd|_tj|j|_||_dS)NF)rG) rHrIrrJrZ SimBurstTablerLrMrN)rOrPrQrrrrRszSGBurstInjectionSet.__init__Nr#c Cs|jttfkr tdt|j|}tjtj}t|j |}t|j |}t |j} x|j D]} | j } d} d} d}| d| }| |ksd||krqdtt| jt| jt| jt| t|t|j\}}t|jjdd|j|jd}t|jjdd|j|jd}|jt| 7_|jt| 7_t|j |krDqdt|| j}||j}| ||dqdW|jjdd|jdd<dS)aAdd injections (as seen by a particular detector) to a time series. Parameters ---------- strain : TimeSeries Time series to inject signals into, of type float32 or float64. detector_name : string Name of the detector used for projecting injections. f_lower : {None, float}, optional Low-frequency cutoff for injected signals. If None, use value provided by each injection. distance_scale: {1, foat}, optional Factor to scale the distance of an injection with. The default is no scaling. Returns ------- None Raises ------ TypeError For invalid types of `strain`. z-Strain dtype must be float32 or float64, not g$@grWN)raepoch)rZrr r[rAr,r\r]r^r0r_r`rMr/rZSimBurstSineGaussianqZ frequencyZhrssrarr*ZdeltaTrrr1r2r3re)rOrjr9rXr<rmrnrorprqr(r_rtZ eccentricityr8r0r:r;rurrrrws:    zSGBurstInjectionSet.apply)Nr#)rrrrrRrwrrrrrsr)r#)>rrnumpyrr,rhr6abcrrrrrrrZ pycbc.optrZpycbc.waveformrr r r r1r Z pycbc.typesr rrrZpycbc.detectorrZpycbc.conversionsrZ pycbc.filterrZpycbc.ioZpycbc.io.ligolwrZligo.lwrHrrZimport_optionalrrZr`r$r+r?rErrFrrrrrrrrrrrrrrsZ         *H'rq"g