Coverage Report

Created: 2025-07-18 06:09

/src/cpython/Objects/stringlib/partition.h
Line
Count
Source (jump to first uncovered line)
1
/* stringlib: partition implementation */
2
3
#ifndef STRINGLIB_FASTSEARCH_H
4
#  error must include "stringlib/fastsearch.h" before including this module
5
#endif
6
7
#if !STRINGLIB_MUTABLE && !defined(STRINGLIB_GET_EMPTY)
8
#  error "STRINGLIB_GET_EMPTY must be defined if STRINGLIB_MUTABLE is zero"
9
#endif
10
11
12
Py_LOCAL_INLINE(PyObject*)
13
STRINGLIB(partition)(PyObject* str_obj,
14
                    const STRINGLIB_CHAR* str, Py_ssize_t str_len,
15
                    PyObject* sep_obj,
16
                    const STRINGLIB_CHAR* sep, Py_ssize_t sep_len)
17
7.13M
{
18
7.13M
    PyObject* out;
19
7.13M
    Py_ssize_t pos;
20
21
7.13M
    if (sep_len == 0) {
22
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
23
0
        return NULL;
24
0
    }
25
26
7.13M
    out = PyTuple_New(3);
27
7.13M
    if (!out)
28
0
        return NULL;
29
30
7.13M
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_SEARCH);
31
32
7.13M
    if (pos < 0) {
33
#if STRINGLIB_MUTABLE
34
0
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, str_len));
35
0
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
36
0
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(NULL, 0));
37
38
0
        if (PyErr_Occurred()) {
39
0
            Py_DECREF(out);
40
0
            return NULL;
41
0
        }
42
#else
43
1.12M
        Py_INCREF(str_obj);
44
1.12M
        PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj);
45
1.12M
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
46
        assert(empty != NULL);
47
1.12M
        Py_INCREF(empty);
48
1.12M
        PyTuple_SET_ITEM(out, 1, empty);
49
1.12M
        Py_INCREF(empty);
50
1.12M
        PyTuple_SET_ITEM(out, 2, empty);
51
#endif
52
0
        return out;
53
1.12M
    }
54
55
6.00M
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
56
6.00M
    Py_INCREF(sep_obj);
57
6.00M
    PyTuple_SET_ITEM(out, 1, sep_obj);
58
6.00M
    pos += sep_len;
59
6.00M
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
60
61
6.00M
    if (PyErr_Occurred()) {
62
0
        Py_DECREF(out);
63
0
        return NULL;
64
0
    }
65
66
6.00M
    return out;
67
6.00M
}
Unexecuted instantiation: bytesobject.c:stringlib_partition
unicodeobject.c:asciilib_partition
Line
Count
Source
17
2.45M
{
18
2.45M
    PyObject* out;
19
2.45M
    Py_ssize_t pos;
20
21
2.45M
    if (sep_len == 0) {
22
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
23
0
        return NULL;
24
0
    }
25
26
2.45M
    out = PyTuple_New(3);
27
2.45M
    if (!out)
28
0
        return NULL;
29
30
2.45M
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_SEARCH);
31
32
2.45M
    if (pos < 0) {
33
#if STRINGLIB_MUTABLE
34
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, str_len));
35
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
36
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(NULL, 0));
37
38
        if (PyErr_Occurred()) {
39
            Py_DECREF(out);
40
            return NULL;
41
        }
42
#else
43
812k
        Py_INCREF(str_obj);
44
812k
        PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj);
45
812k
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
46
812k
        assert(empty != NULL);
47
812k
        Py_INCREF(empty);
48
812k
        PyTuple_SET_ITEM(out, 1, empty);
49
812k
        Py_INCREF(empty);
50
812k
        PyTuple_SET_ITEM(out, 2, empty);
51
812k
#endif
52
812k
        return out;
53
812k
    }
54
55
1.64M
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
56
1.64M
    Py_INCREF(sep_obj);
57
1.64M
    PyTuple_SET_ITEM(out, 1, sep_obj);
58
1.64M
    pos += sep_len;
59
1.64M
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
60
61
1.64M
    if (PyErr_Occurred()) {
62
0
        Py_DECREF(out);
63
0
        return NULL;
64
0
    }
65
66
1.64M
    return out;
67
1.64M
}
unicodeobject.c:ucs1lib_partition
Line
Count
Source
17
4.59M
{
18
4.59M
    PyObject* out;
19
4.59M
    Py_ssize_t pos;
20
21
4.59M
    if (sep_len == 0) {
22
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
23
0
        return NULL;
24
0
    }
25
26
4.59M
    out = PyTuple_New(3);
27
4.59M
    if (!out)
28
0
        return NULL;
29
30
4.59M
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_SEARCH);
31
32
4.59M
    if (pos < 0) {
33
#if STRINGLIB_MUTABLE
34
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, str_len));
35
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
36
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(NULL, 0));
37
38
        if (PyErr_Occurred()) {
39
            Py_DECREF(out);
40
            return NULL;
41
        }
42
#else
43
295k
        Py_INCREF(str_obj);
44
295k
        PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj);
45
295k
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
46
295k
        assert(empty != NULL);
47
295k
        Py_INCREF(empty);
48
295k
        PyTuple_SET_ITEM(out, 1, empty);
49
295k
        Py_INCREF(empty);
50
295k
        PyTuple_SET_ITEM(out, 2, empty);
51
295k
#endif
52
295k
        return out;
53
295k
    }
54
55
4.29M
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
56
4.29M
    Py_INCREF(sep_obj);
57
4.29M
    PyTuple_SET_ITEM(out, 1, sep_obj);
58
4.29M
    pos += sep_len;
59
4.29M
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
60
61
4.29M
    if (PyErr_Occurred()) {
62
0
        Py_DECREF(out);
63
0
        return NULL;
64
0
    }
65
66
4.29M
    return out;
67
4.29M
}
unicodeobject.c:ucs2lib_partition
Line
Count
Source
17
69.3k
{
18
69.3k
    PyObject* out;
19
69.3k
    Py_ssize_t pos;
20
21
69.3k
    if (sep_len == 0) {
22
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
23
0
        return NULL;
24
0
    }
25
26
69.3k
    out = PyTuple_New(3);
27
69.3k
    if (!out)
28
0
        return NULL;
29
30
69.3k
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_SEARCH);
31
32
69.3k
    if (pos < 0) {
33
#if STRINGLIB_MUTABLE
34
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, str_len));
35
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
36
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(NULL, 0));
37
38
        if (PyErr_Occurred()) {
39
            Py_DECREF(out);
40
            return NULL;
41
        }
42
#else
43
17.4k
        Py_INCREF(str_obj);
44
17.4k
        PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj);
45
17.4k
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
46
17.4k
        assert(empty != NULL);
47
17.4k
        Py_INCREF(empty);
48
17.4k
        PyTuple_SET_ITEM(out, 1, empty);
49
17.4k
        Py_INCREF(empty);
50
17.4k
        PyTuple_SET_ITEM(out, 2, empty);
51
17.4k
#endif
52
17.4k
        return out;
53
17.4k
    }
54
55
51.8k
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
56
51.8k
    Py_INCREF(sep_obj);
57
51.8k
    PyTuple_SET_ITEM(out, 1, sep_obj);
58
51.8k
    pos += sep_len;
59
51.8k
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
60
61
51.8k
    if (PyErr_Occurred()) {
62
0
        Py_DECREF(out);
63
0
        return NULL;
64
0
    }
65
66
51.8k
    return out;
67
51.8k
}
unicodeobject.c:ucs4lib_partition
Line
Count
Source
17
8.58k
{
18
8.58k
    PyObject* out;
19
8.58k
    Py_ssize_t pos;
20
21
8.58k
    if (sep_len == 0) {
22
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
23
0
        return NULL;
24
0
    }
25
26
8.58k
    out = PyTuple_New(3);
27
8.58k
    if (!out)
28
0
        return NULL;
29
30
8.58k
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_SEARCH);
31
32
8.58k
    if (pos < 0) {
33
#if STRINGLIB_MUTABLE
34
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, str_len));
35
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
36
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(NULL, 0));
37
38
        if (PyErr_Occurred()) {
39
            Py_DECREF(out);
40
            return NULL;
41
        }
42
#else
43
3.35k
        Py_INCREF(str_obj);
44
3.35k
        PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj);
45
3.35k
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
46
3.35k
        assert(empty != NULL);
47
3.35k
        Py_INCREF(empty);
48
3.35k
        PyTuple_SET_ITEM(out, 1, empty);
49
3.35k
        Py_INCREF(empty);
50
3.35k
        PyTuple_SET_ITEM(out, 2, empty);
51
3.35k
#endif
52
3.35k
        return out;
53
3.35k
    }
54
55
5.22k
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
56
5.22k
    Py_INCREF(sep_obj);
57
5.22k
    PyTuple_SET_ITEM(out, 1, sep_obj);
58
5.22k
    pos += sep_len;
59
5.22k
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
60
61
5.22k
    if (PyErr_Occurred()) {
62
0
        Py_DECREF(out);
63
0
        return NULL;
64
0
    }
65
66
5.22k
    return out;
67
5.22k
}
Unexecuted instantiation: bytearrayobject.c:stringlib_partition
68
69
Py_LOCAL_INLINE(PyObject*)
70
STRINGLIB(rpartition)(PyObject* str_obj,
71
                     const STRINGLIB_CHAR* str, Py_ssize_t str_len,
72
                     PyObject* sep_obj,
73
                     const STRINGLIB_CHAR* sep, Py_ssize_t sep_len)
74
9.64k
{
75
9.64k
    PyObject* out;
76
9.64k
    Py_ssize_t pos;
77
78
9.64k
    if (sep_len == 0) {
79
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
80
0
        return NULL;
81
0
    }
82
83
9.64k
    out = PyTuple_New(3);
84
9.64k
    if (!out)
85
0
        return NULL;
86
87
9.64k
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_RSEARCH);
88
89
9.64k
    if (pos < 0) {
90
#if STRINGLIB_MUTABLE
91
0
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(NULL, 0));
92
0
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
93
0
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str, str_len));
94
95
0
        if (PyErr_Occurred()) {
96
0
            Py_DECREF(out);
97
0
            return NULL;
98
0
        }
99
#else
100
2.08k
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
101
        assert(empty != NULL);
102
2.08k
        Py_INCREF(empty);
103
2.08k
        PyTuple_SET_ITEM(out, 0, empty);
104
2.08k
        Py_INCREF(empty);
105
2.08k
        PyTuple_SET_ITEM(out, 1, empty);
106
2.08k
        Py_INCREF(str_obj);
107
2.08k
        PyTuple_SET_ITEM(out, 2, (PyObject*) str_obj);
108
#endif
109
0
        return out;
110
2.08k
    }
111
112
7.56k
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
113
7.56k
    Py_INCREF(sep_obj);
114
7.56k
    PyTuple_SET_ITEM(out, 1, sep_obj);
115
7.56k
    pos += sep_len;
116
7.56k
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
117
118
7.56k
    if (PyErr_Occurred()) {
119
0
        Py_DECREF(out);
120
0
        return NULL;
121
0
    }
122
123
7.56k
    return out;
124
7.56k
}
Unexecuted instantiation: bytesobject.c:stringlib_rpartition
unicodeobject.c:asciilib_rpartition
Line
Count
Source
74
9.64k
{
75
9.64k
    PyObject* out;
76
9.64k
    Py_ssize_t pos;
77
78
9.64k
    if (sep_len == 0) {
79
0
        PyErr_SetString(PyExc_ValueError, "empty separator");
80
0
        return NULL;
81
0
    }
82
83
9.64k
    out = PyTuple_New(3);
84
9.64k
    if (!out)
85
0
        return NULL;
86
87
9.64k
    pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_RSEARCH);
88
89
9.64k
    if (pos < 0) {
90
#if STRINGLIB_MUTABLE
91
        PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(NULL, 0));
92
        PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
93
        PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str, str_len));
94
95
        if (PyErr_Occurred()) {
96
            Py_DECREF(out);
97
            return NULL;
98
        }
99
#else
100
2.08k
        PyObject *empty = (PyObject*)STRINGLIB_GET_EMPTY();
101
2.08k
        assert(empty != NULL);
102
2.08k
        Py_INCREF(empty);
103
2.08k
        PyTuple_SET_ITEM(out, 0, empty);
104
2.08k
        Py_INCREF(empty);
105
2.08k
        PyTuple_SET_ITEM(out, 1, empty);
106
2.08k
        Py_INCREF(str_obj);
107
2.08k
        PyTuple_SET_ITEM(out, 2, (PyObject*) str_obj);
108
2.08k
#endif
109
2.08k
        return out;
110
2.08k
    }
111
112
7.56k
    PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
113
7.56k
    Py_INCREF(sep_obj);
114
7.56k
    PyTuple_SET_ITEM(out, 1, sep_obj);
115
7.56k
    pos += sep_len;
116
7.56k
    PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
117
118
7.56k
    if (PyErr_Occurred()) {
119
0
        Py_DECREF(out);
120
0
        return NULL;
121
0
    }
122
123
7.56k
    return out;
124
7.56k
}
Unexecuted instantiation: unicodeobject.c:ucs1lib_rpartition
Unexecuted instantiation: unicodeobject.c:ucs2lib_rpartition
Unexecuted instantiation: unicodeobject.c:ucs4lib_rpartition
Unexecuted instantiation: bytearrayobject.c:stringlib_rpartition
125