U /e1@sddlmZddlZeeZddlZddlmZddl Z ddl m Z ddl mZmZmZddlmZddlmZd d lmZmZed Zd Zd dddddddgZd)ddZddZddZdiifddZddZd d!Z d"d#Z!d$d%Z"d&d'Z#d(Z$dS)*) annotationsN)Iterable) ColorSpec)ColumnarDataSourceColumnDataSource GlyphRenderer)import_optional) nice_join)pop_legend_kwarg update_legendZpandas)create_renderer make_glyph pop_visualsnameZ coordinatesZ x_range_nameZ y_range_namelevelviewZvisibleZmutedcCs\ddddddddd d d d d dddddg}|rP|j}dd|D}t|}||S|dSdS)Nz#1f77b4z#ff7f0ez#ffbb78z#2ca02cz#98df8az#d62728z#ff9896z#9467bdz#c5b0d5z#8c564bz#c49c94z#e377c2z#f7b6d2z#7f7f7fz#bcbd22z#dbdb8dz#17becfz#9edae5cSsg|]}|jdkr|qS)r)Z__view_model__.0xrDs z%get_default_color..r) rendererslen)plotcolorsrZ num_renderersrrrget_default_color5s2 rcKspt|}t|}t|}|d}t||}g}|t||||7}|t||||7}|rjttt|ddt||d|ddid} tdd |Drt||d |d } nd} td d |Drt||d |d } nd} t||d|ddid} t |||} t ||| }t ||| }t ||| }t ||| }t f| |p0d|p8d||pBdd|}|j ||rlt ||||S)N data_sourceand)Z conjuctionZ nonselection_alphag?)prefixdefaultsoverride_defaultscss|]}|dVqdS) selection_N startswithrrrr fsz"create_renderer..r%)r"r#css|]}|dVqdS)hover_Nr&rrrrr(lsr)Zmuted_g?auto)glyphnonselection_glyphselection_glyph hover_glyph muted_glyph)_convert_data_sourcer _pop_renderer_argsr_process_sequence_literals RuntimeError_GLYPH_SOURCE_MSGr anyrrrappendr ) glyphclassrkwargsis_user_sourceZ legend_kwargZ renderer_kwssourceZ glyph_visuals incompatible_literal_spec_valuesZnonselection_visualsZselection_visualsZ hover_visualsZ muted_visualsr+r,r-r.r/Zglyph_rendererrrrrOsH        rcCs(|dkr dS|}|||f|S)N)copyupdate)r7kwsextrarrrrs  rc Cs,|}|dd|ddi}|dt|ddtt}}t|}tt|D]} t| \} } || |kr| || || <nf| |kr|| |kr||| || <n@| |kr|| || <n*| |kr|| || <n| |kr|| || <| |krb| | qb|D]} | || dq|S)a Applies basic cascading logic to deduce properties for a glyph. Args: glyphclass : the type of glyph being handled props (dict) : Maps properties and prefixed properties to their values. Keys in `props` matching `glyphclass` visual properties (those of 'line_', 'fill_', 'hatch_' or 'text_') with added `prefix` will get popped, other keys will be ignored. Keys take the form '[{prefix}][{feature}_]{trait}'. Only {feature} must not contain underscores. Keys of the form '{prefix}{trait}' work as lower precedence aliases for {trait} for all {features}, as long as the glyph has no property called {trait}. I.e. this won't apply to "width" in a `rect` glyph. Ex: {'fill_color': 'blue', 'selection_line_width': 0.5} prefix (str) : Prefix used when accessing `props`. Ex: 'selection_' override_defaults (dict) : Explicitly provided fallback based on '{trait}', in case property not set in `props`. Ex. 'width' here may be used for 'selection_line_width'. defaults (dict) : Property fallback, in case prefixed property not in `props` or `override_defaults`. Ex. 'line_width' here may be used for 'selection_line_width'. Returns: result (dict) : Resulting properties for the instance (no prefixes). Notes: Feature trait 'text_color', as well as traits 'color' and 'alpha', have ultimate defaults in case those can't be deduced. Z text_colorZblackZ hatch_colorcolorr!g?N) r< setdefaultrdictsetZ propertiesfilter _is_visual_split_feature_traitpopadd) r7propsr"r#r$Ztrait_defaultsresultZtraitsZ prop_namesr_traitrrrrs2+        rc Cs|dddk }|r|d}t|tsz t|}WnLtk r~}z.dt|d|}t|tdW5d}~XYnX||d<|S)Nr:zFailed to auto-convert z' to ColumnDataSource. Original error: r) get isinstancerr ValueErrortypewith_tracebacksysexc_info)r8r9r:errmsgrrrr0s  (r0cs(fddtD}dt|d<|S)Ncs i|]}|kr||qSr)rH)rattrr8rr sz&_pop_renderer_args..r:r) RENDERER_ARGSrHr)r8rKrrXrr1s  r1c Cstg}|}|D]X\}}t|ts*qt|tr6q||kr@qt|trLqt||trj|||rjqt|tj rFt||tr*|j dkr|j dkrn|j dkr|j dkrnt|j j dkr|j dkrn\|j dks|j j dkr|j dkr|j ddkrn,td|d |j d d tt|j d n|j dkrFtd |d|rX||q|j||d|||<q|S)NZuint32r Zuint8Ufr)zPColor columns need to be of type uint32[N], uint8[N] or uint8/float[N, {3, 4}] (z is [z, ]zColumns need to be 1D (z is not))r) dataspecsitemsrOrrCstrrZis_color_tuple_shapenpZndarrayZdtypendimkindshaper3joinmapr6rI)r7r8r:r9r;ravarvalrrrr2s>   ..   r2cCs&|dd}t|dkr|S|dgS)zAFeature is up to first '_'. Ex. 'line_color' => ['line', 'color']rLr rN)splitr)ftrrrrG3s rGcCst|\}}|dko|dk S)z&Whether a feature trait name is visual)linefillZhatchtextglobalN)rG)rmZfeaturerMrrrrF8s rFaH Expected %s to reference fields in the supplied data source. When a 'source' argument is passed to a glyph method, values that are sequences (like lists or arrays) must come from references to data columns in the source. For instance, as an example: source = ColumnDataSource(data=dict(x=a_list, y=an_array)) p.circle(x='x', y='y', source=source, ...) # pass column names and a source Alternatively, *all* data sequences may be provided as literals as long as a source is *not* provided: p.circle(x=a_list, y=an_array, ...) # pass actual sequences and no source )N)% __future__rlogging getLogger__name__logrScollections.abcrZnumpyrdZcore.propertiesrmodelsrrrZutil.dependenciesr Z util.stringr Z_legendsr r pd__all__rZrrrrr0r1r2rGrFr4rrrr s8       =W1