Specialize 'PyLong_FromLongLong'.

Author: fangerer

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/TruffleCextBuiltins.java

@@ -1863,4 +1863,39 @@ Object createCache(TruffleObject ptrToResolveHandle) {
             return new HandleCache(ptrToResolveHandle);
         }
     }
+
+    @Builtin(name = "PyLong_FromLongLong", fixedNumOfPositionalArgs = 2)
+    @GenerateNodeFactory
+    abstract static class PyLong_FromLongLong extends PythonBinaryBuiltinNode {
+        @Specialization(guards = "signed != 0")
+        Object doSignedInt(int n, @SuppressWarnings("unused") int signed,
+                        @Cached("create()") CExtNodes.ToSulongNode toSulongNode) {
+            return toSulongNode.execute(n);
+        }
+
+        @Specialization(guards = "signed == 0")
+        Object doUnsignedInt(int n, @SuppressWarnings("unused") int signed,
+                        @Cached("create()") CExtNodes.ToSulongNode toSulongNode) {
+            if (n < 0) {
+                return toSulongNode.execute(n & 0xFFFFFFFFL);
+            }
+            return toSulongNode.execute(n);
+        }
+
+        @Specialization(guards = "signed != 0")
+        Object doSignedLong(long n, @SuppressWarnings("unused") int signed,
+                        @Cached("create()") CExtNodes.ToSulongNode toSulongNode) {
+            return toSulongNode.execute(n);
+        }
+
+        @Specialization(guards = "signed == 0")
+        Object doUnsignedLong(long n, @SuppressWarnings("unused") int signed,
+                        @Cached("create()") CExtNodes.ToSulongNode toSulongNode) {
+            if (n < 0) {
+                CompilerDirectives.transferToInterpreter();
+                throw new UnsupportedOperationException();
+            }
+            return toSulongNode.execute(n);
+        }
+    }
 }

graalpython/lib-graalpython/python_cext.py

@@ -314,14 +314,6 @@ def PyList_Size(listObj):
 
 ##################### LONG
 
-@may_raise
-def PyLong_FromLongLong(n, signed):
-    if signed:
-        return int(n)
-    else:
-        return int(n & 0xffffffffffffffff)
-
-
 @may_raise(-1)
 def PyLong_AsPrimitive(n, signed, size, descr):
     if isinstance(n, int):