622a19f0ea
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@26225 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
356 lines
10 KiB
Python
356 lines
10 KiB
Python
#----------------------------------------------------------------------------
|
|
# Name: BrowseImage.py
|
|
# Purpose: Display and Select Image Files
|
|
#
|
|
# Author: Lorne White
|
|
#
|
|
# Version: 1.0
|
|
# Date: January 29, 2002
|
|
# Licence: wxWindows license
|
|
#----------------------------------------------------------------------------
|
|
# 1.0 Release
|
|
# Create list of all available image file types
|
|
# View "All Image" File Types as default filter
|
|
# Sort the file list
|
|
# Use newer "re" function for patterns
|
|
#
|
|
#----------------------------------------------------------------------------
|
|
#
|
|
# 12/08/2003 - Jeff Grimmett (grimmtooth@softhome.net)
|
|
#
|
|
# o Updated for wx namespace
|
|
# o Corrected a nasty bug or two - see comments below.
|
|
# o There was a duplicate ImageView.DrawImage() method. Que?
|
|
#
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
import os
|
|
import sys
|
|
|
|
import wx
|
|
|
|
dir_path = os.getcwd()
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def ConvertBMP(file_nm):
|
|
if file_nm is None:
|
|
return None
|
|
|
|
fl_fld = os.path.splitext(file_nm)
|
|
ext = fl_fld[1]
|
|
ext = ext[1:].lower()
|
|
|
|
image = wx.Image(file_nm, wx.BITMAP_TYPE_ANY)
|
|
return image
|
|
|
|
|
|
def GetSize(file_nm): # for scaling image values
|
|
image = ConvertBMP(file_nm)
|
|
bmp = image.ConvertToBitmap()
|
|
size = bmp.GetWidth(), bmp.GetHeight()
|
|
return size
|
|
|
|
|
|
class ImageView(wx.Window):
|
|
def __init__(self, parent, id=-1, pos=wx.DefaultPosition, size=wx.DefaultSize):
|
|
wx.Window.__init__(self, parent, id, pos, size)
|
|
self.win = parent
|
|
self.image = None
|
|
self.back_color = 'WHITE'
|
|
self.border_color = 'BLACK'
|
|
|
|
# Changed API of wx uses tuples for size and pos now.
|
|
self.image_sizex = size[0]
|
|
self.image_sizey = size[1]
|
|
self.image_posx = pos[0]
|
|
self.image_posy = pos[1]
|
|
self.Bind(wx.EVT_PAINT, self.OnPaint)
|
|
|
|
def OnPaint(self, event):
|
|
dc = wx.PaintDC(self)
|
|
self.DrawImage(dc)
|
|
|
|
|
|
def SetValue(self, file_nm): # display the selected file in the panel
|
|
image = ConvertBMP(file_nm)
|
|
self.image = image
|
|
self.Refresh()
|
|
|
|
def DrawBorder(self, dc):
|
|
brush = wx.Brush(wx.NamedColour(self.back_color), wx.SOLID)
|
|
dc.SetBrush(brush)
|
|
dc.SetPen(wx.Pen(wx.NamedColour(self.border_color), 1))
|
|
dc.DrawRectangle((0, 0), (self.image_sizex, self.image_sizey))
|
|
|
|
def DrawImage(self, dc):
|
|
try:
|
|
image = self.image
|
|
except:
|
|
return
|
|
|
|
self.DrawBorder(dc)
|
|
|
|
if image is None:
|
|
return
|
|
|
|
bmp = image.ConvertToBitmap()
|
|
|
|
iwidth = bmp.GetWidth() # dimensions of image file
|
|
iheight = bmp.GetHeight()
|
|
|
|
diffx = (self.image_sizex - iwidth)/2 # center calc
|
|
if iwidth >= self.image_sizex -10: # if image width fits in window adjust
|
|
diffx = 5
|
|
iwidth = self.image_sizex - 10
|
|
|
|
diffy = (self.image_sizey - iheight)/2 # center calc
|
|
if iheight >= self.image_sizey - 10: # if image height fits in window adjust
|
|
diffy = 5
|
|
iheight = self.image_sizey - 10
|
|
|
|
image.Rescale(iwidth, iheight) # rescale to fit the window
|
|
image.ConvertToBitmap()
|
|
bmp = image.ConvertToBitmap()
|
|
dc.DrawBitmap(bmp, (diffx, diffy)) # draw the image to window
|
|
|
|
|
|
class ImageDialog(wx.Dialog):
|
|
def __init__(self, parent, set_dir = None):
|
|
wx.Dialog.__init__(self, parent, -1, "Image Browser", wx.DefaultPosition, (400, 400))
|
|
|
|
self.x_pos = 30 # initial display positions
|
|
self.y_pos = 20
|
|
self.delta = 20
|
|
|
|
size = wx.Size(80, 25)
|
|
|
|
self.set_dir = os.getcwd()
|
|
self.set_file = None
|
|
|
|
if set_dir != None:
|
|
if os.path.exists(set_dir): # set to working directory if nothing set
|
|
self.set_dir = set_dir
|
|
|
|
self.dir_x = self.x_pos
|
|
self.dir_y = self.y_pos
|
|
self.DisplayDir() # display the directory value
|
|
|
|
self.y_pos = self.y_pos + self.delta
|
|
|
|
btn = wx.Button(self, 12331, ' Set Directory ', (self.x_pos, self.y_pos), size)
|
|
self.Bind(wx.EVT_BUTTON, self.SetDirect, btn)
|
|
|
|
self.type_posy = self.y_pos # save the y position for the image type combo
|
|
|
|
self.fl_ext = '*.bmp' # initial setting for file filtering
|
|
self.GetFiles() # get the file list
|
|
|
|
self.y_pos = self.y_pos + self.delta + 10
|
|
|
|
self.list_height = 150
|
|
|
|
# List of Labels
|
|
self.tb = tb = wx.ListBox(self, -1, (self.x_pos, self.y_pos),
|
|
(160, self.list_height), self.fl_list,
|
|
wx.LB_SINGLE )
|
|
self.Bind(wx.EVT_LISTBOX, self.OnListClick, tb)
|
|
self.Bind(wx.EVT_LISTBOX_DCLICK, self.OnListDClick, tb)
|
|
|
|
width, height = self.tb.GetSize()
|
|
image_posx = self.x_pos + width + 20 # positions for setting the image window
|
|
image_posy = self.y_pos
|
|
image_sizex = 150
|
|
image_sizey = self.list_height
|
|
|
|
self.fl_types = [
|
|
"All Images", "Bmp", "Gif", "Png", "Jpg", "Ico", "Pnm",
|
|
"Pcx", "Tif", "All Files"
|
|
]
|
|
|
|
self.fl_ext_types = {
|
|
"All Images": "All",
|
|
"Bmp": "*.bmp",
|
|
"Gif": "*.gif",
|
|
"Png": "*.png",
|
|
"Jpg": "*.jpg",
|
|
"Ico": "*.ico",
|
|
"Pnm": "*.pnm",
|
|
"Pcx": "*.pcx",
|
|
"Tif": "*.tif",
|
|
"All Files": "*.*"
|
|
}
|
|
|
|
self.set_type = self.fl_types[0] # initial file filter setting
|
|
self.fl_ext = self.fl_ext_types[self.set_type]
|
|
|
|
self.sel_type = wx.ComboBox(self, -1, self.set_type, (image_posx , self.type_posy),
|
|
(150, -1), self.fl_types, wx.CB_DROPDOWN)
|
|
self.Bind(wx.EVT_COMBOBOX, self.OnSetType, self.sel_type)
|
|
|
|
self.image_view = ImageView( self, pos=(image_posx, image_posy),
|
|
size=(image_sizex, image_sizey))
|
|
|
|
self.y_pos = self.y_pos + height + 20
|
|
|
|
btn = wx.Button(self, -1, ' Select ', (100, self.y_pos), size)
|
|
self.Bind(wx.EVT_BUTTON, self.OnOk, btn)
|
|
|
|
wx.Button(self, wx.ID_CANCEL, 'Cancel', (250, self.y_pos), size)
|
|
|
|
self.y_pos = self.y_pos + self.delta
|
|
fsize = (400, self.y_pos + 50) # resize dialog for final vertical position
|
|
self.SetSize(fsize)
|
|
|
|
self.ResetFiles()
|
|
|
|
def GetFiles(self): # get the file list using directory and extension values
|
|
if self.fl_ext == "All":
|
|
all_files = []
|
|
|
|
for ftypes in self.fl_types[1:-1]: # get list of all available image types
|
|
filter = self.fl_ext_types[ftypes]
|
|
#print "filter = ", filter
|
|
self.fl_val = FindFiles(self, self.set_dir, filter)
|
|
all_files = all_files + self.fl_val.files # add to list of files
|
|
|
|
self.fl_list = all_files
|
|
else:
|
|
self.fl_val = FindFiles(self, self.set_dir, self.fl_ext)
|
|
self.fl_list = self.fl_val.files
|
|
|
|
self.fl_list.sort() # sort the file list
|
|
|
|
def DisplayDir(self): # display the working directory
|
|
wx.StaticText(self, -1, self.set_dir, (self.dir_x, self.dir_y), (250, -1))
|
|
|
|
def OnSetType(self, event):
|
|
val = event.GetString() # get file type value
|
|
self.fl_ext = self.fl_ext_types[val]
|
|
self.ResetFiles()
|
|
|
|
def OnListDClick(self, event):
|
|
self.OnOk(0)
|
|
|
|
def OnListClick(self, event):
|
|
val = event.GetSelection()
|
|
self.SetListValue(val)
|
|
|
|
def SetListValue(self, val):
|
|
file_nm = self.fl_list[val]
|
|
self.set_file = file_val = os.path.join(self.set_dir, file_nm)
|
|
self.image_view.SetValue(file_val)
|
|
|
|
def SetDirect(self, event): # set the new directory
|
|
dlg = wx.DirDialog(self)
|
|
dlg.SetPath(self.set_dir)
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
self.set_dir = dlg.GetPath()
|
|
self.ResetFiles()
|
|
|
|
dlg.Destroy()
|
|
|
|
def ResetFiles(self): # refresh the display with files and initial image
|
|
self.DisplayDir()
|
|
self.GetFiles()
|
|
|
|
# Changed 12/8/03 jmg
|
|
#
|
|
# o Clear listbox first
|
|
# o THEN check to see if there are any valid files of the selected
|
|
# type,
|
|
# o THEN if we have any files to display, set the listbox up,
|
|
#
|
|
# OTHERWISE
|
|
#
|
|
# o Leave it cleared
|
|
# o Clear the image viewer.
|
|
#
|
|
# This avoids a nasty assert error.
|
|
#
|
|
self.tb.Clear()
|
|
|
|
if len(self.fl_list):
|
|
self.tb.Set(self.fl_list)
|
|
|
|
try:
|
|
self.tb.SetSelection(0)
|
|
self.SetListValue(0)
|
|
except:
|
|
self.image_view.SetValue(None)
|
|
else:
|
|
self.image_view.SetValue(None)
|
|
|
|
def GetFile(self):
|
|
return self.set_file
|
|
|
|
def GetDirectory(self):
|
|
return self.set_dir
|
|
|
|
def OnCancel(self, event):
|
|
self.result = None
|
|
self.EndModal(wx.ID_CANCEL)
|
|
|
|
def OnOk(self, event):
|
|
self.result = self.set_file
|
|
self.EndModal(wx.ID_OK)
|
|
|
|
|
|
def OnFileDlg(self):
|
|
dlg = wx.FileDialog(self, "Choose an Image File", ".", "",
|
|
"Bmp (*.bmp)|*.bmp|JPEG (*.jpg)|*.jpg", wx.OPEN)
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
path = dlg.GetPath()
|
|
else:
|
|
path = None
|
|
|
|
dlg.Destroy()
|
|
return path
|
|
|
|
class FindFiles:
|
|
def __init__(self, parent, dir, mask):
|
|
filelist = []
|
|
dirlist = [".."]
|
|
self.dir = dir
|
|
self.file = ""
|
|
mask = mask.upper()
|
|
pattern = self.MakeRegex(mask)
|
|
|
|
for i in os.listdir(dir):
|
|
if i == "." or i == "..":
|
|
continue
|
|
|
|
path = os.path.join(dir, i)
|
|
path = path.upper()
|
|
value = i.upper()
|
|
|
|
if pattern.match(value) != None:
|
|
filelist.append(i)
|
|
|
|
self.files = filelist
|
|
|
|
def MakeRegex(self, pattern):
|
|
import re
|
|
f = "" # Set up a regex for file names
|
|
|
|
for ch in pattern:
|
|
if ch == "*":
|
|
f = f + ".*"
|
|
elif ch == ".":
|
|
f = f + "\."
|
|
elif ch == "?":
|
|
f = f + "."
|
|
else:
|
|
f = f + ch
|
|
|
|
return re.compile(f+'$')
|
|
|
|
def StripExt(self, file_nm):
|
|
fl_fld = os.path.splitext(file_nm)
|
|
fl_name = fl_fld[0]
|
|
ext = fl_fld[1]
|
|
return ext[1:]
|