mirror/OCRmyPDF
1
0
Fork 0
mirror of https://github.com/ocrmypdf/OCRmyPDF.git synced 2026-05-05 13:16:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

leptonica: fix Pillow conversion for 1-bit and 8-bit gray images

Browse Source
This commit is contained in:
James R. Barlow
2016-10-26 13:10:13 -07:00
parent 7eca8508fd
commit 203966d86b
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
ocrmypdf/leptonica.py
View File

@@ -206,26 +206,27 @@ class Pix:
# Leptonica manages data in words, so it implicitly does an endian
# swap. Tell Pillow about this when it reads the data.
pix = self
if sys.byteorder == 'little':
if self.mode == 'RGB':
raw_mode = 'XBGR'
elif self.mode == 'RGBA':
raw_mode = 'ABGR'
elif self.mode == '1':
raw_mode = '1;I'
pix = Pix(lept.pixEndianByteSwapNew(pix._pix))
else:
raw_mode = self.mode
pix = Pix(lept.pixEndianByteSwapNew(pix._pix))
else:
raw_mode = self.mode # no endian swap needed
size = (self._pix.w, self._pix.h)
bytecount = self._pix.wpl * 4 * self._pix.h
buf = ffi.buffer(self._pix.data, bytecount)
size = (pix._pix.w, pix._pix.h)
bytecount = pix._pix.wpl * 4 * pix._pix.h
buf = ffi.buffer(pix._pix.data, bytecount)
stride = pix._pix.wpl * 4
im_raw = Image.frombytes(self.mode, size, buf, 'raw', raw_mode)
# Leptonica stores images in 32-bit words
# Need to crop the any trailing amount
box = (0, 0, self.width, self.height)
im = im_raw.crop(box)
im = Image.frombytes(self.mode, size, buf, 'raw', raw_mode, stride)
return im