@quartic/bokehjs
Version:
Interactive, novel data visualization
139 lines (138 loc) • 5.71 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var extend = function (child, parent) { for (var key in parent) {
if (hasProp.call(parent, key))
child[key] = parent[key];
} function ctor() { this.constructor = child; } ctor.prototype = parent.prototype; child.prototype = new ctor(); child.__super__ = parent.prototype; return child; }, hasProp = {}.hasOwnProperty;
var annotation_1 = require("./annotation");
var arrow_head_1 = require("./arrow_head");
var column_data_source_1 = require("../sources/column_data_source");
var p = require("core/properties");
var math_1 = require("core/util/math");
exports.ArrowView = (function (superClass) {
extend(ArrowView, superClass);
function ArrowView() {
return ArrowView.__super__.constructor.apply(this, arguments);
}
ArrowView.prototype.initialize = function (options) {
ArrowView.__super__.initialize.call(this, options);
if (this.model.source == null) {
this.model.source = new column_data_source_1.ColumnDataSource();
}
this.canvas = this.plot_model.canvas;
this.xmapper = this.plot_view.frame.x_mappers[this.model.x_range_name];
this.ymapper = this.plot_view.frame.y_mappers[this.model.y_range_name];
return this.set_data(this.model.source);
};
ArrowView.prototype.bind_bokeh_events = function () {
this.listenTo(this.model, 'change', this.plot_view.request_render);
return this.listenTo(this.model.source, 'change', function () {
this.set_data(this.model.source);
return this.plot_view.request_render();
});
};
ArrowView.prototype.set_data = function (source) {
ArrowView.__super__.set_data.call(this, source);
return this.visuals.warm_cache(source);
};
ArrowView.prototype._map_data = function () {
var end, start, x_name, y_name;
if (this.model.start_units === 'data') {
start = this.plot_view.map_to_screen(this._x_start, this._y_start, x_name = this.model.x_range_name, y_name = this.model.y_range_name);
}
else {
start = [this.canvas.v_vx_to_sx(this._x_start), this.canvas.v_vy_to_sy(this._y_start)];
}
if (this.model.end_units === 'data') {
end = this.plot_view.map_to_screen(this._x_end, this._y_end, x_name = this.model.x_range_name, y_name = this.model.y_range_name);
}
else {
end = [this.canvas.v_vx_to_sx(this._x_end), this.canvas.v_vy_to_sy(this._y_end)];
}
return [start, end];
};
ArrowView.prototype.render = function () {
var ctx, ref;
if (!this.model.visible) {
return;
}
ctx = this.plot_view.canvas_view.ctx;
ctx.save();
ref = this._map_data(), this.start = ref[0], this.end = ref[1];
if (this.model.end != null) {
this._arrow_head(ctx, "render", this.model.end, this.start, this.end);
}
if (this.model.start != null) {
this._arrow_head(ctx, "render", this.model.start, this.end, this.start);
}
ctx.beginPath();
ctx.rect(0, 0, this.canvas.width, this.canvas.height);
if (this.model.end != null) {
this._arrow_head(ctx, "clip", this.model.end, this.start, this.end);
}
if (this.model.start != null) {
this._arrow_head(ctx, "clip", this.model.start, this.end, this.start);
}
ctx.closePath();
ctx.clip();
this._arrow_body(ctx);
return ctx.restore();
};
ArrowView.prototype._arrow_body = function (ctx) {
var i, j, ref, results;
if (!this.visuals.line.doit) {
return;
}
results = [];
for (i = j = 0, ref = this._x_start.length; 0 <= ref ? j < ref : j > ref; i = 0 <= ref ? ++j : --j) {
this.visuals.line.set_vectorize(ctx, i);
ctx.beginPath();
ctx.moveTo(this.start[0][i], this.start[1][i]);
ctx.lineTo(this.end[0][i], this.end[1][i]);
results.push(ctx.stroke());
}
return results;
};
ArrowView.prototype._arrow_head = function (ctx, action, head, start, end) {
var angle, i, j, ref, results;
results = [];
for (i = j = 0, ref = this._x_start.length; 0 <= ref ? j < ref : j > ref; i = 0 <= ref ? ++j : --j) {
angle = Math.PI / 2 + math_1.atan2([start[0][i], start[1][i]], [end[0][i], end[1][i]]);
ctx.save();
ctx.translate(end[0][i], end[1][i]);
ctx.rotate(angle);
if (action === "render") {
head.render(ctx);
}
else if (action === "clip") {
head.clip(ctx);
}
results.push(ctx.restore());
}
return results;
};
return ArrowView;
})(annotation_1.AnnotationView);
exports.Arrow = (function (superClass) {
extend(Arrow, superClass);
function Arrow() {
return Arrow.__super__.constructor.apply(this, arguments);
}
Arrow.prototype.default_view = exports.ArrowView;
Arrow.prototype.type = 'Arrow';
Arrow.mixins(['line']);
Arrow.define({
x_start: [p.NumberSpec],
y_start: [p.NumberSpec],
start_units: [p.String, 'data'],
start: [p.Instance, null],
x_end: [p.NumberSpec],
y_end: [p.NumberSpec],
end_units: [p.String, 'data'],
end: [p.Instance, new arrow_head_1.OpenHead({})],
source: [p.Instance],
x_range_name: [p.String, 'default'],
y_range_name: [p.String, 'default']
});
return Arrow;
})(annotation_1.Annotation);