Cache native type for (builtin) classes.

Author: fangerer

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/cext/PythonObjectNativeWrapperMR.java

@@ -62,6 +62,7 @@
 import com.oracle.graal.python.builtins.objects.cext.NativeWrappers.PythonClassNativeWrapper;
 import com.oracle.graal.python.builtins.objects.cext.NativeWrappers.PythonNativeWrapper;
 import com.oracle.graal.python.builtins.objects.cext.NativeWrappers.PythonObjectNativeWrapper;
+import com.oracle.graal.python.builtins.objects.cext.PythonObjectNativeWrapperMRFactory.GetSulongTypeNodeGen;
 import com.oracle.graal.python.builtins.objects.cext.PythonObjectNativeWrapperMRFactory.PAsPointerNodeGen;
 import com.oracle.graal.python.builtins.objects.cext.PythonObjectNativeWrapperMRFactory.ReadNativeMemberNodeGen;
 import com.oracle.graal.python.builtins.objects.cext.PythonObjectNativeWrapperMRFactory.ToPyObjectNodeGen;
@@ -84,6 +85,7 @@
 import com.oracle.graal.python.builtins.objects.object.PythonObject;
 import com.oracle.graal.python.builtins.objects.slice.PSlice;
 import com.oracle.graal.python.builtins.objects.str.PString;
+import com.oracle.graal.python.builtins.objects.type.LazyPythonClass;
 import com.oracle.graal.python.builtins.objects.type.PythonBuiltinClass;
 import com.oracle.graal.python.builtins.objects.type.PythonClass;
 import com.oracle.graal.python.nodes.PGuards;
@@ -135,11 +137,51 @@ public class PythonObjectNativeWrapperMR {
     @SuppressWarnings("unknown-message")
     @Resolve(message = "com.oracle.truffle.llvm.spi.GetDynamicType")
     abstract static class GetDynamicTypeNode extends Node {
-        @Child GetClassNode getClass = GetClassNode.create();
-        @Child AsPythonObjectNode getDelegate = AsPythonObjectNode.create();
+        @Child private GetLazyClassNode getLazyClassNode = GetLazyClassNode.create();
+        @Child private GetSulongTypeNode getSulongTypeNode = GetSulongTypeNode.create();
+        @Child private AsPythonObjectNode getDelegate = AsPythonObjectNode.create();
 
         public Object access(PythonNativeWrapper object) {
-            PythonClass klass = getClass.execute(getDelegate.execute(object));
+            return getSulongTypeNode.execute(getLazyClassNode.execute(getDelegate.execute(object)));
+        }
+    }
+
+    abstract static class GetSulongTypeNode extends PNodeWithContext {
+
+        private final ConditionProfile profile = ConditionProfile.createBinaryProfile();
+
+        public abstract Object execute(LazyPythonClass clazz);
+
+        @Specialization(guards = "clazz == cachedClass", limit = "10")
+        Object doBuiltinCached(@SuppressWarnings("unused") PythonBuiltinClassType clazz,
+                        @Cached("clazz") @SuppressWarnings("unused") PythonBuiltinClassType cachedClass,
+                        @Cached("getSulongTypeForBuiltinClass(clazz)") Object sulongType) {
+            return sulongType;
+        }
+
+        @Specialization(replaces = "doBuiltinCached")
+        Object doBuiltinGeneric(PythonBuiltinClassType clazz) {
+            return getSulongTypeForBuiltinClass(clazz);
+        }
+
+        @Specialization(assumptions = "singleContextAssumption()", guards = "clazz == cachedClass")
+        Object doGeneric(@SuppressWarnings("unused") PythonClass clazz,
+                        @Cached("clazz") @SuppressWarnings("unused") PythonClass cachedClass,
+                        @Cached("doGeneric(clazz)") Object sulongType) {
+            return sulongType;
+        }
+
+        @Specialization
+        Object doGeneric(PythonClass clazz) {
+            return getSulongTypeForClass(clazz);
+        }
+
+        protected Object getSulongTypeForBuiltinClass(PythonBuiltinClassType clazz) {
+            PythonClass pythonClass = getPythonClass(clazz, profile);
+            return getSulongTypeForClass(pythonClass);
+        }
+
+        private static Object getSulongTypeForClass(PythonClass klass) {
             Object sulongType = klass.getSulongType();
             if (sulongType == null) {
                 CompilerDirectives.transferToInterpreter();
@@ -163,6 +205,11 @@ private static Object findBuiltinClass(PythonClass klass) {
             }
             return sulongType;
         }
+
+        public static GetSulongTypeNode create() {
+            return GetSulongTypeNodeGen.create();
+        }
+
     }
 
     @Resolve(message = "READ")