Simple method to count colors in an image

This commit is contained in:
Vik Paruchuri
2013-01-29 17:15:02 -05:00
parent 255405d0bf
commit 5d4ad4b8ca
2 changed files with 29 additions and 1 deletions

View File

@@ -617,7 +617,6 @@ class CombinedOpenEndedModule(XModule):
return progress_object
class CombinedOpenEndedDescriptor(XmlDescriptor, EditingDescriptor):
"""
Module for adding combined open ended questions

View File

@@ -0,0 +1,29 @@
from PIL import Image
TRUSTED_IMAGE_DOMAINS = [
'wikipedia.com',
'wikipedia.net',
'wikipedia.org'
]
MAX_IMAGE_DIM = 150
MAX_COLORS_TO_COUNT = 16
MAX_COLORS = 5
class ImageProperties(object):
def __init__(self, image):
self.image = image
image_size = self.image.size
if image_size[0]> MAX_IMAGE_DIM or image_size[1] > MAX_IMAGE_DIM:
self.image = self.image.resize((MAX_IMAGE_DIM, MAX_IMAGE_DIM))
self.image_size = self.image.size
def count_colors(self):
colors = self.image.getcolors(MAX_COLORS_TO_COUNT)
if colors is None:
colors = MAX_COLORS_TO_COUNT
else:
colors = len(colors)