import gtk

from Sketch import Sketch
from SampleSketch import SampleSketch
from SketchWidget import SketchWidget

class Window:
    def __init__(self, title):
        self.mode_buttons = {}

        self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.window.set_size_request(600, 500)
        self.window.set_title(title)
        self.window.connect("destroy", self.destroy_event)
        self.window.connect("key-press-event", self.key_press_event)

        ss = SampleSketch()

        ss.solve()

        sw = SketchWidget(ss, self.window)

        self.toolbar = gtk.Toolbar()
        self.toolbar.set_style(gtk.TOOLBAR_BOTH)
        ttb = gtk.ToggleToolButton()
        ttb.set_label('Select')
        ttb.connect('toggled', self.mode_toggle_event)
        ttb.set_active(True)
        self.toolbar.insert(ttb, -1)
        self.mode_buttons[ttb] = 'select'
        ttb = gtk.ToggleToolButton()
        ttb.set_label('Line')
        ttb.connect('toggled', self.mode_toggle_event)
        self.toolbar.insert(ttb, -1)
        self.mode_buttons[ttb] = 'line'
        ttb = gtk.ToggleToolButton()
        ttb.set_label('Circle')
        ttb.connect('toggled', self.mode_toggle_event)
        self.toolbar.insert(ttb, -1)
        self.mode_buttons[ttb] = 'circle'

        self.statusbar = gtk.Statusbar()

        vbox = gtk.VBox()
        vbox.pack_start(self.toolbar, expand = False)
        vbox.pack_start(sw.widget, expand = True)
        vbox.pack_start(self.statusbar, expand = False)

        self.window.add(vbox)
        self.window.show_all()

        sw.widget.grab_focus()

    def main(self):
        gtk.main()

    def destroy_event(self, widget, data=None):
        gtk.main_quit()

    def key_press_event(self, widget, event, data=None):
        if event.keyval == gtk.gdk.keyval_from_name("Escape"):
            gtk.main_quit()
        return False

    def mode_toggle_event(self, ttb):
        if not ttb.get_active():
            for b in self.mode_buttons:
                if b.get_active():
                    break
            else:
                ttb.set_active(True)
        else:
            for b in self.mode_buttons:
                if b != ttb:
                    b.set_active(False)
        return True