[mypyc] Add experimental C extension librt.vecs (part 1/2)

mypy/typeshed/stubs/librt/librt/vecs.pyi

@@ -0,0 +1,21 @@
+from typing import TypeVar, Generic, Iterable, Iterator, overload
+from mypy_extensions import i64
+
+T = TypeVar("T")
+
+class vec(Generic[T]):
+    @overload
+    def __init__(self) -> None: ...
+    @overload
+    def __init__(self, items: Iterable[T], /) -> None: ...
+    def __len__(self) -> i64: ...
+    @overload
+    def __getitem__(self, i: i64, /) -> T: ...
+    @overload
+    def __getitem__(self, i: slice, /) -> vec[T]: ...
+    def __setitem__(self, i: i64, o: T, /) -> None: ...
+    def __iter__(self) -> Iterator[T]: ...
+
+def append(v: vec[T], o: T, /) -> vec[T]: ...
+def remove(v: vec[T], o: T, /) -> vec[T]: ...
+def pop(v: vec[T], i: i64 = -1, /) -> tuple[vec[T], T]: ...

mypyc/build.py

@@ -104,6 +104,20 @@ class ModDesc(NamedTuple):
         ],
         ["base64"],
     ),
+    ModDesc(
+        "librt.vecs",
+        [
+            "vecs/librt_vecs.c",
+            "vecs/vec_i64.c",
+            "vecs/vec_i32.c",
+            "vecs/vec_i16.c",
+            "vecs/vec_u8.c",
+            "vecs/vec_float.c",
+            "vecs/vec_bool.c",
+        ],
+        ["vecs/librt_vecs.h", "vecs/vec_template.c"],
+        ["vecs"],
+    ),
 ]
 
 try:

mypyc/ir/deps.py

@@ -48,6 +48,7 @@ def get_header(self) -> str:
 
 LIBRT_STRINGS: Final = Capsule("librt.strings")
 LIBRT_BASE64: Final = Capsule("librt.base64")
+LIBRT_VECS: Final = Capsule("librt.vecs")
 
 BYTES_EXTRA_OPS: Final = SourceDep("bytes_extra_ops.c")
 BYTES_WRITER_EXTRA_OPS: Final = SourceDep("byteswriter_extra_ops.c")

mypyc/lib-rt/setup.py

@@ -132,5 +132,19 @@ def run(self) -> None:
                 include_dirs=[".", "base64"],
                 extra_compile_args=cflags,
             ),
+            Extension(
+                "librt.vecs",
+                [
+                    "vecs/librt_vecs.c",
+                    "vecs/vec_i64.c",
+                    "vecs/vec_i32.c",
+                    "vecs/vec_i16.c",
+                    "vecs/vec_u8.c",
+                    "vecs/vec_float.c",
+                    "vecs/vec_bool.c",
+                ],
+                include_dirs=[".", "vecs"],
+                extra_compile_args=cflags,
+            ),
         ]
     )