From d73a85b82eb6d6168250f999606ab6970aa7aeef Mon Sep 17 00:00:00 2001
From: Andrew Senetar <arsenetar@gmail.com>
Date: Sat, 30 Apr 2022 05:11:54 -0500
Subject: [PATCH] Add type hints for compiled modules

---
 core/pe/block.pyi | 13 +++++++++++++
 core/pe/cache.pyi |  6 ++++++
 qt/pe/block.pyi   |  7 +++++++
 3 files changed, 26 insertions(+)
 create mode 100644 core/pe/block.pyi
 create mode 100644 core/pe/cache.pyi
 create mode 100644 qt/pe/block.pyi

diff --git a/core/pe/block.pyi b/core/pe/block.pyi
new file mode 100644
index 00000000..0da84be9
--- /dev/null
+++ b/core/pe/block.pyi
@@ -0,0 +1,13 @@
+from typing import Tuple, List, Union, Sequence
+
+_block = Tuple[int, int, int]
+
+class NoBlocksError(Exception): ...  # noqa: E302, E701
+class DifferentBlockCountError(Exception): ...  # noqa E701
+
+def getblock(image: object) -> Union[_block, None]: ...  # noqa: E302
+def getblocks2(image: object, block_count_per_side: int) -> Union[List[_block], None]: ...
+def diff(first: _block, second: _block) -> int: ...
+def avgdiff(  # noqa: E302
+    first: Sequence[_block], second: Sequence[_block], limit: int = 768, min_iterations: int = 1
+) -> Union[int, None]: ...
diff --git a/core/pe/cache.pyi b/core/pe/cache.pyi
new file mode 100644
index 00000000..fbf1e8c8
--- /dev/null
+++ b/core/pe/cache.pyi
@@ -0,0 +1,6 @@
+from typing import Union, Tuple, List
+
+_block = Tuple[int, int, int]
+
+def colors_to_string(colors: List[_block]) -> str: ...  # noqa: E302
+def string_to_colors(s: str) -> Union[List[_block], None]: ...
diff --git a/qt/pe/block.pyi b/qt/pe/block.pyi
new file mode 100644
index 00000000..17954a92
--- /dev/null
+++ b/qt/pe/block.pyi
@@ -0,0 +1,7 @@
+from typing import Tuple, List, Union
+from PyQt5.QtGui import QImage
+
+_block = Tuple[int, int, int]
+
+def getblock(image: QImage) -> _block: ...  # noqa: E302
+def getblocks(image: QImage, block_count_per_side: int) -> Union[List[_block], None]: ...