这听起来像是应该在任何处理精灵的东西中实现的东西,但在这里我们将实现我们自己的精灵分割器。
我们在这里需要做的第一件事是提取单个对象。在这种情况下,只需要确定一个像素是否是背景像素。如果我们假设原点是背景像素,那么我们就完成了:
from PIL import Image
def sprite_mask(img, bg_point=(0, 0)):
width, height = img.size
im = img.load()
bg = im[bg_point]
mask_img = Image.new('L', img.size)
mask = mask_img.load()
for x in xrange(width):
for y in xrange(height):
if im[x, y] != bg:
mask[x, y] = 255
return mask_img, bg
如果您保存mask
上面创建的图像并打开它,您会看到以下内容(我在您的空窗口中添加了一个矩形):
有了上图,如果我们想加入其他内部的精灵(如添加的矩形,见上图),接下来我们需要填充它的洞。这是另一个简单的规则:如果从 [0, 0] 处的点无法到达某个点,那么它就是一个洞,必须将其填充。剩下的就是在单独的图像中分离每个精灵。这是通过连接组件标记来完成的。对于每个组件,我们获取其轴对齐的边界框以定义块的尺寸,然后我们从原始图像中复制属于给定组件的点。为了简短起见,以下代码scipy
用于这些任务:
import sys
import numpy
from scipy.ndimage import label, morphology
def split_sprite(img, mask, bg, join_interior=True, basename='sprite_%d.png'):
im = img.load()
m = numpy.array(mask, dtype=numpy.uint8)
if join_interior:
m = morphology.binary_fill_holes(m)
lbl, ncc = label(m, numpy.ones((3, 3)))
for i in xrange(1, ncc + 1):
px, py = numpy.nonzero(lbl == i)
xmin, xmax, ymin, ymax = px.min(), px.max(), py.min(), py.max()
sprite = Image.new(img.mode, (ymax - ymin + 1, xmax - xmin + 1), bg)
sp = sprite.load()
for x, y in zip(px, py):
x, y = int(x), int(y)
sp[y - int(ymin), x - int(xmin)] = im[y, x]
name = basename % i
sprite.save(name)
print "Wrote %s" % name
sprite = Image.open(sys.argv[1])
mask, bg = sprite_mask(sprite)
split_sprite(sprite, mask, bg)
现在您拥有与问题中包含的完全相同的所有部分(sprite_1.png、sprite_2.png、...、sprite_8.png)。