FlatCAM/cirkuix.py

import threading
from gi.repository import Gtk
from gi.repository import Gdk
from gi.repository import GLib

from matplotlib.figure import Figure
from numpy import arange, sin, pi
from matplotlib.backends.backend_gtk3agg import FigureCanvasGTK3Agg as FigureCanvas
#from matplotlib.backends.backend_gtk3cairo import FigureCanvasGTK3Cairo as FigureCanvas
#from matplotlib.backends.backend_cairo import FigureCanvasCairo as FigureCanvas

from camlib import *


class CirkuixObj:
    form_getters = {}

    form_setters = {}

    # Instance of the application to which these are related.
    # The app should set this value.
    # TODO: Move the definitions of form_getters and form_setters
    # TODO: to their respective classes and use app to reference
    # TODO: the builder.
    app = None

    def __init__(self, name):
        self.name = name
        self.axes = None  # Matplotlib axes
        self.options = {}

    def setup_axes(self, figure):
        if self.axes is None:
            self.axes = figure.add_axes([0.05, 0.05, 0.9, 0.9], label=self.name)
        elif self.axes not in figure.axes:
            figure.add_axes(self.axes)

        self.axes.patch.set_visible(False)  # No background
        self.axes.set_aspect(1)

        return self.axes

    def set_options(self, options):
        for name in options:
            self.options[name] = options[name]
        return

    def to_form(self):
        for name in self.options:
            if name in self.form_setters:
                self.form_setters[name](self.options[name])

    def read_form(self):
        for name in self.form_getters:
            self.options[name] = self.form_getters[name]()


class CirkuixGerber(CirkuixObj, Gerber):

    def __init__(self, name):
        Gerber.__init__(self)
        CirkuixObj.__init__(self, name)

        self.options = {
            "plot": True,
            "mergepolys": True,
            "multicolored": False,
            "solid": False,
            "isotooldia": 0.4/25.4,
            "cutoutmargin": 0.2,
            "cutoutgapsize": 0.15,
            "gaps": "tb",
            "noncoppermargin": 0.0
        }

    def build_ui(self):
        print "cirkuixgerber.build_ui()"
        osw = self.app.builder.get_object("offscrwindow_gerber")
        #box1 = self.app.builder.get_object("box_gerber")
        #osw.remove(box1)
        sw = self.app.builder.get_object("sw_gerber")
        osw.remove(sw)
        vp = self.app.builder.get_object("vp_gerber")
        vp.override_background_color(Gtk.StateType.NORMAL, Gdk.RGBA(1, 1, 1, 1))
        self.app.notebook.append_page(sw, Gtk.Label("Selection"))
        entry_name = self.app.builder.get_object("entry_gerbername")
        entry_name.set_text(self.name)
        entry_name.connect("activate", self.app.on_activate_name)
        self.to_form()
        sw.show()

    def plot(self, figure):
        self.setup_axes(figure)

        self.create_geometry()

        geometry = None
        if self.options["mergepolys"]:
            geometry = self.solid_geometry
        else:
            geometry = self.buffered_paths + \
                        [poly['polygon'] for poly in self.regions] + \
                        self.flash_geometry

        linespec = None
        if self.options["multicolored"]:
            linespec = '-'
        else:
            linespec = 'k-'

        for poly in geometry:
            x, y = poly.exterior.xy
            self.axes.plot(x, y, linespec)
            for ints in poly.interiors:
                x, y = ints.coords.xy
                self.axes.plot(x, y, linespec)


class CirkuixExcellon(CirkuixObj, Excellon):

    def __init__(self, name):
        Excellon.__init__(self)
        CirkuixObj.__init__(self, name)

        self.options = {
            "plot": True,
            "solid": False,
            "multicolored": False
        }

    def build_ui(self):
        print "build_excellon_ui()"
        osw = self.app.builder.get_object("offscrwindow_excellon")
        #box1 = self.app.builder.get_object("box_excellon")
        #osw.remove(box1)
        sw = self.app.builder.get_object("sw_excellon")
        osw.remove(sw)
        vp = self.app.builder.get_object("vp_excellon")
        vp.override_background_color(Gtk.StateType.NORMAL, Gdk.RGBA(1, 1, 1, 1))
        self.app.notebook.append_page(sw, Gtk.Label("Selection"))
        entry_name = self.app.builder.get_object("entry_excellonname")
        entry_name.set_text(self.name)
        entry_name.connect("activate", self.app.on_activate_name)
        self.to_form()
        sw.show()

    def plot(self, figure):
        self.setup_axes(figure)
        self.create_geometry()

        # Plot excellon
        for geo in self.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-')


class CirkuixCNCjob(CirkuixObj, CNCjob):
    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)

        self.options = {
            "plot": True,
            "solid": False,
            "tooldia": 0.4/25.4  # 0.4mm in inches
        }

    def build_ui(self):
        print "build_cncjob_ui()"
        osw = self.app.builder.get_object("offscrwindow_cncjob")
        #box1 = self.app.builder.get_object("box_cncjob")
        #osw.remove(box1)
        sw = self.app.builder.get_object("sw_cncjob")
        osw.remove(sw)
        vp = self.app.builder.get_object("vp_cncjob")
        vp.override_background_color(Gtk.StateType.NORMAL, Gdk.RGBA(1, 1, 1, 1))
        self.app.notebook.append_page(sw, Gtk.Label("Selection"))
        entry_name = self.app.builder.get_object("entry_cncjobname")
        entry_name.set_text(self.name)
        entry_name.connect("activate", self.app.on_activate_name)
        self.to_form()
        sw.show()

    def plot(self, figure):
        self.setup_axes(figure)
        self.plot2(self.axes, tooldia=self.options["tooldia"])


class CirkuixGeometry(CirkuixObj, Geometry):
    def __init__(self, name):
        CirkuixObj.__init__(self, name)
        self.options = {"plot": True,
                        "solid": False,
                        "multicolored": False}

        self.options = {
            "plot": True,
            "solid": False,
            "multicolored": False,
            "cutz": -0.002,
            "travelz": 0.1,
            "feedrate": 5.0
        }

    def build_ui(self):
        print "build_geometry_ui()"
        osw = self.app.builder.get_object("offscrwindow_geometry")
        #box1 = self.app.builder.get_object("box_geometry")
        #osw.remove(box1)
        sw = self.app.builder.get_object("sw_geometry")
        osw.remove(sw)
        vp = self.app.builder.get_object("vp_geometry")
        vp.override_background_color(Gtk.StateType.NORMAL, Gdk.RGBA(1, 1, 1, 1))
        self.app.notebook.append_page(sw, Gtk.Label("Selection"))
        entry_name = self.app.builder.get_object("entry_geometryname")
        entry_name.set_text(self.name)
        entry_name.connect("activate", self.app.on_activate_name)
        self.to_form()
        sw.show()

    def plot(self, figure):
        self.setup_axes(figure)

        for geo in self.solid_geometry:

            if type(geo) == Polygon:
                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, 'r-')
                continue

            if type(geo) == LineString or type(geo) == LinearRing:
                x, y = geo.coords.xy
                self.axes.plot(x, y, 'r-')
                continue


class App:
    def __init__(self):
        # Needed to interact with the GUI from other threads.
        GLib.threads_init()

        ########################################
        ##                GUI                 ##
        ########################################       
        self.gladefile = "cirkuix.ui"
        self.builder = Gtk.Builder()
        self.builder.add_from_file(self.gladefile)
        self.window = self.builder.get_object("window1")
        self.window.set_title("Cirkuix")
        self.position_label = 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)
        
        ## Make plot area ##
        self.figure = None
        self.axes = None
        self.canvas = None
        self.setup_plot()
        
        self.setup_component_viewer()
        self.setup_component_editor()
        
        ########################################
        ##               DATA                 ##
        ########################################
        self.setup_obj_classes()
        self.stuff = {}    # CirkuixObj's by name
        self.mouse = None  # Mouse coordinates over plot
        
        # What is selected by the user. It is
        # a key if self.stuff
        self.selected_item_name = None

        # For debugging only
        def someThreadFunc(self):
            print "Hello World!"
        t = threading.Thread(target=someThreadFunc, args=(self,))
        t.start()

        ########################################
        ##              START                 ##
        ########################################
        self.window.show_all()
        Gtk.main()
        
    def setup_plot(self):
        self.figure = Figure(dpi=50)
        self.axes = self.figure.add_axes([0.05, 0.05, 0.9, 0.9], label="base", alpha=0.0)
        self.axes.set_aspect(1)
        #t = arange(0.0,5.0,0.01)
        #s = sin(2*pi*t)
        #self.axes.plot(t,s)
        self.axes.grid()
        self.figure.patch.set_visible(False)
        
        self.canvas = FigureCanvas(self.figure)  # a Gtk.DrawingArea
        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.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)

    def setup_obj_classes(self):
        CirkuixObj.app = self

        CirkuixGerber.form_getters = {
            "plot": self.builder.get_object("cb_gerber_plot").get_active,
            "mergepolys": self.builder.get_object("cb_gerber_mergepolys").get_active,
            "solid": self.builder.get_object("cb_gerber_solid").get_active,
            "multicolored": self.builder.get_object("cb_gerber_multicolored").get_active,
            "isotooldia": lambda: self.get_eval("entry_gerberisotooldia"),
            "cutoutmargin": lambda: self.get_eval("entry_gerber_cutout_margin"),
            "cutoutgapsize": lambda: self.get_eval("entry_gerber_cutout_gapsize"),
            "gaps": lambda: self.get_radio_value({"rb_2tb": "tb", "rb_2lr": "lr", "rb_4": "4"}),
            "noncoppermargin": lambda: self.get_eval("entry_gerber_noncopper_margin")
        }

        CirkuixGerber.form_setters = {
            "plot": self.builder.get_object("cb_gerber_plot").set_active,
            "mergepolys": self.builder.get_object("cb_gerber_mergepolys").set_active,
            "solid": self.builder.get_object("cb_gerber_solid").set_active,
            "multicolored": self.builder.get_object("cb_gerber_multicolored").set_active,
            "isotooldia": lambda x: self.builder.get_object("entry_gerberisotooldia").set_text(str(x)),
            "cutoutmargin": lambda x: self.builder.get_object("entry_gerber_cutout_margin").set_text(str(x)),
            "cutoutgapsize": lambda x: self.builder.get_object("entry_gerber_cutout_gapsize").set_text(str(x)),
            "gaps": lambda x: self.builder.get_object("cb_gerber_solid").set_active(
                                    {"tb": "rb_2tb", "lr": "rb_2lr", "4": "rb_4"}[x]),
            "noncoppermargin": lambda x: self.builder.get_object("entry_gerber_noncopper_margin").set_text(str(x))
        }

        CirkuixExcellon.form_getters = {
            "plot": self.builder.get_object("cb_excellon_plot").get_active,
            "solid": self.builder.get_object("cb_excellon_solid").get_active,
            "multicolored": self.builder.get_object("cb_excellon_multicolored").get_active
        }

        CirkuixExcellon.form_setters = {
            "plot": self.builder.get_object("cb_excellon_plot").set_active,
            "solid": self.builder.get_object("cb_excellon_solid").set_active,
            "multicolored": self.builder.get_object("cb_excellon_multicolored").set_active
        }

        CirkuixCNCjob.form_getters = {
            "plot": self.builder.get_object("cb_cncjob_plot").get_active,
            "solid": self.builder.get_object("cb_cncjob_solid").get_active,
            "multicolored": self.builder.get_object("cb_cncjob_multicolored").get_active,
            "tooldia": lambda: self.get_eval("entry_cncjob_tooldia")
        }

        CirkuixCNCjob.form_setters = {
            "plot": self.builder.get_object("cb_cncjob_plot").set_active,
            "solid": self.builder.get_object("cb_cncjob_solid").set_active,
            "tooldia": lambda x: self.builder.get_object("entry_cncjob_tooldia").set_text(str(x))
        }

        CirkuixGeometry.form_getters = {
            "plot": self.builder.get_object("cb_geometry_plot").get_active,
            "solid": self.builder.get_object("cb_geometry_solid").get_active,
            "multicolored": self.builder.get_object("cb_geometry_multicolored").get_active,
            "cutz": lambda: self.get_eval("entry_geometry_cutz"),
            "travelz": lambda: self.get_eval("entry_geometry_travelz"),
            "feedrate": lambda: self.get_eval("entry_geometry_feedrate")
        }

        CirkuixGeometry.form_setters = {
            "plot": self.builder.get_object("cb_geometry_plot").set_active,
            "solid": self.builder.get_object("cb_geometry_solid").set_active,
            "multicolored": self.builder.get_object("cb_geometry_multicolored").set_active,
            "cutz": lambda x: self.builder.get_object("entry_geometry_cutz").set_text(str(x)),
            "travelz": lambda x: self.builder.get_object("entry_geometry_travelz").set_text(str(x)),
            "feedrate": lambda x: self.builder.get_object("entry_geometry_feedrate").set_text(str(x))
        }

    def setup_component_viewer(self):
        """
        List or Tree where whatever has been loaded or created is
        displayed.
        """
        self.store = Gtk.ListStore(str)
        self.tree = Gtk.TreeView(self.store)
        #self.list = Gtk.ListBox()
        self.tree_select  = self.tree.get_selection()
        self.signal_id = self.tree_select.connect("changed", self.on_tree_selection_changed)
        renderer = Gtk.CellRendererText()
        column = Gtk.TreeViewColumn("Title", renderer, text=0)
        self.tree.append_column(column)
        self.builder.get_object("notebook1").append_page(self.tree, Gtk.Label("Project"))
        
    def setup_component_editor(self):
        box1 = Gtk.Box(Gtk.Orientation.VERTICAL)
        label1 = Gtk.Label("Choose an item from Project")
        box1.pack_start(label1, False, False, 1)
        self.builder.get_object("notebook1").append_page(box1, Gtk.Label("Selection"))

    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 is None:
            center = [(xmin+xmax)/2.0, (ymin+ymax)/2.0]

        # For keeping the point at the pointer location
        relx = (xmax-center[0])/width
        rely = (ymax-center[1])/height

        new_width = width/factor
        new_height = height/factor

        xmin = center[0]-new_width*(1-relx)
        xmax = center[0]+new_width*relx
        ymin = center[1]-new_height*(1-rely)
        ymax = center[1]+new_height*rely

        for name in self.stuff:
            self.stuff[name].axes.set_xlim((xmin, xmax))
            self.stuff[name].axes.set_ylim((ymin, ymax))
        self.axes.set_xlim((xmin, xmax))
        self.axes.set_ylim((ymin, ymax))

        self.canvas.queue_draw()

    def build_list(self):
        self.store.clear()
        for key in self.stuff:
            self.store.append([key])

    def get_radio_value(self, radio_set):
        """
        Returns the radio_set[key] if the radiobutton
        whose name is key is active.
        """
        for name in radio_set:
            if self.builder.get_object(name).get_active():
                return radio_set[name]

    def plot_all(self):
        self.clear_plots()
        
        for i in self.stuff:
            self.stuff[i].plot(self.figure)
        
        self.on_zoom_fit(None)
        self.axes.grid()
        self.canvas.queue_draw()
        
    def clear_plots(self):
        self.axes.cla()
        self.figure.clf()
        self.figure.add_axes(self.axes)
        self.canvas.queue_draw()

    def get_eval(self, widget_name):
        value = self.builder.get_object(widget_name).get_text()
        return eval(value)

    def set_list_selection(self, name):
        iter = self.store.get_iter_first()
        while iter is not None and self.store[iter][0] != name:
            iter = self.store.iter_next(iter)
        self.tree_select.unselect_all()
        self.tree_select.select_iter(iter)

    ########################################
    ##         EVENT HANDLERS             ##
    ########################################
    def on_gerber_generate_noncopper(self, widget):
        gerber = self.stuff[self.selected_item_name]
        gerber.read_form()

        name = self.selected_item_name + "_noncopper"

        bounding_box = gerber.solid_geometry.envelope.buffer(gerber.options["noncoppermargin"])

        non_copper = bounding_box.difference(gerber.solid_geometry)

        geometry = CirkuixGeometry(name)
        geometry.solid_geometry = non_copper

        self.stuff[name] = geometry
        self.build_list()


    def on_gerber_generate_cutout(self, widget):
        margin = self.get_eval("entry_gerber_cutout_margin")
        gap_size = self.get_eval("entry_gerber_cutout_gapsize")
        gerber = self.stuff[self.selected_item_name]
        minx, miny, maxx, maxy = gerber.bounds()
        minx -= margin
        maxx += margin
        miny -= margin
        maxy += margin
        midx = 0.5 * (minx + maxx)
        midy = 0.5 * (miny + maxy)
        hgap = 0.5 * gap_size
        pts = [[midx-hgap, maxy],
               [minx, maxy],
               [minx, midy+hgap],
               [minx, midy-hgap],
               [minx, miny],
               [midx-hgap, miny],
               [midx+hgap, miny],
               [maxx, miny],
               [maxx, midy-hgap],
               [maxx, midy+hgap],
               [maxx, maxy],
               [midx+hgap, maxy]]
        cases = {"tb": [[pts[0], pts[1], pts[4], pts[5]],
                        [pts[6], pts[7], pts[10], pts[11]]],
                 "lr": [[pts[9], pts[10], pts[1], pts[2]],
                        [pts[3], pts[4], pts[7], pts[8]]],
                 "4": [[pts[0], pts[1], pts[2]],
                       [pts[3], pts[4], pts[5]],
                       [pts[6], pts[7], pts[8]],
                       [pts[9], pts[10], pts[11]]]}
        name = self.selected_item_name + "_cutout"
        geometry = CirkuixGeometry(name)
        cuts = None
        if self.builder.get_object("rb_2tb").get_active():
            cuts = cases["tb"]
        elif self.builder.get_object("rb_2lr").get_active():
            cuts = cases["lr"]
        else:
            cuts = cases["4"]
        geometry.solid_geometry = cascaded_union([LineString(segment) for segment in cuts])

        # Add to App and update.
        self.stuff[name] = geometry
        self.build_list()

    def on_eval_update(self, widget):
        """
        Modifies the content of a Gtk.Entry by running
        eval() on its contents and puting it back as a
        string.
        """
        # TODO: error handling here
        widget.set_text(str(eval(widget.get_text())))

    def on_generate_isolation(self, widget):
        print "Generating Isolation Geometry:"
        # Get required info
        tooldia = self.builder.get_object("entry_gerberisotooldia").get_text()
        tooldia = eval(tooldia)
        print "tooldia:", tooldia
        
        # Generate
        iso = self.stuff[self.selected_item_name].isolation_geometry(tooldia/2.0)
        # TODO: This will break if there is something with this name already        
        iso_name = self.selected_item_name + "_iso"
        geo = CirkuixGeometry(iso_name)
        geo.solid_geometry = iso

        # Add to App and update.        
        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)
        
        #self.tree.get_selection().disconnect(self.signal_id)
        self.build_list() # Update the items list
        #self.signal_id = self.tree.get_selection().connect(
        #                     "changed", self.on_tree_selection_changed)
                             
        self.plot_all()
        #self.notebook.set_current_page(1)
                             
    def on_replot(self, widget):
        self.plot_all()
    
    def on_clear_plots(self, widget):
        self.clear_plots()
        
    def on_activate_name(self, entry):
        '''
        Hitting 'Enter' after changing the name of an item
        updates the item dictionary and re-builds the item list.
        '''
        print "Changing name"
        self.tree.get_selection().disconnect(self.signal_id)
        new_name = entry.get_text()  # Get from form
        self.stuff[new_name] = self.stuff.pop(self.selected_item_name)  # Update dictionary
        self.selected_item_name = new_name  # Update selection name
        self.build_list()  # Update the items list
        self.signal_id = self.tree.get_selection().connect(
                             "changed", self.on_tree_selection_changed)
                             
    def on_tree_selection_changed(self, selection):
        model, treeiter = selection.get_selected()

        if treeiter is not None:
            print "You selected", model[treeiter][0]
        else:
            return  # TODO: Clear "Selected" page
        
        self.selected_item_name = model[treeiter][0]
        # Remove the current selection page
        # from the notebook
        # TODO: Assuming it was last page or #2. Find the right page
        self.builder.get_object("notebook1").remove_page(2)

        self.stuff[self.selected_item_name].build_ui()

        # Determine the kind of item selected
        #kind = self.stuff[model[treeiter][0]].kind
        
        # Build the UI
        # builder = {"gerber": self.build_gerber_ui,
        #            "excellon": self.build_excellon_ui,
        #            "cncjob": self.build_cncjob_ui,
        #            "geometry": self.build_geometry_ui}
        # builder[kind]()
    
    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 file_chooser_action(self, 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,
            Gtk.FileChooserAction.OPEN,
            (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
             Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
        response = dialog.run()
        if response == Gtk.ResponseType.OK:
            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()    
    
    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.stuff[name] = gerber

            self.progress_bar.set_text("Plotting ...")
            self.progress_bar.set_fraction(0.6)

            #self.plot_gerber(gerber)
            gerber.plot(self.figure)
            gerber.axes.set_alpha(0.0)
            self.on_zoom_fit(None)

            self.progress_bar.set_text("Done!")
            self.progress_bar.set_fraction(1.0)

            self.notebook.set_current_page(1)
            self.set_list_selection(name)

            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.stuff[name] = excellon

            self.progress_bar.set_text("Plotting ...")
            self.progress_bar.set_fraction(0.6)

            #self.plot_excellon(excellon)
            excellon.plot(self.figure)
            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()
            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.stuff[name] = job

            self.progress_bar.set_text("Plotting ...")
            self.progress_bar.set_fraction(0.6)

            #self.plot_cncjob(job)
            job.plot(self.figure)
            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):
        try: # May fail in case mouse not within axes
            self.position_label.set_label("X: %.4f   Y: %.4f"%(
                                         event.xdata, event.ydata))
            self.mouse = [event.xdata, event.ydata]
        except:
            self.position_label.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.stuff)
        width = xmax-xmin
        height = ymax-ymin
        r = width/height
        
        Fw, Fh = self.canvas.get_width_height()
        Fr = float(Fw)/Fh
        print "Window aspect ratio:", Fr
        print "Data aspect ratio:", r
        
        #self.axes.set_xlim((xmin-0.05*width, xmax+0.05*width))
        #self.axes.set_ylim((ymin-0.05*height, ymax+0.05*height))
        
        if r > Fr:
            #self.axes.set_xlim((xmin-0.05*width, xmax+0.05*width))
            xmin -= 0.05*width
            xmax += 0.05*width
            ycenter = (ymin+ymax)/2.0
            newheight = height*r/Fr
            ymin = ycenter-newheight/2.0
            ymax = ycenter+newheight/2.0
            #self.axes.set_ylim((ycenter-newheight/2.0, ycenter+newheight/2.0))
        else:
            #self.axes.set_ylim((ymin-0.05*height, ymax+0.05*height))
            ymin -= 0.05*height
            ymax += 0.05*height
            xcenter = (xmax+ymin)/2.0
            newwidth = width*Fr/r
            xmin = xcenter-newwidth/2.0
            xmax = xcenter+newwidth/2.0
            #self.axes.set_xlim((xcenter-newwidth/2.0, xcenter+newwidth/2.0))

        for name in self.stuff:
            self.stuff[name].axes.set_xlim((xmin, xmax))
            self.stuff[name].axes.set_ylim((ymin, ymax))
        self.axes.set_xlim((xmin, xmax))
        self.axes.set_ylim((ymin, ymax))

        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 event.key == '2': # 2
            self.zoom(1/1.5, self.mouse)
            return
            
        if event.key == '3': # 3
            self.zoom(1.5, self.mouse)
            return

app = App()