python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-12-26 18:06:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5844 from radarhere/numpy_array

Browse Source
This commit is contained in:
Hugo van Kemenade 2021-11-21 13:06:15 +02:00 committed by GitHub
commit 059503b9c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/PIL/Image.py
View File

@ -656,6 +656,10 @@ class Image:
raise ValueError("Could not save to PNG for display") from e raise ValueError("Could not save to PNG for display") from e
return b.getvalue() return b.getvalue()
class _ArrayData:
def __init__(self, new):
self.__array_interface__ = new
def __array__(self, dtype=None): def __array__(self, dtype=None):
# numpy array interface support # numpy array interface support
import numpy as np import numpy as np
@ -672,10 +676,7 @@ class Image:
else: else:
new["data"] = self.tobytes() new["data"] = self.tobytes()
class ArrayData: return np.array(self._ArrayData(new), dtype)
__array_interface__ = new
return np.array(ArrayData(), dtype)
def __getstate__(self): def __getstate__(self):
return [self.info, self.mode, self.size, self.getpalette(), self.tobytes()] return [self.info, self.mode, self.size, self.getpalette(), self.tobytes()]