Added progress bar. Opening, parsing and plotting files (File->Open...) is now done in the background and shown in progress bar.

This commit is contained in:
Juan Pablo Caram
2014-01-12 22:07:24 -05:00
parent 1e85c54266
commit d664d40ad0
4 changed files with 401 additions and 117 deletions

View File

@@ -1,35 +1,35 @@
from gi.repository import Gtk
import matplotlib.pyplot as plt
plt.ioff()
import threading
from gi.repository import Gtk
from matplotlib.figure import Figure
from numpy import arange, sin, pi
from matplotlib.backends.backend_gtk3agg import FigureCanvasGTK3Agg as FigureCanvas
#from matplotlib.backends.backend_gtk3 import NavigationToolbar2GTK3 as NavigationToolbar
#from matplotlib.backends.backend_gtk3cairo import FigureCanvasGTK3Cairo as FigureCanvas
#from matplotlib.backends.backend_cairo import FigureCanvasCairo as FigureCanvas
#import cairo
from camlib import *
class CirkuixObj:
def __init__(self, name, kind):
self.name = name
self.kind = kind
class CirkuixGerber(CirkuixObj, Gerber):
def __init__(self, name):
Gerber.__init__(self)
CirkuixObj.__init__(self, name, "gerber")
self.fields = [{"name":"plot",
"type":bool,
"value":True,
"get":None,
"set":None,
"onchange":None},
self.fields = [{"name": "plot",
"type": bool,
"value": True,
"get": None,
"set": None,
"onchange": None},
{}]
class CirkuixExcellon(CirkuixObj, Excellon):
def __init__(self, name):
Excellon.__init__(self)
@@ -37,13 +37,17 @@ class CirkuixExcellon(CirkuixObj, Excellon):
self.options = {"plot": True,
"solid": False,
"multicolored": False}
class CirkuixCNCjob(CirkuixObj, CNCjob):
def __init__(self, name):
CNCjob.__init__(self)
def __init__(self, name, units="in", kind="generic", z_move=0.1,
feedrate=3.0, z_cut=-0.002, tooldia=0.0):
CNCjob.__init__(self, units=units, kind=kind, z_move=z_move,
feedrate=feedrate, z_cut=z_cut, tooldia=tooldia)
CirkuixObj.__init__(self, name, "cncjob")
self.options = {"plot": True}
class CirkuixGeometry(CirkuixObj, Geometry):
def __init__(self, name):
CirkuixObj.__init__(self, name, "geometry")
@@ -51,9 +55,10 @@ class CirkuixGeometry(CirkuixObj, Geometry):
"solid": False,
"multicolored": False}
class CirkuixObjForm:
def __init__(self, container, Cobj):
self.Cobj = Cobj
def __init__(self, container, cobj):
self.cobj = cobj
self.container = container
self.fields = {}
@@ -68,13 +73,16 @@ class CirkuixObjForm:
def get_entry_text(entry):
return entry.get_text()
def get_entry_int(entry):
return int(entry.get_text())
def get_entry_float(entry):
return float(entry.get_text())
def get_entry_eval(entry):
return eval(entry.get_text)
@@ -85,6 +93,7 @@ getters = {"entry_text":get_entry_text,
setters = {"entry"}
class App:
def __init__(self):
@@ -99,7 +108,10 @@ class App:
self.positionLabel = self.builder.get_object("label3")
self.grid = self.builder.get_object("grid1")
self.notebook = self.builder.get_object("notebook1")
self.info_label = self.builder.get_object("label_status")
self.progress_bar = self.builder.get_object("progressbar")
self.progress_bar.set_show_text(True)
## Event handling ##
self.builder.connect_signals(self)
@@ -115,7 +127,7 @@ class App:
########################################
## DATA ##
########################################
self.stuff = {} # CirkuixObj's by name
self.stuff = {} # CirkuixObj's by name
self.mouse = None
@@ -131,7 +143,7 @@ class App:
def plot_setup(self):
self.figure = Figure(dpi=50)
self.axes = self.figure.add_axes([0.05,0.05,0.9,0.9])
self.axes = self.figure.add_axes([0.05, 0.05, 0.9, 0.9])
self.axes.set_aspect(1)
#t = arange(0.0,5.0,0.01)
#s = sin(2*pi*t)
@@ -148,23 +160,29 @@ class App:
########################################
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.set_can_focus(True) # For key press
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.grid.attach(self.canvas,0,0,600,400)
self.grid.attach(self.canvas, 0, 0, 600, 400)
def info(self, text):
"""
Show text on the status bar.
"""
self.info_label.set_text(text)
def zoom(self, factor, center=None):
'''
"""
Zooms the plot by factor around a given
center point. Takes care of re-drawing.
'''
"""
xmin, xmax = self.axes.get_xlim()
ymin, ymax = self.axes.get_ylim()
width = xmax-xmin
height = ymax-ymin
if center == None:
if center is None:
center = [(xmin+xmax)/2.0, (ymin+ymax)/2.0]
# For keeping the point at the pointer location
@@ -225,10 +243,7 @@ class App:
def plot_cncjob(self, job):
#job.gcode_parse()
tooldia_text = self.builder.get_object("entry_tooldia").get_text()
tooldia_val = eval(tooldia_text)
job.plot2(self.axes, tooldia=tooldia_val)
job.plot2(self.axes)
self.canvas.queue_draw()
def plot_geometry(self, geometry):
@@ -331,6 +346,10 @@ class App:
self.axes.cla()
self.canvas.queue_draw()
def get_eval(self, widget_name):
value = self.builder.get_object(widget_name).get_text()
return eval(value)
########################################
## EVENT HANDLERS ##
########################################
@@ -356,7 +375,38 @@ class App:
self.stuff[iso_name] = geo
self.build_list()
def on_generate_cncjob(self, widget):
print "Generating CNC job"
# Get required info
cutz = self.get_eval("entry_geometry_cutz")
travelz = self.get_eval("entry_geometry_travelz")
feedrate = self.get_eval("entry_geometry_feedrate")
geometry = self.stuff[self.selected_item_name]
job_name = self.selected_item_name + "_cnc"
job = CirkuixCNCjob(job_name, z_move = travelz, z_cut = cutz, feedrate = feedrate)
job.generate_from_geometry(geometry.solid_geometry)
job.gcode_parse()
job.create_geometry()
# Add to App and update.
self.stuff[job_name] = job
self.build_list()
def on_cncjob_tooldia_activate(self, widget):
job = self.stuff[self.selected_item_name]
tooldia = self.get_eval("entry_cncjob_tooldia")
job.tooldia = tooldia
print "Changing tool diameter to:", tooldia
def on_cncjob_exportgcode(self, widget):
def on_success(self, filename):
cncjob = self.stuff[self.selected_item_name]
f = open(filename, 'w')
f.write(cncjob.gcode)
f.close()
print "Saved to:", filename
self.file_chooser_save_action(on_success)
def on_delete(self, widget):
self.stuff.pop(self.selected_item_name)
@@ -426,7 +476,7 @@ class App:
def file_chooser_action(self, on_success):
'''
Opens the file chooser and runs on_success
Opens the file chooser and runs on_success on a separate thread
upon completion of valid file choice.
'''
dialog = Gtk.FileChooserDialog("Please choose a file", self.window,
@@ -435,50 +485,130 @@ class App:
Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
response = dialog.run()
if response == Gtk.ResponseType.OK:
on_success(self, dialog.get_filename())
filename = dialog.get_filename()
dialog.destroy()
t = threading.Thread(target=on_success, args=(self, filename))
t.daemon = True
t.start()
#on_success(self, filename)
elif response == Gtk.ResponseType.CANCEL:
print("Cancel clicked")
dialog.destroy()
dialog.destroy()
def file_chooser_save_action(self, on_success):
'''
Opens the file chooser and runs on_success
upon completion of valid file choice.
'''
dialog = Gtk.FileChooserDialog("Save file", self.window,
Gtk.FileChooserAction.SAVE,
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
Gtk.STOCK_SAVE, Gtk.ResponseType.OK))
dialog.set_current_name("Untitled")
response = dialog.run()
if response == Gtk.ResponseType.OK:
filename = dialog.get_filename()
dialog.destroy()
on_success(self, filename)
elif response == Gtk.ResponseType.CANCEL:
print("Cancel clicked")
dialog.destroy()
def on_fileopengerber(self, param):
def on_success(self, filename):
self.progress_bar.set_text("Opening Gerber ...")
self.progress_bar.set_fraction(0.1)
name = filename.split('/')[-1].split('\\')[-1]
gerber = CirkuixGerber(name)
self.progress_bar.set_text("Parsing ...")
self.progress_bar.set_fraction(0.2)
gerber.parse_file(filename)
self.store.append([name])
#self.gerbers.append(gerber)
self.stuff[name] = gerber
self.progress_bar.set_text("Plotting ...")
self.progress_bar.set_fraction(0.6)
self.plot_gerber(gerber)
self.on_zoom_fit(None)
self.progress_bar.set_text("Done!")
self.progress_bar.set_fraction(1.0)
def clear_bar(bar):
bar.set_text("")
bar.set_fraction(0.0)
threading.Timer(1, clear_bar, args=(self.progress_bar,)).start()
self.file_chooser_action(on_success)
def on_fileopenexcellon(self, param):
def on_success(self, filename):
self.progress_bar.set_text("Opening Excellon ...")
self.progress_bar.set_fraction(0.1)
name = filename.split('/')[-1].split('\\')[-1]
excellon = CirkuixExcellon(name)
self.progress_bar.set_text("Parsing ...")
self.progress_bar.set_fraction(0.2)
excellon.parse_file(filename)
self.store.append([name])
#self.excellons.append(excellon)
self.stuff[name] = excellon
self.progress_bar.set_text("Plotting ...")
self.progress_bar.set_fraction(0.6)
self.plot_excellon(excellon)
self.on_zoom_fit(None)
self.progress_bar.set_text("Done!")
self.progress_bar.set_fraction(1.0)
def clear_bar(bar):
bar.set_text("")
bar.set_fraction(0.0)
threading.Timer(1, clear_bar, args=(self.progress_bar,)).start()
self.file_chooser_action(on_success)
def on_fileopengcode(self, param):
def on_success(self, filename):
self.progress_bar.set_text("Opening G-Code ...")
self.progress_bar.set_fraction(0.1)
name = filename.split('/')[-1].split('\\')[-1]
f = open(filename)
gcode = f.read()
f.close()
job = CirkuixCNCjob(name)
tooldia = self.get_eval("entry_tooldia")
job = CirkuixCNCjob(name, tooldia=tooldia)
job.gcode = gcode
self.progress_bar.set_text("Parsing ...")
self.progress_bar.set_fraction(0.2)
job.gcode_parse()
job.create_geometry()
self.store.append([name])
#self.cncjobs.append(job)
self.stuff[name] = job
self.progress_bar.set_text("Plotting ...")
self.progress_bar.set_fraction(0.6)
self.plot_cncjob(job)
self.on_zoom_fit(None)
self.progress_bar.set_text("Done!")
self.progress_bar.set_fraction(1.0)
def clear_bar(bar):
bar.set_text("")
bar.set_fraction(0.0)
threading.Timer(1, clear_bar, args=(self.progress_bar,)).start()
self.file_chooser_action(on_success)
def on_mouse_move_over_plot(self, event):