PySolFC/pysollib/pysolgtk/selecttile.py

##---------------------------------------------------------------------------##
##
## PySol -- a Python Solitaire game
##
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program; see the file COPYING.
## If not, write to the Free Software Foundation, Inc.,
## 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
##
##---------------------------------------------------------------------------##


# imports

## import os, string, sys, types
import gobject, gtk
from gtk import gdk

## # PySol imports
## from pysollib.mfxutil import destruct, Struct, KwStruct
from pysollib.resource import CSI
from pysollib.mfxutil import kwdefault, KwStruct

# Toolkit imports
## from tkutil import loadImage
from tkwidget import MfxDialog
from tkcanvas import MfxCanvas
from tkutil import setTransient


class SelectTileDialogWithPreview(MfxDialog):

    def __init__(self, parent, title, app, manager, key=None, **kw):
        kw = self.initKw(kw)
        MfxDialog.__init__(self, parent, title, **kw)
        #
        top_box, bottom_box = self.createBox()
        #
        if key is None:
            key = manager.getSelected()
        self.app = app
        self.manager = manager
        self.key = key
        self.preview_key = -1
        self.all_keys = []
        self.table_color = app.opt.table_color

        sw = gtk.ScrolledWindow()
        sw.set_shadow_type(gtk.SHADOW_ETCHED_IN)
        sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        top_box.pack_start(sw)

        #
        model = self._create_tree_model(manager, key)
        treeview = gtk.TreeView(model)
        treeview.set_rules_hint(True)
        treeview.set_headers_visible(False)

        renderer = gtk.CellRendererText()
        renderer.set_property('xalign', 0.0)

        column = gtk.TreeViewColumn('Tiles', renderer, text=0)
        column.set_clickable(True)
        treeview.append_column(column)

        sw.add(treeview)
        treeview.expand_all()

        selection = treeview.get_selection()
        selection.connect('changed', self.treeview_show_selected)

        treeview.connect('row-activated', self.row_activated)
        self.treeview = treeview

        #
        self.preview = MfxCanvas(top_box) # width=w2
        top_box.pack_end(self.preview)
        self.preview.show()

        self.createButtons(bottom_box, kw)

        self.updatePreview(key)

        self.show_all()
        gtk.main()


    def _getSelected(self):
        selection = self.treeview.get_selection()
        model, path = selection.get_selected_rows()
        if not path:
            return None
        iter = model.get_iter(path[0])
        index = model.get_value(iter, 1)
        if index < 0:
            return None
        return self.all_keys[index]


    def row_activated(self, w, row, col):
        print 'row_activated_event', row, col


    def treeview_show_selected(self, w):
        key = self._getSelected()
        self.updatePreview(key)


    def _create_tree_model(self, manager, key):
        self.all_keys = []
        index = 0
        #
        model = gtk.TreeStore(gobject.TYPE_STRING,
                              gobject.TYPE_INT)
        #
        iter = model.append(None)
        model.set(iter, 0, _('Solid color'), 1, -1)
        for color, value in ((_('Blue'),   '#0082df'),
                             (_('Green'),  '#008200'),
                             (_('Navy'),   '#000086'),
                             (_('Olive'),  '#868200'),
                             (_('Orange'), '#f79600'),
                             (_('Teal'),   '#008286'),):
            child_iter = model.append(iter)
            model.set(child_iter, 0, color, 1, index)
            self.all_keys.append(value)
            index += 1
        #
        tiles = manager.getAllSortedByName()
        tiles = filter(lambda obj: not obj.error, tiles)
        tiles = filter(lambda tile: tile.index > 0 and tile.filename, tiles)
        #
        iter = model.append(None)
        model.set(iter, 0, _('All Backgrounds'), 1, -1)
        if tiles:
            for tile in tiles:
                child_iter = model.append(iter)
                model.set(child_iter, 0, tile.name, 1, index)
                self.all_keys.append(tile.index)
                index += 1
        else:
            child_iter = model.append(iter)
            model.set(child_iter, 0, _('(no tiles)'), 1, -1)

        return model


    def updatePreview(self, key):
        ##print 'updatePreview:', key
        if key is None:
            return
        if key == self.preview_key:
            return
        canvas = self.preview
        canvas.deleteAllItems()
        if type(key) is str:
            # solid color
            canvas.config(bg=key)
            ##canvas.setTile(None)
            ##canvas.setTextColor(None)
            self.preview_key = key
            self.table_color = key
        else:
            # image
            tile = self.manager.get(key)
            if tile:
                if self.preview.setTile(self.app, key):
                    return
            self.preview_key = -1


    def initKw(self, kw):
        kwdefault(kw,
                  strings=(_('&OK'), _('&Solid color...'), _('&Cancel'),),
                  default=0,
                  resizable=1,
                  font=None,
                  padx=10, pady=10,
                  width=600, height=400,
                  ##~ buttonpadx=10, buttonpady=5,
                  )
        return MfxDialog.initKw(self, kw)


    def _colorselOkClicked(self, w, d):
        c = d.colorsel.get_current_color()
        c = '#%02x%02x%02x' % (c.red/256, c.green/256, c.blue/256)
        d.destroy()
        self.updatePreview(c)
        selection = self.treeview.get_selection()
        selection.unselect_all()


    def createColorsel(self):
        win = gtk.ColorSelectionDialog('Select table color')
        win.help_button.destroy()
        win.set_position(gtk.WIN_POS_CENTER_ON_PARENT)
        if type(self.preview_key) is str:
            color =  self.preview_key
        else:
            color = self.app.opt.table_color
        win.colorsel.set_current_color(gdk.color_parse(color))
        win.connect('delete_event', lambda w, e: win.destroy())
        win.ok_button.connect('clicked', self._colorselOkClicked, win)
        win.cancel_button.connect('clicked', lambda w: win.destroy())
        setTransient(win, self)
        win.show()


    def done(self, button):
        b = button.get_data('user_data')
        if b == 1:
            self.createColorsel()
            return
        if b == 0:
            self.key = self._getSelected()
            if not self.key:
                self.key = self.preview_key
        self.status = 0
        self.button = b
        self.hide()
        self.quit()