blob: 32539bf7acaba6614af728fcf94a9b72dc162803 [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);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000348 if (res != NULL)
349 {
350 *res = NUL;
351 p = s;
352 while (*p != NUL)
353 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000354 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000355 {
356 c = (*mb_ptr2char)(p);
357 if (vim_isprintc(c))
Bram Moolenaarc667da52019-11-30 20:52:27 +0100358 STRNCAT(res, p, l); // append printable multi-byte char
Bram Moolenaar071d4272004-06-13 20:20:40 +0000359 else
360 transchar_hex(res + STRLEN(res), c);
361 p += l;
362 }
363 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000364 STRCAT(res, transchar_byte(*p++));
365 }
366 }
367 return res;
368}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000369
Bram Moolenaar071d4272004-06-13 20:20:40 +0000370/*
Bram Moolenaar217ad922005-03-20 22:37:15 +0000371 * Convert the string "str[orglen]" to do ignore-case comparing. Uses the
372 * current locale.
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000373 * When "buf" is NULL returns an allocated string (NULL for out-of-memory).
374 * Otherwise puts the result in "buf[buflen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000375 */
376 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100377str_foldcase(
378 char_u *str,
379 int orglen,
380 char_u *buf,
381 int buflen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000382{
383 garray_T ga;
384 int i;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000385 int len = orglen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000386
387#define GA_CHAR(i) ((char_u *)ga.ga_data)[i]
kylo252ae6f1d82022-02-16 19:24:07 +0000388#define GA_PTR(i) ((char_u *)ga.ga_data + (i))
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000389#define STR_CHAR(i) (buf == NULL ? GA_CHAR(i) : buf[i])
kylo252ae6f1d82022-02-16 19:24:07 +0000390#define STR_PTR(i) (buf == NULL ? GA_PTR(i) : buf + (i))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000391
Bram Moolenaarc667da52019-11-30 20:52:27 +0100392 // Copy "str" into "buf" or allocated memory, unmodified.
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000393 if (buf == NULL)
394 {
395 ga_init2(&ga, 1, 10);
396 if (ga_grow(&ga, len + 1) == FAIL)
397 return NULL;
398 mch_memmove(ga.ga_data, str, (size_t)len);
399 ga.ga_len = len;
400 }
401 else
402 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100403 if (len >= buflen) // Ugly!
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000404 len = buflen - 1;
405 mch_memmove(buf, str, (size_t)len);
406 }
407 if (buf == NULL)
408 GA_CHAR(len) = NUL;
409 else
410 buf[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000411
Bram Moolenaarc667da52019-11-30 20:52:27 +0100412 // Make each character lower case.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000413 i = 0;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000414 while (STR_CHAR(i) != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000415 {
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000416 if (enc_utf8 || (has_mbyte && MB_BYTE2LEN(STR_CHAR(i)) > 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000417 {
418 if (enc_utf8)
419 {
Bram Moolenaarb9839212008-06-28 11:03:50 +0000420 int c = utf_ptr2char(STR_PTR(i));
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100421 int olen = utf_ptr2len(STR_PTR(i));
Bram Moolenaarb9839212008-06-28 11:03:50 +0000422 int lc = utf_tolower(c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423
Bram Moolenaarc667da52019-11-30 20:52:27 +0100424 // Only replace the character when it is not an invalid
425 // sequence (ASCII character or more than one byte) and
426 // utf_tolower() doesn't return the original character.
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100427 if ((c < 0x80 || olen > 1) && c != lc)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000428 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100429 int nlen = utf_char2len(lc);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000430
Bram Moolenaarc667da52019-11-30 20:52:27 +0100431 // If the byte length changes need to shift the following
432 // characters forward or backward.
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100433 if (olen != nlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000434 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100435 if (nlen > olen)
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000436 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100437 if (buf == NULL
438 ? ga_grow(&ga, nlen - olen + 1) == FAIL
439 : len + nlen - olen >= buflen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000440 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100441 // out of memory, keep old char
Bram Moolenaar071d4272004-06-13 20:20:40 +0000442 lc = c;
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100443 nlen = olen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000444 }
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000445 }
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100446 if (olen != nlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000447 {
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000448 if (buf == NULL)
449 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100450 STRMOVE(GA_PTR(i) + nlen, GA_PTR(i) + olen);
451 ga.ga_len += nlen - olen;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000452 }
453 else
454 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +0100455 STRMOVE(buf + i + nlen, buf + i + olen);
456 len += nlen - olen;
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000457 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000458 }
459 }
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000460 (void)utf_char2bytes(lc, STR_PTR(i));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000461 }
462 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100463 // skip to next multi-byte char
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000464 i += (*mb_ptr2len)(STR_PTR(i));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000465 }
466 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000467 {
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000468 if (buf == NULL)
469 GA_CHAR(i) = TOLOWER_LOC(GA_CHAR(i));
470 else
471 buf[i] = TOLOWER_LOC(buf[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000472 ++i;
473 }
474 }
475
Bram Moolenaar6ebb1142005-01-25 21:58:26 +0000476 if (buf == NULL)
477 return (char_u *)ga.ga_data;
478 return buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000479}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000480
481/*
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100482 * Catch 22: g_chartab[] can't be initialized before the options are
Bram Moolenaar071d4272004-06-13 20:20:40 +0000483 * initialized, and initializing options may cause transchar() to be called!
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100484 * When chartab_initialized == FALSE don't use g_chartab[].
Bram Moolenaar071d4272004-06-13 20:20:40 +0000485 * Does NOT work for multi-byte characters, c must be <= 255.
486 * Also doesn't work for the first byte of a multi-byte, "c" must be a
487 * character!
488 */
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200489static char_u transchar_charbuf[7];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000490
491 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100492transchar(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000493{
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200494 return transchar_buf(curbuf, c);
495}
496
497 char_u *
498transchar_buf(buf_T *buf, int c)
499{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500 int i;
501
502 i = 0;
Bram Moolenaarc667da52019-11-30 20:52:27 +0100503 if (IS_SPECIAL(c)) // special key code, display as ~@ char
Bram Moolenaar071d4272004-06-13 20:20:40 +0000504 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200505 transchar_charbuf[0] = '~';
506 transchar_charbuf[1] = '@';
Bram Moolenaar071d4272004-06-13 20:20:40 +0000507 i = 2;
508 c = K_SECOND(c);
509 }
510
Bram Moolenaar424bcae2022-01-31 14:59:41 +0000511 if ((!chartab_initialized && ((c >= ' ' && c <= '~')))
512 || (c < 256 && vim_isprintc_strict(c)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000513 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100514 // printable character
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200515 transchar_charbuf[i] = c;
516 transchar_charbuf[i + 1] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000517 }
518 else
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200519 transchar_nonprint(buf, transchar_charbuf + i, c);
520 return transchar_charbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000521}
522
Bram Moolenaar071d4272004-06-13 20:20:40 +0000523/*
524 * Like transchar(), but called with a byte instead of a character. Checks
525 * for an illegal UTF-8 byte.
526 */
527 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100528transchar_byte(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529{
530 if (enc_utf8 && c >= 0x80)
531 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200532 transchar_nonprint(curbuf, transchar_charbuf, c);
533 return transchar_charbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534 }
535 return transchar(c);
536}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000537
538/*
539 * Convert non-printable character to two or more printable characters in
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200540 * "buf[]". "charbuf" needs to be able to hold five bytes.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000541 * Does NOT work for multi-byte characters, c must be <= 255.
542 */
543 void
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200544transchar_nonprint(buf_T *buf, char_u *charbuf, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545{
546 if (c == NL)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100547 c = NUL; // we use newline in place of a NUL
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200548 else if (c == CAR && get_fileformat(buf) == EOL_MAC)
Bram Moolenaarc667da52019-11-30 20:52:27 +0100549 c = NL; // we use CR in place of NL in this case
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550
Bram Moolenaarc667da52019-11-30 20:52:27 +0100551 if (dy_flags & DY_UHEX) // 'display' has "uhex"
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200552 transchar_hex(charbuf, c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000553
Bram Moolenaarc667da52019-11-30 20:52:27 +0100554 else if (c <= 0x7f) // 0x00 - 0x1f and 0x7f
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200556 charbuf[0] = '^';
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200557 charbuf[1] = c ^ 0x40; // DEL displayed as ^?
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200558 charbuf[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000559 }
=?UTF-8?q?Dundar=20G=C3=B6c?=f01a6532022-03-09 13:00:54 +0000560 else if (enc_utf8)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561 {
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 >= ' ' + 0x80 && c <= '~' + 0x80) // 0xa0 - 0xfe
Bram Moolenaar071d4272004-06-13 20:20:40 +0000565 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200566 charbuf[0] = '|';
567 charbuf[1] = c - 0x80;
568 charbuf[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000569 }
Bram Moolenaarc667da52019-11-30 20:52:27 +0100570 else // 0x80 - 0x9f and 0xff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000571 {
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200572 charbuf[0] = '~';
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200573 charbuf[1] = (c - 0x80) ^ 0x40; // 0xff displayed as ~?
Bram Moolenaar32ee6272020-06-10 14:16:49 +0200574 charbuf[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000575 }
576}
577
578 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100579transchar_hex(char_u *buf, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000580{
581 int i = 0;
582
583 buf[0] = '<';
Bram Moolenaar071d4272004-06-13 20:20:40 +0000584 if (c > 255)
585 {
586 buf[++i] = nr2hex((unsigned)c >> 12);
587 buf[++i] = nr2hex((unsigned)c >> 8);
588 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000589 buf[++i] = nr2hex((unsigned)c >> 4);
Bram Moolenaar0ab2a882009-05-13 10:51:08 +0000590 buf[++i] = nr2hex((unsigned)c);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000591 buf[++i] = '>';
592 buf[++i] = NUL;
593}
594
595/*
596 * Convert the lower 4 bits of byte "c" to its hex character.
597 * Lower case letters are used to avoid the confusion of <F1> being 0xf1 or
598 * function key 1.
599 */
Bram Moolenaar0ab2a882009-05-13 10:51:08 +0000600 static unsigned
Bram Moolenaar7454a062016-01-30 15:14:10 +0100601nr2hex(unsigned c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000602{
603 if ((c & 0xf) <= 9)
604 return (c & 0xf) + '0';
605 return (c & 0xf) - 10 + 'a';
606}
607
608/*
609 * Return number of display cells occupied by byte "b".
610 * Caller must make sure 0 <= b <= 255.
611 * For multi-byte mode "b" must be the first byte of a character.
612 * A TAB is counted as two cells: "^I".
613 * For UTF-8 mode this will return 0 for bytes >= 0x80, because the number of
614 * cells depends on further bytes.
615 */
616 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100617byte2cells(int b)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000618{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000619 if (enc_utf8 && b >= 0x80)
620 return 0;
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100621 return (g_chartab[b] & CT_CELL_MASK);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000622}
623
624/*
625 * Return number of display cells occupied by character "c".
626 * "c" can be a special key (negative number) in which case 3 or 4 is returned.
627 * A TAB is counted as two cells: "^I" or four: "<09>".
628 */
629 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100630char2cells(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631{
632 if (IS_SPECIAL(c))
633 return char2cells(K_SECOND(c)) + 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 if (c >= 0x80)
635 {
Bram Moolenaarc667da52019-11-30 20:52:27 +0100636 // UTF-8: above 0x80 need to check the value
Bram Moolenaar071d4272004-06-13 20:20:40 +0000637 if (enc_utf8)
638 return utf_char2cells(c);
Bram Moolenaarc667da52019-11-30 20:52:27 +0100639 // DBCS: double-byte means double-width, except for euc-jp with first
640 // byte 0x8e
Bram Moolenaar071d4272004-06-13 20:20:40 +0000641 if (enc_dbcs != 0 && c >= 0x100)
642 {
643 if (enc_dbcs == DBCS_JPNU && ((unsigned)c >> 8) == 0x8e)
644 return 1;
645 return 2;
646 }
647 }
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100648 return (g_chartab[c & 0xff] & CT_CELL_MASK);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000649}
650
651/*
652 * Return number of display cells occupied by character at "*p".
653 * A TAB is counted as two cells: "^I" or four: "<09>".
654 */
655 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100656ptr2cells(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000657{
Bram Moolenaar398649e2022-08-04 15:03:48 +0100658 if (!has_mbyte)
659 return byte2cells(*p);
660
Bram Moolenaarc667da52019-11-30 20:52:27 +0100661 // For UTF-8 we need to look at more bytes if the first byte is >= 0x80.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000662 if (enc_utf8 && *p >= 0x80)
663 return utf_ptr2cells(p);
Bram Moolenaarc667da52019-11-30 20:52:27 +0100664 // For DBCS we can tell the cell count from the first byte.
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100665 return (g_chartab[*p] & CT_CELL_MASK);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000666}
667
668/*
Bram Moolenaar06af6022012-01-26 13:40:08 +0100669 * Return the number of character cells string "s" will take on the screen,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 * counting TABs as two characters: "^I".
671 */
672 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100673vim_strsize(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000674{
675 return vim_strnsize(s, (int)MAXCOL);
676}
677
678/*
Bram Moolenaar06af6022012-01-26 13:40:08 +0100679 * Return the number of character cells string "s[len]" will take on the
680 * screen, counting TABs as two characters: "^I".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000681 */
682 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100683vim_strnsize(char_u *s, int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000684{
685 int size = 0;
686
687 while (*s != NUL && --len >= 0)
Bram Moolenaar398649e2022-08-04 15:03:48 +0100688 {
689 int l = (*mb_ptr2len)(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000690
Bram Moolenaar398649e2022-08-04 15:03:48 +0100691 size += ptr2cells(s);
692 s += l;
693 len -= l - 1;
694 }
Bram Moolenaar13505972019-01-24 15:04:48 +0100695
Bram Moolenaar071d4272004-06-13 20:20:40 +0000696 return size;
697}
698
699/*
700 * Return the number of characters 'c' will take on the screen, taking
701 * into account the size of a tab.
702 * Use a define to make it fast, this is used very often!!!
703 * Also see getvcol() below.
704 */
705
Bram Moolenaar04958cb2018-06-23 19:23:02 +0200706#ifdef FEAT_VARTABS
707# define RET_WIN_BUF_CHARTABSIZE(wp, buf, p, col) \
kylo252ae6f1d82022-02-16 19:24:07 +0000708 if (*(p) == TAB && (!(wp)->w_p_list || (wp)->w_lcs_chars.tab1)) \
Bram Moolenaar04958cb2018-06-23 19:23:02 +0200709 { \
710 return tabstop_padding(col, (buf)->b_p_ts, (buf)->b_p_vts_array); \
711 } \
712 else \
713 return ptr2cells(p);
714#else
715# define RET_WIN_BUF_CHARTABSIZE(wp, buf, p, col) \
Bram Moolenaareed9d462021-02-15 20:38:25 +0100716 if (*(p) == TAB && (!(wp)->w_p_list || wp->w_lcs_chars.tab1)) \
Bram Moolenaar071d4272004-06-13 20:20:40 +0000717 { \
718 int ts; \
719 ts = (buf)->b_p_ts; \
720 return (int)(ts - (col % ts)); \
721 } \
722 else \
723 return ptr2cells(p);
Bram Moolenaar04958cb2018-06-23 19:23:02 +0200724#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000725
Bram Moolenaar071d4272004-06-13 20:20:40 +0000726 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100727chartabsize(char_u *p, colnr_T col)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000728{
729 RET_WIN_BUF_CHARTABSIZE(curwin, curbuf, p, col)
730}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000731
732#ifdef FEAT_LINEBREAK
733 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100734win_chartabsize(win_T *wp, char_u *p, colnr_T col)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000735{
736 RET_WIN_BUF_CHARTABSIZE(wp, wp->w_buffer, p, col)
737}
738#endif
739
740/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100741 * Return the number of characters the string "s" will take on the screen,
Bram Moolenaardc536092010-07-18 15:45:49 +0200742 * taking into account the size of a tab.
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100743 * Does not handle text properties, since "s" is not a buffer line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000744 */
745 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100746linetabsize(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000747{
Bram Moolenaardc536092010-07-18 15:45:49 +0200748 return linetabsize_col(0, s);
749}
750
751/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100752 * Like linetabsize(), but "s" starts at column "startcol".
Bram Moolenaardc536092010-07-18 15:45:49 +0200753 */
754 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100755linetabsize_col(int startcol, char_u *s)
Bram Moolenaardc536092010-07-18 15:45:49 +0200756{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100757 chartabsize_T cts;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100759 init_chartabsize_arg(&cts, curwin, 0, startcol, s, s);
760 while (*cts.cts_ptr != NUL)
761 cts.cts_vcol += lbr_chartabsize_adv(&cts);
Bram Moolenaar49a90792022-08-09 18:25:23 +0100762#ifdef FEAT_PROP_POPUP
763 if (cts.cts_has_prop_with_text && cts.cts_ptr == cts.cts_line)
764 {
765 // check for virtual text in an empty line
766 (void)lbr_chartabsize_adv(&cts);
767 cts.cts_vcol += cts.cts_cur_text_width;
768 }
769#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100770 clear_chartabsize_arg(&cts);
771 return (int)cts.cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000772}
773
774/*
775 * Like linetabsize(), but for a given window instead of the current one.
776 */
777 int
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100778win_linetabsize(win_T *wp, linenr_T lnum, char_u *line, colnr_T len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000779{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100780 chartabsize_T cts;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000781
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100782 init_chartabsize_arg(&cts, wp, lnum, 0, line, line);
Bram Moolenaar49a90792022-08-09 18:25:23 +0100783 win_linetabsize_cts(&cts, len);
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
Bram Moolenaar49a90792022-08-09 18:25:23 +0100788 void
789win_linetabsize_cts(chartabsize_T *cts, colnr_T len)
790{
791#ifdef FEAT_PROP_POPUP
792 cts->cts_with_trailing = len == MAXCOL;
793#endif
794 for ( ; *cts->cts_ptr != NUL && (len == MAXCOL || cts->cts_ptr < cts->cts_line + len);
795 MB_PTR_ADV(cts->cts_ptr))
796 cts->cts_vcol += win_lbr_chartabsize(cts, NULL);
797#ifdef FEAT_PROP_POPUP
798 // check for a virtual text on an empty line
799 if (cts->cts_has_prop_with_text && *cts->cts_ptr == NUL
800 && cts->cts_ptr == cts->cts_line)
801 {
802 (void)win_lbr_chartabsize(cts, NULL);
803 cts->cts_vcol += cts->cts_cur_text_width;
804 }
805#endif
806}
807
Bram Moolenaar071d4272004-06-13 20:20:40 +0000808/*
Bram Moolenaar81695252004-12-29 20:58:21 +0000809 * Return TRUE if 'c' is a normal identifier character:
810 * Letters and characters from the 'isident' option.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000811 */
812 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100813vim_isIDc(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000814{
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100815 return (c > 0 && c < 0x100 && (g_chartab[c] & CT_ID_CHAR));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000816}
817
818/*
Bram Moolenaare3d1f4c2021-04-06 20:21:59 +0200819 * Like vim_isIDc() but not using the 'isident' option: letters, numbers and
820 * underscore.
821 */
822 int
823vim_isNormalIDc(int c)
824{
825 return ASCII_ISALNUM(c) || c == '_';
826}
827
828/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000829 * return TRUE if 'c' is a keyword character: Letters and characters from
Bram Moolenaarcaa55b62017-01-10 13:51:09 +0100830 * 'iskeyword' option for the current buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831 * For multi-byte characters mb_get_class() is used (builtin rules).
832 */
833 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100834vim_iswordc(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835{
Bram Moolenaar9d182dd2013-01-23 15:53:15 +0100836 return vim_iswordc_buf(c, curbuf);
837}
838
839 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100840vim_iswordc_buf(int c, buf_T *buf)
Bram Moolenaar9d182dd2013-01-23 15:53:15 +0100841{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000842 if (c >= 0x100)
843 {
844 if (enc_dbcs != 0)
Bram Moolenaar0ab2a882009-05-13 10:51:08 +0000845 return dbcs_class((unsigned)c >> 8, (unsigned)(c & 0xff)) >= 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000846 if (enc_utf8)
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100847 return utf_class_buf(c, buf) >= 2;
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100848 return FALSE;
849 }
850 return (c > 0 && GET_CHARTAB(buf, c) != 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000851}
852
853/*
854 * Just like vim_iswordc() but uses a pointer to the (multi-byte) character.
855 */
856 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100857vim_iswordp(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000858{
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100859 return vim_iswordp_buf(p, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000860}
861
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100863vim_iswordp_buf(char_u *p, buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000864{
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100865 int c = *p;
866
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100867 if (has_mbyte && MB_BYTE2LEN(c) > 1)
868 c = (*mb_ptr2char)(p);
Bram Moolenaar4019cf92017-01-28 16:39:34 +0100869 return vim_iswordc_buf(c, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000870}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000871
872/*
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100873 * Return TRUE if 'c' is a valid file-name character as specified with the
874 * 'isfname' option.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000875 * Assume characters above 0x100 are valid (multi-byte).
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100876 * To be used for commands like "gf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000877 */
878 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100879vim_isfilec(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000880{
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100881 return (c >= 0x100 || (c > 0 && (g_chartab[c] & CT_FNAME_CHAR)));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000882}
883
884/*
Bram Moolenaarbc49c5f2022-08-04 13:01:48 +0100885 * Return TRUE if 'c' is a valid file-name character, including characters left
886 * out of 'isfname' to make "gf" work, such as comma, space, '@', etc.
887 */
888 int
889vim_is_fname_char(int c)
890{
891 return vim_isfilec(c) || c == ',' || c == ' ' || c == '@';
892}
893
894/*
Bram Moolenaardd87969c2007-08-21 13:07:12 +0000895 * return TRUE if 'c' is a valid file-name character or a wildcard character
896 * Assume characters above 0x100 are valid (multi-byte).
897 * Explicitly interpret ']' as a wildcard character as mch_has_wildcard("]")
898 * returns false.
899 */
900 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100901vim_isfilec_or_wc(int c)
Bram Moolenaardd87969c2007-08-21 13:07:12 +0000902{
903 char_u buf[2];
904
905 buf[0] = (char_u)c;
906 buf[1] = NUL;
907 return vim_isfilec(c) || c == ']' || mch_has_wildcard(buf);
908}
909
910/*
Bram Moolenaar3317d5e2017-04-08 19:12:06 +0200911 * Return TRUE if 'c' is a printable character.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000912 * Assume characters above 0x100 are printable (multi-byte), except for
913 * Unicode.
914 */
915 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100916vim_isprintc(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000917{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000918 if (enc_utf8 && c >= 0x100)
919 return utf_printable(c);
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100920 return (c >= 0x100 || (c > 0 && (g_chartab[c] & CT_PRINT_CHAR)));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000921}
922
923/*
924 * Strict version of vim_isprintc(c), don't return TRUE if "c" is the head
925 * byte of a double-byte character.
926 */
927 int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100928vim_isprintc_strict(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000929{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000930 if (enc_dbcs != 0 && c < 0x100 && MB_BYTE2LEN(c) > 1)
931 return FALSE;
932 if (enc_utf8 && c >= 0x100)
933 return utf_printable(c);
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +0100934 return (c >= 0x100 || (c > 0 && (g_chartab[c] & CT_PRINT_CHAR)));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000935}
936
937/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100938 * Prepare the structure passed to chartabsize functions.
939 * "line" is the start of the line, "ptr" is the first relevant character.
940 * When "lnum" is zero do not use text properties that insert text.
941 */
942 void
943init_chartabsize_arg(
944 chartabsize_T *cts,
945 win_T *wp,
946 linenr_T lnum,
947 colnr_T col,
948 char_u *line,
949 char_u *ptr)
950{
Bram Moolenaar3f79b612022-08-01 12:06:40 +0100951 CLEAR_POINTER(cts);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100952 cts->cts_win = wp;
953 cts->cts_lnum = lnum;
954 cts->cts_vcol = col;
955 cts->cts_line = line;
956 cts->cts_ptr = ptr;
957#ifdef FEAT_PROP_POPUP
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100958 if (lnum > 0)
959 {
960 char_u *prop_start;
961
962 cts->cts_text_prop_count = get_text_props(wp->w_buffer, lnum,
963 &prop_start, FALSE);
964 if (cts->cts_text_prop_count > 0)
965 {
966 // Make a copy of the properties, so that they are properly
967 // aligned.
968 cts->cts_text_props = ALLOC_MULT(textprop_T,
969 cts->cts_text_prop_count);
970 if (cts->cts_text_props == NULL)
971 cts->cts_text_prop_count = 0;
972 else
973 {
974 int i;
975
976 mch_memmove(cts->cts_text_props, prop_start,
977 cts->cts_text_prop_count * sizeof(textprop_T));
978 for (i = 0; i < cts->cts_text_prop_count; ++i)
979 if (cts->cts_text_props[i].tp_id < 0)
980 {
981 cts->cts_has_prop_with_text = TRUE;
982 break;
983 }
984 if (!cts->cts_has_prop_with_text)
985 {
986 // won't use the text properties, free them
Bram Moolenaar3f79b612022-08-01 12:06:40 +0100987 VIM_CLEAR(cts->cts_text_props);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100988 cts->cts_text_prop_count = 0;
989 }
990 }
991 }
992 }
993#endif
994}
995
996/*
997 * Free any allocated item in "cts".
998 */
999 void
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001000clear_chartabsize_arg(chartabsize_T *cts UNUSED)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001001{
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001002#ifdef FEAT_PROP_POPUP
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001003 if (cts->cts_text_prop_count > 0)
Bram Moolenaar34a1f772022-07-26 11:20:48 +01001004 {
Bram Moolenaar3f79b612022-08-01 12:06:40 +01001005 VIM_CLEAR(cts->cts_text_props);
1006 cts->cts_text_prop_count = 0;
Bram Moolenaar34a1f772022-07-26 11:20:48 +01001007 }
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001008#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001009}
1010
1011/*
1012 * Like chartabsize(), but also check for line breaks on the screen and text
1013 * properties that insert text.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001014 */
1015 int
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001016lbr_chartabsize(chartabsize_T *cts)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001017{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001018#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
1019 if (1
1020# ifdef FEAT_LINEBREAK
1021 && !curwin->w_p_lbr && *get_showbreak_value(curwin) == NUL
1022 && !curwin->w_p_bri
1023# endif
1024# ifdef FEAT_PROP_POPUP
1025 && !cts->cts_has_prop_with_text
1026#endif
1027 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001028 {
1029#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001030 if (curwin->w_p_wrap)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001031 return win_nolbr_chartabsize(cts, NULL);
1032 RET_WIN_BUF_CHARTABSIZE(curwin, curbuf, cts->cts_ptr, cts->cts_vcol)
1033#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001034 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001035 return win_lbr_chartabsize(cts, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001036#endif
1037}
1038
1039/*
1040 * Call lbr_chartabsize() and advance the pointer.
1041 */
1042 int
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001043lbr_chartabsize_adv(chartabsize_T *cts)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001044{
1045 int retval;
1046
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001047 retval = lbr_chartabsize(cts);
1048 MB_PTR_ADV(cts->cts_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001049 return retval;
1050}
1051
Bram Moolenaar398649e2022-08-04 15:03:48 +01001052#if defined(FEAT_PROP_POPUP) || defined(PROTO)
1053/*
1054 * Return the cell size of virtual text after truncation.
1055 */
1056 int
1057textprop_size_after_trunc(
1058 win_T *wp,
1059 int below,
1060 int added,
1061 char_u *text,
1062 int *n_used_ptr)
1063{
1064 int space = below ? wp->w_width : added;
1065 int len = (int)STRLEN(text);
1066 int strsize = 0;
1067 int n_used;
1068
1069 // if the remaining size is to small wrap
1070 // anyway and use the next line
1071 if (space < PROP_TEXT_MIN_CELLS)
1072 space += wp->w_width;
1073 for (n_used = 0; n_used < len; n_used += (*mb_ptr2len)(text + n_used))
1074 {
1075 int clen = ptr2cells(text + n_used);
1076
1077 if (strsize + clen > space)
1078 break;
1079 strsize += clen;
1080 }
1081 *n_used_ptr = n_used;
1082 return strsize;
1083}
1084#endif
1085
Bram Moolenaar071d4272004-06-13 20:20:40 +00001086/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001087 * Return the screen size of the character indicated by "cts".
1088 * "cts->cts_cur_text_width" is set to the extra size for a text property that
1089 * inserts text.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001090 * This function is used very often, keep it fast!!!!
1091 *
1092 * If "headp" not NULL, set *headp to the size of what we for 'showbreak'
1093 * string at start of line. Warning: *headp is only set if it's a non-zero
1094 * value, init to 0 before calling.
1095 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001096 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001097win_lbr_chartabsize(
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001098 chartabsize_T *cts,
1099 int *headp UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001100{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001101 win_T *wp = cts->cts_win;
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001102#ifdef FEAT_PROP_POPUP
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001103 char_u *line = cts->cts_line; // start of the line
Bram Moolenaarfe3fb6e2022-07-25 18:35:15 +01001104#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001105 char_u *s = cts->cts_ptr;
1106 colnr_T vcol = cts->cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107#ifdef FEAT_LINEBREAK
1108 int c;
1109 int size;
1110 colnr_T col2;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001111 colnr_T col_adj = 0; // vcol + screen size of tab
Bram Moolenaar071d4272004-06-13 20:20:40 +00001112 colnr_T colmax;
1113 int added;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001114 int mb_added = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001115 int numberextra;
1116 char_u *ps;
1117 int tab_corr = (*s == TAB);
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00001118 int n;
Bram Moolenaaree857022019-11-09 23:26:40 +01001119 char_u *sbr;
Bram Moolenaarcba69522022-08-06 21:03:53 +01001120 int no_sbr = FALSE;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001121#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001123#if defined(FEAT_PROP_POPUP)
1124 cts->cts_cur_text_width = 0;
1125#endif
1126
1127#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001128 /*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001129 * No 'linebreak', 'showbreak', 'breakindent' and text properties that
1130 * insert text: return quickly.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001131 */
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001132 if (1
1133# ifdef FEAT_LINEBREAK
1134 && !wp->w_p_lbr && !wp->w_p_bri && *get_showbreak_value(wp) == NUL
1135# endif
1136# ifdef FEAT_PROP_POPUP
1137 && !cts->cts_has_prop_with_text
1138# endif
1139 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001140#endif
1141 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001142 if (wp->w_p_wrap)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001143 return win_nolbr_chartabsize(cts, headp);
1144 RET_WIN_BUF_CHARTABSIZE(wp, wp->w_buffer, s, vcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001145 }
1146
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001147#if defined(FEAT_LINEBREAK) || defined(FEAT_PROP_POPUP)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001148 /*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001149 * First get the normal size, without 'linebreak' or text properties
Bram Moolenaar071d4272004-06-13 20:20:40 +00001150 */
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001151 size = win_chartabsize(wp, s, vcol);
1152
1153# ifdef FEAT_PROP_POPUP
Bram Moolenaar49a90792022-08-09 18:25:23 +01001154 if (cts->cts_has_prop_with_text)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001155 {
Bram Moolenaare428fa02022-08-09 16:55:41 +01001156 int tab_size = size;
Bram Moolenaar49a90792022-08-09 18:25:23 +01001157 int charlen = *s == NUL ? 1 : mb_ptr2len(s);
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001158 int i;
1159 int col = (int)(s - line);
1160 garray_T *gap = &wp->w_buffer->b_textprop_text;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001161
1162 for (i = 0; i < cts->cts_text_prop_count; ++i)
1163 {
1164 textprop_T *tp = cts->cts_text_props + i;
1165
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001166 // Watch out for the text being deleted. "cts_text_props" is a
1167 // copy, the text prop may actually have been removed from the line.
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001168 if (tp->tp_id < 0
Bram Moolenaar25463612022-08-08 11:07:47 +01001169 && ((tp->tp_col - 1 >= col
Bram Moolenaare428fa02022-08-09 16:55:41 +01001170 && tp->tp_col - 1 < col + charlen)
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001171 || (tp->tp_col == MAXCOL && (s[0] == NUL || s[1] == NUL)
1172 && cts->cts_with_trailing))
1173 && -tp->tp_id - 1 < gap->ga_len)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001174 {
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001175 char_u *p = ((char_u **)gap->ga_data)[-tp->tp_id - 1];
Bram Moolenaarb7963df2022-07-31 17:12:43 +01001176
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001177 if (p != NULL)
Bram Moolenaarb7963df2022-07-31 17:12:43 +01001178 {
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001179 int cells = vim_strsize(p);
Bram Moolenaar398649e2022-08-04 15:03:48 +01001180
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001181 if (tp->tp_col == MAXCOL)
1182 {
1183 int below = (tp->tp_flags & TP_FLAG_ALIGN_BELOW);
Bram Moolenaarf0ccfa42022-08-13 16:41:19 +01001184 int right = (tp->tp_flags & TP_FLAG_ALIGN_RIGHT);
Bram Moolenaarcba69522022-08-06 21:03:53 +01001185 int wrap = (tp->tp_flags & TP_FLAG_WRAP);
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001186 int len = (int)STRLEN(p);
1187 int n_used = len;
1188
Bram Moolenaarc3a483f2022-08-14 19:37:36 +01001189 // The "$" for 'list' mode will go between the EOL and
1190 // the text prop, account for that.
1191 if (wp->w_p_list && wp->w_lcs_chars.eol != NUL)
1192 ++vcol;
1193
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001194 // Keep in sync with where textprop_size_after_trunc()
1195 // is called in win_line().
1196 if (!wrap)
Bram Moolenaar25463612022-08-08 11:07:47 +01001197 {
1198 added = wp->w_width - (vcol + size) % wp->w_width;
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001199 cells = textprop_size_after_trunc(wp,
Bram Moolenaar398649e2022-08-04 15:03:48 +01001200 below, added, p, &n_used);
Bram Moolenaar25463612022-08-08 11:07:47 +01001201 }
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001202 if (below)
1203 cells += wp->w_width - (vcol + size) % wp->w_width;
Bram Moolenaarf0ccfa42022-08-13 16:41:19 +01001204 else if (right)
1205 {
1206 len = wp->w_width - vcol % wp->w_width;
1207 if (len > cells + size)
1208 // add the padding for right-alignment
1209 cells = len - size;
1210 else if (len == 0)
1211 // padding to right-align in the next line
1212 cells += cells > wp->w_width ? 0
1213 :wp->w_width - cells;
1214 }
Bram Moolenaarcba69522022-08-06 21:03:53 +01001215#ifdef FEAT_LINEBREAK
1216 no_sbr = TRUE; // don't use 'showbreak' now
1217#endif
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001218 }
1219 cts->cts_cur_text_width += cells;
Bram Moolenaar28c9f892022-08-14 13:28:55 +01001220 cts->cts_start_incl = tp->tp_flags & TP_FLAG_START_INCL;
Bram Moolenaar2f83cc42022-08-05 11:45:17 +01001221 size += cells;
Bram Moolenaare428fa02022-08-09 16:55:41 +01001222 if (*s == TAB)
1223 {
1224 // tab size changes because of the inserted text
1225 size -= tab_size;
1226 tab_size = win_chartabsize(wp, s, vcol + size);
1227 size += tab_size;
1228 }
Bram Moolenaarb7963df2022-07-31 17:12:43 +01001229 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001230 }
Bram Moolenaar50e75fe2022-08-05 20:25:50 +01001231 if (tp->tp_col != MAXCOL && tp->tp_col - 1 > col)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001232 break;
1233 }
1234 }
1235# endif
1236
1237# ifdef FEAT_LINEBREAK
Bram Moolenaar071d4272004-06-13 20:20:40 +00001238 c = *s;
Bram Moolenaaree739b42014-07-02 19:37:42 +02001239 if (tab_corr)
1240 col_adj = size - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001241
1242 /*
1243 * If 'linebreak' set check at a blank before a non-blank if the line
1244 * needs a break here
1245 */
1246 if (wp->w_p_lbr
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001247 && VIM_ISBREAK(c)
Bram Moolenaar977d0372017-03-12 21:31:58 +01001248 && !VIM_ISBREAK((int)s[1])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001249 && wp->w_p_wrap
Bram Moolenaar4033c552017-09-16 20:54:51 +02001250 && wp->w_width != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001251 {
1252 /*
1253 * Count all characters from first non-blank after a blank up to next
1254 * non-blank after a blank.
1255 */
1256 numberextra = win_col_off(wp);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001257 col2 = vcol;
Bram Moolenaar02631462017-09-22 15:20:32 +02001258 colmax = (colnr_T)(wp->w_width - numberextra - col_adj);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001259 if (vcol >= colmax)
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00001260 {
Bram Moolenaaree739b42014-07-02 19:37:42 +02001261 colmax += col_adj;
1262 n = colmax + win_col_off2(wp);
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00001263 if (n > 0)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001264 colmax += (((vcol - colmax) / n) + 1) * n - col_adj;
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00001265 }
1266
Bram Moolenaar071d4272004-06-13 20:20:40 +00001267 for (;;)
1268 {
1269 ps = s;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001270 MB_PTR_ADV(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001271 c = *s;
1272 if (!(c != NUL
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001273 && (VIM_ISBREAK(c)
1274 || (!VIM_ISBREAK(c)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001275 && (col2 == vcol || !VIM_ISBREAK((int)*ps))))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001276 break;
1277
1278 col2 += win_chartabsize(wp, s, col2);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001279 if (col2 >= colmax) // doesn't fit
Bram Moolenaar071d4272004-06-13 20:20:40 +00001280 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001281 size = colmax - vcol + col_adj;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001282 break;
1283 }
1284 }
1285 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001286 else if (has_mbyte && size == 2 && MB_BYTE2LEN(*s) > 1
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001287 && wp->w_p_wrap && in_win_border(wp, vcol))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001288 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001289 ++size; // Count the ">" in the last column.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001290 mb_added = 1;
1291 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001292
1293 /*
Bram Moolenaar597a4222014-06-25 14:39:50 +02001294 * May have to add something for 'breakindent' and/or 'showbreak'
1295 * string at start of line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001296 * Set *headp to the size of what we add.
Bram Moolenaar21efafe2022-03-03 20:04:03 +00001297 * Do not use 'showbreak' at the NUL after the text.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001298 */
1299 added = 0;
Bram Moolenaarcba69522022-08-06 21:03:53 +01001300 sbr = (c == NUL || no_sbr) ? empty_option : get_showbreak_value(wp);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001301 if ((*sbr != NUL || wp->w_p_bri) && wp->w_p_wrap && vcol != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001302 {
Bram Moolenaard574ea22015-01-14 19:35:14 +01001303 colnr_T sbrlen = 0;
1304 int numberwidth = win_col_off(wp);
1305
1306 numberextra = numberwidth;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001307 vcol += numberextra + mb_added;
1308 if (vcol >= (colnr_T)wp->w_width)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001309 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001310 vcol -= wp->w_width;
Bram Moolenaar02631462017-09-22 15:20:32 +02001311 numberextra = wp->w_width - (numberextra - win_col_off2(wp));
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001312 if (vcol >= numberextra && numberextra > 0)
1313 vcol %= numberextra;
Bram Moolenaaree857022019-11-09 23:26:40 +01001314 if (*sbr != NUL)
Bram Moolenaar1c852102014-10-15 21:26:40 +02001315 {
Bram Moolenaaree857022019-11-09 23:26:40 +01001316 sbrlen = (colnr_T)MB_CHARLEN(sbr);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001317 if (vcol >= sbrlen)
1318 vcol -= sbrlen;
Bram Moolenaar1c852102014-10-15 21:26:40 +02001319 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001320 if (vcol >= numberextra && numberextra > 0)
1321 vcol = vcol % numberextra;
1322 else if (vcol > 0 && numberextra > 0)
1323 vcol += numberwidth - win_col_off2(wp);
Bram Moolenaard574ea22015-01-14 19:35:14 +01001324
1325 numberwidth -= win_col_off2(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001326 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001327 if (vcol == 0 || vcol + size + sbrlen > (colnr_T)wp->w_width)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001328 {
Bram Moolenaar597a4222014-06-25 14:39:50 +02001329 added = 0;
Bram Moolenaaree857022019-11-09 23:26:40 +01001330 if (*sbr != NUL)
Bram Moolenaard574ea22015-01-14 19:35:14 +01001331 {
Bram Moolenaar02631462017-09-22 15:20:32 +02001332 if (size + sbrlen + numberwidth > (colnr_T)wp->w_width)
Bram Moolenaard574ea22015-01-14 19:35:14 +01001333 {
Bram Moolenaar7833dab2019-05-27 22:01:40 +02001334 // calculate effective window width
Bram Moolenaar02631462017-09-22 15:20:32 +02001335 int width = (colnr_T)wp->w_width - sbrlen - numberwidth;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001336 int prev_width = vcol
1337 ? ((colnr_T)wp->w_width - (sbrlen + vcol)) : 0;
Bram Moolenaar7833dab2019-05-27 22:01:40 +02001338
1339 if (width <= 0)
1340 width = (colnr_T)1;
Bram Moolenaaree857022019-11-09 23:26:40 +01001341 added += ((size - prev_width) / width) * vim_strsize(sbr);
Bram Moolenaard574ea22015-01-14 19:35:14 +01001342 if ((size - prev_width) % width)
Bram Moolenaar7833dab2019-05-27 22:01:40 +02001343 // wrapped, add another length of 'sbr'
Bram Moolenaaree857022019-11-09 23:26:40 +01001344 added += vim_strsize(sbr);
Bram Moolenaard574ea22015-01-14 19:35:14 +01001345 }
1346 else
Bram Moolenaaree857022019-11-09 23:26:40 +01001347 added += vim_strsize(sbr);
Bram Moolenaard574ea22015-01-14 19:35:14 +01001348 }
Bram Moolenaar597a4222014-06-25 14:39:50 +02001349 if (wp->w_p_bri)
1350 added += get_breakindent_win(wp, line);
1351
Bram Moolenaar95765082014-08-24 21:19:25 +02001352 size += added;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001353 if (vcol != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001354 added = 0;
1355 }
1356 }
1357 if (headp != NULL)
1358 *headp = added + mb_added;
1359 return size;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001360# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001361#endif
1362}
1363
Bram Moolenaar071d4272004-06-13 20:20:40 +00001364/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001365 * Like win_lbr_chartabsize(), except that we know 'linebreak' is off, 'wrap'
1366 * is on and there are no properties that insert text. This means we need to
1367 * check for a double-byte character that doesn't fit at the end of the screen
1368 * line.
1369 * Only uses "cts_win", "cts_ptr" and "cts_vcol" from "cts".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001370 */
1371 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001372win_nolbr_chartabsize(
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001373 chartabsize_T *cts,
1374 int *headp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001375{
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001376 win_T *wp = cts->cts_win;
1377 char_u *s = cts->cts_ptr;
1378 colnr_T col = cts->cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001379 int n;
1380
Bram Moolenaareed9d462021-02-15 20:38:25 +01001381 if (*s == TAB && (!wp->w_p_list || wp->w_lcs_chars.tab1))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001382 {
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001383# ifdef FEAT_VARTABS
1384 return tabstop_padding(col, wp->w_buffer->b_p_ts,
1385 wp->w_buffer->b_p_vts_array);
1386# else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001387 n = wp->w_buffer->b_p_ts;
1388 return (int)(n - (col % n));
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001389# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001390 }
1391 n = ptr2cells(s);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001392 // Add one cell for a double-width character in the last column of the
1393 // window, displayed with a ">".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001394 if (n == 2 && MB_BYTE2LEN(*s) > 1 && in_win_border(wp, col))
1395 {
1396 if (headp != NULL)
1397 *headp = 1;
1398 return 3;
1399 }
1400 return n;
1401}
1402
1403/*
1404 * Return TRUE if virtual column "vcol" is in the rightmost column of window
1405 * "wp".
1406 */
Bram Moolenaar5843f5f2019-08-20 20:13:45 +02001407 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001408in_win_border(win_T *wp, colnr_T vcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001409{
Bram Moolenaarc667da52019-11-30 20:52:27 +01001410 int width1; // width of first line (after line number)
1411 int width2; // width of further lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00001412
Bram Moolenaarc667da52019-11-30 20:52:27 +01001413 if (wp->w_width == 0) // there is no border
Bram Moolenaar071d4272004-06-13 20:20:40 +00001414 return FALSE;
Bram Moolenaar02631462017-09-22 15:20:32 +02001415 width1 = wp->w_width - win_col_off(wp);
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001416 if ((int)vcol < width1 - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001417 return FALSE;
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001418 if ((int)vcol == width1 - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001419 return TRUE;
1420 width2 = width1 + win_col_off2(wp);
Bram Moolenaar8701cd62009-10-07 14:20:30 +00001421 if (width2 <= 0)
1422 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001423 return ((vcol - width1) % width2 == width2 - 1);
1424}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001425
1426/*
1427 * Get virtual column number of pos.
1428 * start: on the first position of this character (TAB, ctrl)
1429 * cursor: where the cursor is on this character (first char, except for TAB)
1430 * end: on the last position of this character (TAB, ctrl)
1431 *
1432 * This is used very often, keep it fast!
1433 */
1434 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001435getvcol(
1436 win_T *wp,
1437 pos_T *pos,
1438 colnr_T *start,
1439 colnr_T *cursor,
1440 colnr_T *end)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001441{
1442 colnr_T vcol;
Bram Moolenaarc667da52019-11-30 20:52:27 +01001443 char_u *ptr; // points to current char
1444 char_u *posptr; // points to char at pos->col
1445 char_u *line; // start of the line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001446 int incr;
1447 int head;
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001448#ifdef FEAT_VARTABS
1449 int *vts = wp->w_buffer->b_p_vts_array;
1450#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001451 int ts = wp->w_buffer->b_p_ts;
1452 int c;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001453 chartabsize_T cts;
Bram Moolenaar49a90792022-08-09 18:25:23 +01001454#ifdef FEAT_PROP_POPUP
1455 int on_NUL = FALSE;
1456#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001457
1458 vcol = 0;
Bram Moolenaar597a4222014-06-25 14:39:50 +02001459 line = ptr = ml_get_buf(wp->w_buffer, pos->lnum, FALSE);
Bram Moolenaar37d619f2010-03-10 14:46:26 +01001460 if (pos->col == MAXCOL)
Bram Moolenaarc667da52019-11-30 20:52:27 +01001461 posptr = NULL; // continue until the NUL
Bram Moolenaar37d619f2010-03-10 14:46:26 +01001462 else
Bram Moolenaar0c0590d2017-01-28 13:48:10 +01001463 {
Bram Moolenaar94f31922021-12-30 15:29:18 +00001464 colnr_T i;
1465
1466 // In a few cases the position can be beyond the end of the line.
1467 for (i = 0; i < pos->col; ++i)
1468 if (ptr[i] == NUL)
1469 {
1470 pos->col = i;
1471 break;
1472 }
Bram Moolenaar37d619f2010-03-10 14:46:26 +01001473 posptr = ptr + pos->col;
Bram Moolenaar0c0590d2017-01-28 13:48:10 +01001474 if (has_mbyte)
Bram Moolenaarc667da52019-11-30 20:52:27 +01001475 // always start on the first byte
Bram Moolenaar0c0590d2017-01-28 13:48:10 +01001476 posptr -= (*mb_head_off)(line, posptr);
Bram Moolenaar0c0590d2017-01-28 13:48:10 +01001477 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001478
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001479 init_chartabsize_arg(&cts, wp, pos->lnum, 0, line, line);
1480
Bram Moolenaar071d4272004-06-13 20:20:40 +00001481 /*
1482 * This function is used very often, do some speed optimizations.
Bram Moolenaar597a4222014-06-25 14:39:50 +02001483 * When 'list', 'linebreak', 'showbreak' and 'breakindent' are not set
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001484 * and there are no text properties with "text" use a simple loop.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001485 * Also use this when 'list' is set but tabs take their normal size.
1486 */
Bram Moolenaareed9d462021-02-15 20:38:25 +01001487 if ((!wp->w_p_list || wp->w_lcs_chars.tab1 != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001488#ifdef FEAT_LINEBREAK
Bram Moolenaaree857022019-11-09 23:26:40 +01001489 && !wp->w_p_lbr && *get_showbreak_value(wp) == NUL && !wp->w_p_bri
Bram Moolenaar071d4272004-06-13 20:20:40 +00001490#endif
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001491#ifdef FEAT_PROP_POPUP
1492 && !cts.cts_has_prop_with_text
1493#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001494 )
1495 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001496 for (;;)
1497 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001498 head = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001499 c = *ptr;
Bram Moolenaarc667da52019-11-30 20:52:27 +01001500 // make sure we don't go past the end of the line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001501 if (c == NUL)
1502 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001503 incr = 1; // NUL at end of line only takes one column
Bram Moolenaar071d4272004-06-13 20:20:40 +00001504 break;
1505 }
Bram Moolenaarc667da52019-11-30 20:52:27 +01001506 // A tab gets expanded, depending on the current column
Bram Moolenaar071d4272004-06-13 20:20:40 +00001507 if (c == TAB)
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001508#ifdef FEAT_VARTABS
1509 incr = tabstop_padding(vcol, ts, vts);
1510#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001511 incr = ts - (vcol % ts);
Bram Moolenaar04958cb2018-06-23 19:23:02 +02001512#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001513 else
1514 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001515 if (has_mbyte)
1516 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001517 // For utf-8, if the byte is >= 0x80, need to look at
1518 // further bytes to find the cell width.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001519 if (enc_utf8 && c >= 0x80)
1520 incr = utf_ptr2cells(ptr);
1521 else
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +01001522 incr = g_chartab[c] & CT_CELL_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001523
Bram Moolenaarc667da52019-11-30 20:52:27 +01001524 // If a double-cell char doesn't fit at the end of a line
1525 // it wraps to the next line, it's like this char is three
1526 // cells wide.
Bram Moolenaar9c33a7c2008-02-20 13:59:32 +00001527 if (incr == 2 && wp->w_p_wrap && MB_BYTE2LEN(*ptr) > 1
1528 && in_win_border(wp, vcol))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001529 {
1530 ++incr;
1531 head = 1;
1532 }
1533 }
1534 else
Bram Moolenaar88e8f9f2016-01-20 22:48:02 +01001535 incr = g_chartab[c] & CT_CELL_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001536 }
1537
Bram Moolenaarc667da52019-11-30 20:52:27 +01001538 if (posptr != NULL && ptr >= posptr) // character at pos->col
Bram Moolenaar071d4272004-06-13 20:20:40 +00001539 break;
1540
1541 vcol += incr;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001542 MB_PTR_ADV(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001543 }
1544 }
1545 else
1546 {
1547 for (;;)
1548 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001549 // A tab gets expanded, depending on the current column.
1550 // Other things also take up space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001551 head = 0;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001552 incr = win_lbr_chartabsize(&cts, &head);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001553 // make sure we don't go past the end of the line
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001554 if (*cts.cts_ptr == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001555 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001556 incr = 1; // NUL at end of line only takes one column
Bram Moolenaar49a90792022-08-09 18:25:23 +01001557#ifdef FEAT_PROP_POPUP
1558 if (cts.cts_cur_text_width > 0)
1559 incr = cts.cts_cur_text_width;
1560 on_NUL = TRUE;
1561#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001562 break;
1563 }
1564
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001565 if (posptr != NULL && cts.cts_ptr >= posptr)
1566 // character at pos->col
Bram Moolenaar071d4272004-06-13 20:20:40 +00001567 break;
1568
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001569 cts.cts_vcol += incr;
1570 MB_PTR_ADV(cts.cts_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001571 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001572 vcol = cts.cts_vcol;
1573 ptr = cts.cts_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001575 clear_chartabsize_arg(&cts);
1576
Bram Moolenaar071d4272004-06-13 20:20:40 +00001577 if (start != NULL)
1578 *start = vcol + head;
1579 if (end != NULL)
1580 *end = vcol + incr - 1;
1581 if (cursor != NULL)
1582 {
1583 if (*ptr == TAB
Bram Moolenaar24959102022-05-07 20:01:16 +01001584 && (State & MODE_NORMAL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001585 && !wp->w_p_list
1586 && !virtual_active()
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01001587 && !(VIsual_active
1588 && (*p_sel == 'e' || LTOREQ_POS(*pos, VIsual)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001589 )
Bram Moolenaarc667da52019-11-30 20:52:27 +01001590 *cursor = vcol + incr - 1; // cursor at end
Bram Moolenaar071d4272004-06-13 20:20:40 +00001591 else
Bram Moolenaare428fa02022-08-09 16:55:41 +01001592 {
1593#ifdef FEAT_PROP_POPUP
Bram Moolenaar28c9f892022-08-14 13:28:55 +01001594 // in Insert mode, if "start_incl" is true the text gets inserted
1595 // after the virtual text, thus add its width
1596 if (((State & MODE_INSERT) == 0 || cts.cts_start_incl) && !on_NUL)
Bram Moolenaar49a90792022-08-09 18:25:23 +01001597 // cursor is after inserted text, unless on the NUL
Bram Moolenaare428fa02022-08-09 16:55:41 +01001598 vcol += cts.cts_cur_text_width;
1599#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01001600 *cursor = vcol + head; // cursor at start
Bram Moolenaare428fa02022-08-09 16:55:41 +01001601 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001602 }
1603}
1604
1605/*
1606 * Get virtual cursor column in the current window, pretending 'list' is off.
1607 */
1608 colnr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01001609getvcol_nolist(pos_T *posp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001610{
1611 int list_save = curwin->w_p_list;
1612 colnr_T vcol;
1613
1614 curwin->w_p_list = FALSE;
Bram Moolenaardb0eede2018-04-25 22:38:17 +02001615 if (posp->coladd)
1616 getvvcol(curwin, posp, NULL, &vcol, NULL);
1617 else
Bram Moolenaardb0eede2018-04-25 22:38:17 +02001618 getvcol(curwin, posp, NULL, &vcol, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001619 curwin->w_p_list = list_save;
1620 return vcol;
1621}
1622
Bram Moolenaar071d4272004-06-13 20:20:40 +00001623/*
1624 * Get virtual column in virtual mode.
1625 */
1626 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001627getvvcol(
1628 win_T *wp,
1629 pos_T *pos,
1630 colnr_T *start,
1631 colnr_T *cursor,
1632 colnr_T *end)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001633{
1634 colnr_T col;
1635 colnr_T coladd;
1636 colnr_T endadd;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001637 char_u *ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001638
1639 if (virtual_active())
1640 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01001641 // For virtual mode, only want one value
Bram Moolenaar071d4272004-06-13 20:20:40 +00001642 getvcol(wp, pos, &col, NULL, NULL);
1643
1644 coladd = pos->coladd;
1645 endadd = 0;
Bram Moolenaarc667da52019-11-30 20:52:27 +01001646 // Cannot put the cursor on part of a wide character.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001647 ptr = ml_get_buf(wp->w_buffer, pos->lnum, FALSE);
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001648 if (pos->col < (colnr_T)STRLEN(ptr))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001649 {
1650 int c = (*mb_ptr2char)(ptr + pos->col);
1651
1652 if (c != TAB && vim_isprintc(c))
1653 {
Bram Moolenaar0ab2a882009-05-13 10:51:08 +00001654 endadd = (colnr_T)(char2cells(c) - 1);
Bram Moolenaarc667da52019-11-30 20:52:27 +01001655 if (coladd > endadd) // past end of line
Bram Moolenaara5792f52005-11-23 21:25:05 +00001656 endadd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001657 else
1658 coladd = 0;
1659 }
1660 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001661 col += coladd;
1662 if (start != NULL)
1663 *start = col;
1664 if (cursor != NULL)
1665 *cursor = col;
1666 if (end != NULL)
1667 *end = col + endadd;
1668 }
1669 else
1670 getvcol(wp, pos, start, cursor, end);
1671}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001672
Bram Moolenaar071d4272004-06-13 20:20:40 +00001673/*
1674 * Get the leftmost and rightmost virtual column of pos1 and pos2.
1675 * Used for Visual block mode.
1676 */
1677 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001678getvcols(
1679 win_T *wp,
1680 pos_T *pos1,
1681 pos_T *pos2,
1682 colnr_T *left,
1683 colnr_T *right)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001684{
1685 colnr_T from1, from2, to1, to2;
1686
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01001687 if (LT_POSP(pos1, pos2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001688 {
1689 getvvcol(wp, pos1, &from1, NULL, &to1);
1690 getvvcol(wp, pos2, &from2, NULL, &to2);
1691 }
1692 else
1693 {
1694 getvvcol(wp, pos2, &from1, NULL, &to1);
1695 getvvcol(wp, pos1, &from2, NULL, &to2);
1696 }
1697 if (from2 < from1)
1698 *left = from2;
1699 else
1700 *left = from1;
1701 if (to2 > to1)
1702 {
1703 if (*p_sel == 'e' && from2 - 1 >= to1)
1704 *right = from2 - 1;
1705 else
1706 *right = to2;
1707 }
1708 else
1709 *right = to1;
1710}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001711
1712/*
Bram Moolenaarce7eada2021-12-15 15:41:44 +00001713 * Skip over ' ' and '\t'.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001714 */
1715 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001716skipwhite(char_u *q)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001717{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001718 char_u *p = q;
1719
Bram Moolenaarce7eada2021-12-15 15:41:44 +00001720 while (VIM_ISWHITE(*p))
1721 ++p;
1722 return p;
1723}
1724
Dominique Pelle748b3082022-01-08 12:41:16 +00001725#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaarce7eada2021-12-15 15:41:44 +00001726/*
1727 * skip over ' ', '\t' and '\n'.
1728 */
1729 char_u *
1730skipwhite_and_nl(char_u *q)
1731{
1732 char_u *p = q;
1733
1734 while (VIM_ISWHITE(*p) || *p == NL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001735 ++p;
1736 return p;
1737}
Dominique Pelle748b3082022-01-08 12:41:16 +00001738#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001739
1740/*
Bram Moolenaare2e69e42017-09-02 20:30:35 +02001741 * getwhitecols: return the number of whitespace
1742 * columns (bytes) at the start of a given line
1743 */
1744 int
1745getwhitecols_curline()
1746{
1747 return getwhitecols(ml_get_curline());
1748}
1749
1750 int
1751getwhitecols(char_u *p)
1752{
1753 return skipwhite(p) - p;
1754}
1755
1756/*
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001757 * skip over digits
Bram Moolenaar071d4272004-06-13 20:20:40 +00001758 */
1759 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001760skipdigits(char_u *q)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001761{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001762 char_u *p = q;
1763
Bram Moolenaarc667da52019-11-30 20:52:27 +01001764 while (VIM_ISDIGIT(*p)) // skip to next non-digit
Bram Moolenaar071d4272004-06-13 20:20:40 +00001765 ++p;
1766 return p;
1767}
1768
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001769#if defined(FEAT_SYN_HL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001770/*
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001771 * skip over binary digits
1772 */
1773 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001774skipbin(char_u *q)
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001775{
1776 char_u *p = q;
1777
Bram Moolenaarc667da52019-11-30 20:52:27 +01001778 while (vim_isbdigit(*p)) // skip to next non-digit
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001779 ++p;
1780 return p;
1781}
1782
1783/*
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001784 * skip over digits and hex characters
1785 */
1786 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001787skiphex(char_u *q)
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001788{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001789 char_u *p = q;
1790
Bram Moolenaarc667da52019-11-30 20:52:27 +01001791 while (vim_isxdigit(*p)) // skip to next non-digit
Bram Moolenaar75c50c42005-06-04 22:06:24 +00001792 ++p;
1793 return p;
1794}
1795#endif
1796
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001797/*
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001798 * skip to bin digit (or NUL after the string)
1799 */
1800 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001801skiptobin(char_u *q)
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001802{
1803 char_u *p = q;
1804
Bram Moolenaarc667da52019-11-30 20:52:27 +01001805 while (*p != NUL && !vim_isbdigit(*p)) // skip to next digit
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001806 ++p;
1807 return p;
1808}
1809
1810/*
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001811 * skip to digit (or NUL after the string)
1812 */
1813 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001814skiptodigit(char_u *q)
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001815{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001816 char_u *p = q;
1817
Bram Moolenaarc667da52019-11-30 20:52:27 +01001818 while (*p != NUL && !VIM_ISDIGIT(*p)) // skip to next digit
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001819 ++p;
1820 return p;
1821}
1822
1823/*
1824 * skip to hex character (or NUL after the string)
1825 */
1826 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01001827skiptohex(char_u *q)
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001828{
Bram Moolenaar1387a602008-07-24 19:31:11 +00001829 char_u *p = q;
1830
Bram Moolenaarc667da52019-11-30 20:52:27 +01001831 while (*p != NUL && !vim_isxdigit(*p)) // skip to next digit
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001832 ++p;
1833 return p;
1834}
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00001835
Bram Moolenaar071d4272004-06-13 20:20:40 +00001836/*
1837 * Variant of isdigit() that can handle characters > 0x100.
1838 * We don't use isdigit() here, because on some systems it also considers
1839 * superscript 1 to be a digit.
1840 * Use the VIM_ISDIGIT() macro for simple arguments.
1841 */
1842 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001843vim_isdigit(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844{
1845 return (c >= '0' && c <= '9');
1846}
1847
1848/*
1849 * Variant of isxdigit() that can handle characters > 0x100.
1850 * We don't use isxdigit() here, because on some systems it also considers
1851 * superscript 1 to be a digit.
1852 */
1853 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001854vim_isxdigit(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001855{
1856 return (c >= '0' && c <= '9')
1857 || (c >= 'a' && c <= 'f')
1858 || (c >= 'A' && c <= 'F');
1859}
1860
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001861/*
1862 * Corollary of vim_isdigit and vim_isxdigit() that can handle
1863 * characters > 0x100.
1864 */
1865 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001866vim_isbdigit(int c)
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01001867{
1868 return (c == '0' || c == '1');
1869}
1870
Bram Moolenaarc37b6552021-01-07 19:36:30 +01001871 static int
1872vim_isodigit(int c)
1873{
1874 return (c >= '0' && c <= '7');
1875}
1876
Bram Moolenaar78622822005-08-23 21:00:13 +00001877/*
1878 * Vim's own character class functions. These exist because many library
1879 * islower()/toupper() etc. do not work properly: they crash when used with
1880 * invalid values or can't handle latin1 when the locale is C.
1881 * Speed is most important here.
1882 */
1883#define LATIN1LOWER 'l'
1884#define LATIN1UPPER 'U'
1885
Bram Moolenaar6e7c7f32005-08-24 22:16:11 +00001886static char_u latin1flags[257] = " UUUUUUUUUUUUUUUUUUUUUUUUUU llllllllllllllllllllllllll UUUUUUUUUUUUUUUUUUUUUUU UUUUUUUllllllllllllllllllllllll llllllll";
Bram Moolenaar936347b2012-05-25 11:56:22 +02001887static 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";
1888static 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 +00001889
1890 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001891vim_islower(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00001892{
1893 if (c <= '@')
1894 return FALSE;
1895 if (c >= 0x80)
1896 {
1897 if (enc_utf8)
1898 return utf_islower(c);
1899 if (c >= 0x100)
1900 {
1901#ifdef HAVE_ISWLOWER
1902 if (has_mbyte)
1903 return iswlower(c);
1904#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01001905 // islower() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00001906 return FALSE;
1907 }
1908 if (enc_latin1like)
1909 return (latin1flags[c] & LATIN1LOWER) == LATIN1LOWER;
1910 }
1911 return islower(c);
1912}
1913
1914 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001915vim_isupper(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00001916{
1917 if (c <= '@')
1918 return FALSE;
1919 if (c >= 0x80)
1920 {
1921 if (enc_utf8)
1922 return utf_isupper(c);
1923 if (c >= 0x100)
1924 {
1925#ifdef HAVE_ISWUPPER
1926 if (has_mbyte)
1927 return iswupper(c);
1928#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01001929 // islower() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00001930 return FALSE;
1931 }
1932 if (enc_latin1like)
1933 return (latin1flags[c] & LATIN1UPPER) == LATIN1UPPER;
1934 }
1935 return isupper(c);
1936}
1937
1938 int
Bram Moolenaar5921aeb2022-02-19 11:20:12 +00001939vim_isalpha(int c)
1940{
1941 return vim_islower(c) || vim_isupper(c);
1942}
1943
1944 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001945vim_toupper(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00001946{
1947 if (c <= '@')
1948 return c;
Bram Moolenaar3317d5e2017-04-08 19:12:06 +02001949 if (c >= 0x80 || !(cmp_flags & CMP_KEEPASCII))
Bram Moolenaar78622822005-08-23 21:00:13 +00001950 {
1951 if (enc_utf8)
1952 return utf_toupper(c);
1953 if (c >= 0x100)
1954 {
1955#ifdef HAVE_TOWUPPER
1956 if (has_mbyte)
1957 return towupper(c);
1958#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01001959 // toupper() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00001960 return c;
1961 }
1962 if (enc_latin1like)
1963 return latin1upper[c];
1964 }
Bram Moolenaar1cc48202017-04-09 13:41:59 +02001965 if (c < 0x80 && (cmp_flags & CMP_KEEPASCII))
1966 return TOUPPER_ASC(c);
Bram Moolenaar78622822005-08-23 21:00:13 +00001967 return TOUPPER_LOC(c);
1968}
1969
1970 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001971vim_tolower(int c)
Bram Moolenaar78622822005-08-23 21:00:13 +00001972{
1973 if (c <= '@')
1974 return c;
Bram Moolenaar3317d5e2017-04-08 19:12:06 +02001975 if (c >= 0x80 || !(cmp_flags & CMP_KEEPASCII))
Bram Moolenaar78622822005-08-23 21:00:13 +00001976 {
1977 if (enc_utf8)
1978 return utf_tolower(c);
1979 if (c >= 0x100)
1980 {
1981#ifdef HAVE_TOWLOWER
1982 if (has_mbyte)
1983 return towlower(c);
1984#endif
Bram Moolenaarc667da52019-11-30 20:52:27 +01001985 // tolower() can't handle these chars and may crash
Bram Moolenaar78622822005-08-23 21:00:13 +00001986 return c;
1987 }
1988 if (enc_latin1like)
1989 return latin1lower[c];
1990 }
Bram Moolenaar1cc48202017-04-09 13:41:59 +02001991 if (c < 0x80 && (cmp_flags & CMP_KEEPASCII))
1992 return TOLOWER_ASC(c);
Bram Moolenaar78622822005-08-23 21:00:13 +00001993 return TOLOWER_LOC(c);
1994}
Bram Moolenaar78622822005-08-23 21:00:13 +00001995
Bram Moolenaar071d4272004-06-13 20:20:40 +00001996/*
1997 * skiptowhite: skip over text until ' ' or '\t' or NUL.
1998 */
1999 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01002000skiptowhite(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002001{
2002 while (*p != ' ' && *p != '\t' && *p != NUL)
2003 ++p;
2004 return p;
2005}
2006
Bram Moolenaar071d4272004-06-13 20:20:40 +00002007/*
2008 * skiptowhite_esc: Like skiptowhite(), but also skip escaped chars
2009 */
2010 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01002011skiptowhite_esc(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002012{
2013 while (*p != ' ' && *p != '\t' && *p != NUL)
2014 {
2015 if ((*p == '\\' || *p == Ctrl_V) && *(p + 1) != NUL)
2016 ++p;
2017 ++p;
2018 }
2019 return p;
2020}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002021
2022/*
Bram Moolenaaraf377e32021-11-29 12:12:43 +00002023 * Get a number from a string and skip over it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002024 * Note: the argument is a pointer to a char_u pointer!
2025 */
2026 long
Bram Moolenaar7454a062016-01-30 15:14:10 +01002027getdigits(char_u **pp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002028{
2029 char_u *p;
2030 long retval;
2031
2032 p = *pp;
2033 retval = atol((char *)p);
Bram Moolenaarc667da52019-11-30 20:52:27 +01002034 if (*p == '-') // skip negative sign
Bram Moolenaar071d4272004-06-13 20:20:40 +00002035 ++p;
Bram Moolenaarc667da52019-11-30 20:52:27 +01002036 p = skipdigits(p); // skip to next non-digit
Bram Moolenaar071d4272004-06-13 20:20:40 +00002037 *pp = p;
2038 return retval;
2039}
2040
2041/*
Bram Moolenaaraf377e32021-11-29 12:12:43 +00002042 * Like getdigits() but allow for embedded single quotes.
2043 */
2044 long
2045getdigits_quoted(char_u **pp)
2046{
2047 char_u *p = *pp;
2048 long retval = 0;
2049
2050 if (*p == '-')
2051 ++p;
2052 while (VIM_ISDIGIT(*p))
2053 {
2054 if (retval >= LONG_MAX / 10 - 10)
2055 retval = LONG_MAX;
2056 else
2057 retval = retval * 10 - '0' + *p;
2058 ++p;
2059 if (in_vim9script() && *p == '\'' && VIM_ISDIGIT(p[1]))
2060 ++p;
2061 }
2062 if (**pp == '-')
2063 {
2064 if (retval == LONG_MAX)
2065 retval = LONG_MIN;
2066 else
2067 retval = -retval;
2068 }
2069 *pp = p;
2070 return retval;
2071}
2072
2073/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002074 * Return TRUE if "lbuf" is empty or only contains blanks.
2075 */
2076 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002077vim_isblankline(char_u *lbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002078{
2079 char_u *p;
2080
2081 p = skipwhite(lbuf);
2082 return (*p == NUL || *p == '\r' || *p == '\n');
2083}
2084
2085/*
2086 * Convert a string into a long and/or unsigned long, taking care of
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002087 * hexadecimal, octal, and binary numbers. Accepts a '-' sign.
2088 * If "prep" is not NULL, returns a flag to indicate the type of the number:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002089 * 0 decimal
2090 * '0' octal
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002091 * 'O' octal
2092 * 'o' octal
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002093 * 'B' bin
2094 * 'b' bin
Bram Moolenaar071d4272004-06-13 20:20:40 +00002095 * 'X' hex
2096 * 'x' hex
2097 * If "len" is not NULL, the length of the number in characters is returned.
2098 * If "nptr" is not NULL, the signed result is returned in it.
2099 * If "unptr" is not NULL, the unsigned result is returned in it.
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002100 * If "what" contains STR2NR_BIN recognize binary numbers
2101 * If "what" contains STR2NR_OCT recognize octal numbers
2102 * If "what" contains STR2NR_HEX recognize hex numbers
2103 * If "what" contains STR2NR_FORCE always assume bin/oct/hex.
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002104 * If "what" contains STR2NR_QUOTE ignore embedded single quotes
Bram Moolenaarce157752017-10-28 16:07:33 +02002105 * If maxlen > 0, check at a maximum maxlen chars.
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002106 * If strict is TRUE, check the number strictly. return *len = 0 if fail.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002107 */
2108 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002109vim_str2nr(
2110 char_u *start,
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002111 int *prep, // return: type of number 0 = decimal, 'x'
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002112 // or 'X' is hex, '0', 'o' or 'O' is octal,
2113 // 'b' or 'B' is bin
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002114 int *len, // return: detected length of number
2115 int what, // what numbers to recognize
2116 varnumber_T *nptr, // return: signed result
2117 uvarnumber_T *unptr, // return: unsigned result
2118 int maxlen, // max length of string to check
2119 int strict) // check strictly
Bram Moolenaar071d4272004-06-13 20:20:40 +00002120{
2121 char_u *ptr = start;
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002122 int pre = 0; // default is decimal
Bram Moolenaar071d4272004-06-13 20:20:40 +00002123 int negative = FALSE;
Bram Moolenaar22fcfad2016-07-01 18:17:26 +02002124 uvarnumber_T un = 0;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002125 int n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002127 if (len != NULL)
2128 *len = 0;
2129
Bram Moolenaar071d4272004-06-13 20:20:40 +00002130 if (ptr[0] == '-')
2131 {
2132 negative = TRUE;
2133 ++ptr;
2134 }
2135
Bram Moolenaarc667da52019-11-30 20:52:27 +01002136 // Recognize hex, octal, and bin.
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002137 if (ptr[0] == '0' && ptr[1] != '8' && ptr[1] != '9'
2138 && (maxlen == 0 || maxlen > 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002139 {
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002140 pre = ptr[1];
2141 if ((what & STR2NR_HEX)
2142 && (pre == 'X' || pre == 'x') && vim_isxdigit(ptr[2])
2143 && (maxlen == 0 || maxlen > 2))
Bram Moolenaarc667da52019-11-30 20:52:27 +01002144 // hexadecimal
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002145 ptr += 2;
2146 else if ((what & STR2NR_BIN)
2147 && (pre == 'B' || pre == 'b') && vim_isbdigit(ptr[2])
2148 && (maxlen == 0 || maxlen > 2))
Bram Moolenaarc667da52019-11-30 20:52:27 +01002149 // binary
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002150 ptr += 2;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002151 else if ((what & STR2NR_OOCT)
Bram Moolenaarc37b6552021-01-07 19:36:30 +01002152 && (pre == 'O' || pre == 'o') && vim_isodigit(ptr[2])
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002153 && (maxlen == 0 || maxlen > 2))
2154 // octal with prefix "0o"
2155 ptr += 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002156 else
2157 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002158 // decimal or octal, default is decimal
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002159 pre = 0;
2160 if (what & STR2NR_OCT)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002161 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002162 // Don't interpret "0", "08" or "0129" as octal.
Bram Moolenaarce157752017-10-28 16:07:33 +02002163 for (n = 1; n != maxlen && VIM_ISDIGIT(ptr[n]); ++n)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002164 {
2165 if (ptr[n] > '7')
2166 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002167 pre = 0; // can't be octal
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002168 break;
2169 }
Bram Moolenaarc667da52019-11-30 20:52:27 +01002170 pre = '0'; // assume octal
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002171 }
2172 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002173 }
2174 }
2175
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002176 // Do the conversion manually to avoid sscanf() quirks.
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002177 n = 1;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002178 if (pre == 'B' || pre == 'b'
2179 || ((what & STR2NR_BIN) && (what & STR2NR_FORCE)))
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002180 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002181 // bin
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002182 if (pre != 0)
Bram Moolenaarc667da52019-11-30 20:52:27 +01002183 n += 2; // skip over "0b"
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002184 while ('0' <= *ptr && *ptr <= '1')
2185 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002186 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002187 if (un <= UVARNUM_MAX / 2)
2188 un = 2 * un + (uvarnumber_T)(*ptr - '0');
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002189 else
2190 un = UVARNUM_MAX;
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002191 ++ptr;
2192 if (n++ == maxlen)
2193 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002194 if ((what & STR2NR_QUOTE) && *ptr == '\''
2195 && '0' <= ptr[1] && ptr[1] <= '1')
2196 {
2197 ++ptr;
2198 if (n++ == maxlen)
2199 break;
2200 }
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002201 }
2202 }
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002203 else if (pre == 'O' || pre == 'o' ||
2204 pre == '0' || ((what & STR2NR_OCT) && (what & STR2NR_FORCE)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002205 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002206 // octal
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02002207 if (pre != 0 && pre != '0')
2208 n += 2; // skip over "0o"
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002209 while ('0' <= *ptr && *ptr <= '7')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002210 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002211 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002212 if (un <= UVARNUM_MAX / 8)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002213 un = 8 * un + (uvarnumber_T)(*ptr - '0');
2214 else
2215 un = UVARNUM_MAX;
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002216 ++ptr;
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002217 if (n++ == maxlen)
2218 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002219 if ((what & STR2NR_QUOTE) && *ptr == '\''
2220 && '0' <= ptr[1] && ptr[1] <= '7')
2221 {
2222 ++ptr;
2223 if (n++ == maxlen)
2224 break;
2225 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002226 }
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002227 }
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002228 else if (pre != 0 || ((what & STR2NR_HEX) && (what & STR2NR_FORCE)))
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002229 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002230 // hex
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002231 if (pre != 0)
Bram Moolenaarc667da52019-11-30 20:52:27 +01002232 n += 2; // skip over "0x"
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002233 while (vim_isxdigit(*ptr))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002235 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002236 if (un <= UVARNUM_MAX / 16)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002237 un = 16 * un + (uvarnumber_T)hex2nr(*ptr);
2238 else
2239 un = UVARNUM_MAX;
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002240 ++ptr;
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002241 if (n++ == maxlen)
2242 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002243 if ((what & STR2NR_QUOTE) && *ptr == '\'' && vim_isxdigit(ptr[1]))
2244 {
2245 ++ptr;
2246 if (n++ == maxlen)
2247 break;
2248 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002249 }
2250 }
2251 else
2252 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002253 // decimal
Bram Moolenaar071d4272004-06-13 20:20:40 +00002254 while (VIM_ISDIGIT(*ptr))
2255 {
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002256 uvarnumber_T digit = (uvarnumber_T)(*ptr - '0');
2257
Bram Moolenaarc667da52019-11-30 20:52:27 +01002258 // avoid ubsan error for overflow
Bram Moolenaar07ccf7c2018-06-12 17:25:36 +02002259 if (un < UVARNUM_MAX / 10
2260 || (un == UVARNUM_MAX / 10 && digit <= UVARNUM_MAX % 10))
2261 un = 10 * un + digit;
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002262 else
2263 un = UVARNUM_MAX;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002264 ++ptr;
Bram Moolenaar5d1bc782015-07-17 13:03:48 +02002265 if (n++ == maxlen)
2266 break;
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002267 if ((what & STR2NR_QUOTE) && *ptr == '\'' && VIM_ISDIGIT(ptr[1]))
2268 {
2269 ++ptr;
2270 if (n++ == maxlen)
2271 break;
2272 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002273 }
2274 }
Bram Moolenaar1ac90b42019-09-15 14:49:52 +02002275
Bram Moolenaar4b96df52020-01-26 22:00:26 +01002276 // Check for an alphanumeric character immediately following, that is
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002277 // most likely a typo.
2278 if (strict && n - 1 != maxlen && ASCII_ISALNUM(*ptr))
2279 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280
Bram Moolenaar887c1fe2016-01-02 17:56:35 +01002281 if (prep != NULL)
2282 *prep = pre;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002283 if (len != NULL)
2284 *len = (int)(ptr - start);
2285 if (nptr != NULL)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002286 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002287 if (negative) // account for leading '-' for decimal numbers
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002288 {
Bram Moolenaarc667da52019-11-30 20:52:27 +01002289 // avoid ubsan error for overflow
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002290 if (un > VARNUM_MAX)
2291 *nptr = VARNUM_MIN;
2292 else
2293 *nptr = -(varnumber_T)un;
2294 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002295 else
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002296 {
Bram Moolenaar338bf582022-05-22 20:16:32 +01002297 // prevent a larg unsigned number to become negative
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002298 if (un > VARNUM_MAX)
2299 un = VARNUM_MAX;
Bram Moolenaar22fcfad2016-07-01 18:17:26 +02002300 *nptr = (varnumber_T)un;
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002301 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002302 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002303 if (unptr != NULL)
2304 *unptr = un;
2305}
2306
2307/*
2308 * Return the value of a single hex character.
2309 * Only valid when the argument is '0' - '9', 'A' - 'F' or 'a' - 'f'.
2310 */
2311 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002312hex2nr(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002313{
2314 if (c >= 'a' && c <= 'f')
2315 return c - 'a' + 10;
2316 if (c >= 'A' && c <= 'F')
2317 return c - 'A' + 10;
2318 return c - '0';
2319}
2320
Bram Moolenaar071d4272004-06-13 20:20:40 +00002321/*
2322 * Convert two hex characters to a byte.
2323 * Return -1 if one of the characters is not hex.
2324 */
2325 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002326hexhex2nr(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002327{
2328 if (!vim_isxdigit(p[0]) || !vim_isxdigit(p[1]))
2329 return -1;
2330 return (hex2nr(p[0]) << 4) + hex2nr(p[1]);
2331}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002332
2333/*
2334 * Return TRUE if "str" starts with a backslash that should be removed.
Bram Moolenaar2c519cf2019-03-21 21:45:34 +01002335 * For MS-DOS, MSWIN and OS/2 this is only done when the character after the
Bram Moolenaar071d4272004-06-13 20:20:40 +00002336 * backslash is not a normal file name character.
2337 * '$' is a valid file name character, we don't remove the backslash before
2338 * it. This means it is not possible to use an environment variable after a
2339 * backslash. "C:\$VIM\doc" is taken literally, only "$VIM\doc" works.
2340 * Although "\ name" is valid, the backslash in "Program\ files" must be
2341 * removed. Assume a file name doesn't start with a space.
2342 * For multi-byte names, never remove a backslash before a non-ascii
2343 * character, assume that all multi-byte characters are valid file name
2344 * characters.
2345 */
2346 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002347rem_backslash(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002348{
2349#ifdef BACKSLASH_IN_FILENAME
2350 return (str[0] == '\\'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002351 && str[1] < 0x80
Bram Moolenaar071d4272004-06-13 20:20:40 +00002352 && (str[1] == ' '
2353 || (str[1] != NUL
2354 && str[1] != '*'
2355 && str[1] != '?'
2356 && !vim_isfilec(str[1]))));
2357#else
2358 return (str[0] == '\\' && str[1] != NUL);
2359#endif
2360}
2361
2362/*
2363 * Halve the number of backslashes in a file name argument.
2364 * For MS-DOS we only do this if the character after the backslash
2365 * is not a normal file character.
2366 */
2367 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002368backslash_halve(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002369{
2370 for ( ; *p; ++p)
2371 if (rem_backslash(p))
Bram Moolenaar446cb832008-06-24 21:56:24 +00002372 STRMOVE(p, p + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002373}
2374
2375/*
2376 * backslash_halve() plus save the result in allocated memory.
Bram Moolenaare2c453d2019-08-21 14:37:09 +02002377 * However, returns "p" when out of memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002378 */
2379 char_u *
Bram Moolenaar7454a062016-01-30 15:14:10 +01002380backslash_halve_save(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002381{
2382 char_u *res;
2383
2384 res = vim_strsave(p);
2385 if (res == NULL)
2386 return p;
2387 backslash_halve(res);
2388 return res;
2389}