Implement _PyLong_NumBits

Author: msimacek

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_long.py

@@ -471,3 +471,16 @@ class TestPyLong(CPyExtTestCase):
         arguments=["const char* bytes", "Py_ssize_t size", "int little_endian", "int is_signed"],
         cmpfunc=unhandled_error_compare,
     )
+
+    test__PyLong_NumBits = CPyExtFunction(
+        lambda args: args[0].bit_length(),
+        lambda: (
+            (1,),
+            (1230948701328090743,),
+            (-1230948701328090743,),
+        ),
+        resultspec="n",
+        argspec="O",
+        arguments=["PyObject* obj"],
+        cmpfunc=unhandled_error_compare,
+    )

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/cext/PythonCextLongBuiltins.java

@@ -483,4 +483,13 @@ static Object convert(Object charPtr, long size, int littleEndian, int signed,
             return fromByteArray.execute(inliningTarget, bytes, littleEndian != 0, signed != 0);
         }
     }
+
+    @CApiBuiltin(ret = SIZE_T, args = {PyObject}, call = Direct)
+    abstract static class _PyLong_NumBits extends CApiUnaryBuiltinNode {
+        @Specialization
+        static long numBits(Object obj,
+                        @Cached IntBuiltins.BitLengthNode bitLengthNode) {
+            return bitLengthNode.execute(obj);
+        }
+    }
 }

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

@@ -1076,7 +1076,6 @@ public final class CApiFunction {
     @CApiBuiltin(name = "_PyLong_GCD", ret = PyObject, args = {PyObject, PyObject}, call = NotImplemented)
     @CApiBuiltin(name = "_PyLong_Lshift", ret = PyObject, args = {PyObject, SIZE_T}, call = NotImplemented)
     @CApiBuiltin(name = "_PyLong_New", ret = PyLongObject, args = {Py_ssize_t}, call = NotImplemented)
-    @CApiBuiltin(name = "_PyLong_NumBits", ret = SIZE_T, args = {PyObject}, call = NotImplemented)
     @CApiBuiltin(name = "_PyLong_Rshift", ret = PyObject, args = {PyObject, SIZE_T}, call = NotImplemented)
     @CApiBuiltin(name = "_PyLong_Size_t_Converter", ret = Int, args = {PyObject, Pointer}, call = NotImplemented)
     @CApiBuiltin(name = "_PyLong_UnsignedInt_Converter", ret = Int, args = {PyObject, Pointer}, call = NotImplemented)

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/ints/IntBuiltins.java

@@ -3141,7 +3141,9 @@ static int bitCount(PInt i) {
     @Builtin(name = "bit_length", minNumOfPositionalArgs = 1)
     @GenerateNodeFactory
     @TypeSystemReference(PythonArithmeticTypes.class)
-    abstract static class BitLengthNode extends PythonBuiltinNode {
+    public abstract static class BitLengthNode extends PythonUnaryBuiltinNode {
+        public abstract int execute(Object argument);
+
         @Specialization
         static int bitLength(int argument) {
             return Integer.SIZE - Integer.numberOfLeadingZeros(Math.abs(argument));