841919160f
the instances of FlatCAMTool, in any position within the application menu's. install method of FlatCAMTool accept now the following kwargs: 'pos' = it is the menu where we want the Action installed. By default, if no 'pos' arg is provided, the tool will be install in the menutool menu (as before) 'before' = it is the position within the previously selected menu where, before it, we will install our new Action (menu entry) generated by the FlatCAMTool install method. By default, if no 'before' is provided, the action will be installed in the current last position in the menu.
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
from PyQt4 import QtGui
|
|
from FlatCAMTool import FlatCAMTool
|
|
from copy import copy
|
|
from math import sqrt
|
|
|
|
|
|
class Measurement(FlatCAMTool):
|
|
|
|
toolName = "Measurement Tool"
|
|
|
|
def __init__(self, app):
|
|
FlatCAMTool.__init__(self, app)
|
|
|
|
# self.setContentsMargins(0, 0, 0, 0)
|
|
self.layout.setMargin(0)
|
|
self.layout.setContentsMargins(0, 0, 3, 0)
|
|
|
|
self.setSizePolicy(QtGui.QSizePolicy.Ignored, QtGui.QSizePolicy.Maximum)
|
|
|
|
self.point1 = None
|
|
self.point2 = None
|
|
self.label = QtGui.QLabel("Click on a reference point ...")
|
|
self.label.setFrameStyle(QtGui.QFrame.StyledPanel | QtGui.QFrame.Plain)
|
|
self.label.setMargin(3)
|
|
self.layout.addWidget(self.label)
|
|
# self.layout.setMargin(0)
|
|
self.setVisible(False)
|
|
|
|
self.click_subscription = None
|
|
self.move_subscription = None
|
|
|
|
def install(self, icon=None, separator=None, **kwargs):
|
|
FlatCAMTool.install(self, icon, separator, **kwargs)
|
|
self.app.ui.right_layout.addWidget(self)
|
|
self.app.plotcanvas.mpl_connect('key_press_event', self.on_key_press)
|
|
|
|
def run(self):
|
|
self.toggle()
|
|
|
|
def on_click(self, event):
|
|
if self.point1 is None:
|
|
self.point1 = (event.xdata, event.ydata)
|
|
else:
|
|
self.point2 = copy(self.point1)
|
|
self.point1 = (event.xdata, event.ydata)
|
|
self.on_move(event)
|
|
|
|
def on_key_press(self, event):
|
|
if event.key == 'r':
|
|
self.toggle()
|
|
|
|
def toggle(self):
|
|
if self.isVisible():
|
|
self.setVisible(False)
|
|
self.app.plotcanvas.mpl_disconnect(self.move_subscription)
|
|
self.app.plotcanvas.mpl_disconnect(self.click_subscription)
|
|
else:
|
|
self.setVisible(True)
|
|
self.move_subscription = self.app.plotcanvas.mpl_connect('motion_notify_event', self.on_move)
|
|
self.click_subscription = self.app.plotcanvas.mpl_connect('button_press_event', self.on_click)
|
|
|
|
def on_move(self, event):
|
|
if self.point1 is None:
|
|
self.label.setText("Click on a reference point...")
|
|
else:
|
|
try:
|
|
dx = event.xdata - self.point1[0]
|
|
dy = event.ydata - self.point1[1]
|
|
d = sqrt(dx**2 + dy**2)
|
|
self.label.setText("D = %.4f D(x) = %.4f D(y) = %.4f" % (d, dx, dy))
|
|
except TypeError:
|
|
pass
|
|
if self.update is not None:
|
|
self.update()
|