add img utils

vertexi · Feb 17, 2022 · 6f22b02 · 6f22b02
1 parent 6babcd3
commit 6f22b02
Show file tree

Hide file tree

Showing 19 changed files with 79 additions and 0 deletions.
diff --git a/img/gif2jpg.py b/img/gif2jpg.py
diff --git a/img/pikachu-love0.jpg b/img/pikachu-love0.jpg
diff --git a/img/pikachu-love1.jpg b/img/pikachu-love1.jpg
diff --git a/img/pikachu-love10.jpg b/img/pikachu-love10.jpg
diff --git a/img/pikachu-love11.jpg b/img/pikachu-love11.jpg
diff --git a/img/pikachu-love12.jpg b/img/pikachu-love12.jpg
diff --git a/img/pikachu-love13.jpg b/img/pikachu-love13.jpg
diff --git a/img/pikachu-love14.jpg b/img/pikachu-love14.jpg
diff --git a/img/pikachu-love15.jpg b/img/pikachu-love15.jpg
diff --git a/img/pikachu-love2.jpg b/img/pikachu-love2.jpg
diff --git a/img/pikachu-love3.jpg b/img/pikachu-love3.jpg
diff --git a/img/pikachu-love4.jpg b/img/pikachu-love4.jpg
diff --git a/img/pikachu-love5.jpg b/img/pikachu-love5.jpg
diff --git a/img/pikachu-love6.jpg b/img/pikachu-love6.jpg
diff --git a/img/pikachu-love7.jpg b/img/pikachu-love7.jpg
diff --git a/img/pikachu-love8.jpg b/img/pikachu-love8.jpg
diff --git a/img/pikachu-love9.jpg b/img/pikachu-love9.jpg
diff --git a/utils/gif2jpg.py b/utils/gif2jpg.py
@@ -0,0 +1,79 @@
+from PIL import Image
+import pathlib
+import os
+
+
+def gif2jpg(file_name: str, num_key_frames: int, trans_color: tuple,
+            resize: tuple=None, output_dir: str=None):
+    """
+    convert gif to `num_key_frames` images with jpg format
+    :param file_name: gif file name
+    :param num_key_frames: result images number
+    :param trans_color: set converted transparent color in jpg image
+    :param resize: resize image to maximum (width, height) dimension
+    :param output_dir: set output images dir
+    :return:
+    """
+    path = pathlib.PurePath(file_name)
+    file_pure_name = path.name
+    dot_pos = file_pure_name.rfind('.')
+    directory_name = path.parents[0]
+    if dot_pos != -1:
+        file_pure_name = file_pure_name[:dot_pos]
+
+    if output_dir:
+        if not os.path.isdir(output_dir):
+            os.makedirs(output_dir)
+
+    with Image.open(file_name) as im:
+        for i in range(num_key_frames):
+            im.seek(im.n_frames // num_key_frames * i)
+            image = im.convert("RGBA")
+            datas = image.getdata()
+            new_data = []
+            for item in datas:
+                if item[3] == 0:  # if transparent
+                    new_data.append(trans_color)  # set transparent color in jpg
+                else:
+                    new_data.append(tuple(item[:3]))
+            image = Image.new("RGB", im.size)
+            image.getdata()
+            image.putdata(new_data)
+            if resize:
+                image.thumbnail(resize, Image.ANTIALIAS)
+            if output_dir:
+                image.save('{}/{}{}.jpg'.format(output_dir, file_pure_name, i))
+            else:
+                image.save('{}/{}{}.jpg'.format(directory_name, file_pure_name, i))
+
+
+def png2jpg(file_name: str, trans_color: tuple):
+    """
+    convert png file to jpg file
+    :param file_name: png file name
+    :param trans_color: set transparent color in jpg image
+    :return:
+    """
+    with Image.open(file_name) as im:
+        image = im.convert("RGBA")
+        datas = image.getdata()
+        new_data = []
+        for item in datas:
+            if item[3] == 0:  # if transparent
+                new_data.append(trans_color)  # set transparent color in jpg
+            else:
+                new_data.append(tuple(item[:3]))
+        image = Image.new("RGB", im.size)
+        image.getdata()
+        image.putdata(new_data)
+        image.save('{}.jpg'.format(file_name))
+
+
+def create_thumbnail(infile: str, size: tuple):
+    im = Image.open(infile)
+    im.thumbnail(size, Image.ANTIALIAS)
+    im.save(infile)
+
+
+# convert image.gif to 8 jpg images with white background
+gif2jpg("utils/pikachu-love.gif", 16, (0, 0, 0), (100, 100), "img")
diff --git a/img/pikachu-love.gif → utils/pikachu-love.gif b/img/pikachu-love.gif → utils/pikachu-love.gif