wx/gui/options/viewer.py

   1 #!/usr/bin/env python
   2 # -*- coding: UTF-8 -*-
   3 #
   4 # gpdf2swf.py
   5 # graphical user interface for pdf2swf
   6 #
   7 # Part of the swftools package.
   8 #
   9 # Copyright (c) 2008,2009 Matthias Kramm <kramm@quiss.org>
  10 #
  11 # This program is free software; you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation; either version 2 of the License, or
  14 # (at your option) any later version.
  15 #
  16 # This program is distributed in the hope that it will be useful,
  17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 # GNU General Public License for more details.
  20 #
  21 # You should have received a copy of the GNU General Public License
  22 # along with this program; if not, write to the Free Software
  23 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  24
  25 import os
  26 import wx
  27 import operator
  28 import imp
  29 from lib.wordwrap import wordwrap
  30 from gui.boldstatictext import BoldStaticText
  31 import viewers
  32 import gui.fields
  33
  34 # this two are only to satisfy some plugins requirements
  35 import gui.plugin
  36 import shutil
  37
  38 #import viewers.raw
  39 #import viewers.simple
  40 #import viewers.rfx
  41 #import viewers.flexpaper
  42 #import viewers.technoargia
  43
  44 class Viewers:
  45     def __init__(self):
  46         #self.viewers = [
  47         #    'raw',
  48         #    'simple',
  49         #    'rfx',
  50         #    'flexpaper',
  51         #    'technoargia',
  52         #]
  53         self.viewers = []
  54         #self.modules = [
  55         #    viewers.raw,
  56         #    viewers.simple,
  57         #    viewers.rfx,
  58         #    viewers.flexpaper,
  59         #    viewers.technoargia,
  60         #]
  61         self.modules = []
  62
  63         self.list_viewers()
  64         self.import_viewers()
  65
  66     def list_viewers(self):
  67         for file in os.listdir('viewers'):
  68             if (file.startswith('.') or file.startswith('_')
  69                 or file.endswith(".pyc") or not file.endswith('.py')):
  70                 continue
  71             self.viewers.append(os.path.splitext(file)[0])
  72
  73     def import_viewers(self):
  74         for file in self.viewers:
  75             try:
  76                 _temp = imp.load_source("viewers.%s" % file, os.path.join(os.getcwdu(), "viewers/%s.py" % file))
  77                 self.modules.append(_temp)
  78             except Exception, e:
  79                 print "Could not load %s (%s)" % (file, e)
  80
  81
  82
  83 class ViewerBook(wx.Listbook):
  84     def __init__(self, parent):
  85         wx.Listbook.__init__(self, parent, wx.ID_ANY, style=
  86                             wx.LB_DEFAULT
  87                             )
  88
  89         self.__mod = []
  90         self.__viewers = viewers = Viewers()
  91
  92         # make an image list
  93         il = wx.ImageList(102, 102)
  94         self.AssignImageList(il)
  95
  96         # Sort viewers by attribute order.
  97         # The attribute order must be an int from 0..n
  98         viewers.modules.sort(key=operator.attrgetter('order'))
  99
 100         # Make a bunch of panels for the list book
 101         for idx, mod in enumerate(viewers.modules):
 102             bmp = mod.preview.GetBitmap()
 103             il.Add(bmp)
 104
 105             win = self.makePanel(mod)
 106             self.AddPage(win, mod, imageId=idx)
 107
 108
 109     def makePanel(self, mod):
 110         p = wx.Panel(self)
 111         sizer = wx.BoxSizer(wx.VERTICAL)
 112
 113         nameCtrl = BoldStaticText(p, label=mod.name)
 114         sizer.Add(nameCtrl, 0, wx.EXPAND|wx.TOP|wx.BOTTOM, 10)
 115
 116         t = wordwrap(mod.desc, 400, wx.ClientDC(p))
 117         descCtrl = wx.StaticText(p, label=t)
 118         sizer.Add(descCtrl, 0, wx.EXPAND|wx.TOP|wx.BOTTOM, 10)
 119
 120         #swf_options = []
 121
 122         # Avoid showing an empty StaticBox
 123         if (len(mod.swf_options) == 1
 124                and isinstance(mod.swf_options[0], gui.fields.Hidden)
 125            ):
 126             hidden = True
 127         else:
 128             hidden = False
 129
 130         if mod.swf_options and not hidden:
 131             box = wx.StaticBox(p, label=u"SWF")
 132             bsizer = wx.StaticBoxSizer(box, wx.VERTICAL)
 133
 134             flex = wx.FlexGridSizer(rows=len(mod.swf_options), cols=2, hgap=0, vgap=0)
 135             flex.AddGrowableCol(1)
 136
 137             for option in mod.swf_options:
 138                 #swf_options.append(option)
 139                 label, ctrl = option.draw(p)
 140                 if label and ctrl:
 141                     flex.Add(label, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 4)
 142                     flex.Add(ctrl, 0, wx.EXPAND|wx.ALL, 4)
 143
 144             bsizer.Add(flex, 0, wx.EXPAND)
 145             sizer.Add(bsizer, 0, wx.EXPAND)
 146
 147         #vie_options = []
 148         if mod.viewer_options:
 149             box = wx.StaticBox(p, label=u"Viewer")
 150             bsizer = wx.StaticBoxSizer(box, wx.VERTICAL)
 151
 152             flex = wx.FlexGridSizer(rows=len(mod.viewer_options),
 153                                     cols=2, hgap=0, vgap=0)
 154             flex.AddGrowableCol(1)
 155
 156             for option in mod.viewer_options:
 157                 #vie_options.append(option)
 158                 label, ctrl = option.draw(p)
 159                 if label and ctrl:
 160                     flex.Add(label, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 4)
 161                     flex.Add(ctrl, 0, wx.EXPAND|wx.ALL, 4)
 162
 163             bsizer.Add(flex, 0, wx.EXPAND)
 164             sizer.Add(bsizer, 0, wx.EXPAND)
 165
 166         p.SetSizer(sizer)
 167         return p
 168
 169     def AddPage(self, win, mod, select=False, imageId=-1):
 170         wx.Listbook.AddPage(self, win, mod.name, select, imageId)
 171         self.__mod.append(mod)
 172
 173     def __get_options(self):
 174         page = self.GetSelection()
 175         return self.__mod[page]
 176
 177     options = property(__get_options)
 178
 179     def pickle(self):
 180         data = {}
 181         data['selected_viewer'] = self.GetSelection()
 182
 183         for viewer, module in zip(self.__viewers.viewers, self.__viewers.modules):
 184             data[viewer] = {}
 185             for opt in module.swf_options:
 186                 data[viewer][opt.name] = opt.value
 187             for opt in module.viewer_options:
 188                 data[viewer][opt.name] = opt.value
 189
 190         return data
 191
 192     def unpickle(self, data):
 193         if not data:
 194             return
 195
 196         selected_viewer = data.pop('selected_viewer')
 197         self.SetSelection(selected_viewer)
 198
 199         _fields = {}
 200         for viewer, module in zip(self.__viewers.viewers, self.__viewers.modules):
 201             _fields[viewer] = {}
 202
 203             all_opts = module.swf_options + module.viewer_options
 204             for field in all_opts:
 205                 _fields[viewer][field.name] = field
 206
 207         for modname, opts in data.items():
 208             for k, v in opts.items():
 209                 _fields[modname][k].SetValue(v)
 210