homa 0.22__py3-none-any.whl → 0.23__py3-none-any.whl

homa/__init__.py

@@ -2,3 +2,8 @@ from .main import *
 from .filters import *
 from .camera import *
 from .orientation import *
+from .spaces import *
+from .events import *
+from .shapes import *
+
+from .helpers import *

homa/camera.py

@@ -1,21 +1,19 @@
 import cv2
 from .classes.Repository import Repository
+from typing import Iterator
 
 
-def camera(delay=10):
-    capture = cv2.VideoCapture(0)
-
+def camera(delay: int = 10, exit_on: str = "q") -> Iterator[int]:
     pressed_key = None
-
+    capture = cv2.VideoCapture(0)
     frame_number = 0
-
-    while pressed_key != ord("q"):
-        frame_number += 1
+    while pressed_key != ord(exit_on):
         _, frame = capture.read()
-
         Repository.images["camera"] = frame
 
+        frame_number += 1
         yield frame_number
+
         pressed_key = cv2.waitKey(delay)
 
     capture.release()

homa/classes/Repository.py

@@ -1,4 +1,4 @@
-from ..helpers import is_colab
+from ..helpers.environment import is_colab
 
 
 class RepositoryWrapper:
@@ -8,8 +8,7 @@ class RepositoryWrapper:
 
         self.directory = "./"
         self.images = {}
-        self.cameras = {}
-        self.window_counter = 0
+        self.windows = {}
 
         if is_colab():
             from google.colab.patches import cv2_imshow as imshow
@@ -24,9 +23,5 @@ class RepositoryWrapper:
 
         self.imshow = final_imshow
 
-    def get_counter(self):
-        self.window_counter += 1
-        return self.window_counter
-
 
 Repository = RepositoryWrapper()

homa/events.py

@@ -0,0 +1,70 @@
+# enum MouseEventTypes
+# {
+#     EVENT_MOUSEMOVE     = 0,
+#     EVENT_LBUTTONDOWN   = 1,
+#     EVENT_RBUTTONDOWN   = 2,
+#     EVENT_MBUTTONDOWN   = 3,
+#     EVENT_LBUTTONUP     = 4,
+#     EVENT_RBUTTONUP     = 5,
+#     EVENT_MBUTTONUP     = 6,
+#     EVENT_LBUTTONDBLCLK = 7,
+#     EVENT_RBUTTONDBLCLK = 8,
+#     EVENT_MBUTTONDBLCLK = 9,
+#     EVENT_MOUSEWHEEL    = 10,
+#     EVENT_MOUSEHWHEEL   = 11,
+# };
+import cv2
+from .main import win
+from inspect import signature
+
+event_map = {
+    "click":     cv2.EVENT_LBUTTONDOWN,
+    "rclick":    cv2.EVENT_RBUTTONDOWN,
+    "mclick":    cv2.EVENT_MBUTTONDOWN,
+    "mouseup":   cv2.EVENT_LBUTTONUP,
+    "rmouseup":  cv2.EVENT_RBUTTONUP,
+    "dblclick":  cv2.EVENT_LBUTTONDBLCLK,
+    "rdblclick": cv2.EVENT_RBUTTONDBLCLK,
+    "move":      cv2.EVENT_MOUSEMOVE
+}
+
+
+def create_wrapper_function(event_name: str, handler: callable):
+    def wrapper_function(event, x, y, flags, param):
+        if event != event_map[event_name]:
+            return
+
+        argument_count = len(signature(handler).parameters)
+        if argument_count == 2:
+            handler(x, y)
+        elif argument_count == 3:
+            handler(x, y, flags)
+        elif argument_count == 4:
+            handler(x, y, flags, param)
+
+    return wrapper_function
+
+
+def onClick(key: str, handler: callable):
+    win(key)
+    cv2.setMouseCallback(key, create_wrapper_function("click", handler))
+
+
+def onRightClick(key: str, handler: callable):
+    win(key)
+    cv2.setMouseCallback(key, create_wrapper_function("rclick", handler))
+
+
+def onMouseUp(key: str, handler: callable):
+    win(key)
+    cv2.setMouseCallback(key, create_wrapper_function("mouseup", handler))
+
+
+def onMove(key: str, handler: callable):
+    win(key)
+    cv2.setMouseCallback(key, create_wrapper_function("mousemove", handler))
+
+
+def onDoubleClick(key: str, handler: callable):
+    win(key)
+    cv2.setMouseCallback(key, create_wrapper_function("dblclick", handler))

homa/filters.py

@@ -1,10 +1,10 @@
 from typing import List
 from .classes.Repository import Repository
-from .helpers import create_kernel
+from .helpers.kernel import create_kernel
 import cv2
 
 
-def blur(key: str, kernel: int | List[int] = (7, 7), new_key: str | None = None):
+def blur(key: str, kernel: int | List[int] = (7, 7), new_key: str | None = None) -> None:
     if new_key is None:
         new_key = key
 
@@ -14,21 +14,28 @@ def blur(key: str, kernel: int | List[int] = (7, 7), new_key: str | None = None)
     )
 
 
-def sigma(x: float = 0, y: float = 0):
+def median(key: str, kernel: int | List[int] = (7, 7), new_key: str | None = None) -> None:
+    if new_key is None:
+        new_key = key
+
+    Repository.images[new_key] = cv2.medianBlur(
+        Repository.images[key],
+        create_kernel(kernel)
+    )
+
+
+def sigma(x: float = 0, y: float = 0) -> None:
     Repository.sigmaX = x
     Repository.sigmaY = y
 
 
-def gaussian(key: str, kernel: None | List[int] = None, new_key: str | None = None):
+def gaussian(key: str, kernel: None | List[int] = None, new_key: str | None = None) -> None:
     if new_key is None:
         new_key = key
 
-    if isinstance(kernel, int):
-        kernel = (kernel, kernel)
-
     Repository.images[new_key] = cv2.GaussianBlur(
         Repository.images[key],
-        kernel,
+        create_kernel(kernel),
         sigmaX=Repository.sigmaX,
         sigmaY=Repository.sigmaY
     )

homa/helpers/alias.py

@@ -0,0 +1,14 @@
+from ..classes.Repository import Repository
+from ..classes import Collection
+from typing import List
+
+
+def repo(key: str|None = None):
+    if key is None:
+        return Repository.images
+
+    return Repository.images[key]
+
+
+def collection(items: List[any]):
+    return Collection(items)

homa/helpers/environment.py

@@ -0,0 +1,5 @@
+import sys
+
+
+def is_colab() -> bool:
+    return 'google.colab' in sys.modules

homa/helpers/kernel.py

@@ -0,0 +1,15 @@
+from typing import Tuple
+
+
+def create_kernel(value: int | Tuple[int, int]) -> Tuple[int, int]:
+    if isinstance(value, tuple):
+        x, y = value
+
+        x = x - 1 if x % 2 == 0 else x
+        y = y - 1 if y % 2 == 0 else y
+
+        return (x, y)
+
+    value = value - 1 if value % 2 == 0 else value
+
+    return (value, value)

homa/helpers.py

@@ -1,30 +0,0 @@
-import sys
-from typing import List
-from .classes.Collection import Collection
-from .classes.Logger import Logger
-
-
-def is_colab() -> bool:
-    return 'google.colab' in sys.modules
-
-
-def collection(items: List[any]):
-    return Collection(items)
-
-
-def danger(message: str) -> None:
-    Logger.danger(message)
-
-
-def create_kernel(value: int | tuple) -> tuple:
-    if isinstance(value, tuple):
-        x, y = value
-
-        x = x - 1 if x % 2 == 0 else x
-        y = y - 1 if y % 2 == 0 else y
-
-        return (x, y)
-
-    value = value - 1 if value % 2 == 0 else value
-
-    return (value, value)

homa/main.py

@@ -1,8 +1,20 @@
 import cv2
-from .helpers import danger
+from .classes.Logger import Logger
 from .classes.Repository import Repository
 
 
+def destroy(key: str | None = None) -> None:
+    if key is not None:
+        cv2.destroyWindow(key)
+        return
+
+    cv2.destroyAllWindows()
+
+
+def win(key: str):
+    cv2.namedWindow(key)
+
+
 def path(directory: str) -> None:
     Repository.directory = directory
 
@@ -41,16 +53,23 @@ def show(key: any = None, wait: bool = False, window: str = "Homa Window") -> No
 
     if key is not None and not isinstance(key, str):
         Repository.imshow(window, key)
+        Repository.windows[key] = window
 
     elif key is None:
         for key, image in Repository.images.items():
             Repository.imshow(key, image)
+            Repository.windows[key] = key
 
     elif key is not None:
         if key in Repository.images:
             Repository.imshow(key, Repository.images[key])
+            Repository.windows[key] = key
         else:
-            danger(f"No image found with key {key}")
+            Logger.danger(f"No image found with key {key}")
 
     if wait:
         cv2.waitKey(0)
+
+
+def refresh(key: str) -> None:
+    cv2.imshow(Repository.windows[key], Repository.images[key])

homa/orientation.py

@@ -1,6 +1,5 @@
-from typing import List
 from .classes.Repository import Repository
-from .helpers import collection
+from .helpers.alias import collection
 import numpy
 
 
@@ -15,8 +14,11 @@ def stack(*keys, **settings):
         **settings
     }
 
+    if all(isinstance(item, str) for item in keys):
+        keys = collection(keys).map(lambda key: Repository.images[key])
+
     stacked_image = numpy.concatenate(
-        collection(keys).map(lambda key: Repository.images[key]),
+        keys,
         axis=settings["axis"]
     )
 

homa/shapes.py

@@ -0,0 +1,9 @@
+import cv2
+from .main import refresh
+from .helpers.alias import repo
+from .classes.Repository import Repository
+
+
+def circle(key: str, x: int = 0, y: int = 0, r: int = 1, color=(0, 0, 255), thickness: int = 1):
+    cv2.circle(repo(key), (x, y), r, color, thickness)
+    refresh(key)

homa/spaces.py

@@ -0,0 +1,20 @@
+from typing import Tuple
+from .helpers.alias import repo
+
+
+def rgb(key: str) -> Tuple[int, int, int]:
+    image = repo(key)
+
+    if image.shape[2] == 3:
+        # found three channels in stored image
+        # meaning that it has been loaded as a color image
+
+        b = image[:, :, 0]
+        g = image[:, :, 0]
+        r = image[:, :, 0]
+
+        return (r, g, b)
+
+    if image.shape[2] == 1:
+        gray = image[:, :, 0]
+        return (gray, gray, gray)

{homa-0.22.dist-info → homa-0.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: homa
-Version: 0.22
+Version: 0.23
 Maintainer: Taha Shieenavaz
 Maintainer-email: tahashieenavaz@gmail.com
 Description-Content-Type: text/markdown
@@ -49,7 +49,8 @@ image("horse.jpg")
 blur("horse", 7)                    # rewrites "horse" key
 blur("horse", (7, 19))              # rewrites "horse" key
 blur("horse", 9, "blurred horse")   # as a new key in the repository
-show("blurred horse")
+
+showWait("blurred horse")
 ```
 
 ### Gaussian Blur
@@ -62,7 +63,8 @@ image("horse.jpg")
 gaussian("horse", 7)                             # rewrites "horse" key
 gaussian("horse", (7, 19))                       # rewrites "horse" key
 gaussian("horse", 9, "gaussian blurred horse")   # as a new key in the repository
-show("gaussian blurred horse")
+
+showWait("gaussian blurred horse")
 ```
 
 ## Stacking

homa-0.23.dist-info/RECORD

@@ -0,0 +1,25 @@
+homa/__init__.py,sha256=fImpDDouSHoDdq-q-O0p8RirzGTgXXRI4Ol_LU4f9sI,182
+homa/camera.py,sha256=PHZb3sX0EFdwLDixYindluVDftg6JZlG_Ssbd3hgiNE,464
+homa/constants.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+homa/events.py,sha256=Q077YcaM8-VjfKYhPQCQ7dasdu-mACKMf5cupiLNDoY,1946
+homa/filters.py,sha256=lEwUZtCUSH59D4J9lWQ-d0ObSk_ikZBGRnNAZWMnpPo,1058
+homa/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+homa/main.py,sha256=1bRjOrU-Frdlj5a1cLvM2pUVLpk3K9EK6J0bFBC5hQU,1774
+homa/orientation.py,sha256=kR7mEmbjCdyxNeQ7UuGpwTFD6EaNbXmErzaln3gpzfk,804
+homa/shapes.py,sha256=MVZWkOtqkby8KiL04mK4nrirDGeQM3k8dDEZIpDt8AU,283
+homa/spaces.py,sha256=yEIXofkM0ebvRBbRp-e7epETHW_xmwSrCoTyYUYV-uA,460
+homa/classes/Collection.py,sha256=7hskbC8wxTAC3hHmFYXtzR5W05lsA4GaYTzjiez_6Zg,222
+homa/classes/Logger.py,sha256=2VjXJzBLMW5joArFpu2wGrbWUw7S-77z7LNhKUG1Iow,93
+homa/classes/Repository.py,sha256=w7hlbj6QIWNV2lWNs5-j6OXacSm6UGQ9DtX1zxaUA7c,589
+homa/classes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+homa/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+homa/helpers/alias.py,sha256=a-wq7UUnT_nYIh5nP-bc_EZb-0zF5VfnKwVhSuLSjw8,288
+homa/helpers/environment.py,sha256=I7rD9B5zv3W8lkspntcshm5PZZRRSW81OqIjKOdZW6U,78
+homa/helpers/kernel.py,sha256=sAdIW92coy0BoAPlk5Ez4sE5iFU207TU7OfjP7sPSkM,330
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/test_images.py,sha256=4k5tDcMc2YZOKAq3wNQnbJ_cO1Y_DQ-YUn82yzy__ww,772
+homa-0.23.dist-info/LICENSE,sha256=js3WDbJn9k5EN6sy1uuP2QBXxyPgS5DjO4Bf5yE35hQ,1072
+homa-0.23.dist-info/METADATA,sha256=rBFjHYZz3mSqydiHvPLzLoTGAbMtkJF18zVnqPZvcUk,2071
+homa-0.23.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+homa-0.23.dist-info/top_level.txt,sha256=i1-Cx_lkcOD5QJEK9JYp1uDEBhtshAN7No-99mhpOZ0,11
+homa-0.23.dist-info/RECORD,,

homa-0.23.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+homa
+tests

tests/test_images.py

@@ -0,0 +1,27 @@
+from homa import *
+
+import unittest
+
+
+class LoadingImagesTest(unittest.TestCase):
+    def setUp(self):
+        pass
+
+    def test_can_load_images_into_repository(self):
+        image("italy.jpg", "italy")
+        self.assertIn("italy", repo())
+
+    def test_it_does_not_load_extra_images(self):
+        image("italy.jpg", "italy")
+        self.assertEqual(len(repo()), 1)
+
+        image("italy.jpg", "italy again")
+        self.assertEqual(len(repo()), 2)
+
+    def test_it_can_load_images_as_black_and_white(self):
+        image("italy.jpg", "italy", color=False)
+        self.assertEqual(len(repo("italy").shape), 2)
+
+    def test_it_can_load_images_as_colorful(self):
+        image("italy.jpg", "italy", color=True)
+        self.assertEqual(len(repo("italy").shape), 3)

homa-0.22.dist-info/RECORD

@@ -1,16 +0,0 @@
-homa/__init__.py,sha256=BKmHKbWAjj61ESPbgQTmsQ7GH_NjRBowkD15QOIZBf0,92
-homa/camera.py,sha256=KZf1r7olUkMRmX5c2qcDskMAq19cKWN5gTwj24pOM9o,391
-homa/constants.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-homa/filters.py,sha256=oIo6dsB8OnCAkSDlYUzOySxEhlUI4D6ZHkKvO2GvuP4,821
-homa/helpers.py,sha256=iS-MzPtcfwJYxmhM9NtWfWmWuLUwnAKaN3ZxxVA4SHw,593
-homa/main.py,sha256=g0Uk271_qE8xq6Gs2hVMGNI73Sp7zjPbP-jiaaFndfo,1346
-homa/orientation.py,sha256=0BdvozqBuuscMjG9zSl6YFt__MS4zM52Ja04SjbxIpk,749
-homa/classes/Collection.py,sha256=7hskbC8wxTAC3hHmFYXtzR5W05lsA4GaYTzjiez_6Zg,222
-homa/classes/Logger.py,sha256=2VjXJzBLMW5joArFpu2wGrbWUw7S-77z7LNhKUG1Iow,93
-homa/classes/Repository.py,sha256=c0F_okpC6d8ppsOQ5X6eUD2dfHmIiVInNrGNe6b-3ss,705
-homa/classes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-homa-0.22.dist-info/LICENSE,sha256=js3WDbJn9k5EN6sy1uuP2QBXxyPgS5DjO4Bf5yE35hQ,1072
-homa-0.22.dist-info/METADATA,sha256=VpZ5AVoNJwnwHp0o8mSryTdO9oghE1u8Ha5zJfnoi4Y,2061
-homa-0.22.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-homa-0.22.dist-info/top_level.txt,sha256=tmOfy2tuaAwc3W5-i6j61_vYJsXgR4ivBWkhJ3ZtJDc,5
-homa-0.22.dist-info/RECORD,,

homa-0.22.dist-info/top_level.txt

@@ -1 +0,0 @@
-homa