1 files changed, 92 insertions, 0 deletions
diff --git a/setup/enginecombobox.py b/setup/enginecombobox.py
new file mode 100644
index 0000000..c8eaa15
--- /dev/null
+++ b/setup/enginecombobox.py
@@ -0,0 +1,92 @@
+# vim:set et sts=4 sw=4:
+#
+# ibus - The Input Bus
+#
+# Copyright (c) 2007-2008 Huang Peng <shawn.p.huang@gmail.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2 of the License, or (at your option) any later version.
+#
+# This library 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this program; if not, write to the
+# Free Software Foundation, Inc., 59 Temple Place, Suite 330,
+# Boston, MA  02111-1307  USA
+
+import gtk
+import gobject
+import ibus
+
+from icon import load_icon
+
+class EngineComboBox(gtk.ComboBox):
+    def __init__(self, engines):
+        super(EngineComboBox, self).__init__()
+
+        self.__model = gtk.TreeStore(gobject.TYPE_PYOBJECT)
+
+        lang = {}
+        for e in engines:
+            l = ibus.get_language_name(e.language)
+            if l not in lang:
+                lang[l] = []
+            lang[l].append(e)
+
+        for l in lang.keys():
+            iter1 = self.__model.append(None)
+            self.__model.set(iter1, 0, l)
+            for e in lang[l]:
+                iter2 = self.__model.append(iter1)
+                self.__model.set(iter2, 0, e)
+
+        self.set_model(self.__model)
+
+        renderer = gtk.CellRendererPixbuf()
+        renderer.set_property("xalign", 0)
+        self.pack_start(renderer, False)
+        self.set_cell_data_func(renderer, self.__icon_cell_data_cb)
+
+        renderer = gtk.CellRendererText()
+        renderer.set_property("xalign", 0)
+        self.pack_start(renderer, True)
+        self.set_cell_data_func(renderer, self.__name_cell_data_cb)
+
+    def __icon_cell_data_cb(self, celllayout, renderer, model, iter):
+        engine = self.__model.get_value(iter, 0)
+
+        icon_size = gtk.icon_size_lookup(gtk.ICON_SIZE_LARGE_TOOLBAR)[0]
+        if isinstance (engine, str) or isinstance (engine, unicode):
+            renderer.set_property("visible", False)
+            renderer.set_property("sensitive", False)
+        else:
+            renderer.set_property("visible", True)
+            renderer.set_property("sensitive", True)
+            pixbuf = load_icon(engine.icon, gtk.ICON_SIZE_LARGE_TOOLBAR)
+            if pixbuf == None:
+                pixbuf = load_icon("default-engine", gtk.ICON_SIZE_LARGE_TOOLBAR)
+            if pixbuf == None:
+                pixbuf = load_icon("gtk-missing-image", gtk.ICON_SIZE_LARGE_TOOLBAR)
+            renderer.set_property("pixbuf", pixbuf)
+
+    def __name_cell_data_cb(self, celllayout, renderer, model, iter):
+        engine = self.__model.get_value(iter, 0)
+
+        if isinstance (engine, str) or isinstance (engine, unicode):
+            renderer.set_property("sensitive", False)
+            renderer.set_property("text", engine)
+        else:
+            renderer.set_property("sensitive", True)
+            renderer.set_property("text", engine.longname)
+
+    def get_active_engine(self):
+        iter = self.get_active_iter()
+        if iter == None:
+            return None
+        return self.get_model()[iter][0]
+