| Index: imageConversion.py | 
| =================================================================== | 
| new file mode 100644 | 
| --- /dev/null | 
| +++ b/imageConversion.py | 
| @@ -0,0 +1,123 @@ | 
| +# coding: utf-8 | 
| + | 
| +# This file is part of the Adblock Plus build tools, | 
| +# Copyright (C) 2006-2013 Eyeo GmbH | 
| +# | 
| +# Adblock Plus is free software: you can redistribute it and/or modify | 
| +# it under the terms of the GNU General Public License version 3 as | 
| +# published by the Free Software Foundation. | 
| +# | 
| +# Adblock Plus is distributed in the hope that it will be useful, | 
| +# but WITHOUT ANY WARRANTY; without even the implied warranty of | 
| +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
| +# GNU General Public License for more details. | 
| +# | 
| +# You should have received a copy of the GNU General Public License | 
| +# along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>. | 
| + | 
| +import os | 
| +import re | 
| +from StringIO import StringIO | 
| + | 
| +import PIL.Image | 
| +import PIL.ImageOps | 
| + | 
| +def get_alpha(image): | 
| +  if image.mode in ('RGBA', 'LA'): | 
| +    return image.split()[image.getbands().index('A')] | 
| + | 
| +  # images with palette doesn't have an alpha channel, but | 
| +  # can store transparent pixels by using 0 as image data. | 
| +  # In order to don't discard those information, we create an | 
| +  # alpha channel from all the pixels that are not transparent. | 
| +  if image.mode == 'P': | 
| +    return image.point([0] + [255] * 255, 'L') | 
| + | 
| +def ensure_same_mode(im1, im2): | 
| +  # if both images already have the same mode (and palette, in | 
| +  # case of mode P), don't convert anything. Images with mode P, | 
| +  # and a different palette, are the only case where images | 
| +  # using the same mode, will be incompatible with each other. | 
| +  if im1.mode == im2.mode and (im1.mode != 'P' or im1.getpalette() == im2.getpalette()): | 
| +    return (im1, im2) | 
| + | 
| +  # if any given image has a mode that supports colors convert both | 
| +  # images to RGB or RGBA, otherwise convert both images to L or LA. | 
| +  # If any given image has an alpha channel (or mode P which | 
| +  # can store transparent pixels too) convert both images | 
| +  # to RGBA or LA, otherwise convert both images to RGB or L. | 
| +  mode = max( | 
| +    PIL.Image.getmodebase(im1.mode), | 
| +    PIL.Image.getmodebase(im2.mode), | 
| + | 
| +    key=('L', 'RGB').index | 
| +  ) | 
| + | 
| +  if any(im.mode in ('RGBA', 'LA', 'P') for im in (im1, im2)): | 
| +    mode += 'A' | 
| + | 
| +  return ( | 
| +    im1 if im1.mode == mode else im1.convert(mode), | 
| +    im2 if im2.mode == mode else im2.convert(mode), | 
| +  ) | 
| + | 
| +def filter_grayscale(image, baseDir): | 
| +  alpha = get_alpha(image) | 
| +  image = image.convert('L') | 
| + | 
| +  if alpha: | 
| +    image.putalpha(alpha) | 
| + | 
| +  return image | 
| + | 
| +def filter_contrastToAlpha(image, baseDir): | 
| +  alpha = PIL.Image.new('L', image.size, 255) | 
| +  alpha.paste(image, mask=get_alpha(image)) | 
| +  alpha = PIL.ImageOps.invert(alpha) | 
| +  alpha = PIL.ImageOps.autocontrast(alpha) | 
| + | 
| +  return PIL.Image.merge('LA', [PIL.Image.new('L', image.size), alpha]) | 
| + | 
| +def filter_blend(image, baseDir, *args): | 
| +  if len(args) == 2: | 
| +    filename, opacity = args | 
| + | 
| +    overlay = PIL.Image.open(os.path.join( | 
| +      baseDir, | 
| +      *filename.split('/') | 
| +    )) | 
| +  else: | 
| +    red, green, blue, opacity = args | 
| + | 
| +    overlay = PIL.Image.new('RGB', image.size, ( | 
| +      int(red), | 
| +      int(green), | 
| +      int(blue), | 
| +    )) | 
| + | 
| +    # if the background image has an alpha channel copy it to | 
| +    # the overlay, so that transparent areas stay transparent. | 
| +    alpha = get_alpha(image) | 
| +    if alpha: | 
| +      overlay.putalpha(alpha) | 
| + | 
| +  image, overlay = ensure_same_mode(image, overlay) | 
| +  return PIL.Image.blend(image, overlay, float(opacity)) | 
| + | 
| +def convertImages(params, files): | 
| +  metadata = params['metadata'] | 
| + | 
| +  for filename, chain in metadata.items('convert_img'): | 
| +    baseDir = os.path.dirname(metadata.option_source('convert_img', filename)) | 
| +    steps = re.split(r'\s*->\s*', chain) | 
| +    image = PIL.Image.open(os.path.join(baseDir, *steps.pop(0).split('/'))) | 
| + | 
| +    for step in steps: | 
| +      filter, args = re.match(r'([^(]+)(?:\((.*)\))?', step).groups() | 
| +      args = re.split(r'\s*,\s*', args) if args else () | 
| +      image = globals()['filter_' + filter](image, baseDir, *args) | 
| + | 
| +    f = StringIO() | 
| +    f.name = filename | 
| +    image.save(f) | 
| +    files[filename] = f.getvalue() | 
|  |