U f{jZ @sddlmZddlmZmZddlZddlZddlm Z ddl Z ddl Z ddlmZddlmZddlmZmZmZmZmZddlmZmZmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1d d Z2ddZ3dZddZ4e.GdddZ5ddZ6Gddde7Z8e*d[ddddZ9e*d\ddZ:d]dd Z;d^d!d"Ze*d`d)d*Z?d+d,Z@d-d.ZAe.dad0d1ZBd2d3ZCe.dbd4d5ZDdcd6d7ZEddd8d9ZFd:d;ZGd= 2 Number of colors on the gradient. Returns ------- colors : tuple of tuples, len = n N RGBA tuples that occupy a gradient in value (low to high) but share saturation and hue with the input color. g$@r#gffffff?g?c3s|]}|fVqdSNr0).0v_)r(r)r0r1 csz _split_colors..css |]\}}}t|||VqdSr4)r)r5Zh_Zs_r6r0r0r1r7dsc3s |]\}}}|||fVqdSr4r0)r5r_Zg_Zb_)r.r0r1r7es)rnpsqrtr&r'linspacetuple) r/nr+r,r-r*Zgradient_rangeZv_maxZv_minZ hsv_colorsZ rgb_colorsZ rgba_colorsr0)r.r(r)r1 _split_colorsIs r>Fhsvc Csd}||krtd|ddlm}|||}tdd|d}|||d}|rt|D]6\}} tt|d|| kdrVtd ||fqV|S) aProduce a list of n unique RGBA color tuples based on a colormap. Parameters ---------- n : int Number of colors. bytes : bool Return colors as integers values between 0 and 255 (instead of floats between 0 and 1). cmap : str Which colormap to use. Returns ------- colors : array, shape (n, 4) RGBA color values. iz(Can't produce more than %i unique colorsr)get_cmaprF)bytesNzPCould not get %d unique colors from %s colormap. Try using a different colormap.) NotImplementedErrorZ matplotlib.cmr@r9r; enumerateanyall RuntimeError) r=bytes_cmapZn_maxr@cmposcolorsiicr0r0r1 _n_colorsis    rNc @seZdZdZed3ddddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ d4ddZd5ddZed6ddZed7d d!Zed8d#d$Zd9d%d&Zd:d'd(Zed;d*d+ZedLabelaA FreeSurfer/MNE label with vertices restricted to one hemisphere. Labels can be combined with the ``+`` operator: * Duplicate vertices are removed. * If duplicate vertices have conflicting position values, an error is raised. * Values of duplicate vertices are summed. Parameters ---------- vertices : array, shape (N,) Vertex indices (0 based). pos : array, shape (N, 3) | None Locations in meters. If None, then zeros are used. values : array, shape (N,) | None Values at the vertices. If None, then ones are used. hemi : 'lh' | 'rh' Hemisphere to which the label applies. comment : str Kept as information but not used by the object itself. name : str Kept as information but not used by the object itself. filename : str Kept as information but not used by the object itself. %(subject_label)s color : None | matplotlib color Default label color and alpha (e.g., ``(1., 0., 0., 1.)`` for red). %(verbose)s Attributes ---------- color : None | tuple Default label color, represented as RGBA tuple with values between 0 and 1. comment : str Comment from the first line of the label file. hemi : 'lh' | 'rh' Hemisphere. name : None | str A name for the label. It is OK to change that attribute manually. pos : array, shape (N, 3) Locations in meters. subject : str | None The label subject. It is best practice to set this to the proper value on initialization, but it can also be set manually. values : array, shape (N,) Values at the vertices. vertices : array, shape (N,) Vertex indices (0 based) r0Nrc  Cs>t|tstdt|t|t}tt| tdkrHtd| dk rfddl m } | | } |dkr~t t|}n t|}|dkrtt|df}n t|}t|t|krt|ksntd|dkr|dk rt|dd}||_||_||_||_||_td|dd |_| |_||_||_dS) Nzhemi must be a string, not %srz-Vertices must be ordered in increasing order.)colorConverterzFvertices, values and pos need to have same length (number of vertices)F raise_error) isinstancestr ValueErrortyper9asarrayintrDdiffastypeZmatplotlib.colorsrRZto_rgbaoneslenzerosopbasenameverticesrJvalueshemicommentrsubjectr/namefilename) selfrdrJrerfrgrirjrhr/rrRr0r0r1__init__s6      "zLabel.__init__cCsf|d|_|d|_|d|_|d|_|d|_|dd|_|dd|_|d|_|d |_ dS) NrdrJrerfrgrhr/rirj) rdrJrerfrggetrhr/rirj)rkstater0r0r1 __setstate__s      zLabel.__setstate__c Cs0t|j|j|j|j|j|j|j|j|j d }|S)N) rdrJrerfrgrhr/rirj) dictrdrJrerfrgrhr/rirj)rkoutr0r0r1 __getstate__s zLabel.__getstate__cCsL|jdkrdn|jd}||jdk r.t|jnd7}t|}d||j|fS)N unknown, , unnamedz