@bokeh/bokehjs
Version:
Interactive, novel data visualization
101 lines • 3.5 kB
JavaScript
import { isField } from "../../../core/vectorization";
import { dict } from "../../../core/util/object";
import { isArray } from "../../../core/util/types";
import { EditTool, EditToolView } from "./edit_tool";
import { GlyphRenderer } from "../../renderers/glyph_renderer";
import { tool_icon_freehand_draw } from "../../../styles/icons.css";
export class FreehandDrawToolView extends EditToolView {
static __name__ = "FreehandDrawToolView";
_draw(ev, mode, emit = false) {
if (!this.model.active) {
return;
}
const renderer = this.model.renderers[0];
const point = this._map_drag(ev.sx, ev.sy, renderer);
if (point == null) {
return;
}
const [x, y] = point;
const { glyph, data_source } = renderer;
const xkey = isField(glyph.xs) ? glyph.xs.field : null;
const ykey = isField(glyph.ys) ? glyph.ys.field : null;
const data = dict(data_source.data);
if (mode == "new") {
this._pop_glyphs(data_source, this.model.num_objects);
if (xkey != null) {
data_source.get_array(xkey).push([x]);
}
if (ykey != null) {
data_source.get_array(ykey).push([y]);
}
this._pad_empty_columns(data_source, [xkey, ykey]);
}
else if (mode == "add") {
if (xkey != null) {
const column = data.get(xkey) ?? [];
const xidx = column.length - 1;
let xs = data_source.get_array(xkey)[xidx];
if (!isArray(xs)) {
xs = Array.from(xs);
column[xidx] = xs;
}
xs.push(x);
}
if (ykey != null) {
const column = data.get(ykey) ?? [];
const yidx = column.length - 1;
let ys = data_source.get_array(ykey)[yidx];
if (!isArray(ys)) {
ys = Array.from(ys);
column[yidx] = ys;
}
ys.push(y);
}
}
this._emit_cds_changes(data_source, true, true, emit);
}
_pan_start(ev) {
this._draw(ev, "new");
}
_pan(ev) {
this._draw(ev, "add");
}
_pan_end(ev) {
this._draw(ev, "add", true);
}
_tap(ev) {
this._select_event(ev, this._select_mode(ev), this.model.renderers);
}
_keyup(ev) {
if (!this.model.active || !this._mouse_in_frame) {
return;
}
for (const renderer of this.model.renderers) {
if (ev.key == "Escape") {
renderer.data_source.selection_manager.clear();
}
else if (ev.key == "Backspace") {
this._delete_selected(renderer);
}
}
}
}
export class FreehandDrawTool extends EditTool {
static __name__ = "FreehandDrawTool";
constructor(attrs) {
super(attrs);
}
static {
this.prototype.default_view = FreehandDrawToolView;
this.define(({ Int, List, Ref }) => ({
num_objects: [Int, 0],
renderers: [List(Ref((GlyphRenderer))), []],
}));
this.register_alias("freehand_draw", () => new FreehandDrawTool());
}
tool_name = "Freehand Draw Tool";
tool_icon = tool_icon_freehand_draw;
event_type = ["pan", "tap"];
default_order = 3;
}
//# sourceMappingURL=freehand_draw_tool.js.map