annotate core/image.py @ 670:c5c0f8604c4b
Update requirements.txt to include ftfy.
author |
Brian Neal <bgneal@gmail.com> |
date |
Sat, 25 May 2013 17:44:18 -0500 |
parents |
ee87ea74d46b |
children |
234726f5a47a |
rev |
line source |
bgneal@265
|
1 """
|
bgneal@265
|
2 This file contains common utility functions for manipulating images for
|
bgneal@265
|
3 the rest of the applications in the project.
|
bgneal@265
|
4 """
|
bgneal@265
|
5 from PIL import ImageFile
|
bgneal@265
|
6 from PIL import Image
|
bgneal@265
|
7
|
bgneal@265
|
8
|
bgneal@265
|
9 def parse_image(file):
|
bgneal@265
|
10 """
|
bgneal@265
|
11 Returns a PIL Image from the supplied Django file object.
|
bgneal@265
|
12 Throws IOError if the file does not parse as an image file or some other
|
bgneal@265
|
13 I/O error occurred.
|
bgneal@265
|
14
|
bgneal@265
|
15 """
|
bgneal@265
|
16 parser = ImageFile.Parser()
|
bgneal@265
|
17 for chunk in file.chunks():
|
bgneal@265
|
18 parser.feed(chunk)
|
bgneal@265
|
19 image = parser.close()
|
bgneal@265
|
20 return image
|
bgneal@265
|
21
|
bgneal@265
|
22
|
bgneal@265
|
23 def downscale_image_square(image, size):
|
bgneal@265
|
24 """
|
bgneal@265
|
25 Scale an image to the square dimensions given by size (in pixels).
|
bgneal@265
|
26 The new image is returned.
|
bgneal@265
|
27 If the image is already smaller than (size, size) then no scaling
|
bgneal@265
|
28 is performed and the image is returned unchanged.
|
bgneal@265
|
29
|
bgneal@265
|
30 """
|
bgneal@265
|
31 # don't upscale
|
bgneal@265
|
32 if (size, size) >= image.size:
|
bgneal@265
|
33 return image
|
bgneal@265
|
34
|
bgneal@265
|
35 (w, h) = image.size
|
bgneal@265
|
36 if w > h:
|
bgneal@265
|
37 diff = (w - h) / 2
|
bgneal@265
|
38 image = image.crop((diff, 0, w - diff, h))
|
bgneal@265
|
39 elif h > w:
|
bgneal@265
|
40 diff = (h - w) / 2
|
bgneal@265
|
41 image = image.crop((0, diff, w, h - diff))
|
bgneal@265
|
42 image = image.resize((size, size), Image.ANTIALIAS)
|
bgneal@265
|
43 return image
|