blob: d5394b4a31020cb980ef9799dcb6c76214f8f0fe [file] [log] [blame]
Fredrik Lundh58b5e842006-05-26 19:24:53 +00001/* stringlib: find/index implementation */
2
3#ifndef STRINGLIB_FIND_H
4#define STRINGLIB_FIND_H
5
6#ifndef STRINGLIB_FASTSEARCH_H
7#error must include "stringlib/fastsearch.h" before including this module
8#endif
9
10Py_LOCAL(Py_ssize_t)
11stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
Fredrik Lundhe6e43c82006-05-26 19:48:07 +000012 const STRINGLIB_CHAR* sub, Py_ssize_t sub_len,
13 Py_ssize_t offset)
Fredrik Lundh58b5e842006-05-26 19:24:53 +000014{
Fredrik Lundhe6e43c82006-05-26 19:48:07 +000015 Py_ssize_t pos;
Fredrik Lundh58b5e842006-05-26 19:24:53 +000016
Fredrik Lundhe6e43c82006-05-26 19:48:07 +000017 if (sub_len == 0)
18 return offset;
19
20 pos = fastsearch(str, str_len, sub, sub_len, FAST_SEARCH);
21
22 if (pos >= 0)
23 pos += offset;
24
25 return pos;
Fredrik Lundh58b5e842006-05-26 19:24:53 +000026}
27
28Py_LOCAL(Py_ssize_t)
29stringlib_rfind(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
Fredrik Lundhe6e43c82006-05-26 19:48:07 +000030 const STRINGLIB_CHAR* sub, Py_ssize_t sub_len,
31 Py_ssize_t offset)
Fredrik Lundh58b5e842006-05-26 19:24:53 +000032{
33 Py_ssize_t pos;
34
35 /* XXX - create reversefastsearch helper! */
36 if (sub_len == 0)
Fredrik Lundhe6e43c82006-05-26 19:48:07 +000037 pos = str_len + offset;
Fredrik Lundh58b5e842006-05-26 19:24:53 +000038 else {
39 Py_ssize_t j;
40 pos = -1;
41 for (j = str_len - sub_len; j >= 0; --j)
42 if (STRINGLIB_CMP(str+j, sub, sub_len) == 0) {
Fredrik Lundhe6e43c82006-05-26 19:48:07 +000043 pos = j + offset;
Fredrik Lundh58b5e842006-05-26 19:24:53 +000044 break;
45 }
46 }
47
48 return pos;
49}
50
51#endif
52
53/*
54Local variables:
55c-basic-offset: 4
56indent-tabs-mode: nil
57End:
58*/