[wxWidgets.git] / wxPython / wx / lib / imagebrowser.py

#----------------------------------------------------------------------------
# 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:]
Commit	Line	Data
d14a1e28 RD	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	#----------------------------------------------------------------------------
d14a1e28 RD	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
b881fc78 RD	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	#
1fded56b	25
d14a1e28 RD	26	#---------------------------------------------------------------------------
d14a1e28 RD	27
b881fc78 RD	28	import os
	29	import sys
	30
	31	import wx
	32
d14a1e28 RD	33	dir_path = os.getcwd()
	34
	35	#---------------------------------------------------------------------------
	36
	37	def 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()
d14a1e28	44
b881fc78	45	image = wx.Image(file_nm, wx.BITMAP_TYPE_ANY)
d14a1e28 RD	46	return image
d14a1e28 RD	47
b881fc78	48
d14a1e28 RD	49	def 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
b881fc78 RD	55
	56	class 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)
d14a1e28 RD	59	self.win = parent
	60	self.image = None
	61	self.back_color = 'WHITE'
	62	self.border_color = 'BLACK'
	63
b881fc78 RD	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)
d14a1e28 RD	70
d14a1e28 RD	71	def OnPaint(self, event):
b881fc78	72	dc = wx.PaintDC(self)
d14a1e28 RD	73	self.DrawImage(dc)
d14a1e28 RD	74
d14a1e28 RD	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):
b881fc78	82	brush = wx.Brush(wx.NamedColour(self.back_color), wx.SOLID)
d14a1e28	83	dc.SetBrush(brush)
b881fc78 RD	84	dc.SetPen(wx.Pen(wx.NamedColour(self.border_color), 1))
b881fc78 RD	85	dc.DrawRectangle((0, 0), (self.image_sizex, self.image_sizey))
d14a1e28 RD	86
	87	def DrawImage(self, dc):
	88	try:
	89	image = self.image
	90	except:
	91	return
	92
	93	self.DrawBorder(dc)
b881fc78	94
d14a1e28 RD	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()
b881fc78	116	dc.DrawBitmap(bmp, (diffx, diffy)) # draw the image to window
d14a1e28 RD	117
d14a1e28 RD	118
b881fc78	119	class ImageDialog(wx.Dialog):
d14a1e28	120	def __init__(self, parent, set_dir = None):
b881fc78	121	wx.Dialog.__init__(self, parent, -1, "Image Browser", wx.DefaultPosition, (400, 400))
d14a1e28 RD	122
	123	self.x_pos = 30 # initial display positions
	124	self.y_pos = 20
	125	self.delta = 20
	126
d4b73b1b	127	size = wx.Size(80, 25)
d14a1e28 RD	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.DisplayDir() # display the directory value
	139
	140	self.y_pos = self.y_pos + self.delta
	141
622a19f0	142	btn = wx.Button(self, 12331, ' Set Directory ', (self.x_pos, self.y_pos), size)
b881fc78	143	self.Bind(wx.EVT_BUTTON, self.SetDirect, btn)
d14a1e28 RD	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
	152	self.list_height = 150
	153
	154	# List of Labels
b881fc78 RD	155	self.tb = tb = wx.ListBox(self, -1, (self.x_pos, self.y_pos),
	156	(160, self.list_height), self.fl_list,
	157	wx.LB_SINGLE )
	158	self.Bind(wx.EVT_LISTBOX, self.OnListClick, tb)
	159	self.Bind(wx.EVT_LISTBOX_DCLICK, self.OnListDClick, tb)
d14a1e28	160
b881fc78	161	width, height = self.tb.GetSize()
d14a1e28 RD	162	image_posx = self.x_pos + width + 20 # positions for setting the image window
	163	image_posy = self.y_pos
	164	image_sizex = 150
	165	image_sizey = self.list_height
	166
b881fc78 RD	167	self.fl_types = [
	168	"All Images", "Bmp", "Gif", "Png", "Jpg", "Ico", "Pnm",
	169	"Pcx", "Tif", "All Files"
	170	]
	171
	172	self.fl_ext_types = {
	173	"All Images": "All",
	174	"Bmp": "*.bmp",
	175	"Gif": "*.gif",
	176	"Png": "*.png",
	177	"Jpg": "*.jpg",
	178	"Ico": "*.ico",
	179	"Pnm": "*.pnm",
	180	"Pcx": "*.pcx",
	181	"Tif": "*.tif",
	182	"All Files": "."
	183	}
d14a1e28 RD	184
	185	self.set_type = self.fl_types[0] # initial file filter setting
	186	self.fl_ext = self.fl_ext_types[self.set_type]
	187
b881fc78 RD	188	self.sel_type = wx.ComboBox(self, -1, self.set_type, (image_posx , self.type_posy),
	189	(150, -1), self.fl_types, wx.CB_DROPDOWN)
	190	self.Bind(wx.EVT_COMBOBOX, self.OnSetType, self.sel_type)
d14a1e28	191
b881fc78 RD	192	self.image_view = ImageView( self, pos=(image_posx, image_posy),
b881fc78 RD	193	size=(image_sizex, image_sizey))
d14a1e28 RD	194
	195	self.y_pos = self.y_pos + height + 20
	196
622a19f0	197	btn = wx.Button(self, -1, ' Select ', (100, self.y_pos), size)
b881fc78	198	self.Bind(wx.EVT_BUTTON, self.OnOk, btn)
d14a1e28	199
b881fc78	200	wx.Button(self, wx.ID_CANCEL, 'Cancel', (250, self.y_pos), size)
d14a1e28 RD	201
d14a1e28 RD	202	self.y_pos = self.y_pos + self.delta
b881fc78	203	fsize = (400, self.y_pos + 50) # resize dialog for final vertical position
d14a1e28 RD	204	self.SetSize(fsize)
	205
	206	self.ResetFiles()
	207
	208	def GetFiles(self): # get the file list using directory and extension values
	209	if self.fl_ext == "All":
	210	all_files = []
b881fc78	211
d14a1e28 RD	212	for ftypes in self.fl_types[1:-1]: # get list of all available image types
d14a1e28 RD	213	filter = self.fl_ext_types[ftypes]
b881fc78	214	#print "filter = ", filter
d14a1e28 RD	215	self.fl_val = FindFiles(self, self.set_dir, filter)
d14a1e28 RD	216	all_files = all_files + self.fl_val.files # add to list of files
b881fc78	217
d14a1e28 RD	218	self.fl_list = all_files
	219	else:
	220	self.fl_val = FindFiles(self, self.set_dir, self.fl_ext)
	221	self.fl_list = self.fl_val.files
	222
	223	self.fl_list.sort() # sort the file list
	224
	225	def DisplayDir(self): # display the working directory
b881fc78	226	wx.StaticText(self, -1, self.set_dir, (self.dir_x, self.dir_y), (250, -1))
d14a1e28 RD	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
b881fc78	246	dlg = wx.DirDialog(self)
d14a1e28	247	dlg.SetPath(self.set_dir)
b881fc78 RD	248
b881fc78 RD	249	if dlg.ShowModal() == wx.ID_OK:
d14a1e28 RD	250	self.set_dir = dlg.GetPath()
d14a1e28 RD	251	self.ResetFiles()
b881fc78	252
d14a1e28 RD	253	dlg.Destroy()
	254
	255	def ResetFiles(self): # refresh the display with files and initial image
	256	self.DisplayDir()
	257	self.GetFiles()
b881fc78 RD	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:
d14a1e28 RD	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
b881fc78	294	self.EndModal(wx.ID_CANCEL)
d14a1e28 RD	295
	296	def OnOk(self, event):
	297	self.result = self.set_file
b881fc78	298	self.EndModal(wx.ID_OK)
d14a1e28 RD	299
	300
	301	def OnFileDlg(self):
b881fc78 RD	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:
d14a1e28 RD	306	path = dlg.GetPath()
	307	else:
	308	path = None
b881fc78	309
d14a1e28 RD	310	dlg.Destroy()
	311	return path
	312
	313	class 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)
b881fc78	321
d14a1e28 RD	322	for i in os.listdir(dir):
	323	if i == "." or i == "..":
	324	continue
b881fc78	325
d14a1e28 RD	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
b881fc78	338
d14a1e28 RD	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
b881fc78	348
d14a1e28 RD	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:]