将PIL图像转换为Cairo ImageSurface

Question

我正在尝试从PIL图像创建一个cairo ImageSurface,到目前为止我的代码是:

im = Image.open(filename)
imstr = im.tostring()
a = array.array('B', imstr)
height, width = im.size
stride = cairo.ImageSurface.format_stride_for_width(cairo.FORMAT_RGB24, width)
return cairo.ImageSurface.create_for_data(a, cairo.FORMAT_ARGB24, width, height, stride)

但是这给了我

TypeError: buffer is not long enough.

我真的不明白为什么会这样,也许我对图像格式不太了解.

我正在使用cairo 1.10.

Answer 1

开罗的create_for_data()需要一个可写缓冲区对象(一个字符串可以用作缓冲区对象,但它不可写),并且它只支持每像素32位数据(RGBA或RGB后跟一个未使用的字节).另一方面,PIL提供24bpp RGB只读缓冲区对象.

我建议你告诉PIL添加一个alpha通道,然后将PIL缓冲区转换为numpy数组,以获得Cairo的可写缓冲区.

im = Image.open(filename)
im.putalpha(256) # create alpha channel
arr = numpy.array(im)
height, width, channels = arr.shape
surface = cairo.ImageSurface.create_for_data(arr, cairo.FORMAT_RGB24, width, height)

Answer 2

接受的版本在以下情况下无法正常工作：

• 您的图片有颜色
• 您的图像不透明
• 您的图像处于与RGB（A）不同的模式

在开罗图像中，颜色的值乘以alpha的值，然后使用本机CPU字节序将它们存储为32位字。这意味着PIL映像：

r1 g1 b1 a1 r2 g2 b2 a2 ...

存储在开罗小端CPU中的方式为：

b1*a1 g1*a1 r1*a1 a1 b2*a2 g2*a2 r2*a2 a2 ...

在大字节序CPU中为：

a1 r1*a1 b1*a1 g1*a1 a2 r2*a2 g2*a2 b2*a2 ...

这是一个在没有NumPy依赖项的小端机器上正常运行的版本：

def pil2cairo(im):
    """Transform a PIL Image into a Cairo ImageSurface."""

    assert sys.byteorder == 'little', 'We don\'t support big endian'
    if im.mode != 'RGBA':
        im = im.convert('RGBA')

    s = im.tostring('raw', 'BGRA')
    a = array.array('B', s)
    dest = cairo.ImageSurface(cairo.FORMAT_ARGB32, im.size[0], im.size[1])
    ctx = cairo.Context(dest)
    non_premult_src_wo_alpha = cairo.ImageSurface.create_for_data(
        a, cairo.FORMAT_RGB24, im.size[0], im.size[1])
    non_premult_src_alpha = cairo.ImageSurface.create_for_data(
        a, cairo.FORMAT_ARGB32, im.size[0], im.size[1])
    ctx.set_source_surface(non_premult_src_wo_alpha)
    ctx.mask_surface(non_premult_src_alpha)
    return dest

在这里，我用开罗做预乘法。我也尝试使用NumPy进行预乘法，但是结果较慢。在我的计算机（Mac OS X，2.13GHz Intel Core 2 Duo）中，此功能需要1秒钟才能转换出6000x6000像素的图像，而5ms可以转换到500x500像素的图像。