bartool/flatcam-wsl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed g-code arc parse/plot

Browse Source
This commit is contained in:
Juan Pablo Caram
2014-01-08 01:46:50 -05:00
parent 3cb9e444c0
commit 145496b4ae
9 changed files with 546 additions and 423 deletions
Download Patch File Download Diff File Expand all files Collapse all files

340
cirkuix.py
View File

@@ -29,22 +29,26 @@ class App:
## Event handling ##
self.builder.connect_signals(self)
## Make plot area ##
self.figure = None
self.axes = None
self.canvas = None
## Make plot area ##
self.mplpaint()
self.window.show_all()
########################################
## DATA ##
########################################
self.gerbers = []
self.excellons = []
self.cncjobs = []
self.mouse = None
########################################
## START ##
########################################
self.window.show_all()
Gtk.main()
def mplpaint(self):
@@ -58,169 +62,24 @@ class App:
self.axes.grid()
#a.patch.set_visible(False) Background of the axes
self.figure.patch.set_visible(False)
#self.figure.tight_layout()
self.canvas = FigureCanvas(self.figure) # a Gtk.DrawingArea
#self.canvas.set_size_request(600,400)
self.canvas.set_hexpand(1)
self.canvas.set_vexpand(1)
########################################
## EVENTS ##
########################################
self.canvas.mpl_connect('button_press_event', self.on_click_over_plot)
self.canvas.mpl_connect('motion_notify_event', self.on_mouse_move_over_plot)
##self.canvas.mpl_connect('scroll_event', self.on_scroll_over_plot)
##self.canvas.mpl_connect('key_press_event', self.on_key_over_plot)
self.canvas.set_hexpand(1)
self.canvas.set_vexpand(1)
self.canvas.set_can_focus(True) # For key press
self.canvas.mpl_connect('key_press_event', self.on_key_over_plot)
self.canvas.mpl_connect('scroll_event', self.on_scroll_over_plot)
#self.builder.get_object("viewport2").add(self.canvas)
self.grid.attach(self.canvas,0,0,600,400)
#self.builder.get_object("scrolledwindow1").add(self.canvas)
def on_filequit(self, param):
print "quit from menu"
self.window.destroy()
Gtk.main_quit()
def on_closewindow(self, param):
print "quit from X"
self.window.destroy()
Gtk.main_quit()
def on_fileopengerber(self, param):
print "File->Open Gerber"
dialog = Gtk.FileChooserDialog("Please choose a file", self.window,
Gtk.FileChooserAction.OPEN,
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
response = dialog.run()
if response == Gtk.ResponseType.OK:
## Load the file ##
print("Open clicked")
print("File selected: " + dialog.get_filename())
gerber = Gerber()
gerber.parse_file(dialog.get_filename())
self.gerbers.append(gerber)
self.plot_gerber(gerber)
## End ##
elif response == Gtk.ResponseType.CANCEL:
print("Cancel clicked")
dialog.destroy()
def on_fileopenexcellon(self, param):
print "File->Open Excellon"
dialog = Gtk.FileChooserDialog("Please choose a file", self.window,
Gtk.FileChooserAction.OPEN,
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
response = dialog.run()
if response == Gtk.ResponseType.OK:
## Load the file ##
print("Open clicked")
print("File selected: " + dialog.get_filename())
excellon = Excellon()
excellon.parse_file(dialog.get_filename())
self.excellons.append(excellon)
self.plot_excellon(excellon)
## End ##
elif response == Gtk.ResponseType.CANCEL:
print("Cancel clicked")
dialog.destroy()
def plot_gerber(self, gerber):
gerber.create_geometry()
# Options
mergepolys = self.builder.get_object("cb_mergepolys").get_active()
multicolored = self.builder.get_object("cb_multicolored").get_active()
geometry = None
if mergepolys:
geometry = gerber.solid_geometry
else:
geometry = gerber.buffered_paths + \
[poly['polygon'] for poly in gerber.regions] + \
gerber.flash_geometry
linespec = None
if multicolored:
linespec = '-'
else:
linespec = 'k-'
#f = Figure(dpi=75)
#a = f.add_subplot(111)
#a.set_aspect(1)
for poly in geometry:
x, y = poly.exterior.xy
#a.plot(x, y)
self.axes.plot(x, y, linespec)
for ints in poly.interiors:
x, y = ints.coords.xy
self.axes.plot(x, y, linespec)
#f.tight_layout()
#canvas = FigureCanvas(f) # a Gtk.DrawingArea
#canvas.set_size_request(600,400)
#self.grid.attach(canvas,1,1,600,400)
#self.window.show_all()
def plot_excellon(self, excellon):
excellon.create_geometry()
# Plot excellon
for geo in excellon.solid_geometry:
x, y = geo.exterior.coords.xy
self.axes.plot(x, y, 'r-')
for ints in geo.interiors:
x, y = ints.coords.xy
self.axes.plot(x, y, 'g-')
def on_mouse_move_over_plot(self, event):
try: # May fail in case mouse not within axes
self.positionLabel.set_label("X: %.4f Y: %.4f"%(
event.xdata, event.ydata))
self.mouse = [event.xdata, event.ydata]
except:
self.positionLabel.set_label("X: --- Y: ---")
self.mouse = None
def on_click_over_plot(self, event):
print 'button=%d, x=%d, y=%d, xdata=%f, ydata=%f'%(
event.button, event.x, event.y, event.xdata, event.ydata)
def get_bounds(self):
xmin = Inf
ymin = Inf
xmax = -Inf
ymax = -Inf
geometry_sets = [self.gerbers, self.excellons]
for gs in geometry_sets:
for g in gs:
gxmin, gymin, gxmax, gymax = g.solid_geometry.bounds
xmin = min([xmin, gxmin])
ymin = min([ymin, gymin])
xmax = max([xmax, gxmax])
ymax = max([ymax, gymax])
return [xmin, ymin, xmax, ymax]
def on_zoom_in(self, event):
self.zoom(1.5)
return
def on_zoom_out(self, event):
self.zoom(1/1.5)
def on_zoom_fit(self, event):
xmin, ymin, xmax, ymax = self.get_bounds()
width = xmax-xmin
height = ymax-ymin
self.axes.set_xlim((xmin-0.05*width, xmax+0.05*width))
self.axes.set_ylim((ymin-0.05*height, ymax+0.05*height))
self.canvas.queue_draw()
return
def zoom(self, factor, center=None):
xmin, xmax = self.axes.get_xlim()
ymin, ymax = self.axes.get_ylim()
@@ -241,34 +100,163 @@ class App:
self.axes.set_ylim((center[1]-new_height*(1-rely), center[1]+new_height*rely))
self.canvas.queue_draw()
def plot_gerber(self, gerber):
gerber.create_geometry()
# def on_scroll_over_plot(self, event):
# print "Scroll"
# center = [event.xdata, event.ydata]
# if sign(event.step):
# self.zoom(1.5, center=center)
# else:
# self.zoom(1/1.5, center=center)
#
# def on_window_scroll(self, event):
# print "Scroll"
#
# def on_key_over_plot(self, event):
# print 'you pressed', event.key, event.xdata, event.ydata
# Options
mergepolys = self.builder.get_object("cb_mergepolys").get_active()
multicolored = self.builder.get_object("cb_multicolored").get_active()
def on_window_key_press(self, widget, event):
print event.get_keycode(), event.get_keyval()
val = int(event.get_keyval()[1])
geometry = None
if mergepolys:
geometry = gerber.solid_geometry
else:
geometry = gerber.buffered_paths + \
[poly['polygon'] for poly in gerber.regions] + \
gerber.flash_geometry
if val == 49: # 1
linespec = None
if multicolored:
linespec = '-'
else:
linespec = 'k-'
for poly in geometry:
x, y = poly.exterior.xy
#a.plot(x, y)
self.axes.plot(x, y, linespec)
for ints in poly.interiors:
x, y = ints.coords.xy
self.axes.plot(x, y, linespec)
def plot_excellon(self, excellon):
excellon.create_geometry()
# Plot excellon
for geo in excellon.solid_geometry:
x, y = geo.exterior.coords.xy
self.axes.plot(x, y, 'r-')
for ints in geo.interiors:
x, y = ints.coords.xy
self.axes.plot(x, y, 'g-')
def plot_cncjob(self, job):
job.create_gcode_geometry()
tooldia_text = self.builder.get_object("entry_tooldia").get_text()
tooldia_val = eval(tooldia_text)
job.plot2(self.axes, tooldia=tooldia_val)
return
def file_chooser_action(self, on_success):
dialog = Gtk.FileChooserDialog("Please choose a file", self.window,
Gtk.FileChooserAction.OPEN,
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
response = dialog.run()
if response == Gtk.ResponseType.OK:
on_success(self, dialog.get_filename())
elif response == Gtk.ResponseType.CANCEL:
print("Cancel clicked")
dialog.destroy()
########################################
## EVENT HANDLERS ##
########################################
def on_filequit(self, param):
print "quit from menu"
self.window.destroy()
Gtk.main_quit()
def on_closewindow(self, param):
print "quit from X"
self.window.destroy()
Gtk.main_quit()
def on_fileopengerber(self, param):
def on_success(self, filename):
gerber = Gerber()
gerber.parse_file(filename)
self.gerbers.append(gerber)
self.plot_gerber(gerber)
self.file_chooser_action(on_success)
def on_fileopenexcellon(self, param):
def on_success(self, filename):
excellon = Excellon()
excellon.parse_file(filename)
self.excellons.append(excellon)
self.plot_excellon(excellon)
self.file_chooser_action(on_success)
def on_fileopengcode(self, param):
def on_success(self, filename):
f = open(filename)
gcode = f.read()
f.close()
job = CNCjob()
job.gcode = gcode
self.cncjobs.append(job)
self.plot_cncjob(job)
self.file_chooser_action(on_success)
def on_mouse_move_over_plot(self, event):
try: # May fail in case mouse not within axes
self.positionLabel.set_label("X: %.4f Y: %.4f"%(
event.xdata, event.ydata))
self.mouse = [event.xdata, event.ydata]
except:
self.positionLabel.set_label("")
self.mouse = None
def on_click_over_plot(self, event):
# For key presses
self.canvas.grab_focus()
print 'button=%d, x=%d, y=%d, xdata=%f, ydata=%f'%(
event.button, event.x, event.y, event.xdata, event.ydata)
def on_zoom_in(self, event):
self.zoom(1.5)
return
def on_zoom_out(self, event):
self.zoom(1/1.5)
def on_zoom_fit(self, event):
xmin, ymin, xmax, ymax = get_bounds([self.gerbers, self.excellons])
width = xmax-xmin
height = ymax-ymin
self.axes.set_xlim((xmin-0.05*width, xmax+0.05*width))
self.axes.set_ylim((ymin-0.05*height, ymax+0.05*height))
self.canvas.queue_draw()
return
def on_scroll_over_plot(self, event):
print "Scroll"
center = [event.xdata, event.ydata]
if sign(event.step):
self.zoom(1.5, center=center)
else:
self.zoom(1/1.5, center=center)
def on_window_scroll(self, event):
print "Scroll"
def on_key_over_plot(self, event):
print 'you pressed', event.key, event.xdata, event.ydata
if event.key == '1': # 1
self.on_zoom_fit(None)
return
if val == 50: # 2
if event.key == '2': # 2
self.zoom(1/1.5, self.mouse)
return
if val == 51: # 3
if event.key == '3': # 3
self.zoom(1.5, self.mouse)
return