Implement C API function 'PySet_New' and 'PyFrozenSet_New'.

Author: fangerer

graalpython/com.oracle.graal.python.cext/src/capi.h

@@ -74,6 +74,7 @@ POLYGLOT_DECLARE_TYPE(PyBytesObject);
 POLYGLOT_DECLARE_STRUCT(_longobject);
 POLYGLOT_DECLARE_TYPE(PyCapsule);
 POLYGLOT_DECLARE_TYPE(PyMemoryViewObject);
+POLYGLOT_DECLARE_TYPE(PySetObject);
 
 
 extern void* to_java(PyObject* obj);

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

@@ -40,3 +40,19 @@
 
 PyTypeObject PySet_Type = PY_TRUFFLE_TYPE("set", &PyType_Type, Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC);
 PyTypeObject PyFrozenSet_Type = PY_TRUFFLE_TYPE("frozenset", &PyType_Type, Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC);
+
+PyObject * PySet_New(PyObject *iterable) {
+	void* result = polyglot_invoke(PY_TRUFFLE_CEXT, "PySet_New", to_java(iterable));
+	if (result == ERROR_MARKER) {
+		return NULL;
+	}
+	return to_sulong(result);
+}
+
+PyObject * PyFrozenSet_New(PyObject *iterable) {
+	void* result = polyglot_invoke(PY_TRUFFLE_CEXT, "PyFrozenSet_New", to_java(iterable));
+	if (result == ERROR_MARKER) {
+		return NULL;
+	}
+	return to_sulong(result);
+}

graalpython/lib-graalpython/python_cext.py

@@ -152,6 +152,25 @@ def PyDict_DelItem(dictObj, key):
     return 0
 
 
+##################### SET, FROZENSET
+
+
+@may_raise
+def PySet_New(iterable):
+    if iterable:
+        return set(iterable)
+    else:
+        return set()
+
+
+@may_raise
+def PyFrozenSet_New(iterable):
+    if iterable:
+        return frozenset(iterable)
+    else:
+        return frozenset()
+
+
 ##################### MAPPINGPROXY