vincent/PiPlot2D
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify src folder in piplot folder

Browse Source
This commit is contained in:
sinseman44
2020-11-15 07:54:19 +00:00
parent e55f8ea1a2
commit 3a2275a30f
6 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
piplot/__init__.py Normal file
View File

22
piplot/config.py Normal file
View File

@@ -0,0 +1,22 @@
# Hardware config.
# HD44780 LCD CONFIG
LCD_RS = 11
LCD_E = 5
LCD_DATA4 = 6
LCD_DATA5 = 13
LCD_DATA6 = 19
LCD_DATA7 = 26
NUM_COLS = 16
NUM_ROWS = 2
# BUTTONS CONFIG
PG_UP = 17
PG_DOWN = 22
PG_OK = 27
# rootfs config.
PYCNC_EXEC="/usr/local/bin/pycnc"
GCODE_REPO="/srv/ftp"

93
piplot/main.py Executable file
View File

@@ -0,0 +1,93 @@
#!/usr/bin/env python3
import os
import sys
import time
import RPi.GPIO as GPIO
from RPLCD.gpio import CharLCD
from src.config import *
from src.menu import menu
###################################################################
GPIO.setwarnings(False)
lcd = CharLCD(pin_rs=LCD_RS,
pin_e=LCD_E,
pin_rw=None,
pins_data=[LCD_DATA4, LCD_DATA5, LCD_DATA6, LCD_DATA7],
pin_backlight=None,
numbering_mode=GPIO.BCM,
cols=NUM_COLS,
rows=NUM_ROWS,
dotsize=8,
charmap='A00')
arrow = (
0b00000,
0b10000,
0b11000,
0b11100,
0b11110,
0b11100,
0b11000,
0b10000,
)
tilde = (
0b00000,
0b00000,
0b00000,
0b01101,
0b10101,
0b10110,
0b00000,
0b00000,
)
lcd.create_char(0, arrow)
lcd.create_char(1, tilde)
menu = menu(lcd,
cols=NUM_COLS,
rows=NUM_ROWS,
pycnc=PYCNC_EXEC,
gcode_path=GCODE_REPO)
def pg_up_cb(channel):
''' page up button callback '''
if GPIO.input(channel) == GPIO.HIGH:
menu.set_up()
def pg_down_cb(channel):
''' page down button callback '''
if GPIO.input(channel) == GPIO.HIGH:
menu.set_down()
def pg_ok_cb(channel):
''' page ok button callback '''
if GPIO.input(channel) == GPIO.HIGH:
menu.set_ok()
def init():
''' init pins and display main menu '''
try:
GPIO.setup(PG_UP, GPIO.IN)
GPIO.setup(PG_DOWN, GPIO.IN)
GPIO.setup(PG_OK, GPIO.IN)
GPIO.add_event_detect(PG_UP, GPIO.RISING, callback=pg_up_cb, bouncetime=50)
GPIO.add_event_detect(PG_DOWN, GPIO.RISING, callback=pg_down_cb, bouncetime=50)
GPIO.add_event_detect(PG_OK, GPIO.RISING, callback=pg_ok_cb, bouncetime=50)
except:
e = sys.exc_info()[1]
print("Error : {}".format(e))
sys.exit(1)
menu.init_menu()
def main():
init()
while True:
time.sleep(1)
lcd.close()
###################################################################
if __name__ == '__main__':
main()

217
piplot/menu.py Normal file
View File

@@ -0,0 +1,217 @@
#!/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()