在别人的代码基础上做了一点修改,修正裁剪不准确的问题。
脚本用法请参考https://blog.youkuaiyun.com/linuxchen/article/details/16865645
主要是result_box不对。
result_box=(
( sizelist[0] - width )/2,
( sizelist[1] - height )/2,
( sizelist[0] + width )/2,
( sizelist[1] + height )/2
)
完整代码如下
#!python
import os,sys
import logging
from xml.etree import ElementTree
from PIL import Image
def tree_to_dict(tree):
d = {}
for index, item in enumerate(tree):
if item.tag == 'key':
if tree[index+1].tag == 'string':
d[item.text] = tree[index + 1].text
elif tree[index + 1].tag == 'true':
d[item.text] = True
elif tree[index + 1].tag == 'false':
d[item.text] = False
elif tree[index+1].tag == 'dict':
d[item.text] = tree_to_dict(tree[index+1])
return d
def gen_png_from_plist(plist_filename, png_filename):
file_path = plist_filename.replace('.plist', '')
big_image = Image.open(png_filename)
root = ElementTree.fromstring(open(plist_filename, 'r').read())
plist_dict = tree_to_dict(root[0])
to_list = lambda x: x.replace('{','').replace('}','').split(',')
for k,v in plist_dict['frames'].items():
rectlist = to_list(v['frame'])
width = int( rectlist[3] if v['rotated'] else rectlist[2] )
height = int( rectlist[2] if v['rotated'] else rectlist[3] )
box=(
int(rectlist[0]),
int(rectlist[1]),
int(rectlist[0]) + width,
int(rectlist[1]) + height,
)
sizelist = [ int(x) for x in to_list(v['sourceSize'])]
rect_on_big = big_image.crop(box)
if v['rotated']:
rect_on_big = rect_on_big.rotate(90)
print "witdh: ", width
print "offset: ", v['offset']
print "x: ", to_list(v['offset'])[0]
s = (sizelist[0] - width)/2
d = sizelist[0]
offset = [ int(x) for x in to_list(v['offset'])]
print ( sizelist[0] - width )/2 + offset[0],
print ( sizelist[1] - height )/2 + offset[1],
print ( sizelist[0] + width )/2 + offset[0],
print ( sizelist[1] + height )/2 + offset[1]
result_image = Image.new('RGBA', sizelist, (0,0,0,0))
if v['rotated']:
result_box=(
( sizelist[0] - height )/2 + offset[0],
( sizelist[1] - width )/2 - offset[1],
( sizelist[0] + height )/2 + offset[0],
( sizelist[1] + width )/2 - offset[1]
)
else:
result_box=(
( sizelist[0] - width )/2 + offset[0],
( sizelist[1] - height )/2 - offset[1],
( sizelist[0] + width )/2 + offset[0],
( sizelist[1] + height )/2 - offset[1]
)
result_image.paste(rect_on_big, result_box, mask=0)
if not os.path.isdir(file_path):
os.mkdir(file_path)
outfile = (file_path+'/' + k).replace('gift_', '')
print outfile, "generated"
result_image.save(outfile)
if __name__ == '__main__':
filename = sys.argv[1]
plist_filename = filename + '.plist'
png_filename = filename + '.png'
if (os.path.exists(plist_filename) and os.path.exists(png_filename)):
gen_png_from_plist( plist_filename, png_filename )
else:
print "make sure you have boith plist and png files in the same directory"