B Y @sdZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZd d lmZmZGd d d eZGd ddeZdS)z )absolute_importN) iteritems)HasProps) GlyphRenderer)ColumnDataSource)AnyColorDictEitherFloatInstanceListString)Column ColumnLabelcsTeZdZdZeeeeedddZee e e eddZ e e ddZee eZeeeZeeddZed d dZed d Zed d dZe dd Ze dd Ze dd Ze dd Ze dd Ze dd ZfddZddZddZ e!ddZ"e!ddZ#ddZ$ddZ%dd Z&d!d"Z'd#d$Z(d%d&Z)d'd(Z*d)d*Z+d+d,Z,e-d-d.Z.Z/S)/CompositeGlyphaYRepresents a subset of data. A collection of hetero or homogeneous glyph renderers which represent a subset of data. The purpose of the composite glyph is to abstract away the details of constructing glyphs, based on the details of a subset of data, from the grouping operations that a generalized builders must implement. In general, the Builder operates at the full column oriented data source level, segmenting and assigning attributes from a large selection, while the composite glyphs will typically be passed an array-like structures with one or more singular attributes to apply. Another way to explain the concept is that the Builder operates as the groupby, as in pandas, while the CompositeGlyph operates as the function used in the apply. What is the responsibility of the Composite Glyph? - Produce GlyphRenderers - Apply any aggregations - Tag the GlyphRenderers with the group label - Apply transforms due to chart operations - Note: Operations require implementation of special methods NonezIdentifies the subset of data.)defaulthelpaU Array-like values, which are used as the input to the composite glyph. Most composite glyphs add their own representation of one or more values-like columns/arrays that they receive as inputs. These are compiled together for generating `source`, `data`, and `df` by the individual composite glyphs. )rzThe data source used for the contained glyph renderers. Simple glyphs part of the composite glyph might not use the column data source.z\A list of chart operations that can be applied to manipulate their visual depiction.ZgrayzsA high level color. Some glyphs will implement more specific color attributes for parts or specific glyphs.)rZblackz5A default outline color for contained glyphs.g?g?gc sJ|d}t|s"t|r,|g|d<tt|jf||dS)Nvalues)getrZis_validr superr__init__setup)self propertiesZvals) __class__.lib/python3.7/site-packages/bkcharts/models.pyrOs   zCompositeGlyph.__init__cCs*dd|D|_|jdk r&|dS)z=Build renderers and data source and set sources on renderers.cSsg|]}|qSrr).0rendererrrr Xsz(CompositeGlyph.setup..N)build_renderers renderersrefresh)rrrrrVs zCompositeGlyph.setupcCs`|jdk r\|}|dk r\t|tr,t|}t|tsH|dk rHtdn |||_|dS)zrUpdate the GlyphRenderers. .. note: this method would be called after data is added. Nz2build_source must return dict or ColumnDataSource.) r$ build_source isinstancedictr TypeErroradd_chart_indexsource _set_sources)rdatar+rrrr%\s    zCompositeGlyph.refreshcCs|jdk r|jjSiSdS)N)r+r-)rrrrr-qs zCompositeGlyph.datacCs|jrt|jStSdS)N)r-pdZ DataFrame)rrrrdfxs zCompositeGlyph.dfcCst|tr|}|j}nd}d|krtt|dkrtt|d}|jg||d<t|jtrx$t |jD]\}}|g|||<qrW|dk r||_|S|SdS)aFAdd identifier of the data group as a column for each row. Args: data (dict or `ColumnDataSource`): can be the type of data used internally to ColumnDataSource, or a ColumnDataSource. Returns: dict or `ColumnDataSource`: returns the same type of data provided NZ chart_indexr) r'rr-lenlistkeysrlabelr(r)rr-r+Zn_rowscolvalrrrr*s  zCompositeGlyph.add_chart_indexccs tVdS)N)r)rrrrr#szCompositeGlyph.build_rendererscCsi}|jdk rd|ji}|S)Nr)r)rr-rrrr&s  zCompositeGlyph.build_sourcecCsx|jD] }|j|_qWdS)zStore reference to source in each GlyphRenderer. .. note:: if the glyphs that are part of the composite glyph differ, you may have to override this method and handle the sources manually. N)r$r+Z data_source)rr!rrrr,s zCompositeGlyph._set_sourcescCsdS)zBA special method the `stack` function applies to composite glyphs.Nr)rglyphsrrr __stack__szCompositeGlyph.__stack__cCsdS)zCA special method the `jitter` function applies to composite glyphs.Nr)rr6rrr __jitter__szCompositeGlyph.__jitter__cCsdS)zBA special method the `dodge` function applies to composite glyphs.Nr)rr6rrr __dodge__szCompositeGlyph.__dodge__cCsdS)zDA special method the `overlay` function applies to composite glyphs.Nr)rr6rrr __overlay__szCompositeGlyph.__overlay__cCsdS)Nr)rrrrapply_operationsszCompositeGlyph.apply_operationscCs0i}x&t|jD]\}}|jdd||<qW|S)NT)Z withbases)rr6Zclass_properties)clsZpropsnameZglyphrrrglyph_propertiesszCompositeGlyph.glyph_properties)0__name__ __module__ __qualname____doc__r rr rr3rr rr rr+r rr$r6Z operationsrZcolorZ fill_colorZ line_colorZ fill_alphaZ line_alphaZ left_bufferZ right_bufferZ top_bufferZ bottom_bufferrrr%propertyr-r/r*r#r&r,r7r8r9r:r; classmethodr> __classcell__rr)rrrsJ            # rc@sXeZdZdZeeeddZeddZ eddZ e e ee ddZ ddZd d d Zd S) CollisionModifiera5Models an special type of operation that alters how glyphs interact. Used to handle the manipulation of glyphs for operations, such as stacking. The list of `CompositeGlyph`s can either be input into the `CollisionModifier` as keyword args, or added individually with the `add_glyph` method. zAA list of composite glyphs, to apply the modification to.)rz#The name of the collision modifier.zzThe name of the method that will be utilized on the composite glyphs. This method must exist on all `comp_glyphs`.zcSome collision modifiers might require column labels to apply the operation in relation to.cCs|j|dS)N) comp_glyphsappend)rZ comp_glyphrrr add_glyphszCollisionModifier.add_glyphNcCsPt|jdkr||_t|jdkrs     (D