FlatCAM/cirkuix.py

from gi.repository import Gtk

import matplotlib.pyplot as plt
plt.ioff()

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 App:
    def __init__(self):
        
        ########################################
        ##                GUI                 ##
        ########################################
        self.gladefile = "cirkuix.ui"
        self.builder = Gtk.Builder()
        self.builder.add_from_file(self.gladefile)
        self.window = self.builder.get_object("window1")
        self.positionLabel = self.builder.get_object("label3")      
        self.grid = self.builder.get_object("grid1")
        
        ## Event handling ##
        self.builder.connect_signals(self)
        
        self.figure = None
        self.axes = None
        self.canvas = None
        
        ## Make plot area ##
        self.mplpaint()
        self.window.show_all()
        
        ########################################
        ##               DATA                 ##
        ########################################
        self.gerbers = []
        self.excellons = []
        
        self.mouse = None
        
        Gtk.main()
        
    def mplpaint(self):
        self.figure = Figure(dpi=50)
        #self.axes = self.figure.add_subplot(111)
        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)
        #self.axes.plot(t,s)
        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.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.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()
        width = xmax-xmin
        height = ymax-ymin
        
        if center == 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
        
        self.axes.set_xlim((center[0]-new_width*(1-relx), center[0]+new_width*relx))
        self.axes.set_ylim((center[1]-new_height*(1-rely), center[1]+new_height*rely))
        
        self.canvas.queue_draw()
        
#    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
        
    def on_window_key_press(self, widget, event):
        print event.get_keycode(), event.get_keyval()
        val = int(event.get_keyval()[1])
        
        if val == 49: # 1
            self.on_zoom_fit(None)
            return
            
        if val == 50: # 2
            self.zoom(1/1.5, self.mouse)
            return
            
        if val == 51: # 3
            self.zoom(1.5, self.mouse)
            return

app = App()