python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-10-31 16:07:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add support for few more formats

Browse Source
This commit is contained in:
REDxEYE 2022-08-14 01:35:00 +03:00
parent ed70519396
commit 3e7843745b
2 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/PIL/VtfImagePlugin.py
View File

@ -125,13 +125,11 @@ VTFHeader = NamedTuple(
("resource_count", int),
],
)
RGB_FORMATS = (
RGB_FORMATS = (VtfPF.DXT1,
VtfPF.RGB888,
VtfPF.BGR888,
VtfPF.UV88,
)
VtfPF.UV88,)
RGBA_FORMATS = (
VtfPF.DXT1,
VtfPF.DXT1_ONEBITALPHA,
VtfPF.DXT3,
VtfPF.DXT5,
@ -230,8 +228,13 @@ def _write_image(fp: BufferedIOBase, im: Image.Image, pixel_format: VtfPF):
im = im.convert("LA")
elif pixel_format == VtfPF.UV88:
encoder = "raw"
if im.mode == "RGB" or im.mode == "RGBA":
r, g, *_ = im.split()
im = Image.merge("LA", (r, g))
im = Image.merge('LA', (r, g))
elif im.mode == "LA":
pass
else:
raise VTFException(f'Cannot encode {im.mode} as {pixel_format}')
encoder_args = ("LA", 0, 0)
else:
raise VTFException(f"Unsupported pixel format: {pixel_format!r}")
@ -283,8 +286,9 @@ class VtfImageFile(ImageFile.ImageFile):
# flags = CompiledVtfFlags(header.flags)
pixel_format = VtfPF(header.pixel_format)
low_format = VtfPF(header.low_pixel_format)
if pixel_format in RGB_FORMATS:
if pixel_format == VtfPF.DXT1: # Special case for DXT1
self.mode = "RGBA"
elif pixel_format in RGB_FORMATS:
self.mode = "RGB"
elif pixel_format in RGBA_FORMATS:
self.mode = "RGBA"
@ -316,6 +320,12 @@ class VtfImageFile(ImageFile.ImageFile):
tile = ("raw", (0, 0) + self.size, data_start, ("RGBA", 0, 1))
elif pixel_format in (VtfPF.RGB888,):
tile = ("raw", (0, 0) + self.size, data_start, ("RGB", 0, 1))
elif pixel_format in (VtfPF.BGR888,):
tile = ("raw", (0, 0) + self.size, data_start, ("BGR", 0, 1))
elif pixel_format in (VtfPF.BGRA8888,):
tile = ("raw", (0, 0) + self.size, data_start, ("BGRA", 0, 1))
elif pixel_format in (VtfPF.UV88,):
tile = ("raw", (0, 0) + self.size, data_start, ("LA", 0, 1))
elif pixel_format in L_FORMATS:
tile = ("raw", (0, 0) + self.size, data_start, ("L", 0, 1))
elif pixel_format in LA_FORMATS:

1
src/encode.c
View File

@ -384,7 +384,6 @@ PyImaging_BcnEncoderNew(PyObject *self, PyObject *args) {
ImagingEncoderObject *encoder;
char *mode;
char *actual;
int n = 0;
char *pixel_format = "";
if (!PyArg_ParseTuple(args, "si|s", &mode, &n, &pixel_format)) {