wxPython/wxPython/lib/imagebrowser.py

   1 #----------------------------------------------------------------------------
   2 # Name:         BrowseImage.py
   3 # Purpose:      Display and Select Image Files
   4 #
   5 # Author:       Lorne White
   6 #
   7 # Version:      0.9
   8 # Date:         August 15, 2001
   9 # Licence:      wxWindows license
  10 #----------------------------------------------------------------------------
  11
  12 import os, sys, string
  13 from wxPython.wx import *
  14 dir_path = os.getcwd()
  15
  16 #---------------------------------------------------------------------------
  17
  18 def ConvertBMP(file_nm):
  19     if file_nm is None:
  20         return None
  21
  22     fl_fld = os.path.splitext(file_nm)
  23     ext = fl_fld[1]
  24     ext = string.lower(ext[1:])
  25     if ext == 'bmp':
  26         image = wxImage(file_nm, wxBITMAP_TYPE_BMP)
  27     elif ext == 'gif':
  28         image = wxImage(file_nm, wxBITMAP_TYPE_GIF)
  29     elif ext == 'png':
  30         image = wxImage(file_nm, wxBITMAP_TYPE_PNG)
  31     elif ext == 'jpg':
  32         image = wxImage(file_nm, wxBITMAP_TYPE_JPEG)
  33     else:
  34         image = None
  35
  36     return image
  37
  38 def GetSize(file_nm):       # for scaling image values
  39     image = ConvertBMP(file_nm)
  40     bmp = image.ConvertToBitmap()
  41     size = bmp.GetWidth(), bmp.GetHeight()
  42     return size
  43
  44 class ImageView(wxWindow):
  45     def __init__(self, parent, id=-1, pos=wxDefaultPosition, size=wxDefaultSize):
  46         wxWindow.__init__(self, parent, id, pos, size)
  47         self.win = parent
  48         self.image = None
  49         self.back_color = 'WHITE'
  50         self.border_color = 'BLACK'
  51
  52         self.image_sizex = size.width
  53         self.image_sizey = size.height
  54         self.image_posx = pos.x
  55         self.image_posy = pos.y
  56         EVT_PAINT(self, self.OnPaint)
  57
  58         wxInitAllImageHandlers()
  59
  60     def OnPaint(self, event):
  61         dc = wxPaintDC(self)
  62         self.DrawImage(dc)
  63
  64     def DrawImage(self, dc):
  65         dc.BeginDrawing()
  66         self.DrawImage(dc)
  67         dc.EndDrawing()
  68
  69     def SetValue(self, file_nm):    # display the selected file in the panel
  70         image = ConvertBMP(file_nm)
  71         self.image = image
  72         self.Refresh()
  73
  74     def DrawBorder(self, dc):
  75         brush = wxBrush(wxNamedColour(self.back_color), wxSOLID)
  76         dc.SetBrush(brush)
  77         dc.SetPen(wxPen(wxNamedColour(self.border_color), 1))
  78         dc.DrawRectangle(0, 0, self.image_sizex, self.image_sizey)
  79
  80     def DrawImage(self, dc):
  81         try:
  82             image = self.image
  83         except:
  84             return
  85
  86         self.DrawBorder(dc)
  87         if image is None:
  88             return
  89
  90         bmp = image.ConvertToBitmap()
  91
  92         iwidth = bmp.GetWidth()   # dimensions of image file
  93         iheight = bmp.GetHeight()
  94
  95         diffx = (self.image_sizex - iwidth)/2   # center calc
  96         if iwidth >= self.image_sizex -10:      # if image width fits in window adjust
  97             diffx = 5
  98             iwidth = self.image_sizex - 10
  99
 100         diffy = (self.image_sizey - iheight)/2  # center calc
 101         if iheight >= self.image_sizey - 10:    # if image height fits in window adjust
 102             diffy = 5
 103             iheight = self.image_sizey - 10
 104
 105         image.Rescale(iwidth, iheight)      # rescale to fit the window
 106         image.ConvertToBitmap()
 107         bmp = image.ConvertToBitmap()
 108         dc.DrawBitmap(bmp, diffx, diffy)        # draw the image to window
 109
 110
 111 class ImageDialog(wxDialog):
 112     def __init__(self, parent, set_dir = None):
 113         wxDialog.__init__(self, parent, -1, "Image Browser", wxPyDefaultPosition, wxSize(400, 400))
 114
 115         self.x_pos = 30     # initial display positions
 116         self.y_pos = 20
 117         self.delta = 20
 118
 119         size = wxSize(80, 25)
 120
 121         self.set_dir = os.getcwd()
 122
 123         if set_dir != None:
 124             if os.path.exists(set_dir):     # set to working directory if nothing set
 125                 self.set_dir = set_dir
 126
 127         self.dir_x = self.x_pos
 128         self.dir_y = self.y_pos
 129         self.DisplayDir()       # display the directory value
 130
 131         self.y_pos = self.y_pos + self.delta
 132
 133         mID = NewId()
 134         wxButton(self, mID, ' Set Directory ', wxPoint(self.x_pos, self.y_pos), size).SetDefault()
 135         EVT_BUTTON(self, mID, self.SetDirect)
 136
 137         self.type_posy = self.y_pos     # save the y position for the image type combo
 138
 139         self.fl_ext = '*.bmp'   # initial setting for file filtering
 140         self.GetFiles()     # get the file list
 141
 142         self.y_pos = self.y_pos + self.delta + 10
 143
 144         self.list_height = 150
 145
 146     # List of Labels
 147         mID = NewId()
 148         self.tb = tb = wxListBox(self, mID, wxPoint(self.x_pos, self.y_pos), wxSize(160, self.list_height), self.fl_list, wxLB_SINGLE)
 149         EVT_LISTBOX(self, mID, self.OnListClick)
 150         EVT_LISTBOX_DCLICK(self, mID, self.OnListDClick)
 151
 152         width, height = self.tb.GetSizeTuple()
 153         image_posx = self.x_pos + width + 20       # positions for setting the image window
 154         image_posy = self.y_pos
 155         image_sizex = 150
 156         image_sizey = self.list_height
 157
 158         self.fl_types = ["Bmp", "Gif", "Png", "Jpg"]
 159         self.fl_ext_types = { "Bmp": "*.bmp", "Gif": "*.gif", "Png": "*.png", "Jpg": "*.jpg" }
 160
 161         self.set_type = self.fl_types[0]    # initial file filter setting
 162
 163         mID = NewId()
 164         self.sel_type = wxComboBox(self, mID, self.set_type, wxPoint(image_posx , self.type_posy), wxSize(150, -1), self.fl_types, wxCB_DROPDOWN)
 165         EVT_COMBOBOX(self, mID, self.OnSetType)
 166
 167         self.image_view = ImageView(self, pos=wxPoint(image_posx, image_posy), size=wxSize(image_sizex, image_sizey))
 168
 169         self.y_pos = self.y_pos + height + 20
 170
 171         mID = NewId()
 172         wxButton(self, mID, ' Select ', wxPoint(100, self.y_pos), size).SetDefault()
 173         EVT_BUTTON(self, mID, self.OnOk)
 174
 175         wxButton(self, wxID_CANCEL, 'Cancel', wxPoint(250, self.y_pos), size)
 176
 177         self.y_pos = self.y_pos + self.delta
 178         fsize = wxSize(400, self.y_pos + 50)    # resize dialog for final vertical position
 179         self.SetSize(fsize)
 180
 181         self.ResetFiles()
 182
 183     def GetFiles(self):     # get the file list using directory and extension values
 184         self.fl_val = FindFiles(self, self.set_dir, self.fl_ext)
 185         self.fl_list = self.fl_val.files
 186
 187     def DisplayDir(self):       # display the working directory
 188         wxStaticText(self, -1, self.set_dir, wxPoint(self.dir_x, self.dir_y), wxSize(250, -1))
 189
 190     def OnSetType(self, event):
 191         val = event.GetString()      # get file type value
 192         self.fl_ext = self.fl_ext_types[val]
 193         self.ResetFiles()
 194
 195     def OnListDClick(self, event):
 196         self.OnOk(0)
 197
 198     def OnListClick(self, event):
 199         val = event.GetSelection()
 200         self.SetListValue(val)
 201
 202     def SetListValue(self, val):
 203         file_nm = self.fl_list[val]
 204         self.set_file = file_val = os.path.join(self.set_dir, file_nm)
 205         self.image_view.SetValue(file_val)
 206
 207     def SetDirect(self, event):     # set the new directory
 208         dlg = wxDirDialog(self)
 209         dlg.SetPath(self.set_dir)
 210         if dlg.ShowModal() == wxID_OK:
 211             self.set_dir = dlg.GetPath()
 212             self.ResetFiles()
 213         dlg.Destroy()
 214
 215     def ResetFiles(self):   # refresh the display with files and initial image
 216         self.DisplayDir()
 217         self.GetFiles()
 218         self.tb.Set(self.fl_list)
 219         try:
 220             self.tb.SetSelection(0)
 221             self.SetListValue(0)
 222         except:
 223             self.image_view.SetValue(None)
 224
 225     def GetFile(self):
 226         return self.set_file
 227
 228     def GetDirectory(self):
 229         return self.set_dir
 230
 231     def OnCancel(self, event):
 232         self.result = None
 233         self.EndModal(wxID_CANCEL)
 234
 235     def OnOk(self, event):
 236         self.result = self.set_file
 237         self.EndModal(wxID_OK)
 238
 239
 240 def OnFileDlg(self):
 241     dlg = wxFileDialog(self, "Choose an Image File", ".", "", "Bmp (*.bmp)|*.bmp|JPEG (*.jpg)|*.jpg", wxOPEN)
 242     if dlg.ShowModal() == wxID_OK:
 243         path = dlg.GetPath()
 244     else:
 245         path = None
 246     dlg.Destroy()
 247     return path
 248
 249 class FindFiles:
 250     def __init__(self, parent, dir, mask):
 251         filelist = []
 252         dirlist = [".."]
 253         self.dir = dir
 254         self.file = ""
 255         mask = string.upper(mask)
 256         self.MakeRegex(mask)
 257         for i in os.listdir(dir):
 258             if i == "." or i == "..":
 259                 continue
 260             path = os.path.join(dir, i)
 261             path = string.upper(path)
 262             value = string.upper(i)
 263
 264             if self.regex.match(value) == len(value):
 265                 filelist.append(i)
 266
 267             self.files = filelist
 268
 269     def MakeRegex(self, pattern):
 270         import regex
 271         f = ""  # Set up a regex for file names
 272         for ch in pattern:
 273             if ch == "*":
 274                 f = f + ".*"
 275             elif ch == ".":
 276                 f = f + "\."
 277             elif ch == "?":
 278                 f = f + "."
 279             else:
 280                 f = f + ch
 281
 282         self.regex = regex.compile(f)
 283
 284     def StripExt(self, file_nm):
 285         fl_fld = os.path.splitext(file_nm)
 286         fl_name = fl_fld[0]
 287         ext = fl_fld[1]
 288         return ext[1:]