wxPython/wxPython/tools/img2img.py

   1 #----------------------------------------------------------------------
   2 # Name:        wxPython.tools.img2img
   3 # Purpose:     Common routines for the image converter utilities.
   4 #
   5 # Author:      Robin Dunn
   6 #
   7 # RCS-ID:      $Id$
   8 # Copyright:   (c) 2002 by Total Control Software
   9 # Licence:     wxWindows license
  10 #----------------------------------------------------------------------
  11
  12
  13 import sys, os, glob, getopt
  14 from wxPython.wx import *
  15
  16 if wxPlatform == "__WXGTK__":
  17     # some bitmap related things need to have a wxApp initialized...
  18     app = wxPySimpleApp()
  19
  20 wxInitAllImageHandlers()
  21
  22 def convert(file, maskClr, outputDir, outputName, outType, outExt):
  23     if os.path.splitext(file)[1].lower() == ".ico":
  24         icon = wxIcon(file, wxBITMAP_TYPE_ICO)
  25         img = wxBitmapFromIcon(icon)
  26     else:
  27         img = wxBitmap(file, wxBITMAP_TYPE_ANY)
  28
  29     if not img.Ok():
  30         return 0, file + " failed to load!"
  31     else:
  32         if maskClr:
  33             om = img.GetMask()
  34             mask = wxMaskColour(img, maskClr)
  35             img.SetMask(mask)
  36             if om is not None:
  37                 om.Destroy()
  38         if outputName:
  39             newname = outputName
  40         else:
  41             newname = os.path.join(outputDir,
  42                                    os.path.basename(os.path.splitext(file)[0]) + outExt)
  43         if img.SaveFile(newname, outType):
  44             return 1, file + " converted to " + newname
  45         else:
  46             img = wxImageFromBitmap(img)
  47             if img.SaveFile(newname, outType):
  48                 return 1, "ok"
  49             else:
  50                 return 0, file + " failed to save!"
  51
  52
  53
  54
  55 def main(args, outType, outExt, doc):
  56     if not args or ("-h" in args):
  57         print doc
  58         return
  59
  60     outputDir = ""
  61     maskClr = None
  62     outputName = None
  63
  64     try:
  65         opts, fileArgs = getopt.getopt(args, "m:n:o:")
  66     except getopt.GetoptError:
  67         print __doc__
  68         return
  69
  70     for opt, val in opts:
  71         if opt == "-m":
  72             maskClr = val
  73         elif opt == "-n":
  74             outputName = val
  75         elif opt == "-o":
  76             outputDir = val
  77
  78     if not fileArgs:
  79         print doc
  80         return
  81
  82     for arg in fileArgs:
  83         for file in glob.glob(arg):
  84             if not os.path.isfile(file):
  85                 continue
  86             ok, msg = convert(file, maskClr, outputDir, outputName,
  87                               outType, outExt)
  88             print msg
  89