find.h

/* stringlib: find/index implementation */

#ifndefSTRINGLIB_FIND_H
#defineSTRINGLIB_FIND_H

#ifndefSTRINGLIB_FASTSEARCH_H
#error must include "stringlib/fastsearch.h" before including this module
#endif

Py_LOCAL_INLINE(Py_ssize_t)
stringlib_find(constSTRINGLIB_CHAR*str, Py_ssize_tstr_len,
constSTRINGLIB_CHAR*sub, Py_ssize_tsub_len,
Py_ssize_toffset)
{
Py_ssize_tpos;

if (str_len<0)
return-1;
if (sub_len==0)
returnoffset;

pos=fastsearch(str, str_len, sub, sub_len, -1, FAST_SEARCH);

if (pos >= 0)
pos+=offset;

returnpos;
}

Py_LOCAL_INLINE(Py_ssize_t)
stringlib_rfind(constSTRINGLIB_CHAR*str, Py_ssize_tstr_len,
constSTRINGLIB_CHAR*sub, Py_ssize_tsub_len,
Py_ssize_toffset)
{
Py_ssize_tpos;

if (str_len<0)
return-1;
if (sub_len==0)
returnstr_len+offset;

pos=fastsearch(str, str_len, sub, sub_len, -1, FAST_RSEARCH);

if (pos >= 0)
pos+=offset;

returnpos;
}

/* helper macro to fixup start/end slice values */
#defineADJUST_INDICES(start, end, len) \
 if (end > len) \
 end = len; \
 else if (end < 0) { \
 end += len; \
 if (end < 0) \
 end = 0; \
 } \
 if (start < 0) { \
 start += len; \
 if (start < 0) \
 start = 0; \
 }

Py_LOCAL_INLINE(Py_ssize_t)
stringlib_find_slice(constSTRINGLIB_CHAR*str, Py_ssize_tstr_len,
constSTRINGLIB_CHAR*sub, Py_ssize_tsub_len,
Py_ssize_tstart, Py_ssize_tend)
{
ADJUST_INDICES(start, end, str_len);
returnstringlib_find(str+start, end-start, sub, sub_len, start);
}

Py_LOCAL_INLINE(Py_ssize_t)
stringlib_rfind_slice(constSTRINGLIB_CHAR*str, Py_ssize_tstr_len,
constSTRINGLIB_CHAR*sub, Py_ssize_tsub_len,
Py_ssize_tstart, Py_ssize_tend)
{
ADJUST_INDICES(start, end, str_len);
returnstringlib_rfind(str+start, end-start, sub, sub_len, start);
}

#ifdefSTRINGLIB_WANT_CONTAINS_OBJ

Py_LOCAL_INLINE(int)
stringlib_contains_obj(PyObject*str, PyObject*sub)
{
returnstringlib_find(
STRINGLIB_STR(str), STRINGLIB_LEN(str),
STRINGLIB_STR(sub), STRINGLIB_LEN(sub), 0
 ) !=-1;
}

#endif/* STRINGLIB_WANT_CONTAINS_OBJ */

/*
This function is a helper for the "find" family (find, rfind, index,
rindex) and for count, startswith and endswith, because they all have
the same behaviour for the arguments.

It does not touch the variables received until it knows everything 
is ok.
*/

#defineFORMAT_BUFFER_SIZE 50

Py_LOCAL_INLINE(int)
stringlib_parse_args_finds(constchar*function_name, PyObject*args,
PyObject**subobj,
Py_ssize_t*start, Py_ssize_t*end)
{
PyObject*tmp_subobj;
Py_ssize_ttmp_start=0;
Py_ssize_ttmp_end=PY_SSIZE_T_MAX;
PyObject*obj_start=Py_None, *obj_end=Py_None;
charformat[FORMAT_BUFFER_SIZE] ="O|OO:";
size_tlen=strlen(format);

strncpy(format+len, function_name, FORMAT_BUFFER_SIZE-len-1);
format[FORMAT_BUFFER_SIZE-1] ='\0';

if (!PyArg_ParseTuple(args, format, &tmp_subobj, &obj_start, &obj_end))
return0;

/* To support None in "start" and "end" arguments, meaning
 the same as if they were not passed.
 */
if (obj_start!=Py_None)
if (!_PyEval_SliceIndex(obj_start, &tmp_start))
return0;
if (obj_end!=Py_None)
if (!_PyEval_SliceIndex(obj_end, &tmp_end))
return0;

*start=tmp_start;
*end=tmp_end;
*subobj=tmp_subobj;
return1;
}

#undef FORMAT_BUFFER_SIZE

#ifSTRINGLIB_IS_UNICODE

/*
Wraps stringlib_parse_args_finds() and additionally ensures that the
first argument is a unicode object.

Note that we receive a pointer to the pointer of the substring object,
so when we create that object in this function we don't DECREF it,
because it continues living in the caller functions (those functions, 
after finishing using the substring, must DECREF it).
*/

Py_LOCAL_INLINE(int)
stringlib_parse_args_finds_unicode(constchar*function_name, PyObject*args,
PyUnicodeObject**substring,
Py_ssize_t*start, Py_ssize_t*end)
{
PyObject*tmp_substring;

if(stringlib_parse_args_finds(function_name, args, &tmp_substring,
start, end)) {
tmp_substring=PyUnicode_FromObject(tmp_substring);
if (!tmp_substring)
return0;
*substring= (PyUnicodeObject*)tmp_substring;
return1;
 }
return0;
}

#endif/* STRINGLIB_IS_UNICODE */

#endif/* STRINGLIB_FIND_H */