WIP: new mypyc primitives for str.count

BobTheBuidler · web-flow · commit 4a9f0068b37b · 2025-06-10T01:05:03.000-04:00
diff --git a/mypyc/lib-rt/str_ops.c b/mypyc/lib-rt/str_ops.c
@@ -511,6 +511,10 @@ PyObject *CPy_Encode(PyObject *obj, PyObject *encoding, PyObject *errors) {
     }
 }
 
+Py_ssize_t CPyStr_CountAll(PyObject *unicode, PyObject *substring) {
+    Py_ssize_t end = PyUnicode_GET_LENGTH(self);
+    return PyUnicode_Count(unicode, substring, 0, end);
+
 
 CPyTagged CPyStr_Ord(PyObject *obj) {
     Py_ssize_t s = PyUnicode_GET_LENGTH(obj);

Original file line number	Diff line number	Diff line change
`@@ -511,6 +511,10 @@ PyObject CPy_Encode(PyObject obj, PyObject encoding, PyObject errors) {`
`511`	`511`	`}`
`512`	`512`	`}`
`513`	`513`
	`514`	`+Py_ssize_t CPyStr_CountAll(PyObject unicode, PyObject substring) {`
	`515`	`+ Py_ssize_t end = PyUnicode_GET_LENGTH(self);`
	`516`	`+ return PyUnicode_Count(unicode, substring, 0, end);`
	`517`	`+`
`514`	`518`
`515`	`519`	`CPyTagged CPyStr_Ord(PyObject *obj) {`
`516`	`520`	`Py_ssize_t s = PyUnicode_GET_LENGTH(obj);`