ref: 1ccd0cd04d75840861e26f6c5f6ac73396d43d8a
dir: /sys/src/cmd/python/Objects/stringlib/partition.h/
/* stringlib: partition implementation */ #ifndef STRINGLIB_PARTITION_H #define STRINGLIB_PARTITION_H #ifndef STRINGLIB_FASTSEARCH_H #error must include "stringlib/fastsearch.h" before including this module #endif Py_LOCAL_INLINE(PyObject*) stringlib_partition( PyObject* str_obj, const STRINGLIB_CHAR* str, Py_ssize_t str_len, PyObject* sep_obj, const STRINGLIB_CHAR* sep, Py_ssize_t sep_len ) { PyObject* out; Py_ssize_t pos; if (sep_len == 0) { PyErr_SetString(PyExc_ValueError, "empty separator"); return NULL; } out = PyTuple_New(3); if (!out) return NULL; pos = fastsearch(str, str_len, sep, sep_len, FAST_SEARCH); if (pos < 0) { Py_INCREF(str_obj); PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj); Py_INCREF(STRINGLIB_EMPTY); PyTuple_SET_ITEM(out, 1, (PyObject*) STRINGLIB_EMPTY); Py_INCREF(STRINGLIB_EMPTY); PyTuple_SET_ITEM(out, 2, (PyObject*) STRINGLIB_EMPTY); return out; } PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos)); Py_INCREF(sep_obj); PyTuple_SET_ITEM(out, 1, sep_obj); pos += sep_len; PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos)); if (PyErr_Occurred()) { Py_DECREF(out); return NULL; } return out; } Py_LOCAL_INLINE(PyObject*) stringlib_rpartition( PyObject* str_obj, const STRINGLIB_CHAR* str, Py_ssize_t str_len, PyObject* sep_obj, const STRINGLIB_CHAR* sep, Py_ssize_t sep_len ) { PyObject* out; Py_ssize_t pos, j; if (sep_len == 0) { PyErr_SetString(PyExc_ValueError, "empty separator"); return NULL; } out = PyTuple_New(3); if (!out) return NULL; /* XXX - create reversefastsearch helper! */ pos = -1; for (j = str_len - sep_len; j >= 0; --j) if (STRINGLIB_CMP(str+j, sep, sep_len) == 0) { pos = j; break; } if (pos < 0) { Py_INCREF(STRINGLIB_EMPTY); PyTuple_SET_ITEM(out, 0, (PyObject*) STRINGLIB_EMPTY); Py_INCREF(STRINGLIB_EMPTY); PyTuple_SET_ITEM(out, 1, (PyObject*) STRINGLIB_EMPTY); Py_INCREF(str_obj); PyTuple_SET_ITEM(out, 2, (PyObject*) str_obj); return out; } PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos)); Py_INCREF(sep_obj); PyTuple_SET_ITEM(out, 1, sep_obj); pos += sep_len; PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos)); if (PyErr_Occurred()) { Py_DECREF(out); return NULL; } return out; } #endif /* Local variables: c-basic-offset: 4 indent-tabs-mode: nil End: */