B ~d' @sdZddlmZddlmZddlZddlmZddl m Z yddl m Z Wn ek rlddlm Z YnXdZeed d Zd d Zdd d ddd d dd d ZdddZdddZdS)z-Utilities for signal-processing with windows )wraps)ceilN) get_window)expit) _win_equivz(Duncan Macleod c Osytt||f||Sttfk r.YnXt|ttfrPt||f||St|}|j dkslt d|j |ks~t d|S)z\Wrapper around :func:`scipy.signal.get_window` to allow pre-computed window arrays. z+multi-dimensional windows are not supportedzwindow array wrong size) _get_windowfloat TypeError ValueError isinstancestrtuplenumpyZasarrayndimAssertionErrorsize)ZwindowZNxargskwargsr_/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/gwpy/signal/window.pyr%s rcCsD|dkrdSyt|jStk r>td|YnXdS)a)Find the canonical name for the given window in scipy.signal Parameters ---------- name : `str` the name of the window you want Returns ------- realname : `str` the name of the window as implemented in `scipy.signal.window` Raises ------- ValueError if ``name`` cannot be resolved to a window function in `scipy.signal` Examples -------- >>> from gwpy.signal.window import canonical_name >>> canonical_name('hann') 'hann' >>> canonical_name('ksr') 'kaiser' planckz3no window function in scipy.signal equivalent to %rN)lowerWINDOWS__name__KeyErrorr )namerrrcanonical_name9s rg?gx&?g?gˡE?) ZboxcarZbartlettZbarthannZblackmanharrisZflattopZhannZhammingZnuttallZtriangc Csy t|}Wn.tk r:}ztt|Wdd}~XYnXy t|}Wn tk rhtd|YnX|r~tt||S|S)aReturns the recommended fractional overlap for the given window If ``nfft`` is given, the return is in samples Parameters ---------- name : `str` the name of the window you are using nfft : `int`, optional the length of the window Returns ------- rov : `float`, `int` the recommended overlap (ROV) for the given window, in samples if ``nfft` is given (`int`), otherwise fractional (`float`) Examples -------- >>> from gwpy.signal.window import recommended_overlap >>> recommended_overlap('hann') 0.5 >>> recommended_overlap('blackmanharris', nfft=128) 85 Nz$no recommended overlap for %r window)rrr r ROVintr)rZnfftexcZrovrrrrecommended_overlapms  r!cst|}rV|dd9<tfddtdD}|dt| 9<r||dd9<tfddtdD}|||dt| 9<|S)aReturn a Planck taper window. Parameters ---------- N : `int` Number of samples in the output window nleft : `int`, optional Number of samples to taper on the left, should be less than `N/2` nright : `int`, optional Number of samples to taper on the right, should be less than `N/2` Returns ------- w : `ndarray` The window, with the maximum value normalized to 1 and at least one end tapered smoothly to 0. Examples -------- To taper 0.1 seconds on both ends of one second of data sampled at 2048 Hz: >>> from gwpy.signal.window import planck >>> w = planck(2048, nleft=205, nright=205) References ---------- .. [1] McKechan, D.J.A., Robinson, C., and Sathyaprakash, B.S. (April 2010). "A tapering window for time-domain templates and simulated signals in the detection of gravitational waves from coalescing compact binaries". Classical and Quantum Gravity 27 (8). :doi:`10.1088/0264-9381/27/8/084020` .. [2] Wikipedia, "Window function", https://en.wikipedia.org/wiki/Window_function#Planck-taper_window rcs$g|]}d|d|qS)g?r).0k)nleftrr szplanck..rcs&g|]} d|d|qS)g?r)r"r#)nrightrrr%s)rZonesarrayranger)Nr$r&wZzleftZzrightr)r$r&rrs' "r)N)rr)__doc__ functoolsrmathrrZ scipy.signalrrZ scipy.specialrZscipy.signal.windows._windowsrr ImportErrorZscipy.signal.windows.windows __author__rrr!rrrrrs.    ( ,