U f{jt<@sxdZddlmZddlmZddlZddlmZddlm Z dd d Z ddddddd gfddZ dddZ dddZ dS) z0Functions to plot on circle as for connectivity.)cycle)partialN)plt_show)_validate_typeZT c spt|}t|krtd|dk rtj|tjd}t||ksPt|dkrXtdt|dkrtt|dkrtdt|}nd}d}fdd |Dttt|krtd d |||} |r(|| d 7}|dk r(|ddkr(||d 7}|dkr$|ddnd}tj|tj d| } || d<|dk r^| ||7<t | } | S) aCreate layout arranging nodes on a circle. Parameters ---------- node_names : list of str Node names. node_order : list of str List with node names defining the order in which the nodes are arranged. Must have the elements as node_names but the order can be different. The nodes are arranged clockwise starting at "start_pos" degrees. start_pos : float Angle in degrees that defines where the first node is plotted. start_between : bool If True, the layout starts with the position between the nodes. This is the same as adding "180. / len(node_names)" to start_pos. group_boundaries : None | array-like List of of boundaries between groups at which point a "group_sep" will be inserted. E.g. "[0, len(node_names) / 2]" will create two groups. group_sep : float Group separation angle in degrees. See "group_boundaries". Returns ------- node_angles : array, shape=(n_node_names,) Node angles in degrees. z2node_order has to be the same length as node_namesNZdtyperz7"group_boundaries" has to be between 0 and n_nodes - 1.rz3"group_boundaries" must have non-decreasing values.csg|]}|qS)index).0name node_orderr M/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/viz/circle.py Csz#circular_layout..znode_order has repeated entriesgv@r) len ValueErrornparrayint64anydiffuniqueonesZfloat64Zcumsum) node_namesr start_posZ start_betweengroup_boundariesZ group_sepn_nodesZ boundariesZ n_group_sepZnode_sep node_anglesr rrcircular_layouts8       r! c Cs|j|krdS|jdkr|d|jkr6|dks}?}@|@d!krd"}An |@d7}@d#}A|j<|?d$||>||@d%|Ad| d& q|dk r|j=|| |d'|r|jj>| |?| |d(}B|B@t | |tA}C|dk r4|CjB|d)|dk rJ|CjB|d*| jC|Bfd+|i|C}D|D|DjEjFd,}E|DjEjG|d-|jH|E| d.|rtItJ| |||d/}F| jKLd0|FtM|| |fS)1Nr) PolarAxesz3node_angles has to be the same length as node_namesrF)ZendpointgeAg?csg|]}|tqSr )float)r i)rspectralr rrsz-_plot_connectivity_circle..rz+indices has to be provided if con.ndim == 1z#con has to be 1D or a square matrix)r;r;)Zfigsize facecolorT)polarr rEgcsg|] }|qSr r r ind) con_draw_idxr rrscsg|] }|qSr r rF)sort_idxr rrsr g?)fill edgecolor linewidthalphar"z.9center)widthbottomrLZlwrDalignileftrightg"@anchor)r,ZrotationZ rotation_modeZhorizontalalignmentZverticalalignmentcolor)rVZfontsize)ZcmapZnorm)shrink)rUr/Z yticklabels)Z labelsize)rV)r.r/r0rr Zbutton_press_event)NZmatplotlib.pyplotZpyplotZmatplotlib.pathpathZmatplotlib.patchesr&Zmatplotlib.projections.polarr>rrrrr$ZlinspaceZ diag_indicesminr%rcmrBAttributeErrorZSpectralr+ndimshapeZ tril_indices isinstancestrZget_cmapZfigureZ add_subplotZ set_facecolorZ set_xticksZ set_yticksZset_ylimZspinesr)sortZravelwhereZargsortmaxzerosrr(randomZmtrandZ RandomStateuniformZ zeros_liker'r@PathZMOVETOZCURVE4ZLINETOZ PathPatchZ add_patchrbartext set_titleZScalarMappableZ NormalizeZ set_arraydictupdatecolorbarZgetpr/ZaxesZ tick_paramsZsetprr5r*Z mpl_connectr)Gconrr0Zn_linesr Z node_widthZ node_height node_colorsrDZ textcolorZnode_edgecolorrMcolormapvminvmaxrltitleZ colorbar_sizeZ colorbar_posZfontsize_titleZfontsize_namesZfontsize_colorbarpaddingr/ interactiveZnode_linewidthshowZpltZm_pathZ m_patchesr>Zdist_matr.Z con_threshZcon_absZvrangeZ nodes_n_conrAjrngZn_conZ noise_maxZ start_noiseZ end_noiseZnodes_n_con_seenstartendZcon_val_scaledpost0Zr0t1r1ZvertscodesrXrVpatchheightZbarsrgZ angles_degrZ angle_radZ angle_degZhasmZcolorbar_kwargscbZ cb_ytickscallbackr )rHrrIrBr_plot_connectivity_circlezs*                              "   "              rc s6ddlm}t|tdttdfdtttdfddk rXfdd|D}t|}ttd d | D}t |}dk r|D]}|krt d |d qt}t |}fd d |D} dgt || } |j d| t | d} nd} ||} t t | t | ft j} t|D]J\}}||D]6}| |}|||}|| ||f<| ||f<q>q.||ddd}t| |ddt |gd}d|kr|j|dd|kr|jddd|kr|j| dd|kr|jddd|kr|jddd |kr(|j| d!t| | f|S)"aPlot labels for each channel in a circle plot. .. note:: This primarily makes sense for sEEG channels where each channel can be assigned an anatomical label as the electrode passes through various brain areas. Parameters ---------- labels : dict Lists of labels (values) associated with each channel (keys). colors : dict The color (value) for each label (key). picks : list | tuple The channels to consider. **kwargs : kwargs Keyword arguments for :func:`mne_connectivity.viz.plot_connectivity_circle`. Returns ------- fig : instance of matplotlib.figure.Figure The figure handle. axes : instance of matplotlib.projections.polar.PolarAxes The subplot handle. r)LinearSegmentedColormaplabelsNcolorspickscsi|]\}}|kr||qSr r )r kv)rr r zsz.plot_channel_labels_circle..cSsg|]}|D]}|q qSr r )r vallabelr r rr|sz.plot_channel_labels_circle..zNo color provided for z in `colors`csg|] }|qSr r )r r)rr rrsr6 label_cmap)NrCr)rrr )r rlF)rlrn)rnrp)rprqr)rqro)ro)Zmatplotlib.colorsrrrjlisttupleitemskeyssetvaluesrr from_listrrcnanr'r r!rkr)rrrkwargsrZch_namesZ all_labelsZn_labelsrZ label_colorsrnrrrmidxZch_nameZnode_idxZ label_colorrr r )rrrplot_channel_labels_circleZs^                    r)rTNr )NNNNNNr6r7r6r8r9NNTNNNr:r;r;r<NTr=T)NN)__doc__ itertoolsr functoolsrnumpyrutilsrrr!r5rrr r r rsR     I  a