+#!/bin/env python
+#----------------------------------------------------------------------------
+# Name: BrowseImage.py
+# Purpose: Display and Select Image Files
+#
+# Author: Lorne White
+#
+# Version: 0.6
+# Date: March 27, 2001
+# Licence: wxWindows license
+#----------------------------------------------------------------------------
+
+import os, sys, string
+from wxPython.wx import *
+dir_path = os.getcwd()
+
+#---------------------------------------------------------------------------
+
+def ConvertBMP(file_nm):
+ fl_fld = os.path.splitext(file_nm)
+ ext = fl_fld[1]
+ ext = string.lower(ext[1:])
+ if ext == 'bmp':
+ image = wxImage(file_nm, wxBITMAP_TYPE_BMP)
+ elif ext == 'gif':
+ image = wxImage(file_nm, wxBITMAP_TYPE_GIF)
+ elif ext == 'png':
+ image = wxImage(file_nm, wxBITMAP_TYPE_PNG)
+ elif ext == 'jpg':
+ image = wxImage(file_nm, wxBITMAP_TYPE_JPEG)
+ else:
+ image = None
+
+ return image
+
+class ImageView:
+ def __init__(self, iparent, ipos, isize):
+ self.win = iparent
+ self.ImagePanel = wxPanel(size = isize, parent = iparent, id = -1, name = 'backgroundPanel', style = wxSIMPLE_BORDER | wxCLIP_CHILDREN, pos= ipos)
+ self.clear = wxColour(255, 255, 255)
+ self.ImagePanel.SetBackgroundColour(self.clear) # clear the panel
+
+ self.image_sizex = isize.width
+ self.image_sizey = isize.height
+ self.image_posx = ipos.x
+ self.image_posy = ipos.y
+
+ wxInitAllImageHandlers()
+
+ def SetValue(self, file_nm): # display the selected file in the panel
+ image = ConvertBMP(file_nm).ConvertToBitmap()
+ if image is None:
+ return
+
+ iwidth = image.GetWidth() # dimensions of image file
+ iheight = image.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
+
+ self.ClearPanel()
+ wxStaticBitmap(self.win, -1, image, wxPoint(self.image_posx+diffx, self.image_posy+diffy), wxSize(iwidth, iheight ))
+
+ def ClearPanel(self): # clear the image panel
+ self.ImagePanel.SetBackgroundColour(self.clear)
+ self.ImagePanel.Refresh()
+
+class ImageDialog(wxDialog):
+ def __init__(self, parent, set_dir = None):
+ wxDialog.__init__(self, parent, -1, "Image Browser", wxPyDefaultPosition, wxSize(400, 400))
+
+ self.x_pos = 30 # initial display positions
+ self.y_pos = 20
+ self.delta = 20
+
+ size = wxSize(80, 25)
+
+ self.set_dir = os.getcwd()
+
+ 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
+
+ mID = NewId()
+ wxButton(self, mID, ' Set Directory ', wxPoint(self.x_pos, self.y_pos), size).SetDefault()
+ EVT_BUTTON(self, mID, self.SetDirect)
+
+ 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
+ mID = NewId()
+ self.tb = tb = wxListBox(self, mID, wxPoint(self.x_pos, self.y_pos), wxSize(160, self.list_height), self.fl_list, wxLB_SINGLE)
+ EVT_LISTBOX(self, mID, self.OnListClick)
+ EVT_LISTBOX_DCLICK(self, mID, self.OnListDClick)
+
+ width, height = self.tb.GetSizeTuple()
+ 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 = ["Bmp", "Gif", "Png", "Jpg"]
+ self.fl_ext_types = { "Bmp": "*.bmp", "Gif": "*.gif", "Png": "*.png", "Jpg": "*.jpg" }
+
+ self.set_type = self.fl_types[0] # initial file filter setting
+
+ mID = NewId()
+ self.sel_type = wxComboBox(self, mID, self.set_type, wxPoint(image_posx , self.type_posy), wxSize(150, -1), self.fl_types, wxCB_DROPDOWN)
+ EVT_COMBOBOX(self, mID, self.OnSetType)
+
+ self.image_view = ImageView(self, wxPoint(image_posx, image_posy), wxSize(image_sizex, image_sizey))
+
+ self.y_pos = self.y_pos + height + 20
+
+ mID = NewId()
+ wxButton(self, mID, ' Select ', wxPoint(100, self.y_pos), size).SetDefault()
+ EVT_BUTTON(self, mID, self.OnOk)
+
+ wxButton(self, wxID_CANCEL, 'Cancel', wxPoint(250, self.y_pos), size)
+
+ self.y_pos = self.y_pos + self.delta
+ fsize = wxSize(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
+ self.fl_val = FindFiles(self, self.set_dir, self.fl_ext)
+ self.fl_list = self.fl_val.files
+
+ def DisplayDir(self): # display the working directory
+ wxStaticText(self, -1, self.set_dir, wxPoint(self.dir_x, self.dir_y), wxSize(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 = wxDirDialog(self)
+ dlg.SetPath(self.set_dir)
+ if dlg.ShowModal() == wxID_OK:
+ self.set_dir = dlg.GetPath()
+ self.ResetFiles()
+ dlg.Destroy()
+
+ def ResetFiles(self): # refresh the display with files and initial image
+ self.image_view.ClearPanel()
+ self.DisplayDir()
+ self.GetFiles()
+ self.tb.Set(self.fl_list)
+ try:
+ self.tb.SetSelection(0)
+ self.SetListValue(0)
+ except:
+ pass
+
+ def GetFile(self):
+ return self.set_file
+
+ def OnCancel(self, event):
+ self.result = None
+ self.EndModal(wxID_CANCEL)
+
+ def OnOk(self, event):
+ self.result = self.set_file
+ self.EndModal(wxID_OK)
+
+
+def OnFileDlg(self):
+ dlg = wxFileDialog(self, "Choose an Image File", ".", "", "Bmp (*.bmp)|*.bmp|JPEG (*.jpg)|*.jpg", wxOPEN)
+ if dlg.ShowModal() == wxID_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 = string.upper(mask)
+ self.MakeRegex(mask)
+ for i in os.listdir(dir):
+ if i == "." or i == "..":
+ continue
+ path = os.path.join(dir, i)
+ path = string.upper(path)
+ value = string.upper(i)
+
+ if self.regex.match(value) == len(value):
+ filelist.append(i)
+
+ self.files = filelist
+
+ def MakeRegex(self, pattern):
+ import regex
+ 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
+
+ self.regex = regex.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:]
projects / wxWidgets.git / commitdiff
