blob: 45b16aa3ab7a9584381919a3d5c0a52e8e90a311 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10#include "vim.h"
11
Bram Moolenaar13505972019-01-24 15:04:48 +010012#if defined(HAVE_WCHAR_H)
Bram Moolenaarc667da52019-11-30 20:52:27 +010013# include <wchar.h> // for towupper() and towlower()
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#endif
15
Bram Moolenaar7f9969c2022-07-25 18:13:54 +010016static int win_nolbr_chartabsize(chartabsize_T *cts, int *headp);
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010017static unsigned nr2hex(unsigned c);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
19static int chartab_initialized = FALSE;
20
Bram Moolenaarc667da52019-11-30 20:52:27 +010021// b_chartab[] is an array of 32 bytes, each bit representing one of the
22// characters 0-255.
Bram Moolenaar071d4272004-06-13 20:20:40 +000023#define SET_CHARTAB(buf, c) (buf)->b_chartab[(unsigned)(c) >> 3] |= (1 << ((c) & 0x7))
24#define RESET_CHARTAB(buf, c) (buf)->b_chartab[(unsigned)(c) >> 3] &= ~(1 << ((c) & 0x7))
25#define GET_CHARTAB(buf, c) ((buf)->b_chartab[(unsigned)(c) >> 3] & (1 << ((c) & 0x7)))
26
Bram Moolenaarc667da52019-11-30 20:52:27 +010027// table used below, see init_chartab() for an explanation
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010028static char_u g_chartab[256];
29
Bram Moolenaar071d4272004-06-13 20:20:40 +000030/*
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010031 * Flags for g_chartab[].
32 */
Bram Moolenaarc667da52019-11-30 20:52:27 +010033#define CT_CELL_MASK 0x07 // mask: nr of display cells (1, 2 or 4)
34#define CT_PRINT_CHAR 0x10 // flag: set for printable chars
35#define CT_ID_CHAR 0x20 // flag: set for ID chars
36#define CT_FNAME_CHAR 0x40 // flag: set for file name chars
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010037
Bram Moolenaar5843f5f2019-08-20 20:13:45 +020038static int in_win_border(win_T *wp, colnr_T vcol);
39
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010040/*
41 * Fill g_chartab[]. Also fills curbuf->b_chartab[] with flags for keyword
Bram Moolenaar071d4272004-06-13 20:20:40 +000042 * characters for current buffer.
43 *
44 * Depends on the option settings 'iskeyword', 'isident', 'isfname',
45 * 'isprint' and 'encoding'.
46 *
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010047 * The index in g_chartab[] depends on 'encoding':
Bram Moolenaar071d4272004-06-13 20:20:40 +000048 * - For non-multi-byte index with the byte (same as the character).
49 * - For DBCS index with the first byte.
50 * - For UTF-8 index with the character (when first byte is up to 0x80 it is
51 * the same as the character, if the first byte is 0x80 and above it depends
52 * on further bytes).
53 *
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010054 * The contents of g_chartab[]:
Bram Moolenaar071d4272004-06-13 20:20:40 +000055 * - The lower two bits, masked by CT_CELL_MASK, give the number of display
56 * cells the character occupies (1 or 2). Not valid for UTF-8 above 0x80.
57 * - CT_PRINT_CHAR bit is set when the character is printable (no need to
58 * translate the character before displaying it). Note that only DBCS
59 * characters can have 2 display cells and still be printable.
60 * - CT_FNAME_CHAR bit is set when the character can be in a file name.
61 * - CT_ID_CHAR bit is set when the character can be in an identifier.
62 *
63 * Return FAIL if 'iskeyword', 'isident', 'isfname' or 'isprint' option has an
64 * error, OK otherwise.
65 */
66 int
Bram Moolenaar7454a062016-01-30 15:14:10 +010067init_chartab(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +000068{
69 return buf_init_chartab(curbuf, TRUE);
70}
71
72 int
Bram Moolenaar7454a062016-01-30 15:14:10 +010073buf_init_chartab(
74 buf_T *buf,
Bram Moolenaarc667da52019-11-30 20:52:27 +010075 int global) // FALSE: only set buf->b_chartab[]
Bram Moolenaar071d4272004-06-13 20:20:40 +000076{
77 int c;
78 int c2;
79 char_u *p;
80 int i;
81 int tilde;
82 int do_isalpha;
83
84 if (global)
85 {
86 /*
87 * Set the default size for printable characters:
88 * From <Space> to '~' is 1 (printable), others are 2 (not printable).
89 * This also inits all 'isident' and 'isfname' flags to FALSE.
Bram Moolenaar071d4272004-06-13 20:20:40 +000090 */
91 c = 0;
92 while (c < ' ')
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010093 g_chartab[c++] = (dy_flags & DY_UHEX) ? 4 : 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 while (c <= '~')
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +010095 g_chartab[c++] = 1 + CT_PRINT_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +000096 while (c < 256)
97 {
Bram Moolenaarc667da52019-11-30 20:52:27 +010098 // UTF-8: bytes 0xa0 - 0xff are printable (latin1)
Bram Moolenaar071d4272004-06-13 20:20:40 +000099 if (enc_utf8 && c >= 0xa0)
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100100 g_chartab[c++] = CT_PRINT_CHAR + 1;
Bram Moolenaarc667da52019-11-30 20:52:27 +0100101 // euc-jp characters starting with 0x8e are single width
Bram Moolenaar071d4272004-06-13 20:20:40 +0000102 else if (enc_dbcs == DBCS_JPNU && c == 0x8e)
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100103 g_chartab[c++] = CT_PRINT_CHAR + 1;
Bram Moolenaarc667da52019-11-30 20:52:27 +0100104 // other double-byte chars can be printable AND double-width
Bram Moolenaar071d4272004-06-13 20:20:40 +0000105 else if (enc_dbcs != 0 && MB_BYTE2LEN(c) == 2)
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100106 g_chartab[c++] = CT_PRINT_CHAR + 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000107 else
Bram Moolenaarc667da52019-11-30 20:52:27 +0100108 // the rest is unprintable by default
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100109 g_chartab[c++] = (dy_flags & DY_UHEX) ? 4 : 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000110 }
111
Bram Moolenaarc667da52019-11-30 20:52:27 +0100112 // Assume that every multi-byte char is a filename character.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000113 for (c = 1; c < 256; ++c)
114 if ((enc_dbcs != 0 && MB_BYTE2LEN(c) > 1)
115 || (enc_dbcs == DBCS_JPNU && c == 0x8e)
116 || (enc_utf8 && c >= 0xa0))
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100117 g_chartab[c] |= CT_FNAME_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000118 }
119
120 /*
121 * Init word char flags all to FALSE
122 */
Bram Moolenaara80faa82020-04-12 19:37:17 +0200123 CLEAR_FIELD(buf->b_chartab);
Bram Moolenaar6bb68362005-03-22 23:03:44 +0000124 if (enc_dbcs != 0)
125 for (c = 0; c < 256; ++c)
126 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100127 // double-byte characters are probably word characters
Bram Moolenaar6bb68362005-03-22 23:03:44 +0000128 if (MB_BYTE2LEN(c) == 2)
129 SET_CHARTAB(buf, c);
130 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000131
Bram Moolenaar071d4272004-06-13 20:20:40 +0000132 /*
133 * In lisp mode the '-' character is included in keywords.
134 */
135 if (buf->b_p_lisp)
136 SET_CHARTAB(buf, '-');
Bram Moolenaar071d4272004-06-13 20:20:40 +0000137
Bram Moolenaarc667da52019-11-30 20:52:27 +0100138 // Walk through the 'isident', 'iskeyword', 'isfname' and 'isprint'
139 // options Each option is a list of characters, character numbers or
140 // ranges, separated by commas, e.g.: "200-210,x,#-178,-"
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 for (i = global ? 0 : 3; i <= 3; ++i)
142 {
143 if (i == 0)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100144 p = p_isi; // first round: 'isident'
Bram Moolenaar071d4272004-06-13 20:20:40 +0000145 else if (i == 1)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100146 p = p_isp; // second round: 'isprint'
Bram Moolenaar071d4272004-06-13 20:20:40 +0000147 else if (i == 2)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100148 p = p_isf; // third round: 'isfname'
149 else // i == 3
150 p = buf->b_p_isk; // fourth round: 'iskeyword'
Bram Moolenaar071d4272004-06-13 20:20:40 +0000151
152 while (*p)
153 {
154 tilde = FALSE;
155 do_isalpha = FALSE;
156 if (*p == '^' && p[1] != NUL)
157 {
158 tilde = TRUE;
159 ++p;
160 }
161 if (VIM_ISDIGIT(*p))
162 c = getdigits(&p);
Dominique Pelle4781d6f2021-05-18 21:46:31 +0200163 else if (has_mbyte)
Bram Moolenaar183bb3e2009-09-11 12:02:34 +0000164 c = mb_ptr2char_adv(&p);
165 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166 c = *p++;
167 c2 = -1;
168 if (*p == '-' && p[1] != NUL)
169 {
170 ++p;
171 if (VIM_ISDIGIT(*p))
172 c2 = getdigits(&p);
Dominique Pelle4781d6f2021-05-18 21:46:31 +0200173 else if (has_mbyte)
Bram Moolenaar2ac5e602009-11-03 15:04:20 +0000174 c2 = mb_ptr2char_adv(&p);
175 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176 c2 = *p++;
177 }
Bram Moolenaar2ac5e602009-11-03 15:04:20 +0000178 if (c <= 0 || c >= 256 || (c2 < c && c2 != -1) || c2 >= 256
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 || !(*p == NUL || *p == ','))
180 return FAIL;
181
Bram Moolenaarc667da52019-11-30 20:52:27 +0100182 if (c2 == -1) // not a range
Bram Moolenaar071d4272004-06-13 20:20:40 +0000183 {
184 /*
185 * A single '@' (not "@-@"):
186 * Decide on letters being ID/printable/keyword chars with
187 * standard function isalpha(). This takes care of locale for
188 * single-byte characters).
189 */
190 if (c == '@')
191 {
192 do_isalpha = TRUE;
193 c = 1;
194 c2 = 255;
195 }
196 else
197 c2 = c;
198 }
199 while (c <= c2)
200 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100201 // Use the MB_ functions here, because isalpha() doesn't
202 // work properly when 'encoding' is "latin1" and the locale is
203 // "C".
Bram Moolenaar14184a32019-02-16 15:10:30 +0100204 if (!do_isalpha || MB_ISLOWER(c) || MB_ISUPPER(c))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000205 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100206 if (i == 0) // (re)set ID flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000207 {
208 if (tilde)
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100209 g_chartab[c] &= ~CT_ID_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 else
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100211 g_chartab[c] |= CT_ID_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000212 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100213 else if (i == 1) // (re)set printable
Bram Moolenaar071d4272004-06-13 20:20:40 +0000214 {
Bram Moolenaar424bcae2022-01-31 14:59:41 +0000215 if ((c < ' ' || c > '~'
Bram Moolenaar13505972019-01-24 15:04:48 +0100216 // For double-byte we keep the cell width, so
217 // that we can detect it from the first byte.
218 ) && !(enc_dbcs && MB_BYTE2LEN(c) == 2))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000219 {
220 if (tilde)
221 {
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100222 g_chartab[c] = (g_chartab[c] & ~CT_CELL_MASK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000223 + ((dy_flags & DY_UHEX) ? 4 : 2);
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100224 g_chartab[c] &= ~CT_PRINT_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000225 }
226 else
227 {
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100228 g_chartab[c] = (g_chartab[c] & ~CT_CELL_MASK)
229 + 1;
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100230 g_chartab[c] |= CT_PRINT_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000231 }
232 }
233 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100234 else if (i == 2) // (re)set fname flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000235 {
236 if (tilde)
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100237 g_chartab[c] &= ~CT_FNAME_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000238 else
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100239 g_chartab[c] |= CT_FNAME_CHAR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000240 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100241 else // i == 3 (re)set keyword flag
Bram Moolenaar071d4272004-06-13 20:20:40 +0000242 {
243 if (tilde)
244 RESET_CHARTAB(buf, c);
245 else
246 SET_CHARTAB(buf, c);
247 }
248 }
249 ++c;
250 }
Bram Moolenaar309379f2013-02-06 16:26:26 +0100251
252 c = *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000253 p = skip_to_option_part(p);
Bram Moolenaar309379f2013-02-06 16:26:26 +0100254 if (c == ',' && *p == NUL)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100255 // Trailing comma is not allowed.
Bram Moolenaar309379f2013-02-06 16:26:26 +0100256 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000257 }
258 }
259 chartab_initialized = TRUE;
260 return OK;
261}
262
263/*
264 * Translate any special characters in buf[bufsize] in-place.
265 * The result is a string with only printable characters, but if there is not
266 * enough room, not all characters will be translated.
267 */
268 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100269trans_characters(
270 char_u *buf,
271 int bufsize)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000272{
Bram Moolenaarc667da52019-11-30 20:52:27 +0100273 int len; // length of string needing translation
274 int room; // room in buffer after string
275 char_u *trs; // translated character
276 int trs_len; // length of trs[]
Bram Moolenaar071d4272004-06-13 20:20:40 +0000277
278 len = (int)STRLEN(buf);
279 room = bufsize - len;
280 while (*buf != 0)
281 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100282 // Assume a multi-byte character doesn't need translation.
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000283 if (has_mbyte && (trs_len = (*mb_ptr2len)(buf)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000284 len -= trs_len;
285 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000286 {
287 trs = transchar_byte(*buf);
288 trs_len = (int)STRLEN(trs);
289 if (trs_len > 1)
290 {
291 room -= trs_len - 1;
292 if (room <= 0)
293 return;
294 mch_memmove(buf + trs_len, buf + 1, (size_t)len);
295 }
296 mch_memmove(buf, trs, (size_t)trs_len);
297 --len;
298 }
299 buf += trs_len;
300 }
301}
302
Bram Moolenaar071d4272004-06-13 20:20:40 +0000303/*
304 * Translate a string into allocated memory, replacing special chars with
305 * printable chars. Returns NULL when out of memory.
306 */
307 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100308transstr(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000309{
310 char_u *res;
311 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000312 int l, len, c;
313 char_u hexbuf[11];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000314
Bram Moolenaar071d4272004-06-13 20:20:40 +0000315 if (has_mbyte)
316 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100317 // Compute the length of the result, taking account of unprintable
318 // multi-byte characters.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000319 len = 0;
320 p = s;
321 while (*p != NUL)
322 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000323 if ((l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000324 {
325 c = (*mb_ptr2char)(p);
326 p += l;
327 if (vim_isprintc(c))
328 len += l;
329 else
330 {
331 transchar_hex(hexbuf, c);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +0000332 len += (int)STRLEN(hexbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000333 }
334 }
335 else
336 {
337 l = byte2cells(*p++);
338 if (l > 0)
339 len += l;
340 else
Bram Moolenaarc667da52019-11-30 20:52:27 +0100341 len += 4; // illegal byte sequence
Bram Moolenaar071d4272004-06-13 20:20:40 +0000342 }
343 }
Bram Moolenaar964b3742019-05-24 18:54:09 +0200344 res = alloc(len + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000345 }
346 else
Bram Moolenaar964b3742019-05-24 18:54:09 +0200347 res = alloc(vim_strsize(s) + 1);
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000348
349 if (res == NULL)
350 return NULL;
351
352 *res = NUL;
353 p = s;
354 while (*p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000355 {
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000356 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000357 {
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000358 c = (*mb_ptr2char)(p);
359 if (vim_isprintc(c))
360 STRNCAT(res, p, l); // append printable multi-byte char
Bram Moolenaar071d4272004-06-13 20:20:40 +0000361 else
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000362 transchar_hex(res + STRLEN(res), c);
363 p += l;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000364 }
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000365 else
366 STRCAT(res, transchar_byte(*p++));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000367 }
368 return res;
369}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000370
Bram Moolenaar071d4272004-06-13 20:20:40 +0000371/*
Bram Moolenaar217ad922005-03-20 22:37:15 +0000372 * Convert the string "str[orglen]" to do ignore-case comparing. Uses the
373 * current locale.
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000374 * When "buf" is NULL returns an allocated string (NULL for out-of-memory).
375 * Otherwise puts the result in "buf[buflen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000376 */
377 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100378str_foldcase(
379 char_u *str,
380 int orglen,
381 char_u *buf,
382 int buflen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000383{
384 garray_T ga;
385 int i;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000386 int len = orglen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000387
388#define GA_CHAR(i) ((char_u *)ga.ga_data)[i]
kylo252ae6f1d82022-02-16 19:24:07 +0000389#define GA_PTR(i) ((char_u *)ga.ga_data + (i))
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000390#define STR_CHAR(i) (buf == NULL ? GA_CHAR(i) : buf[i])
kylo252ae6f1d82022-02-16 19:24:07 +0000391#define STR_PTR(i) (buf == NULL ? GA_PTR(i) : buf + (i))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000392
Bram Moolenaarc667da52019-11-30 20:52:27 +0100393 // Copy "str" into "buf" or allocated memory, unmodified.
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000394 if (buf == NULL)
395 {
396 ga_init2(&ga, 1, 10);
397 if (ga_grow(&ga, len + 1) == FAIL)
398 return NULL;
399 mch_memmove(ga.ga_data, str, (size_t)len);
400 ga.ga_len = len;
401 }
402 else
403 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100404 if (len >= buflen) // Ugly!
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000405 len = buflen - 1;
406 mch_memmove(buf, str, (size_t)len);
407 }
408 if (buf == NULL)
409 GA_CHAR(len) = NUL;
410 else
411 buf[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412
Bram Moolenaarc667da52019-11-30 20:52:27 +0100413 // Make each character lower case.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000414 i = 0;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000415 while (STR_CHAR(i) != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000416 {
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000417 if (enc_utf8 || (has_mbyte && MB_BYTE2LEN(STR_CHAR(i)) > 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418 {
419 if (enc_utf8)
420 {
Bram Moolenaarb9839212008-06-28 11:03:50 +0000421 int c = utf_ptr2char(STR_PTR(i));
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100422 int olen = utf_ptr2len(STR_PTR(i));
Bram Moolenaarb9839212008-06-28 11:03:50 +0000423 int lc = utf_tolower(c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000424
Bram Moolenaarc667da52019-11-30 20:52:27 +0100425 // Only replace the character when it is not an invalid
426 // sequence (ASCII character or more than one byte) and
427 // utf_tolower() doesn't return the original character.
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100428 if ((c < 0x80 || olen > 1) && c != lc)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000429 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100430 int nlen = utf_char2len(lc);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000431
Bram Moolenaarc667da52019-11-30 20:52:27 +0100432 // If the byte length changes need to shift the following
433 // characters forward or backward.
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100434 if (olen != nlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000435 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100436 if (nlen > olen)
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000437 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100438 if (buf == NULL
439 ? ga_grow(&ga, nlen - olen + 1) == FAIL
440 : len + nlen - olen >= buflen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000441 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100442 // out of memory, keep old char
Bram Moolenaar071d4272004-06-13 20:20:40 +0000443 lc = c;
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100444 nlen = olen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000445 }
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000446 }
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100447 if (olen != nlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000448 {
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000449 if (buf == NULL)
450 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100451 STRMOVE(GA_PTR(i) + nlen, GA_PTR(i) + olen);
452 ga.ga_len += nlen - olen;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000453 }
454 else
455 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100456 STRMOVE(buf + i + nlen, buf + i + olen);
457 len += nlen - olen;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000458 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000459 }
460 }
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000461 (void)utf_char2bytes(lc, STR_PTR(i));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000462 }
463 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100464 // skip to next multi-byte char
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000465 i += (*mb_ptr2len)(STR_PTR(i));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000466 }
467 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000468 {
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000469 if (buf == NULL)
470 GA_CHAR(i) = TOLOWER_LOC(GA_CHAR(i));
471 else
472 buf[i] = TOLOWER_LOC(buf[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000473 ++i;
474 }
475 }
476
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000477 if (buf == NULL)
478 return (char_u *)ga.ga_data;
479 return buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000480}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000481
482/*
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100483 * Catch 22: g_chartab[] can't be initialized before the options are
Bram Moolenaar071d4272004-06-13 20:20:40 +0000484 * initialized, and initializing options may cause transchar() to be called!
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100485 * When chartab_initialized == FALSE don't use g_chartab[].
Bram Moolenaar071d4272004-06-13 20:20:40 +0000486 * Does NOT work for multi-byte characters, c must be <= 255.
487 * Also doesn't work for the first byte of a multi-byte, "c" must be a
488 * character!
489 */
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200490static char_u transchar_charbuf[7];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000491
492 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100493transchar(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000494{
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200495 return transchar_buf(curbuf, c);
496}
497
498 char_u *
499transchar_buf(buf_T *buf, int c)
500{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000501 int i;
502
503 i = 0;
Bram Moolenaarc667da52019-11-30 20:52:27 +0100504 if (IS_SPECIAL(c)) // special key code, display as ~@ char
Bram Moolenaar071d4272004-06-13 20:20:40 +0000505 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200506 transchar_charbuf[0] = '~';
507 transchar_charbuf[1] = '@';
Bram Moolenaar071d4272004-06-13 20:20:40 +0000508 i = 2;
509 c = K_SECOND(c);
510 }
511
Bram Moolenaar424bcae2022-01-31 14:59:41 +0000512 if ((!chartab_initialized && ((c >= ' ' && c <= '~')))
513 || (c < 256 && vim_isprintc_strict(c)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000514 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100515 // printable character
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200516 transchar_charbuf[i] = c;
517 transchar_charbuf[i + 1] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000518 }
519 else
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200520 transchar_nonprint(buf, transchar_charbuf + i, c);
521 return transchar_charbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000522}
523
Bram Moolenaar071d4272004-06-13 20:20:40 +0000524/*
525 * Like transchar(), but called with a byte instead of a character. Checks
cero19881d87e112023-02-16 15:03:12 +0000526 * for an illegal UTF-8 byte. Uses 'fileformat' of the current buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000527 */
528 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100529transchar_byte(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000530{
cero19881d87e112023-02-16 15:03:12 +0000531 return transchar_byte_buf(curbuf, c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000532}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000533
534/*
cero19881d87e112023-02-16 15:03:12 +0000535 * Like transchar_buf(), but called with a byte instead of a character. Checks
536 * for an illegal UTF-8 byte. Uses 'fileformat' of "buf", unless it is NULL.
537 */
538 char_u *
539transchar_byte_buf(buf_T *buf, int c)
540{
541 if (enc_utf8 && c >= 0x80)
542 {
543 transchar_nonprint(buf, transchar_charbuf, c);
544 return transchar_charbuf;
545 }
546 return transchar_buf(buf, c);
547}
548/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000549 * Convert non-printable character to two or more printable characters in
Bram Moolenaara9a6b032023-02-05 18:00:42 +0000550 * "charbuf[]". "charbuf" needs to be able to hold five bytes.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000551 * Does NOT work for multi-byte characters, c must be <= 255.
552 */
553 void
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200554transchar_nonprint(buf_T *buf, char_u *charbuf, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555{
556 if (c == NL)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100557 c = NUL; // we use newline in place of a NUL
cero19881d87e112023-02-16 15:03:12 +0000558 else if (buf != NULL && c == CAR && get_fileformat(buf) == EOL_MAC)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100559 c = NL; // we use CR in place of NL in this case
Bram Moolenaar071d4272004-06-13 20:20:40 +0000560
Bram Moolenaarc667da52019-11-30 20:52:27 +0100561 if (dy_flags & DY_UHEX) // 'display' has "uhex"
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200562 transchar_hex(charbuf, c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000563
Bram Moolenaarc667da52019-11-30 20:52:27 +0100564 else if (c <= 0x7f) // 0x00 - 0x1f and 0x7f
Bram Moolenaar071d4272004-06-13 20:20:40 +0000565 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200566 charbuf[0] = '^';
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200567 charbuf[1] = c ^ 0x40; // DEL displayed as ^?
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200568 charbuf[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000569 }
=?UTF-8?q?Dundar=20G=C3=B6c?=f01a6532022-03-09 13:00:54 +0000570 else if (enc_utf8)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000571 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200572 transchar_hex(charbuf, c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000573 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100574 else if (c >= ' ' + 0x80 && c <= '~' + 0x80) // 0xa0 - 0xfe
Bram Moolenaar071d4272004-06-13 20:20:40 +0000575 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200576 charbuf[0] = '|';
577 charbuf[1] = c - 0x80;
578 charbuf[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000579 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100580 else // 0x80 - 0x9f and 0xff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000581 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200582 charbuf[0] = '~';
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200583 charbuf[1] = (c - 0x80) ^ 0x40; // 0xff displayed as ~?
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200584 charbuf[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000585 }
586}
587
588 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100589transchar_hex(char_u *buf, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590{
591 int i = 0;
592
593 buf[0] = '<';
Bram Moolenaar071d4272004-06-13 20:20:40 +0000594 if (c > 255)
595 {
596 buf[++i] = nr2hex((unsigned)c >> 12);
597 buf[++i] = nr2hex((unsigned)c >> 8);
598 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000599 buf[++i] = nr2hex((unsigned)c >> 4);
Bram Moolenaar0ab2a882009-05-13 10:51:08 +0000600 buf[++i] = nr2hex((unsigned)c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000601 buf[++i] = '>';
602 buf[++i] = NUL;
603}
604
605/*
606 * Convert the lower 4 bits of byte "c" to its hex character.
607 * Lower case letters are used to avoid the confusion of <F1> being 0xf1 or
608 * function key 1.
609 */
Bram Moolenaar0ab2a882009-05-13 10:51:08 +0000610 static unsigned
Bram Moolenaar7454a062016-01-30 15:14:10 +0100611nr2hex(unsigned c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000612{
613 if ((c & 0xf) <= 9)
614 return (c & 0xf) + '0';
615 return (c & 0xf) - 10 + 'a';
616}
617
618/*
619 * Return number of display cells occupied by byte "b".
620 * Caller must make sure 0 <= b <= 255.
621 * For multi-byte mode "b" must be the first byte of a character.
622 * A TAB is counted as two cells: "^I".
623 * For UTF-8 mode this will return 0 for bytes >= 0x80, because the number of
624 * cells depends on further bytes.
625 */
626 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100627byte2cells(int b)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000628{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000629 if (enc_utf8 && b >= 0x80)
630 return 0;
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100631 return (g_chartab[b] & CT_CELL_MASK);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000632}
633
634/*
635 * Return number of display cells occupied by character "c".
636 * "c" can be a special key (negative number) in which case 3 or 4 is returned.
637 * A TAB is counted as two cells: "^I" or four: "<09>".
638 */
639 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100640char2cells(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000641{
642 if (IS_SPECIAL(c))
643 return char2cells(K_SECOND(c)) + 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644 if (c >= 0x80)
645 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100646 // UTF-8: above 0x80 need to check the value
Bram Moolenaar071d4272004-06-13 20:20:40 +0000647 if (enc_utf8)
648 return utf_char2cells(c);
Bram Moolenaarc667da52019-11-30 20:52:27 +0100649 // DBCS: double-byte means double-width, except for euc-jp with first
650 // byte 0x8e
Bram Moolenaar071d4272004-06-13 20:20:40 +0000651 if (enc_dbcs != 0 && c >= 0x100)
652 {
653 if (enc_dbcs == DBCS_JPNU && ((unsigned)c >> 8) == 0x8e)
654 return 1;
655 return 2;
656 }
657 }
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100658 return (g_chartab[c & 0xff] & CT_CELL_MASK);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659}
660
661/*
662 * Return number of display cells occupied by character at "*p".
663 * A TAB is counted as two cells: "^I" or four: "<09>".
664 */
665 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100666ptr2cells(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000667{
Bram Moolenaar398649e2022-08-04 15:03:48 +0100668 if (!has_mbyte)
669 return byte2cells(*p);
670
Bram Moolenaarc667da52019-11-30 20:52:27 +0100671 // For UTF-8 we need to look at more bytes if the first byte is >= 0x80.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 if (enc_utf8 && *p >= 0x80)
673 return utf_ptr2cells(p);
Bram Moolenaarc667da52019-11-30 20:52:27 +0100674 // For DBCS we can tell the cell count from the first byte.
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100675 return (g_chartab[*p] & CT_CELL_MASK);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000676}
677
678/*
Bram Moolenaar06af6022012-01-26 13:40:08 +0100679 * Return the number of character cells string "s" will take on the screen,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000680 * counting TABs as two characters: "^I".
681 */
682 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100683vim_strsize(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000684{
685 return vim_strnsize(s, (int)MAXCOL);
686}
687
688/*
Bram Moolenaar06af6022012-01-26 13:40:08 +0100689 * Return the number of character cells string "s[len]" will take on the
690 * screen, counting TABs as two characters: "^I".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000691 */
692 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100693vim_strnsize(char_u *s, int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000694{
695 int size = 0;
696
697 while (*s != NUL && --len >= 0)
Bram Moolenaar398649e2022-08-04 15:03:48 +0100698 {
699 int l = (*mb_ptr2len)(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000700
Bram Moolenaar398649e2022-08-04 15:03:48 +0100701 size += ptr2cells(s);
702 s += l;
703 len -= l - 1;
704 }
Bram Moolenaar13505972019-01-24 15:04:48 +0100705
Bram Moolenaar071d4272004-06-13 20:20:40 +0000706 return size;
707}
708
709/*
710 * Return the number of characters 'c' will take on the screen, taking
711 * into account the size of a tab.
712 * Use a define to make it fast, this is used very often!!!
713 * Also see getvcol() below.
714 */
715
Bram Moolenaar04958cb2018-06-23 19:23:02 +0200716#ifdef FEAT_VARTABS
717# define RET_WIN_BUF_CHARTABSIZE(wp, buf, p, col) \
kylo252ae6f1d82022-02-16 19:24:07 +0000718 if (*(p) == TAB && (!(wp)->w_p_list || (wp)->w_lcs_chars.tab1)) \
Bram Moolenaar04958cb2018-06-23 19:23:02 +0200719 { \
720 return tabstop_padding(col, (buf)->b_p_ts, (buf)->b_p_vts_array); \
721 } \
722 else \
723 return ptr2cells(p);
724#else
725# define RET_WIN_BUF_CHARTABSIZE(wp, buf, p, col) \
Bram Moolenaareed9d462021-02-15 20:38:25 +0100726 if (*(p) == TAB && (!(wp)->w_p_list || wp->w_lcs_chars.tab1)) \
Bram Moolenaar071d4272004-06-13 20:20:40 +0000727 { \
728 int ts; \
729 ts = (buf)->b_p_ts; \
730 return (int)(ts - (col % ts)); \
731 } \
732 else \
733 return ptr2cells(p);
Bram Moolenaar04958cb2018-06-23 19:23:02 +0200734#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000735
Bram Moolenaar071d4272004-06-13 20:20:40 +0000736 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100737chartabsize(char_u *p, colnr_T col)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000738{
739 RET_WIN_BUF_CHARTABSIZE(curwin, curbuf, p, col)
740}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741
742#ifdef FEAT_LINEBREAK
743 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100744win_chartabsize(win_T *wp, char_u *p, colnr_T col)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000745{
746 RET_WIN_BUF_CHARTABSIZE(wp, wp->w_buffer, p, col)
747}
748#endif
749
750/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100751 * Return the number of characters the string "s" will take on the screen,
Bram Moolenaardc536092010-07-18 15:45:49 +0200752 * taking into account the size of a tab.
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100753 * Does not handle text properties, since "s" is not a buffer line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754 */
755 int
Bram Moolenaarc9121f72022-10-14 20:09:04 +0100756linetabsize_str(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000757{
Bram Moolenaardc536092010-07-18 15:45:49 +0200758 return linetabsize_col(0, s);
759}
760
761/*
Bram Moolenaarc9121f72022-10-14 20:09:04 +0100762 * Like linetabsize_str(), but "s" starts at column "startcol".
Bram Moolenaardc536092010-07-18 15:45:49 +0200763 */
764 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100765linetabsize_col(int startcol, char_u *s)
Bram Moolenaardc536092010-07-18 15:45:49 +0200766{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100767 chartabsize_T cts;
Christian Brabandteff20eb2024-05-15 21:35:36 +0200768 vimlong_T vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000769
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100770 init_chartabsize_arg(&cts, curwin, 0, startcol, s, s);
Christian Brabandteff20eb2024-05-15 21:35:36 +0200771 vcol = cts.cts_vcol;
772
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100773 while (*cts.cts_ptr != NUL)
Christian Brabandteff20eb2024-05-15 21:35:36 +0200774 {
775 vcol += lbr_chartabsize_adv(&cts);
776 if (vcol > MAXCOL)
777 {
778 cts.cts_vcol = MAXCOL;
779 break;
780 }
781 else
782 cts.cts_vcol = (int)vcol;
783 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100784 clear_chartabsize_arg(&cts);
785 return (int)cts.cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000786}
787
788/*
Bram Moolenaarc9121f72022-10-14 20:09:04 +0100789 * Like linetabsize_str(), but for a given window instead of the current one.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000790 */
791 int
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100792win_linetabsize(win_T *wp, linenr_T lnum, char_u *line, colnr_T len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000793{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100794 chartabsize_T cts;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000795
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100796 init_chartabsize_arg(&cts, wp, lnum, 0, line, line);
Bram Moolenaar49a90792022-08-09 18:25:23 +0100797 win_linetabsize_cts(&cts, len);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100798 clear_chartabsize_arg(&cts);
799 return (int)cts.cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000800}
801
Bram Moolenaarc9121f72022-10-14 20:09:04 +0100802/*
803 * Return the number of cells line "lnum" of window "wp" will take on the
804 * screen, taking into account the size of a tab and text properties.
805 */
806 int
807linetabsize(win_T *wp, linenr_T lnum)
808{
809 return win_linetabsize(wp, lnum,
810 ml_get_buf(wp->w_buffer, lnum, FALSE), (colnr_T)MAXCOL);
811}
812
Dylan Thacker-Smithb2d124c2024-03-24 09:43:25 +0100813/*
814 * Like linetabsize(), but excludes 'above'/'after'/'right'/'below' aligned
815 * virtual text, while keeping inline virtual text.
816 */
817 int
818linetabsize_no_outer(win_T *wp, linenr_T lnum)
819{
820#ifndef FEAT_PROP_POPUP
821 return linetabsize(wp, lnum);
822#else
823 chartabsize_T cts;
824 char_u *line = ml_get_buf(wp->w_buffer, lnum, FALSE);
825
826 init_chartabsize_arg(&cts, wp, lnum, 0, line, line);
827
828 if (cts.cts_text_prop_count)
829 {
zeertzjqd9be94c2024-07-14 10:20:20 +0200830 int write_idx = 0;
831 for (int read_idx = 0; read_idx < cts.cts_text_prop_count; read_idx++)
832 {
833 textprop_T *tp = &cts.cts_text_props[read_idx];
834 if (tp->tp_col != MAXCOL)
835 {
836 if (read_idx != write_idx)
837 cts.cts_text_props[write_idx] = *tp;
838 write_idx++;
839 }
840 }
841 cts.cts_text_prop_count = write_idx;
842 if (cts.cts_text_prop_count == 0)
843 VIM_CLEAR(cts.cts_text_props);
Dylan Thacker-Smithb2d124c2024-03-24 09:43:25 +0100844 }
845
846 win_linetabsize_cts(&cts, (colnr_T)MAXCOL);
847 clear_chartabsize_arg(&cts);
848 return (int)cts.cts_vcol;
849#endif
850}
851
Bram Moolenaar49a90792022-08-09 18:25:23 +0100852 void
853win_linetabsize_cts(chartabsize_T *cts, colnr_T len)
854{
Christian Brabandteff20eb2024-05-15 21:35:36 +0200855 vimlong_T vcol = cts->cts_vcol;
Bram Moolenaar49a90792022-08-09 18:25:23 +0100856#ifdef FEAT_PROP_POPUP
857 cts->cts_with_trailing = len == MAXCOL;
858#endif
859 for ( ; *cts->cts_ptr != NUL && (len == MAXCOL || cts->cts_ptr < cts->cts_line + len);
860 MB_PTR_ADV(cts->cts_ptr))
Christian Brabandteff20eb2024-05-15 21:35:36 +0200861 {
862 vcol += win_lbr_chartabsize(cts, NULL);
863 if (vcol > MAXCOL)
864 {
865 cts->cts_vcol = MAXCOL;
866 break;
867 }
868 else
869 cts->cts_vcol = (int)vcol;
870 }
Bram Moolenaar49a90792022-08-09 18:25:23 +0100871#ifdef FEAT_PROP_POPUP
Ibbya6ab5e62023-08-20 20:24:18 +0200872 // check for a virtual text at the end of a line or on an empty line
zeertzjqe3daa062023-08-27 19:11:46 +0200873 if (len == MAXCOL && cts->cts_has_prop_with_text && *cts->cts_ptr == NUL)
Bram Moolenaar49a90792022-08-09 18:25:23 +0100874 {
875 (void)win_lbr_chartabsize(cts, NULL);
Christian Brabandteff20eb2024-05-15 21:35:36 +0200876 vcol += cts->cts_cur_text_width;
Bram Moolenaar55a27d82023-02-12 18:03:57 +0000877 // when properties are above or below the empty line must also be
878 // counted
Ibbya6ab5e62023-08-20 20:24:18 +0200879 if (cts->cts_ptr == cts->cts_line && cts->cts_prop_lines > 0)
Christian Brabandteff20eb2024-05-15 21:35:36 +0200880 ++vcol;
881 cts->cts_vcol = vcol > MAXCOL ? MAXCOL : (int)vcol;
Bram Moolenaar49a90792022-08-09 18:25:23 +0100882 }
883#endif
884}
885
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886/*
Bram Moolenaar81695252004-12-29 20:58:21 +0000887 * Return TRUE if 'c' is a normal identifier character:
888 * Letters and characters from the 'isident' option.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000889 */
890 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100891vim_isIDc(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000892{
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100893 return (c > 0 && c < 0x100 && (g_chartab[c] & CT_ID_CHAR));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000894}
895
896/*
Bram Moolenaare3d1f4c2021-04-06 20:21:59 +0200897 * Like vim_isIDc() but not using the 'isident' option: letters, numbers and
898 * underscore.
899 */
900 int
901vim_isNormalIDc(int c)
902{
903 return ASCII_ISALNUM(c) || c == '_';
904}
905
906/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000907 * return TRUE if 'c' is a keyword character: Letters and characters from
Bram Moolenaarcaa55b62017-01-10 13:51:09 +0100908 * 'iskeyword' option for the current buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 * For multi-byte characters mb_get_class() is used (builtin rules).
910 */
911 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100912vim_iswordc(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000913{
Bram Moolenaar9d182dd2013-01-23 15:53:15 +0100914 return vim_iswordc_buf(c, curbuf);
915}
916
917 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100918vim_iswordc_buf(int c, buf_T *buf)
Bram Moolenaar9d182dd2013-01-23 15:53:15 +0100919{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000920 if (c >= 0x100)
921 {
922 if (enc_dbcs != 0)
Bram Moolenaar0ab2a882009-05-13 10:51:08 +0000923 return dbcs_class((unsigned)c >> 8, (unsigned)(c & 0xff)) >= 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000924 if (enc_utf8)
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100925 return utf_class_buf(c, buf) >= 2;
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100926 return FALSE;
927 }
928 return (c > 0 && GET_CHARTAB(buf, c) != 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000929}
930
931/*
932 * Just like vim_iswordc() but uses a pointer to the (multi-byte) character.
933 */
934 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100935vim_iswordp(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000936{
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100937 return vim_iswordp_buf(p, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000938}
939
Bram Moolenaar071d4272004-06-13 20:20:40 +0000940 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100941vim_iswordp_buf(char_u *p, buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000942{
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100943 int c = *p;
944
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100945 if (has_mbyte && MB_BYTE2LEN(c) > 1)
946 c = (*mb_ptr2char)(p);
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100947 return vim_iswordc_buf(c, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000948}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000949
950/*
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100951 * Return TRUE if 'c' is a valid file-name character as specified with the
952 * 'isfname' option.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000953 * Assume characters above 0x100 are valid (multi-byte).
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100954 * To be used for commands like "gf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000955 */
956 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100957vim_isfilec(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000958{
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100959 return (c >= 0x100 || (c > 0 && (g_chartab[c] & CT_FNAME_CHAR)));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000960}
961
Dominique Pellee764d1b2023-03-12 21:20:59 +0000962#if defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000963/*
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100964 * Return TRUE if 'c' is a valid file-name character, including characters left
965 * out of 'isfname' to make "gf" work, such as comma, space, '@', etc.
966 */
967 int
968vim_is_fname_char(int c)
969{
970 return vim_isfilec(c) || c == ',' || c == ' ' || c == '@';
971}
Dominique Pellee764d1b2023-03-12 21:20:59 +0000972#endif
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100973
974/*
Bram Moolenaardd87969c2007-08-21 13:07:12 +0000975 * return TRUE if 'c' is a valid file-name character or a wildcard character
976 * Assume characters above 0x100 are valid (multi-byte).
977 * Explicitly interpret ']' as a wildcard character as mch_has_wildcard("]")
978 * returns false.
979 */
980 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100981vim_isfilec_or_wc(int c)
Bram Moolenaardd87969c2007-08-21 13:07:12 +0000982{
983 char_u buf[2];
984
985 buf[0] = (char_u)c;
986 buf[1] = NUL;
987 return vim_isfilec(c) || c == ']' || mch_has_wildcard(buf);
988}
989
990/*
Bram Moolenaar3317d5e2017-04-08 19:12:06 +0200991 * Return TRUE if 'c' is a printable character.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000992 * Assume characters above 0x100 are printable (multi-byte), except for
993 * Unicode.
994 */
995 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100996vim_isprintc(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000997{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000998 if (enc_utf8 && c >= 0x100)
999 return utf_printable(c);
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +01001000 return (c >= 0x100 || (c > 0 && (g_chartab[c] & CT_PRINT_CHAR)));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001001}
1002
1003/*
1004 * Strict version of vim_isprintc(c), don't return TRUE if "c" is the head
1005 * byte of a double-byte character.
1006 */
1007 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001008vim_isprintc_strict(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001009{
Bram Moolenaar071d4272004-06-13 20:20:40 +00001010 if (enc_dbcs != 0 && c < 0x100 && MB_BYTE2LEN(c) > 1)
1011 return FALSE;
1012 if (enc_utf8 && c >= 0x100)
1013 return utf_printable(c);
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +01001014 return (c >= 0x100 || (c > 0 && (g_chartab[c] & CT_PRINT_CHAR)));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001015}
1016
1017/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001018 * Prepare the structure passed to chartabsize functions.
1019 * "line" is the start of the line, "ptr" is the first relevant character.
1020 * When "lnum" is zero do not use text properties that insert text.
1021 */
1022 void
1023init_chartabsize_arg(
1024 chartabsize_T *cts,
1025 win_T *wp,
Bram Moolenaare5a420f2022-09-07 21:46:56 +01001026 linenr_T lnum UNUSED,
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001027 colnr_T col,
1028 char_u *line,
1029 char_u *ptr)
1030{
Bram Moolenaar3f79b612022-08-01 12:06:40 +01001031 CLEAR_POINTER(cts);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001032 cts->cts_win = wp;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001033 cts->cts_vcol = col;
1034 cts->cts_line = line;
1035 cts->cts_ptr = ptr;
zeertzjq5b0722b2024-01-17 20:42:53 +01001036#ifdef FEAT_LINEBREAK
1037 cts->cts_bri_size = -1;
1038#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001039#ifdef FEAT_PROP_POPUP
Bram Moolenaar702bd6c2022-09-14 16:09:57 +01001040 if (lnum > 0 && !ignore_text_props)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001041 {
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001042 char_u *prop_start;
1043 int count;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001044
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001045 count = get_text_props(wp->w_buffer, lnum, &prop_start, FALSE);
1046 cts->cts_text_prop_count = count;
1047 if (count > 0)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001048 {
1049 // Make a copy of the properties, so that they are properly
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001050 // aligned. Make it twice as long for the sorting below.
1051 cts->cts_text_props = ALLOC_MULT(textprop_T, count * 2);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001052 if (cts->cts_text_props == NULL)
1053 cts->cts_text_prop_count = 0;
1054 else
1055 {
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001056 int i;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001057
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001058 mch_memmove(cts->cts_text_props + count, prop_start,
1059 count * sizeof(textprop_T));
1060 for (i = 0; i < count; ++i)
Bram Moolenaar89469d12022-12-02 20:46:26 +00001061 {
1062 textprop_T *tp = cts->cts_text_props + i + count;
1063 if (tp->tp_id < 0
1064 && text_prop_type_valid(wp->w_buffer, tp))
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001065 {
1066 cts->cts_has_prop_with_text = TRUE;
1067 break;
1068 }
Bram Moolenaar89469d12022-12-02 20:46:26 +00001069 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001070 if (!cts->cts_has_prop_with_text)
1071 {
1072 // won't use the text properties, free them
Bram Moolenaar3f79b612022-08-01 12:06:40 +01001073 VIM_CLEAR(cts->cts_text_props);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001074 cts->cts_text_prop_count = 0;
1075 }
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001076 else
1077 {
1078 int *text_prop_idxs;
1079
1080 // Need to sort the array to get any truncation right.
1081 // Do the sorting in the second part of the array, then
1082 // move the sorted props to the first part of the array.
1083 text_prop_idxs = ALLOC_MULT(int, count);
1084 if (text_prop_idxs != NULL)
1085 {
1086 for (i = 0; i < count; ++i)
1087 text_prop_idxs[i] = i + count;
1088 sort_text_props(curbuf, cts->cts_text_props,
1089 text_prop_idxs, count);
1090 // Here we want the reverse order.
1091 for (i = 0; i < count; ++i)
1092 cts->cts_text_props[count - i - 1] =
1093 cts->cts_text_props[text_prop_idxs[i]];
1094 vim_free(text_prop_idxs);
1095 }
1096 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001097 }
1098 }
1099 }
1100#endif
1101}
1102
1103/*
1104 * Free any allocated item in "cts".
1105 */
1106 void
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001107clear_chartabsize_arg(chartabsize_T *cts UNUSED)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001108{
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001109#ifdef FEAT_PROP_POPUP
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001110 if (cts->cts_text_prop_count > 0)
Bram Moolenaar34a1f772022-07-26 11:20:48 +01001111 {
Bram Moolenaar3f79b612022-08-01 12:06:40 +01001112 VIM_CLEAR(cts->cts_text_props);
1113 cts->cts_text_prop_count = 0;
Bram Moolenaar34a1f772022-07-26 11:20:48 +01001114 }
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001115#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001116}
1117
1118/*
1119 * Like chartabsize(), but also check for line breaks on the screen and text
1120 * properties that insert text.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001121 */
1122 int
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001123lbr_chartabsize(chartabsize_T *cts)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001124{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001125#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
1126 if (1
1127# ifdef FEAT_LINEBREAK
1128 && !curwin->w_p_lbr && *get_showbreak_value(curwin) == NUL
1129 && !curwin->w_p_bri
1130# endif
1131# ifdef FEAT_PROP_POPUP
1132 && !cts->cts_has_prop_with_text
1133#endif
1134 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001135 {
1136#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001137 if (curwin->w_p_wrap)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001138 return win_nolbr_chartabsize(cts, NULL);
1139 RET_WIN_BUF_CHARTABSIZE(curwin, curbuf, cts->cts_ptr, cts->cts_vcol)
1140#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001141 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001142 return win_lbr_chartabsize(cts, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001143#endif
1144}
1145
1146/*
1147 * Call lbr_chartabsize() and advance the pointer.
1148 */
1149 int
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001150lbr_chartabsize_adv(chartabsize_T *cts)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001151{
1152 int retval;
1153
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001154 retval = lbr_chartabsize(cts);
1155 MB_PTR_ADV(cts->cts_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001156 return retval;
1157}
1158
1159/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001160 * Return the screen size of the character indicated by "cts".
1161 * "cts->cts_cur_text_width" is set to the extra size for a text property that
1162 * inserts text.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001163 * This function is used very often, keep it fast!!!!
1164 *
zeertzjqb557f482023-08-22 22:07:34 +02001165 * If "headp" not NULL, set "*headp" to the size of 'showbreak'/'breakindent'
1166 * included in the return value.
1167 * When "cts->cts_max_head_vcol" is positive, only count in "*headp" the size
1168 * of 'showbreak'/'breakindent' before "cts->cts_max_head_vcol".
1169 * When "cts->cts_max_head_vcol" is negative, only count in "*headp" the size
1170 * of 'showbreak'/'breakindent' before where cursor should be placed.
1171 *
1172 * Warning: "*headp" may not be set if it's 0, init to 0 before calling.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001173 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001174 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001175win_lbr_chartabsize(
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001176 chartabsize_T *cts,
1177 int *headp UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001178{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001179 win_T *wp = cts->cts_win;
Martin Tournoijba43e762022-10-13 22:12:15 +01001180#if defined(FEAT_PROP_POPUP) || defined(FEAT_LINEBREAK)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001181 char_u *line = cts->cts_line; // start of the line
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001182#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001183 char_u *s = cts->cts_ptr;
1184 colnr_T vcol = cts->cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001185#ifdef FEAT_LINEBREAK
Bram Moolenaar071d4272004-06-13 20:20:40 +00001186 int size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001187 int mb_added = 0;
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00001188 int n;
Bram Moolenaaree857022019-11-09 23:26:40 +01001189 char_u *sbr;
Bram Moolenaarcba69522022-08-06 21:03:53 +01001190 int no_sbr = FALSE;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001191#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001192
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001193#if defined(FEAT_PROP_POPUP)
1194 cts->cts_cur_text_width = 0;
Bram Moolenaar04e0ed12022-09-10 20:00:56 +01001195 cts->cts_first_char = 0;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001196#endif
1197
1198#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001199 /*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001200 * No 'linebreak', 'showbreak', 'breakindent' and text properties that
1201 * insert text: return quickly.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001202 */
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001203 if (1
1204# ifdef FEAT_LINEBREAK
1205 && !wp->w_p_lbr && !wp->w_p_bri && *get_showbreak_value(wp) == NUL
1206# endif
1207# ifdef FEAT_PROP_POPUP
1208 && !cts->cts_has_prop_with_text
1209# endif
1210 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001211#endif
1212 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001213 if (wp->w_p_wrap)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001214 return win_nolbr_chartabsize(cts, headp);
1215 RET_WIN_BUF_CHARTABSIZE(wp, wp->w_buffer, s, vcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001216 }
1217
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001218#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
zeertzjq11939512023-08-23 20:58:01 +02001219 int has_lcs_eol = wp->w_p_list && wp->w_lcs_chars.eol != NUL;
1220
Bram Moolenaar071d4272004-06-13 20:20:40 +00001221 /*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001222 * First get the normal size, without 'linebreak' or text properties
Bram Moolenaar071d4272004-06-13 20:20:40 +00001223 */
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001224 size = win_chartabsize(wp, s, vcol);
Dylan Thacker-Smithda0c9132024-02-27 20:25:10 +01001225 if (*s == NUL)
1226 {
1227 // 1 cell for EOL list char (if present), as opposed to the two cell ^@
1228 // for a NUL character in the text.
1229 size = has_lcs_eol ? 1 : 0;
1230 }
zeertzjqac2d8812023-08-31 18:07:30 +02001231# ifdef FEAT_LINEBREAK
1232 int is_doublewidth = has_mbyte && size == 2 && MB_BYTE2LEN(*s) > 1;
1233# endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001234
1235# ifdef FEAT_PROP_POPUP
Bram Moolenaar49a90792022-08-09 18:25:23 +01001236 if (cts->cts_has_prop_with_text)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001237 {
Bram Moolenaare428fa02022-08-09 16:55:41 +01001238 int tab_size = size;
Bram Moolenaar49a90792022-08-09 18:25:23 +01001239 int charlen = *s == NUL ? 1 : mb_ptr2len(s);
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001240 int i;
1241 int col = (int)(s - line);
1242 garray_T *gap = &wp->w_buffer->b_textprop_text;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001243
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001244 // The "$" for 'list' mode will go between the EOL and
1245 // the text prop, account for that.
zeertzjq11939512023-08-23 20:58:01 +02001246 if (has_lcs_eol)
1247 {
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001248 ++vcol;
zeertzjq11939512023-08-23 20:58:01 +02001249 --size;
1250 }
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001251
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001252 for (i = 0; i < cts->cts_text_prop_count; ++i)
1253 {
Bram Moolenaarc8bf59e2022-08-28 16:39:22 +01001254 textprop_T *tp = cts->cts_text_props + i;
1255 int col_off = win_col_off(wp);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001256
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001257 // Watch out for the text being deleted. "cts_text_props" is a
1258 // copy, the text prop may actually have been removed from the line.
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001259 if (tp->tp_id < 0
Bram Moolenaar25463612022-08-08 11:07:47 +01001260 && ((tp->tp_col - 1 >= col
Bram Moolenaare428fa02022-08-09 16:55:41 +01001261 && tp->tp_col - 1 < col + charlen)
Bram Moolenaar04e0ed12022-09-10 20:00:56 +01001262 || (tp->tp_col == MAXCOL
1263 && ((tp->tp_flags & TP_FLAG_ALIGN_ABOVE)
1264 ? col == 0
zeertzjqe3daa062023-08-27 19:11:46 +02001265 : s[0] == NUL && cts->cts_with_trailing)))
Bram Moolenaar4ce1f992022-12-19 13:31:06 +00001266 && -tp->tp_id - 1 < gap->ga_len)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001267 {
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001268 char_u *p = ((char_u **)gap->ga_data)[-tp->tp_id - 1];
Bram Moolenaarb7963df2022-07-31 17:12:43 +01001269
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001270 if (p != NULL)
Bram Moolenaarb7963df2022-07-31 17:12:43 +01001271 {
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001272 int cells;
Bram Moolenaar398649e2022-08-04 15:03:48 +01001273
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001274 if (tp->tp_col == MAXCOL)
1275 {
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001276 int n_extra = (int)STRLEN(p);
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001277
Bram Moolenaarf167c7b2022-10-09 21:53:58 +01001278 cells = text_prop_position(wp, tp, vcol,
Bram Moolenaarc8bf59e2022-08-28 16:39:22 +01001279 (vcol + size) % (wp->w_width - col_off) + col_off,
Bram Moolenaar56a40fe2022-12-06 14:17:57 +00001280 &n_extra, &p, NULL, NULL, FALSE);
zeertzjqce53e3e2023-09-01 18:49:30 +02001281# ifdef FEAT_LINEBREAK
Bram Moolenaarcba69522022-08-06 21:03:53 +01001282 no_sbr = TRUE; // don't use 'showbreak' now
zeertzjqce53e3e2023-09-01 18:49:30 +02001283# endif
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001284 }
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001285 else
1286 cells = vim_strsize(p);
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001287 cts->cts_cur_text_width += cells;
Bram Moolenaar04e0ed12022-09-10 20:00:56 +01001288 if (tp->tp_flags & TP_FLAG_ALIGN_ABOVE)
1289 cts->cts_first_char += cells;
zeertzjqce53e3e2023-09-01 18:49:30 +02001290 else
1291 size += cells;
Bram Moolenaar28c9f892022-08-14 13:28:55 +01001292 cts->cts_start_incl = tp->tp_flags & TP_FLAG_START_INCL;
Bram Moolenaare428fa02022-08-09 16:55:41 +01001293 if (*s == TAB)
1294 {
1295 // tab size changes because of the inserted text
1296 size -= tab_size;
1297 tab_size = win_chartabsize(wp, s, vcol + size);
1298 size += tab_size;
1299 }
Bram Moolenaar55a27d82023-02-12 18:03:57 +00001300 if (tp->tp_col == MAXCOL && (tp->tp_flags
1301 & (TP_FLAG_ALIGN_ABOVE | TP_FLAG_ALIGN_BELOW)))
1302 // count extra line for property above/below
1303 ++cts->cts_prop_lines;
Bram Moolenaarb7963df2022-07-31 17:12:43 +01001304 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001305 }
Bram Moolenaar50e75fe2022-08-05 20:25:50 +01001306 if (tp->tp_col != MAXCOL && tp->tp_col - 1 > col)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001307 break;
1308 }
zeertzjq11939512023-08-23 20:58:01 +02001309 if (has_lcs_eol)
1310 {
Bram Moolenaarf396ce82022-08-23 18:39:37 +01001311 --vcol;
zeertzjq11939512023-08-23 20:58:01 +02001312 ++size;
1313 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001314 }
1315# endif
1316
1317# ifdef FEAT_LINEBREAK
zeertzjqac2d8812023-08-31 18:07:30 +02001318 if (is_doublewidth && wp->w_p_wrap && in_win_border(wp, vcol + size - 2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001319 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001320 ++size; // Count the ">" in the last column.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001321 mb_added = 1;
1322 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001323
1324 /*
Bram Moolenaar597a4222014-06-25 14:39:50 +02001325 * May have to add something for 'breakindent' and/or 'showbreak'
zeertzjq11939512023-08-23 20:58:01 +02001326 * string at the start of a screen line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001327 */
zeertzjqb557f482023-08-22 22:07:34 +02001328 int head = mb_added;
zeertzjq11939512023-08-23 20:58:01 +02001329 sbr = no_sbr ? empty_option : get_showbreak_value(wp);
1330 // When "size" is 0, no new screen line is started.
1331 if (size > 0 && wp->w_p_wrap && (*sbr != NUL || wp->w_p_bri))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001332 {
zeertzjqb557f482023-08-22 22:07:34 +02001333 int col_off_prev = win_col_off(wp);
1334 int width2 = wp->w_width - col_off_prev + win_col_off2(wp);
zeertzjqb557f482023-08-22 22:07:34 +02001335 colnr_T wcol = vcol + col_off_prev;
zeertzjqce53e3e2023-09-01 18:49:30 +02001336# ifdef FEAT_PROP_POPUP
zeertzjq6e55e852023-08-30 16:55:09 +02001337 wcol -= wp->w_virtcol_first_char;
zeertzjqce53e3e2023-09-01 18:49:30 +02001338# endif
zeertzjq1d3e0e82023-08-28 21:20:16 +02001339 colnr_T max_head_vcol = cts->cts_max_head_vcol;
1340 int added = 0;
1341
zeertzjqb557f482023-08-22 22:07:34 +02001342 // cells taken by 'showbreak'/'breakindent' before current char
1343 int head_prev = 0;
1344 if (wcol >= wp->w_width)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001345 {
zeertzjqb557f482023-08-22 22:07:34 +02001346 wcol -= wp->w_width;
1347 col_off_prev = wp->w_width - width2;
1348 if (wcol >= width2 && width2 > 0)
1349 wcol %= width2;
Bram Moolenaaree857022019-11-09 23:26:40 +01001350 if (*sbr != NUL)
zeertzjqb557f482023-08-22 22:07:34 +02001351 head_prev += vim_strsize(sbr);
1352 if (wp->w_p_bri)
zeertzjq5b0722b2024-01-17 20:42:53 +01001353 {
1354 if (cts->cts_bri_size < 0)
1355 cts->cts_bri_size = get_breakindent_win(wp, line);
1356 head_prev += cts->cts_bri_size;
1357 }
zeertzjqb557f482023-08-22 22:07:34 +02001358 if (wcol < head_prev)
Bram Moolenaard574ea22015-01-14 19:35:14 +01001359 {
zeertzjq1d3e0e82023-08-28 21:20:16 +02001360 head_prev -= wcol;
1361 wcol += head_prev;
zeertzjqb557f482023-08-22 22:07:34 +02001362 added += head_prev;
1363 if (max_head_vcol <= 0 || vcol < max_head_vcol)
1364 head += head_prev;
1365 }
zeertzjq6e55e852023-08-30 16:55:09 +02001366 else
1367 head_prev = 0;
zeertzjq1d3e0e82023-08-28 21:20:16 +02001368 wcol += col_off_prev;
1369 }
zeertzjqb557f482023-08-22 22:07:34 +02001370
zeertzjq1d3e0e82023-08-28 21:20:16 +02001371 if (wcol + size > wp->w_width)
1372 {
zeertzjqb557f482023-08-22 22:07:34 +02001373 // cells taken by 'showbreak'/'breakindent' halfway current char
1374 int head_mid = 0;
1375 if (*sbr != NUL)
1376 head_mid += vim_strsize(sbr);
1377 if (wp->w_p_bri)
zeertzjq5b0722b2024-01-17 20:42:53 +01001378 {
1379 if (cts->cts_bri_size < 0)
1380 cts->cts_bri_size = get_breakindent_win(wp, line);
1381 head_mid += cts->cts_bri_size;
1382 }
zeertzjq5532d3b2024-03-28 10:04:25 +01001383 if (head_mid > 0)
zeertzjqb557f482023-08-22 22:07:34 +02001384 {
zeertzjq6a389722023-08-27 19:04:14 +02001385 // Calculate effective window width.
zeertzjq11939512023-08-23 20:58:01 +02001386 int prev_rem = wp->w_width - wcol;
1387 int width = width2 - head_mid;
Bram Moolenaar7833dab2019-05-27 22:01:40 +02001388
zeertzjq11939512023-08-23 20:58:01 +02001389 if (width <= 0)
1390 width = 1;
zeertzjq6a389722023-08-27 19:04:14 +02001391 // Divide "size - prev_rem" by "width", rounding up.
zeertzjq11939512023-08-23 20:58:01 +02001392 int cnt = (size - prev_rem + width - 1) / width;
1393 added += cnt * head_mid;
zeertzjqb557f482023-08-22 22:07:34 +02001394
zeertzjq11939512023-08-23 20:58:01 +02001395 if (max_head_vcol == 0 || vcol + size + added < max_head_vcol)
1396 head += cnt * head_mid;
1397 else if (max_head_vcol > vcol + head_prev + prev_rem)
1398 head += (max_head_vcol - (vcol + head_prev + prev_rem)
zeertzjqb557f482023-08-22 22:07:34 +02001399 + width2 - 1) / width2 * head_mid;
zeertzjqce53e3e2023-09-01 18:49:30 +02001400# ifdef FEAT_PROP_POPUP
zeertzjq11939512023-08-23 20:58:01 +02001401 else if (max_head_vcol < 0)
1402 {
1403 int off = 0;
zeertzjq6e55e852023-08-30 16:55:09 +02001404 if (*s != NUL
zeertzjq11939512023-08-23 20:58:01 +02001405 && ((State & MODE_NORMAL) || cts->cts_start_incl))
1406 off += cts->cts_cur_text_width;
1407 if (off >= prev_rem)
1408 head += (1 + (off - prev_rem) / width) * head_mid;
Bram Moolenaard574ea22015-01-14 19:35:14 +01001409 }
zeertzjqce53e3e2023-09-01 18:49:30 +02001410# endif
Bram Moolenaard574ea22015-01-14 19:35:14 +01001411 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001412 }
zeertzjq1d3e0e82023-08-28 21:20:16 +02001413
1414 size += added;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001415 }
zeertzjq6e55e852023-08-30 16:55:09 +02001416
Bram Moolenaar071d4272004-06-13 20:20:40 +00001417 if (headp != NULL)
zeertzjqb557f482023-08-22 22:07:34 +02001418 *headp = head;
zeertzjq6e55e852023-08-30 16:55:09 +02001419
zeertzjq703f9bc2024-01-25 21:27:13 +01001420 int need_lbr = FALSE;
zeertzjq6e55e852023-08-30 16:55:09 +02001421 /*
1422 * If 'linebreak' set check at a blank before a non-blank if the line
zeertzjq703f9bc2024-01-25 21:27:13 +01001423 * needs a break here.
zeertzjq6e55e852023-08-30 16:55:09 +02001424 */
zeertzjq703f9bc2024-01-25 21:27:13 +01001425 if (wp->w_p_lbr && wp->w_p_wrap && wp->w_width != 0
1426 && VIM_ISBREAK((int)s[0]) && !VIM_ISBREAK((int)s[1]))
Christian Brabandtdd75fcf2023-10-11 21:51:19 +02001427 {
1428 char_u *t = cts->cts_line;
Christian Brabandtcd6ee692023-10-14 11:29:28 +02001429 while (VIM_ISBREAK((int)t[0]))
Christian Brabandtdd75fcf2023-10-11 21:51:19 +02001430 t++;
zeertzjq703f9bc2024-01-25 21:27:13 +01001431 // 'linebreak' is only needed when not in leading whitespace.
1432 need_lbr = s >= t;
Christian Brabandtdd75fcf2023-10-11 21:51:19 +02001433 }
zeertzjq703f9bc2024-01-25 21:27:13 +01001434 if (need_lbr)
zeertzjq6e55e852023-08-30 16:55:09 +02001435 {
1436 /*
1437 * Count all characters from first non-blank after a blank up to next
1438 * non-blank after a blank.
1439 */
1440 int numberextra = win_col_off(wp);
1441 colnr_T col_adj = size - 1;
1442 colnr_T colmax = (colnr_T)(wp->w_width - numberextra - col_adj);
1443 if (vcol >= colmax)
1444 {
1445 colmax += col_adj;
1446 n = colmax + win_col_off2(wp);
1447 if (n > 0)
1448 colmax += (((vcol - colmax) / n) + 1) * n - col_adj;
1449 }
1450
1451 colnr_T vcol2 = vcol;
1452 for (;;)
1453 {
1454 char_u *ps = s;
1455 MB_PTR_ADV(s);
1456 int c = *s;
1457 if (!(c != NUL
1458 && (VIM_ISBREAK(c)
1459 || (!VIM_ISBREAK(c)
1460 && (vcol2 == vcol || !VIM_ISBREAK((int)*ps))))))
1461 break;
1462
1463 vcol2 += win_chartabsize(wp, s, vcol2);
1464 if (vcol2 >= colmax) // doesn't fit
1465 {
1466 size = colmax - vcol + col_adj;
1467 break;
1468 }
1469 }
1470 }
1471
zeertzjqce53e3e2023-09-01 18:49:30 +02001472# ifdef FEAT_PROP_POPUP
1473 size += cts->cts_first_char;
1474# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475 return size;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001476# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001477#endif
1478}
1479
Bram Moolenaar071d4272004-06-13 20:20:40 +00001480/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001481 * Like win_lbr_chartabsize(), except that we know 'linebreak' is off, 'wrap'
1482 * is on and there are no properties that insert text. This means we need to
1483 * check for a double-byte character that doesn't fit at the end of the screen
1484 * line.
1485 * Only uses "cts_win", "cts_ptr" and "cts_vcol" from "cts".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001486 */
1487 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001488win_nolbr_chartabsize(
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001489 chartabsize_T *cts,
1490 int *headp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001491{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001492 win_T *wp = cts->cts_win;
1493 char_u *s = cts->cts_ptr;
1494 colnr_T col = cts->cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001495 int n;
1496
Bram Moolenaareed9d462021-02-15 20:38:25 +01001497 if (*s == TAB && (!wp->w_p_list || wp->w_lcs_chars.tab1))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001498 {
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001499# ifdef FEAT_VARTABS
1500 return tabstop_padding(col, wp->w_buffer->b_p_ts,
1501 wp->w_buffer->b_p_vts_array);
1502# else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001503 n = wp->w_buffer->b_p_ts;
1504 return (int)(n - (col % n));
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001505# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001506 }
1507 n = ptr2cells(s);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001508 // Add one cell for a double-width character in the last column of the
1509 // window, displayed with a ">".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001510 if (n == 2 && MB_BYTE2LEN(*s) > 1 && in_win_border(wp, col))
1511 {
1512 if (headp != NULL)
1513 *headp = 1;
1514 return 3;
1515 }
1516 return n;
1517}
1518
1519/*
1520 * Return TRUE if virtual column "vcol" is in the rightmost column of window
1521 * "wp".
1522 */
Bram Moolenaar5843f5f2019-08-20 20:13:45 +02001523 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001524in_win_border(win_T *wp, colnr_T vcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525{
Bram Moolenaarc667da52019-11-30 20:52:27 +01001526 int width1; // width of first line (after line number)
1527 int width2; // width of further lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00001528
Bram Moolenaarc667da52019-11-30 20:52:27 +01001529 if (wp->w_width == 0) // there is no border
Bram Moolenaar071d4272004-06-13 20:20:40 +00001530 return FALSE;
Bram Moolenaar02631462017-09-22 15:20:32 +02001531 width1 = wp->w_width - win_col_off(wp);
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001532 if ((int)vcol < width1 - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001533 return FALSE;
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001534 if ((int)vcol == width1 - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001535 return TRUE;
1536 width2 = width1 + win_col_off2(wp);
Bram Moolenaar8701cd62009-10-07 14:20:30 +00001537 if (width2 <= 0)
1538 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001539 return ((vcol - width1) % width2 == width2 - 1);
1540}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001541
1542/*
1543 * Get virtual column number of pos.
1544 * start: on the first position of this character (TAB, ctrl)
1545 * cursor: where the cursor is on this character (first char, except for TAB)
1546 * end: on the last position of this character (TAB, ctrl)
1547 *
1548 * This is used very often, keep it fast!
1549 */
1550 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001551getvcol(
1552 win_T *wp,
1553 pos_T *pos,
1554 colnr_T *start,
1555 colnr_T *cursor,
1556 colnr_T *end)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001557{
1558 colnr_T vcol;
Bram Moolenaarc667da52019-11-30 20:52:27 +01001559 char_u *ptr; // points to current char
Bram Moolenaarc667da52019-11-30 20:52:27 +01001560 char_u *line; // start of the line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001561 int incr;
1562 int head;
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001563#ifdef FEAT_VARTABS
1564 int *vts = wp->w_buffer->b_p_vts_array;
1565#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001566 int ts = wp->w_buffer->b_p_ts;
1567 int c;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001568 chartabsize_T cts;
Bram Moolenaar49a90792022-08-09 18:25:23 +01001569#ifdef FEAT_PROP_POPUP
1570 int on_NUL = FALSE;
1571#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001572
1573 vcol = 0;
Bram Moolenaar597a4222014-06-25 14:39:50 +02001574 line = ptr = ml_get_buf(wp->w_buffer, pos->lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001575
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001576 init_chartabsize_arg(&cts, wp, pos->lnum, 0, line, line);
zeertzjqb557f482023-08-22 22:07:34 +02001577 cts.cts_max_head_vcol = -1;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001578
Bram Moolenaar071d4272004-06-13 20:20:40 +00001579 /*
1580 * This function is used very often, do some speed optimizations.
Bram Moolenaar597a4222014-06-25 14:39:50 +02001581 * When 'list', 'linebreak', 'showbreak' and 'breakindent' are not set
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001582 * and there are no text properties with "text" use a simple loop.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001583 * Also use this when 'list' is set but tabs take their normal size.
1584 */
Bram Moolenaareed9d462021-02-15 20:38:25 +01001585 if ((!wp->w_p_list || wp->w_lcs_chars.tab1 != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001586#ifdef FEAT_LINEBREAK
Bram Moolenaaree857022019-11-09 23:26:40 +01001587 && !wp->w_p_lbr && *get_showbreak_value(wp) == NUL && !wp->w_p_bri
Bram Moolenaar071d4272004-06-13 20:20:40 +00001588#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001589#ifdef FEAT_PROP_POPUP
1590 && !cts.cts_has_prop_with_text
1591#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001592 )
1593 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001594 for (;;)
1595 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001596 head = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001597 c = *ptr;
Bram Moolenaarc667da52019-11-30 20:52:27 +01001598 // make sure we don't go past the end of the line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001599 if (c == NUL)
1600 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001601 incr = 1; // NUL at end of line only takes one column
Bram Moolenaar071d4272004-06-13 20:20:40 +00001602 break;
1603 }
Bram Moolenaarc667da52019-11-30 20:52:27 +01001604 // A tab gets expanded, depending on the current column
Bram Moolenaar071d4272004-06-13 20:20:40 +00001605 if (c == TAB)
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001606#ifdef FEAT_VARTABS
1607 incr = tabstop_padding(vcol, ts, vts);
1608#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001609 incr = ts - (vcol % ts);
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001610#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001611 else
1612 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001613 if (has_mbyte)
1614 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001615 // For utf-8, if the byte is >= 0x80, need to look at
1616 // further bytes to find the cell width.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001617 if (enc_utf8 && c >= 0x80)
1618 incr = utf_ptr2cells(ptr);
1619 else
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +01001620 incr = g_chartab[c] & CT_CELL_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001621
Bram Moolenaarc667da52019-11-30 20:52:27 +01001622 // If a double-cell char doesn't fit at the end of a line
1623 // it wraps to the next line, it's like this char is three
1624 // cells wide.
Bram Moolenaar9c33a7c2008-02-20 13:59:32 +00001625 if (incr == 2 && wp->w_p_wrap && MB_BYTE2LEN(*ptr) > 1
1626 && in_win_border(wp, vcol))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001627 {
1628 ++incr;
1629 head = 1;
1630 }
1631 }
1632 else
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +01001633 incr = g_chartab[c] & CT_CELL_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001634 }
1635
zeertzjq4ea37f82024-01-17 20:52:13 +01001636 char_u *next_ptr = ptr + (*mb_ptr2len)(ptr);
1637 if (next_ptr - line > pos->col) // character at pos->col
Bram Moolenaar071d4272004-06-13 20:20:40 +00001638 break;
1639
1640 vcol += incr;
zeertzjq4ea37f82024-01-17 20:52:13 +01001641 ptr = next_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001642 }
1643 }
1644 else
1645 {
1646 for (;;)
1647 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001648 // A tab gets expanded, depending on the current column.
1649 // Other things also take up space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001650 head = 0;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001651 incr = win_lbr_chartabsize(&cts, &head);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001652 // make sure we don't go past the end of the line
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001653 if (*cts.cts_ptr == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001654 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001655 incr = 1; // NUL at end of line only takes one column
Bram Moolenaar49a90792022-08-09 18:25:23 +01001656#ifdef FEAT_PROP_POPUP
1657 if (cts.cts_cur_text_width > 0)
1658 incr = cts.cts_cur_text_width;
1659 on_NUL = TRUE;
1660#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001661 break;
1662 }
Bram Moolenaar04e0ed12022-09-10 20:00:56 +01001663#ifdef FEAT_PROP_POPUP
1664 if (cursor == &wp->w_virtcol && cts.cts_ptr == cts.cts_line)
1665 // do not count the virtual text above for w_curswant
1666 wp->w_virtcol_first_char = cts.cts_first_char;
1667#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001668
zeertzjq4ea37f82024-01-17 20:52:13 +01001669 char_u *next_ptr = cts.cts_ptr + (*mb_ptr2len)(cts.cts_ptr);
1670 if (next_ptr - line > pos->col) // character at pos->col
Bram Moolenaar071d4272004-06-13 20:20:40 +00001671 break;
1672
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001673 cts.cts_vcol += incr;
zeertzjq4ea37f82024-01-17 20:52:13 +01001674 cts.cts_ptr = next_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001675 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001676 vcol = cts.cts_vcol;
1677 ptr = cts.cts_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001678 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001679 clear_chartabsize_arg(&cts);
1680
Bram Moolenaar071d4272004-06-13 20:20:40 +00001681 if (start != NULL)
1682 *start = vcol + head;
1683 if (end != NULL)
1684 *end = vcol + incr - 1;
1685 if (cursor != NULL)
1686 {
1687 if (*ptr == TAB
Bram Moolenaar24959102022-05-07 20:01:16 +01001688 && (State & MODE_NORMAL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001689 && !wp->w_p_list
1690 && !virtual_active()
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01001691 && !(VIsual_active
1692 && (*p_sel == 'e' || LTOREQ_POS(*pos, VIsual)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001693 )
Bram Moolenaarc667da52019-11-30 20:52:27 +01001694 *cursor = vcol + incr - 1; // cursor at end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001695 else
Bram Moolenaare428fa02022-08-09 16:55:41 +01001696 {
1697#ifdef FEAT_PROP_POPUP
Bram Moolenaar28c9f892022-08-14 13:28:55 +01001698 // in Insert mode, if "start_incl" is true the text gets inserted
1699 // after the virtual text, thus add its width
1700 if (((State & MODE_INSERT) == 0 || cts.cts_start_incl) && !on_NUL)
Bram Moolenaar49a90792022-08-09 18:25:23 +01001701 // cursor is after inserted text, unless on the NUL
Bram Moolenaare428fa02022-08-09 16:55:41 +01001702 vcol += cts.cts_cur_text_width;
Bram Moolenaar88b79cb2022-09-10 22:32:14 +01001703 else
1704 // insertion also happens after the "above" virtual text
1705 vcol += cts.cts_first_char;
Bram Moolenaare428fa02022-08-09 16:55:41 +01001706#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01001707 *cursor = vcol + head; // cursor at start
Bram Moolenaare428fa02022-08-09 16:55:41 +01001708 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001709 }
1710}
1711
1712/*
1713 * Get virtual cursor column in the current window, pretending 'list' is off.
1714 */
1715 colnr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01001716getvcol_nolist(pos_T *posp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001717{
1718 int list_save = curwin->w_p_list;
1719 colnr_T vcol;
1720
1721 curwin->w_p_list = FALSE;
Bram Moolenaardb0eede2018-04-25 22:38:17 +02001722 if (posp->coladd)
1723 getvvcol(curwin, posp, NULL, &vcol, NULL);
1724 else
Bram Moolenaardb0eede2018-04-25 22:38:17 +02001725 getvcol(curwin, posp, NULL, &vcol, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001726 curwin->w_p_list = list_save;
1727 return vcol;
1728}
1729
Bram Moolenaar071d4272004-06-13 20:20:40 +00001730/*
1731 * Get virtual column in virtual mode.
1732 */
1733 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001734getvvcol(
1735 win_T *wp,
1736 pos_T *pos,
1737 colnr_T *start,
1738 colnr_T *cursor,
1739 colnr_T *end)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001740{
1741 colnr_T col;
1742 colnr_T coladd;
1743 colnr_T endadd;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001744 char_u *ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001745
1746 if (virtual_active())
1747 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001748 // For virtual mode, only want one value
Bram Moolenaar071d4272004-06-13 20:20:40 +00001749 getvcol(wp, pos, &col, NULL, NULL);
1750
1751 coladd = pos->coladd;
1752 endadd = 0;
Bram Moolenaarc667da52019-11-30 20:52:27 +01001753 // Cannot put the cursor on part of a wide character.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001754 ptr = ml_get_buf(wp->w_buffer, pos->lnum, FALSE);
zeertzjq94b7c322024-03-12 21:50:32 +01001755 if (pos->col < ml_get_buf_len(wp->w_buffer, pos->lnum))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001756 {
1757 int c = (*mb_ptr2char)(ptr + pos->col);
1758
1759 if (c != TAB && vim_isprintc(c))
1760 {
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001761 endadd = (colnr_T)(char2cells(c) - 1);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001762 if (coladd > endadd) // past end of line
Bram Moolenaara5792f52005-11-23 21:25:05 +00001763 endadd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001764 else
1765 coladd = 0;
1766 }
1767 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001768 col += coladd;
1769 if (start != NULL)
1770 *start = col;
1771 if (cursor != NULL)
1772 *cursor = col;
1773 if (end != NULL)
1774 *end = col + endadd;
1775 }
1776 else
1777 getvcol(wp, pos, start, cursor, end);
1778}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001779
Bram Moolenaar071d4272004-06-13 20:20:40 +00001780/*
1781 * Get the leftmost and rightmost virtual column of pos1 and pos2.
1782 * Used for Visual block mode.
1783 */
1784 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001785getvcols(
1786 win_T *wp,
1787 pos_T *pos1,
1788 pos_T *pos2,
1789 colnr_T *left,
1790 colnr_T *right)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001791{
1792 colnr_T from1, from2, to1, to2;
1793
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01001794 if (LT_POSP(pos1, pos2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795 {
1796 getvvcol(wp, pos1, &from1, NULL, &to1);
1797 getvvcol(wp, pos2, &from2, NULL, &to2);
1798 }
1799 else
1800 {
1801 getvvcol(wp, pos2, &from1, NULL, &to1);
1802 getvvcol(wp, pos1, &from2, NULL, &to2);
1803 }
1804 if (from2 < from1)
1805 *left = from2;
1806 else
1807 *left = from1;
1808 if (to2 > to1)
1809 {
1810 if (*p_sel == 'e' && from2 - 1 >= to1)
1811 *right = from2 - 1;
1812 else
1813 *right = to2;
1814 }
1815 else
1816 *right = to1;
1817}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001818
1819/*
Bram Moolenaarce7eada2021-12-15 15:41:44 +00001820 * Skip over ' ' and '\t'.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821 */
1822 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001823skipwhite(char_u *q)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001824{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001825 char_u *p = q;
1826
Bram Moolenaarce7eada2021-12-15 15:41:44 +00001827 while (VIM_ISWHITE(*p))
1828 ++p;
1829 return p;
1830}
1831
Dominique Pelle748b3082022-01-08 12:41:16 +00001832#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaarce7eada2021-12-15 15:41:44 +00001833/*
1834 * skip over ' ', '\t' and '\n'.
1835 */
1836 char_u *
1837skipwhite_and_nl(char_u *q)
1838{
1839 char_u *p = q;
1840
1841 while (VIM_ISWHITE(*p) || *p == NL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842 ++p;
1843 return p;
1844}
Dominique Pelle748b3082022-01-08 12:41:16 +00001845#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001846
1847/*
Bram Moolenaare2e69e42017-09-02 20:30:35 +02001848 * getwhitecols: return the number of whitespace
1849 * columns (bytes) at the start of a given line
1850 */
1851 int
Yegappan Lakshmanana23a11b2023-02-21 14:27:41 +00001852getwhitecols_curline(void)
Bram Moolenaare2e69e42017-09-02 20:30:35 +02001853{
1854 return getwhitecols(ml_get_curline());
1855}
1856
1857 int
1858getwhitecols(char_u *p)
1859{
1860 return skipwhite(p) - p;
1861}
1862
1863/*
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001864 * skip over digits
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865 */
1866 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001867skipdigits(char_u *q)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001868{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001869 char_u *p = q;
1870
Bram Moolenaarc667da52019-11-30 20:52:27 +01001871 while (VIM_ISDIGIT(*p)) // skip to next non-digit
Bram Moolenaar071d4272004-06-13 20:20:40 +00001872 ++p;
1873 return p;
1874}
1875
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001876#if defined(FEAT_SYN_HL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001877/*
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001878 * skip over binary digits
1879 */
1880 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001881skipbin(char_u *q)
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001882{
1883 char_u *p = q;
1884
Bram Moolenaarc667da52019-11-30 20:52:27 +01001885 while (vim_isbdigit(*p)) // skip to next non-digit
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001886 ++p;
1887 return p;
1888}
1889
1890/*
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001891 * skip over digits and hex characters
1892 */
1893 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001894skiphex(char_u *q)
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001895{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001896 char_u *p = q;
1897
Bram Moolenaarc667da52019-11-30 20:52:27 +01001898 while (vim_isxdigit(*p)) // skip to next non-digit
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001899 ++p;
1900 return p;
1901}
1902#endif
1903
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001904/*
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001905 * skip to bin digit (or NUL after the string)
1906 */
1907 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001908skiptobin(char_u *q)
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001909{
1910 char_u *p = q;
1911
Bram Moolenaarc667da52019-11-30 20:52:27 +01001912 while (*p != NUL && !vim_isbdigit(*p)) // skip to next digit
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001913 ++p;
1914 return p;
1915}
1916
1917/*
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001918 * skip to digit (or NUL after the string)
1919 */
1920 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001921skiptodigit(char_u *q)
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001922{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001923 char_u *p = q;
1924
Bram Moolenaarc667da52019-11-30 20:52:27 +01001925 while (*p != NUL && !VIM_ISDIGIT(*p)) // skip to next digit
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001926 ++p;
1927 return p;
1928}
1929
1930/*
1931 * skip to hex character (or NUL after the string)
1932 */
1933 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001934skiptohex(char_u *q)
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001935{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001936 char_u *p = q;
1937
Bram Moolenaarc667da52019-11-30 20:52:27 +01001938 while (*p != NUL && !vim_isxdigit(*p)) // skip to next digit
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001939 ++p;
1940 return p;
1941}
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001942
Bram Moolenaar071d4272004-06-13 20:20:40 +00001943/*
1944 * Variant of isdigit() that can handle characters > 0x100.
1945 * We don't use isdigit() here, because on some systems it also considers
1946 * superscript 1 to be a digit.
1947 * Use the VIM_ISDIGIT() macro for simple arguments.
1948 */
1949 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001950vim_isdigit(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951{
1952 return (c >= '0' && c <= '9');
1953}
1954
1955/*
1956 * Variant of isxdigit() that can handle characters > 0x100.
1957 * We don't use isxdigit() here, because on some systems it also considers
1958 * superscript 1 to be a digit.
1959 */
1960 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001961vim_isxdigit(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001962{
1963 return (c >= '0' && c <= '9')
1964 || (c >= 'a' && c <= 'f')
1965 || (c >= 'A' && c <= 'F');
1966}
1967
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001968/*
1969 * Corollary of vim_isdigit and vim_isxdigit() that can handle
1970 * characters > 0x100.
1971 */
1972 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001973vim_isbdigit(int c)
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001974{
1975 return (c == '0' || c == '1');
1976}
1977
Bram Moolenaarc37b6552021-01-07 19:36:30 +01001978 static int
1979vim_isodigit(int c)
1980{
1981 return (c >= '0' && c <= '7');
1982}
1983
Bram Moolenaar78622822005-08-23 21:00:13 +00001984/*
1985 * Vim's own character class functions. These exist because many library
1986 * islower()/toupper() etc. do not work properly: they crash when used with
1987 * invalid values or can't handle latin1 when the locale is C.
1988 * Speed is most important here.
1989 */
1990#define LATIN1LOWER 'l'
1991#define LATIN1UPPER 'U'
1992
Bram Moolenaar6e7c7f32005-08-24 22:16:11 +00001993static char_u latin1flags[257] = " UUUUUUUUUUUUUUUUUUUUUUUUUU llllllllllllllllllllllllll UUUUUUUUUUUUUUUUUUUUUUU UUUUUUUllllllllllllllllllllllll llllllll";
Bram Moolenaar936347b2012-05-25 11:56:22 +02001994static char_u latin1upper[257] = " !\"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`ABCDEFGHIJKLMNOPQRSTUVWXYZ{|}~\x7f\x80\x81\x82\x83\x84\x85\x86\x87\x88\x89\x8a\x8b\x8c\x8d\x8e\x8f\x90\x91\x92\x93\x94\x95\x96\x97\x98\x99\x9a\x9b\x9c\x9d\x9e\x9f\xa0\xa1\xa2\xa3\xa4\xa5\xa6\xa7\xa8\xa9\xaa\xab\xac\xad\xae\xaf\xb0\xb1\xb2\xb3\xb4\xb5\xb6\xb7\xb8\xb9\xba\xbb\xbc\xbd\xbe\xbf\xc0\xc1\xc2\xc3\xc4\xc5\xc6\xc7\xc8\xc9\xca\xcb\xcc\xcd\xce\xcf\xd0\xd1\xd2\xd3\xd4\xd5\xd6\xd7\xd8\xd9\xda\xdb\xdc\xdd\xde\xdf\xc0\xc1\xc2\xc3\xc4\xc5\xc6\xc7\xc8\xc9\xca\xcb\xcc\xcd\xce\xcf\xd0\xd1\xd2\xd3\xd4\xd5\xd6\xf7\xd8\xd9\xda\xdb\xdc\xdd\xde\xff";
1995static char_u latin1lower[257] = " !\"#$%&'()*+,-./0123456789:;<=>?@abcdefghijklmnopqrstuvwxyz[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~\x7f\x80\x81\x82\x83\x84\x85\x86\x87\x88\x89\x8a\x8b\x8c\x8d\x8e\x8f\x90\x91\x92\x93\x94\x95\x96\x97\x98\x99\x9a\x9b\x9c\x9d\x9e\x9f\xa0\xa1\xa2\xa3\xa4\xa5\xa6\xa7\xa8\xa9\xaa\xab\xac\xad\xae\xaf\xb0\xb1\xb2\xb3\xb4\xb5\xb6\xb7\xb8\xb9\xba\xbb\xbc\xbd\xbe\xbf\xe0\xe1\xe2\xe3\xe4\xe5\xe6\xe7\xe8\xe9\xea\xeb\xec\xed\xee\xef\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xd7\xf8\xf9\xfa\xfb\xfc\xfd\xfe\xdf\xe0\xe1\xe2\xe3\xe4\xe5\xe6\xe7\xe8\xe9\xea\xeb\xec\xed\xee\xef\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7\xf8\xf9\xfa\xfb\xfc\xfd\xfe\xff";
Bram Moolenaar78622822005-08-23 21:00:13 +00001996
1997 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001998vim_islower(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00001999{
2000 if (c <= '@')
2001 return FALSE;
2002 if (c >= 0x80)
2003 {
2004 if (enc_utf8)
2005 return utf_islower(c);
2006 if (c >= 0x100)
2007 {
2008#ifdef HAVE_ISWLOWER
2009 if (has_mbyte)
2010 return iswlower(c);
2011#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01002012 // islower() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00002013 return FALSE;
2014 }
2015 if (enc_latin1like)
2016 return (latin1flags[c] & LATIN1LOWER) == LATIN1LOWER;
2017 }
Keith Thompson184f71c2024-01-04 21:19:04 +01002018 return SAFE_islower(c);
Bram Moolenaar78622822005-08-23 21:00:13 +00002019}
2020
2021 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002022vim_isupper(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00002023{
2024 if (c <= '@')
2025 return FALSE;
2026 if (c >= 0x80)
2027 {
2028 if (enc_utf8)
2029 return utf_isupper(c);
2030 if (c >= 0x100)
2031 {
2032#ifdef HAVE_ISWUPPER
2033 if (has_mbyte)
2034 return iswupper(c);
2035#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01002036 // islower() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00002037 return FALSE;
2038 }
2039 if (enc_latin1like)
2040 return (latin1flags[c] & LATIN1UPPER) == LATIN1UPPER;
2041 }
Keith Thompson184f71c2024-01-04 21:19:04 +01002042 return SAFE_isupper(c);
Bram Moolenaar78622822005-08-23 21:00:13 +00002043}
2044
2045 int
Bram Moolenaar5921aeb2022-02-19 11:20:12 +00002046vim_isalpha(int c)
2047{
2048 return vim_islower(c) || vim_isupper(c);
2049}
2050
2051 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002052vim_toupper(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00002053{
2054 if (c <= '@')
2055 return c;
Bram Moolenaar3317d5e2017-04-08 19:12:06 +02002056 if (c >= 0x80 || !(cmp_flags & CMP_KEEPASCII))
Bram Moolenaar78622822005-08-23 21:00:13 +00002057 {
2058 if (enc_utf8)
2059 return utf_toupper(c);
2060 if (c >= 0x100)
2061 {
2062#ifdef HAVE_TOWUPPER
2063 if (has_mbyte)
2064 return towupper(c);
2065#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01002066 // toupper() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00002067 return c;
2068 }
2069 if (enc_latin1like)
2070 return latin1upper[c];
2071 }
Bram Moolenaar1cc48202017-04-09 13:41:59 +02002072 if (c < 0x80 && (cmp_flags & CMP_KEEPASCII))
2073 return TOUPPER_ASC(c);
Bram Moolenaar78622822005-08-23 21:00:13 +00002074 return TOUPPER_LOC(c);
2075}
2076
2077 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002078vim_tolower(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00002079{
2080 if (c <= '@')
2081 return c;
Bram Moolenaar3317d5e2017-04-08 19:12:06 +02002082 if (c >= 0x80 || !(cmp_flags & CMP_KEEPASCII))
Bram Moolenaar78622822005-08-23 21:00:13 +00002083 {
2084 if (enc_utf8)
2085 return utf_tolower(c);
2086 if (c >= 0x100)
2087 {
2088#ifdef HAVE_TOWLOWER
2089 if (has_mbyte)
2090 return towlower(c);
2091#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01002092 // tolower() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00002093 return c;
2094 }
2095 if (enc_latin1like)
2096 return latin1lower[c];
2097 }
Bram Moolenaar1cc48202017-04-09 13:41:59 +02002098 if (c < 0x80 && (cmp_flags & CMP_KEEPASCII))
2099 return TOLOWER_ASC(c);
Bram Moolenaar78622822005-08-23 21:00:13 +00002100 return TOLOWER_LOC(c);
2101}
Bram Moolenaar78622822005-08-23 21:00:13 +00002102
Bram Moolenaar071d4272004-06-13 20:20:40 +00002103/*
2104 * skiptowhite: skip over text until ' ' or '\t' or NUL.
2105 */
2106 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01002107skiptowhite(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002108{
2109 while (*p != ' ' && *p != '\t' && *p != NUL)
2110 ++p;
2111 return p;
2112}
2113
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114/*
2115 * skiptowhite_esc: Like skiptowhite(), but also skip escaped chars
2116 */
2117 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01002118skiptowhite_esc(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002119{
2120 while (*p != ' ' && *p != '\t' && *p != NUL)
2121 {
2122 if ((*p == '\\' || *p == Ctrl_V) && *(p + 1) != NUL)
2123 ++p;
2124 ++p;
2125 }
2126 return p;
2127}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002128
2129/*
Bram Moolenaaraf377e32021-11-29 12:12:43 +00002130 * Get a number from a string and skip over it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131 * Note: the argument is a pointer to a char_u pointer!
2132 */
2133 long
Bram Moolenaar7454a062016-01-30 15:14:10 +01002134getdigits(char_u **pp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002135{
2136 char_u *p;
2137 long retval;
2138
2139 p = *pp;
2140 retval = atol((char *)p);
Bram Moolenaarc667da52019-11-30 20:52:27 +01002141 if (*p == '-') // skip negative sign
Bram Moolenaar071d4272004-06-13 20:20:40 +00002142 ++p;
Bram Moolenaarc667da52019-11-30 20:52:27 +01002143 p = skipdigits(p); // skip to next non-digit
Bram Moolenaar071d4272004-06-13 20:20:40 +00002144 *pp = p;
2145 return retval;
2146}
2147
2148/*
Bram Moolenaaraf377e32021-11-29 12:12:43 +00002149 * Like getdigits() but allow for embedded single quotes.
2150 */
2151 long
2152getdigits_quoted(char_u **pp)
2153{
2154 char_u *p = *pp;
2155 long retval = 0;
2156
2157 if (*p == '-')
2158 ++p;
2159 while (VIM_ISDIGIT(*p))
2160 {
2161 if (retval >= LONG_MAX / 10 - 10)
2162 retval = LONG_MAX;
2163 else
2164 retval = retval * 10 - '0' + *p;
2165 ++p;
2166 if (in_vim9script() && *p == '\'' && VIM_ISDIGIT(p[1]))
2167 ++p;
2168 }
2169 if (**pp == '-')
2170 {
2171 if (retval == LONG_MAX)
2172 retval = LONG_MIN;
2173 else
2174 retval = -retval;
2175 }
2176 *pp = p;
2177 return retval;
2178}
2179
2180/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002181 * Return TRUE if "lbuf" is empty or only contains blanks.
2182 */
2183 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002184vim_isblankline(char_u *lbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002185{
2186 char_u *p;
2187
2188 p = skipwhite(lbuf);
2189 return (*p == NUL || *p == '\r' || *p == '\n');
2190}
2191
2192/*
2193 * Convert a string into a long and/or unsigned long, taking care of
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002194 * hexadecimal, octal, and binary numbers. Accepts a '-' sign.
2195 * If "prep" is not NULL, returns a flag to indicate the type of the number:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002196 * 0 decimal
2197 * '0' octal
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002198 * 'O' octal
2199 * 'o' octal
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002200 * 'B' bin
2201 * 'b' bin
Bram Moolenaar071d4272004-06-13 20:20:40 +00002202 * 'X' hex
2203 * 'x' hex
2204 * If "len" is not NULL, the length of the number in characters is returned.
2205 * If "nptr" is not NULL, the signed result is returned in it.
2206 * If "unptr" is not NULL, the unsigned result is returned in it.
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002207 * If "what" contains STR2NR_BIN recognize binary numbers
2208 * If "what" contains STR2NR_OCT recognize octal numbers
2209 * If "what" contains STR2NR_HEX recognize hex numbers
2210 * If "what" contains STR2NR_FORCE always assume bin/oct/hex.
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002211 * If "what" contains STR2NR_QUOTE ignore embedded single quotes
Bram Moolenaarce157752017-10-28 16:07:33 +02002212 * If maxlen > 0, check at a maximum maxlen chars.
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002213 * If strict is TRUE, check the number strictly. return *len = 0 if fail.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002214 */
2215 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002216vim_str2nr(
2217 char_u *start,
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002218 int *prep, // return: type of number 0 = decimal, 'x'
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002219 // or 'X' is hex, '0', 'o' or 'O' is octal,
2220 // 'b' or 'B' is bin
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002221 int *len, // return: detected length of number
2222 int what, // what numbers to recognize
2223 varnumber_T *nptr, // return: signed result
2224 uvarnumber_T *unptr, // return: unsigned result
2225 int maxlen, // max length of string to check
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002226 int strict, // check strictly
2227 int *overflow) // when not NULL set to TRUE for overflow
Bram Moolenaar071d4272004-06-13 20:20:40 +00002228{
2229 char_u *ptr = start;
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002230 int pre = 0; // default is decimal
Bram Moolenaar071d4272004-06-13 20:20:40 +00002231 int negative = FALSE;
Bram Moolenaar22fcfad2016-07-01 18:17:26 +02002232 uvarnumber_T un = 0;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002233 int n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002235 if (len != NULL)
2236 *len = 0;
2237
Bram Moolenaar071d4272004-06-13 20:20:40 +00002238 if (ptr[0] == '-')
2239 {
2240 negative = TRUE;
2241 ++ptr;
2242 }
2243
Bram Moolenaarc667da52019-11-30 20:52:27 +01002244 // Recognize hex, octal, and bin.
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002245 if (ptr[0] == '0' && ptr[1] != '8' && ptr[1] != '9'
2246 && (maxlen == 0 || maxlen > 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002247 {
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002248 pre = ptr[1];
2249 if ((what & STR2NR_HEX)
2250 && (pre == 'X' || pre == 'x') && vim_isxdigit(ptr[2])
2251 && (maxlen == 0 || maxlen > 2))
Bram Moolenaarc667da52019-11-30 20:52:27 +01002252 // hexadecimal
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002253 ptr += 2;
2254 else if ((what & STR2NR_BIN)
2255 && (pre == 'B' || pre == 'b') && vim_isbdigit(ptr[2])
2256 && (maxlen == 0 || maxlen > 2))
Bram Moolenaarc667da52019-11-30 20:52:27 +01002257 // binary
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002258 ptr += 2;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002259 else if ((what & STR2NR_OOCT)
Bram Moolenaarc37b6552021-01-07 19:36:30 +01002260 && (pre == 'O' || pre == 'o') && vim_isodigit(ptr[2])
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002261 && (maxlen == 0 || maxlen > 2))
2262 // octal with prefix "0o"
2263 ptr += 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002264 else
2265 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002266 // decimal or octal, default is decimal
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002267 pre = 0;
2268 if (what & STR2NR_OCT)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002269 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002270 // Don't interpret "0", "08" or "0129" as octal.
Bram Moolenaarce157752017-10-28 16:07:33 +02002271 for (n = 1; n != maxlen && VIM_ISDIGIT(ptr[n]); ++n)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002272 {
2273 if (ptr[n] > '7')
2274 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002275 pre = 0; // can't be octal
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002276 break;
2277 }
Bram Moolenaarc667da52019-11-30 20:52:27 +01002278 pre = '0'; // assume octal
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002279 }
2280 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281 }
2282 }
2283
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002284 // Do the conversion manually to avoid sscanf() quirks.
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002285 n = 1;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002286 if (pre == 'B' || pre == 'b'
2287 || ((what & STR2NR_BIN) && (what & STR2NR_FORCE)))
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002288 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002289 // bin
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002290 if (pre != 0)
Bram Moolenaarc667da52019-11-30 20:52:27 +01002291 n += 2; // skip over "0b"
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002292 while ('0' <= *ptr && *ptr <= '1')
2293 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002294 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002295 if (un <= UVARNUM_MAX / 2)
2296 un = 2 * un + (uvarnumber_T)(*ptr - '0');
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002297 else
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002298 {
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002299 un = UVARNUM_MAX;
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002300 if (overflow != NULL)
2301 *overflow = TRUE;
2302 }
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002303 ++ptr;
2304 if (n++ == maxlen)
2305 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002306 if ((what & STR2NR_QUOTE) && *ptr == '\''
2307 && '0' <= ptr[1] && ptr[1] <= '1')
2308 {
2309 ++ptr;
2310 if (n++ == maxlen)
2311 break;
2312 }
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002313 }
2314 }
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002315 else if (pre == 'O' || pre == 'o' ||
2316 pre == '0' || ((what & STR2NR_OCT) && (what & STR2NR_FORCE)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002317 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002318 // octal
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002319 if (pre != 0 && pre != '0')
2320 n += 2; // skip over "0o"
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002321 while ('0' <= *ptr && *ptr <= '7')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002322 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002323 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002324 if (un <= UVARNUM_MAX / 8)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002325 un = 8 * un + (uvarnumber_T)(*ptr - '0');
2326 else
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002327 {
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002328 un = UVARNUM_MAX;
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002329 if (overflow != NULL)
2330 *overflow = TRUE;
2331 }
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002332 ++ptr;
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002333 if (n++ == maxlen)
2334 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002335 if ((what & STR2NR_QUOTE) && *ptr == '\''
2336 && '0' <= ptr[1] && ptr[1] <= '7')
2337 {
2338 ++ptr;
2339 if (n++ == maxlen)
2340 break;
2341 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002342 }
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002343 }
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002344 else if (pre != 0 || ((what & STR2NR_HEX) && (what & STR2NR_FORCE)))
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002345 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002346 // hex
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002347 if (pre != 0)
Bram Moolenaarc667da52019-11-30 20:52:27 +01002348 n += 2; // skip over "0x"
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002349 while (vim_isxdigit(*ptr))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002350 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002351 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002352 if (un <= UVARNUM_MAX / 16)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002353 un = 16 * un + (uvarnumber_T)hex2nr(*ptr);
2354 else
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002355 {
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002356 un = UVARNUM_MAX;
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002357 if (overflow != NULL)
2358 *overflow = TRUE;
2359 }
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002360 ++ptr;
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002361 if (n++ == maxlen)
2362 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002363 if ((what & STR2NR_QUOTE) && *ptr == '\'' && vim_isxdigit(ptr[1]))
2364 {
2365 ++ptr;
2366 if (n++ == maxlen)
2367 break;
2368 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002369 }
2370 }
2371 else
2372 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002373 // decimal
Bram Moolenaar071d4272004-06-13 20:20:40 +00002374 while (VIM_ISDIGIT(*ptr))
2375 {
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002376 uvarnumber_T digit = (uvarnumber_T)(*ptr - '0');
2377
Bram Moolenaarc667da52019-11-30 20:52:27 +01002378 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002379 if (un < UVARNUM_MAX / 10
2380 || (un == UVARNUM_MAX / 10 && digit <= UVARNUM_MAX % 10))
2381 un = 10 * un + digit;
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002382 else
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002383 {
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002384 un = UVARNUM_MAX;
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002385 if (overflow != NULL)
2386 *overflow = TRUE;
2387 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002388 ++ptr;
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002389 if (n++ == maxlen)
2390 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002391 if ((what & STR2NR_QUOTE) && *ptr == '\'' && VIM_ISDIGIT(ptr[1]))
2392 {
2393 ++ptr;
2394 if (n++ == maxlen)
2395 break;
2396 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002397 }
2398 }
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002399
Bram Moolenaar4b96df52020-01-26 22:00:26 +01002400 // Check for an alphanumeric character immediately following, that is
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002401 // most likely a typo.
2402 if (strict && n - 1 != maxlen && ASCII_ISALNUM(*ptr))
2403 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002405 if (prep != NULL)
2406 *prep = pre;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002407 if (len != NULL)
2408 *len = (int)(ptr - start);
2409 if (nptr != NULL)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002410 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002411 if (negative) // account for leading '-' for decimal numbers
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002412 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002413 // avoid ubsan error for overflow
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002414 if (un > VARNUM_MAX)
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002415 {
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002416 *nptr = VARNUM_MIN;
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002417 if (overflow != NULL)
2418 *overflow = TRUE;
2419 }
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002420 else
2421 *nptr = -(varnumber_T)un;
2422 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002423 else
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002424 {
dundargocc57b5bc2022-11-02 13:30:51 +00002425 // prevent a large unsigned number to become negative
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002426 if (un > VARNUM_MAX)
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002427 {
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002428 un = VARNUM_MAX;
Bram Moolenaar5fb78c32023-03-04 20:47:39 +00002429 if (overflow != NULL)
2430 *overflow = TRUE;
2431 }
Bram Moolenaar22fcfad2016-07-01 18:17:26 +02002432 *nptr = (varnumber_T)un;
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002433 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002434 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002435 if (unptr != NULL)
2436 *unptr = un;
2437}
2438
2439/*
2440 * Return the value of a single hex character.
2441 * Only valid when the argument is '0' - '9', 'A' - 'F' or 'a' - 'f'.
2442 */
2443 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002444hex2nr(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002445{
2446 if (c >= 'a' && c <= 'f')
2447 return c - 'a' + 10;
2448 if (c >= 'A' && c <= 'F')
2449 return c - 'A' + 10;
2450 return c - '0';
2451}
2452
Bram Moolenaar071d4272004-06-13 20:20:40 +00002453/*
2454 * Convert two hex characters to a byte.
2455 * Return -1 if one of the characters is not hex.
2456 */
2457 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002458hexhex2nr(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002459{
2460 if (!vim_isxdigit(p[0]) || !vim_isxdigit(p[1]))
2461 return -1;
2462 return (hex2nr(p[0]) << 4) + hex2nr(p[1]);
2463}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002464
2465/*
2466 * Return TRUE if "str" starts with a backslash that should be removed.
Bram Moolenaar2c519cf2019-03-21 21:45:34 +01002467 * For MS-DOS, MSWIN and OS/2 this is only done when the character after the
Bram Moolenaar071d4272004-06-13 20:20:40 +00002468 * backslash is not a normal file name character.
2469 * '$' is a valid file name character, we don't remove the backslash before
2470 * it. This means it is not possible to use an environment variable after a
2471 * backslash. "C:\$VIM\doc" is taken literally, only "$VIM\doc" works.
2472 * Although "\ name" is valid, the backslash in "Program\ files" must be
2473 * removed. Assume a file name doesn't start with a space.
2474 * For multi-byte names, never remove a backslash before a non-ascii
2475 * character, assume that all multi-byte characters are valid file name
2476 * characters.
2477 */
2478 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002479rem_backslash(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002480{
2481#ifdef BACKSLASH_IN_FILENAME
2482 return (str[0] == '\\'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002483 && str[1] < 0x80
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 && (str[1] == ' '
2485 || (str[1] != NUL
2486 && str[1] != '*'
2487 && str[1] != '?'
2488 && !vim_isfilec(str[1]))));
2489#else
2490 return (str[0] == '\\' && str[1] != NUL);
2491#endif
2492}
2493
2494/*
2495 * Halve the number of backslashes in a file name argument.
2496 * For MS-DOS we only do this if the character after the backslash
2497 * is not a normal file character.
2498 */
2499 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002500backslash_halve(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002501{
2502 for ( ; *p; ++p)
2503 if (rem_backslash(p))
Bram Moolenaar446cb832008-06-24 21:56:24 +00002504 STRMOVE(p, p + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002505}
2506
2507/*
2508 * backslash_halve() plus save the result in allocated memory.
Bram Moolenaare2c453d2019-08-21 14:37:09 +02002509 * However, returns "p" when out of memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002510 */
2511 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01002512backslash_halve_save(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002513{
2514 char_u *res;
2515
2516 res = vim_strsave(p);
2517 if (res == NULL)
2518 return p;
2519 backslash_halve(res);
2520 return res;
2521}