Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: imageConversion.py

Issue 4826241328742400: Made import of PIL compatible to environments where you have to import Image directly (Closed)
Patch Set: Created Nov. 15, 2013, 9:40 a.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # coding: utf-8 1 # coding: utf-8
2 2
3 # This file is part of the Adblock Plus build tools, 3 # This file is part of the Adblock Plus build tools,
4 # Copyright (C) 2006-2013 Eyeo GmbH 4 # Copyright (C) 2006-2013 Eyeo GmbH
5 # 5 #
6 # Adblock Plus is free software: you can redistribute it and/or modify 6 # Adblock Plus is free software: you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License version 3 as 7 # it under the terms of the GNU General Public License version 3 as
8 # published by the Free Software Foundation. 8 # published by the Free Software Foundation.
9 # 9 #
10 # Adblock Plus is distributed in the hope that it will be useful, 10 # Adblock Plus is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of 11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details. 13 # GNU General Public License for more details.
14 # 14 #
15 # You should have received a copy of the GNU General Public License 15 # You should have received a copy of the GNU General Public License
16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. 16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>.
17 17
18 import os 18 import os
19 import re 19 import re
20 from StringIO import StringIO 20 from StringIO import StringIO
21 21
22 import PIL.Image 22 try:
23 from PIL import Image
24 from PIL import ImageOps
25 except ImportError:
26 import Image
27 import ImageOps
23 28
24 def get_alpha(image): 29 def get_alpha(image):
25 if image.mode in ('RGBA', 'LA'): 30 if image.mode in ('RGBA', 'LA'):
26 return image.split()[image.getbands().index('A')] 31 return image.split()[image.getbands().index('A')]
27 32
28 if image.mode == 'P': 33 if image.mode == 'P':
29 transparency = image.info.get('transparency') 34 transparency = image.info.get('transparency')
30 35
31 if transparency is not None: 36 if transparency is not None:
32 table = [255] * 256 37 table = [255] * 256
33 table[transparency] = 0 38 table[transparency] = 0
34 39
35 return image.point(table, 'L') 40 return image.point(table, 'L')
36 41
37 def ensure_same_mode(im1, im2): 42 def ensure_same_mode(im1, im2):
38 # if both images already have the same mode (and palette, in 43 # if both images already have the same mode (and palette, in
39 # case of mode P), don't convert anything. Images with mode P, 44 # case of mode P), don't convert anything. Images with mode P,
40 # and a different palette, are the only case where images 45 # and a different palette, are the only case where images
41 # using the same mode, will be incompatible with each other. 46 # using the same mode, will be incompatible with each other.
42 if im1.mode == im2.mode and (im1.mode != 'P' or im1.getpalette() == im2.getpal ette()): 47 if im1.mode == im2.mode and (im1.mode != 'P' or im1.getpalette() == im2.getpal ette()):
43 return (im1, im2) 48 return (im1, im2)
44 49
45 # if any given image has a mode that supports colors convert both 50 # if any given image has a mode that supports colors convert both
46 # images to RGB or RGBA, otherwise convert both images to L or LA. 51 # images to RGB or RGBA, otherwise convert both images to L or LA.
47 # If any given image has an alpha channel (or mode P which 52 # If any given image has an alpha channel (or mode P which
48 # can store transparent pixels too) convert both images 53 # can store transparent pixels too) convert both images
49 # to RGBA or LA, otherwise convert both images to RGB or L. 54 # to RGBA or LA, otherwise convert both images to RGB or L.
50 mode = max( 55 mode = max(
51 PIL.Image.getmodebase(im1.mode), 56 Image.getmodebase(im1.mode),
52 PIL.Image.getmodebase(im2.mode), 57 Image.getmodebase(im2.mode),
53 58
54 key=('L', 'RGB').index 59 key=('L', 'RGB').index
55 ) 60 )
56 61
57 if any(im.mode in ('RGBA', 'LA', 'P') for im in (im1, im2)): 62 if any(im.mode in ('RGBA', 'LA', 'P') for im in (im1, im2)):
58 mode += 'A' 63 mode += 'A'
59 64
60 return ( 65 return (
61 im1 if im1.mode == mode else im1.convert(mode), 66 im1 if im1.mode == mode else im1.convert(mode),
62 im2 if im2.mode == mode else im2.convert(mode), 67 im2 if im2.mode == mode else im2.convert(mode),
63 ) 68 )
64 69
65 def filter_grayscale(image, baseDir): 70 def filter_grayscale(image, baseDir):
66 alpha = get_alpha(image) 71 alpha = get_alpha(image)
67 image = image.convert('L') 72 image = image.convert('L')
68 73
69 if alpha: 74 if alpha:
70 image.putalpha(alpha) 75 image.putalpha(alpha)
71 76
72 return image 77 return image
73 78
74 def filter_contrastToAlpha(image, baseDir): 79 def filter_contrastToAlpha(image, baseDir):
75 import PIL.ImageOps 80 alpha = Image.new('L', image.size, 255)
81 alpha.paste(image, mask=get_alpha(image))
82 alpha = ImageOps.invert(alpha)
83 alpha = ImageOps.autocontrast(alpha)
76 84
77 alpha = PIL.Image.new('L', image.size, 255) 85 return Image.merge('LA', [Image.new('L', image.size), alpha])
78 alpha.paste(image, mask=get_alpha(image))
79 alpha = PIL.ImageOps.invert(alpha)
80 alpha = PIL.ImageOps.autocontrast(alpha)
81
82 return PIL.Image.merge('LA', [PIL.Image.new('L', image.size), alpha])
83 86
84 def filter_blend(image, baseDir, *args): 87 def filter_blend(image, baseDir, *args):
85 if len(args) == 2: 88 if len(args) == 2:
86 filename, opacity = args 89 filename, opacity = args
87 90
88 overlay = PIL.Image.open(os.path.join( 91 overlay = Image.open(os.path.join(
89 baseDir, 92 baseDir,
90 *filename.split('/') 93 *filename.split('/')
91 )) 94 ))
92 else: 95 else:
93 red, green, blue, opacity = args 96 red, green, blue, opacity = args
94 97
95 overlay = PIL.Image.new('RGB', image.size, ( 98 overlay = Image.new('RGB', image.size, (
96 int(red), 99 int(red),
97 int(green), 100 int(green),
98 int(blue), 101 int(blue),
99 )) 102 ))
100 103
101 # if the background image has an alpha channel copy it to 104 # if the background image has an alpha channel copy it to
102 # the overlay, so that transparent areas stay transparent. 105 # the overlay, so that transparent areas stay transparent.
103 alpha = get_alpha(image) 106 alpha = get_alpha(image)
104 if alpha: 107 if alpha:
105 overlay.putalpha(alpha) 108 overlay.putalpha(alpha)
106 109
107 image, overlay = ensure_same_mode(image, overlay) 110 image, overlay = ensure_same_mode(image, overlay)
108 return PIL.Image.blend(image, overlay, float(opacity)) 111 return Image.blend(image, overlay, float(opacity))
109 112
110 def convertImages(params, files): 113 def convertImages(params, files):
111 metadata = params['metadata'] 114 metadata = params['metadata']
112 115
113 for filename, chain in metadata.items('convert_img'): 116 for filename, chain in metadata.items('convert_img'):
114 baseDir = os.path.dirname(metadata.option_source('convert_img', filename)) 117 baseDir = os.path.dirname(metadata.option_source('convert_img', filename))
115 steps = re.split(r'\s*->\s*', chain) 118 steps = re.split(r'\s*->\s*', chain)
116 image = PIL.Image.open(os.path.join(baseDir, *steps.pop(0).split('/'))) 119 image = Image.open(os.path.join(baseDir, *steps.pop(0).split('/')))
117 120
118 for step in steps: 121 for step in steps:
119 filter, args = re.match(r'([^(]+)(?:\((.*)\))?', step).groups() 122 filter, args = re.match(r'([^(]+)(?:\((.*)\))?', step).groups()
120 args = re.split(r'\s*,\s*', args) if args else () 123 args = re.split(r'\s*,\s*', args) if args else ()
121 image = globals()['filter_' + filter](image, baseDir, *args) 124 image = globals()['filter_' + filter](image, baseDir, *args)
122 125
123 f = StringIO() 126 f = StringIO()
124 f.name = filename 127 f.name = filename
125 image.save(f) 128 image.save(f)
126 files[filename] = f.getvalue() 129 files[filename] = f.getvalue()
OLDNEW
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld