使用PIL加载RGBA位图

Question

我尝试使用PIL将32位位图转换为32位PNG.

from PIL import Image
im = Image.open('example.bmp')
print im.mode
# it prints 'RGB', but expected was 'RGBA'
im.save('output.png', format='PNG')

预期的图像模式是'RGBA',但实际上我得到'RGB'.

我也尝试了以下代码,但它不起作用.

from PIL import Image
im = Image.open('example.bmp')
im = im.convert('RGBA')
im.save('output.png', format='PNG')

Answer 1

好的，这里有一些可以开始的事情。因为我不知道你的 BMP 文件具体是哪种格式，所以我只处理了我碰巧拥有的带有完整 alpha 通道的 BMP 的特定情况。我在这里处理的 BMP 类型可以通过使用 ImageMagick 将带有 alpha 的 PNG 转换为 BMP 来获得。这将创建所谓的“BITMAPV5”。根据您的描述，您没有 BitmapV5（因为 PIL 甚至无法打开它），因此我们需要进行迭代讨论来解决您的具体情况。

因此，您要么需要一个新的文件解码器，要么需要一个修补的BmpImagePlugin.py. PIL 手册中描述了如何执行前者。对于后者，您显然需要发送补丁并希望将其纳入下一个 PIL 版本。我的重点是创建一个新的解码器：

from PIL import ImageFile, BmpImagePlugin

_i16, _i32 = BmpImagePlugin.i16, BmpImagePlugin.i32

class BmpAlphaImageFile(ImageFile.ImageFile):
    format = "BMP+Alpha"
    format_description = "BMP with full alpha channel"

    def _open(self):
        s = self.fp.read(14)
        if s[:2] != 'BM':
            raise SyntaxError("Not a BMP file")
        offset = _i32(s[10:])

        self._read_bitmap(offset)

    def _read_bitmap(self, offset):

        s = self.fp.read(4)
        s += ImageFile._safe_read(self.fp, _i32(s) - 4)

        if len(s) not in (40, 108, 124):
            # Only accept BMP v3, v4, and v5.
            raise IOError("Unsupported BMP header type (%d)" % len(s))

        bpp = _i16(s[14:])
        if bpp != 32:
            # Only accept BMP with alpha.
            raise IOError("Unsupported BMP pixel depth (%d)" % bpp)

        compression = _i32(s[16:])
        if compression == 3:
            # BI_BITFIELDS compression
            mask = (_i32(self.fp.read(4)), _i32(self.fp.read(4)),
                    _i32(self.fp.read(4)), _i32(self.fp.read(4)))
            # XXX Handle mask.
        elif compression != 0:
            # Only accept uncompressed BMP.
            raise IOError("Unsupported BMP compression (%d)" % compression)

        self.mode, rawmode = 'RGBA', 'BGRA'

        self.size = (_i32(s[4:]), _i32(s[8:]))
        direction = -1
        if s[11] == '\xff':
            # upside-down storage
            self.size = self.size[0], 2**32 - self.size[1]
            direction = 0

        self.info["compression"] = compression

        # data descriptor
        self.tile = [("raw", (0, 0) + self.size, offset,
            (rawmode, 0, direction))]

为了正确使用它，规范的方式应该执行：

from PIL import Image
Image.register_open(BmpAlphaImageFile.format, BmpAlphaImageFile)
# XXX register_save

Image.register_extension(BmpAlphaImageFile.format, ".bmp")

问题是已经有一个用于处理“.bmp”的插件，并且我没有费心去了解如何在前面添加这个新扩展名，以便在使用 BmpImagePlugin 之前使用它（我也不知道它是否是可以在 PIL 中做这样的事情）。说了这么多，我其实直接使用了代码，如下：

from BmpAlphaImagePlugin import BmpAlphaImageFile

x = BmpAlphaImageFile('gearscolor.bmp')
print x.mode
x.save('abc1.png')

其中 gearscolor.bmp 是具有完整 Alpha 通道的示例位图，如前所述。生成的 png 与 alpha 数据一起保存。如果你检查BmpImagePlugin.py的代码，你会发现我重用了它的大部分代码。