From 90d5539e0f2cc1521493a63097b7b6b65971cd1f Mon Sep 17 00:00:00 2001
From: Mohamed Ali ZORGATI
 <46978677+mohamedali-zorgati1997@users.noreply.github.com>
Date: Wed, 6 Apr 2022 20:08:44 +0100
Subject: [PATCH] ENH: Add alpha channel support for png files (#614)

---
 Scripts/pdf-image-extractor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/Scripts/pdf-image-extractor.py b/Scripts/pdf-image-extractor.py
index 601309c29..c1ac03238 100644
--- 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")