218 lines
8.0 KiB
Python
218 lines
8.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os, sys
|
|
|
|
###################################################################
|
|
|
|
class menu:
|
|
# Main Menu :
|
|
# |-> Files : Display gcode files in FTP directory
|
|
# |-> Commands : Display custom commands (homing, setup pen, etc ...)
|
|
# |-> Poweroff : Power off the system
|
|
# |-> Reboot : Reboot the system
|
|
|
|
main_menu = ['Files',
|
|
'Commands',
|
|
'Poweroff',
|
|
'Reboot']
|
|
|
|
cmd_menu = ['..',
|
|
'Homing',
|
|
'Set up Pen',
|
|
'Set down Pen',
|
|
'Set 1cm X axis',
|
|
'Set 1cm Y axis',
|
|
'Set rectangle']
|
|
|
|
files_menu = ['..']
|
|
|
|
def __init__(self, lcd=None, cols=0, rows=0, pycnc="", gcode_path=""):
|
|
''' constructor '''
|
|
self.lcd = lcd
|
|
self.pycnc_path=pycnc
|
|
self.gcode_path=gcode_path
|
|
self.max_cols = cols
|
|
self.max_rows = rows
|
|
self.current_cursor = 0
|
|
self.current_menu = self.main_menu
|
|
for dirs,r,files in os.walk(self.gcode_path):
|
|
for f in files:
|
|
if len(f) > 14:
|
|
f = f[:13]+'~'
|
|
self.files_menu.append(f)
|
|
|
|
def __refresh_menu(self):
|
|
''' refresh current menu '''
|
|
pos = 0
|
|
if self.current_cursor % self.max_rows == 0:
|
|
pos = self.current_cursor
|
|
else:
|
|
pos = self.current_cursor - 1
|
|
for idx, menu in enumerate(self.current_menu[pos:]):
|
|
self.lcd.cursor_pos = (idx, 2)
|
|
if menu.endswith('~'):
|
|
menu = menu[:-1]+'\x01'
|
|
self.lcd.write_string(menu)
|
|
if idx == self.max_rows - 1:
|
|
break
|
|
|
|
def __drawing(self, filename):
|
|
''' drawing gcode '''
|
|
self.lcd.clear()
|
|
self.lcd.home()
|
|
self.lcd.write_string(" drawing ...")
|
|
os.system(self.pycnc_path + ' ' + os.path.join(self.gcode_path, filename))
|
|
self.lcd.clear()
|
|
self.lcd.cursor_pos = (self.current_cursor % self.max_rows, 0)
|
|
self.lcd.write_string('\x00')
|
|
self.__refresh_menu()
|
|
|
|
def __pre_cmd(self):
|
|
''' pre command '''
|
|
self.lcd.clear()
|
|
self.lcd.home()
|
|
self.lcd.write_string(" waiting ...")
|
|
|
|
def __post_cmd(self):
|
|
''' post command '''
|
|
self.lcd.clear()
|
|
self.lcd.cursor_pos = (self.current_cursor % self.max_rows, 0)
|
|
self.lcd.write_string('\x00')
|
|
self.__refresh_menu()
|
|
|
|
def __homing(self):
|
|
''' set homing command '''
|
|
self.__pre_cmd()
|
|
with open("/tmp/homing.gcode", 'w') as f:
|
|
f.write("G28 (Homing)")
|
|
os.system(self.pycnc_path + ' ' + os.path.join("/tmp/homing.gcode"))
|
|
self.__post_cmd()
|
|
|
|
def __set_up_pen(self):
|
|
''' set up pen '''
|
|
self.__pre_cmd()
|
|
with open("/tmp/set_up_pen.gcode", 'w') as f:
|
|
f.write("M300 S50 (UP Pen)")
|
|
os.system(self.pycnc_path + ' ' + os.path.join("/tmp/set_up_pen.gcode"))
|
|
self.__post_cmd()
|
|
|
|
def __set_down_pen(self):
|
|
''' set down pen '''
|
|
self.__pre_cmd()
|
|
with open("/tmp/set_down_pen.gcode", 'w') as f:
|
|
f.write("M300 S30 (DOWN Pen)")
|
|
os.system(self.pycnc_path + ' ' + os.path.join("/tmp/set_down_pen.gcode"))
|
|
self.__post_cmd()
|
|
|
|
def __set_line_1cm(self, axis='X'):
|
|
''' set a 1cm line in X or Y axis '''
|
|
self.__pre_cmd()
|
|
with open("/tmp/set_line.gcode", 'w') as f:
|
|
f.write("M300 S50 (UP Pen)\n")
|
|
f.write("G28 (Homing)\n")
|
|
f.write("G1 X10 Y10 F1000.0\n")
|
|
f.write("M300 S30 (DOWN Pen)\n")
|
|
if axis.startswith('X'):
|
|
f.write("G1 Y15 F1000.0\n")
|
|
f.write("G1 Y12.5 F1000.0\n")
|
|
f.write("G1 X20 F1000.0\n")
|
|
f.write("G1 Y10 F1000.0\n")
|
|
f.write("G1 Y15 F1000.0\n")
|
|
elif axis.startswith('Y'):
|
|
f.write("G1 X15 F1000.0\n")
|
|
f.write("G1 X12.5 F1000.0\n")
|
|
f.write("G1 Y20 F1000.0\n")
|
|
f.write("G1 X10 F1000.0\n")
|
|
f.write("G1 X15 F1000.0\n")
|
|
f.write("M300 S50 (UP Pen)\n")
|
|
f.write("G28 (Homing)\n")
|
|
os.system(self.pycnc_path + ' ' + os.path.join("/tmp/set_line.gcode"))
|
|
self.__post_cmd()
|
|
|
|
def __set_rect(self):
|
|
''' set a rectangle '''
|
|
self.__pre_cmd()
|
|
with open("/tmp/set_rect.gcode", 'w') as f:
|
|
f.write("G28 (Homing)\n")
|
|
f.write("G1 X30 Y0 F2000.0\n")
|
|
f.write("G1 X30 Y30 F2000.0\n")
|
|
f.write("G1 X0 Y30 F2000.0\n")
|
|
f.write("G1 X0 Y0 F2000.0\n")
|
|
os.system(self.pycnc_path + ' ' + os.path.join("/tmp/set_rect.gcode"))
|
|
self.__post_cmd()
|
|
|
|
def init_menu(self):
|
|
''' display menu '''
|
|
self.lcd.clear()
|
|
self.lcd.cursor_pos = (self.current_cursor, 0)
|
|
self.lcd.write_string('\x00')
|
|
self.__refresh_menu()
|
|
|
|
def set_down(self):
|
|
''' set down menu '''
|
|
if self.current_cursor < len(self.current_menu) - 1:
|
|
self.lcd.clear()
|
|
self.current_cursor += 1
|
|
self.lcd.cursor_pos = (self.current_cursor % self.max_rows, 0)
|
|
self.lcd.write_string('\x00')
|
|
self.__refresh_menu()
|
|
|
|
def set_up(self):
|
|
''' set up menu '''
|
|
if self.current_cursor > 0:
|
|
self.lcd.clear()
|
|
self.current_cursor -= 1
|
|
self.lcd.cursor_pos = (self.current_cursor % self.max_rows, 0)
|
|
self.lcd.write_string('\x00')
|
|
self.__refresh_menu()
|
|
|
|
def set_ok(self):
|
|
''' set ok '''
|
|
if self.current_menu == self.main_menu:
|
|
if self.current_menu[self.current_cursor].startswith('Commands'):
|
|
self.current_menu = self.cmd_menu
|
|
elif self.current_menu[self.current_cursor].startswith('Files'):
|
|
self.current_menu = self.files_menu
|
|
self.files_menu = ['..']
|
|
for dirs,r,files in os.walk(self.gcode_path):
|
|
for f in files:
|
|
if len(f) > 14:
|
|
f = f[:13]+'~'
|
|
self.files_menu.append(f)
|
|
elif self.current_menu[self.current_cursor].startswith('Poweroff'):
|
|
self.lcd.clear()
|
|
self.lcd.home()
|
|
self.lcd.write_string(" Bye Bye ...")
|
|
os.system('sudo systemctl poweroff')
|
|
exit(0)
|
|
elif self.current_menu[self.current_cursor].startswith('Reboot'):
|
|
self.lcd.clear()
|
|
self.lcd.home()
|
|
self.lcd.write_string(" Bye Bye ...")
|
|
os.system('sudo systemctl reboot')
|
|
exit(0)
|
|
self.current_cursor = 0
|
|
self.init_menu()
|
|
else:
|
|
if self.current_menu == self.cmd_menu:
|
|
if self.current_menu[self.current_cursor].startswith('Homing'):
|
|
self.__homing()
|
|
elif self.current_menu[self.current_cursor].startswith('Set up Pen'):
|
|
self.__set_up_pen()
|
|
elif self.current_menu[self.current_cursor].startswith('Set down Pen'):
|
|
self.__set_down_pen()
|
|
elif self.current_menu[self.current_cursor].startswith('Set 1cm X axis'):
|
|
self.__set_line_1cm('X')
|
|
elif self.current_menu[self.current_cursor].startswith('Set 1cm Y axis'):
|
|
self.__set_line_1cm('Y')
|
|
elif self.current_menu[self.current_cursor].startswith('Set rectangle'):
|
|
self.__set_rect()
|
|
elif self.current_menu == self.files_menu:
|
|
if not self.current_menu[self.current_cursor].startswith('..'):
|
|
self.__drawing(self.current_menu[self.current_cursor])
|
|
|
|
if self.current_menu[self.current_cursor].startswith('..'):
|
|
self.current_menu = self.main_menu
|
|
self.current_cursor = 0
|
|
self.init_menu()
|