Skip to content

Commit

Permalink
Changed module and codec names
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed May 14, 2015
1 parent 799e831 commit 98fa49e
Show file tree
Hide file tree
Showing 2 changed files with 35 additions and 27 deletions.
42 changes: 19 additions & 23 deletions PIL/features.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,19 @@
from PIL import Image

modules = {
"PIL CORE": "PIL._imaging",
"TKINTER": "PIL._imagingtk",
"FREETYPE2": "PIL._imagingft",
"LITTLECMS2": "PIL._imagingcms",
"WEBP": "PIL._webp",
"Transparent WEBP": ("WEBP", "WebPDecoderBuggyAlpha")
"pil": "PIL._imaging",
"tkinter": "PIL._imagingtk",
"freetype2": "PIL._imagingft",
"littlecms2": "PIL._imagingcms",
"webp": "PIL._webp",
"transp_webp": ("WEBP", "WebPDecoderBuggyAlpha")
}


def check_module(feature):
if feature not in modules:
raise ValueError("Unknown module %s" % feature)

module = modules[feature]

method_to_call = None
Expand All @@ -34,38 +37,31 @@ def check_module(feature):

def get_supported_modules():
supported_modules = []
for feature in get_all_modules():
for feature in modules:
if check_module(feature):
supported_modules.append(feature)
return supported_modules


def get_all_modules():
# While the dictionary keys could be used here,
# a static list is used to maintain order
return ["PIL CORE", "TKINTER", "FREETYPE2",
"LITTLECMS2", "WEBP", "Transparent WEBP"]

codecs = {
"JPEG": "jpeg",
"JPEG 2000": "jpeg2k",
"ZLIB (PNG/ZIP)": "zip",
"LIBTIFF": "libtiff"
"jpg": "jpeg",
"jpg_2000": "jpeg2k",
"zlib": "zip",
"libtiff": "libtiff"
}


def check_codec(feature):
if feature not in codecs:
raise ValueError("Unknown codec %s" % feature)

codec = codecs[feature]

return codec + "_encoder" in dir(Image.core)


def get_supported_codecs():
supported_codecs = []
for feature in get_all_codecs():
for feature in codecs:
if check_codec(feature):
supported_codecs.append(feature)
return supported_codecs


def get_all_codecs():
return ["JPEG", "JPEG 2000", "ZLIB (PNG/ZIP)", "LIBTIFF"]
20 changes: 16 additions & 4 deletions selftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -174,8 +174,15 @@ def testimage():
print("Python modules loaded from", os.path.dirname(Image.__file__))
print("Binary modules loaded from", os.path.dirname(Image.core.__file__))
print("-"*68)
for feature in features.get_all_modules():
supported = features.check_module(feature)
for name, feature in [
("pil", "PIL CORE"),
("tkinter", "TKINTER"),
("freetype2", "FREETYPE2"),
("littlecms2", "LITTLECMS2"),
("webp", "WEBP"),
("transp_webp", "Transparent WEBP")
]:
supported = features.check_module(name)

if supported is None:
# A method was being tested, but the module required
Expand All @@ -185,8 +192,13 @@ def testimage():
print("---", feature, "support ok")
else:
print("***", feature, "support not installed")
for feature in features.get_all_codecs():
if features.check_codec(feature):
for name, feature in [
("jpg", "JPEG"),
("jpg_2000", "JPEG 2000"),
("zlib", "ZLIB (PNG/ZIP)"),
("libtiff", "LIBTIFF")
]:
if features.check_codec(name):
print("---", feature, "support ok")
else:
print("***", feature, "support not installed")
Expand Down

0 comments on commit 98fa49e

Please sign in to comment.