#!/usr/bin/env python3 # vim: sts=4 sw=4 et # GladeVcp actions # # Copyright (c) 2010 Pavel Shramov # # 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. import gi gi.require_version("Gtk","3.0") from gi.repository import Gtk from gi.repository import GObject from gi.repository import GLib import os import time import re, string from .hal_widgets import _HalWidgetBase import linuxcnc from hal_glib import GStat import hal from gladevcp.core import Info, Action INFO = Info() ACTION = Action() _ = lambda x: x class _EMCStaticHolder: def __init__(self): # Delay init... self.linuxcnc = None self.stat = None self.gstat = None def get(self): if not self.linuxcnc: self.linuxcnc = linuxcnc.command() if not self.gstat: self.gstat = GStat() return self.linuxcnc, self.gstat.stat, self.gstat class _EMCStatic: holder = _EMCStaticHolder() def get(self): return self.holder.get() class _EMC_ActionBase(_HalWidgetBase): _gproperties = {'name': (GObject.TYPE_STRING, 'Name', 'Action name', "", GObject.ParamFlags.READWRITE|GObject.ParamFlags.CONSTRUCT) } linuxcnc_static = _EMCStatic() def _hal_init(self): self.linuxcnc, self.stat, self.gstat = self.linuxcnc_static.get() self._stop_emission = False # if 'NO_FORCE_HOMING' is true, MDI commands are allowed before homing. inifile = os.environ.get('INI_FILE_NAME', '/dev/null') ini = linuxcnc.ini(inifile) self.no_f_home = int(ini.find("TRAJ", "NO_FORCE_HOMING") or 0) def machine_on(self): self.stat.poll() return self.stat.task_state > linuxcnc.STATE_OFF def is_auto_mode(self): self.stat.poll() print(self.stat.task_mode, linuxcnc.MODE_AUTO) return self.stat.task_mode == linuxcnc.MODE_AUTO def is_file_loaded(self): self.stat.poll() print("file name:",self.stat.file) if self.stat.file: return True else: return False def is_all_homed(self): self.stat.poll() homed_count = 0 for i,h in enumerate(self.stat.homed): #Don't worry about joint to axis mapping if h: homed_count +=1 if homed_count == self.stat.joints: return True return False def no_home_required(self): return self.no_f_home def safe_handler(self, f): def _f(self, *a, **kw): if self._stop_emission: return return f(self, *a, **kw) return _f def set_active_safe(self, active): self._stop_emission = True self.set_active(active) self._stop_emission = False def do_get_property(self, property): name = property.name.replace('-', '_') if name == 'name': return self.get_name() elif name == 'label': return self.get_label() elif name == 'tooltip': return self.get_tooltip() elif name == 'stock_id': return self.get_stock_id() else: raise AttributeError("Unknown property: %s" % property.name) def do_set_property(self, property, value): name = property.name.replace('-', '_') if name == 'name': if value: self.set_name(value) elif name == 'label': self.set_label(value) elif name == 'tooltip': self.set_tooltip(value) elif name == 'stock_id': self.set_stock_id(value) else: raise AttributeError("Unknown property: %s" % property.name) return True class _EMC_Action(Gtk.Action, _EMC_ActionBase): __gproperties__ = _EMC_ActionBase._gproperties def __init__(self, name=None): Gtk.Action.__init__(self, None, None, None, None) self._stop_emission = False self.connect('activate', self.safe_handler(self.on_activate)) def set_active_safe(self, a): return #XXX: Override set_active with nop def on_activate(self, w): return True class _EMC_ToggleAction(Gtk.ToggleAction, _EMC_ActionBase): __gproperties__ = _EMC_ActionBase._gproperties def __init__(self, name=None): Gtk.ToggleAction.__init__(self, None, None, None, None) self._stop_emission = False self.connect('toggled', self.safe_handler(self.on_toggled)) # XXX: Override nop in _EMC_Action set_active_safe = _EMC_ActionBase.set_active_safe def on_toggled(self, w): return True class _EMC_RadioAction(Gtk.RadioAction, _EMC_ToggleAction): __gproperties__ = _EMC_ToggleAction._gproperties def __init__(self, name=None): Gtk.RadioAction.__init__(self, None, None, None, None, 0) self._stop_emission = False self.connect('toggled', self.safe_handler(self.on_toggled)) def on_toggled(self, w): if not w.get_active(): return return self.on_activate(w) class EMC_Stat(GStat, _EMC_ActionBase): __gtype_name__ = 'EMC_Stat' def __init__(self): stat = self.linuxcnc_static.get()[1] GStat.__init__(self, stat) def _hal_init(self): pass def _action(klass, f, *a, **kw): class _C(_EMC_Action): __gtype_name__ = klass def on_activate(self, w): print(klass) f(self, *a, **kw) return _C EMC_Action_ESTOP = _action('EMC_Action_ESTOP', lambda s: s.linuxcnc.state(linuxcnc.STATE_ESTOP)) EMC_Action_ESTOP_RESET = _action('EMC_Action_ESTOP_RESET', lambda s: s.linuxcnc.state(linuxcnc.STATE_ESTOP_RESET)) EMC_Action_ON = _action('EMC_Action_ON', lambda s: s.linuxcnc.state(linuxcnc.STATE_ON)) EMC_Action_OFF = _action('EMC_Action_OFF', lambda s: s.linuxcnc.state(linuxcnc.STATE_OFF)) class EMC_ToggleAction_ESTOP(_EMC_ToggleAction): __gtype_name__ = 'EMC_ToggleAction_ESTOP' def _hal_init(self): _EMC_ToggleAction._hal_init(self) self.set_active_safe(True) self.gstat.connect('state-estop', lambda w: self.set_active_safe(True)) self.gstat.connect('state-estop-reset', lambda w: self.set_active_safe(False)) def on_toggled(self, w): if self.get_active(): print('Issuing ESTOP') self.linuxcnc.state(linuxcnc.STATE_ESTOP) else: print('Issuing ESTOP RESET') self.linuxcnc.state(linuxcnc.STATE_ESTOP_RESET) class EMC_ToggleAction_Power(_EMC_ToggleAction): __gtype_name__ = 'EMC_ToggleAction_Power' def _hal_init(self): _EMC_ToggleAction._hal_init(self) self.set_active_safe(False) self.set_sensitive(False) self.gstat.connect('state-on', lambda w: self.set_active_safe(True)) self.gstat.connect('state-off', lambda w: self.set_active_safe(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop-reset', lambda w: self.set_sensitive(True)) def on_toggled(self, w): if self.get_active(): print('Issuing ON') self.linuxcnc.state(linuxcnc.STATE_ON) else: print('Issuing OFF') self.linuxcnc.state(linuxcnc.STATE_OFF) class EMC_RadioAction_ESTOP(_EMC_RadioAction): __gtype_name__ = 'EMC_RadioAction_ESTOP' def _hal_init(self): _EMC_RadioAction._hal_init(self) self.set_active_safe(True) self.gstat.connect('state-estop', lambda w: self.set_active_safe(True)) def on_activate(self, w): self.linuxcnc.state(linuxcnc.STATE_ESTOP) class EMC_RadioAction_ESTOP_RESET(_EMC_RadioAction): __gtype_name__ = 'EMC_RadioAction_ESTOP_RESET' def _hal_init(self): _EMC_RadioAction._hal_init(self) self.set_active_safe(False) self.gstat.connect('state-estop-reset', lambda w: self.set_active_safe(True)) def on_activate(self, w): self.linuxcnc.state(linuxcnc.STATE_ESTOP_RESET) class EMC_RadioAction_ON(_EMC_RadioAction): __gtype_name__ = 'EMC_RadioAction_ON' def _hal_init(self): _EMC_RadioAction._hal_init(self) self.set_active_safe(True) self.gstat.connect('state-on', lambda w: self.set_active_safe(True)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop-reset', lambda w: self.set_sensitive(True)) def on_activate(self, w): self.linuxcnc.state(linuxcnc.STATE_ON) class EMC_RadioAction_OFF(_EMC_RadioAction): __gtype_name__ = 'EMC_RadioAction_OFF' def _hal_init(self): _EMC_RadioAction._hal_init(self) self.set_active_safe(False) self.gstat.connect('state-off', lambda w: self.set_active_safe(True)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop-reset', lambda w: self.set_sensitive(True)) def on_activate(self, w): self.linuxcnc.state(linuxcnc.STATE_OFF) def running(s, do_poll=True): if do_poll: s.poll() return s.task_mode == linuxcnc.MODE_AUTO and s.interp_state != linuxcnc.INTERP_IDLE def ensure_mode(s, c, *modes): s.poll() if not modes: return False if s.task_mode in modes: return True if running(s, do_poll=False): return False c.mode(modes[0]) c.wait_complete() return True class EMC_Action_Python(_EMC_Action): __gtype_name__ = 'EMC_Action_Python' command = GObject.property(type=str, default='', nick='Python Command') is_homed = GObject.property(type=bool, default=True, nick='Must Be Homed', blurb='Machine Must be homed for widgets to be sensitive to input') is_on = GObject.property(type=bool, default=True, nick='Must Be On', blurb='Machine Must be On for widgets to be sensitive to input') is_idle = GObject.property(type=bool, default=True, nick='Must Be Idle', blurb='Machine Must be Idle for widgets to be sensitive to input') requires_manual = GObject.property(type=bool, default=False, nick='Preset Manual Mode', blurb='Preset Manual Mode before command') requires_mdi = GObject.property(type=bool, default=False, nick='Preset MDI Mode', blurb='Preset MDI Mode before command') requires_auto = GObject.property(type=bool, default=False, nick='Preset Auto Mode', blurb='Preset Auto Mode before command') def _hal_init(self): _EMC_Action._hal_init(self) self.set_sensitive(False) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) if self.is_on: self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) if self.is_homed: self.gstat.connect('interp-idle', lambda w: self.set_sensitive(self.machine_on() and ( self.is_all_homed() or self.no_home_required() ) )) else: self.gstat.connect('interp-idle', lambda w: self.set_sensitive(self.machine_on()) ) if self.is_idle: self.gstat.connect('interp-run', lambda w: self.set_sensitive(False)) if self.is_homed: self.gstat.connect('all-homed', lambda w: self.set_sensitive(self.machine_on())) def on_activate(self, w): self._globalParameter = { 'EXT':self._panel_instance.get_handler_obj(), 'GSTAT':self.gstat,'CMD':self.linuxcnc,'STAT':self.stat, 'ACTION':ACTION, 'HAL':hal, 'INFO':INFO} self._localsParameter = {'dir': dir, 'self': self,'linuxcnc':linuxcnc} if self.requires_manual: ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_MAN) elif self.requires_mdi: ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_MDI) elif self.requires_auto: ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_AUTO) exec(self.command, self._globalParameter, self._localsParameter) class EMC_Action_Run(_EMC_Action): __gtype_name__ = 'EMC_Action_Run' program_start_line = GObject.property(type=int, default=0, minimum=0, nick='Restart line', blurb='Restart line number-Usually 0 - program start') reset_line = GObject.property(type=int, default=0, minimum=0, nick='Restart line after restarting once', blurb='Line number that will be set afterthe next restart. -usually 0 - program start') def set_restart_line(self,line,resetline=0): self.program_start_line = line self.reset_line = resetline def on_activate(self, w): ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_AUTO) self.linuxcnc.auto(linuxcnc.AUTO_RUN, self.program_start_line) self.program_start_line = self.reset_line class EMC_Action_Step(_EMC_Action): __gtype_name__ = 'EMC_Action_Step' def _hal_init(self): _EMC_Action._hal_init(self) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_sensitive(self.machine_on())) def on_activate(self, w): ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_AUTO) self.linuxcnc.auto(linuxcnc.AUTO_STEP) class EMC_Action_Pause(_EMC_Action): __gtype_name__ = 'EMC_Action_Pause' def on_activate(self, w): self.stat.poll() if self.stat.task_mode != linuxcnc.MODE_AUTO or\ self.stat.interp_state not in (linuxcnc.INTERP_READING, linuxcnc.INTERP_WAITING): return ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_AUTO) self.linuxcnc.auto(linuxcnc.AUTO_PAUSE) class EMC_Action_Resume(_EMC_Action): __gtype_name__ = 'EMC_Action_Resume' def on_activate(self, w): print("RESUME") self.stat.poll() if not self.stat.paused: return if self.stat.task_mode not in (linuxcnc.MODE_AUTO, linuxcnc.MODE_MDI): return ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_AUTO, linuxcnc.MODE_MDI) self.linuxcnc.auto(linuxcnc.AUTO_RESUME) class EMC_Action_Stop(_EMC_Action): __gtype_name__ = 'EMC_Action_Stop' def on_activate(self, w): self.linuxcnc.abort() self.linuxcnc.wait_complete() class EMC_ToggleAction_Run(_EMC_ToggleAction, EMC_Action_Run): __gtype_name__ = 'EMC_ToggleAction_Run' program_start_line = GObject.property(type=int, default=0, minimum=0, nick='Restart line', blurb='Restart line number-Usually 0 - program start') reset_line = GObject.property(type=int, default=0, minimum=0, nick='Restart line after restarting once', blurb='Line number that will be set afterthe next restart. -usually 0 - program start') def _hal_init(self): _EMC_ToggleAction._hal_init(self) self.set_active_safe(False) self.set_sensitive(False) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect( 'interp-idle', lambda w: self.set_sensitive( self.machine_on() and ( self.is_all_homed() or self.no_home_required() ) and self.is_file_loaded() ) ) self.gstat.connect('interp-idle', lambda w: self.set_active_safe(False)) self.gstat.connect('interp-run', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-run', lambda w: self.set_active_safe(True)) self.gstat.connect('all-homed', lambda w: self.set_sensitive( self.machine_on() and self.is_file_loaded() )) self.gstat.connect('file-loaded', self.file_loaded_check) def file_loaded_check(self,widget,filename): self.set_sensitive( self.machine_on() and (self.is_all_homed() or self.no_home_required()) ) def set_restart_line(self,line,resetline=0): self.program_start_line = line self.reset_line = resetline def on_toggled(self, w): if self.get_active(): return self.on_activate(w) class EMC_ToggleAction_Stop(_EMC_ToggleAction, EMC_Action_Stop): __gtype_name__ = "EMC_ToggleAction_Stop" def _hal_init(self): _EMC_ToggleAction._hal_init(self) self.set_active_safe(True) self.set_sensitive(False) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_active_safe(True)) self.gstat.connect('interp-run', lambda w: self.set_sensitive(self.machine_on())) self.gstat.connect('interp-run', lambda w: self.set_active_safe(False)) def on_toggled(self, w): if self.get_active(): return self.on_activate(w) class EMC_ToggleAction_Pause(_EMC_ToggleAction, EMC_Action_Pause): __gtype_name__ = "EMC_ToggleAction_Pause" def _hal_init(self): _EMC_ToggleAction._hal_init(self) self.resume = EMC_Action_Resume() self.resume._hal_init() self.set_active_safe(True) self.set_sensitive(False) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_active_safe(False)) self.gstat.connect('interp-run', lambda w: self.set_sensitive(self.machine_on())) self.gstat.connect('interp-run', lambda w: self.set_active_safe(False)) self.gstat.connect('interp-paused', lambda w: self.set_active_safe(True)) self.gstat.connect('interp-waiting', lambda w: self.set_active_safe(False)) def on_toggled(self, w): if self.get_active(): return self.on_activate(w) else: return self.resume.on_activate(self.resume) class HalTemplate(string.Template): idpattern = '[_a-z][-._a-z0-9]*' class FloatComp: def __init__(self, comp): self.comp = comp def __getitem__(self, k): v = float(self.comp[k]) return "%f" % v class EMC_Action_MDI(_EMC_Action): __gtype_name__ = 'EMC_Action_MDI' command = GObject.property(type=str, default='', nick='MDI Command') def _hal_init(self): _EMC_Action._hal_init(self) self.set_sensitive(False) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_sensitive(self.machine_on() and ( self.is_all_homed() or self.no_home_required() ) )) self.gstat.connect('interp-run', lambda w: self.set_sensitive(False)) self.gstat.connect('all-homed', lambda w: self.set_sensitive(self.machine_on())) def on_activate(self, w): ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_MDI) template = HalTemplate(self.command) cmd = template.substitute(FloatComp(self.hal)) self.linuxcnc.mdi(cmd) class EMC_ToggleAction_MDI(_EMC_ToggleAction, EMC_Action_MDI): __gtype_name__ = 'EMC_ToggleAction_MDI' __gsignals__ = { 'mdi-command-start': (GObject.SignalFlags.RUN_FIRST, GObject.TYPE_NONE, ()), 'mdi-command-stop': (GObject.SignalFlags.RUN_FIRST, GObject.TYPE_NONE, ()), } command = GObject.property(type=str, default='', nick='MDI Command') def _hal_init(self): _EMC_ToggleAction._hal_init(self) EMC_Action_MDI._hal_init(self) def on_toggled(self, w): if not self.get_active(): return self.set_sensitive(False) self.emit('mdi-command-start') self.on_activate(w) GLib.timeout_add(100, self.wait_complete) def wait_complete(self): if self.linuxcnc.wait_complete(0) in [-1, linuxcnc.RCS_EXEC]: return True self.emit('mdi-command-stop') self.set_active_safe(False) self.set_sensitive(self.machine_on()) return False class EMC_Action_UnHome(_EMC_Action): __gtype_name__ = 'EMC_Action_Unhome' axis = GObject.property(type=int, default=-1, minimum=-1, nick='Axis', blurb='Axis to unhome. -1 to unhome all') def _hal_init(self): _EMC_Action._hal_init(self) self.set_sensitive(False) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_sensitive(self.machine_on())) self.gstat.connect('interp-run', lambda w: self.set_sensitive(False)) def on_activate(self, w): ensure_mode(self.stat, self.linuxcnc, linuxcnc.MODE_MANUAL) self.linuxcnc.teleop_enable(False) self.linuxcnc.unhome(self.axis) def prompt_areyousure(type, message, secondary=None): dialog = Gtk.MessageDialog(parent=None, modal=True, message_type=type, text=message) dialog.add_buttons(Gtk.STOCK_YES, Gtk.ResponseType.YES, Gtk.STOCK_NO, Gtk.ResponseType.NO) dialog.set_keep_above(True) if secondary: dialog.format_secondary_text(secondary) r = dialog.run() dialog.destroy() return r == Gtk.ResponseType.YES class EMC_Action_Home(_EMC_Action): __gtype_name__ = 'EMC_Action_Home' axis = GObject.property(type=int, default=-1, minimum=-1, nick='Axis', blurb='Axis to home. -1 to home all') confirm_homed = GObject.property(type=bool, default=False, nick='Confirm rehoming', blurb='Ask user if axis is already homed') def _hal_init(self): _EMC_Action._hal_init(self) self.set_sensitive(False) self.gstat.connect('state-off', lambda w: self.set_sensitive(False)) self.gstat.connect('state-estop', lambda w: self.set_sensitive(False)) self.gstat.connect('interp-idle', lambda w: self.set_sensitive(self.machine_on())) self.gstat.connect('interp-run', lambda w: self.set_sensitive(False)) def homed(self): if self.axis != -1: return self.stat.homed[self.axis] for i,h in enumerate(self.stat.homed): if h and self.stat.axis_mask & (1<