Compiled versions of 'any' and 'all'.

Author: braxtonmckee

typed_python/compiler/python_object_representation.py

@@ -72,6 +72,7 @@
 from typed_python.compiler.type_wrappers.bytes_wrapper import BytesWrapper, BytesMaketransWrapper
 from typed_python.compiler.type_wrappers.python_object_of_type_wrapper import PythonObjectOfTypeWrapper
 from typed_python.compiler.type_wrappers.abs_wrapper import AbsWrapper
+from typed_python.compiler.type_wrappers.all_any_wrapper import AllWrapper, AnyWrapper
 from typed_python.compiler.type_wrappers.min_max_wrapper import MinWrapper, MaxWrapper
 from typed_python.compiler.type_wrappers.repr_wrapper import ReprWrapper
 from types import ModuleType
@@ -262,6 +263,12 @@ def pythonObjectRepresentation(context, f, owningGlobalScopeAndName=None):
     if f is max:
         return TypedExpression(context, native_ast.nullExpr, MaxWrapper(), False)
 
+    if f is any:
+        return TypedExpression(context, native_ast.nullExpr, AnyWrapper(), False)
+
+    if f is all:
+        return TypedExpression(context, native_ast.nullExpr, AllWrapper(), False)
+
     if f is repr:
         return TypedExpression(context, native_ast.nullExpr, ReprWrapper(), False)
 

typed_python/compiler/tests/any_all_compilation_test.py

@@ -0,0 +1,34 @@
+#   Copyright 2017-2023 typed_python Authors
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+
+from typed_python import Entrypoint, TupleOf
+
+
+def test_compiles_any_and_all():
+    @Entrypoint
+    def callAny(x):
+        return any(x)
+
+    @Entrypoint
+    def callAll(x):
+        return all(x)
+
+    assert callAll.resultTypeFor(list).typeRepresentation is bool
+    assert callAny.resultTypeFor(list).typeRepresentation is bool
+
+    for T in [list, TupleOf(int)]:
+        assert callAny(T([1, 2, 3])) == any(T([1, 2, 3]))
+        assert callAll(T([1, 2, 3])) == all(T([1, 2, 3]))
+        assert callAny(T([0, 0, 0])) == any(T([0, 0, 0]))
+        assert callAll(T([0, 0, 0])) == all(T([0, 0, 0]))

typed_python/compiler/type_wrappers/all_any_wrapper.py

@@ -0,0 +1,62 @@
+#   Copyright 2017-2019 typed_python Authors
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+
+from typed_python.compiler.type_wrappers.wrapper import Wrapper
+import typed_python.compiler.native_ast as native_ast
+
+
+def all(iterable):
+    for a in iterable:
+        if not a:
+            return False
+
+    return True
+
+
+def any(iterable):
+    for a in iterable:
+        if a:
+            return True
+
+    return False
+
+
+class AllWrapper(Wrapper):
+    is_pod = True
+    is_empty = False
+    is_pass_by_ref = False
+
+    def __init__(self):
+        super().__init__(print)
+
+    def getNativeLayoutType(self):
+        return native_ast.Type.Void()
+
+    def convert_call(self, context, expr, args, kwargs):
+        return context.call_py_function(all, args, kwargs)
+
+
+class AnyWrapper(Wrapper):
+    is_pod = True
+    is_empty = False
+    is_pass_by_ref = False
+
+    def __init__(self):
+        super().__init__(print)
+
+    def getNativeLayoutType(self):
+        return native_ast.Type.Void()
+
+    def convert_call(self, context, expr, args, kwargs):
+        return context.call_py_function(all, args, kwargs)