From 1768db5a32efb96ded029c153fab1fab240362c2 Mon Sep 17 00:00:00 2001
From: shekhar-cis <shekhar.c@cisinlabs.com>
Date: Thu, 25 Aug 2022 12:48:26 +0530
Subject: [PATCH] Add identicon generator to generate default profile picture

---
 src/bitmessagekivy/identiconGeneration.py | 79 +++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 src/bitmessagekivy/identiconGeneration.py

diff --git a/src/bitmessagekivy/identiconGeneration.py b/src/bitmessagekivy/identiconGeneration.py
new file mode 100644
index 00000000..2d6578a7
--- /dev/null
+++ b/src/bitmessagekivy/identiconGeneration.py
@@ -0,0 +1,79 @@
+"""
+Core classes for loading images and converting them to a Texture.
+The raw image data can be keep in memory for further access
+"""
+import hashlib
+from io import BytesIO
+
+from PIL import Image
+from kivy.core.image import Image as CoreImage
+from kivy.uix.image import Image as kiImage
+
+
+# constants
+RESOLUTION = 300, 300
+V_RESOLUTION = 7, 7
+BACKGROUND_COLOR = 255, 255, 255, 255
+MODE = "RGB"
+
+
+def generate(Generate_string=None):
+    """Generating string"""
+    hash_string = generate_hash(Generate_string)
+    color = random_color(hash_string)
+    image = Image.new(MODE, V_RESOLUTION, BACKGROUND_COLOR)
+    image = generate_image(image, color, hash_string)
+    image = image.resize(RESOLUTION, 0)
+    data = BytesIO()
+    image.save(data, format='png')
+    data.seek(0)
+    # yes you actually need this
+    im = CoreImage(BytesIO(data.read()), ext='png')
+    beeld = kiImage()
+    # only use this line in first code instance
+    beeld.texture = im.texture
+    return beeld
+
+
+def generate_hash(string):
+    """Generating hash"""
+    try:
+        # make input case insensitive
+        string = str.lower(string)
+        hash_object = hashlib.md5(str.encode(string))
+        print(hash_object.hexdigest())
+        # returned object is a hex string
+        return hash_object.hexdigest()
+    except IndexError:
+        print("Error: Please enter a string as an argument.")
+
+
+def random_color(hash_string):
+    """Getting random color"""
+    # remove first three digits from hex string
+    split = 6
+    rgb = hash_string[:split]
+    split = 2
+    r = rgb[:split]
+    g = rgb[split:2 * split]
+    b = rgb[2 * split:3 * split]
+    color = (int(r, 16), int(g, 16), int(b, 16), 0xFF)
+    return color
+
+
+def generate_image(image, color, hash_string):
+    """Generating images"""
+    hash_string = hash_string[6:]
+    lower_x = 1
+    lower_y = 1
+    upper_x = int(V_RESOLUTION[0] / 2) + 1
+    upper_y = V_RESOLUTION[1] - 1
+    limit_x = V_RESOLUTION[0] - 1
+    index = 0
+    for x in range(lower_x, upper_x):
+        for y in range(lower_y, upper_y):
+            if int(hash_string[index], 16) % 2 == 0:
+                image.putpixel((x, y), color)
+                image.putpixel((limit_x - x, y), color)
+            index = index + 1
+    return image