util/LocaleString: add IsIncompleteCharMB()
[ncmpc-debian.git] / src / util / LocaleString.cxx
1 /*
2  * Copyright 2018 Max Kellermann <max.kellermann@gmail.com>
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * - Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *
11  * - Redistributions in binary form must reproduce the above copyright
12  * notice, this list of conditions and the following disclaimer in the
13  * documentation and/or other materials provided with the
14  * distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
19  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
20  * FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
23  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
25  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
27  * OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30 #include "LocaleString.hxx"
31
32 #include <cwchar>
33
34 #include <string.h>
35
36 bool
37 IsIncompleteCharMB(const char *s, size_t n)
38 {
39         auto mb = std::mbstate_t();
40         const std::size_t length = std::mbrlen(s, n, &mb);
41         return length == std::size_t(-2);
42 }
43
44 std::size_t
45 CharSizeMB(const char *s, size_t n)
46 {
47         auto mb = std::mbstate_t();
48         const std::size_t length = std::mbrlen(s, n, &mb);
49         if (length == std::size_t(-2))
50                 return n;
51
52         if (length == std::size_t(-1))
53                 return 1;
54
55         return length;
56 }
57
58 const char *
59 PrevCharMB(const char *start, const char *reference)
60 {
61         const char *p = reference;
62
63         while (p > start) {
64                 --p;
65
66                 auto mb = std::mbstate_t();
67                 const std::size_t length = std::mbrlen(p, reference - p, &mb);
68                 if (length != std::size_t(-1))
69                         break;
70         }
71
72         return p;
73 }
74
75 size_t
76 StringWidthMB(const char *s, size_t length)
77 {
78         const char *const end = s + length;
79         auto state = std::mbstate_t();
80
81         size_t width = 0;
82         while (s < end) {
83                 wchar_t w;
84                 std::size_t n = std::mbrtowc(&w, s, end - s, &state);
85                 if (n == std::size_t(-2))
86                         break;
87
88                 if (n == std::size_t(-1) || n == 0) {
89                         ++s;
90                 } else {
91                         s += n;
92                         int cw = wcwidth(w);
93                         if (cw > 0)
94                                 width += cw;
95                 }
96         }
97
98         return width;
99 }
100
101 size_t
102 StringWidthMB(const char *s)
103 {
104         return StringWidthMB(s, strlen(s));
105 }