Add type aliases

ddacw · ddacw · commit fc638921c723 · 2021-07-07T19:41:57.000+02:00
diff --git a/modules/python/src2/gen2.py b/modules/python/src2/gen2.py
@@ -215,6 +215,14 @@ def prefixed_lines():
 
 """
 
+stub_type_aliases = {
+    "Mat": "np.ndarray",
+    "Rect": "Sequence[int]",
+    "Size": "Sequence[int]",
+    "Size2f": "Sequence[float]",
+    "UMat": "np.ndarray"
+}
+
 class FormatStrings:
     string = 's'
     unsigned_char = 'b'
@@ -1334,8 +1342,13 @@ def process_isalgorithm(classinfo):
         for name, classinfo in self.classes.items():
             process_isalgorithm(classinfo)
 
-        # header and enums for stub
+        # stub header and type aliases
         self.code_stubs.write(stub_header)
+        for alias, type in sorted(stub_type_aliases.items()):
+            self.code_stubs.write("{} = {}\n".format(alias, type))
+        self.code_stubs.write("\n")
+
+        # stub enums
         for ns_name, ns in self.namespaces.items():
             self.code_stubs.write("# {}\n".format(ns_name))
             for name in sorted(ns.consts):
