home *** CD-ROM | disk | FTP | other *** search
/ PC Welt 2006 November (DVD) / PCWELT_11_2006.ISO / casper / filesystem.squashfs / usr / lib / gimp / 2.0 / python / gimpui.py < prev    next >
Encoding:
Python Source  |  2006-07-10  |  13.3 KB  |  404 lines
  1. #   Gimp-Python - allows the writing of Gimp plugins in Python.
  2. #   Copyright (C) 1997  James Henstridge <james@daa.com.au>
  3. #
  4. #    This program is free software; you can redistribute it and/or modify
  5. #   it under the terms of the GNU General Public License as published by
  6. #   the Free Software Foundation; either version 2 of the License, or
  7. #   (at your option) any later version.
  8. #
  9. #   This program is distributed in the hope that it will be useful,
  10. #   but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12. #   GNU General Public License for more details.
  13. #
  14. #   You should have received a copy of the GNU General Public License
  15. #   along with this program; if not, write to the Free Software
  16. #   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  17.  
  18. '''This module implements the UI items found in the libgimpui library.
  19. It requires pygtk to work.  These functions take use to callbacks -- one
  20. is a constraint function, and the other is the callback object.  The
  21. constraint function takes an image object as its first argument, and
  22. a drawable object as its second if appropriate.  The callback functions
  23. get the selected object as their first argument, and the user data as
  24. the second.
  25.  
  26. It also implements a number of selector widgets, which can be used to select
  27. various gimp data types.  Each of these selectors takes default as an argument
  28. to the constructor, and has a get_value() method for retrieving the result.
  29. '''
  30.  
  31. import pygtk
  32. pygtk.require('2.0')
  33.  
  34. import gtk, gobject, gimp
  35.  
  36. def _callbackWrapper(menu_item, callback, data):
  37.     callback(menu_item.get_data("Gimp-ID"), data)
  38.  
  39. def _createMenu(items, callback, data):
  40.     menu = gtk.Menu()
  41.     if not items:
  42.         items = [("(none)", None)]
  43.     for label, id in items:
  44.         menu_item = gtk.MenuItem(label)
  45.         menu_item.set_data("Gimp-ID", id)
  46.         menu.add(menu_item)
  47.         if callback:
  48.             menu_item.connect("activate", _callbackWrapper,
  49.                               callback, data)
  50.         menu_item.show()
  51.     return menu
  52.         
  53.  
  54. def ImageMenu(constraint=None, callback=None, data=None):
  55.     items = []
  56.     for img in gimp.image_list():
  57.         if constraint and not constraint(img):
  58.             continue
  59.         if not img.filename:
  60.             filename = img.name
  61.         else:
  62.             filename = img.filename
  63.         items.append((filename, img))
  64.     items.sort()
  65.     return _createMenu(items, callback, data)
  66.  
  67. def LayerMenu(constraint=None, callback=None, data=None):
  68.     items = []
  69.     for img in gimp.image_list():
  70.         filename = img.filename
  71.         if not filename:
  72.             filename = img.name
  73.         for layer in img.layers:
  74.             if constraint and not constraint(img, layer):
  75.                 continue
  76.             name = filename + "/" + layer.name
  77.             items.append((name, layer))
  78.     items.sort()
  79.     return _createMenu(items, callback, data)
  80.  
  81. def ChannelMenu(constraint=None, callback=None, data=None):
  82.     items = []
  83.     for img in gimp.image_list():
  84.         filename = img.filename
  85.         if not filename:
  86.             filename = img.name
  87.         for channel in img.channels:
  88.             if constraint and not constraint(img, channel):
  89.                 continue
  90.             name = filename + "/" + channel.name
  91.             items.append((name, channel))
  92.     items.sort()
  93.     return _createMenu(items, callback, data)
  94.  
  95. def DrawableMenu(constraint=None, callback=None, data=None):
  96.     items = []
  97.     for img in gimp.image_list():
  98.         filename = img.filename
  99.         if not filename:
  100.             filename = img.name
  101.         for drawable in img.layers + img.channels:
  102.             if constraint and not constraint(img, drawable):
  103.                 continue
  104.             name = filename + "/" + drawable.name
  105.             items.append((name, drawable))
  106.     items.sort()
  107.     return _createMenu(items, callback, data)
  108.  
  109. class ImageSelector(gtk.OptionMenu):
  110.     def __init__(self, default=None):
  111.         gtk.OptionMenu.__init__(self)
  112.         self.menu = ImageMenu(None, self.clicked)
  113.         self.set_menu(self.menu)
  114.         if default == None:
  115.             self.selected = self.menu.children()[0].get_data("Gimp-ID")
  116.             self.set_history (0)
  117.         else:
  118.             self.selected = default
  119.             children = self.menu.children()
  120.             for child in range(len(children)):
  121.                 if children[child].get_data("Gimp-ID") == default:
  122.                     self.set_history(child)
  123.                     break
  124.     def clicked(self, img, data=None):
  125.         self.selected = img
  126.     def get_value(self):
  127.         return self.selected
  128.  
  129. class LayerSelector(gtk.OptionMenu):
  130.     def __init__(self, default=None):
  131.         gtk.OptionMenu.__init__(self)
  132.         self.menu = LayerMenu(None, self.clicked)
  133.         self.set_menu(self.menu)
  134.         if default == None:
  135.             self.selected = self.menu.children()[0].get_data("Gimp-ID")
  136.             self.set_history (0)
  137.         else:
  138.             self.selected = default
  139.             children = self.menu.children()
  140.             for child in range(len(children)):
  141.                 if children[child].get_data("Gimp-ID") == default:
  142.                     self.set_history(child)
  143.                     break
  144.     def clicked(self, layer, data=None):
  145.         self.selected = layer
  146.     def get_value(self):
  147.         return self.selected
  148.  
  149. class ChannelSelector(gtk.OptionMenu):
  150.     def __init__(self, default=None):
  151.         gtk.OptionMenu.__init__(self)
  152.         self.menu = ChannelMenu(None, self.clicked)
  153.         self.set_menu(self.menu)
  154.         if default == None:
  155.             self.selected = self.menu.children()[0].get_data("Gimp-ID")
  156.             self.set_history (0)
  157.         else:
  158.             self.selected = default
  159.             children = self.menu.children()
  160.             for child in range(len(children)):
  161.                 if children[child].get_data("Gimp-ID") == default:
  162.                     self.set_history(child)
  163.                     break
  164.     def clicked(self, channel, data=None):
  165.         self.selected = channel
  166.     def get_value(self):
  167.         return self.selected
  168.  
  169. class DrawableSelector(gtk.OptionMenu):
  170.     def __init__(self, default=None):
  171.         gtk.OptionMenu.__init__(self)
  172.         self.menu = DrawableMenu(None, self.clicked)
  173.         self.set_menu(self.menu)
  174.         if default == None:
  175.             self.selected = self.menu.children()[0].get_data("Gimp-ID")
  176.             self.set_history (0)
  177.         else:
  178.             self.selected = default
  179.             children = self.menu.children()
  180.             for child in range(len(children)):
  181.                 if children[child].get_data("Gimp-ID") == default:
  182.                     self.set_history(child)
  183.                     break
  184.     def clicked(self, drawable, data=None):
  185.         self.selected = drawable
  186.     def get_value(self):
  187.         return self.selected
  188.  
  189. class ColourSelector(gtk.Button):
  190.     def __init__(self, default=(255, 0, 0)):
  191.         gtk.Button.__init__(self)
  192.         self.set_size_request(100, 20)
  193.  
  194.         self.colour = self.get_colormap().alloc_color(*map(lambda x: x*0x101,
  195.                                                            default))
  196.         self.update_colour()
  197.  
  198.         self.connect("clicked", self.show_dialog)
  199.  
  200.     def update_colour(self):
  201.         style = self.get_style().copy()
  202.         style.bg[gtk.STATE_NORMAL] = self.colour
  203.         style.bg[gtk.STATE_PRELIGHT] = self.colour
  204.         self.set_style(style)
  205.         self.queue_draw()
  206.  
  207.     def show_dialog(self, button):
  208.         dialog = gtk.ColorSelectionDialog("Colour")
  209.         dialog.colorsel.set_current_color(self.colour)
  210.         dialog.show()
  211.         response = dialog.run()
  212.         if response == gtk.RESPONSE_OK:
  213.             self.colour = dialog.colorsel.get_current_color()
  214.             self.update_colour()
  215.         dialog.destroy()
  216.             
  217.     def get_value(self):
  218.         return (self.colour.red/257, self.colour.green/257,
  219.                 self.colour.blue/257)
  220.  
  221. class _Selector(gtk.HBox):
  222.     def __init__(self):
  223.         gtk.HBox.__init__(self, gtk.FALSE, 5)
  224.         self.entry = gtk.Entry()
  225.         self.pack_start(self.entry)
  226.         self.entry.show()
  227.         self.button = gtk.Button("Browse...")
  228.         self.button.connect("clicked", self.show_dialog)
  229.         self.pack_start(self.button, expand=gtk.FALSE)
  230.         self.button.show()
  231.  
  232.         self.dialog = gtk.Dialog(self.get_title(), None, 0,
  233.                                  (gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
  234.                                   gtk.STOCK_OK, gtk.RESPONSE_OK))
  235.  
  236.         swin = gtk.ScrolledWindow()
  237.         swin.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
  238.         self.dialog.vbox.pack_start(swin)
  239.         swin.show()
  240.                 
  241.         items = map(None, self.get_list())
  242.  
  243.         store = gtk.ListStore(gobject.TYPE_STRING)
  244.         treeview = gtk.TreeView(store)
  245.  
  246.         treeview.set_headers_visible(gtk.FALSE)
  247.         column = gtk.TreeViewColumn('Items', gtk.CellRendererText(), text=0)
  248.         treeview.append_column(column)
  249.  
  250.         self.tv_selection = treeview.get_selection()
  251.  
  252.         self.selected = self.get_default()
  253.         self.entry.set_text(self.selected)
  254.  
  255.         sel_path = None
  256.         items.sort()
  257.  
  258.         for s in items:
  259.             iter = store.append()
  260.             store.set(iter, 0, s)
  261.  
  262.             if s == self.selected:
  263.                 sel_path = store.get_path(iter)
  264.  
  265.         if sel_path:
  266.             self.tv_selection.select_path(sel_path)
  267.  
  268.         swin.add(treeview)
  269.         treeview.show()
  270.  
  271.         self.dialog.set_default_size(300, 225)
  272.  
  273.     def show_dialog(self, button):
  274.         self.dialog.show()
  275.         response = self.dialog.run()
  276.         self.dialog.hide()
  277.  
  278.         if response == gtk.RESPONSE_OK:
  279.             self.selected = self.get_selection()
  280.             self.entry.set_text(self.selected)
  281.  
  282.     def get_selection(self):
  283.         store, iter = self.tv_selection.get_selected()
  284.         return store.get_value(iter, 0)
  285.  
  286.     def get_value(self):
  287.         return self.selected
  288.  
  289. class PatternSelector(_Selector):
  290.     def __init__(self, default=""):
  291.         self.default = default
  292.         _Selector.__init__(self)
  293.     def get_default(self):
  294.         return self.default
  295.     def get_title(self):
  296.         return "Patterns"
  297.     def get_list(self):
  298.         num, patterns = gimp.pdb.gimp_patterns_get_list(None)
  299.         return patterns
  300.  
  301. class BrushSelector(_Selector):
  302.     def __init__(self, default=""):
  303.         self.default = default
  304.         _Selector.__init__(self)
  305.     def get_default(self):
  306.         return self.default
  307.     def get_title(self):
  308.         return "Brushes"
  309.     def get_list(self):
  310.         num, brushes = gimp.pdb.gimp_brushes_get_list(None)
  311.         return brushes
  312.  
  313. class GradientSelector(_Selector):
  314.     def __init__(self, default=""):
  315.         self.default = default
  316.         _Selector.__init__(self)
  317.     def get_default(self):
  318.         return self.default
  319.     def get_title(self):
  320.         return "Gradients"
  321.     def get_list(self):
  322.         num, gradients = gimp.pdb.gimp_gradients_get_list(None)
  323.         return gradients
  324.  
  325. class PaletteSelector(_Selector):
  326.     def __init__(self, default=""):
  327.         self.default = default
  328.         _Selector.__init__(self)
  329.     def get_default(self):
  330.         return self.default
  331.     def get_title(self):
  332.         return "Palettes"
  333.     def get_list(self):
  334.         num, palettes = gimp.pdb.gimp_palettes_get_list("")
  335.         return palettes    
  336.     
  337. class FontSelector(gtk.HBox):
  338.     def __init__(self, default="Sans"):
  339.         gtk.HBox.__init__(self, gtk.FALSE, 5)
  340.         self.entry = gtk.Entry()
  341.         self.pack_start(self.entry)
  342.         self.entry.show()
  343.         self.button = gtk.Button("Browse...")
  344.         self.button.connect("clicked", self.show_dialog)
  345.         self.pack_start(self.button, expand=gtk.FALSE)
  346.         self.button.show()
  347.  
  348.         self.dialog = gtk.FontSelectionDialog("Fonts")
  349.         self.dialog.set_default_size(400, 300)
  350.         def delete_event(win, event):
  351.             win.hide()
  352.             return gtk.TRUE
  353.         self.dialog.connect("delete_event", delete_event)
  354.  
  355.         self.dialog.set_font_name(default)
  356.         self.selected = default
  357.         self.entry.set_text(self.selected)
  358.         
  359.     def show_dialog(self, button):
  360.         self.dialog.show()
  361.         response = self.dialog.run()
  362.         self.dialog.hide()
  363.  
  364.         if response == gtk.RESPONSE_OK:
  365.             self.selected = self.dialog.get_font_name()
  366.             self.entry.set_text(self.selected)
  367.  
  368.     def get_value(self):
  369.         return self.selected
  370.                 
  371. class FileSelector(gtk.HBox):
  372.     def __init__(self, default=""):
  373.         gtk.HBox.__init__(self, gtk.FALSE, 5)
  374.         self.entry = gtk.Entry()
  375.         self.pack_start(self.entry)
  376.         self.entry.show()
  377.         self.button = gtk.Button("...")
  378.         self.button.connect("clicked", self.show_dialog)
  379.         self.pack_start(self.button, expand=gtk.FALSE)
  380.         self.button.show()
  381.  
  382.         self.dialog = gtk.FileSelection("Files")
  383.         self.dialog.set_default_size(400, 300)
  384.         def delete_event(win, event):
  385.             win.hide()
  386.             return gtk.TRUE
  387.         self.dialog.connect("delete_event", delete_event)
  388.  
  389.         self.dialog.set_filename(default)
  390.         self.selected = self.dialog.get_filename()
  391.         self.entry.set_text(self.selected)
  392.                 
  393.     def show_dialog(self, button):
  394.         self.dialog.show()
  395.         response = self.dialog.run()
  396.         self.dialog.hide()
  397.  
  398.         if response == gtk.RESPONSE_OK:
  399.             self.selected = self.dialog.get_filename()
  400.             self.entry.set_text(self.selected)
  401.  
  402.     def get_value(self):
  403.         return self.selected
  404.