Laurent@814: #!/usr/bin/env python Laurent@814: # -*- coding: utf-8 -*- Laurent@814: Laurent@814: #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor Laurent@814: #based on the plcopen standard. Laurent@814: # Laurent@814: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD Laurent@814: # Laurent@814: #See COPYING file for copyrights details. Laurent@814: # Laurent@814: #This library is free software; you can redistribute it and/or Laurent@814: #modify it under the terms of the GNU General Public Laurent@814: #License as published by the Free Software Foundation; either Laurent@814: #version 2.1 of the License, or (at your option) any later version. Laurent@814: # Laurent@814: #This library is distributed in the hope that it will be useful, Laurent@814: #but WITHOUT ANY WARRANTY; without even the implied warranty of Laurent@814: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Laurent@814: #General Public License for more details. Laurent@814: # Laurent@814: #You should have received a copy of the GNU General Public Laurent@814: #License along with this library; if not, write to the Free Software Laurent@814: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA Laurent@814: Laurent@814: import numpy Laurent@814: import math Laurent@814: Laurent@814: import wx Laurent@814: import wx.lib.plot as plot Laurent@814: import wx.lib.buttons Laurent@814: Laurent@814: from graphics.GraphicCommons import DebugViewer, MODE_SELECTION, MODE_MOTION Laurent@814: from EditorPanel import EditorPanel Laurent@814: from util.BitmapLibrary import GetBitmap Laurent@814: Laurent@814: colours = ['blue', 'red', 'green', 'yellow', 'orange', 'purple', 'brown', 'cyan', Laurent@814: 'pink', 'grey'] Laurent@814: markers = ['circle', 'dot', 'square', 'triangle', 'triangle_down', 'cross', 'plus', 'circle'] Laurent@814: Laurent@814: Laurent@814: #------------------------------------------------------------------------------- Laurent@814: # Debug Variable Graphic Viewer class Laurent@814: #------------------------------------------------------------------------------- Laurent@814: Laurent@814: SECOND = 1000000000 Laurent@814: MINUTE = 60 * SECOND Laurent@814: HOUR = 60 * MINUTE Laurent@814: Laurent@814: ZOOM_VALUES = map(lambda x:("x %.1f" % x, x), [math.sqrt(2) ** i for i in xrange(8)]) Laurent@814: RANGE_VALUES = map(lambda x: (str(x), x), [25 * 2 ** i for i in xrange(6)]) Laurent@814: TIME_RANGE_VALUES = [("%ds" % i, i * SECOND) for i in (1, 2, 5, 10, 20, 30)] + \ Laurent@814: [("%dm" % i, i * MINUTE) for i in (1, 2, 5, 10, 20, 30)] + \ Laurent@814: [("%dh" % i, i * HOUR) for i in (1, 2, 3, 6, 12, 24)] Laurent@814: Laurent@814: class GraphicViewer(EditorPanel, DebugViewer): Laurent@814: Laurent@814: def _init_Editor(self, prnt): Laurent@814: self.Editor = wx.Panel(prnt) Laurent@814: Laurent@814: main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0) Laurent@814: main_sizer.AddGrowableCol(0) Laurent@814: main_sizer.AddGrowableRow(0) Laurent@814: Laurent@814: self.Canvas = plot.PlotCanvas(self.Editor, name='Canvas') Laurent@814: def _axisInterval(spec, lower, upper): Laurent@814: if spec == 'border': Laurent@814: if lower == upper: Laurent@814: return lower - 0.5, upper + 0.5 Laurent@814: else: Laurent@814: border = (upper - lower) * 0.05 Laurent@814: return lower - border, upper + border Laurent@814: else: Laurent@814: return plot.PlotCanvas._axisInterval(self.Canvas, spec, lower, upper) Laurent@814: self.Canvas._axisInterval = _axisInterval Laurent@814: self.Canvas.SetYSpec('border') Laurent@814: self.Canvas.canvas.Bind(wx.EVT_LEFT_DOWN, self.OnCanvasLeftDown) Laurent@814: self.Canvas.canvas.Bind(wx.EVT_LEFT_UP, self.OnCanvasLeftUp) Laurent@814: self.Canvas.canvas.Bind(wx.EVT_MIDDLE_DOWN, self.OnCanvasMiddleDown) Laurent@814: self.Canvas.canvas.Bind(wx.EVT_MIDDLE_UP, self.OnCanvasMiddleUp) Laurent@814: self.Canvas.canvas.Bind(wx.EVT_MOTION, self.OnCanvasMotion) Laurent@814: self.Canvas.canvas.Bind(wx.EVT_SIZE, self.OnCanvasResize) Laurent@814: main_sizer.AddWindow(self.Canvas, 0, border=0, flag=wx.GROW) Laurent@814: Laurent@814: range_sizer = wx.FlexGridSizer(cols=10, hgap=5, rows=1, vgap=0) Laurent@814: range_sizer.AddGrowableCol(5) Laurent@814: range_sizer.AddGrowableRow(0) Laurent@814: main_sizer.AddSizer(range_sizer, 0, border=5, flag=wx.GROW|wx.ALL) Laurent@814: Laurent@814: range_label = wx.StaticText(self.Editor, label=_('Range:')) Laurent@814: range_sizer.AddWindow(range_label, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL) Laurent@814: Laurent@814: self.CanvasRange = wx.ComboBox(self.Editor, Laurent@814: size=wx.Size(100, 28), style=wx.CB_READONLY) Laurent@814: self.Bind(wx.EVT_COMBOBOX, self.OnRangeChanged, self.CanvasRange) Laurent@814: range_sizer.AddWindow(self.CanvasRange, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL) Laurent@814: Laurent@814: zoom_label = wx.StaticText(self.Editor, label=_('Zoom:')) Laurent@814: range_sizer.AddWindow(zoom_label, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL) Laurent@814: Laurent@814: self.CanvasZoom = wx.ComboBox(self.Editor, Laurent@814: size=wx.Size(70, 28), style=wx.CB_READONLY) Laurent@814: self.Bind(wx.EVT_COMBOBOX, self.OnZoomChanged, self.CanvasZoom) Laurent@814: range_sizer.AddWindow(self.CanvasZoom, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL) Laurent@814: Laurent@814: position_label = wx.StaticText(self.Editor, label=_('Position:')) Laurent@814: range_sizer.AddWindow(position_label, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL) Laurent@814: Laurent@814: self.CanvasPosition = wx.ScrollBar(self.Editor, Laurent@814: size=wx.Size(0, 16), style=wx.SB_HORIZONTAL) Laurent@814: self.CanvasPosition.SetScrollbar(0, 10, 100, 10) Laurent@814: self.CanvasPosition.Bind(wx.EVT_SCROLL_THUMBTRACK, Laurent@814: self.OnPositionChanging, self.CanvasPosition) Laurent@814: self.CanvasPosition.Bind(wx.EVT_SCROLL_LINEUP, Laurent@814: self.OnPositionChanging, self.CanvasPosition) Laurent@814: self.CanvasPosition.Bind(wx.EVT_SCROLL_LINEDOWN, Laurent@814: self.OnPositionChanging, self.CanvasPosition) Laurent@814: self.CanvasPosition.Bind(wx.EVT_SCROLL_PAGEUP, Laurent@814: self.OnPositionChanging, self.CanvasPosition) Laurent@814: self.CanvasPosition.Bind(wx.EVT_SCROLL_PAGEDOWN, Laurent@814: self.OnPositionChanging, self.CanvasPosition) Laurent@814: range_sizer.AddWindow(self.CanvasPosition, 0, border=5, flag=wx.GROW|wx.ALL) Laurent@814: Laurent@814: self.ResetButton = wx.lib.buttons.GenBitmapButton(self.Editor, Laurent@814: bitmap=GetBitmap("reset"), size=wx.Size(28, 28), style=wx.NO_BORDER) Laurent@814: self.ResetButton.SetToolTipString(_("Clear the graph values")) Laurent@814: self.Bind(wx.EVT_BUTTON, self.OnResetButton, self.ResetButton) Laurent@814: range_sizer.AddWindow(self.ResetButton, 0, border=0, flag=0) Laurent@814: Laurent@814: self.CurrentButton = wx.lib.buttons.GenBitmapButton(self.Editor, Laurent@814: bitmap=GetBitmap("current"), size=wx.Size(28, 28), style=wx.NO_BORDER) Laurent@814: self.CurrentButton.SetToolTipString(_("Go to current value")) Laurent@814: self.Bind(wx.EVT_BUTTON, self.OnCurrentButton, self.CurrentButton) Laurent@814: range_sizer.AddWindow(self.CurrentButton, 0, border=0, flag=0) Laurent@814: Laurent@814: self.ResetZoomOffsetButton = wx.lib.buttons.GenBitmapButton(self.Editor, Laurent@814: bitmap=GetBitmap("fit"), size=wx.Size(28, 28), style=wx.NO_BORDER) Laurent@814: self.ResetZoomOffsetButton.SetToolTipString(_("Reset zoom and offset")) Laurent@814: self.Bind(wx.EVT_BUTTON, self.OnResetZoomOffsetButton, Laurent@814: self.ResetZoomOffsetButton) Laurent@814: range_sizer.AddWindow(self.ResetZoomOffsetButton, 0, border=0, flag=0) Laurent@814: Laurent@814: self.ExportGraphButton = wx.lib.buttons.GenBitmapButton(self.Editor, Laurent@814: bitmap=GetBitmap("export_graph"), size=wx.Size(28, 28), style=wx.NO_BORDER) Laurent@814: self.ExportGraphButton.SetToolTipString(_("Export graph values to clipboard")) Laurent@814: self.Bind(wx.EVT_BUTTON, self.OnExportGraphButtonClick, Laurent@814: self.ExportGraphButton) Laurent@814: range_sizer.AddWindow(self.ExportGraphButton, 0, border=0, flag=0) Laurent@814: Laurent@814: self.Editor.SetSizer(main_sizer) Laurent@814: Laurent@814: self.Editor.Bind(wx.EVT_MOUSEWHEEL, self.OnCanvasMouseWheel) Laurent@814: Laurent@814: def __init__(self, parent, window, producer, instancepath = ""): Laurent@814: EditorPanel.__init__(self, parent, "", window, None) Laurent@814: DebugViewer.__init__(self, producer, True, False) Laurent@814: Laurent@814: self.InstancePath = instancepath Laurent@814: self.RangeValues = None Laurent@814: self.CursorIdx = None Laurent@814: self.LastCursor = None Laurent@814: self.CurrentMousePos = None Laurent@814: self.CurrentMotionValue = None Laurent@814: self.Dragging = False Laurent@814: Laurent@814: # Initialize Viewer mode to Selection mode Laurent@814: self.Mode = MODE_SELECTION Laurent@814: Laurent@887: self.Data = numpy.array([]).reshape(0, 2) Laurent@814: self.StartTick = 0 Laurent@814: self.StartIdx = 0 Laurent@814: self.EndIdx = 0 Laurent@814: self.MinValue = None Laurent@814: self.MaxValue = None Laurent@814: self.YCenter = 0 Laurent@814: self.CurrentZoom = 1.0 Laurent@814: self.Fixed = False Laurent@814: self.Ticktime = self.DataProducer.GetTicktime() Laurent@814: self.RefreshCanvasRange() Laurent@814: Laurent@814: for zoom_txt, zoom in ZOOM_VALUES: Laurent@814: self.CanvasZoom.Append(zoom_txt) Laurent@814: self.CanvasZoom.SetSelection(0) Laurent@814: Laurent@814: self.AddDataConsumer(self.InstancePath.upper(), self) Laurent@814: Laurent@814: def __del__(self): Laurent@814: DebugViewer.__del__(self) Laurent@814: self.RemoveDataConsumer(self) Laurent@814: Laurent@814: def GetTitle(self): Laurent@814: if len(self.InstancePath) > 15: Laurent@814: return "..." + self.InstancePath[-12:] Laurent@814: return self.InstancePath Laurent@814: Laurent@814: # Changes Viewer mode Laurent@814: def SetMode(self, mode): Laurent@814: if self.Mode != mode or mode == MODE_SELECTION: Laurent@814: if self.Mode == MODE_MOTION: Laurent@814: wx.CallAfter(self.Canvas.canvas.SetCursor, wx.NullCursor) Laurent@814: self.Mode = mode Laurent@814: if self.Mode == MODE_MOTION: Laurent@814: wx.CallAfter(self.Canvas.canvas.SetCursor, wx.StockCursor(wx.CURSOR_HAND)) Laurent@814: Laurent@814: def ResetView(self, register=False): Laurent@887: self.Data = numpy.array([]).reshape(0, 2) Laurent@814: self.StartTick = 0 Laurent@814: self.StartIdx = 0 Laurent@814: self.EndIdx = 0 Laurent@814: self.MinValue = None Laurent@814: self.MaxValue = None Laurent@814: self.CursorIdx = None Laurent@814: self.Fixed = False Laurent@814: self.Ticktime = self.DataProducer.GetTicktime() Laurent@814: if register: Laurent@814: self.AddDataConsumer(self.InstancePath.upper(), self) Laurent@814: self.ResetLastCursor() Laurent@814: self.RefreshCanvasRange() Laurent@814: self.RefreshView() Laurent@814: Laurent@814: def RefreshNewData(self, *args, **kwargs): Laurent@814: self.RefreshView(*args, **kwargs) Laurent@814: DebugViewer.RefreshNewData(self) Laurent@814: Laurent@814: def GetNearestData(self, tick, adjust): Laurent@887: ticks = self.Data[:, 0] Laurent@814: new_cursor = numpy.argmin(abs(ticks - tick)) Laurent@814: if adjust == -1 and ticks[new_cursor] > tick and new_cursor > 0: Laurent@814: new_cursor -= 1 Laurent@887: elif adjust == 1 and ticks[new_cursor] < tick and new_cursor < len(ticks): Laurent@814: new_cursor += 1 Laurent@814: return new_cursor Laurent@814: Laurent@814: def GetBounds(self): Laurent@814: if self.StartIdx is None or self.EndIdx is None: Laurent@814: self.StartIdx = self.GetNearestData(self.StartTick, -1) Laurent@814: self.EndIdx = self.GetNearestData(self.StartTick + self.CurrentRange, 1) Laurent@814: Laurent@814: def ResetBounds(self): Laurent@814: self.StartIdx = None Laurent@814: self.EndIdx = None Laurent@814: Laurent@814: def RefreshCanvasRange(self): Laurent@814: if self.Ticktime == 0 and self.RangeValues != RANGE_VALUES: Laurent@814: self.RangeValues = RANGE_VALUES Laurent@814: self.CanvasRange.Clear() Laurent@814: for text, value in RANGE_VALUES: Laurent@814: self.CanvasRange.Append(text) Laurent@814: self.CanvasRange.SetStringSelection(RANGE_VALUES[0][0]) Laurent@814: self.CurrentRange = RANGE_VALUES[0][1] Laurent@814: elif self.RangeValues != TIME_RANGE_VALUES: Laurent@814: self.RangeValues = TIME_RANGE_VALUES Laurent@814: self.CanvasRange.Clear() Laurent@814: for text, value in TIME_RANGE_VALUES: Laurent@814: self.CanvasRange.Append(text) Laurent@814: self.CanvasRange.SetStringSelection(TIME_RANGE_VALUES[0][0]) Laurent@814: self.CurrentRange = TIME_RANGE_VALUES[0][1] / self.Ticktime Laurent@814: Laurent@814: def RefreshView(self, force=False): Laurent@814: self.Freeze() Laurent@887: if force or not self.Fixed or (len(self.Data) > 0 and self.StartTick + self.CurrentRange > self.Data[-1, 0]): Laurent@814: if (self.MinValue is not None and Laurent@814: self.MaxValue is not None and Laurent@814: self.MinValue != self.MaxValue): Laurent@814: Yrange = float(self.MaxValue - self.MinValue) / self.CurrentZoom Laurent@814: else: Laurent@814: Yrange = 2. / self.CurrentZoom Laurent@814: Laurent@887: if not force and not self.Fixed and len(self.Data) > 0: Laurent@887: self.YCenter = max(self.Data[-1, 1] - Yrange / 2, Laurent@814: min(self.YCenter, Laurent@887: self.Data[-1, 1] + Yrange / 2)) Laurent@814: Laurent@814: var_name = self.InstancePath.split(".")[-1] Laurent@814: Laurent@814: self.GetBounds() Laurent@887: self.VariableGraphic = plot.PolyLine(self.Data[self.StartIdx:self.EndIdx + 1], Laurent@814: legend=var_name, colour=colours[0]) Laurent@814: self.GraphicsObject = plot.PlotGraphics([self.VariableGraphic], _("%s Graphics") % var_name, _("Tick"), _("Values")) Laurent@814: self.Canvas.Draw(self.GraphicsObject, Laurent@814: xAxis=(self.StartTick, self.StartTick + self.CurrentRange), Laurent@814: yAxis=(self.YCenter - Yrange * 1.1 / 2., self.YCenter + Yrange * 1.1 / 2.)) Laurent@814: Laurent@814: # Reset and draw cursor Laurent@814: self.ResetLastCursor() Laurent@814: self.RefreshCursor() Laurent@814: Laurent@814: self.RefreshScrollBar() Laurent@814: Laurent@814: self.Thaw() Laurent@814: Laurent@814: def GetInstancePath(self): Laurent@814: return self.InstancePath Laurent@814: Laurent@814: def IsViewing(self, tagname): Laurent@814: return self.InstancePath == tagname Laurent@814: Laurent@814: def NewValue(self, tick, value, forced=False): Laurent@814: value = {True:1., False:0.}.get(value, float(value)) Laurent@887: self.Data = numpy.append(self.Data, [[float(tick), value]], axis=0) Laurent@814: if self.MinValue is None: Laurent@814: self.MinValue = value Laurent@814: else: Laurent@814: self.MinValue = min(self.MinValue, value) Laurent@814: if self.MaxValue is None: Laurent@814: self.MaxValue = value Laurent@814: else: Laurent@814: self.MaxValue = max(self.MaxValue, value) Laurent@814: if not self.Fixed or tick < self.StartTick + self.CurrentRange: Laurent@814: self.GetBounds() Laurent@887: while int(self.Data[self.StartIdx, 0]) < tick - self.CurrentRange: Laurent@814: self.StartIdx += 1 Laurent@814: self.EndIdx += 1 Laurent@887: self.StartTick = self.Data[self.StartIdx, 0] Laurent@814: self.NewDataAvailable() Laurent@814: Laurent@814: def RefreshScrollBar(self): Laurent@887: if len(self.Data) > 0: Laurent@814: self.GetBounds() Laurent@887: pos = int(self.Data[self.StartIdx, 0] - self.Data[0, 0]) Laurent@887: range = int(self.Data[-1, 0] - self.Data[0, 0]) Laurent@814: else: Laurent@814: pos = 0 Laurent@814: range = 0 Laurent@814: self.CanvasPosition.SetScrollbar(pos, self.CurrentRange, range, self.CurrentRange) Laurent@814: Laurent@814: def RefreshRange(self): Laurent@887: if len(self.Data) > 0: Laurent@887: if self.Fixed and self.Data[-1, 0] - self.Data[0, 0] < self.CurrentRange: Laurent@814: self.Fixed = False Laurent@814: self.ResetBounds() Laurent@814: if self.Fixed: Laurent@887: self.StartTick = min(self.StartTick, self.Data[-1, 0] - self.CurrentRange) Laurent@814: else: Laurent@887: self.StartTick = max(self.Data[0, 0], self.Data[-1, 0] - self.CurrentRange) Laurent@814: self.RefreshView(True) Laurent@814: Laurent@814: def OnRangeChanged(self, event): Laurent@814: try: Laurent@814: if self.Ticktime == 0: Laurent@814: self.CurrentRange = self.RangeValues[self.CanvasRange.GetSelection()][1] Laurent@814: else: Laurent@814: self.CurrentRange = self.RangeValues[self.CanvasRange.GetSelection()][1] / self.Ticktime Laurent@814: except ValueError, e: Laurent@814: self.CanvasRange.SetValue(str(self.CurrentRange)) Laurent@814: wx.CallAfter(self.RefreshRange) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnZoomChanged(self, event): Laurent@814: self.CurrentZoom = ZOOM_VALUES[self.CanvasZoom.GetSelection()][1] Laurent@814: wx.CallAfter(self.RefreshView, True) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnPositionChanging(self, event): Laurent@887: if len(self.Data) > 0: Laurent@814: self.ResetBounds() Laurent@887: self.StartTick = self.Data[0, 0] + event.GetPosition() Laurent@814: self.Fixed = True Laurent@814: self.NewDataAvailable(True) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnResetButton(self, event): Laurent@814: self.Fixed = False Laurent@814: self.ResetView() Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCurrentButton(self, event): Laurent@887: if len(self.Data) > 0: Laurent@814: self.ResetBounds() Laurent@887: self.StartTick = max(self.Data[0, 0], self.Data[-1, 0] - self.CurrentRange) Laurent@814: self.Fixed = False Laurent@814: self.NewDataAvailable(True) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnResetZoomOffsetButton(self, event): Laurent@887: if len(self.Data) > 0: Laurent@814: self.YCenter = (self.MaxValue + self.MinValue) / 2 Laurent@814: else: Laurent@814: self.YCenter = 0.0 Laurent@814: self.CurrentZoom = 1.0 Laurent@814: self.CanvasZoom.SetSelection(0) Laurent@814: wx.CallAfter(self.RefreshView, True) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnExportGraphButtonClick(self, event): Laurent@887: data_copy = self.Data[:] Laurent@814: text = "tick;%s;\n" % self.InstancePath Laurent@814: for tick, value in data_copy: Laurent@814: text += "%d;%.3f;\n" % (tick, value) Laurent@814: self.ParentWindow.SetCopyBuffer(text) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasLeftDown(self, event): Laurent@814: self.Fixed = True Laurent@814: self.Canvas.canvas.CaptureMouse() Laurent@887: if len(self.Data) > 0: Laurent@814: if self.Mode == MODE_SELECTION: Laurent@814: self.Dragging = True Laurent@814: pos = self.Canvas.PositionScreenToUser(event.GetPosition()) Laurent@814: self.CursorIdx = self.GetNearestData(pos[0], -1) Laurent@814: self.RefreshCursor() Laurent@814: elif self.Mode == MODE_MOTION: Laurent@814: self.GetBounds() Laurent@814: self.CurrentMousePos = event.GetPosition() Laurent@887: self.CurrentMotionValue = self.Data[self.StartIdx, 0] Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasLeftUp(self, event): Laurent@814: self.Dragging = False Laurent@814: if self.Mode == MODE_MOTION: Laurent@814: self.CurrentMousePos = None Laurent@814: self.CurrentMotionValue = None Laurent@814: if self.Canvas.canvas.HasCapture(): Laurent@814: self.Canvas.canvas.ReleaseMouse() Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasMiddleDown(self, event): Laurent@814: self.Fixed = True Laurent@814: self.Canvas.canvas.CaptureMouse() Laurent@887: if len(self.Data) > 0: Laurent@814: self.GetBounds() Laurent@814: self.CurrentMousePos = event.GetPosition() Laurent@887: self.CurrentMotionValue = self.Data[self.StartIdx, 0] Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasMiddleUp(self, event): Laurent@814: self.CurrentMousePos = None Laurent@814: self.CurrentMotionValue = None Laurent@814: if self.Canvas.canvas.HasCapture(): Laurent@814: self.Canvas.canvas.ReleaseMouse() Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasMotion(self, event): Laurent@814: if self.Mode == MODE_SELECTION and self.Dragging: Laurent@814: pos = self.Canvas.PositionScreenToUser(event.GetPosition()) Laurent@814: graphics, xAxis, yAxis = self.Canvas.last_draw Laurent@814: self.CursorIdx = self.GetNearestData(max(xAxis[0], min(pos[0], xAxis[1])), -1) Laurent@814: self.RefreshCursor() Laurent@887: elif self.CurrentMousePos is not None and len(self.Data) > 0: Laurent@814: oldpos = self.Canvas.PositionScreenToUser(self.CurrentMousePos) Laurent@814: newpos = self.Canvas.PositionScreenToUser(event.GetPosition()) Laurent@814: self.CurrentMotionValue += oldpos[0] - newpos[0] Laurent@814: self.YCenter += oldpos[1] - newpos[1] Laurent@814: self.ResetBounds() Laurent@887: self.StartTick = max(self.Data[0, 0], min(self.CurrentMotionValue, self.Data[-1, 0] - self.CurrentRange)) Laurent@814: self.CurrentMousePos = event.GetPosition() Laurent@814: self.NewDataAvailable(True) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasMouseWheel(self, event): Laurent@814: if self.CurrentMousePos is None: Laurent@814: rotation = event.GetWheelRotation() / event.GetWheelDelta() Laurent@814: if event.ShiftDown(): Laurent@814: current = self.CanvasRange.GetSelection() Laurent@814: new = max(0, min(current - rotation, len(self.RangeValues) - 1)) Laurent@814: if new != current: Laurent@814: if self.Ticktime == 0: Laurent@814: self.CurrentRange = self.RangeValues[new][1] Laurent@814: else: Laurent@814: self.CurrentRange = self.RangeValues[new][1] / self.Ticktime Laurent@814: self.CanvasRange.SetStringSelection(self.RangeValues[new][0]) Laurent@814: wx.CallAfter(self.RefreshRange) Laurent@814: else: Laurent@814: current = self.CanvasZoom.GetSelection() Laurent@814: new = max(0, min(current + rotation, len(ZOOM_VALUES) - 1)) Laurent@814: if new != current: Laurent@814: self.CurrentZoom = ZOOM_VALUES[new][1] Laurent@814: self.CanvasZoom.SetStringSelection(ZOOM_VALUES[new][0]) Laurent@814: wx.CallAfter(self.RefreshView, True) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnCanvasResize(self, event): Laurent@814: self.ResetLastCursor() Laurent@814: wx.CallAfter(self.RefreshCursor) Laurent@814: event.Skip() Laurent@814: Laurent@814: ## Reset the last cursor Laurent@814: def ResetLastCursor(self): Laurent@814: self.LastCursor = None Laurent@814: Laurent@814: ## Draw the cursor on graphic Laurent@814: # @param dc The draw canvas Laurent@814: # @param cursor The cursor parameters Laurent@814: def DrawCursor(self, dc, cursor, value): Laurent@814: if self.StartTick <= cursor <= self.StartTick + self.CurrentRange: Laurent@814: # Prepare temporary dc for drawing Laurent@814: width = self.Canvas._Buffer.GetWidth() Laurent@814: height = self.Canvas._Buffer.GetHeight() Laurent@814: tmp_Buffer = wx.EmptyBitmap(width, height) Laurent@814: dcs = wx.MemoryDC() Laurent@814: dcs.SelectObject(tmp_Buffer) Laurent@814: dcs.Clear() Laurent@814: dcs.BeginDrawing() Laurent@814: Laurent@814: dcs.SetPen(wx.Pen(wx.RED)) Laurent@814: dcs.SetBrush(wx.Brush(wx.RED, wx.SOLID)) Laurent@814: dcs.SetFont(self.Canvas._getFont(self.Canvas._fontSizeAxis)) Laurent@814: Laurent@814: # Calculate clipping region Laurent@814: graphics, xAxis, yAxis = self.Canvas.last_draw Laurent@814: p1 = numpy.array([xAxis[0], yAxis[0]]) Laurent@814: p2 = numpy.array([xAxis[1], yAxis[1]]) Laurent@814: cx, cy, cwidth, cheight = self.Canvas._point2ClientCoord(p1, p2) Laurent@814: Laurent@814: px, py = self.Canvas.PositionUserToScreen((float(cursor), 0.)) Laurent@814: Laurent@814: # Draw line cross drawing for diaplaying time cursor Laurent@814: dcs.DrawLine(px, cy + 1, px, cy + cheight - 1) Laurent@814: Laurent@814: lines = ("X:%d\nY:%f" % (cursor, value)).splitlines() Laurent@814: Laurent@814: wtext = 0 Laurent@814: for line in lines: Laurent@814: w, h = dcs.GetTextExtent(line) Laurent@814: wtext = max(wtext, w) Laurent@814: Laurent@814: offset = 0 Laurent@814: for line in lines: Laurent@814: # Draw time cursor date Laurent@814: dcs.DrawText(line, min(px + 3, cx + cwidth - wtext), cy + 3 + offset) Laurent@814: w, h = dcs.GetTextExtent(line) Laurent@814: offset += h Laurent@814: Laurent@814: dcs.EndDrawing() Laurent@814: Laurent@814: #this will erase if called twice Laurent@814: dc.Blit(0, 0, width, height, dcs, 0, 0, wx.EQUIV) #(NOT src) XOR dst Laurent@814: Laurent@814: ## Refresh the variable cursor. Laurent@814: # @param dc The draw canvas Laurent@814: def RefreshCursor(self, dc=None): Laurent@814: if self: Laurent@814: if dc is None: Laurent@814: dc = wx.BufferedDC(wx.ClientDC(self.Canvas.canvas), self.Canvas._Buffer) Laurent@814: Laurent@814: # Erase previous time cursor if drawn Laurent@814: if self.LastCursor is not None: Laurent@814: self.DrawCursor(dc, *self.LastCursor) Laurent@814: Laurent@814: # Draw new time cursor Laurent@814: if self.CursorIdx is not None: Laurent@887: self.LastCursor = self.Data[self.CursorIdx] Laurent@814: self.DrawCursor(dc, *self.LastCursor)