import os import sys import random import re import urllib from Config import * from Pb import Pb import simplejson as json from PIL import Image import uuid _fuse_mode="Pin_Light" class PbPattern(Pb): example_params = { # "pattern_url" : "http://asdf.us/impattern/patterns/1.png", "pattern_data" : '{"matrix":[["0","0","0","0","0","1","0","0","0","0"],["0","0","0","0","1","1","1","0","0","0"],["0","0","1","1","1","0","1","0","0","0"],["0","1","1","0","0","0","0","0","0","0"],["0","1","0","0","1","0","0","0","0","0"],["0","1","0","0","1","0","0","0","1","0"],["0","1","0","0","1","1","0","0","1","0"],["0","1","0","0","0","1","1","1","1","0"],["0","1","1","1","1","0","0","0","0","0"],["0","0","0","0","1","0","0","0","0","0"]],"width":"10","height":"10"}', # "username" : "garfield", "image_url" : "http://i.asdf.us/im/be/PinkHijab_1425078647_reye.gif", } def __init__(self, pattern_url=None, pattern_data=None, username=None, image_url=None, ): super(PbPattern,self).__init__(); self.params.set_val("image_url", url, value_type="img_url"); self.params.set_val("pattern_url", url, value_type="img_url"); self.params.set_val("pattern_data", url, value_type="json"); self.params.set_val("username", username, value_type="string"); self.basename, self._format = self._get_filename(); #FIXME omit file extension for downloaded files #lets go back to this in a second self._pattern_file = os.path.join(WORKING_DIR, "IMPATTERNTMP_PTN{}_{}.{}".format(self.basename, self._pid, self._format)) # this self._download(self.params.image_url, self._downloaded_file) self.filename = "{}.{}".format(self.basename, self._format) self.filepath = os.path.join(WORKING_DIR, self.filename) if self.params['pattern_url']: self._download(self.params['pattern_url'], self._pattern_file) elif self.params['pattern_data']: self._from_pattern_data() else: sys.stderr.write("pattern must be supplied as json array or as a png url") raise ValueError; def _from_pattern_data(self): def boolToColor(boolean): if boolean: return (0,0,0,255); else: return (255,255,255,255) specs = json.loads(self.params.pattern_data); if int(specs['width']) > 100 or int(specs['height']) > 100: raise ValueError sys.stderr.write("height and width need to be less than 100 px") img = Image.new('RGBA', (int(specs['width']), int(specs['height']))); pixels = img.load(); for i in range(0, len(specs['matrix'])): for j in range(0, len(specs['matrix'][i])): pixels[j,i] = boolToColor(int(specs['matrix'][i][j])); img.save(self._pattern_file, "PNG") #first step def _make_canvas(self): _width, _height = self.dimensions(self._downloaded_file) # same here cmd = [BIN_CONVERT, "-size", _width + "x" + _height, "canvas:transparent", self.filepath] self._call_cmd(cmd) #second step use the Canvas as a background def _make_mask(self): #tile the pattern pattern on the canvas cmd = [BIN_COMPOSITE,"-tile", self._pattern_file, self.filepath, self.filepath]; self._call_cmd(cmd) #fuse the tiled file to create a mask #convert thebg.gif -compose Dst_In null: thefile.gif -matte -layers composite new.gif cmd = [BIN_CONVERT, self.filepath, "-compose", "Dst_In", "null:", self._downloaded_file, "-matte", "-layers", "composite", self.filepath] self._call_cmd(cmd) #third step def _fuse_mask(self, fuse_mode=_fuse_mode): cmd = [BIN_CONVERT, "-dispose", "2", self.filepath, "null:", self._downloaded_file, "-matte", "-compose", fuse_mode, "-layers", "composite", self.filepath] self._call_cmd(cmd) def create(self): self._make_canvas(); self._make_mask() self._fuse_mask();