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