ENH: Add alpha channel support for png files (#614)

py-pdf · Apr 6, 2022 · 90d5539 · 90d5539
1 parent 7528347
commit 90d5539
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/Scripts/pdf-image-extractor.py b/Scripts/pdf-image-extractor.py
@@ -34,6 +34,9 @@
                 if '/Filter' in xObject[obj]:
                     if xObject[obj]['/Filter'] == '/FlateDecode':
                         img = Image.frombytes(mode, size, data)
+                        if "/SMask" in xObject[obj]: # add alpha channel
+                            alpha = Image.frombytes("L", size, xObject[obj]["/SMask"].getData())
+                            img.putalpha(alpha)
                         img.save(obj[1:] + ".png")
                     elif xObject[obj]['/Filter'] == '/DCTDecode':
                         img = open(obj[1:] + ".jpg", "wb")