|
814
|
1 |
#!/usr/bin/env python
|
|
|
2 |
# -*- coding: utf-8 -*-
|
|
|
3 |
|
|
|
4 |
#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor
|
|
|
5 |
#based on the plcopen standard.
|
|
|
6 |
#
|
|
|
7 |
#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
|
|
|
8 |
#
|
|
|
9 |
#See COPYING file for copyrights details.
|
|
|
10 |
#
|
|
|
11 |
#This library is free software; you can redistribute it and/or
|
|
|
12 |
#modify it under the terms of the GNU General Public
|
|
|
13 |
#License as published by the Free Software Foundation; either
|
|
|
14 |
#version 2.1 of the License, or (at your option) any later version.
|
|
|
15 |
#
|
|
|
16 |
#This library is distributed in the hope that it will be useful,
|
|
|
17 |
#but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
18 |
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
19 |
#General Public License for more details.
|
|
|
20 |
#
|
|
|
21 |
#You should have received a copy of the GNU General Public
|
|
|
22 |
#License along with this library; if not, write to the Free Software
|
|
|
23 |
#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
24 |
|
|
|
25 |
import numpy
|
|
|
26 |
import math
|
|
|
27 |
|
|
|
28 |
import wx
|
|
|
29 |
import wx.lib.plot as plot
|
|
|
30 |
import wx.lib.buttons
|
|
|
31 |
|
|
|
32 |
from graphics.GraphicCommons import DebugViewer, MODE_SELECTION, MODE_MOTION
|
|
|
33 |
from EditorPanel import EditorPanel
|
|
|
34 |
from util.BitmapLibrary import GetBitmap
|
|
|
35 |
|
|
|
36 |
colours = ['blue', 'red', 'green', 'yellow', 'orange', 'purple', 'brown', 'cyan',
|
|
|
37 |
'pink', 'grey']
|
|
|
38 |
markers = ['circle', 'dot', 'square', 'triangle', 'triangle_down', 'cross', 'plus', 'circle']
|
|
|
39 |
|
|
|
40 |
|
|
|
41 |
#-------------------------------------------------------------------------------
|
|
|
42 |
# Debug Variable Graphic Viewer class
|
|
|
43 |
#-------------------------------------------------------------------------------
|
|
|
44 |
|
|
|
45 |
SECOND = 1000000000
|
|
|
46 |
MINUTE = 60 * SECOND
|
|
|
47 |
HOUR = 60 * MINUTE
|
|
|
48 |
|
|
|
49 |
ZOOM_VALUES = map(lambda x:("x %.1f" % x, x), [math.sqrt(2) ** i for i in xrange(8)])
|
|
|
50 |
RANGE_VALUES = map(lambda x: (str(x), x), [25 * 2 ** i for i in xrange(6)])
|
|
|
51 |
TIME_RANGE_VALUES = [("%ds" % i, i * SECOND) for i in (1, 2, 5, 10, 20, 30)] + \
|
|
|
52 |
[("%dm" % i, i * MINUTE) for i in (1, 2, 5, 10, 20, 30)] + \
|
|
|
53 |
[("%dh" % i, i * HOUR) for i in (1, 2, 3, 6, 12, 24)]
|
|
|
54 |
|
|
|
55 |
class GraphicViewer(EditorPanel, DebugViewer):
|
|
|
56 |
|
|
|
57 |
def _init_Editor(self, prnt):
|
|
|
58 |
self.Editor = wx.Panel(prnt)
|
|
|
59 |
|
|
|
60 |
main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)
|
|
|
61 |
main_sizer.AddGrowableCol(0)
|
|
|
62 |
main_sizer.AddGrowableRow(0)
|
|
|
63 |
|
|
|
64 |
self.Canvas = plot.PlotCanvas(self.Editor, name='Canvas')
|
|
|
65 |
def _axisInterval(spec, lower, upper):
|
|
|
66 |
if spec == 'border':
|
|
|
67 |
if lower == upper:
|
|
|
68 |
return lower - 0.5, upper + 0.5
|
|
|
69 |
else:
|
|
|
70 |
border = (upper - lower) * 0.05
|
|
|
71 |
return lower - border, upper + border
|
|
|
72 |
else:
|
|
|
73 |
return plot.PlotCanvas._axisInterval(self.Canvas, spec, lower, upper)
|
|
|
74 |
self.Canvas._axisInterval = _axisInterval
|
|
|
75 |
self.Canvas.SetYSpec('border')
|
|
|
76 |
self.Canvas.canvas.Bind(wx.EVT_LEFT_DOWN, self.OnCanvasLeftDown)
|
|
|
77 |
self.Canvas.canvas.Bind(wx.EVT_LEFT_UP, self.OnCanvasLeftUp)
|
|
|
78 |
self.Canvas.canvas.Bind(wx.EVT_MIDDLE_DOWN, self.OnCanvasMiddleDown)
|
|
|
79 |
self.Canvas.canvas.Bind(wx.EVT_MIDDLE_UP, self.OnCanvasMiddleUp)
|
|
|
80 |
self.Canvas.canvas.Bind(wx.EVT_MOTION, self.OnCanvasMotion)
|
|
|
81 |
self.Canvas.canvas.Bind(wx.EVT_SIZE, self.OnCanvasResize)
|
|
|
82 |
main_sizer.AddWindow(self.Canvas, 0, border=0, flag=wx.GROW)
|
|
|
83 |
|
|
|
84 |
range_sizer = wx.FlexGridSizer(cols=10, hgap=5, rows=1, vgap=0)
|
|
|
85 |
range_sizer.AddGrowableCol(5)
|
|
|
86 |
range_sizer.AddGrowableRow(0)
|
|
|
87 |
main_sizer.AddSizer(range_sizer, 0, border=5, flag=wx.GROW|wx.ALL)
|
|
|
88 |
|
|
|
89 |
range_label = wx.StaticText(self.Editor, label=_('Range:'))
|
|
|
90 |
range_sizer.AddWindow(range_label, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
|
91 |
|
|
|
92 |
self.CanvasRange = wx.ComboBox(self.Editor,
|
|
|
93 |
size=wx.Size(100, 28), style=wx.CB_READONLY)
|
|
|
94 |
self.Bind(wx.EVT_COMBOBOX, self.OnRangeChanged, self.CanvasRange)
|
|
|
95 |
range_sizer.AddWindow(self.CanvasRange, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
|
96 |
|
|
|
97 |
zoom_label = wx.StaticText(self.Editor, label=_('Zoom:'))
|
|
|
98 |
range_sizer.AddWindow(zoom_label, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
|
99 |
|
|
|
100 |
self.CanvasZoom = wx.ComboBox(self.Editor,
|
|
|
101 |
size=wx.Size(70, 28), style=wx.CB_READONLY)
|
|
|
102 |
self.Bind(wx.EVT_COMBOBOX, self.OnZoomChanged, self.CanvasZoom)
|
|
|
103 |
range_sizer.AddWindow(self.CanvasZoom, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
|
104 |
|
|
|
105 |
position_label = wx.StaticText(self.Editor, label=_('Position:'))
|
|
|
106 |
range_sizer.AddWindow(position_label, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
|
107 |
|
|
|
108 |
self.CanvasPosition = wx.ScrollBar(self.Editor,
|
|
|
109 |
size=wx.Size(0, 16), style=wx.SB_HORIZONTAL)
|
|
|
110 |
self.CanvasPosition.SetScrollbar(0, 10, 100, 10)
|
|
|
111 |
self.CanvasPosition.Bind(wx.EVT_SCROLL_THUMBTRACK,
|
|
|
112 |
self.OnPositionChanging, self.CanvasPosition)
|
|
|
113 |
self.CanvasPosition.Bind(wx.EVT_SCROLL_LINEUP,
|
|
|
114 |
self.OnPositionChanging, self.CanvasPosition)
|
|
|
115 |
self.CanvasPosition.Bind(wx.EVT_SCROLL_LINEDOWN,
|
|
|
116 |
self.OnPositionChanging, self.CanvasPosition)
|
|
|
117 |
self.CanvasPosition.Bind(wx.EVT_SCROLL_PAGEUP,
|
|
|
118 |
self.OnPositionChanging, self.CanvasPosition)
|
|
|
119 |
self.CanvasPosition.Bind(wx.EVT_SCROLL_PAGEDOWN,
|
|
|
120 |
self.OnPositionChanging, self.CanvasPosition)
|
|
|
121 |
range_sizer.AddWindow(self.CanvasPosition, 0, border=5, flag=wx.GROW|wx.ALL)
|
|
|
122 |
|
|
|
123 |
self.ResetButton = wx.lib.buttons.GenBitmapButton(self.Editor,
|
|
|
124 |
bitmap=GetBitmap("reset"), size=wx.Size(28, 28), style=wx.NO_BORDER)
|
|
|
125 |
self.ResetButton.SetToolTipString(_("Clear the graph values"))
|
|
|
126 |
self.Bind(wx.EVT_BUTTON, self.OnResetButton, self.ResetButton)
|
|
|
127 |
range_sizer.AddWindow(self.ResetButton, 0, border=0, flag=0)
|
|
|
128 |
|
|
|
129 |
self.CurrentButton = wx.lib.buttons.GenBitmapButton(self.Editor,
|
|
|
130 |
bitmap=GetBitmap("current"), size=wx.Size(28, 28), style=wx.NO_BORDER)
|
|
|
131 |
self.CurrentButton.SetToolTipString(_("Go to current value"))
|
|
|
132 |
self.Bind(wx.EVT_BUTTON, self.OnCurrentButton, self.CurrentButton)
|
|
|
133 |
range_sizer.AddWindow(self.CurrentButton, 0, border=0, flag=0)
|
|
|
134 |
|
|
|
135 |
self.ResetZoomOffsetButton = wx.lib.buttons.GenBitmapButton(self.Editor,
|
|
|
136 |
bitmap=GetBitmap("fit"), size=wx.Size(28, 28), style=wx.NO_BORDER)
|
|
|
137 |
self.ResetZoomOffsetButton.SetToolTipString(_("Reset zoom and offset"))
|
|
|
138 |
self.Bind(wx.EVT_BUTTON, self.OnResetZoomOffsetButton,
|
|
|
139 |
self.ResetZoomOffsetButton)
|
|
|
140 |
range_sizer.AddWindow(self.ResetZoomOffsetButton, 0, border=0, flag=0)
|
|
|
141 |
|
|
|
142 |
self.ExportGraphButton = wx.lib.buttons.GenBitmapButton(self.Editor,
|
|
|
143 |
bitmap=GetBitmap("export_graph"), size=wx.Size(28, 28), style=wx.NO_BORDER)
|
|
|
144 |
self.ExportGraphButton.SetToolTipString(_("Export graph values to clipboard"))
|
|
|
145 |
self.Bind(wx.EVT_BUTTON, self.OnExportGraphButtonClick,
|
|
|
146 |
self.ExportGraphButton)
|
|
|
147 |
range_sizer.AddWindow(self.ExportGraphButton, 0, border=0, flag=0)
|
|
|
148 |
|
|
|
149 |
self.Editor.SetSizer(main_sizer)
|
|
|
150 |
|
|
|
151 |
self.Editor.Bind(wx.EVT_MOUSEWHEEL, self.OnCanvasMouseWheel)
|
|
|
152 |
|
|
|
153 |
def __init__(self, parent, window, producer, instancepath = ""):
|
|
|
154 |
EditorPanel.__init__(self, parent, "", window, None)
|
|
|
155 |
DebugViewer.__init__(self, producer, True, False)
|
|
|
156 |
|
|
|
157 |
self.InstancePath = instancepath
|
|
|
158 |
self.RangeValues = None
|
|
|
159 |
self.CursorIdx = None
|
|
|
160 |
self.LastCursor = None
|
|
|
161 |
self.CurrentMousePos = None
|
|
|
162 |
self.CurrentMotionValue = None
|
|
|
163 |
self.Dragging = False
|
|
|
164 |
|
|
|
165 |
# Initialize Viewer mode to Selection mode
|
|
|
166 |
self.Mode = MODE_SELECTION
|
|
|
167 |
|
|
|
168 |
self.Datas = []
|
|
|
169 |
self.StartTick = 0
|
|
|
170 |
self.StartIdx = 0
|
|
|
171 |
self.EndIdx = 0
|
|
|
172 |
self.MinValue = None
|
|
|
173 |
self.MaxValue = None
|
|
|
174 |
self.YCenter = 0
|
|
|
175 |
self.CurrentZoom = 1.0
|
|
|
176 |
self.Fixed = False
|
|
|
177 |
self.Ticktime = self.DataProducer.GetTicktime()
|
|
|
178 |
self.RefreshCanvasRange()
|
|
|
179 |
|
|
|
180 |
for zoom_txt, zoom in ZOOM_VALUES:
|
|
|
181 |
self.CanvasZoom.Append(zoom_txt)
|
|
|
182 |
self.CanvasZoom.SetSelection(0)
|
|
|
183 |
|
|
|
184 |
self.AddDataConsumer(self.InstancePath.upper(), self)
|
|
|
185 |
|
|
|
186 |
def __del__(self):
|
|
|
187 |
DebugViewer.__del__(self)
|
|
|
188 |
self.RemoveDataConsumer(self)
|
|
|
189 |
|
|
|
190 |
def GetTitle(self):
|
|
|
191 |
if len(self.InstancePath) > 15:
|
|
|
192 |
return "..." + self.InstancePath[-12:]
|
|
|
193 |
return self.InstancePath
|
|
|
194 |
|
|
|
195 |
# Changes Viewer mode
|
|
|
196 |
def SetMode(self, mode):
|
|
|
197 |
if self.Mode != mode or mode == MODE_SELECTION:
|
|
|
198 |
if self.Mode == MODE_MOTION:
|
|
|
199 |
wx.CallAfter(self.Canvas.canvas.SetCursor, wx.NullCursor)
|
|
|
200 |
self.Mode = mode
|
|
|
201 |
if self.Mode == MODE_MOTION:
|
|
|
202 |
wx.CallAfter(self.Canvas.canvas.SetCursor, wx.StockCursor(wx.CURSOR_HAND))
|
|
|
203 |
|
|
|
204 |
def ResetView(self, register=False):
|
|
|
205 |
self.Datas = []
|
|
|
206 |
self.StartTick = 0
|
|
|
207 |
self.StartIdx = 0
|
|
|
208 |
self.EndIdx = 0
|
|
|
209 |
self.MinValue = None
|
|
|
210 |
self.MaxValue = None
|
|
|
211 |
self.CursorIdx = None
|
|
|
212 |
self.Fixed = False
|
|
|
213 |
self.Ticktime = self.DataProducer.GetTicktime()
|
|
|
214 |
if register:
|
|
|
215 |
self.AddDataConsumer(self.InstancePath.upper(), self)
|
|
|
216 |
self.ResetLastCursor()
|
|
|
217 |
self.RefreshCanvasRange()
|
|
|
218 |
self.RefreshView()
|
|
|
219 |
|
|
|
220 |
def RefreshNewData(self, *args, **kwargs):
|
|
|
221 |
self.RefreshView(*args, **kwargs)
|
|
|
222 |
DebugViewer.RefreshNewData(self)
|
|
|
223 |
|
|
|
224 |
def GetNearestData(self, tick, adjust):
|
|
|
225 |
ticks = numpy.array(zip(*self.Datas)[0])
|
|
|
226 |
new_cursor = numpy.argmin(abs(ticks - tick))
|
|
|
227 |
if adjust == -1 and ticks[new_cursor] > tick and new_cursor > 0:
|
|
|
228 |
new_cursor -= 1
|
|
|
229 |
elif adjust == 1 and ticks[new_cursor] < tick and new_cursor < len(self.Datas):
|
|
|
230 |
new_cursor += 1
|
|
|
231 |
return new_cursor
|
|
|
232 |
|
|
|
233 |
def GetBounds(self):
|
|
|
234 |
if self.StartIdx is None or self.EndIdx is None:
|
|
|
235 |
self.StartIdx = self.GetNearestData(self.StartTick, -1)
|
|
|
236 |
self.EndIdx = self.GetNearestData(self.StartTick + self.CurrentRange, 1)
|
|
|
237 |
|
|
|
238 |
def ResetBounds(self):
|
|
|
239 |
self.StartIdx = None
|
|
|
240 |
self.EndIdx = None
|
|
|
241 |
|
|
|
242 |
def RefreshCanvasRange(self):
|
|
|
243 |
if self.Ticktime == 0 and self.RangeValues != RANGE_VALUES:
|
|
|
244 |
self.RangeValues = RANGE_VALUES
|
|
|
245 |
self.CanvasRange.Clear()
|
|
|
246 |
for text, value in RANGE_VALUES:
|
|
|
247 |
self.CanvasRange.Append(text)
|
|
|
248 |
self.CanvasRange.SetStringSelection(RANGE_VALUES[0][0])
|
|
|
249 |
self.CurrentRange = RANGE_VALUES[0][1]
|
|
|
250 |
elif self.RangeValues != TIME_RANGE_VALUES:
|
|
|
251 |
self.RangeValues = TIME_RANGE_VALUES
|
|
|
252 |
self.CanvasRange.Clear()
|
|
|
253 |
for text, value in TIME_RANGE_VALUES:
|
|
|
254 |
self.CanvasRange.Append(text)
|
|
|
255 |
self.CanvasRange.SetStringSelection(TIME_RANGE_VALUES[0][0])
|
|
|
256 |
self.CurrentRange = TIME_RANGE_VALUES[0][1] / self.Ticktime
|
|
|
257 |
|
|
|
258 |
def RefreshView(self, force=False):
|
|
|
259 |
self.Freeze()
|
|
|
260 |
if force or not self.Fixed or (len(self.Datas) > 0 and self.StartTick + self.CurrentRange > self.Datas[-1][0]):
|
|
|
261 |
if (self.MinValue is not None and
|
|
|
262 |
self.MaxValue is not None and
|
|
|
263 |
self.MinValue != self.MaxValue):
|
|
|
264 |
Yrange = float(self.MaxValue - self.MinValue) / self.CurrentZoom
|
|
|
265 |
else:
|
|
|
266 |
Yrange = 2. / self.CurrentZoom
|
|
|
267 |
|
|
|
268 |
if not force and not self.Fixed and len(self.Datas) > 0:
|
|
|
269 |
self.YCenter = max(self.Datas[-1][1] - Yrange / 2,
|
|
|
270 |
min(self.YCenter,
|
|
|
271 |
self.Datas[-1][1] + Yrange / 2))
|
|
|
272 |
|
|
|
273 |
var_name = self.InstancePath.split(".")[-1]
|
|
|
274 |
|
|
|
275 |
self.GetBounds()
|
|
|
276 |
self.VariableGraphic = plot.PolyLine(self.Datas[self.StartIdx:self.EndIdx + 1],
|
|
|
277 |
legend=var_name, colour=colours[0])
|
|
|
278 |
self.GraphicsObject = plot.PlotGraphics([self.VariableGraphic], _("%s Graphics") % var_name, _("Tick"), _("Values"))
|
|
|
279 |
self.Canvas.Draw(self.GraphicsObject,
|
|
|
280 |
xAxis=(self.StartTick, self.StartTick + self.CurrentRange),
|
|
|
281 |
yAxis=(self.YCenter - Yrange * 1.1 / 2., self.YCenter + Yrange * 1.1 / 2.))
|
|
|
282 |
|
|
|
283 |
# Reset and draw cursor
|
|
|
284 |
self.ResetLastCursor()
|
|
|
285 |
self.RefreshCursor()
|
|
|
286 |
|
|
|
287 |
self.RefreshScrollBar()
|
|
|
288 |
|
|
|
289 |
self.Thaw()
|
|
|
290 |
|
|
|
291 |
def GetInstancePath(self):
|
|
|
292 |
return self.InstancePath
|
|
|
293 |
|
|
|
294 |
def IsViewing(self, tagname):
|
|
|
295 |
return self.InstancePath == tagname
|
|
|
296 |
|
|
|
297 |
def NewValue(self, tick, value, forced=False):
|
|
|
298 |
value = {True:1., False:0.}.get(value, float(value))
|
|
|
299 |
self.Datas.append((float(tick), value))
|
|
|
300 |
if self.MinValue is None:
|
|
|
301 |
self.MinValue = value
|
|
|
302 |
else:
|
|
|
303 |
self.MinValue = min(self.MinValue, value)
|
|
|
304 |
if self.MaxValue is None:
|
|
|
305 |
self.MaxValue = value
|
|
|
306 |
else:
|
|
|
307 |
self.MaxValue = max(self.MaxValue, value)
|
|
|
308 |
if not self.Fixed or tick < self.StartTick + self.CurrentRange:
|
|
|
309 |
self.GetBounds()
|
|
|
310 |
while int(self.Datas[self.StartIdx][0]) < tick - self.CurrentRange:
|
|
|
311 |
self.StartIdx += 1
|
|
|
312 |
self.EndIdx += 1
|
|
|
313 |
self.StartTick = self.Datas[self.StartIdx][0]
|
|
|
314 |
self.NewDataAvailable()
|
|
|
315 |
|
|
|
316 |
def RefreshScrollBar(self):
|
|
|
317 |
if len(self.Datas) > 0:
|
|
|
318 |
self.GetBounds()
|
|
|
319 |
pos = int(self.Datas[self.StartIdx][0] - self.Datas[0][0])
|
|
|
320 |
range = int(self.Datas[-1][0] - self.Datas[0][0])
|
|
|
321 |
else:
|
|
|
322 |
pos = 0
|
|
|
323 |
range = 0
|
|
|
324 |
self.CanvasPosition.SetScrollbar(pos, self.CurrentRange, range, self.CurrentRange)
|
|
|
325 |
|
|
|
326 |
def RefreshRange(self):
|
|
|
327 |
if len(self.Datas) > 0:
|
|
|
328 |
if self.Fixed and self.Datas[-1][0] - self.Datas[0][0] < self.CurrentRange:
|
|
|
329 |
self.Fixed = False
|
|
|
330 |
self.ResetBounds()
|
|
|
331 |
if self.Fixed:
|
|
|
332 |
self.StartTick = min(self.StartTick, self.Datas[-1][0] - self.CurrentRange)
|
|
|
333 |
else:
|
|
|
334 |
self.StartTick = max(self.Datas[0][0], self.Datas[-1][0] - self.CurrentRange)
|
|
|
335 |
self.RefreshView(True)
|
|
|
336 |
|
|
|
337 |
def OnRangeChanged(self, event):
|
|
|
338 |
try:
|
|
|
339 |
if self.Ticktime == 0:
|
|
|
340 |
self.CurrentRange = self.RangeValues[self.CanvasRange.GetSelection()][1]
|
|
|
341 |
else:
|
|
|
342 |
self.CurrentRange = self.RangeValues[self.CanvasRange.GetSelection()][1] / self.Ticktime
|
|
|
343 |
except ValueError, e:
|
|
|
344 |
self.CanvasRange.SetValue(str(self.CurrentRange))
|
|
|
345 |
wx.CallAfter(self.RefreshRange)
|
|
|
346 |
event.Skip()
|
|
|
347 |
|
|
|
348 |
def OnZoomChanged(self, event):
|
|
|
349 |
self.CurrentZoom = ZOOM_VALUES[self.CanvasZoom.GetSelection()][1]
|
|
|
350 |
wx.CallAfter(self.RefreshView, True)
|
|
|
351 |
event.Skip()
|
|
|
352 |
|
|
|
353 |
def OnPositionChanging(self, event):
|
|
|
354 |
if len(self.Datas) > 0:
|
|
|
355 |
self.ResetBounds()
|
|
|
356 |
self.StartTick = self.Datas[0][0] + event.GetPosition()
|
|
|
357 |
self.Fixed = True
|
|
|
358 |
self.NewDataAvailable(True)
|
|
|
359 |
event.Skip()
|
|
|
360 |
|
|
|
361 |
def OnResetButton(self, event):
|
|
|
362 |
self.Fixed = False
|
|
|
363 |
self.ResetView()
|
|
|
364 |
event.Skip()
|
|
|
365 |
|
|
|
366 |
def OnCurrentButton(self, event):
|
|
|
367 |
if len(self.Datas) > 0:
|
|
|
368 |
self.ResetBounds()
|
|
|
369 |
self.StartTick = max(self.Datas[0][0], self.Datas[-1][0] - self.CurrentRange)
|
|
|
370 |
self.Fixed = False
|
|
|
371 |
self.NewDataAvailable(True)
|
|
|
372 |
event.Skip()
|
|
|
373 |
|
|
|
374 |
def OnResetZoomOffsetButton(self, event):
|
|
|
375 |
if len(self.Datas) > 0:
|
|
|
376 |
self.YCenter = (self.MaxValue + self.MinValue) / 2
|
|
|
377 |
else:
|
|
|
378 |
self.YCenter = 0.0
|
|
|
379 |
self.CurrentZoom = 1.0
|
|
|
380 |
self.CanvasZoom.SetSelection(0)
|
|
|
381 |
wx.CallAfter(self.RefreshView, True)
|
|
|
382 |
event.Skip()
|
|
|
383 |
|
|
|
384 |
def OnExportGraphButtonClick(self, event):
|
|
|
385 |
data_copy = self.Datas[:]
|
|
|
386 |
text = "tick;%s;\n" % self.InstancePath
|
|
|
387 |
for tick, value in data_copy:
|
|
|
388 |
text += "%d;%.3f;\n" % (tick, value)
|
|
|
389 |
self.ParentWindow.SetCopyBuffer(text)
|
|
|
390 |
event.Skip()
|
|
|
391 |
|
|
|
392 |
def OnCanvasLeftDown(self, event):
|
|
|
393 |
self.Fixed = True
|
|
|
394 |
self.Canvas.canvas.CaptureMouse()
|
|
|
395 |
if len(self.Datas) > 0:
|
|
|
396 |
if self.Mode == MODE_SELECTION:
|
|
|
397 |
self.Dragging = True
|
|
|
398 |
pos = self.Canvas.PositionScreenToUser(event.GetPosition())
|
|
|
399 |
self.CursorIdx = self.GetNearestData(pos[0], -1)
|
|
|
400 |
self.RefreshCursor()
|
|
|
401 |
elif self.Mode == MODE_MOTION:
|
|
|
402 |
self.GetBounds()
|
|
|
403 |
self.CurrentMousePos = event.GetPosition()
|
|
|
404 |
self.CurrentMotionValue = self.Datas[self.StartIdx][0]
|
|
|
405 |
event.Skip()
|
|
|
406 |
|
|
|
407 |
def OnCanvasLeftUp(self, event):
|
|
|
408 |
self.Dragging = False
|
|
|
409 |
if self.Mode == MODE_MOTION:
|
|
|
410 |
self.CurrentMousePos = None
|
|
|
411 |
self.CurrentMotionValue = None
|
|
|
412 |
if self.Canvas.canvas.HasCapture():
|
|
|
413 |
self.Canvas.canvas.ReleaseMouse()
|
|
|
414 |
event.Skip()
|
|
|
415 |
|
|
|
416 |
def OnCanvasMiddleDown(self, event):
|
|
|
417 |
self.Fixed = True
|
|
|
418 |
self.Canvas.canvas.CaptureMouse()
|
|
|
419 |
if len(self.Datas) > 0:
|
|
|
420 |
self.GetBounds()
|
|
|
421 |
self.CurrentMousePos = event.GetPosition()
|
|
|
422 |
self.CurrentMotionValue = self.Datas[self.StartIdx][0]
|
|
|
423 |
event.Skip()
|
|
|
424 |
|
|
|
425 |
def OnCanvasMiddleUp(self, event):
|
|
|
426 |
self.CurrentMousePos = None
|
|
|
427 |
self.CurrentMotionValue = None
|
|
|
428 |
if self.Canvas.canvas.HasCapture():
|
|
|
429 |
self.Canvas.canvas.ReleaseMouse()
|
|
|
430 |
event.Skip()
|
|
|
431 |
|
|
|
432 |
def OnCanvasMotion(self, event):
|
|
|
433 |
if self.Mode == MODE_SELECTION and self.Dragging:
|
|
|
434 |
pos = self.Canvas.PositionScreenToUser(event.GetPosition())
|
|
|
435 |
graphics, xAxis, yAxis = self.Canvas.last_draw
|
|
|
436 |
self.CursorIdx = self.GetNearestData(max(xAxis[0], min(pos[0], xAxis[1])), -1)
|
|
|
437 |
self.RefreshCursor()
|
|
|
438 |
elif self.CurrentMousePos is not None and len(self.Datas) > 0:
|
|
|
439 |
oldpos = self.Canvas.PositionScreenToUser(self.CurrentMousePos)
|
|
|
440 |
newpos = self.Canvas.PositionScreenToUser(event.GetPosition())
|
|
|
441 |
self.CurrentMotionValue += oldpos[0] - newpos[0]
|
|
|
442 |
self.YCenter += oldpos[1] - newpos[1]
|
|
|
443 |
self.ResetBounds()
|
|
|
444 |
self.StartTick = max(self.Datas[0][0], min(self.CurrentMotionValue, self.Datas[-1][0] - self.CurrentRange))
|
|
|
445 |
self.CurrentMousePos = event.GetPosition()
|
|
|
446 |
self.NewDataAvailable(True)
|
|
|
447 |
event.Skip()
|
|
|
448 |
|
|
|
449 |
def OnCanvasMouseWheel(self, event):
|
|
|
450 |
if self.CurrentMousePos is None:
|
|
|
451 |
rotation = event.GetWheelRotation() / event.GetWheelDelta()
|
|
|
452 |
if event.ShiftDown():
|
|
|
453 |
current = self.CanvasRange.GetSelection()
|
|
|
454 |
new = max(0, min(current - rotation, len(self.RangeValues) - 1))
|
|
|
455 |
if new != current:
|
|
|
456 |
if self.Ticktime == 0:
|
|
|
457 |
self.CurrentRange = self.RangeValues[new][1]
|
|
|
458 |
else:
|
|
|
459 |
self.CurrentRange = self.RangeValues[new][1] / self.Ticktime
|
|
|
460 |
self.CanvasRange.SetStringSelection(self.RangeValues[new][0])
|
|
|
461 |
wx.CallAfter(self.RefreshRange)
|
|
|
462 |
else:
|
|
|
463 |
current = self.CanvasZoom.GetSelection()
|
|
|
464 |
new = max(0, min(current + rotation, len(ZOOM_VALUES) - 1))
|
|
|
465 |
if new != current:
|
|
|
466 |
self.CurrentZoom = ZOOM_VALUES[new][1]
|
|
|
467 |
self.CanvasZoom.SetStringSelection(ZOOM_VALUES[new][0])
|
|
|
468 |
wx.CallAfter(self.RefreshView, True)
|
|
|
469 |
event.Skip()
|
|
|
470 |
|
|
|
471 |
def OnCanvasResize(self, event):
|
|
|
472 |
self.ResetLastCursor()
|
|
|
473 |
wx.CallAfter(self.RefreshCursor)
|
|
|
474 |
event.Skip()
|
|
|
475 |
|
|
|
476 |
## Reset the last cursor
|
|
|
477 |
def ResetLastCursor(self):
|
|
|
478 |
self.LastCursor = None
|
|
|
479 |
|
|
|
480 |
## Draw the cursor on graphic
|
|
|
481 |
# @param dc The draw canvas
|
|
|
482 |
# @param cursor The cursor parameters
|
|
|
483 |
def DrawCursor(self, dc, cursor, value):
|
|
|
484 |
if self.StartTick <= cursor <= self.StartTick + self.CurrentRange:
|
|
|
485 |
# Prepare temporary dc for drawing
|
|
|
486 |
width = self.Canvas._Buffer.GetWidth()
|
|
|
487 |
height = self.Canvas._Buffer.GetHeight()
|
|
|
488 |
tmp_Buffer = wx.EmptyBitmap(width, height)
|
|
|
489 |
dcs = wx.MemoryDC()
|
|
|
490 |
dcs.SelectObject(tmp_Buffer)
|
|
|
491 |
dcs.Clear()
|
|
|
492 |
dcs.BeginDrawing()
|
|
|
493 |
|
|
|
494 |
dcs.SetPen(wx.Pen(wx.RED))
|
|
|
495 |
dcs.SetBrush(wx.Brush(wx.RED, wx.SOLID))
|
|
|
496 |
dcs.SetFont(self.Canvas._getFont(self.Canvas._fontSizeAxis))
|
|
|
497 |
|
|
|
498 |
# Calculate clipping region
|
|
|
499 |
graphics, xAxis, yAxis = self.Canvas.last_draw
|
|
|
500 |
p1 = numpy.array([xAxis[0], yAxis[0]])
|
|
|
501 |
p2 = numpy.array([xAxis[1], yAxis[1]])
|
|
|
502 |
cx, cy, cwidth, cheight = self.Canvas._point2ClientCoord(p1, p2)
|
|
|
503 |
|
|
|
504 |
px, py = self.Canvas.PositionUserToScreen((float(cursor), 0.))
|
|
|
505 |
|
|
|
506 |
# Draw line cross drawing for diaplaying time cursor
|
|
|
507 |
dcs.DrawLine(px, cy + 1, px, cy + cheight - 1)
|
|
|
508 |
|
|
|
509 |
lines = ("X:%d\nY:%f" % (cursor, value)).splitlines()
|
|
|
510 |
|
|
|
511 |
wtext = 0
|
|
|
512 |
for line in lines:
|
|
|
513 |
w, h = dcs.GetTextExtent(line)
|
|
|
514 |
wtext = max(wtext, w)
|
|
|
515 |
|
|
|
516 |
offset = 0
|
|
|
517 |
for line in lines:
|
|
|
518 |
# Draw time cursor date
|
|
|
519 |
dcs.DrawText(line, min(px + 3, cx + cwidth - wtext), cy + 3 + offset)
|
|
|
520 |
w, h = dcs.GetTextExtent(line)
|
|
|
521 |
offset += h
|
|
|
522 |
|
|
|
523 |
dcs.EndDrawing()
|
|
|
524 |
|
|
|
525 |
#this will erase if called twice
|
|
|
526 |
dc.Blit(0, 0, width, height, dcs, 0, 0, wx.EQUIV) #(NOT src) XOR dst
|
|
|
527 |
|
|
|
528 |
## Refresh the variable cursor.
|
|
|
529 |
# @param dc The draw canvas
|
|
|
530 |
def RefreshCursor(self, dc=None):
|
|
|
531 |
if self:
|
|
|
532 |
if dc is None:
|
|
|
533 |
dc = wx.BufferedDC(wx.ClientDC(self.Canvas.canvas), self.Canvas._Buffer)
|
|
|
534 |
|
|
|
535 |
# Erase previous time cursor if drawn
|
|
|
536 |
if self.LastCursor is not None:
|
|
|
537 |
self.DrawCursor(dc, *self.LastCursor)
|
|
|
538 |
|
|
|
539 |
# Draw new time cursor
|
|
|
540 |
if self.CursorIdx is not None:
|
|
|
541 |
self.LastCursor = self.Datas[self.CursorIdx]
|
|
|
542 |
self.DrawCursor(dc, *self.LastCursor)
|