Add PyTuple_Pack2

eendebakpt · eendebakpt · commit f2431b38e716 · 2024-04-24T00:26:04.000+02:00
diff --git a/Include/tupleobject.h b/Include/tupleobject.h
@@ -33,6 +33,7 @@ PyAPI_FUNC(PyObject *) PyTuple_GetItem(PyObject *, Py_ssize_t);
 PyAPI_FUNC(int) PyTuple_SetItem(PyObject *, Py_ssize_t, PyObject *);
 PyAPI_FUNC(PyObject *) PyTuple_GetSlice(PyObject *, Py_ssize_t, Py_ssize_t);
 PyAPI_FUNC(PyObject *) PyTuple_Pack(Py_ssize_t, ...);
+PyAPI_FUNC(PyObject*) PyTuple_Pack2(PyObject* a, PyObject* b);
 
 #ifndef Py_LIMITED_API
 #  define Py_CPYTHON_TUPLEOBJECT_H
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c
@@ -356,7 +356,7 @@ pairwise_next(pairwiseobject *po)
         return NULL;
     }
     /* Future optimization: Reuse the result tuple as we do in enumerate() */
-    result = PyTuple_Pack(2, old, new);
+    result = PyTuple_Pack2(old, new);
     Py_XSETREF(po->old, new);
     Py_DECREF(old);
     return result;
diff --git a/Objects/tupleobject.c b/Objects/tupleobject.c
@@ -151,6 +151,21 @@ _PyTuple_MaybeUntrack(PyObject *op)
     _PyObject_GC_UNTRACK(op);
 }
 
+PyObject*
+PyTuple_Pack2(PyObject *a, PyObject *b)
+{
+    assert(a!=NULL);
+    assert(b != NULL);
+    PyTupleObject* result = tuple_alloc(2);
+    if (result == NULL) {
+        return NULL;
+    }
+    result->ob_item[0] = Py_NewRef(a);
+    result->ob_item[1] = Py_NewRef(b);
+    _PyObject_GC_TRACK(result);
+    return (PyObject*)result;
+}
+
 PyObject *
 PyTuple_Pack(Py_ssize_t n, ...)
 {

Original file line number	Diff line number	Diff line change
`@@ -356,7 +356,7 @@ pairwise_next(pairwiseobject *po)`
`356`	`356`	`return NULL;`
`357`	`357`	`}`
`358`	`358`	`/* Future optimization: Reuse the result tuple as we do in enumerate() */`
`359`		`- result = PyTuple_Pack(2, old, new);`
	`359`	`+ result = PyTuple_Pack2(old, new);`
`360`	`360`	`Py_XSETREF(po->old, new);`
`361`	`361`	`Py_DECREF(old);`
`362`	`362`	`return result;`