]> git.saurik.com Git - wxWidgets.git/blame_incremental - wxPython/wx/lib/imagebrowser.py
docstring update
[wxWidgets.git] / wxPython / wx / lib / imagebrowser.py
... / ...
CommitLineData
1#----------------------------------------------------------------------------
2# Name: BrowseImage.py
3# Purpose: Display and Select Image Files
4#
5# Author: Lorne White
6#
7# Version: 1.0
8# Date: January 29, 2002
9# Licence: wxWindows license
10#----------------------------------------------------------------------------
11# 1.0 Release
12# Create list of all available image file types
13# View "All Image" File Types as default filter
14# Sort the file list
15# Use newer "re" function for patterns
16#
17#----------------------------------------------------------------------------
18#
19# 12/08/2003 - Jeff Grimmett (grimmtooth@softhome.net)
20#
21# o Updated for wx namespace
22# o Corrected a nasty bug or two - see comments below.
23# o There was a duplicate ImageView.DrawImage() method. Que?
24#
25
26#---------------------------------------------------------------------------
27
28import os
29import sys
30
31import wx
32
33dir_path = os.getcwd()
34
35#---------------------------------------------------------------------------
36
37def ConvertBMP(file_nm):
38 if file_nm is None:
39 return None
40
41 fl_fld = os.path.splitext(file_nm)
42 ext = fl_fld[1]
43 ext = ext[1:].lower()
44
45 image = wx.Image(file_nm, wx.BITMAP_TYPE_ANY)
46 return image
47
48
49def GetSize(file_nm): # for scaling image values
50 image = ConvertBMP(file_nm)
51 bmp = image.ConvertToBitmap()
52 size = bmp.GetWidth(), bmp.GetHeight()
53 return size
54
55
56class ImageView(wx.Window):
57 def __init__(self, parent, id=-1, pos=wx.DefaultPosition, size=wx.DefaultSize):
58 wx.Window.__init__(self, parent, id, pos, size)
59 self.win = parent
60 self.image = None
61 self.back_color = 'WHITE'
62 self.border_color = 'BLACK'
63
64 # Changed API of wx uses tuples for size and pos now.
65 self.image_sizex = size[0]
66 self.image_sizey = size[1]
67 self.image_posx = pos[0]
68 self.image_posy = pos[1]
69 self.Bind(wx.EVT_PAINT, self.OnPaint)
70
71 def OnPaint(self, event):
72 dc = wx.PaintDC(self)
73 self.DrawImage(dc)
74
75
76 def SetValue(self, file_nm): # display the selected file in the panel
77 image = ConvertBMP(file_nm)
78 self.image = image
79 self.Refresh()
80
81 def DrawBorder(self, dc):
82 brush = wx.Brush(wx.NamedColour(self.back_color), wx.SOLID)
83 dc.SetBrush(brush)
84 dc.SetPen(wx.Pen(wx.NamedColour(self.border_color), 1))
85 dc.DrawRectangle(0, 0, self.image_sizex, self.image_sizey)
86
87 def DrawImage(self, dc):
88 try:
89 image = self.image
90 except:
91 return
92
93 self.DrawBorder(dc)
94
95 if image is None:
96 return
97
98 bmp = image.ConvertToBitmap()
99
100 iwidth = bmp.GetWidth() # dimensions of image file
101 iheight = bmp.GetHeight()
102
103 diffx = (self.image_sizex - iwidth)/2 # center calc
104 if iwidth >= self.image_sizex -10: # if image width fits in window adjust
105 diffx = 5
106 iwidth = self.image_sizex - 10
107
108 diffy = (self.image_sizey - iheight)/2 # center calc
109 if iheight >= self.image_sizey - 10: # if image height fits in window adjust
110 diffy = 5
111 iheight = self.image_sizey - 10
112
113 image.Rescale(iwidth, iheight) # rescale to fit the window
114 image.ConvertToBitmap()
115 bmp = image.ConvertToBitmap()
116 dc.DrawBitmap(bmp, diffx, diffy) # draw the image to window
117
118
119class ImageDialog(wx.Dialog):
120 def __init__(self, parent, set_dir = None):
121 wx.Dialog.__init__(self, parent, -1, "Image Browser", wx.DefaultPosition, (400, 400))
122
123 self.x_pos = 30 # initial display positions
124 self.y_pos = 20
125 self.delta = 20
126
127 size = wx.Size(80, -1)
128
129 self.set_dir = os.getcwd()
130 self.set_file = None
131
132 if set_dir != None:
133 if os.path.exists(set_dir): # set to working directory if nothing set
134 self.set_dir = set_dir
135
136 self.dir_x = self.x_pos
137 self.dir_y = self.y_pos
138 self.dir = wx.StaticText(self, -1, self.set_dir, (self.dir_x, self.dir_y), (250, -1))
139
140 self.y_pos = self.y_pos + self.delta
141
142 btn = wx.Button(self, 12331, ' Set Directory ', (self.x_pos, self.y_pos))
143 self.Bind(wx.EVT_BUTTON, self.SetDirect, btn)
144
145 self.type_posy = self.y_pos # save the y position for the image type combo
146
147 self.fl_ext = '*.bmp' # initial setting for file filtering
148 self.GetFiles() # get the file list
149
150 self.y_pos = self.y_pos + self.delta + 10
151 self.list_height = 150
152
153 # List of Labels
154 self.tb = tb = wx.ListBox(self, -1, (self.x_pos, self.y_pos),
155 (160, self.list_height), self.fl_list,
156 wx.LB_SINGLE )
157 self.Bind(wx.EVT_LISTBOX, self.OnListClick, tb)
158 self.Bind(wx.EVT_LISTBOX_DCLICK, self.OnListDClick, tb)
159
160 width, height = self.tb.GetSize()
161 image_posx = self.x_pos + width + 20 # positions for setting the image window
162 image_posy = self.y_pos
163 image_sizex = 150
164 image_sizey = self.list_height
165
166 self.fl_types = [
167 "All Images", "Bmp", "Gif", "Png", "Jpg", "Ico", "Pnm",
168 "Pcx", "Tif", "All Files"
169 ]
170
171 self.fl_ext_types = {
172 "All Images": "All",
173 "Bmp": "*.bmp",
174 "Gif": "*.gif",
175 "Png": "*.png",
176 "Jpg": "*.jpg",
177 "Ico": "*.ico",
178 "Pnm": "*.pnm",
179 "Pcx": "*.pcx",
180 "Tif": "*.tif",
181 "All Files": "*.*"
182 }
183
184 self.set_type = self.fl_types[0] # initial file filter setting
185 self.fl_ext = self.fl_ext_types[self.set_type]
186
187 self.sel_type = wx.ComboBox(self, -1, self.set_type, (image_posx , self.type_posy),
188 (150, -1), self.fl_types, wx.CB_DROPDOWN)
189 self.Bind(wx.EVT_COMBOBOX, self.OnSetType, self.sel_type)
190
191 self.image_view = ImageView( self, pos=(image_posx, image_posy),
192 size=(image_sizex, image_sizey))
193
194 self.y_pos = self.y_pos + height + 20
195
196 btn = wx.Button(self, -1, ' Select ', (100, self.y_pos), size)
197 self.Bind(wx.EVT_BUTTON, self.OnOk, btn)
198
199 wx.Button(self, wx.ID_CANCEL, 'Cancel', (250, self.y_pos), size)
200
201 self.y_pos = self.y_pos + self.delta
202 fsize = (400, self.y_pos + 50) # resize dialog for final vertical position
203 self.SetSize(fsize)
204
205 self.ResetFiles()
206
207 def GetFiles(self): # get the file list using directory and extension values
208 if self.fl_ext == "All":
209 all_files = []
210
211 for ftypes in self.fl_types[1:-1]: # get list of all available image types
212 filter = self.fl_ext_types[ftypes]
213 #print "filter = ", filter
214 self.fl_val = FindFiles(self, self.set_dir, filter)
215 all_files = all_files + self.fl_val.files # add to list of files
216
217 self.fl_list = all_files
218 else:
219 self.fl_val = FindFiles(self, self.set_dir, self.fl_ext)
220 self.fl_list = self.fl_val.files
221
222 self.fl_list.sort() # sort the file list
223
224 def DisplayDir(self): # display the working directory
225 if self.dir:
226 self.dir.SetLabel(self.set_dir)
227
228 def OnSetType(self, event):
229 val = event.GetString() # get file type value
230 self.fl_ext = self.fl_ext_types[val]
231 self.ResetFiles()
232
233 def OnListDClick(self, event):
234 self.OnOk(0)
235
236 def OnListClick(self, event):
237 val = event.GetSelection()
238 self.SetListValue(val)
239
240 def SetListValue(self, val):
241 file_nm = self.fl_list[val]
242 self.set_file = file_val = os.path.join(self.set_dir, file_nm)
243 self.image_view.SetValue(file_val)
244
245 def SetDirect(self, event): # set the new directory
246 dlg = wx.DirDialog(self)
247 dlg.SetPath(self.set_dir)
248
249 if dlg.ShowModal() == wx.ID_OK:
250 self.set_dir = dlg.GetPath()
251 self.ResetFiles()
252
253 dlg.Destroy()
254
255 def ResetFiles(self): # refresh the display with files and initial image
256 self.DisplayDir()
257 self.GetFiles()
258
259 # Changed 12/8/03 jmg
260 #
261 # o Clear listbox first
262 # o THEN check to see if there are any valid files of the selected
263 # type,
264 # o THEN if we have any files to display, set the listbox up,
265 #
266 # OTHERWISE
267 #
268 # o Leave it cleared
269 # o Clear the image viewer.
270 #
271 # This avoids a nasty assert error.
272 #
273 self.tb.Clear()
274
275 if len(self.fl_list):
276 self.tb.Set(self.fl_list)
277
278 try:
279 self.tb.SetSelection(0)
280 self.SetListValue(0)
281 except:
282 self.image_view.SetValue(None)
283 else:
284 self.image_view.SetValue(None)
285
286 def GetFile(self):
287 return self.set_file
288
289 def GetDirectory(self):
290 return self.set_dir
291
292 def OnCancel(self, event):
293 self.result = None
294 self.EndModal(wx.ID_CANCEL)
295
296 def OnOk(self, event):
297 self.result = self.set_file
298 self.EndModal(wx.ID_OK)
299
300
301def OnFileDlg(self):
302 dlg = wx.FileDialog(self, "Choose an Image File", ".", "",
303 "Bmp (*.bmp)|*.bmp|JPEG (*.jpg)|*.jpg", wx.OPEN)
304
305 if dlg.ShowModal() == wx.ID_OK:
306 path = dlg.GetPath()
307 else:
308 path = None
309
310 dlg.Destroy()
311 return path
312
313class FindFiles:
314 def __init__(self, parent, dir, mask):
315 filelist = []
316 dirlist = [".."]
317 self.dir = dir
318 self.file = ""
319 mask = mask.upper()
320 pattern = self.MakeRegex(mask)
321
322 for i in os.listdir(dir):
323 if i == "." or i == "..":
324 continue
325
326 path = os.path.join(dir, i)
327 path = path.upper()
328 value = i.upper()
329
330 if pattern.match(value) != None:
331 filelist.append(i)
332
333 self.files = filelist
334
335 def MakeRegex(self, pattern):
336 import re
337 f = "" # Set up a regex for file names
338
339 for ch in pattern:
340 if ch == "*":
341 f = f + ".*"
342 elif ch == ".":
343 f = f + "\."
344 elif ch == "?":
345 f = f + "."
346 else:
347 f = f + ch
348
349 return re.compile(f+'$')
350
351 def StripExt(self, file_nm):
352 fl_fld = os.path.splitext(file_nm)
353 fl_name = fl_fld[0]
354 ext = fl_fld[1]
355 return ext[1:]