[GR-57219] Add custom class registry for foreign objects

PullRequest: graalpython/3467

Author: Sowasvonbot

Diff for: graalpython/com.oracle.graal.python.benchmarks/python/harness.py

@@ -267,8 +267,8 @@ def __init__(self, bench_file, bench_args=None, iterations=1, warmup=-1, warmup_
 
     @staticmethod
     def get_bench_module(bench_file):
-        name = bench_file.rpartition("/")[2].partition(".")[0].replace('.py', '')
-        directory = bench_file.rpartition("/")[0]
+        name = bench_file.rpartition(os.sep)[2].partition(".")[0].replace('.py', '')
+        directory = bench_file.rpartition(os.sep)[0]
         pkg = []
         while any(f.endswith("__init__.py") for f in os.listdir(directory)):
             directory, slash, postfix = directory.rpartition("/")

Diff for: graalpython/com.oracle.graal.python.benchmarks/python/micro/java-register.py

@@ -0,0 +1,84 @@
+# Copyright (c) 2024, 2024, Oracle and/or its affiliates. All rights reserved.
+# DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+#
+# The Universal Permissive License (UPL), Version 1.0
+#
+# Subject to the condition set forth below, permission is hereby granted to any
+# person obtaining a copy of this software, associated documentation and/or
+# data (collectively the "Software"), free of charge and under any and all
+# copyright rights in the Software, and any and all patent rights owned or
+# freely licensable by each licensor hereunder covering either (i) the
+# unmodified Software as contributed to or provided by such licensor, or (ii)
+# the Larger Works (as defined below), to deal in both
+#
+# (a) the Software, and
+#
+# (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
+# one is included with the Software each a "Larger Work" to which the Software
+# is contributed by such licensors),
+#
+# without restriction, including without limitation the rights to copy, create
+# derivative works of, display, perform, and distribute the Software and make,
+# use, sell, offer for sale, import, export, have made, and have sold the
+# Software and the Larger Work(s), and to sublicense the foregoing rights on
+# either these or other terms.
+#
+# This license is subject to the following condition:
+#
+# The above copyright notice and either this complete permission notice or at a
+# minimum a reference to the UPL must be included in all copies or substantial
+# portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+
+# This benchmark hits the GetRegisteredClassNodes as much as possible with one registration
+
+import polyglot
+from java.util import LinkedList
+from java.util import List
+
+
+
+@polyglot.interop_type(List)
+class JList:
+    def append(self, value):
+        self.add(value)
+
+    def __getitem__(self, item):
+        return self.get(item)
+
+    def get_value(self):
+        return sum(self)
+
+    def length(self):
+        return self.size()
+
+
+# igv: function_root_create_list_at
+def create_list(num: int, l: JList):
+    for i in range(num):
+        l.add(LinkedList())
+        for j in range(i):
+            l[i].add(j)
+
+    for index, li in enumerate(l):
+        assert li.length() * (li.length() - 1) / 2 == li.get_value()
+
+
+def measure(num):
+    j_list = LinkedList()
+    create_list(num, j_list)
+
+
+def __benchmark__(num=2_000):
+    measure(num)
+
+if __name__ == '__main__':
+    __benchmark__()

Diff for: graalpython/com.oracle.graal.python.frozen/freeze_modules.py

@@ -6,16 +6,15 @@
 """Freeze specified modules
 """
 
-from collections import namedtuple
+import argparse
 import marshal
-import os
 import ntpath
+import os
 import posixpath
-import sys
-import argparse
 import shutil
-
+import sys
 from _sha256 import sha256
+from collections import namedtuple
 
 FROZEN_ONLY = os.path.join(os.path.dirname(__file__), "flag.py")
 
@@ -111,7 +110,7 @@ def add_graalpython_core():
         l.append(f"{modname} : {modname} = {modpath}")
     for name in [
         "__graalpython__",
-        "_interop_behavior",
+        "_polyglot",
         "_sre",
         "_sysconfig",
         "_weakref",

Diff for: graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_interop_behavior.py

@@ -38,16 +38,16 @@
 # SOFTWARE.
 
 import sys
-from unittest import skipIf
-from . import CPyExtType
+from unittest import skipIf, TestCase
 
+from . import CPyExtType
 
 if sys.implementation.name == "graalpy":
     import polyglot
     from __graalpython__ import is_native
     is_windows = sys.platform == "win32"
 
-    class TestPyStructNumericSequenceTypes(object):
+    class TestPyStructNumericSequenceTypes(TestCase):
 
         @skipIf(is_native, "not supported in native mode")
         def test_interop_assertions(self):
@@ -131,6 +131,34 @@ def test_native_object_interop_behavior_extension(self):
                 # print("Error : ", e)
                 assert False
 
+        @skipIf(is_native, "not supported in native mode")
+        def test_decorator_usage(self):
+            class MyType(object):
+                data = 100000000000
+
+            t = MyType()
+            with self.assertRaises(ValueError):
+                # Fail, because argument is not a type
+                @polyglot.interop_behavior(t)
+                class MyTypeBehavior:
+                    @staticmethod
+                    def is_number(_):
+                        return True
+
+            with self.assertRaises(ValueError):
+                # Fail, because method decorated
+                @polyglot.interop_behavior(MyType)
+                def is_number(_):
+                    return True
+
+            # multiple decorator and custom CPyExtType should work
+            @polyglot.interop_behavior(CPyExtType("dummy", ""))
+            @polyglot.interop_behavior(MyType)
+            class MyTypeBehavior:
+                @staticmethod
+                def is_number(_):
+                    return True
+
         @skipIf(is_windows, "GR-51663: fails on windows")
         def test_native_sequence_interop(self):
             MySequenceNativeType = CPyExtType("MySequenceNativeType",