B }d@sLdZddlmZddlmZddlmZddlmZdZ Gd d d eZ d S) zCoherence plots ) OrderedDict)Plot)label_to_latex)Spectrumz&Joseph Areeda csheZdZdZdZdZfddZefddZfdd Z d d Z d d Z ddZ fddZ ZS) CoherencezYPlot coherence between a reference time series and one or more other time series coherencercsBtj|||jjp|jd|_d|jkr>|jdd|_dS)Nr,)super__init__argsrefZ chan_listref_chansplit)selfr kwargs) __class___/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/gwpy/cli/coherence.pyr 's zCoherence.__init__cst|}|jddd|S)Nz--refz7Reference channel against which others will be compared)help)r arg_channels add_argument)clsparsergroup)rrrr.s zCoherence.arg_channelscs>|jdkrd|_|jdkr2|js&d|_|js2d|_t|S)NZlinearrg?)ZyscaleZyminZymaxr _finalize_arguments)rr )rrrr5s  zCoherence._finalize_argumentscCsdS)zText for y-axis label rr)rrrr get_ylabel?szCoherence.get_ylabelcCs d|jS)zFStart of default super title, first channel is appended to it z Coherence: )r)rrrr get_suptitleDszCoherence.get_suptitlec Cs|j}t|j}|j}|dd||f|dk r:||9}|dd|jt}xX|jD]N}|j}y||||j j <WqZt k rt||<||||j j <YqZXqZWt |j |jd}|}g|_x|D]}|||j} x||D]t} ||| }|j| |||jd} | } t|jdkr8| d |jj7} |jrHt| } |j| | d |j| qWqW|jd kr|jsd||_|S) z9Generate the coherence plot from all time series rz-Calculating spectrum secpfft: %s, overlap: %sNzReference channel: )figsizedpi) fftlengthoverlapwindowrz, )labellog)r floatZsecpfftr#r&rrZ timeseriesspanZchannelnameKeyErrorrr r!ZgcaZspectrapopr r$len start_listepochZgpsZusetexrplotappendZxscaleZxmin) rr r"r#groupsZseriessegr/ZaxZreftsr)Zcohr%rrr make_plotIsF        zCoherence.make_plotcs t}|dk r|d|S)z)Create a legend for this product NzCoherence with:)r set_legend set_title)rleg)rrrr4}s  zCoherence.set_legend)__name__ __module__ __qualname____doc__actionZ MIN_DATASETSr classmethodrrrrr3r4 __classcell__rr)rrrs  4rN) r: collectionsrr/rZplot.texrZspectrumr __author__rrrrrs