211 lines
8.6 KiB
Python
211 lines
8.6 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# QTVcp Widget
|
|
# Copyright (c) 2019 Chris Morley
|
|
#
|
|
# This program 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 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program 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.
|
|
###############################################################################
|
|
|
|
from qtpy.QtWidgets import QToolButton, QMenu, QAction
|
|
from qtpy.QtCore import Property
|
|
from qtpy.QtGui import QIcon
|
|
|
|
from qtvcp.widgets.widget_baseclass import _HalWidgetBase
|
|
from qtvcp.core import Status, Action, Info
|
|
from qtvcp import logger
|
|
|
|
# Instiniate the libraries with global reference
|
|
# STATUS gives us status messages from LinuxCNC
|
|
# AUX_PRGM holds helper program loader
|
|
# INI holds INI details
|
|
# ACTION gives commands to LinuxCNC
|
|
# LOG is for running code logging
|
|
STATUS = Status()
|
|
INFO = Info()
|
|
ACTION = Action()
|
|
LOG = logger.getLogger(__name__)
|
|
# Force the log level for this module
|
|
#LOG.setLevel(logger.DEBUG) # One of DEBUG, INFO, WARNING, ERROR, CRITICAL
|
|
|
|
class OffsetToolButton(QToolButton, _HalWidgetBase):
|
|
def __init__(self, parent=None):
|
|
super(OffsetToolButton, self).__init__(parent)
|
|
self._axis = ''
|
|
self._last = 0
|
|
self._block_signal = False
|
|
self.dialog_code = 'CALCULATOR'
|
|
self.display_units_mm = 0
|
|
|
|
self.settingMenu = QMenu(self)
|
|
self.setMenu(self.settingMenu)
|
|
|
|
Button = QAction(QIcon('exit24.png'), 'Set Current Tool Position', self)
|
|
Button.triggered.connect(self.setOffset)
|
|
self.settingMenu.addAction(Button)
|
|
|
|
Button = QAction(QIcon('exit24.png'), 'Adjust Current Tool Position', self)
|
|
Button.triggered.connect(self.adjustOffset)
|
|
self.settingMenu.addAction(Button)
|
|
|
|
Button = QAction(QIcon('exit24.png'), 'Zero Current Tool Position', self)
|
|
Button.triggered.connect(self.zeroOffset)
|
|
self.settingMenu.addAction(Button)
|
|
|
|
Button = QAction(QIcon('exit24.png'), 'Set Tool Offset Directly', self)
|
|
Button.triggered.connect(self.setDirectOffset)
|
|
self.settingMenu.addAction(Button)
|
|
|
|
Button = QAction(QIcon('exit24.png'), 'Reset To Last', self)
|
|
Button.triggered.connect(self.last)
|
|
self.settingMenu.addAction(Button)
|
|
|
|
def _hal_init(self):
|
|
def homed_on_test():
|
|
return (STATUS.machine_is_on()
|
|
and (STATUS.get_current_tool() not in(0,-1))
|
|
and (STATUS.is_all_homed()
|
|
or INFO.NO_HOME_REQUIRED))
|
|
STATUS.connect('metric-mode-changed', self._switch_units)
|
|
STATUS.connect('state-off', lambda w: self.settingMenu.setEnabled(False))
|
|
STATUS.connect('state-estop', lambda w: self.settingMenu.setEnabled(False))
|
|
STATUS.connect('interp-idle', lambda w: self.settingMenu.setEnabled(homed_on_test()))
|
|
STATUS.connect('interp-run', lambda w: self.settingMenu.setEnabled(False))
|
|
STATUS.connect('all-homed', lambda w: self.settingMenu.setEnabled(homed_on_test()))
|
|
STATUS.connect('not-all-homed', lambda w, data: self.settingMenu.setEnabled(False))
|
|
STATUS.connect('interp-paused', lambda w: self.settingMenu.setEnabled(homed_on_test()))
|
|
STATUS.connect('tool-in-spindle-changed', lambda w,t:self.settingMenu.setEnabled(homed_on_test()))
|
|
STATUS.connect('general',self.return_value)
|
|
|
|
# process the STATUS return message
|
|
def return_value(self, w, message):
|
|
num = message['RETURN']
|
|
code = bool(message.get('ID') == '%s__'% self.objectName())
|
|
name = bool(message.get('NAME') == self.dialog_code)
|
|
if code and name and num is not None:
|
|
LOG.debug('message return:{}'.format (message))
|
|
self._last = message.get('CURRENT')
|
|
axis = message.get('AXIS')
|
|
fixture = message.get('FIXTURE') or False
|
|
direct = message.get('DIRECT') or False
|
|
adjust = message.get('ADJUST') or False
|
|
if direct:
|
|
if adjust:
|
|
num = self._last - num
|
|
ACTION.SET_DIRECT_TOOL_OFFSET(axis,num)
|
|
STATUS.emit('update-machine-log', 'Set Direct tool offset of Axis {}: {} to {}'.format(axis, self._last, num), 'TIME,SUCCESS')
|
|
else:
|
|
ACTION.SET_TOOL_OFFSET(axis,num,fixture)
|
|
STATUS.emit('update-machine-log', 'Adjust Tool offset of Axis {}, {} to {}'.format(axis, self._last, num), 'TIME,SUCCESS')
|
|
|
|
def zeroOffset(self):
|
|
axis, tool, now = self._get_current_info(self._axis)
|
|
if axis:
|
|
self._last = now
|
|
fixture = False
|
|
ACTION.SET_TOOL_OFFSET(axis, 0, fixture)
|
|
STATUS.emit('update-machine-log', 'Zeroed Tool Offset {} of Tool {}'.format(axis, tool), 'TIME,SUCCESS')
|
|
|
|
def setOffset(self):
|
|
axis, tool, now = self._get_current_info(self._axis)
|
|
if axis:
|
|
mess = {'NAME':self.dialog_code,'ID':'%s__' % self.objectName(),
|
|
'AXIS':axis,'CURRENT':now,
|
|
'TITLE':'Set Axis {} Current Position by setting Offset of Tool {}'.format(axis, tool)}
|
|
STATUS.emit('dialog-request', mess)
|
|
LOG.debug('message sent:{}'.format (mess))
|
|
|
|
def adjustOffset(self):
|
|
axis, tool, now = self._get_current_info(self._axis)
|
|
if axis:
|
|
mess = {'NAME':self.dialog_code,'ID':'%s__' % self.objectName(),
|
|
'AXIS':axis,'CURRENT':now,'ADJUST':True, 'DIRECT':True,
|
|
'TITLE':'Adjust Axis {} Current Position by Offsetting Tool {} by Entered Amount'.format(axis, tool, now)}
|
|
STATUS.emit('dialog-request', mess)
|
|
LOG.debug('message sent:{}'.format (mess))
|
|
|
|
def setDirectOffset(self):
|
|
axis, tool, now = self._get_current_info(self._axis)
|
|
if axis:
|
|
mess = {'NAME':self.dialog_code,'ID':'%s__' % self.objectName(),
|
|
'AXIS':axis,'CURRENT':now,'DIRECT':True,
|
|
'TITLE':'Set Axis {} Offset of Tool {}'.format(axis, tool)}
|
|
STATUS.emit('dialog-request', mess)
|
|
LOG.debug('message sent:{}'.format (mess))
|
|
|
|
def last(self):
|
|
axis, tool, now = self._get_current_info(self._axis)
|
|
if axis:
|
|
fixture = False
|
|
ACTION.SET_DIRECT_TOOL_OFFSET(axis, self._last)
|
|
STATUS.emit('update-machine-log', 'Reset Tool {} Offset To Last {}'.format(tool, self._last), 'TIME,SUCCESS')
|
|
self._last = now
|
|
|
|
def _get_current_info(self, axis):
|
|
tool = STATUS.get_current_tool()
|
|
if STATUS.is_joint_mode() or not tool:
|
|
return None, None, None
|
|
if axis == '':
|
|
axis = STATUS.get_selected_axis()
|
|
j = "XYZABCUVW"
|
|
try:
|
|
jnum = j.find(axis)
|
|
except:
|
|
return None, None, None
|
|
offset = STATUS.stat.tool_offset[jnum]
|
|
return axis, tool, offset
|
|
|
|
def _switch_units(self, widget, data):
|
|
self.display_units_mm = data
|
|
if self.display_units_mm != INFO.MACHINE_IS_METRIC:
|
|
self._last = INFO.convert_units(self._last)
|
|
|
|
#########################################################################
|
|
# This is how designer can interact with our widget properties.
|
|
# designer will show the Property properties in the editor
|
|
# it will use the get set and reset calls to do those actions
|
|
#
|
|
########################################################################
|
|
|
|
def set_axis(self, data):
|
|
if data.upper() in('X','Y','Z','A','B','C','U','V','W'):
|
|
self._axis = str(data.upper())
|
|
def get_axis(self):
|
|
return self._axis
|
|
def reset_axis(self):
|
|
self._axis = ''
|
|
axis_letter = Property(str, get_axis, set_axis, reset_axis)
|
|
|
|
|
|
def set_dialog_code(self, data):
|
|
self.dialog_code = data
|
|
def get_dialog_code(self):
|
|
return self.dialog_code
|
|
def reset_dialog_code(self):
|
|
self.dialog_code = 'CALCULATOR'
|
|
dialog_code_string = Property(str, get_dialog_code, set_dialog_code, reset_dialog_code)
|
|
|
|
def __getitem__(self, item):
|
|
return getattr(self, item)
|
|
def __setitem__(self, item, value):
|
|
return setattr(self, item, value)
|
|
|
|
# for testing without editor:
|
|
def main():
|
|
import sys
|
|
from qtpy.QtWidgets import QApplication
|
|
app = QApplication(sys.argv)
|
|
widget = OffsetToolButton()
|
|
widget.show()
|
|
|
|
sys.exit(app.exec())
|
|
if __name__ == "__main__":
|
|
main()
|