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

Seems to work fine now, added some more sample images

Browse Source
This commit is contained in:
MitchellHansen
2019-06-04 16:41:58 -07:00
parent aa8a92870c
commit dc85e7a58c
8 changed files with 70 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.idea/other.xml generated Normal file
View File

@@ -0,0 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="PySciProjectComponent">
<option name="PY_SCI_VIEW_SUGGESTED" value="true" />
</component>
</project>

29
GCodeRenderer.py
View File

@@ -16,11 +16,17 @@ class Renderer():
def clear_screen(self): def clear_screen(self):
self.svg_context.rectangle(0, 0, self.settings.canvas_x, self.settings.canvas_y) self.svg_surface.finish()
self.svg_context.set_source_rgba(1, 1, 1, 1.0) self.svg_surface = cairo.SVGSurface("tmp/rendered-output-t.svg", self.settings.canvas_x, self.settings.canvas_y)
self.svg_context.fill() self.svg_context = cairo.Context(self.svg_surface)
self.svg_context.set_source_rgba(0, 0, 0, 1.0) self.svg_context.scale(1, 1)
self.svg_context.stroke() self.svg_context.set_line_width(0.1)
# self.svg_context.rectangle(0, 0, self.settings.canvas_x, self.settings.canvas_y)
# self.svg_context.set_source_rgba(1, 1, 1, 1.0)
# self.svg_context.fill()
# self.svg_context.set_source_rgba(0, 0, 0, 1.0)
# self.svg_context.stroke()
# Render GCODE from the gcode-output.gcode output file that was generated in convert_gcode # Render GCODE from the gcode-output.gcode output file that was generated in convert_gcode
def render_gcode(self): def render_gcode(self):
@@ -116,18 +122,5 @@ class Renderer():
self.svg_surface = cairo.SVGSurface("tmp/rendered-output-t.svg", self.settings.canvas_x, self.settings.canvas_y) self.svg_surface = cairo.SVGSurface("tmp/rendered-output-t.svg", self.settings.canvas_x, self.settings.canvas_y)
self.svg_context = cairo.Context(self.svg_surface) self.svg_context = cairo.Context(self.svg_surface)
# def render(self):
# self.clear_screen()
# # self.render_gcode()
# #
# # if self.label is not None:
# # self.label.pack_forget()
# #
# # # Apply the rendered gcode image to the UI
# # self.image_ref = ImageTk.PhotoImage(
# # Image.frombuffer("RGBA", (self.bed_actual_x, self.bed_actual_y), self.png_surface.get_data().tobytes(), "raw", "BGRA", 0, 1))
# # self.label = Label(self, image=self.image_ref)
# # self.label.pack(expand=True, fill="both")
def toggle_flip_markers(self): def toggle_flip_markers(self):
self.settings.lift_markers = not self.settings.lift_markers self.settings.lift_markers = not self.settings.lift_markers

15
ImageConverter.py
View File

@@ -7,6 +7,8 @@ class ImageConverter:
# mkbitmap settings # mkbitmap settings
self.highpass_filter = 0 self.highpass_filter = 0
self.blur = 0 self.blur = 0
# potrace settings
self.turd = 0
# This function takes a file and runs it through mogrify, mkbitmap, and finally potrace. # This function takes a file and runs it through mogrify, mkbitmap, and finally potrace.
# The flow of the intermediate files is # The flow of the intermediate files is
@@ -39,12 +41,17 @@ class ImageConverter:
print("Running potrace...") print("Running potrace...")
start = time.time() start = time.time()
subprocess.call(["potrace",
# "-t", "0.1", potrace_args = ["potrace",
"-z", "white", "-z", "white",
"-b", "svg", "-b", "svg",
"input-images/{}-n.pbm".format(base_name), "input-images/{}-n.pbm".format(base_name),
"--rotate", "0", "--rotate", "0",
"-o", "tmp/conversion-output.svg", "-o", "tmp/conversion-output.svg"]
])
if settings.turd > 0:
potrace_args.append(["-t", settings.turd])
subprocess.call(potrace_args)
print("Run took [{:.2f}] seconds\n".format(time.time() - start)) print("Run took [{:.2f}] seconds\n".format(time.time() - start))

25
Svg2GcodeConverter.py
View File

@@ -19,9 +19,10 @@ def triangulate_lengths(settings, dest_xy):
return left_pulley_length, right_pulley_length return left_pulley_length, right_pulley_length
# http://paulbourke.net/geometry/circlesphere/
# https://math.stackexchange.com/questions/187107/calculate-coordinates-of-3rd-point-vertex-of-a-scalene-triangle-if-angles-and
# http://xaktly.com/MathNonRightTrig.html
def untriangulate_lengths(settings, x, y): def untriangulate_lengths(settings, x, y):
result = [0, 0]
r0 = x r0 = x
r1 = y r1 = y
@@ -30,6 +31,9 @@ def untriangulate_lengths(settings, x, y):
a = (pow(r0, 2) - pow(r1, 2) + pow(r2, 2)) / (2 * r2) a = (pow(r0, 2) - pow(r1, 2) + pow(r2, 2)) / (2 * r2)
h = math.sqrt(pow(r0, 2) - pow(a, 2)) h = math.sqrt(pow(r0, 2) - pow(a, 2))
a = a + settings.left_pulley_x_offset
h = h - settings.pulley_y_droop
return a, h return a, h
@@ -96,8 +100,8 @@ class Svg2GcodeConverter:
max_x_dim = max(bounding_x_max, bounding_x_min) max_x_dim = max(bounding_x_max, bounding_x_min)
max_y_dim = max(bounding_y_max, bounding_y_min) max_y_dim = max(bounding_y_max, bounding_y_min)
scale_x = self.settings.canvas_x / max_x_dim scale_x = self.settings.canvas_x / max_x_dim
scale_y = self.settings.canvas_y / max_y_dim scale_y = self.settings.canvas_y / max_y_dim
scale = min(scale_x, scale_y) scale = min(scale_x, scale_y)
print("Scaling to : {:.5f}\n".format(scale)) print("Scaling to : {:.5f}\n".format(scale))
@@ -106,8 +110,6 @@ class Svg2GcodeConverter:
gcode = "" gcode = ""
gcode += self.gcode_preamble gcode += self.gcode_preamble
current_position = (self.settings.canvas_x/2, self.settings.pulley_y_droop)
# Walk through the paths and create the GCODE # Walk through the paths and create the GCODE
for path in paths: for path in paths:
@@ -155,11 +157,8 @@ class Svg2GcodeConverter:
for i in pos: for i in pos:
evals.append(curve.evaluate(i)) evals.append(curve.evaluate(i))
#gcode += "G1 X{:.3f} Y{:.3f}\n".format(start_x, start_y)
lengths = triangulate_lengths(self.settings, (start_x, start_y)) lengths = triangulate_lengths(self.settings, (start_x, start_y))
gcode += "; Setting down tip at beginning of line ({}, {})\n".format(start_x, start_y) # gcode += "; Setting down tip at beginning of line ({}, {})\n".format(start_x, start_y)
gcode += "G1 X{:.3f} Y{:.3f}\n".format(lengths[0], lengths[1]) gcode += "G1 X{:.3f} Y{:.3f}\n".format(lengths[0], lengths[1])
gcode += "G1 Z{:.3f} \n".format(0) gcode += "G1 Z{:.3f} \n".format(0)
@@ -167,17 +166,17 @@ class Svg2GcodeConverter:
x = i[0][0] x = i[0][0]
y = i[1][0] y = i[1][0]
tmp_len = triangulate_lengths(self.settings, (x * scale, y * scale)) tmp_len = triangulate_lengths(self.settings, (x * scale, y * scale))
gcode += "; Continuing the line ({}, {})\n".format(x * scale, y * scale) # gcode += "; Continuing the line ({}, {})\n".format(x * scale, y * scale)
gcode += "G1 X{:.3f} Y{:.3f}\n".format(tmp_len[0], tmp_len[1]) gcode += "G1 X{:.3f} Y{:.3f}\n".format(tmp_len[0], tmp_len[1])
if isinstance(part, Line): if isinstance(part, Line):
start_len = triangulate_lengths(self.settings, (start_x, start_y)) start_len = triangulate_lengths(self.settings, (start_x, start_y))
end_len = triangulate_lengths(self.settings, (end_x, end_y)) end_len = triangulate_lengths(self.settings, (end_x, end_y))
gcode += "; Setting down tip at beginning of line ({}, {})\n".format(start_x, start_y) # gcode += "; Setting down tip at beginning of line ({}, {})\n".format(start_x, start_y)
gcode += "G1 X{:.3f} Y{:.3f}\n".format(start_len[0], start_len[1]) gcode += "G1 X{:.3f} Y{:.3f}\n".format(start_len[0], start_len[1])
gcode += "G1 Z{:.3f} \n".format(0) gcode += "G1 Z{:.3f} \n".format(0)
gcode += "; Moving tip to the end of the line ({}, {})\n".format(end_x, end_y) # gcode += "; Moving tip to the end of the line ({}, {})\n".format(end_x, end_y)
gcode += "G1 X{:.3f} Y{:.3f}\n".format(end_len[0], end_len[1]) gcode += "G1 X{:.3f} Y{:.3f}\n".format(end_len[0], end_len[1])
gcode += self.gcode_end gcode += self.gcode_end

BIN
input-images/funky-bird.jpg Executable file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 392 KiB

BIN
input-images/leaf.jpg Executable file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 465 KiB

BIN
input-images/new-york.jpg Executable file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 1.6 MiB

46
main.py
View File

@@ -11,7 +11,6 @@ from ImageConverter import ImageConverter
from Simulator import Simulator from Simulator import Simulator
class Settings: class Settings:
def __init__(self): def __init__(self):
@@ -19,8 +18,8 @@ class Settings:
# ============ HARDCODED VALUES =========== # ============ HARDCODED VALUES ===========
# Canvas size # Canvas size
self.canvas_x = 300 self.canvas_x = 1000
self.canvas_y = 300 self.canvas_y = 1000
# The position of the pulley centers in relation to the top left and right of the canvas # The position of the pulley centers in relation to the top left and right of the canvas
self.left_pulley_x_offset = -40 self.left_pulley_x_offset = -40
@@ -50,6 +49,8 @@ class Tracer(Tk):
def update_blur_value(self, value): def update_blur_value(self, value):
self.blur = value self.blur = value
def update_turd_value(self, value):
self.turd = value
def __init__(self): def __init__(self):
@@ -83,17 +84,20 @@ class Tracer(Tk):
self.image_ref = None self.image_ref = None
# Initialize TK # Initialize TK
self.geometry("{}x{}".format(500, 500)) self.geometry("{}x{}".format(800, 800))
self.n = Notebook(self, width= 400, height =400) self.tab_bar = Notebook(self, width= 400, height =400)
self.n.pack(fill=BOTH, expand=1) self.tab_bar.pack(fill=BOTH, expand=1)
self.f1 = Frame(self.n) self.converted_image_tab = Frame(self.tab_bar)
self.f2 = Frame(self.n) self.original_image_tab = Frame(self.tab_bar)
self.rightframe = Frame(self) self.rightframe = Frame(self)
self.rightframe.pack(side=RIGHT) self.rightframe.pack(side=RIGHT)
self.centerframe = Frame(self)
self.centerframe.pack(side=BOTTOM)
self.image_select_button = Button(self.rightframe, text="Select Image", command=self.file_select_callback) self.image_select_button = Button(self.rightframe, text="Select Image", command=self.file_select_callback)
self.image_select_button.pack() self.image_select_button.pack()
@@ -106,14 +110,24 @@ class Tracer(Tk):
self.lift_markers_checkbox = Checkbutton(self.rightframe, text="Lift Markers", command=self.cairo_renderer.toggle_flip_markers) self.lift_markers_checkbox = Checkbutton(self.rightframe, text="Lift Markers", command=self.cairo_renderer.toggle_flip_markers)
self.lift_markers_checkbox.pack() self.lift_markers_checkbox.pack()
self.highpass_slider = Scale(self.rightframe, command=self.update_highpass_value, resolution=0.1, to=15) self.highpass_label = Label(self.centerframe, text="Highpass filter", fg="black")
self.highpass_label.pack()
self.highpass_slider = Scale(self.centerframe, command=self.update_highpass_value, resolution=0.0, to=15, orient=HORIZONTAL)
self.highpass_slider.set(self.image_converter_settings.highpass_filter) self.highpass_slider.set(self.image_converter_settings.highpass_filter)
self.highpass_slider.pack() self.highpass_slider.pack()
self.blur_slider = Scale(self.rightframe, command=self.update_blur_value, resolution=0.1, to=5) self.blur_label = Label(self.centerframe, text="Blur", fg="black")
self.blur_label.pack()
self.blur_slider = Scale(self.centerframe, command=self.update_blur_value, resolution=0.0, to=5, orient=HORIZONTAL)
self.blur_slider.set(self.image_converter_settings.blur) self.blur_slider.set(self.image_converter_settings.blur)
self.blur_slider.pack() self.blur_slider.pack()
self.turd_label = Label(self.centerframe, text="Turds", fg="black")
self.turd_label.pack()
self.turd_slider = Scale(self.centerframe, command=self.update_turd_value, resolution=0.0, to=5, orient=HORIZONTAL)
self.turd_slider.set(self.image_converter_settings.turd)
self.turd_slider.pack()
# Start TK # Start TK
self.mainloop() self.mainloop()
@@ -139,8 +153,8 @@ class Tracer(Tk):
self.cairo_renderer.clear_screen() self.cairo_renderer.clear_screen()
self.cairo_renderer.render_gcode() self.cairo_renderer.render_gcode()
self.f1.pack_forget() self.converted_image_tab.pack_forget()
self.f2.pack_forget() self.original_image_tab.pack_forget()
if self.label is not None: if self.label is not None:
self.label.pack_forget() self.label.pack_forget()
@@ -152,14 +166,14 @@ class Tracer(Tk):
# scale = self.winfo_width() / pil_image.width # scale = self.winfo_width() / pil_image.width
# pil_image = pil_image.resize((int(scale * pil_image.width), int(scale * pil_image.height))) # pil_image = pil_image.resize((int(scale * pil_image.width), int(scale * pil_image.height)))
self.image_ref = ImageTk.PhotoImage(pil_image) self.image_ref = ImageTk.PhotoImage(pil_image)
self.label = Label(self.f1, image=self.image_ref) self.label = Label(self.converted_image_tab, image=self.image_ref)
self.n.add(self.f1, text="Converted") self.tab_bar.add(self.converted_image_tab, text="Converted")
self.label.pack(expand=True, fill="both") self.label.pack(expand=True, fill="both")
self.pic = ImageTk.PhotoImage(file="input-images/{}".format(self.filename)) self.pic = ImageTk.PhotoImage(file="input-images/{}".format(self.filename))
self.label1 = Label(self.f2, image=self.pic) self.label1 = Label(self.original_image_tab, image=self.pic)
self.n.add(self.f2, text="Original") self.tab_bar.add(self.original_image_tab, text="Original")
self.label1.pack(expand=True, fill="both") self.label1.pack(expand=True, fill="both")
def render_simulation(self): def render_simulation(self):