| /* 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(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(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; |
| |
| if (sep_len == 0) { |
| PyErr_SetString(PyExc_ValueError, "empty separator"); |
| return NULL; |
| } |
| |
| out = PyTuple_New(3); |
| if (!out) |
| return NULL; |
| |
| /* XXX - create reversefastsearch helper! */ |
| if (sep_len == 0) |
| pos = str_len; |
| else { |
| Py_ssize_t j; |
| 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(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; |
| } |
| |
| #endif |
| |
| /* |
| Local variables: |
| c-basic-offset: 4 |
| indent-tabs-mode: nil |
| End: |
| */ |