/
box_select_tool.coffee
134 lines (106 loc) · 3.5 KB
/
box_select_tool.coffee
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
_ = require "underscore"
SelectTool = require "./select_tool"
BoxAnnotation = require "../../annotations/box_annotation"
p = require "../../../core/properties"
class BoxSelectToolView extends SelectTool.View
_pan_start: (e) ->
canvas = @plot_view.canvas
@_baseboint = [
canvas.sx_to_vx(e.bokeh.sx)
canvas.sy_to_vy(e.bokeh.sy)
]
return null
_pan: (e) ->
canvas = @plot_view.canvas
curpoint = [
canvas.sx_to_vx(e.bokeh.sx)
canvas.sy_to_vy(e.bokeh.sy)
]
frame = @plot_model.frame
dims = @model.dimensions
[vxlim, vylim] = @model._get_dim_limits(@_baseboint, curpoint, frame, dims)
@model.overlay.update({left: vxlim[0], right: vxlim[1], top: vylim[1], bottom: vylim[0]})
if @model.select_every_mousemove
append = e.srcEvent.shiftKey ? false
@_select(vxlim, vylim, false, append)
return null
_pan_end: (e) ->
canvas = @plot_view.canvas
curpoint = [
canvas.sx_to_vx(e.bokeh.sx)
canvas.sy_to_vy(e.bokeh.sy)
]
frame = @plot_model.frame
dims = @model.dimensions
[vxlim, vylim] = @model._get_dim_limits(@_baseboint, curpoint, frame, dims)
append = e.srcEvent.shiftKey ? false
@_select(vxlim, vylim, true, append)
@model.overlay.update({left: null, right: null, top: null, bottom: null})
@_baseboint = null
@plot_view.push_state('box_select', {selection: @plot_view.get_selection()})
return null
_select: ([vx0, vx1], [vy0, vy1], final, append=false) ->
geometry = {
type: 'rect'
vx0: vx0
vx1: vx1
vy0: vy0
vy1: vy1
}
for r in @model.computed_renderers
ds = r.data_source
sm = ds.selection_manager
sm.select(@, @plot_view.renderer_views[r.id], geometry, final, append)
if @model.callback?
@_emit_callback(geometry)
@_save_geometry(geometry, final, append)
return null
_emit_callback: (geometry) ->
r = @model.computed_renderers[0]
canvas = @plot_model.canvas
frame = @plot_model.frame
geometry['sx0'] = canvas.vx_to_sx(geometry.vx0)
geometry['sx1'] = canvas.vx_to_sx(geometry.vx1)
geometry['sy0'] = canvas.vy_to_sy(geometry.vy0)
geometry['sy1'] = canvas.vy_to_sy(geometry.vy1)
xmapper = frame.x_mappers[r.x_range_name]
ymapper = frame.y_mappers[r.y_range_name]
geometry['x0'] = xmapper.map_from_target(geometry.vx0)
geometry['x1'] = xmapper.map_from_target(geometry.vx1)
geometry['y0'] = ymapper.map_from_target(geometry.vy0)
geometry['y1'] = ymapper.map_from_target(geometry.vy1)
@model.callback.execute(@model, {geometry: geometry})
return
DEFAULT_BOX_OVERLAY = () -> new BoxAnnotation.Model({
level: "overlay"
render_mode: "css"
top_units: "screen"
left_units: "screen"
bottom_units: "screen"
right_units: "screen"
fill_color: "lightgrey"
fill_alpha: 0.5
line_color: "black"
line_alpha: 1.0
line_width: 2
line_dash: [4, 4]
})
class BoxSelectTool extends SelectTool.Model
default_view: BoxSelectToolView
type: "BoxSelectTool"
tool_name: "Box Select"
icon: "bk-tool-icon-box-select"
event_type: "pan"
default_order: 30
@define {
dimensions: [ p.Dimensions, "both" ]
select_every_mousemove: [ p. Bool, false ]
callback: [ p.Instance ]
overlay: [ p.Instance, DEFAULT_BOX_OVERLAY ]
}
@getters {
tooltip: () -> @_get_dim_tooltip(@tool_name, @dimensions)
}
module.exports =
Model: BoxSelectTool
View: BoxSelectToolView