stripe.py

stripe.py

# coding: utf-8

import appex
import Image
import photos

def main():
   global appex
   global Image
   if not appex.is_running_extension():
     print ('Running in Pythonista app, using test image...')
     imglist = photos.pick_image(multi=True)
   else:
     imglist = appex.get_images()
   if imglist:
     import appex
     from PIL import Image, ImageDraw
     
     firstImage = imglist[0]
     (w, h) = firstImage.size
     
     images = [firstImage]
     
     for q in imglist[1:]:
         print("%d,%d %d,%d" % (w,h,q.size[0],q.size[1]))
         if q.size[0] != w or q.size[1] != h:
             raise "Must match in dimensions"
         images.append(q)
     slices = len(images)
     sliceWidth = (1.0*w) / slices
     leftEdge = 0
     rightEdge = -1
     chopWidth = -1
     outputImage = Image.new("RGB", (w, h))
     draw = ImageDraw.Draw(outputImage)
     borders = []
     for chop in images:
         rightEdge = leftEdge + int(sliceWidth + 0.5)
         if rightEdge > w:
             rightEdge = w
         chopWidth = rightEdge - leftEdge + 1
         chopped = chop.crop((leftEdge,0,rightEdge,h))
         outputImage.paste(chopped, (leftEdge,0))
         borders.append(rightEdge)
         leftEdge = rightEdge
     for border in borders[:-1]:
         draw.line((border-2,0, border-2,h), fill="white", width=4)
     outputImage.show()
   else:
     print('No input image found')

if __name__ == '__main__':
   main()