Implement C API function 'PySet_Size'

Author: fangerer

graalpython/com.oracle.graal.python.cext/src/setobject.c

@@ -53,6 +53,12 @@ PyObject * PyFrozenSet_New(PyObject *iterable) {
     return UPCALL_CEXT_O(_jls_PyFrozenSet_New, native_to_java(iterable));
 }
 
+typedef Py_ssize_t (*set_size_fun_t)(PyObject *anyset);
+UPCALL_TYPED_ID(PySet_Size, set_size_fun_t);
+Py_ssize_t PySet_Size(PyObject *anyset) {
+    return PySet_GET_SIZE(anyset);
+}
+
 UPCALL_ID(PySet_Add);
 int PySet_Add(PyObject *set, PyObject *key) {
     return UPCALL_CEXT_I(_jls_PySet_Add, native_to_java(set), native_to_java(key));

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

@@ -198,6 +198,7 @@
 import com.oracle.graal.python.builtins.objects.object.PythonObject;
 import com.oracle.graal.python.builtins.objects.object.PythonObjectLibrary;
 import com.oracle.graal.python.builtins.objects.set.PBaseSet;
+import com.oracle.graal.python.builtins.objects.set.PSet;
 import com.oracle.graal.python.builtins.objects.str.NativeCharSequence;
 import com.oracle.graal.python.builtins.objects.str.PString;
 import com.oracle.graal.python.builtins.objects.traceback.GetTracebackNode;
@@ -4529,4 +4530,14 @@ private static Object[] unwrapArray(Object ptr, InteropLibrary ptrLib, ToJavaNod
             throw CompilerDirectives.shouldNotReachHere();
         }
     }
+
+    @Builtin(name = "PySet_Size", minNumOfPositionalArgs = 1)
+    @GenerateNodeFactory
+    abstract static class PySetSize extends PythonUnaryBuiltinNode {
+        @Specialization
+        static int doSet(PSet type,
+                        @CachedLibrary(limit = "3") HashingStorageLibrary lib) {
+            return lib.length(type.getDictStorage());
+        }
+    }
 }