mitchellhansen/Trac3r-python
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added UI

Browse Source
This commit is contained in:
MitchellHansen
2019-01-09 18:16:52 -08:00
parent 3b74e57e1d
commit a8d9efb6bb
2 changed files with 240 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
activate.sh
View File

@@ -1,3 +1,4 @@
sudo apt-get install python3-pil python3-pil.imagetk
sudo apt install libcairo2-dev pkg-config sudo apt install libcairo2-dev pkg-config
virtualenv -p python3 .env virtualenv -p python3 .env
source .env/bin/activate source .env/bin/activate

380
main.py
View File

@@ -1,162 +1,139 @@
import math from tkinter import Tk, Label, filedialog, Button
from PIL import Image, ImageTk
from svgpathtools import svg2paths, Line, QuadraticBezier, CubicBezier
import cairo, subprocess, bezier, os, math, time
import numpy as np import numpy as np
touch_height = 20
raise_height = 2
head_x_offset = 50
speed = 500
lift_markers = True
PREAMBLE = ''' class GCoder(Tk):
G1 Z20 def __init__(self):
M107 super().__init__()
M190 S0
M104 S0
G28 ; home all axes
G0 F{1}
G1 Z{0}
G1 Z{0}
'''.format(touch_height + raise_height, speed)
FINISH = """ # Setup the file structure
G1 Z{0} F7000 if not os.path.exists("output"):
M104 S0
G28 X0 Y0
M84
""".format(75)
import cairo, subprocess, bezier, os
from svgpathtools import svg2paths, Line, QuadraticBezier, CubicBezier
# Setup the file structure
if not os.path.exists("output"):
os.makedirs("output") os.makedirs("output")
# Convert the bmp to a vector svg # Height at which the pen touches and draws on the surface
file_name = "geom" self.touch_height = 20
# How far to raise the pen tip to raise it off the page
self.raise_height = 2
# The inherent offset from true 0 we have from the pen bracket
self.head_x_offset = 50
# XY movement speed
self.speed = 500
# Weather we render lift markers
self.lift_markers = True
subprocess.call(["mogrify", "-format", "bmp", "input-images/{}.svg".format(file_name)]) # X and Y offsets to place the image on A11 paper
self.offset_x = 75 + self.head_x_offset
self.offset_y = 20
subprocess.call(["mkbitmap", "input-images/{}.bmp".format(file_name), "-x", # Bed dimensions to fit A11 paper
self.bed_max_x = 280
self.bed_min_x = self.offset_x
self.bed_max_y = 280
self.bed_min_y = 20
self.started = False
self.gcode_preamble = '''
G91 ; Set to relative mode for the initial pen lift
G1 Z20 ; Lift head by 20
G90 ; Set back to absolute position mode
M107 ; Fan off
M190 S0 ; Set bed temp
M104 S0 ; Set nozzle temp
G28 ; home all axes
G0 F{1} ; Set the feed rate
G1 Z{0} ; Move the pen to just above the paper
'''.format(self.touch_height + self.raise_height, self.speed)
self.gcode_end = '''
G1 Z{0} F7000 ; Raise the pen high up so we can fit a cap onto it
M104 S0 ; Set the nozzle to 0
G28 X0 Y0 ; Home back to (0,0) for (x,y)
M84 ; Turn off the motors
'''.format(75)
w, h = 300, 300
self.geometry("{}x{}".format(w, h))
self.surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, 300, 300)
self.context = cairo.Context(self.surface)
self.context.scale(1, 1)
self.context.set_line_width(0.4)
self.button = Button(self, text="Select Image", command=self.file_select_callback)
self.button.pack()
self.mainloop()
def file_select_callback(self):
filepath = filedialog.askopenfilename(initialdir=".", title="Select file",
filetypes=(("jpeg files", "*.jpg"), ("all files", "*.*")))
if len(filepath) is 0:
return
self.context.rectangle(0, 0, 300, 300)
self.context.set_source_rgba(1, 1, 1, 1.0)
self.context.fill()
self.context.set_source_rgba(0, 0, 0, 1.0)
filename = os.path.basename(filepath)
self.convert_image(filename)
self.convert_gcode()
self.render_gcode()
self._image_ref = ImageTk.PhotoImage(
Image.frombuffer("RGBA", (300, 300), self.surface.get_data().tobytes(), "raw", "BGRA", 0, 1))
self.label = Label(self, image=self._image_ref)
self.label.pack(expand=True, fill="both")
def convert_image(self, file_name):
base_name = file_name.split(".")[0]
print("Converting input file [{}]".format(file_name))
print("Running mogrify...")
start = time.time()
subprocess.call(["mogrify", "-format", "bmp", "input-images/{}".format(file_name)])
print("Run took [{:.2f}] seconds".format(time.time() - start))
print("Running mkbitmap...")
start = time.time()
subprocess.call(["mkbitmap", "input-images/{}.bmp".format(base_name), "-x",
"-f", "15", "-f", "15",
#"-b", "0", # "-b", "0",
"-o", "input-images/{}-n.bmp".format(file_name) "-o", "input-images/{}-n.bmp".format(base_name)
]) ])
print("Run took [{:.2f}] seconds".format(time.time() - start))
subprocess.call(["potrace", print("Running potrace...")
start = time.time()
subprocess.call(["potrace",
"-t", "20", "-t", "20",
"-z", "white", "-z", "white",
"-b", "svg", "-b", "svg",
"input-images/{}-n.bmp".format(file_name), "input-images/{}-n.bmp".format(base_name),
"--rotate", "90", "--rotate", "0",
"-o", "tmp/conversion-output.svg", "-o", "tmp/conversion-output.svg",
]) ])
print("Run took [{:.2f}] seconds\n".format(time.time() - start))
# read in the svg def render_gcode(self):
paths, attributes = svg2paths("tmp/conversion-output.svg")
gcode = "" file = open("output/gcode-output.gcode", "r")
gcode += PREAMBLE
started = False
scale = 0.0045
offset_x = 75 + head_x_offset
offset_y = 20
# Walk through the paths and create the GCODE
for path in paths:
previous_x = None
previous_y = None
# rotated = path.rotated(90)
for part in path:
start = part.start
end = part.end
start_x = start.real * scale + offset_x
start_y = start.imag * scale + offset_y
end_x = end.real * scale + offset_x
end_y = end.imag * scale + offset_y
# Check to see if the endpoint of the last cycle continues and wether we need to lift the pen or not
lift = True
if previous_x is not None and previous_y is not None:
if abs(start.real - previous_x) < 30 and abs(start.imag - previous_y) < 30:
lift = False
# if the pen needs to lift,
# if lift:
previous_x = end.real
previous_y = end.imag
if lift:
gcode += "G1 Z{}\n".format(raise_height + touch_height)
else:
gcode += "# NOT LIFTING\n"
if isinstance(part, CubicBezier):
nodes = np.asfortranarray([
[start.real, part.control1.real, part.control2.real, end.real],
[start.imag, part.control1.imag, part.control2.imag, end.imag],
])
curve = bezier.Curve.from_nodes(nodes)
evals = []
pos = np.linspace(0.1, 1, 10)
for i in pos:
evals.append(curve.evaluate(i))
gcode += "G1 X{} Y{}\n".format(start_x, start_y)
gcode += "G1 Z{} \n".format(touch_height)
for i in evals:
x = i[0][0]
y = i[1][0]
gcode += "G1 X{} Y{}\n".format(x * scale + offset_x, y * scale + offset_y)
#gcode += "G1 X{} Y{}\n".format(end.real * scale + offset_x, end.imag * scale + offset_y)
if isinstance(part, Line):
gcode += "G1 X{} Y{}\n".format(start_x, start_y)
gcode += "G1 Z{} \n".format(touch_height)
gcode += "G1 X{} Y{}\n".format(end_x, end_y)
gcode += FINISH
output_gcode = open("output/gcode-output.gcode", "w")
output_gcode.write(gcode)
output_gcode.close()
file = open("output/gcode-output.gcode", "r")
x = None
y = None
with cairo.SVGSurface("rendered-output.svg", 300, 300) as surface:
context = cairo.Context(surface)
context.scale(1, 1)
context.set_line_width(0.4)
largest_x = 0 largest_x = 0
largest_y = 0 largest_y = 0
smallest_x = 300 smallest_x = 300
smallest_y = 300 smallest_y = 300
x = None
y = None
for line in file: for line in file:
@@ -177,12 +154,12 @@ with cairo.SVGSurface("rendered-output.svg", 300, 300) as surface:
y = float(operand[1:]) y = float(operand[1:])
if y > largest_y: largest_y = y if y > largest_y: largest_y = y
if y < smallest_y: smallest_y = y if y < smallest_y: smallest_y = y
elif operand.startswith("Z{}".format(touch_height + raise_height)): elif operand.startswith("Z{}".format(self.touch_height + self.raise_height)):
# signify a lift # signify a lift
if prev_x is not None and prev_y is not None and lift_markers: if prev_x is not None and prev_y is not None and self.lift_markers:
context.arc(prev_x, prev_y, 0.5, 0, 2*math.pi) self.context.arc(prev_x - self.head_x_offset, prev_y, 0.5, 0, 2*math.pi)
context.stroke() self.context.stroke()
prev_x = None prev_x = None
prev_y = None prev_y = None
@@ -190,23 +167,130 @@ with cairo.SVGSurface("rendered-output.svg", 300, 300) as surface:
y = None y = None
if (prev_x != x and prev_x is not None) or (prev_y != y and prev_y is not None): if (prev_x != x and prev_x is not None) or (prev_y != y and prev_y is not None):
context.line_to(prev_x, prev_y) self.context.line_to(prev_x - self.head_x_offset, prev_y)
context.line_to(x, y) self.context.line_to(x - self.head_x_offset, y)
context.stroke() self.context.stroke()
print("Largest X : " + str(largest_x)) print("Largest X : " + str(largest_x))
print("Largest Y : " + str(largest_y))
print("Smallest X : " + str(smallest_x)) print("Smallest X : " + str(smallest_x))
print("Largest Y : " + str(largest_y))
print("Smallest Y : " + str(smallest_y)) print("Smallest Y : " + str(smallest_y))
if largest_x > 280: if largest_x > self.bed_max_x:
print("X OVERFLOW") print("X OVERFLOW")
if largest_y > 280: if largest_y > self.bed_max_y:
print("Y OVERFLOW") print("Y OVERFLOW")
if smallest_x < 125: if smallest_x < self.bed_min_x:
print("X_UNDERFLOW") print("X_UNDERFLOW")
if smallest_y < 20: if smallest_y < self.bed_min_y:
print("Y_UNDERFLOW") print("Y_UNDERFLOW")
def convert_gcode(self):
# read in the svg
paths, attributes = svg2paths("tmp/conversion-output.svg")
bounding_x_max = None
bounding_x_min = None
bounding_y_max = None
bounding_y_min = None
for path in paths:
bbox = path.bbox()
if bounding_x_max is None or bbox[0] > bounding_x_max:
bounding_x_max = bbox[0]
if bounding_x_min is None or bbox[1] < bounding_x_min:
bounding_x_min = bbox[1]
if bounding_y_max is None or bbox[2] > bounding_y_max:
bounding_y_max = bbox[2]
if bounding_y_min is None or bbox[3] > bounding_y_min:
bounding_y_min = bbox[3]
print("Maximum X : {}".format(bounding_x_max))
print("Minimum Y : {}".format(bounding_x_min))
print("Maximum X : {}".format(bounding_y_max))
print("Minimum Y : {}".format(bounding_y_min))
max_dim = max(bounding_x_max, bounding_x_min, bounding_y_max, bounding_y_min)
scale = (300 - self.offset_x) / max_dim
print("Scaling to : {}\n".format(scale))
# Start the gcode
gcode = ""
gcode += self.gcode_preamble
# Walk through the paths and create the GCODE
for path in paths:
previous_x = None
previous_y = None
for part in path:
start = part.start
end = part.end
start_x = start.real * scale + self.offset_x
start_y = start.imag * scale + self.offset_y
end_x = end.real * scale + self.offset_x
end_y = end.imag * scale + self.offset_y
# Check to see if the endpoint of the last cycle continues and wether we need to lift the pen or not
lift = True
if previous_x is not None and previous_y is not None:
if abs(start.real - previous_x) < 30 and abs(start.imag - previous_y) < 30:
lift = False
# if the pen needs to lift,
# if lift:
previous_x = end.real
previous_y = end.imag
if lift:
gcode += "G1 Z{}\n".format(self.raise_height + self.touch_height)
else:
gcode += "# NOT LIFTING\n"
if isinstance(part, CubicBezier):
nodes = np.asfortranarray([
[start.real, part.control1.real, part.control2.real, end.real],
[start.imag, part.control1.imag, part.control2.imag, end.imag],
])
curve = bezier.Curve.from_nodes(nodes)
evals = []
pos = np.linspace(0.1, 1, 10)
for i in pos:
evals.append(curve.evaluate(i))
gcode += "G1 X{} Y{}\n".format(start_x, start_y)
gcode += "G1 Z{} \n".format(self.touch_height)
for i in evals:
x = i[0][0]
y = i[1][0]
gcode += "G1 X{} Y{}\n".format(x * scale + self.offset_x, y * scale + self.offset_y)
if isinstance(part, Line):
gcode += "G1 X{} Y{}\n".format(start_x, start_y)
gcode += "G1 Z{} \n".format(self.touch_height)
gcode += "G1 X{} Y{}\n".format(end_x, end_y)
gcode += self.gcode_end
output_gcode = open("output/gcode-output.gcode", "w")
output_gcode.write(gcode)
output_gcode.close()
if __name__ == "__main__":
GCoder()