scripts/imgconv.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   5 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   6 #
   7 import sys
   8 import os
   9 import shutil
  10 from os.path import splitext, dirname, basename, realpath
  11 from PIL import Image, ImageFilter, ImageEnhance, ImageOps
  12
  13
  14 def resize(image, max_width, max_height):
  15     """Resize image so it's not wider than max_width and not higher than max_height."""
  16     width, height = image.size
  17     ratio = max(1.0, float(width) / max_width, float(height) / max_height)
  18     new_width, new_height = int(width / ratio), int(height / ratio)
  19     return image.resize((new_width, new_height), Image.ANTIALIAS)
  20
  21
  22 def crop(image, ratio, from_right=False):
  23     """Crop image to ratio of current width."""
  24     width, height = image.size
  25     new_width = width * ratio
  26     if from_right:
  27         bounds = (int(width - new_width), 0, int(width), int(height))
  28     else:
  29         bounds = (0, 0, int(new_width), int(height))
  30     image = image.crop(bounds)
  31     image.load()
  32     return image
  33
  34
  35 def ratio(image):
  36     """Return width to height ratio of image."""
  37     width, height = image.size
  38     return float(width) / height
  39
  40
  41 def try_creating(directory):
  42     try:
  43         os.mkdir(directory)
  44     except:
  45         pass
  46
  47
  48 output_dir = realpath(os.getcwd()) + '/output'
  49 # big_output_dir = output_dir + '/big'
  50 tmp_output_dir = output_dir + '/tmp'
  51
  52 try_creating(output_dir)
  53 # try_creating(big_output_dir)
  54 try_creating(tmp_output_dir)
  55
  56
  57 for file_name in sys.argv[1:]:
  58     base_name, ext = splitext(file_name)
  59     try:
  60         image = Image.open(file_name)
  61     except IOError, e:
  62         sys.stderr.write('\nerror:%s:%s\n' % (file_name, e.message))
  63         continue
  64
  65     # Check ratio
  66     if ratio(image) > 1:
  67         images = [crop(image, 0.6), crop(image, 0.6, from_right=True)]
  68     else:
  69         images = [image]
  70
  71     for i, image in enumerate(images):
  72         image_name = '%s.%d.png' % (basename(base_name), i)
  73
  74         # Save files
  75         small_image = resize(image, 640, 960)
  76         small_image = small_image.convert('L')
  77         small_image = ImageOps.autocontrast(small_image, cutoff=85)
  78         # small_image = small_image.filter(ImageFilter.SHARPEN)
  79         small_image.save(tmp_output_dir + '/' + image_name)
  80
  81         os.system('pngnq -n 128 -s 1 -e .png -d "%s" -f "%s"' % (
  82             output_dir,
  83             tmp_output_dir + '/' + image_name,
  84         ))
  85         os.remove(tmp_output_dir + '/' + image_name)
  86
  87         # big_image = resize(image, 960, 1440)
  88         # big_image = big_image.convert('L')
  89         # big_image = big_image.filter(ImageFilter.SHARPEN)
  90         # big_image.save(tmp_output_dir + '/' + image_name, optimize=True)
  91         # os.system('pngnq -n 16 -s 1 -e .png -d "%s" -f "%s"' % (
  92         #     big_output_dir,
  93         #     tmp_output_dir + '/' + image_name,
  94         # ))
  95         # os.remove(tmp_output_dir + '/' + image_name)
  96
  97     sys.stderr.write('.')
  98
  99 shutil.rmtree(tmp_output_dir)