Skip to content

Commit

Permalink
Moved _Tile to ImageFile
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed Dec 1, 2023
1 parent e1ef612 commit bb61b10
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 18 deletions.
19 changes: 12 additions & 7 deletions src/PIL/DdsImagePlugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
Full text of the CC0 license:
https://creativecommons.org/publicdomain/zero/1.0/
"""

import io
import struct
import sys
Expand Down Expand Up @@ -460,9 +461,11 @@ def _open(self):

extents = (0, 0) + self.size
if n:
self.tile = [Image._Tile("bcn", extents, offset, (n, self.pixel_format))]
self.tile = [
ImageFile._Tile("bcn", extents, offset, (n, self.pixel_format))
]
else:
self.tile = [Image._Tile("raw", extents, 0, rawmode or self.mode)]
self.tile = [ImageFile._Tile("raw", extents, 0, rawmode or self.mode)]

def load_seek(self, pos):
pass
Expand Down Expand Up @@ -494,8 +497,8 @@ def _save(im, fp, filename):
rgba_mask.append(0xFF000000 if alpha else 0)

flags = DDSD.CAPS | DDSD.HEIGHT | DDSD.WIDTH | DDSD.PITCH | DDSD.PIXELFORMAT
bit_count = len(im.getbands()) * 8
stride = (im.width * bit_count + 7) // 8
bitcount = len(im.getbands()) * 8
pitch = (im.width * bitcount + 7) // 8

fp.write(
o32(DDS_MAGIC)
Expand All @@ -505,17 +508,19 @@ def _save(im, fp, filename):
flags, # flags
im.height,
im.width,
stride, # pitch
pitch,
0, # depth
0, # mipmaps
)
+ struct.pack("11I", *((0,) * 11)) # reserved
# pfsize, pfflags, fourcc, bitcount
+ struct.pack("<4I", 32, pixel_flags, 0, bit_count)
+ struct.pack("<4I", 32, pixel_flags, 0, bitcount)
+ struct.pack("<4I", *rgba_mask) # dwRGBABitMask
+ struct.pack("<5I", DDSCAPS.TEXTURE, 0, 0, 0, 0)
)
ImageFile._save(im, fp, [Image._Tile("raw", (0, 0) + im.size, 0, (rawmode, 0, 1))])
ImageFile._save(
im, fp, [ImageFile._Tile("raw", (0, 0) + im.size, 0, (rawmode, 0, 1))]
)


def _accept(prefix):
Expand Down
10 changes: 0 additions & 10 deletions src/PIL/Image.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@
#
# See the README file for information on usage and redistribution.
#
from __future__ import annotations

import atexit
import builtins
Expand All @@ -39,7 +38,6 @@
from collections.abc import Callable, MutableMapping
from enum import IntEnum
from pathlib import Path
from typing import NamedTuple

try:
import defusedxml.ElementTree as ElementTree
Expand Down Expand Up @@ -208,13 +206,6 @@ class Quantize(IntEnum):
FIXED = core.FIXED


class _Tile(NamedTuple):
encoder_name: str
extents: tuple[int, int, int, int]
offset: int
args: tuple | str | None


# --------------------------------------------------------------------
# Registries

Expand Down Expand Up @@ -705,7 +696,6 @@ def __getstate__(self):

def __setstate__(self, state):
Image.__init__(self)
self.tile: list[_Tile] = []
info, mode, size, palette, data = state
self.info = info
self._mode = mode
Expand Down
10 changes: 9 additions & 1 deletion src/PIL/ImageFile.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@
import itertools
import struct
import sys
from typing import NamedTuple

from . import Image
from ._util import is_path
Expand Down Expand Up @@ -78,6 +79,13 @@ def _tilesort(t):
return t[2]


class _Tile(NamedTuple):
encoder_name: str
extents: tuple[int, int, int, int]
offset: int
args: tuple | str | None


#
# --------------------------------------------------------------------
# ImageFile base class
Expand Down Expand Up @@ -522,7 +530,7 @@ def _save(im, fp, tile, bufsize=0):
fp.flush()


def _encode_tile(im, fp, tile: list[Image._Tile], bufsize, fh, exc=None):
def _encode_tile(im, fp, tile: list[_Tile], bufsize, fh, exc=None):
for encoder_name, extents, offset, args in tile:
if offset > 0:
fp.seek(offset)
Expand Down

0 comments on commit bb61b10

Please sign in to comment.