xref: /vim-8.2.3635/src/fold.c (revision d99df423)
1071d4279SBram Moolenaar /* vim:set ts=8 sts=4 sw=4:
2071d4279SBram Moolenaar  * vim600:fdm=marker fdl=1 fdc=3:
3071d4279SBram Moolenaar  *
4071d4279SBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
5071d4279SBram Moolenaar  *
6071d4279SBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
7071d4279SBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
8071d4279SBram Moolenaar  * See README.txt for an overview of the Vim source code.
9071d4279SBram Moolenaar  */
10071d4279SBram Moolenaar 
11071d4279SBram Moolenaar /*
12071d4279SBram Moolenaar  * fold.c: code for folding
13071d4279SBram Moolenaar  */
14071d4279SBram Moolenaar 
15071d4279SBram Moolenaar #include "vim.h"
16071d4279SBram Moolenaar 
17071d4279SBram Moolenaar #if defined(FEAT_FOLDING) || defined(PROTO)
18071d4279SBram Moolenaar 
19071d4279SBram Moolenaar /* local declarations. {{{1 */
20071d4279SBram Moolenaar /* typedef fold_T {{{2 */
21071d4279SBram Moolenaar /*
22071d4279SBram Moolenaar  * The toplevel folds for each window are stored in the w_folds growarray.
23071d4279SBram Moolenaar  * Each toplevel fold can contain an array of second level folds in the
24071d4279SBram Moolenaar  * fd_nested growarray.
25071d4279SBram Moolenaar  * The info stored in both growarrays is the same: An array of fold_T.
26071d4279SBram Moolenaar  */
27071d4279SBram Moolenaar typedef struct
28071d4279SBram Moolenaar {
29071d4279SBram Moolenaar     linenr_T	fd_top;		/* first line of fold; for nested fold
30071d4279SBram Moolenaar 				 * relative to parent */
31071d4279SBram Moolenaar     linenr_T	fd_len;		/* number of lines in the fold */
32071d4279SBram Moolenaar     garray_T	fd_nested;	/* array of nested folds */
33071d4279SBram Moolenaar     char	fd_flags;	/* see below */
34071d4279SBram Moolenaar     char	fd_small;	/* TRUE, FALSE or MAYBE: fold smaller than
35071d4279SBram Moolenaar 				   'foldminlines'; MAYBE applies to nested
36071d4279SBram Moolenaar 				   folds too */
37071d4279SBram Moolenaar } fold_T;
38071d4279SBram Moolenaar 
39071d4279SBram Moolenaar #define FD_OPEN		0	/* fold is open (nested ones can be closed) */
40071d4279SBram Moolenaar #define FD_CLOSED	1	/* fold is closed */
41071d4279SBram Moolenaar #define FD_LEVEL	2	/* depends on 'foldlevel' (nested folds too) */
42071d4279SBram Moolenaar 
43071d4279SBram Moolenaar #define MAX_LEVEL	20	/* maximum fold depth */
44071d4279SBram Moolenaar 
45071d4279SBram Moolenaar /* static functions {{{2 */
46d25c16e2SBram Moolenaar static void newFoldLevelWin(win_T *wp);
47d25c16e2SBram Moolenaar static int checkCloseRec(garray_T *gap, linenr_T lnum, int level);
48d25c16e2SBram Moolenaar static int foldFind(garray_T *gap, linenr_T lnum, fold_T **fpp);
49d25c16e2SBram Moolenaar static int foldLevelWin(win_T *wp, linenr_T lnum);
50d25c16e2SBram Moolenaar static void checkupdate(win_T *wp);
51d25c16e2SBram Moolenaar static void setFoldRepeat(linenr_T lnum, long count, int do_open);
52d25c16e2SBram Moolenaar static linenr_T setManualFold(linenr_T lnum, int opening, int recurse, int *donep);
53d25c16e2SBram Moolenaar static linenr_T setManualFoldWin(win_T *wp, linenr_T lnum, int opening, int recurse, int *donep);
54d25c16e2SBram Moolenaar static void foldOpenNested(fold_T *fpr);
55d25c16e2SBram Moolenaar static void deleteFoldEntry(garray_T *gap, int idx, int recursive);
56d25c16e2SBram Moolenaar static void foldMarkAdjustRecurse(garray_T *gap, linenr_T line1, linenr_T line2, long amount, long amount_after);
57d25c16e2SBram Moolenaar static int getDeepestNestingRecurse(garray_T *gap);
58d25c16e2SBram Moolenaar static int check_closed(win_T *win, fold_T *fp, int *use_levelp, int level, int *maybe_smallp, linenr_T lnum_off);
59d25c16e2SBram Moolenaar static void checkSmall(win_T *wp, fold_T *fp, linenr_T lnum_off);
60d25c16e2SBram Moolenaar static void setSmallMaybe(garray_T *gap);
61d25c16e2SBram Moolenaar static void foldCreateMarkers(linenr_T start, linenr_T end);
62d25c16e2SBram Moolenaar static void foldAddMarker(linenr_T lnum, char_u *marker, int markerlen);
63d25c16e2SBram Moolenaar static void deleteFoldMarkers(fold_T *fp, int recursive, linenr_T lnum_off);
64d25c16e2SBram Moolenaar static void foldDelMarker(linenr_T lnum, char_u *marker, int markerlen);
65d25c16e2SBram Moolenaar static void foldUpdateIEMS(win_T *wp, linenr_T top, linenr_T bot);
66d25c16e2SBram Moolenaar static void parseMarker(win_T *wp);
67071d4279SBram Moolenaar 
68071d4279SBram Moolenaar static char *e_nofold = N_("E490: No fold found");
69071d4279SBram Moolenaar 
70071d4279SBram Moolenaar /*
71071d4279SBram Moolenaar  * While updating the folds lines between invalid_top and invalid_bot have an
72071d4279SBram Moolenaar  * undefined fold level.  Only used for the window currently being updated.
73071d4279SBram Moolenaar  */
74071d4279SBram Moolenaar static linenr_T invalid_top = (linenr_T)0;
75071d4279SBram Moolenaar static linenr_T invalid_bot = (linenr_T)0;
76071d4279SBram Moolenaar 
77071d4279SBram Moolenaar /*
78071d4279SBram Moolenaar  * When using 'foldexpr' we sometimes get the level of the next line, which
79071d4279SBram Moolenaar  * calls foldlevel() to get the level of the current line, which hasn't been
80071d4279SBram Moolenaar  * stored yet.  To get around this chicken-egg problem the level of the
81071d4279SBram Moolenaar  * previous line is stored here when available.  prev_lnum is zero when the
82071d4279SBram Moolenaar  * level is not available.
83071d4279SBram Moolenaar  */
84071d4279SBram Moolenaar static linenr_T prev_lnum = 0;
85071d4279SBram Moolenaar static int prev_lnum_lvl = -1;
86071d4279SBram Moolenaar 
87071d4279SBram Moolenaar /* Flags used for "done" argument of setManualFold. */
88071d4279SBram Moolenaar #define DONE_NOTHING	0
89071d4279SBram Moolenaar #define DONE_ACTION	1	/* did close or open a fold */
90071d4279SBram Moolenaar #define DONE_FOLD	2	/* did find a fold */
91071d4279SBram Moolenaar 
92071d4279SBram Moolenaar static int foldstartmarkerlen;
93071d4279SBram Moolenaar static char_u *foldendmarker;
94071d4279SBram Moolenaar static int foldendmarkerlen;
95071d4279SBram Moolenaar 
96071d4279SBram Moolenaar /* Exported folding functions. {{{1 */
97071d4279SBram Moolenaar /* copyFoldingState() {{{2 */
98071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO)
99071d4279SBram Moolenaar /*
100071d4279SBram Moolenaar  * Copy that folding state from window "wp_from" to window "wp_to".
101071d4279SBram Moolenaar  */
102071d4279SBram Moolenaar     void
103071d4279SBram Moolenaar copyFoldingState(wp_from, wp_to)
104071d4279SBram Moolenaar     win_T	*wp_from;
105071d4279SBram Moolenaar     win_T	*wp_to;
106071d4279SBram Moolenaar {
107071d4279SBram Moolenaar     wp_to->w_fold_manual = wp_from->w_fold_manual;
108071d4279SBram Moolenaar     wp_to->w_foldinvalid = wp_from->w_foldinvalid;
109071d4279SBram Moolenaar     cloneFoldGrowArray(&wp_from->w_folds, &wp_to->w_folds);
110071d4279SBram Moolenaar }
111071d4279SBram Moolenaar #endif
112071d4279SBram Moolenaar 
113071d4279SBram Moolenaar /* hasAnyFolding() {{{2 */
114071d4279SBram Moolenaar /*
115071d4279SBram Moolenaar  * Return TRUE if there may be folded lines in the current window.
116071d4279SBram Moolenaar  */
117071d4279SBram Moolenaar     int
118071d4279SBram Moolenaar hasAnyFolding(win)
119071d4279SBram Moolenaar     win_T	*win;
120071d4279SBram Moolenaar {
121071d4279SBram Moolenaar     /* very simple now, but can become more complex later */
122071d4279SBram Moolenaar     return (win->w_p_fen
123071d4279SBram Moolenaar 	    && (!foldmethodIsManual(win) || win->w_folds.ga_len > 0));
124071d4279SBram Moolenaar }
125071d4279SBram Moolenaar 
126071d4279SBram Moolenaar /* hasFolding() {{{2 */
127071d4279SBram Moolenaar /*
128071d4279SBram Moolenaar  * Return TRUE if line "lnum" in the current window is part of a closed
129071d4279SBram Moolenaar  * fold.
130071d4279SBram Moolenaar  * When returning TRUE, *firstp and *lastp are set to the first and last
131071d4279SBram Moolenaar  * lnum of the sequence of folded lines (skipped when NULL).
132071d4279SBram Moolenaar  */
133071d4279SBram Moolenaar     int
134071d4279SBram Moolenaar hasFolding(lnum, firstp, lastp)
135071d4279SBram Moolenaar     linenr_T	lnum;
136071d4279SBram Moolenaar     linenr_T	*firstp;
137071d4279SBram Moolenaar     linenr_T	*lastp;
138071d4279SBram Moolenaar {
139071d4279SBram Moolenaar     return hasFoldingWin(curwin, lnum, firstp, lastp, TRUE, NULL);
140071d4279SBram Moolenaar }
141071d4279SBram Moolenaar 
142071d4279SBram Moolenaar /* hasFoldingWin() {{{2 */
143071d4279SBram Moolenaar     int
144071d4279SBram Moolenaar hasFoldingWin(win, lnum, firstp, lastp, cache, infop)
145071d4279SBram Moolenaar     win_T	*win;
146071d4279SBram Moolenaar     linenr_T	lnum;
147071d4279SBram Moolenaar     linenr_T	*firstp;
148071d4279SBram Moolenaar     linenr_T	*lastp;
149071d4279SBram Moolenaar     int		cache;		/* when TRUE: use cached values of window */
150071d4279SBram Moolenaar     foldinfo_T	*infop;		/* where to store fold info */
151071d4279SBram Moolenaar {
152071d4279SBram Moolenaar     int		had_folded = FALSE;
153071d4279SBram Moolenaar     linenr_T	first = 0;
154071d4279SBram Moolenaar     linenr_T	last = 0;
155071d4279SBram Moolenaar     linenr_T	lnum_rel = lnum;
156071d4279SBram Moolenaar     int		x;
157071d4279SBram Moolenaar     fold_T	*fp;
158071d4279SBram Moolenaar     int		level = 0;
159071d4279SBram Moolenaar     int		use_level = FALSE;
160071d4279SBram Moolenaar     int		maybe_small = FALSE;
161071d4279SBram Moolenaar     garray_T	*gap;
162071d4279SBram Moolenaar     int		low_level = 0;;
163071d4279SBram Moolenaar 
164071d4279SBram Moolenaar     checkupdate(win);
165071d4279SBram Moolenaar     /*
166071d4279SBram Moolenaar      * Return quickly when there is no folding at all in this window.
167071d4279SBram Moolenaar      */
168071d4279SBram Moolenaar     if (!hasAnyFolding(win))
169071d4279SBram Moolenaar     {
170071d4279SBram Moolenaar 	if (infop != NULL)
171071d4279SBram Moolenaar 	    infop->fi_level = 0;
172071d4279SBram Moolenaar 	return FALSE;
173071d4279SBram Moolenaar     }
174071d4279SBram Moolenaar 
175071d4279SBram Moolenaar     if (cache)
176071d4279SBram Moolenaar     {
177071d4279SBram Moolenaar 	/*
178071d4279SBram Moolenaar 	 * First look in cached info for displayed lines.  This is probably
179071d4279SBram Moolenaar 	 * the fastest, but it can only be used if the entry is still valid.
180071d4279SBram Moolenaar 	 */
181071d4279SBram Moolenaar 	x = find_wl_entry(win, lnum);
182071d4279SBram Moolenaar 	if (x >= 0)
183071d4279SBram Moolenaar 	{
184071d4279SBram Moolenaar 	    first = win->w_lines[x].wl_lnum;
185071d4279SBram Moolenaar 	    last = win->w_lines[x].wl_lastlnum;
186071d4279SBram Moolenaar 	    had_folded = win->w_lines[x].wl_folded;
187071d4279SBram Moolenaar 	}
188071d4279SBram Moolenaar     }
189071d4279SBram Moolenaar 
190071d4279SBram Moolenaar     if (first == 0)
191071d4279SBram Moolenaar     {
192071d4279SBram Moolenaar 	/*
193071d4279SBram Moolenaar 	 * Recursively search for a fold that contains "lnum".
194071d4279SBram Moolenaar 	 */
195071d4279SBram Moolenaar 	gap = &win->w_folds;
196071d4279SBram Moolenaar 	for (;;)
197071d4279SBram Moolenaar 	{
198071d4279SBram Moolenaar 	    if (!foldFind(gap, lnum_rel, &fp))
199071d4279SBram Moolenaar 		break;
200071d4279SBram Moolenaar 
201071d4279SBram Moolenaar 	    /* Remember lowest level of fold that starts in "lnum". */
202071d4279SBram Moolenaar 	    if (lnum_rel == fp->fd_top && low_level == 0)
203071d4279SBram Moolenaar 		low_level = level + 1;
204071d4279SBram Moolenaar 
205071d4279SBram Moolenaar 	    first += fp->fd_top;
206071d4279SBram Moolenaar 	    last += fp->fd_top;
207071d4279SBram Moolenaar 
208071d4279SBram Moolenaar 	    /* is this fold closed? */
209071d4279SBram Moolenaar 	    had_folded = check_closed(win, fp, &use_level, level,
210071d4279SBram Moolenaar 					       &maybe_small, lnum - lnum_rel);
211071d4279SBram Moolenaar 	    if (had_folded)
212071d4279SBram Moolenaar 	    {
213071d4279SBram Moolenaar 		/* Fold closed: Set last and quit loop. */
214071d4279SBram Moolenaar 		last += fp->fd_len - 1;
215071d4279SBram Moolenaar 		break;
216071d4279SBram Moolenaar 	    }
217071d4279SBram Moolenaar 
218071d4279SBram Moolenaar 	    /* Fold found, but it's open: Check nested folds.  Line number is
219071d4279SBram Moolenaar 	     * relative to containing fold. */
220071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
221071d4279SBram Moolenaar 	    lnum_rel -= fp->fd_top;
222071d4279SBram Moolenaar 	    ++level;
223071d4279SBram Moolenaar 	}
224071d4279SBram Moolenaar     }
225071d4279SBram Moolenaar 
226071d4279SBram Moolenaar     if (!had_folded)
227071d4279SBram Moolenaar     {
228071d4279SBram Moolenaar 	if (infop != NULL)
229071d4279SBram Moolenaar 	{
230071d4279SBram Moolenaar 	    infop->fi_level = level;
231071d4279SBram Moolenaar 	    infop->fi_lnum = lnum - lnum_rel;
232071d4279SBram Moolenaar 	    infop->fi_low_level = low_level == 0 ? level : low_level;
233071d4279SBram Moolenaar 	}
234071d4279SBram Moolenaar 	return FALSE;
235071d4279SBram Moolenaar     }
236071d4279SBram Moolenaar 
23705b20fb1SBram Moolenaar     if (last > win->w_buffer->b_ml.ml_line_count)
23805b20fb1SBram Moolenaar 	last = win->w_buffer->b_ml.ml_line_count;
239071d4279SBram Moolenaar     if (lastp != NULL)
240071d4279SBram Moolenaar 	*lastp = last;
241071d4279SBram Moolenaar     if (firstp != NULL)
242071d4279SBram Moolenaar 	*firstp = first;
243071d4279SBram Moolenaar     if (infop != NULL)
244071d4279SBram Moolenaar     {
245071d4279SBram Moolenaar 	infop->fi_level = level + 1;
246071d4279SBram Moolenaar 	infop->fi_lnum = first;
247071d4279SBram Moolenaar 	infop->fi_low_level = low_level == 0 ? level + 1 : low_level;
248071d4279SBram Moolenaar     }
249071d4279SBram Moolenaar     return TRUE;
250071d4279SBram Moolenaar }
251071d4279SBram Moolenaar 
252071d4279SBram Moolenaar /* foldLevel() {{{2 */
253071d4279SBram Moolenaar /*
254071d4279SBram Moolenaar  * Return fold level at line number "lnum" in the current window.
255071d4279SBram Moolenaar  */
256071d4279SBram Moolenaar     int
257071d4279SBram Moolenaar foldLevel(lnum)
258071d4279SBram Moolenaar     linenr_T	lnum;
259071d4279SBram Moolenaar {
260071d4279SBram Moolenaar     /* While updating the folds lines between invalid_top and invalid_bot have
261071d4279SBram Moolenaar      * an undefined fold level.  Otherwise update the folds first. */
262071d4279SBram Moolenaar     if (invalid_top == (linenr_T)0)
263071d4279SBram Moolenaar 	checkupdate(curwin);
264071d4279SBram Moolenaar     else if (lnum == prev_lnum && prev_lnum_lvl >= 0)
265071d4279SBram Moolenaar 	return prev_lnum_lvl;
266071d4279SBram Moolenaar     else if (lnum >= invalid_top && lnum <= invalid_bot)
267071d4279SBram Moolenaar 	return -1;
268071d4279SBram Moolenaar 
269071d4279SBram Moolenaar     /* Return quickly when there is no folding at all in this window. */
270071d4279SBram Moolenaar     if (!hasAnyFolding(curwin))
271071d4279SBram Moolenaar 	return 0;
272071d4279SBram Moolenaar 
273071d4279SBram Moolenaar     return foldLevelWin(curwin, lnum);
274071d4279SBram Moolenaar }
275071d4279SBram Moolenaar 
276071d4279SBram Moolenaar /* lineFolded()	{{{2 */
277071d4279SBram Moolenaar /*
278071d4279SBram Moolenaar  * Low level function to check if a line is folded.  Doesn't use any caching.
279071d4279SBram Moolenaar  * Return TRUE if line is folded.
280071d4279SBram Moolenaar  * Return FALSE if line is not folded.
281071d4279SBram Moolenaar  * Return MAYBE if the line is folded when next to a folded line.
282071d4279SBram Moolenaar  */
283071d4279SBram Moolenaar     int
284071d4279SBram Moolenaar lineFolded(win, lnum)
285071d4279SBram Moolenaar     win_T	*win;
286071d4279SBram Moolenaar     linenr_T	lnum;
287071d4279SBram Moolenaar {
288071d4279SBram Moolenaar     return foldedCount(win, lnum, NULL) != 0;
289071d4279SBram Moolenaar }
290071d4279SBram Moolenaar 
291071d4279SBram Moolenaar /* foldedCount() {{{2 */
292071d4279SBram Moolenaar /*
293071d4279SBram Moolenaar  * Count the number of lines that are folded at line number "lnum".
294071d4279SBram Moolenaar  * Normally "lnum" is the first line of a possible fold, and the returned
295071d4279SBram Moolenaar  * number is the number of lines in the fold.
296071d4279SBram Moolenaar  * Doesn't use caching from the displayed window.
297071d4279SBram Moolenaar  * Returns number of folded lines from "lnum", or 0 if line is not folded.
298071d4279SBram Moolenaar  * When "infop" is not NULL, fills *infop with the fold level info.
299071d4279SBram Moolenaar  */
300071d4279SBram Moolenaar     long
301071d4279SBram Moolenaar foldedCount(win, lnum, infop)
302071d4279SBram Moolenaar     win_T	*win;
303071d4279SBram Moolenaar     linenr_T	lnum;
304071d4279SBram Moolenaar     foldinfo_T	*infop;
305071d4279SBram Moolenaar {
306071d4279SBram Moolenaar     linenr_T	last;
307071d4279SBram Moolenaar 
308071d4279SBram Moolenaar     if (hasFoldingWin(win, lnum, NULL, &last, FALSE, infop))
309071d4279SBram Moolenaar 	return (long)(last - lnum + 1);
310071d4279SBram Moolenaar     return 0;
311071d4279SBram Moolenaar }
312071d4279SBram Moolenaar 
313071d4279SBram Moolenaar /* foldmethodIsManual() {{{2 */
314071d4279SBram Moolenaar /*
315071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "manual"
316071d4279SBram Moolenaar  */
317071d4279SBram Moolenaar     int
318071d4279SBram Moolenaar foldmethodIsManual(wp)
319071d4279SBram Moolenaar     win_T	*wp;
320071d4279SBram Moolenaar {
321071d4279SBram Moolenaar     return (wp->w_p_fdm[3] == 'u');
322071d4279SBram Moolenaar }
323071d4279SBram Moolenaar 
324071d4279SBram Moolenaar /* foldmethodIsIndent() {{{2 */
325071d4279SBram Moolenaar /*
326071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "indent"
327071d4279SBram Moolenaar  */
328071d4279SBram Moolenaar     int
329071d4279SBram Moolenaar foldmethodIsIndent(wp)
330071d4279SBram Moolenaar     win_T	*wp;
331071d4279SBram Moolenaar {
332071d4279SBram Moolenaar     return (wp->w_p_fdm[0] == 'i');
333071d4279SBram Moolenaar }
334071d4279SBram Moolenaar 
335071d4279SBram Moolenaar /* foldmethodIsExpr() {{{2 */
336071d4279SBram Moolenaar /*
337071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "expr"
338071d4279SBram Moolenaar  */
339071d4279SBram Moolenaar     int
340071d4279SBram Moolenaar foldmethodIsExpr(wp)
341071d4279SBram Moolenaar     win_T	*wp;
342071d4279SBram Moolenaar {
343071d4279SBram Moolenaar     return (wp->w_p_fdm[1] == 'x');
344071d4279SBram Moolenaar }
345071d4279SBram Moolenaar 
346071d4279SBram Moolenaar /* foldmethodIsMarker() {{{2 */
347071d4279SBram Moolenaar /*
348071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "marker"
349071d4279SBram Moolenaar  */
350071d4279SBram Moolenaar     int
351071d4279SBram Moolenaar foldmethodIsMarker(wp)
352071d4279SBram Moolenaar     win_T	*wp;
353071d4279SBram Moolenaar {
354071d4279SBram Moolenaar     return (wp->w_p_fdm[2] == 'r');
355071d4279SBram Moolenaar }
356071d4279SBram Moolenaar 
357071d4279SBram Moolenaar /* foldmethodIsSyntax() {{{2 */
358071d4279SBram Moolenaar /*
359071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "syntax"
360071d4279SBram Moolenaar  */
361071d4279SBram Moolenaar     int
362071d4279SBram Moolenaar foldmethodIsSyntax(wp)
363071d4279SBram Moolenaar     win_T	*wp;
364071d4279SBram Moolenaar {
365071d4279SBram Moolenaar     return (wp->w_p_fdm[0] == 's');
366071d4279SBram Moolenaar }
367071d4279SBram Moolenaar 
368071d4279SBram Moolenaar /* foldmethodIsDiff() {{{2 */
369071d4279SBram Moolenaar /*
370071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "diff"
371071d4279SBram Moolenaar  */
372071d4279SBram Moolenaar     int
373071d4279SBram Moolenaar foldmethodIsDiff(wp)
374071d4279SBram Moolenaar     win_T	*wp;
375071d4279SBram Moolenaar {
376071d4279SBram Moolenaar     return (wp->w_p_fdm[0] == 'd');
377071d4279SBram Moolenaar }
378071d4279SBram Moolenaar 
379071d4279SBram Moolenaar /* closeFold() {{{2 */
380071d4279SBram Moolenaar /*
381071d4279SBram Moolenaar  * Close fold for current window at line "lnum".
382071d4279SBram Moolenaar  * Repeat "count" times.
383071d4279SBram Moolenaar  */
384071d4279SBram Moolenaar     void
385071d4279SBram Moolenaar closeFold(lnum, count)
386071d4279SBram Moolenaar     linenr_T	lnum;
387071d4279SBram Moolenaar     long	count;
388071d4279SBram Moolenaar {
389071d4279SBram Moolenaar     setFoldRepeat(lnum, count, FALSE);
390071d4279SBram Moolenaar }
391071d4279SBram Moolenaar 
392071d4279SBram Moolenaar /* closeFoldRecurse() {{{2 */
393071d4279SBram Moolenaar /*
394071d4279SBram Moolenaar  * Close fold for current window at line "lnum" recursively.
395071d4279SBram Moolenaar  */
396071d4279SBram Moolenaar     void
397071d4279SBram Moolenaar closeFoldRecurse(lnum)
398071d4279SBram Moolenaar     linenr_T	lnum;
399071d4279SBram Moolenaar {
400071d4279SBram Moolenaar     (void)setManualFold(lnum, FALSE, TRUE, NULL);
401071d4279SBram Moolenaar }
402071d4279SBram Moolenaar 
403071d4279SBram Moolenaar /* opFoldRange() {{{2 */
404071d4279SBram Moolenaar /*
405071d4279SBram Moolenaar  * Open or Close folds for current window in lines "first" to "last".
406071d4279SBram Moolenaar  * Used for "zo", "zO", "zc" and "zC" in Visual mode.
407071d4279SBram Moolenaar  */
408071d4279SBram Moolenaar     void
409071d4279SBram Moolenaar opFoldRange(first, last, opening, recurse, had_visual)
410071d4279SBram Moolenaar     linenr_T	first;
411071d4279SBram Moolenaar     linenr_T	last;
412071d4279SBram Moolenaar     int		opening;	/* TRUE to open, FALSE to close */
413071d4279SBram Moolenaar     int		recurse;	/* TRUE to do it recursively */
414071d4279SBram Moolenaar     int		had_visual;	/* TRUE when Visual selection used */
415071d4279SBram Moolenaar {
416071d4279SBram Moolenaar     int		done = DONE_NOTHING;	/* avoid error messages */
417071d4279SBram Moolenaar     linenr_T	lnum;
418071d4279SBram Moolenaar     linenr_T	lnum_next;
419071d4279SBram Moolenaar 
420071d4279SBram Moolenaar     for (lnum = first; lnum <= last; lnum = lnum_next + 1)
421071d4279SBram Moolenaar     {
422071d4279SBram Moolenaar 	lnum_next = lnum;
423071d4279SBram Moolenaar 	/* Opening one level only: next fold to open is after the one going to
424071d4279SBram Moolenaar 	 * be opened. */
425071d4279SBram Moolenaar 	if (opening && !recurse)
426071d4279SBram Moolenaar 	    (void)hasFolding(lnum, NULL, &lnum_next);
427071d4279SBram Moolenaar 	(void)setManualFold(lnum, opening, recurse, &done);
428071d4279SBram Moolenaar 	/* Closing one level only: next line to close a fold is after just
429071d4279SBram Moolenaar 	 * closed fold. */
430071d4279SBram Moolenaar 	if (!opening && !recurse)
431071d4279SBram Moolenaar 	    (void)hasFolding(lnum, NULL, &lnum_next);
432071d4279SBram Moolenaar     }
433071d4279SBram Moolenaar     if (done == DONE_NOTHING)
434071d4279SBram Moolenaar 	EMSG(_(e_nofold));
435071d4279SBram Moolenaar     /* Force a redraw to remove the Visual highlighting. */
436071d4279SBram Moolenaar     if (had_visual)
437071d4279SBram Moolenaar 	redraw_curbuf_later(INVERTED);
438071d4279SBram Moolenaar }
439071d4279SBram Moolenaar 
440071d4279SBram Moolenaar /* openFold() {{{2 */
441071d4279SBram Moolenaar /*
442071d4279SBram Moolenaar  * Open fold for current window at line "lnum".
443071d4279SBram Moolenaar  * Repeat "count" times.
444071d4279SBram Moolenaar  */
445071d4279SBram Moolenaar     void
446071d4279SBram Moolenaar openFold(lnum, count)
447071d4279SBram Moolenaar     linenr_T	lnum;
448071d4279SBram Moolenaar     long	count;
449071d4279SBram Moolenaar {
450071d4279SBram Moolenaar     setFoldRepeat(lnum, count, TRUE);
451071d4279SBram Moolenaar }
452071d4279SBram Moolenaar 
453071d4279SBram Moolenaar /* openFoldRecurse() {{{2 */
454071d4279SBram Moolenaar /*
455071d4279SBram Moolenaar  * Open fold for current window at line "lnum" recursively.
456071d4279SBram Moolenaar  */
457071d4279SBram Moolenaar     void
458071d4279SBram Moolenaar openFoldRecurse(lnum)
459071d4279SBram Moolenaar     linenr_T	lnum;
460071d4279SBram Moolenaar {
461071d4279SBram Moolenaar     (void)setManualFold(lnum, TRUE, TRUE, NULL);
462071d4279SBram Moolenaar }
463071d4279SBram Moolenaar 
464071d4279SBram Moolenaar /* foldOpenCursor() {{{2 */
465071d4279SBram Moolenaar /*
466071d4279SBram Moolenaar  * Open folds until the cursor line is not in a closed fold.
467071d4279SBram Moolenaar  */
468071d4279SBram Moolenaar     void
469071d4279SBram Moolenaar foldOpenCursor()
470071d4279SBram Moolenaar {
471071d4279SBram Moolenaar     int		done;
472071d4279SBram Moolenaar 
473071d4279SBram Moolenaar     checkupdate(curwin);
474071d4279SBram Moolenaar     if (hasAnyFolding(curwin))
475071d4279SBram Moolenaar 	for (;;)
476071d4279SBram Moolenaar 	{
477071d4279SBram Moolenaar 	    done = DONE_NOTHING;
478071d4279SBram Moolenaar 	    (void)setManualFold(curwin->w_cursor.lnum, TRUE, FALSE, &done);
479071d4279SBram Moolenaar 	    if (!(done & DONE_ACTION))
480071d4279SBram Moolenaar 		break;
481071d4279SBram Moolenaar 	}
482071d4279SBram Moolenaar }
483071d4279SBram Moolenaar 
484071d4279SBram Moolenaar /* newFoldLevel() {{{2 */
485071d4279SBram Moolenaar /*
486071d4279SBram Moolenaar  * Set new foldlevel for current window.
487071d4279SBram Moolenaar  */
488071d4279SBram Moolenaar     void
489071d4279SBram Moolenaar newFoldLevel()
490071d4279SBram Moolenaar {
491071d4279SBram Moolenaar     newFoldLevelWin(curwin);
492071d4279SBram Moolenaar 
493071d4279SBram Moolenaar #ifdef FEAT_DIFF
494071d4279SBram Moolenaar     if (foldmethodIsDiff(curwin) && curwin->w_p_scb)
495071d4279SBram Moolenaar     {
496071d4279SBram Moolenaar 	win_T	    *wp;
497071d4279SBram Moolenaar 
498071d4279SBram Moolenaar 	/*
499071d4279SBram Moolenaar 	 * Set the same foldlevel in other windows in diff mode.
500071d4279SBram Moolenaar 	 */
501071d4279SBram Moolenaar 	FOR_ALL_WINDOWS(wp)
502071d4279SBram Moolenaar 	{
503071d4279SBram Moolenaar 	    if (wp != curwin && foldmethodIsDiff(wp) && wp->w_p_scb)
504071d4279SBram Moolenaar 	    {
505071d4279SBram Moolenaar 		wp->w_p_fdl = curwin->w_p_fdl;
506071d4279SBram Moolenaar 		newFoldLevelWin(wp);
507071d4279SBram Moolenaar 	    }
508071d4279SBram Moolenaar 	}
509071d4279SBram Moolenaar     }
510071d4279SBram Moolenaar #endif
511071d4279SBram Moolenaar }
512071d4279SBram Moolenaar 
513071d4279SBram Moolenaar     static void
514071d4279SBram Moolenaar newFoldLevelWin(wp)
515071d4279SBram Moolenaar     win_T	*wp;
516071d4279SBram Moolenaar {
517071d4279SBram Moolenaar     fold_T	*fp;
518071d4279SBram Moolenaar     int		i;
519071d4279SBram Moolenaar 
520071d4279SBram Moolenaar     checkupdate(wp);
521071d4279SBram Moolenaar     if (wp->w_fold_manual)
522071d4279SBram Moolenaar     {
523071d4279SBram Moolenaar 	/* Set all flags for the first level of folds to FD_LEVEL.  Following
524071d4279SBram Moolenaar 	 * manual open/close will then change the flags to FD_OPEN or
525071d4279SBram Moolenaar 	 * FD_CLOSED for those folds that don't use 'foldlevel'. */
526071d4279SBram Moolenaar 	fp = (fold_T *)wp->w_folds.ga_data;
527071d4279SBram Moolenaar 	for (i = 0; i < wp->w_folds.ga_len; ++i)
528071d4279SBram Moolenaar 	    fp[i].fd_flags = FD_LEVEL;
529071d4279SBram Moolenaar 	wp->w_fold_manual = FALSE;
530071d4279SBram Moolenaar     }
531071d4279SBram Moolenaar     changed_window_setting_win(wp);
532071d4279SBram Moolenaar }
533071d4279SBram Moolenaar 
534071d4279SBram Moolenaar /* foldCheckClose() {{{2 */
535071d4279SBram Moolenaar /*
536071d4279SBram Moolenaar  * Apply 'foldlevel' to all folds that don't contain the cursor.
537071d4279SBram Moolenaar  */
538071d4279SBram Moolenaar     void
539071d4279SBram Moolenaar foldCheckClose()
540071d4279SBram Moolenaar {
541071d4279SBram Moolenaar     if (*p_fcl != NUL)	/* can only be "all" right now */
542071d4279SBram Moolenaar     {
543071d4279SBram Moolenaar 	checkupdate(curwin);
544071d4279SBram Moolenaar 	if (checkCloseRec(&curwin->w_folds, curwin->w_cursor.lnum,
545071d4279SBram Moolenaar 							(int)curwin->w_p_fdl))
546071d4279SBram Moolenaar 	    changed_window_setting();
547071d4279SBram Moolenaar     }
548071d4279SBram Moolenaar }
549071d4279SBram Moolenaar 
550071d4279SBram Moolenaar /* checkCloseRec() {{{2 */
551071d4279SBram Moolenaar     static int
552071d4279SBram Moolenaar checkCloseRec(gap, lnum, level)
553071d4279SBram Moolenaar     garray_T	*gap;
554071d4279SBram Moolenaar     linenr_T	lnum;
555071d4279SBram Moolenaar     int		level;
556071d4279SBram Moolenaar {
557071d4279SBram Moolenaar     fold_T	*fp;
558071d4279SBram Moolenaar     int		retval = FALSE;
559071d4279SBram Moolenaar     int		i;
560071d4279SBram Moolenaar 
561071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
562071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
563071d4279SBram Moolenaar     {
564071d4279SBram Moolenaar 	/* Only manually opened folds may need to be closed. */
565071d4279SBram Moolenaar 	if (fp[i].fd_flags == FD_OPEN)
566071d4279SBram Moolenaar 	{
567071d4279SBram Moolenaar 	    if (level <= 0 && (lnum < fp[i].fd_top
568071d4279SBram Moolenaar 				      || lnum >= fp[i].fd_top + fp[i].fd_len))
569071d4279SBram Moolenaar 	    {
570071d4279SBram Moolenaar 		fp[i].fd_flags = FD_LEVEL;
571071d4279SBram Moolenaar 		retval = TRUE;
572071d4279SBram Moolenaar 	    }
573071d4279SBram Moolenaar 	    else
574071d4279SBram Moolenaar 		retval |= checkCloseRec(&fp[i].fd_nested, lnum - fp[i].fd_top,
575071d4279SBram Moolenaar 								   level - 1);
576071d4279SBram Moolenaar 	}
577071d4279SBram Moolenaar     }
578071d4279SBram Moolenaar     return retval;
579071d4279SBram Moolenaar }
580071d4279SBram Moolenaar 
581071d4279SBram Moolenaar /* foldCreateAllowed() {{{2 */
582071d4279SBram Moolenaar /*
583071d4279SBram Moolenaar  * Return TRUE if it's allowed to manually create or delete a fold.
584071d4279SBram Moolenaar  * Give an error message and return FALSE if not.
585071d4279SBram Moolenaar  */
586071d4279SBram Moolenaar     int
587071d4279SBram Moolenaar foldManualAllowed(create)
588071d4279SBram Moolenaar     int		create;
589071d4279SBram Moolenaar {
590071d4279SBram Moolenaar     if (foldmethodIsManual(curwin) || foldmethodIsMarker(curwin))
591071d4279SBram Moolenaar 	return TRUE;
592071d4279SBram Moolenaar     if (create)
593071d4279SBram Moolenaar 	EMSG(_("E350: Cannot create fold with current 'foldmethod'"));
594071d4279SBram Moolenaar     else
595071d4279SBram Moolenaar 	EMSG(_("E351: Cannot delete fold with current 'foldmethod'"));
596071d4279SBram Moolenaar     return FALSE;
597071d4279SBram Moolenaar }
598071d4279SBram Moolenaar 
599071d4279SBram Moolenaar /* foldCreate() {{{2 */
600071d4279SBram Moolenaar /*
601071d4279SBram Moolenaar  * Create a fold from line "start" to line "end" (inclusive) in the current
602071d4279SBram Moolenaar  * window.
603071d4279SBram Moolenaar  */
604071d4279SBram Moolenaar     void
605071d4279SBram Moolenaar foldCreate(start, end)
606071d4279SBram Moolenaar     linenr_T	start;
607071d4279SBram Moolenaar     linenr_T	end;
608071d4279SBram Moolenaar {
609071d4279SBram Moolenaar     fold_T	*fp;
610071d4279SBram Moolenaar     garray_T	*gap;
611071d4279SBram Moolenaar     garray_T	fold_ga;
612071d4279SBram Moolenaar     int		i, j;
613071d4279SBram Moolenaar     int		cont;
614071d4279SBram Moolenaar     int		use_level = FALSE;
615071d4279SBram Moolenaar     int		closed = FALSE;
616071d4279SBram Moolenaar     int		level = 0;
617071d4279SBram Moolenaar     linenr_T	start_rel = start;
618071d4279SBram Moolenaar     linenr_T	end_rel = end;
619071d4279SBram Moolenaar 
620071d4279SBram Moolenaar     if (start > end)
621071d4279SBram Moolenaar     {
622071d4279SBram Moolenaar 	/* reverse the range */
623071d4279SBram Moolenaar 	end = start_rel;
624071d4279SBram Moolenaar 	start = end_rel;
625071d4279SBram Moolenaar 	start_rel = start;
626071d4279SBram Moolenaar 	end_rel = end;
627071d4279SBram Moolenaar     }
628071d4279SBram Moolenaar 
629071d4279SBram Moolenaar     /* When 'foldmethod' is "marker" add markers, which creates the folds. */
630071d4279SBram Moolenaar     if (foldmethodIsMarker(curwin))
631071d4279SBram Moolenaar     {
632071d4279SBram Moolenaar 	foldCreateMarkers(start, end);
633071d4279SBram Moolenaar 	return;
634071d4279SBram Moolenaar     }
635071d4279SBram Moolenaar 
636071d4279SBram Moolenaar     checkupdate(curwin);
637071d4279SBram Moolenaar 
638071d4279SBram Moolenaar     /* Find the place to insert the new fold. */
639071d4279SBram Moolenaar     gap = &curwin->w_folds;
640071d4279SBram Moolenaar     for (;;)
641071d4279SBram Moolenaar     {
642071d4279SBram Moolenaar 	if (!foldFind(gap, start_rel, &fp))
643071d4279SBram Moolenaar 	    break;
644071d4279SBram Moolenaar 	if (fp->fd_top + fp->fd_len > end_rel)
645071d4279SBram Moolenaar 	{
646071d4279SBram Moolenaar 	    /* New fold is completely inside this fold: Go one level deeper. */
647071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
648071d4279SBram Moolenaar 	    start_rel -= fp->fd_top;
649071d4279SBram Moolenaar 	    end_rel -= fp->fd_top;
650071d4279SBram Moolenaar 	    if (use_level || fp->fd_flags == FD_LEVEL)
651071d4279SBram Moolenaar 	    {
652071d4279SBram Moolenaar 		use_level = TRUE;
653071d4279SBram Moolenaar 		if (level >= curwin->w_p_fdl)
654071d4279SBram Moolenaar 		    closed = TRUE;
655071d4279SBram Moolenaar 	    }
656071d4279SBram Moolenaar 	    else if (fp->fd_flags == FD_CLOSED)
657071d4279SBram Moolenaar 		closed = TRUE;
658071d4279SBram Moolenaar 	    ++level;
659071d4279SBram Moolenaar 	}
660071d4279SBram Moolenaar 	else
661071d4279SBram Moolenaar 	{
662071d4279SBram Moolenaar 	    /* This fold and new fold overlap: Insert here and move some folds
663071d4279SBram Moolenaar 	     * inside the new fold. */
664071d4279SBram Moolenaar 	    break;
665071d4279SBram Moolenaar 	}
666071d4279SBram Moolenaar     }
667071d4279SBram Moolenaar 
668071d4279SBram Moolenaar     i = (int)(fp - (fold_T *)gap->ga_data);
669071d4279SBram Moolenaar     if (ga_grow(gap, 1) == OK)
670071d4279SBram Moolenaar     {
671071d4279SBram Moolenaar 	fp = (fold_T *)gap->ga_data + i;
672071d4279SBram Moolenaar 	ga_init2(&fold_ga, (int)sizeof(fold_T), 10);
673071d4279SBram Moolenaar 
674071d4279SBram Moolenaar 	/* Count number of folds that will be contained in the new fold. */
675071d4279SBram Moolenaar 	for (cont = 0; i + cont < gap->ga_len; ++cont)
676071d4279SBram Moolenaar 	    if (fp[cont].fd_top > end_rel)
677071d4279SBram Moolenaar 		break;
678071d4279SBram Moolenaar 	if (cont > 0 && ga_grow(&fold_ga, cont) == OK)
679071d4279SBram Moolenaar 	{
680071d4279SBram Moolenaar 	    /* If the first fold starts before the new fold, let the new fold
681071d4279SBram Moolenaar 	     * start there.  Otherwise the existing fold would change. */
682071d4279SBram Moolenaar 	    if (start_rel > fp->fd_top)
683071d4279SBram Moolenaar 		start_rel = fp->fd_top;
684071d4279SBram Moolenaar 
685071d4279SBram Moolenaar 	    /* When last contained fold isn't completely contained, adjust end
686071d4279SBram Moolenaar 	     * of new fold. */
687071d4279SBram Moolenaar 	    if (end_rel < fp[cont - 1].fd_top + fp[cont - 1].fd_len - 1)
688071d4279SBram Moolenaar 		end_rel = fp[cont - 1].fd_top + fp[cont - 1].fd_len - 1;
689071d4279SBram Moolenaar 	    /* Move contained folds to inside new fold. */
690071d4279SBram Moolenaar 	    mch_memmove(fold_ga.ga_data, fp, sizeof(fold_T) * cont);
691071d4279SBram Moolenaar 	    fold_ga.ga_len += cont;
692071d4279SBram Moolenaar 	    i += cont;
693071d4279SBram Moolenaar 
694071d4279SBram Moolenaar 	    /* Adjust line numbers in contained folds to be relative to the
695071d4279SBram Moolenaar 	     * new fold. */
696071d4279SBram Moolenaar 	    for (j = 0; j < cont; ++j)
697071d4279SBram Moolenaar 		((fold_T *)fold_ga.ga_data)[j].fd_top -= start_rel;
698071d4279SBram Moolenaar 	}
699071d4279SBram Moolenaar 	/* Move remaining entries to after the new fold. */
700071d4279SBram Moolenaar 	if (i < gap->ga_len)
701071d4279SBram Moolenaar 	    mch_memmove(fp + 1, (fold_T *)gap->ga_data + i,
702071d4279SBram Moolenaar 				     sizeof(fold_T) * (gap->ga_len - i));
703071d4279SBram Moolenaar 	gap->ga_len = gap->ga_len + 1 - cont;
704071d4279SBram Moolenaar 
705071d4279SBram Moolenaar 	/* insert new fold */
706071d4279SBram Moolenaar 	fp->fd_nested = fold_ga;
707071d4279SBram Moolenaar 	fp->fd_top = start_rel;
708071d4279SBram Moolenaar 	fp->fd_len = end_rel - start_rel + 1;
709071d4279SBram Moolenaar 
710071d4279SBram Moolenaar 	/* We want the new fold to be closed.  If it would remain open because
711071d4279SBram Moolenaar 	 * of using 'foldlevel', need to adjust fd_flags of containing folds.
712071d4279SBram Moolenaar 	 */
713071d4279SBram Moolenaar 	if (use_level && !closed && level < curwin->w_p_fdl)
714071d4279SBram Moolenaar 	    closeFold(start, 1L);
715071d4279SBram Moolenaar 	if (!use_level)
716071d4279SBram Moolenaar 	    curwin->w_fold_manual = TRUE;
717071d4279SBram Moolenaar 	fp->fd_flags = FD_CLOSED;
718071d4279SBram Moolenaar 	fp->fd_small = MAYBE;
719071d4279SBram Moolenaar 
720071d4279SBram Moolenaar 	/* redraw */
721071d4279SBram Moolenaar 	changed_window_setting();
722071d4279SBram Moolenaar     }
723071d4279SBram Moolenaar }
724071d4279SBram Moolenaar 
725071d4279SBram Moolenaar /* deleteFold() {{{2 */
726071d4279SBram Moolenaar /*
727071d4279SBram Moolenaar  * Delete a fold at line "start" in the current window.
728071d4279SBram Moolenaar  * When "end" is not 0, delete all folds from "start" to "end".
729071d4279SBram Moolenaar  * When "recursive" is TRUE delete recursively.
730071d4279SBram Moolenaar  */
731071d4279SBram Moolenaar     void
732071d4279SBram Moolenaar deleteFold(start, end, recursive, had_visual)
733071d4279SBram Moolenaar     linenr_T	start;
734071d4279SBram Moolenaar     linenr_T	end;
735071d4279SBram Moolenaar     int		recursive;
736071d4279SBram Moolenaar     int		had_visual;	/* TRUE when Visual selection used */
737071d4279SBram Moolenaar {
738071d4279SBram Moolenaar     garray_T	*gap;
739071d4279SBram Moolenaar     fold_T	*fp;
740071d4279SBram Moolenaar     garray_T	*found_ga;
741071d4279SBram Moolenaar     fold_T	*found_fp = NULL;
742071d4279SBram Moolenaar     linenr_T	found_off = 0;
743238b8e20SBram Moolenaar     int		use_level;
744071d4279SBram Moolenaar     int		maybe_small = FALSE;
745071d4279SBram Moolenaar     int		level = 0;
746071d4279SBram Moolenaar     linenr_T	lnum = start;
747071d4279SBram Moolenaar     linenr_T	lnum_off;
748071d4279SBram Moolenaar     int		did_one = FALSE;
749071d4279SBram Moolenaar     linenr_T	first_lnum = MAXLNUM;
750071d4279SBram Moolenaar     linenr_T	last_lnum = 0;
751071d4279SBram Moolenaar 
752071d4279SBram Moolenaar     checkupdate(curwin);
753071d4279SBram Moolenaar 
754071d4279SBram Moolenaar     while (lnum <= end)
755071d4279SBram Moolenaar     {
756071d4279SBram Moolenaar 	/* Find the deepest fold for "start". */
757071d4279SBram Moolenaar 	gap = &curwin->w_folds;
758071d4279SBram Moolenaar 	found_ga = NULL;
759071d4279SBram Moolenaar 	lnum_off = 0;
760238b8e20SBram Moolenaar 	use_level = FALSE;
761071d4279SBram Moolenaar 	for (;;)
762071d4279SBram Moolenaar 	{
763071d4279SBram Moolenaar 	    if (!foldFind(gap, lnum - lnum_off, &fp))
764071d4279SBram Moolenaar 		break;
765071d4279SBram Moolenaar 	    /* lnum is inside this fold, remember info */
766071d4279SBram Moolenaar 	    found_ga = gap;
767071d4279SBram Moolenaar 	    found_fp = fp;
768071d4279SBram Moolenaar 	    found_off = lnum_off;
769071d4279SBram Moolenaar 
770071d4279SBram Moolenaar 	    /* if "lnum" is folded, don't check nesting */
771071d4279SBram Moolenaar 	    if (check_closed(curwin, fp, &use_level, level,
772071d4279SBram Moolenaar 						      &maybe_small, lnum_off))
773071d4279SBram Moolenaar 		break;
774071d4279SBram Moolenaar 
775071d4279SBram Moolenaar 	    /* check nested folds */
776071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
777071d4279SBram Moolenaar 	    lnum_off += fp->fd_top;
778071d4279SBram Moolenaar 	    ++level;
779071d4279SBram Moolenaar 	}
780071d4279SBram Moolenaar 	if (found_ga == NULL)
781071d4279SBram Moolenaar 	{
782071d4279SBram Moolenaar 	    ++lnum;
783071d4279SBram Moolenaar 	}
784071d4279SBram Moolenaar 	else
785071d4279SBram Moolenaar 	{
786071d4279SBram Moolenaar 	    lnum = found_fp->fd_top + found_fp->fd_len + found_off;
787071d4279SBram Moolenaar 
788071d4279SBram Moolenaar 	    if (foldmethodIsManual(curwin))
789071d4279SBram Moolenaar 		deleteFoldEntry(found_ga,
790071d4279SBram Moolenaar 		    (int)(found_fp - (fold_T *)found_ga->ga_data), recursive);
791071d4279SBram Moolenaar 	    else
792071d4279SBram Moolenaar 	    {
793238b8e20SBram Moolenaar 		if (first_lnum > found_fp->fd_top + found_off)
794238b8e20SBram Moolenaar 		    first_lnum = found_fp->fd_top + found_off;
795238b8e20SBram Moolenaar 		if (last_lnum < lnum)
796071d4279SBram Moolenaar 		    last_lnum = lnum;
797238b8e20SBram Moolenaar 		if (!did_one)
798071d4279SBram Moolenaar 		    parseMarker(curwin);
799071d4279SBram Moolenaar 		deleteFoldMarkers(found_fp, recursive, found_off);
800071d4279SBram Moolenaar 	    }
801238b8e20SBram Moolenaar 	    did_one = TRUE;
802071d4279SBram Moolenaar 
803071d4279SBram Moolenaar 	    /* redraw window */
804071d4279SBram Moolenaar 	    changed_window_setting();
805071d4279SBram Moolenaar 	}
806071d4279SBram Moolenaar     }
807071d4279SBram Moolenaar     if (!did_one)
808071d4279SBram Moolenaar     {
809071d4279SBram Moolenaar 	EMSG(_(e_nofold));
810071d4279SBram Moolenaar 	/* Force a redraw to remove the Visual highlighting. */
811071d4279SBram Moolenaar 	if (had_visual)
812071d4279SBram Moolenaar 	    redraw_curbuf_later(INVERTED);
813071d4279SBram Moolenaar     }
814238b8e20SBram Moolenaar     else
815238b8e20SBram Moolenaar 	/* Deleting markers may make cursor column invalid. */
816238b8e20SBram Moolenaar 	check_cursor_col();
817238b8e20SBram Moolenaar 
818071d4279SBram Moolenaar     if (last_lnum > 0)
819071d4279SBram Moolenaar 	changed_lines(first_lnum, (colnr_T)0, last_lnum, 0L);
820071d4279SBram Moolenaar }
821071d4279SBram Moolenaar 
822071d4279SBram Moolenaar /* clearFolding() {{{2 */
823071d4279SBram Moolenaar /*
824071d4279SBram Moolenaar  * Remove all folding for window "win".
825071d4279SBram Moolenaar  */
826071d4279SBram Moolenaar     void
827071d4279SBram Moolenaar clearFolding(win)
828071d4279SBram Moolenaar     win_T	*win;
829071d4279SBram Moolenaar {
830071d4279SBram Moolenaar     deleteFoldRecurse(&win->w_folds);
831071d4279SBram Moolenaar     win->w_foldinvalid = FALSE;
832071d4279SBram Moolenaar }
833071d4279SBram Moolenaar 
834071d4279SBram Moolenaar /* foldUpdate() {{{2 */
835071d4279SBram Moolenaar /*
836071d4279SBram Moolenaar  * Update folds for changes in the buffer of a window.
837071d4279SBram Moolenaar  * Note that inserted/deleted lines must have already been taken care of by
838071d4279SBram Moolenaar  * calling foldMarkAdjust().
839071d4279SBram Moolenaar  * The changes in lines from top to bot (inclusive).
840071d4279SBram Moolenaar  */
841071d4279SBram Moolenaar     void
842071d4279SBram Moolenaar foldUpdate(wp, top, bot)
843071d4279SBram Moolenaar     win_T	*wp;
844071d4279SBram Moolenaar     linenr_T	top;
845071d4279SBram Moolenaar     linenr_T	bot;
846071d4279SBram Moolenaar {
847071d4279SBram Moolenaar     fold_T	*fp;
848071d4279SBram Moolenaar 
849071d4279SBram Moolenaar     /* Mark all folds from top to bot as maybe-small. */
850d5f6933dSBram Moolenaar     (void)foldFind(&wp->w_folds, top, &fp);
851d5f6933dSBram Moolenaar     while (fp < (fold_T *)wp->w_folds.ga_data + wp->w_folds.ga_len
852071d4279SBram Moolenaar 	    && fp->fd_top < bot)
853071d4279SBram Moolenaar     {
854071d4279SBram Moolenaar 	fp->fd_small = MAYBE;
855071d4279SBram Moolenaar 	++fp;
856071d4279SBram Moolenaar     }
857071d4279SBram Moolenaar 
858071d4279SBram Moolenaar     if (foldmethodIsIndent(wp)
859071d4279SBram Moolenaar 	    || foldmethodIsExpr(wp)
860071d4279SBram Moolenaar 	    || foldmethodIsMarker(wp)
861071d4279SBram Moolenaar #ifdef FEAT_DIFF
862071d4279SBram Moolenaar 	    || foldmethodIsDiff(wp)
863071d4279SBram Moolenaar #endif
864071d4279SBram Moolenaar 	    || foldmethodIsSyntax(wp))
865fa6d5afdSBram Moolenaar     {
866fa6d5afdSBram Moolenaar 	int save_got_int = got_int;
867fa6d5afdSBram Moolenaar 
868fa6d5afdSBram Moolenaar 	/* reset got_int here, otherwise it won't work */
869fa6d5afdSBram Moolenaar 	got_int = FALSE;
870071d4279SBram Moolenaar 	foldUpdateIEMS(wp, top, bot);
871fa6d5afdSBram Moolenaar 	got_int |= save_got_int;
872fa6d5afdSBram Moolenaar     }
873071d4279SBram Moolenaar }
874071d4279SBram Moolenaar 
875071d4279SBram Moolenaar /* foldUpdateAll() {{{2 */
876071d4279SBram Moolenaar /*
877071d4279SBram Moolenaar  * Update all lines in a window for folding.
878071d4279SBram Moolenaar  * Used when a fold setting changes or after reloading the buffer.
879071d4279SBram Moolenaar  * The actual updating is postponed until fold info is used, to avoid doing
880071d4279SBram Moolenaar  * every time a setting is changed or a syntax item is added.
881071d4279SBram Moolenaar  */
882071d4279SBram Moolenaar     void
883071d4279SBram Moolenaar foldUpdateAll(win)
884071d4279SBram Moolenaar     win_T	*win;
885071d4279SBram Moolenaar {
886071d4279SBram Moolenaar     win->w_foldinvalid = TRUE;
887071d4279SBram Moolenaar     redraw_win_later(win, NOT_VALID);
888071d4279SBram Moolenaar }
889071d4279SBram Moolenaar 
890071d4279SBram Moolenaar /* foldMoveTo() {{{2 */
891071d4279SBram Moolenaar /*
892071d4279SBram Moolenaar  * If "updown" is FALSE: Move to the start or end of the fold.
893071d4279SBram Moolenaar  * If "updown" is TRUE: move to fold at the same level.
894071d4279SBram Moolenaar  * If not moved return FAIL.
895071d4279SBram Moolenaar  */
896071d4279SBram Moolenaar     int
897071d4279SBram Moolenaar foldMoveTo(updown, dir, count)
898071d4279SBram Moolenaar     int		updown;
899071d4279SBram Moolenaar     int		dir;	    /* FORWARD or BACKWARD */
900071d4279SBram Moolenaar     long	count;
901071d4279SBram Moolenaar {
902071d4279SBram Moolenaar     long	n;
903071d4279SBram Moolenaar     int		retval = FAIL;
904071d4279SBram Moolenaar     linenr_T	lnum_off;
905071d4279SBram Moolenaar     linenr_T	lnum_found;
906071d4279SBram Moolenaar     linenr_T	lnum;
907071d4279SBram Moolenaar     int		use_level;
908071d4279SBram Moolenaar     int		maybe_small;
909071d4279SBram Moolenaar     garray_T	*gap;
910071d4279SBram Moolenaar     fold_T	*fp;
911071d4279SBram Moolenaar     int		level;
912071d4279SBram Moolenaar     int		last;
913071d4279SBram Moolenaar 
914293ee4d4SBram Moolenaar     checkupdate(curwin);
915293ee4d4SBram Moolenaar 
916071d4279SBram Moolenaar     /* Repeat "count" times. */
917071d4279SBram Moolenaar     for (n = 0; n < count; ++n)
918071d4279SBram Moolenaar     {
919071d4279SBram Moolenaar 	/* Find nested folds.  Stop when a fold is closed.  The deepest fold
920071d4279SBram Moolenaar 	 * that moves the cursor is used. */
921071d4279SBram Moolenaar 	lnum_off = 0;
922071d4279SBram Moolenaar 	gap = &curwin->w_folds;
923071d4279SBram Moolenaar 	use_level = FALSE;
924071d4279SBram Moolenaar 	maybe_small = FALSE;
925071d4279SBram Moolenaar 	lnum_found = curwin->w_cursor.lnum;
926071d4279SBram Moolenaar 	level = 0;
927071d4279SBram Moolenaar 	last = FALSE;
928071d4279SBram Moolenaar 	for (;;)
929071d4279SBram Moolenaar 	{
930071d4279SBram Moolenaar 	    if (!foldFind(gap, curwin->w_cursor.lnum - lnum_off, &fp))
931071d4279SBram Moolenaar 	    {
932071d4279SBram Moolenaar 		if (!updown)
933071d4279SBram Moolenaar 		    break;
934071d4279SBram Moolenaar 
935071d4279SBram Moolenaar 		/* When moving up, consider a fold above the cursor; when
936071d4279SBram Moolenaar 		 * moving down consider a fold below the cursor. */
937071d4279SBram Moolenaar 		if (dir == FORWARD)
938071d4279SBram Moolenaar 		{
939071d4279SBram Moolenaar 		    if (fp - (fold_T *)gap->ga_data >= gap->ga_len)
940071d4279SBram Moolenaar 			break;
941071d4279SBram Moolenaar 		    --fp;
942071d4279SBram Moolenaar 		}
943071d4279SBram Moolenaar 		else
944071d4279SBram Moolenaar 		{
945071d4279SBram Moolenaar 		    if (fp == (fold_T *)gap->ga_data)
946071d4279SBram Moolenaar 			break;
947071d4279SBram Moolenaar 		}
948071d4279SBram Moolenaar 		/* don't look for contained folds, they will always move
949071d4279SBram Moolenaar 		 * the cursor too far. */
950071d4279SBram Moolenaar 		last = TRUE;
951071d4279SBram Moolenaar 	    }
952071d4279SBram Moolenaar 
953071d4279SBram Moolenaar 	    if (!last)
954071d4279SBram Moolenaar 	    {
955071d4279SBram Moolenaar 		/* Check if this fold is closed. */
956071d4279SBram Moolenaar 		if (check_closed(curwin, fp, &use_level, level,
957071d4279SBram Moolenaar 						      &maybe_small, lnum_off))
958071d4279SBram Moolenaar 		    last = TRUE;
959071d4279SBram Moolenaar 
960071d4279SBram Moolenaar 		/* "[z" and "]z" stop at closed fold */
961071d4279SBram Moolenaar 		if (last && !updown)
962071d4279SBram Moolenaar 		    break;
963071d4279SBram Moolenaar 	    }
964071d4279SBram Moolenaar 
965071d4279SBram Moolenaar 	    if (updown)
966071d4279SBram Moolenaar 	    {
967071d4279SBram Moolenaar 		if (dir == FORWARD)
968071d4279SBram Moolenaar 		{
969071d4279SBram Moolenaar 		    /* to start of next fold if there is one */
970071d4279SBram Moolenaar 		    if (fp + 1 - (fold_T *)gap->ga_data < gap->ga_len)
971071d4279SBram Moolenaar 		    {
972071d4279SBram Moolenaar 			lnum = fp[1].fd_top + lnum_off;
973071d4279SBram Moolenaar 			if (lnum > curwin->w_cursor.lnum)
974071d4279SBram Moolenaar 			    lnum_found = lnum;
975071d4279SBram Moolenaar 		    }
976071d4279SBram Moolenaar 		}
977071d4279SBram Moolenaar 		else
978071d4279SBram Moolenaar 		{
979071d4279SBram Moolenaar 		    /* to end of previous fold if there is one */
980071d4279SBram Moolenaar 		    if (fp > (fold_T *)gap->ga_data)
981071d4279SBram Moolenaar 		    {
982071d4279SBram Moolenaar 			lnum = fp[-1].fd_top + lnum_off + fp[-1].fd_len - 1;
983071d4279SBram Moolenaar 			if (lnum < curwin->w_cursor.lnum)
984071d4279SBram Moolenaar 			    lnum_found = lnum;
985071d4279SBram Moolenaar 		    }
986071d4279SBram Moolenaar 		}
987071d4279SBram Moolenaar 	    }
988071d4279SBram Moolenaar 	    else
989071d4279SBram Moolenaar 	    {
990071d4279SBram Moolenaar 		/* Open fold found, set cursor to its start/end and then check
991071d4279SBram Moolenaar 		 * nested folds. */
992071d4279SBram Moolenaar 		if (dir == FORWARD)
993071d4279SBram Moolenaar 		{
994071d4279SBram Moolenaar 		    lnum = fp->fd_top + lnum_off + fp->fd_len - 1;
995071d4279SBram Moolenaar 		    if (lnum > curwin->w_cursor.lnum)
996071d4279SBram Moolenaar 			lnum_found = lnum;
997071d4279SBram Moolenaar 		}
998071d4279SBram Moolenaar 		else
999071d4279SBram Moolenaar 		{
1000071d4279SBram Moolenaar 		    lnum = fp->fd_top + lnum_off;
1001071d4279SBram Moolenaar 		    if (lnum < curwin->w_cursor.lnum)
1002071d4279SBram Moolenaar 			lnum_found = lnum;
1003071d4279SBram Moolenaar 		}
1004071d4279SBram Moolenaar 	    }
1005071d4279SBram Moolenaar 
1006071d4279SBram Moolenaar 	    if (last)
1007071d4279SBram Moolenaar 		break;
1008071d4279SBram Moolenaar 
1009071d4279SBram Moolenaar 	    /* Check nested folds (if any). */
1010071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
1011071d4279SBram Moolenaar 	    lnum_off += fp->fd_top;
1012071d4279SBram Moolenaar 	    ++level;
1013071d4279SBram Moolenaar 	}
1014071d4279SBram Moolenaar 	if (lnum_found != curwin->w_cursor.lnum)
1015071d4279SBram Moolenaar 	{
1016071d4279SBram Moolenaar 	    if (retval == FAIL)
1017071d4279SBram Moolenaar 		setpcmark();
1018071d4279SBram Moolenaar 	    curwin->w_cursor.lnum = lnum_found;
1019071d4279SBram Moolenaar 	    curwin->w_cursor.col = 0;
1020071d4279SBram Moolenaar 	    retval = OK;
1021071d4279SBram Moolenaar 	}
1022071d4279SBram Moolenaar 	else
1023071d4279SBram Moolenaar 	    break;
1024071d4279SBram Moolenaar     }
1025071d4279SBram Moolenaar 
1026071d4279SBram Moolenaar     return retval;
1027071d4279SBram Moolenaar }
1028071d4279SBram Moolenaar 
1029071d4279SBram Moolenaar /* foldInitWin() {{{2 */
1030071d4279SBram Moolenaar /*
1031071d4279SBram Moolenaar  * Init the fold info in a new window.
1032071d4279SBram Moolenaar  */
1033071d4279SBram Moolenaar     void
103470b2a56dSBram Moolenaar foldInitWin(new_win)
103570b2a56dSBram Moolenaar     win_T	*new_win;
1036071d4279SBram Moolenaar {
103770b2a56dSBram Moolenaar     ga_init2(&new_win->w_folds, (int)sizeof(fold_T), 10);
1038071d4279SBram Moolenaar }
1039071d4279SBram Moolenaar 
1040071d4279SBram Moolenaar /* find_wl_entry() {{{2 */
1041071d4279SBram Moolenaar /*
1042071d4279SBram Moolenaar  * Find an entry in the win->w_lines[] array for buffer line "lnum".
1043071d4279SBram Moolenaar  * Only valid entries are considered (for entries where wl_valid is FALSE the
1044071d4279SBram Moolenaar  * line number can be wrong).
1045071d4279SBram Moolenaar  * Returns index of entry or -1 if not found.
1046071d4279SBram Moolenaar  */
1047071d4279SBram Moolenaar     int
1048071d4279SBram Moolenaar find_wl_entry(win, lnum)
1049071d4279SBram Moolenaar     win_T	*win;
1050071d4279SBram Moolenaar     linenr_T	lnum;
1051071d4279SBram Moolenaar {
1052071d4279SBram Moolenaar     int		i;
1053071d4279SBram Moolenaar 
1054071d4279SBram Moolenaar     for (i = 0; i < win->w_lines_valid; ++i)
1055071d4279SBram Moolenaar 	if (win->w_lines[i].wl_valid)
1056071d4279SBram Moolenaar 	{
1057071d4279SBram Moolenaar 	    if (lnum < win->w_lines[i].wl_lnum)
1058071d4279SBram Moolenaar 		return -1;
1059071d4279SBram Moolenaar 	    if (lnum <= win->w_lines[i].wl_lastlnum)
1060071d4279SBram Moolenaar 		return i;
1061071d4279SBram Moolenaar 	}
1062071d4279SBram Moolenaar     return -1;
1063071d4279SBram Moolenaar }
1064071d4279SBram Moolenaar 
1065071d4279SBram Moolenaar /* foldAdjustVisual() {{{2 */
1066071d4279SBram Moolenaar /*
1067071d4279SBram Moolenaar  * Adjust the Visual area to include any fold at the start or end completely.
1068071d4279SBram Moolenaar  */
1069071d4279SBram Moolenaar     void
1070071d4279SBram Moolenaar foldAdjustVisual()
1071071d4279SBram Moolenaar {
1072071d4279SBram Moolenaar     pos_T	*start, *end;
1073071d4279SBram Moolenaar     char_u	*ptr;
1074071d4279SBram Moolenaar 
1075071d4279SBram Moolenaar     if (!VIsual_active || !hasAnyFolding(curwin))
1076071d4279SBram Moolenaar 	return;
1077071d4279SBram Moolenaar 
1078071d4279SBram Moolenaar     if (ltoreq(VIsual, curwin->w_cursor))
1079071d4279SBram Moolenaar     {
1080071d4279SBram Moolenaar 	start = &VIsual;
1081071d4279SBram Moolenaar 	end = &curwin->w_cursor;
1082071d4279SBram Moolenaar     }
1083071d4279SBram Moolenaar     else
1084071d4279SBram Moolenaar     {
1085071d4279SBram Moolenaar 	start = &curwin->w_cursor;
1086071d4279SBram Moolenaar 	end = &VIsual;
1087071d4279SBram Moolenaar     }
1088071d4279SBram Moolenaar     if (hasFolding(start->lnum, &start->lnum, NULL))
1089071d4279SBram Moolenaar 	start->col = 0;
1090071d4279SBram Moolenaar     if (hasFolding(end->lnum, NULL, &end->lnum))
1091071d4279SBram Moolenaar     {
1092071d4279SBram Moolenaar 	ptr = ml_get(end->lnum);
1093071d4279SBram Moolenaar 	end->col = (colnr_T)STRLEN(ptr);
1094071d4279SBram Moolenaar 	if (end->col > 0 && *p_sel == 'o')
1095071d4279SBram Moolenaar 	    --end->col;
1096071d4279SBram Moolenaar #ifdef FEAT_MBYTE
1097071d4279SBram Moolenaar 	/* prevent cursor from moving on the trail byte */
1098071d4279SBram Moolenaar 	if (has_mbyte)
1099071d4279SBram Moolenaar 	    mb_adjust_cursor();
1100071d4279SBram Moolenaar #endif
1101071d4279SBram Moolenaar     }
1102071d4279SBram Moolenaar }
1103071d4279SBram Moolenaar 
1104071d4279SBram Moolenaar /* cursor_foldstart() {{{2 */
1105071d4279SBram Moolenaar /*
1106071d4279SBram Moolenaar  * Move the cursor to the first line of a closed fold.
1107071d4279SBram Moolenaar  */
1108071d4279SBram Moolenaar     void
1109071d4279SBram Moolenaar foldAdjustCursor()
1110071d4279SBram Moolenaar {
1111071d4279SBram Moolenaar     (void)hasFolding(curwin->w_cursor.lnum, &curwin->w_cursor.lnum, NULL);
1112071d4279SBram Moolenaar }
1113071d4279SBram Moolenaar 
1114071d4279SBram Moolenaar /* Internal functions for "fold_T" {{{1 */
1115071d4279SBram Moolenaar /* cloneFoldGrowArray() {{{2 */
1116071d4279SBram Moolenaar /*
1117071d4279SBram Moolenaar  * Will "clone" (i.e deep copy) a garray_T of folds.
1118071d4279SBram Moolenaar  *
1119071d4279SBram Moolenaar  * Return FAIL if the operation cannot be completed, otherwise OK.
1120071d4279SBram Moolenaar  */
1121071d4279SBram Moolenaar     void
1122071d4279SBram Moolenaar cloneFoldGrowArray(from, to)
1123071d4279SBram Moolenaar     garray_T	*from;
1124071d4279SBram Moolenaar     garray_T	*to;
1125071d4279SBram Moolenaar {
1126071d4279SBram Moolenaar     int		i;
1127071d4279SBram Moolenaar     fold_T	*from_p;
1128071d4279SBram Moolenaar     fold_T	*to_p;
1129071d4279SBram Moolenaar 
1130071d4279SBram Moolenaar     ga_init2(to, from->ga_itemsize, from->ga_growsize);
1131071d4279SBram Moolenaar     if (from->ga_len == 0 || ga_grow(to, from->ga_len) == FAIL)
1132071d4279SBram Moolenaar 	return;
1133071d4279SBram Moolenaar 
1134071d4279SBram Moolenaar     from_p = (fold_T *)from->ga_data;
1135071d4279SBram Moolenaar     to_p = (fold_T *)to->ga_data;
1136071d4279SBram Moolenaar 
1137071d4279SBram Moolenaar     for (i = 0; i < from->ga_len; i++)
1138071d4279SBram Moolenaar     {
1139071d4279SBram Moolenaar 	to_p->fd_top = from_p->fd_top;
1140071d4279SBram Moolenaar 	to_p->fd_len = from_p->fd_len;
1141071d4279SBram Moolenaar 	to_p->fd_flags = from_p->fd_flags;
1142071d4279SBram Moolenaar 	to_p->fd_small = from_p->fd_small;
1143071d4279SBram Moolenaar 	cloneFoldGrowArray(&from_p->fd_nested, &to_p->fd_nested);
1144071d4279SBram Moolenaar 	++to->ga_len;
1145071d4279SBram Moolenaar 	++from_p;
1146071d4279SBram Moolenaar 	++to_p;
1147071d4279SBram Moolenaar     }
1148071d4279SBram Moolenaar }
1149071d4279SBram Moolenaar 
1150071d4279SBram Moolenaar /* foldFind() {{{2 */
1151071d4279SBram Moolenaar /*
1152071d4279SBram Moolenaar  * Search for line "lnum" in folds of growarray "gap".
1153071d4279SBram Moolenaar  * Set *fpp to the fold struct for the fold that contains "lnum" or
1154071d4279SBram Moolenaar  * the first fold below it (careful: it can be beyond the end of the array!).
1155071d4279SBram Moolenaar  * Returns FALSE when there is no fold that contains "lnum".
1156071d4279SBram Moolenaar  */
1157071d4279SBram Moolenaar     static int
1158071d4279SBram Moolenaar foldFind(gap, lnum, fpp)
1159071d4279SBram Moolenaar     garray_T	*gap;
1160071d4279SBram Moolenaar     linenr_T	lnum;
1161071d4279SBram Moolenaar     fold_T	**fpp;
1162071d4279SBram Moolenaar {
1163071d4279SBram Moolenaar     linenr_T	low, high;
1164071d4279SBram Moolenaar     fold_T	*fp;
1165071d4279SBram Moolenaar     int		i;
1166071d4279SBram Moolenaar 
1167071d4279SBram Moolenaar     /*
1168071d4279SBram Moolenaar      * Perform a binary search.
1169071d4279SBram Moolenaar      * "low" is lowest index of possible match.
1170071d4279SBram Moolenaar      * "high" is highest index of possible match.
1171071d4279SBram Moolenaar      */
1172071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
1173071d4279SBram Moolenaar     low = 0;
1174071d4279SBram Moolenaar     high = gap->ga_len - 1;
1175071d4279SBram Moolenaar     while (low <= high)
1176071d4279SBram Moolenaar     {
1177071d4279SBram Moolenaar 	i = (low + high) / 2;
1178071d4279SBram Moolenaar 	if (fp[i].fd_top > lnum)
1179071d4279SBram Moolenaar 	    /* fold below lnum, adjust high */
1180071d4279SBram Moolenaar 	    high = i - 1;
1181071d4279SBram Moolenaar 	else if (fp[i].fd_top + fp[i].fd_len <= lnum)
1182071d4279SBram Moolenaar 	    /* fold above lnum, adjust low */
1183071d4279SBram Moolenaar 	    low = i + 1;
1184071d4279SBram Moolenaar 	else
1185071d4279SBram Moolenaar 	{
1186071d4279SBram Moolenaar 	    /* lnum is inside this fold */
1187071d4279SBram Moolenaar 	    *fpp = fp + i;
1188071d4279SBram Moolenaar 	    return TRUE;
1189071d4279SBram Moolenaar 	}
1190071d4279SBram Moolenaar     }
1191071d4279SBram Moolenaar     *fpp = fp + low;
1192071d4279SBram Moolenaar     return FALSE;
1193071d4279SBram Moolenaar }
1194071d4279SBram Moolenaar 
1195071d4279SBram Moolenaar /* foldLevelWin() {{{2 */
1196071d4279SBram Moolenaar /*
1197071d4279SBram Moolenaar  * Return fold level at line number "lnum" in window "wp".
1198071d4279SBram Moolenaar  */
1199071d4279SBram Moolenaar     static int
1200071d4279SBram Moolenaar foldLevelWin(wp, lnum)
1201071d4279SBram Moolenaar     win_T	*wp;
1202071d4279SBram Moolenaar     linenr_T	lnum;
1203071d4279SBram Moolenaar {
1204071d4279SBram Moolenaar     fold_T	*fp;
1205071d4279SBram Moolenaar     linenr_T	lnum_rel = lnum;
1206071d4279SBram Moolenaar     int		level =  0;
1207071d4279SBram Moolenaar     garray_T	*gap;
1208071d4279SBram Moolenaar 
1209071d4279SBram Moolenaar     /* Recursively search for a fold that contains "lnum". */
1210071d4279SBram Moolenaar     gap = &wp->w_folds;
1211071d4279SBram Moolenaar     for (;;)
1212071d4279SBram Moolenaar     {
1213071d4279SBram Moolenaar 	if (!foldFind(gap, lnum_rel, &fp))
1214071d4279SBram Moolenaar 	    break;
1215071d4279SBram Moolenaar 	/* Check nested folds.  Line number is relative to containing fold. */
1216071d4279SBram Moolenaar 	gap = &fp->fd_nested;
1217071d4279SBram Moolenaar 	lnum_rel -= fp->fd_top;
1218071d4279SBram Moolenaar 	++level;
1219071d4279SBram Moolenaar     }
1220071d4279SBram Moolenaar 
1221071d4279SBram Moolenaar     return level;
1222071d4279SBram Moolenaar }
1223071d4279SBram Moolenaar 
1224071d4279SBram Moolenaar /* checkupdate() {{{2 */
1225071d4279SBram Moolenaar /*
1226071d4279SBram Moolenaar  * Check if the folds in window "wp" are invalid and update them if needed.
1227071d4279SBram Moolenaar  */
1228071d4279SBram Moolenaar     static void
1229071d4279SBram Moolenaar checkupdate(wp)
1230071d4279SBram Moolenaar     win_T	*wp;
1231071d4279SBram Moolenaar {
1232071d4279SBram Moolenaar     if (wp->w_foldinvalid)
1233071d4279SBram Moolenaar     {
1234071d4279SBram Moolenaar 	foldUpdate(wp, (linenr_T)1, (linenr_T)MAXLNUM); /* will update all */
1235071d4279SBram Moolenaar 	wp->w_foldinvalid = FALSE;
1236071d4279SBram Moolenaar     }
1237071d4279SBram Moolenaar }
1238071d4279SBram Moolenaar 
1239071d4279SBram Moolenaar /* setFoldRepeat() {{{2 */
1240071d4279SBram Moolenaar /*
1241071d4279SBram Moolenaar  * Open or close fold for current window at line "lnum".
1242071d4279SBram Moolenaar  * Repeat "count" times.
1243071d4279SBram Moolenaar  */
1244071d4279SBram Moolenaar     static void
1245fe86f2d7SBram Moolenaar setFoldRepeat(lnum, count, do_open)
1246071d4279SBram Moolenaar     linenr_T	lnum;
1247071d4279SBram Moolenaar     long	count;
1248fe86f2d7SBram Moolenaar     int		do_open;
1249071d4279SBram Moolenaar {
1250071d4279SBram Moolenaar     int		done;
1251071d4279SBram Moolenaar     long	n;
1252071d4279SBram Moolenaar 
1253071d4279SBram Moolenaar     for (n = 0; n < count; ++n)
1254071d4279SBram Moolenaar     {
1255071d4279SBram Moolenaar 	done = DONE_NOTHING;
1256fe86f2d7SBram Moolenaar 	(void)setManualFold(lnum, do_open, FALSE, &done);
1257071d4279SBram Moolenaar 	if (!(done & DONE_ACTION))
1258071d4279SBram Moolenaar 	{
1259071d4279SBram Moolenaar 	    /* Only give an error message when no fold could be opened. */
1260071d4279SBram Moolenaar 	    if (n == 0 && !(done & DONE_FOLD))
1261071d4279SBram Moolenaar 		EMSG(_(e_nofold));
1262071d4279SBram Moolenaar 	    break;
1263071d4279SBram Moolenaar 	}
1264071d4279SBram Moolenaar     }
1265071d4279SBram Moolenaar }
1266071d4279SBram Moolenaar 
1267071d4279SBram Moolenaar /* setManualFold() {{{2 */
1268071d4279SBram Moolenaar /*
1269071d4279SBram Moolenaar  * Open or close the fold in the current window which contains "lnum".
1270071d4279SBram Moolenaar  * Also does this for other windows in diff mode when needed.
1271071d4279SBram Moolenaar  */
1272071d4279SBram Moolenaar     static linenr_T
1273071d4279SBram Moolenaar setManualFold(lnum, opening, recurse, donep)
1274071d4279SBram Moolenaar     linenr_T	lnum;
1275071d4279SBram Moolenaar     int		opening;    /* TRUE when opening, FALSE when closing */
1276071d4279SBram Moolenaar     int		recurse;    /* TRUE when closing/opening recursive */
1277071d4279SBram Moolenaar     int		*donep;
1278071d4279SBram Moolenaar {
1279071d4279SBram Moolenaar #ifdef FEAT_DIFF
1280071d4279SBram Moolenaar     if (foldmethodIsDiff(curwin) && curwin->w_p_scb)
1281071d4279SBram Moolenaar     {
1282071d4279SBram Moolenaar 	win_T	    *wp;
1283071d4279SBram Moolenaar 	linenr_T    dlnum;
1284071d4279SBram Moolenaar 
1285071d4279SBram Moolenaar 	/*
1286071d4279SBram Moolenaar 	 * Do the same operation in other windows in diff mode.  Calculate the
1287071d4279SBram Moolenaar 	 * line number from the diffs.
1288071d4279SBram Moolenaar 	 */
1289071d4279SBram Moolenaar 	FOR_ALL_WINDOWS(wp)
1290071d4279SBram Moolenaar 	{
1291071d4279SBram Moolenaar 	    if (wp != curwin && foldmethodIsDiff(wp) && wp->w_p_scb)
1292071d4279SBram Moolenaar 	    {
1293071d4279SBram Moolenaar 		dlnum = diff_lnum_win(curwin->w_cursor.lnum, wp);
1294071d4279SBram Moolenaar 		if (dlnum != 0)
1295071d4279SBram Moolenaar 		    (void)setManualFoldWin(wp, dlnum, opening, recurse, NULL);
1296071d4279SBram Moolenaar 	    }
1297071d4279SBram Moolenaar 	}
1298071d4279SBram Moolenaar     }
1299071d4279SBram Moolenaar #endif
1300071d4279SBram Moolenaar 
1301071d4279SBram Moolenaar     return setManualFoldWin(curwin, lnum, opening, recurse, donep);
1302071d4279SBram Moolenaar }
1303071d4279SBram Moolenaar 
1304071d4279SBram Moolenaar /* setManualFoldWin() {{{2 */
1305071d4279SBram Moolenaar /*
1306071d4279SBram Moolenaar  * Open or close the fold in window "wp" which contains "lnum".
1307071d4279SBram Moolenaar  * "donep", when not NULL, points to flag that is set to DONE_FOLD when some
1308071d4279SBram Moolenaar  * fold was found and to DONE_ACTION when some fold was opened or closed.
1309071d4279SBram Moolenaar  * When "donep" is NULL give an error message when no fold was found for
1310071d4279SBram Moolenaar  * "lnum", but only if "wp" is "curwin".
1311071d4279SBram Moolenaar  * Return the line number of the next line that could be closed.
1312071d4279SBram Moolenaar  * It's only valid when "opening" is TRUE!
1313071d4279SBram Moolenaar  */
1314071d4279SBram Moolenaar     static linenr_T
1315071d4279SBram Moolenaar setManualFoldWin(wp, lnum, opening, recurse, donep)
1316071d4279SBram Moolenaar     win_T	*wp;
1317071d4279SBram Moolenaar     linenr_T	lnum;
1318071d4279SBram Moolenaar     int		opening;    /* TRUE when opening, FALSE when closing */
1319071d4279SBram Moolenaar     int		recurse;    /* TRUE when closing/opening recursive */
1320071d4279SBram Moolenaar     int		*donep;
1321071d4279SBram Moolenaar {
1322071d4279SBram Moolenaar     fold_T	*fp;
1323071d4279SBram Moolenaar     fold_T	*fp2;
1324071d4279SBram Moolenaar     fold_T	*found = NULL;
1325071d4279SBram Moolenaar     int		j;
1326071d4279SBram Moolenaar     int		level = 0;
1327071d4279SBram Moolenaar     int		use_level = FALSE;
1328071d4279SBram Moolenaar     int		found_fold = FALSE;
1329071d4279SBram Moolenaar     garray_T	*gap;
1330071d4279SBram Moolenaar     linenr_T	next = MAXLNUM;
1331071d4279SBram Moolenaar     linenr_T	off = 0;
1332071d4279SBram Moolenaar     int		done = 0;
1333071d4279SBram Moolenaar 
1334071d4279SBram Moolenaar     checkupdate(wp);
1335071d4279SBram Moolenaar 
1336071d4279SBram Moolenaar     /*
1337071d4279SBram Moolenaar      * Find the fold, open or close it.
1338071d4279SBram Moolenaar      */
1339071d4279SBram Moolenaar     gap = &wp->w_folds;
1340071d4279SBram Moolenaar     for (;;)
1341071d4279SBram Moolenaar     {
1342071d4279SBram Moolenaar 	if (!foldFind(gap, lnum, &fp))
1343071d4279SBram Moolenaar 	{
1344071d4279SBram Moolenaar 	    /* If there is a following fold, continue there next time. */
1345071d4279SBram Moolenaar 	    if (fp < (fold_T *)gap->ga_data + gap->ga_len)
1346071d4279SBram Moolenaar 		next = fp->fd_top + off;
1347071d4279SBram Moolenaar 	    break;
1348071d4279SBram Moolenaar 	}
1349071d4279SBram Moolenaar 
1350071d4279SBram Moolenaar 	/* lnum is inside this fold */
1351071d4279SBram Moolenaar 	found_fold = TRUE;
1352071d4279SBram Moolenaar 
1353071d4279SBram Moolenaar 	/* If there is a following fold, continue there next time. */
1354071d4279SBram Moolenaar 	if (fp + 1 < (fold_T *)gap->ga_data + gap->ga_len)
1355071d4279SBram Moolenaar 	    next = fp[1].fd_top + off;
1356071d4279SBram Moolenaar 
1357071d4279SBram Moolenaar 	/* Change from level-dependent folding to manual. */
1358071d4279SBram Moolenaar 	if (use_level || fp->fd_flags == FD_LEVEL)
1359071d4279SBram Moolenaar 	{
1360071d4279SBram Moolenaar 	    use_level = TRUE;
1361071d4279SBram Moolenaar 	    if (level >= wp->w_p_fdl)
1362071d4279SBram Moolenaar 		fp->fd_flags = FD_CLOSED;
1363071d4279SBram Moolenaar 	    else
1364071d4279SBram Moolenaar 		fp->fd_flags = FD_OPEN;
1365071d4279SBram Moolenaar 	    fp2 = (fold_T *)fp->fd_nested.ga_data;
1366071d4279SBram Moolenaar 	    for (j = 0; j < fp->fd_nested.ga_len; ++j)
1367071d4279SBram Moolenaar 		fp2[j].fd_flags = FD_LEVEL;
1368071d4279SBram Moolenaar 	}
1369071d4279SBram Moolenaar 
1370071d4279SBram Moolenaar 	/* Simple case: Close recursively means closing the fold. */
1371071d4279SBram Moolenaar 	if (!opening && recurse)
1372071d4279SBram Moolenaar 	{
1373071d4279SBram Moolenaar 	    if (fp->fd_flags != FD_CLOSED)
1374071d4279SBram Moolenaar 	    {
1375071d4279SBram Moolenaar 		done |= DONE_ACTION;
1376071d4279SBram Moolenaar 		fp->fd_flags = FD_CLOSED;
1377071d4279SBram Moolenaar 	    }
1378071d4279SBram Moolenaar 	}
1379071d4279SBram Moolenaar 	else if (fp->fd_flags == FD_CLOSED)
1380071d4279SBram Moolenaar 	{
1381071d4279SBram Moolenaar 	    /* When opening, open topmost closed fold. */
1382071d4279SBram Moolenaar 	    if (opening)
1383071d4279SBram Moolenaar 	    {
1384071d4279SBram Moolenaar 		fp->fd_flags = FD_OPEN;
1385071d4279SBram Moolenaar 		done |= DONE_ACTION;
1386071d4279SBram Moolenaar 		if (recurse)
1387071d4279SBram Moolenaar 		    foldOpenNested(fp);
1388071d4279SBram Moolenaar 	    }
1389071d4279SBram Moolenaar 	    break;
1390071d4279SBram Moolenaar 	}
1391071d4279SBram Moolenaar 
1392071d4279SBram Moolenaar 	/* fold is open, check nested folds */
1393071d4279SBram Moolenaar 	found = fp;
1394071d4279SBram Moolenaar 	gap = &fp->fd_nested;
1395071d4279SBram Moolenaar 	lnum -= fp->fd_top;
1396071d4279SBram Moolenaar 	off += fp->fd_top;
1397071d4279SBram Moolenaar 	++level;
1398071d4279SBram Moolenaar     }
1399071d4279SBram Moolenaar     if (found_fold)
1400071d4279SBram Moolenaar     {
1401071d4279SBram Moolenaar 	/* When closing and not recurse, close deepest open fold. */
1402071d4279SBram Moolenaar 	if (!opening && found != NULL)
1403071d4279SBram Moolenaar 	{
1404071d4279SBram Moolenaar 	    found->fd_flags = FD_CLOSED;
1405071d4279SBram Moolenaar 	    done |= DONE_ACTION;
1406071d4279SBram Moolenaar 	}
1407071d4279SBram Moolenaar 	wp->w_fold_manual = TRUE;
1408071d4279SBram Moolenaar 	if (done & DONE_ACTION)
1409071d4279SBram Moolenaar 	    changed_window_setting_win(wp);
1410071d4279SBram Moolenaar 	done |= DONE_FOLD;
1411071d4279SBram Moolenaar     }
1412071d4279SBram Moolenaar     else if (donep == NULL && wp == curwin)
1413071d4279SBram Moolenaar 	EMSG(_(e_nofold));
1414071d4279SBram Moolenaar 
1415071d4279SBram Moolenaar     if (donep != NULL)
1416071d4279SBram Moolenaar 	*donep |= done;
1417071d4279SBram Moolenaar 
1418071d4279SBram Moolenaar     return next;
1419071d4279SBram Moolenaar }
1420071d4279SBram Moolenaar 
1421071d4279SBram Moolenaar /* foldOpenNested() {{{2 */
1422071d4279SBram Moolenaar /*
1423071d4279SBram Moolenaar  * Open all nested folds in fold "fpr" recursively.
1424071d4279SBram Moolenaar  */
1425071d4279SBram Moolenaar     static void
1426071d4279SBram Moolenaar foldOpenNested(fpr)
1427071d4279SBram Moolenaar     fold_T	*fpr;
1428071d4279SBram Moolenaar {
1429071d4279SBram Moolenaar     int		i;
1430071d4279SBram Moolenaar     fold_T	*fp;
1431071d4279SBram Moolenaar 
1432071d4279SBram Moolenaar     fp = (fold_T *)fpr->fd_nested.ga_data;
1433071d4279SBram Moolenaar     for (i = 0; i < fpr->fd_nested.ga_len; ++i)
1434071d4279SBram Moolenaar     {
1435071d4279SBram Moolenaar 	foldOpenNested(&fp[i]);
1436071d4279SBram Moolenaar 	fp[i].fd_flags = FD_OPEN;
1437071d4279SBram Moolenaar     }
1438071d4279SBram Moolenaar }
1439071d4279SBram Moolenaar 
1440071d4279SBram Moolenaar /* deleteFoldEntry() {{{2 */
1441071d4279SBram Moolenaar /*
1442071d4279SBram Moolenaar  * Delete fold "idx" from growarray "gap".
1443071d4279SBram Moolenaar  * When "recursive" is TRUE also delete all the folds contained in it.
1444071d4279SBram Moolenaar  * When "recursive" is FALSE contained folds are moved one level up.
1445071d4279SBram Moolenaar  */
1446071d4279SBram Moolenaar     static void
1447071d4279SBram Moolenaar deleteFoldEntry(gap, idx, recursive)
1448071d4279SBram Moolenaar     garray_T	*gap;
1449071d4279SBram Moolenaar     int		idx;
1450071d4279SBram Moolenaar     int		recursive;
1451071d4279SBram Moolenaar {
1452071d4279SBram Moolenaar     fold_T	*fp;
1453071d4279SBram Moolenaar     int		i;
1454071d4279SBram Moolenaar     long	moved;
1455071d4279SBram Moolenaar     fold_T	*nfp;
1456071d4279SBram Moolenaar 
1457071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data + idx;
1458071d4279SBram Moolenaar     if (recursive || fp->fd_nested.ga_len == 0)
1459071d4279SBram Moolenaar     {
1460071d4279SBram Moolenaar 	/* recursively delete the contained folds */
1461071d4279SBram Moolenaar 	deleteFoldRecurse(&fp->fd_nested);
1462071d4279SBram Moolenaar 	--gap->ga_len;
1463071d4279SBram Moolenaar 	if (idx < gap->ga_len)
1464071d4279SBram Moolenaar 	    mch_memmove(fp, fp + 1, sizeof(fold_T) * (gap->ga_len - idx));
1465071d4279SBram Moolenaar     }
1466071d4279SBram Moolenaar     else
1467071d4279SBram Moolenaar     {
1468c9927c13SBram Moolenaar 	/* Move nested folds one level up, to overwrite the fold that is
1469071d4279SBram Moolenaar 	 * deleted. */
1470071d4279SBram Moolenaar 	moved = fp->fd_nested.ga_len;
1471071d4279SBram Moolenaar 	if (ga_grow(gap, (int)(moved - 1)) == OK)
1472071d4279SBram Moolenaar 	{
1473c9927c13SBram Moolenaar 	    /* Get "fp" again, the array may have been reallocated. */
1474c9927c13SBram Moolenaar 	    fp = (fold_T *)gap->ga_data + idx;
1475c9927c13SBram Moolenaar 
1476071d4279SBram Moolenaar 	    /* adjust fd_top and fd_flags for the moved folds */
1477071d4279SBram Moolenaar 	    nfp = (fold_T *)fp->fd_nested.ga_data;
1478071d4279SBram Moolenaar 	    for (i = 0; i < moved; ++i)
1479071d4279SBram Moolenaar 	    {
1480071d4279SBram Moolenaar 		nfp[i].fd_top += fp->fd_top;
1481071d4279SBram Moolenaar 		if (fp->fd_flags == FD_LEVEL)
1482071d4279SBram Moolenaar 		    nfp[i].fd_flags = FD_LEVEL;
1483071d4279SBram Moolenaar 		if (fp->fd_small == MAYBE)
1484071d4279SBram Moolenaar 		    nfp[i].fd_small = MAYBE;
1485071d4279SBram Moolenaar 	    }
1486071d4279SBram Moolenaar 
1487071d4279SBram Moolenaar 	    /* move the existing folds down to make room */
1488c9927c13SBram Moolenaar 	    if (idx + 1 < gap->ga_len)
1489071d4279SBram Moolenaar 		mch_memmove(fp + moved, fp + 1,
1490c9927c13SBram Moolenaar 				  sizeof(fold_T) * (gap->ga_len - (idx + 1)));
1491071d4279SBram Moolenaar 	    /* move the contained folds one level up */
1492071d4279SBram Moolenaar 	    mch_memmove(fp, nfp, (size_t)(sizeof(fold_T) * moved));
1493071d4279SBram Moolenaar 	    vim_free(nfp);
1494071d4279SBram Moolenaar 	    gap->ga_len += moved - 1;
1495071d4279SBram Moolenaar 	}
1496071d4279SBram Moolenaar     }
1497071d4279SBram Moolenaar }
1498071d4279SBram Moolenaar 
1499071d4279SBram Moolenaar /* deleteFoldRecurse() {{{2 */
1500071d4279SBram Moolenaar /*
1501071d4279SBram Moolenaar  * Delete nested folds in a fold.
1502071d4279SBram Moolenaar  */
1503071d4279SBram Moolenaar     void
1504071d4279SBram Moolenaar deleteFoldRecurse(gap)
1505071d4279SBram Moolenaar     garray_T	*gap;
1506071d4279SBram Moolenaar {
1507071d4279SBram Moolenaar     int		i;
1508071d4279SBram Moolenaar 
1509071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
1510071d4279SBram Moolenaar 	deleteFoldRecurse(&(((fold_T *)(gap->ga_data))[i].fd_nested));
1511071d4279SBram Moolenaar     ga_clear(gap);
1512071d4279SBram Moolenaar }
1513071d4279SBram Moolenaar 
1514071d4279SBram Moolenaar /* foldMarkAdjust() {{{2 */
1515071d4279SBram Moolenaar /*
1516071d4279SBram Moolenaar  * Update line numbers of folds for inserted/deleted lines.
1517071d4279SBram Moolenaar  */
1518071d4279SBram Moolenaar     void
1519071d4279SBram Moolenaar foldMarkAdjust(wp, line1, line2, amount, amount_after)
1520071d4279SBram Moolenaar     win_T	*wp;
1521071d4279SBram Moolenaar     linenr_T	line1;
1522071d4279SBram Moolenaar     linenr_T	line2;
1523071d4279SBram Moolenaar     long	amount;
1524071d4279SBram Moolenaar     long	amount_after;
1525071d4279SBram Moolenaar {
1526071d4279SBram Moolenaar     /* If deleting marks from line1 to line2, but not deleting all those
1527071d4279SBram Moolenaar      * lines, set line2 so that only deleted lines have their folds removed. */
1528071d4279SBram Moolenaar     if (amount == MAXLNUM && line2 >= line1 && line2 - line1 >= -amount_after)
1529071d4279SBram Moolenaar 	line2 = line1 - amount_after - 1;
1530071d4279SBram Moolenaar     /* If appending a line in Insert mode, it should be included in the fold
1531071d4279SBram Moolenaar      * just above the line. */
1532071d4279SBram Moolenaar     if ((State & INSERT) && amount == (linenr_T)1 && line2 == MAXLNUM)
1533071d4279SBram Moolenaar 	--line1;
1534071d4279SBram Moolenaar     foldMarkAdjustRecurse(&wp->w_folds, line1, line2, amount, amount_after);
1535071d4279SBram Moolenaar }
1536071d4279SBram Moolenaar 
1537071d4279SBram Moolenaar /* foldMarkAdjustRecurse() {{{2 */
1538071d4279SBram Moolenaar     static void
1539071d4279SBram Moolenaar foldMarkAdjustRecurse(gap, line1, line2, amount, amount_after)
1540071d4279SBram Moolenaar     garray_T	*gap;
1541071d4279SBram Moolenaar     linenr_T	line1;
1542071d4279SBram Moolenaar     linenr_T	line2;
1543071d4279SBram Moolenaar     long	amount;
1544071d4279SBram Moolenaar     long	amount_after;
1545071d4279SBram Moolenaar {
1546071d4279SBram Moolenaar     fold_T	*fp;
1547071d4279SBram Moolenaar     int		i;
1548071d4279SBram Moolenaar     linenr_T	last;
1549071d4279SBram Moolenaar     linenr_T	top;
1550071d4279SBram Moolenaar 
1551071d4279SBram Moolenaar     /* In Insert mode an inserted line at the top of a fold is considered part
1552071d4279SBram Moolenaar      * of the fold, otherwise it isn't. */
1553071d4279SBram Moolenaar     if ((State & INSERT) && amount == (linenr_T)1 && line2 == MAXLNUM)
1554071d4279SBram Moolenaar 	top = line1 + 1;
1555071d4279SBram Moolenaar     else
1556071d4279SBram Moolenaar 	top = line1;
1557071d4279SBram Moolenaar 
1558071d4279SBram Moolenaar     /* Find the fold containing or just below "line1". */
1559071d4279SBram Moolenaar     (void)foldFind(gap, line1, &fp);
1560071d4279SBram Moolenaar 
1561071d4279SBram Moolenaar     /*
1562071d4279SBram Moolenaar      * Adjust all folds below "line1" that are affected.
1563071d4279SBram Moolenaar      */
1564071d4279SBram Moolenaar     for (i = (int)(fp - (fold_T *)gap->ga_data); i < gap->ga_len; ++i, ++fp)
1565071d4279SBram Moolenaar     {
1566071d4279SBram Moolenaar 	/*
1567071d4279SBram Moolenaar 	 * Check for these situations:
1568071d4279SBram Moolenaar 	 *	  1  2	3
1569071d4279SBram Moolenaar 	 *	  1  2	3
1570071d4279SBram Moolenaar 	 * line1     2	3  4  5
1571071d4279SBram Moolenaar 	 *	     2	3  4  5
1572071d4279SBram Moolenaar 	 *	     2	3  4  5
1573071d4279SBram Moolenaar 	 * line2     2	3  4  5
1574071d4279SBram Moolenaar 	 *		3     5  6
1575071d4279SBram Moolenaar 	 *		3     5  6
1576071d4279SBram Moolenaar 	 */
1577071d4279SBram Moolenaar 
1578071d4279SBram Moolenaar 	last = fp->fd_top + fp->fd_len - 1; /* last line of fold */
1579071d4279SBram Moolenaar 
1580071d4279SBram Moolenaar 	/* 1. fold completely above line1: nothing to do */
1581071d4279SBram Moolenaar 	if (last < line1)
1582071d4279SBram Moolenaar 	    continue;
1583071d4279SBram Moolenaar 
1584071d4279SBram Moolenaar 	/* 6. fold below line2: only adjust for amount_after */
1585071d4279SBram Moolenaar 	if (fp->fd_top > line2)
1586071d4279SBram Moolenaar 	{
1587071d4279SBram Moolenaar 	    if (amount_after == 0)
1588071d4279SBram Moolenaar 		break;
1589071d4279SBram Moolenaar 	    fp->fd_top += amount_after;
1590071d4279SBram Moolenaar 	}
1591071d4279SBram Moolenaar 	else
1592071d4279SBram Moolenaar 	{
1593071d4279SBram Moolenaar 	    if (fp->fd_top >= top && last <= line2)
1594071d4279SBram Moolenaar 	    {
1595071d4279SBram Moolenaar 		/* 4. fold completely contained in range */
1596071d4279SBram Moolenaar 		if (amount == MAXLNUM)
1597071d4279SBram Moolenaar 		{
1598071d4279SBram Moolenaar 		    /* Deleting lines: delete the fold completely */
1599071d4279SBram Moolenaar 		    deleteFoldEntry(gap, i, TRUE);
1600071d4279SBram Moolenaar 		    --i;    /* adjust index for deletion */
1601071d4279SBram Moolenaar 		    --fp;
1602071d4279SBram Moolenaar 		}
1603071d4279SBram Moolenaar 		else
1604071d4279SBram Moolenaar 		    fp->fd_top += amount;
1605071d4279SBram Moolenaar 	    }
1606071d4279SBram Moolenaar 	    else
1607071d4279SBram Moolenaar 	    {
1608071d4279SBram Moolenaar 		if (fp->fd_top < top)
1609071d4279SBram Moolenaar 		{
1610194b94c5SBram Moolenaar 		    /* 2 or 3: need to correct nested folds too */
1611194b94c5SBram Moolenaar 		    foldMarkAdjustRecurse(&fp->fd_nested, line1 - fp->fd_top,
1612194b94c5SBram Moolenaar 				  line2 - fp->fd_top, amount, amount_after);
1613071d4279SBram Moolenaar 		    if (last <= line2)
1614071d4279SBram Moolenaar 		    {
1615071d4279SBram Moolenaar 			/* 2. fold contains line1, line2 is below fold */
1616071d4279SBram Moolenaar 			if (amount == MAXLNUM)
1617071d4279SBram Moolenaar 			    fp->fd_len = line1 - fp->fd_top;
1618071d4279SBram Moolenaar 			else
1619071d4279SBram Moolenaar 			    fp->fd_len += amount;
1620071d4279SBram Moolenaar 		    }
1621071d4279SBram Moolenaar 		    else
1622071d4279SBram Moolenaar 		    {
1623071d4279SBram Moolenaar 			/* 3. fold contains line1 and line2 */
1624071d4279SBram Moolenaar 			fp->fd_len += amount_after;
1625071d4279SBram Moolenaar 		    }
1626071d4279SBram Moolenaar 		}
1627071d4279SBram Moolenaar 		else
1628071d4279SBram Moolenaar 		{
1629194b94c5SBram Moolenaar 		    /* 5. fold is below line1 and contains line2; need to
1630194b94c5SBram Moolenaar 		     * correct nested folds too */
1631194b94c5SBram Moolenaar 		    foldMarkAdjustRecurse(&fp->fd_nested, line1 - fp->fd_top,
1632194b94c5SBram Moolenaar 				  line2 - fp->fd_top, amount,
1633194b94c5SBram Moolenaar 				  amount_after + (fp->fd_top - top));
1634071d4279SBram Moolenaar 		    if (amount == MAXLNUM)
1635071d4279SBram Moolenaar 		    {
1636071d4279SBram Moolenaar 			fp->fd_len -= line2 - fp->fd_top + 1;
1637071d4279SBram Moolenaar 			fp->fd_top = line1;
1638071d4279SBram Moolenaar 		    }
1639071d4279SBram Moolenaar 		    else
1640071d4279SBram Moolenaar 		    {
1641071d4279SBram Moolenaar 			fp->fd_len += amount_after - amount;
1642071d4279SBram Moolenaar 			fp->fd_top += amount;
1643071d4279SBram Moolenaar 		    }
1644071d4279SBram Moolenaar 		}
1645071d4279SBram Moolenaar 	    }
1646071d4279SBram Moolenaar 	}
1647071d4279SBram Moolenaar     }
1648071d4279SBram Moolenaar }
1649071d4279SBram Moolenaar 
1650071d4279SBram Moolenaar /* getDeepestNesting() {{{2 */
1651071d4279SBram Moolenaar /*
1652071d4279SBram Moolenaar  * Get the lowest 'foldlevel' value that makes the deepest nested fold in the
1653071d4279SBram Moolenaar  * current window open.
1654071d4279SBram Moolenaar  */
1655071d4279SBram Moolenaar     int
1656071d4279SBram Moolenaar getDeepestNesting()
1657071d4279SBram Moolenaar {
1658071d4279SBram Moolenaar     checkupdate(curwin);
1659071d4279SBram Moolenaar     return getDeepestNestingRecurse(&curwin->w_folds);
1660071d4279SBram Moolenaar }
1661071d4279SBram Moolenaar 
1662071d4279SBram Moolenaar     static int
1663071d4279SBram Moolenaar getDeepestNestingRecurse(gap)
1664071d4279SBram Moolenaar     garray_T	*gap;
1665071d4279SBram Moolenaar {
1666071d4279SBram Moolenaar     int		i;
1667071d4279SBram Moolenaar     int		level;
1668071d4279SBram Moolenaar     int		maxlevel = 0;
1669071d4279SBram Moolenaar     fold_T	*fp;
1670071d4279SBram Moolenaar 
1671071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
1672071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
1673071d4279SBram Moolenaar     {
1674071d4279SBram Moolenaar 	level = getDeepestNestingRecurse(&fp[i].fd_nested) + 1;
1675071d4279SBram Moolenaar 	if (level > maxlevel)
1676071d4279SBram Moolenaar 	    maxlevel = level;
1677071d4279SBram Moolenaar     }
1678071d4279SBram Moolenaar 
1679071d4279SBram Moolenaar     return maxlevel;
1680071d4279SBram Moolenaar }
1681071d4279SBram Moolenaar 
1682071d4279SBram Moolenaar /* check_closed() {{{2 */
1683071d4279SBram Moolenaar /*
1684071d4279SBram Moolenaar  * Check if a fold is closed and update the info needed to check nested folds.
1685071d4279SBram Moolenaar  */
1686071d4279SBram Moolenaar     static int
1687071d4279SBram Moolenaar check_closed(win, fp, use_levelp, level, maybe_smallp, lnum_off)
1688071d4279SBram Moolenaar     win_T	*win;
1689071d4279SBram Moolenaar     fold_T	*fp;
1690071d4279SBram Moolenaar     int		*use_levelp;	    /* TRUE: outer fold had FD_LEVEL */
1691071d4279SBram Moolenaar     int		level;		    /* folding depth */
1692071d4279SBram Moolenaar     int		*maybe_smallp;	    /* TRUE: outer this had fd_small == MAYBE */
1693071d4279SBram Moolenaar     linenr_T	lnum_off;	    /* line number offset for fp->fd_top */
1694071d4279SBram Moolenaar {
1695071d4279SBram Moolenaar     int		closed = FALSE;
1696071d4279SBram Moolenaar 
1697071d4279SBram Moolenaar     /* Check if this fold is closed.  If the flag is FD_LEVEL this
1698071d4279SBram Moolenaar      * fold and all folds it contains depend on 'foldlevel'. */
1699071d4279SBram Moolenaar     if (*use_levelp || fp->fd_flags == FD_LEVEL)
1700071d4279SBram Moolenaar     {
1701071d4279SBram Moolenaar 	*use_levelp = TRUE;
1702071d4279SBram Moolenaar 	if (level >= win->w_p_fdl)
1703071d4279SBram Moolenaar 	    closed = TRUE;
1704071d4279SBram Moolenaar     }
1705071d4279SBram Moolenaar     else if (fp->fd_flags == FD_CLOSED)
1706071d4279SBram Moolenaar 	closed = TRUE;
1707071d4279SBram Moolenaar 
1708071d4279SBram Moolenaar     /* Small fold isn't closed anyway. */
1709071d4279SBram Moolenaar     if (fp->fd_small == MAYBE)
1710071d4279SBram Moolenaar 	*maybe_smallp = TRUE;
1711071d4279SBram Moolenaar     if (closed)
1712071d4279SBram Moolenaar     {
1713071d4279SBram Moolenaar 	if (*maybe_smallp)
1714071d4279SBram Moolenaar 	    fp->fd_small = MAYBE;
1715071d4279SBram Moolenaar 	checkSmall(win, fp, lnum_off);
1716071d4279SBram Moolenaar 	if (fp->fd_small == TRUE)
1717071d4279SBram Moolenaar 	    closed = FALSE;
1718071d4279SBram Moolenaar     }
1719071d4279SBram Moolenaar     return closed;
1720071d4279SBram Moolenaar }
1721071d4279SBram Moolenaar 
1722071d4279SBram Moolenaar /* checkSmall() {{{2 */
1723071d4279SBram Moolenaar /*
1724071d4279SBram Moolenaar  * Update fd_small field of fold "fp".
1725071d4279SBram Moolenaar  */
1726071d4279SBram Moolenaar     static void
1727071d4279SBram Moolenaar checkSmall(wp, fp, lnum_off)
1728071d4279SBram Moolenaar     win_T	*wp;
1729071d4279SBram Moolenaar     fold_T	*fp;
1730071d4279SBram Moolenaar     linenr_T	lnum_off;	/* offset for fp->fd_top */
1731071d4279SBram Moolenaar {
1732071d4279SBram Moolenaar     int		count;
1733071d4279SBram Moolenaar     int		n;
1734071d4279SBram Moolenaar 
1735071d4279SBram Moolenaar     if (fp->fd_small == MAYBE)
1736071d4279SBram Moolenaar     {
1737071d4279SBram Moolenaar 	/* Mark any nested folds to maybe-small */
1738071d4279SBram Moolenaar 	setSmallMaybe(&fp->fd_nested);
1739071d4279SBram Moolenaar 
1740071d4279SBram Moolenaar 	if (fp->fd_len > curwin->w_p_fml)
1741071d4279SBram Moolenaar 	    fp->fd_small = FALSE;
1742071d4279SBram Moolenaar 	else
1743071d4279SBram Moolenaar 	{
1744071d4279SBram Moolenaar 	    count = 0;
1745071d4279SBram Moolenaar 	    for (n = 0; n < fp->fd_len; ++n)
1746071d4279SBram Moolenaar 	    {
1747071d4279SBram Moolenaar 		count += plines_win_nofold(wp, fp->fd_top + lnum_off + n);
1748071d4279SBram Moolenaar 		if (count > curwin->w_p_fml)
1749071d4279SBram Moolenaar 		{
1750071d4279SBram Moolenaar 		    fp->fd_small = FALSE;
1751071d4279SBram Moolenaar 		    return;
1752071d4279SBram Moolenaar 		}
1753071d4279SBram Moolenaar 	    }
1754071d4279SBram Moolenaar 	    fp->fd_small = TRUE;
1755071d4279SBram Moolenaar 	}
1756071d4279SBram Moolenaar     }
1757071d4279SBram Moolenaar }
1758071d4279SBram Moolenaar 
1759071d4279SBram Moolenaar /* setSmallMaybe() {{{2 */
1760071d4279SBram Moolenaar /*
1761071d4279SBram Moolenaar  * Set small flags in "gap" to MAYBE.
1762071d4279SBram Moolenaar  */
1763071d4279SBram Moolenaar     static void
1764071d4279SBram Moolenaar setSmallMaybe(gap)
1765071d4279SBram Moolenaar     garray_T	*gap;
1766071d4279SBram Moolenaar {
1767071d4279SBram Moolenaar     int		i;
1768071d4279SBram Moolenaar     fold_T	*fp;
1769071d4279SBram Moolenaar 
1770071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
1771071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
1772071d4279SBram Moolenaar 	fp[i].fd_small = MAYBE;
1773071d4279SBram Moolenaar }
1774071d4279SBram Moolenaar 
1775071d4279SBram Moolenaar /* foldCreateMarkers() {{{2 */
1776071d4279SBram Moolenaar /*
1777071d4279SBram Moolenaar  * Create a fold from line "start" to line "end" (inclusive) in the current
1778071d4279SBram Moolenaar  * window by adding markers.
1779071d4279SBram Moolenaar  */
1780071d4279SBram Moolenaar     static void
1781071d4279SBram Moolenaar foldCreateMarkers(start, end)
1782071d4279SBram Moolenaar     linenr_T	start;
1783071d4279SBram Moolenaar     linenr_T	end;
1784071d4279SBram Moolenaar {
1785071d4279SBram Moolenaar     if (!curbuf->b_p_ma)
1786071d4279SBram Moolenaar     {
1787071d4279SBram Moolenaar 	EMSG(_(e_modifiable));
1788071d4279SBram Moolenaar 	return;
1789071d4279SBram Moolenaar     }
1790071d4279SBram Moolenaar     parseMarker(curwin);
1791071d4279SBram Moolenaar 
1792071d4279SBram Moolenaar     foldAddMarker(start, curwin->w_p_fmr, foldstartmarkerlen);
1793071d4279SBram Moolenaar     foldAddMarker(end, foldendmarker, foldendmarkerlen);
1794071d4279SBram Moolenaar 
1795071d4279SBram Moolenaar     /* Update both changes here, to avoid all folds after the start are
1796071d4279SBram Moolenaar      * changed when the start marker is inserted and the end isn't. */
1797071d4279SBram Moolenaar     changed_lines(start, (colnr_T)0, end, 0L);
1798071d4279SBram Moolenaar }
1799071d4279SBram Moolenaar 
1800071d4279SBram Moolenaar /* foldAddMarker() {{{2 */
1801071d4279SBram Moolenaar /*
1802071d4279SBram Moolenaar  * Add "marker[markerlen]" in 'commentstring' to line "lnum".
1803071d4279SBram Moolenaar  */
1804071d4279SBram Moolenaar     static void
1805071d4279SBram Moolenaar foldAddMarker(lnum, marker, markerlen)
1806071d4279SBram Moolenaar     linenr_T	lnum;
1807071d4279SBram Moolenaar     char_u	*marker;
1808071d4279SBram Moolenaar     int		markerlen;
1809071d4279SBram Moolenaar {
1810071d4279SBram Moolenaar     char_u	*cms = curbuf->b_p_cms;
1811071d4279SBram Moolenaar     char_u	*line;
1812071d4279SBram Moolenaar     int		line_len;
1813071d4279SBram Moolenaar     char_u	*newline;
1814071d4279SBram Moolenaar     char_u	*p = (char_u *)strstr((char *)curbuf->b_p_cms, "%s");
1815071d4279SBram Moolenaar 
1816071d4279SBram Moolenaar     /* Allocate a new line: old-line + 'cms'-start + marker + 'cms'-end */
1817071d4279SBram Moolenaar     line = ml_get(lnum);
1818071d4279SBram Moolenaar     line_len = (int)STRLEN(line);
1819071d4279SBram Moolenaar 
1820071d4279SBram Moolenaar     if (u_save(lnum - 1, lnum + 1) == OK)
1821071d4279SBram Moolenaar     {
1822071d4279SBram Moolenaar 	newline = alloc((unsigned)(line_len + markerlen + STRLEN(cms) + 1));
1823071d4279SBram Moolenaar 	if (newline == NULL)
1824071d4279SBram Moolenaar 	    return;
1825071d4279SBram Moolenaar 	STRCPY(newline, line);
1826071d4279SBram Moolenaar 	if (p == NULL)
1827ce0842a6SBram Moolenaar 	    vim_strncpy(newline + line_len, marker, markerlen);
1828071d4279SBram Moolenaar 	else
1829071d4279SBram Moolenaar 	{
1830071d4279SBram Moolenaar 	    STRCPY(newline + line_len, cms);
1831071d4279SBram Moolenaar 	    STRNCPY(newline + line_len + (p - cms), marker, markerlen);
1832071d4279SBram Moolenaar 	    STRCPY(newline + line_len + (p - cms) + markerlen, p + 2);
1833071d4279SBram Moolenaar 	}
1834071d4279SBram Moolenaar 
1835071d4279SBram Moolenaar 	ml_replace(lnum, newline, FALSE);
1836071d4279SBram Moolenaar     }
1837071d4279SBram Moolenaar }
1838071d4279SBram Moolenaar 
1839071d4279SBram Moolenaar /* deleteFoldMarkers() {{{2 */
1840071d4279SBram Moolenaar /*
1841071d4279SBram Moolenaar  * Delete the markers for a fold, causing it to be deleted.
1842071d4279SBram Moolenaar  */
1843071d4279SBram Moolenaar     static void
1844071d4279SBram Moolenaar deleteFoldMarkers(fp, recursive, lnum_off)
1845071d4279SBram Moolenaar     fold_T	*fp;
1846071d4279SBram Moolenaar     int		recursive;
1847071d4279SBram Moolenaar     linenr_T	lnum_off;	/* offset for fp->fd_top */
1848071d4279SBram Moolenaar {
1849071d4279SBram Moolenaar     int		i;
1850071d4279SBram Moolenaar 
1851071d4279SBram Moolenaar     if (recursive)
1852071d4279SBram Moolenaar 	for (i = 0; i < fp->fd_nested.ga_len; ++i)
1853071d4279SBram Moolenaar 	    deleteFoldMarkers((fold_T *)fp->fd_nested.ga_data + i, TRUE,
1854071d4279SBram Moolenaar 						       lnum_off + fp->fd_top);
1855071d4279SBram Moolenaar     foldDelMarker(fp->fd_top + lnum_off, curwin->w_p_fmr, foldstartmarkerlen);
1856071d4279SBram Moolenaar     foldDelMarker(fp->fd_top + lnum_off + fp->fd_len - 1,
1857071d4279SBram Moolenaar 					     foldendmarker, foldendmarkerlen);
1858071d4279SBram Moolenaar }
1859071d4279SBram Moolenaar 
1860071d4279SBram Moolenaar /* foldDelMarker() {{{2 */
1861071d4279SBram Moolenaar /*
1862071d4279SBram Moolenaar  * Delete marker "marker[markerlen]" at the end of line "lnum".
1863071d4279SBram Moolenaar  * Delete 'commentstring' if it matches.
1864071d4279SBram Moolenaar  * If the marker is not found, there is no error message.  Could a missing
1865071d4279SBram Moolenaar  * close-marker.
1866071d4279SBram Moolenaar  */
1867071d4279SBram Moolenaar     static void
1868071d4279SBram Moolenaar foldDelMarker(lnum, marker, markerlen)
1869071d4279SBram Moolenaar     linenr_T	lnum;
1870071d4279SBram Moolenaar     char_u	*marker;
1871071d4279SBram Moolenaar     int		markerlen;
1872071d4279SBram Moolenaar {
1873071d4279SBram Moolenaar     char_u	*line;
1874071d4279SBram Moolenaar     char_u	*newline;
1875071d4279SBram Moolenaar     char_u	*p;
1876071d4279SBram Moolenaar     int		len;
1877071d4279SBram Moolenaar     char_u	*cms = curbuf->b_p_cms;
1878071d4279SBram Moolenaar     char_u	*cms2;
1879071d4279SBram Moolenaar 
1880071d4279SBram Moolenaar     line = ml_get(lnum);
1881071d4279SBram Moolenaar     for (p = line; *p != NUL; ++p)
1882071d4279SBram Moolenaar 	if (STRNCMP(p, marker, markerlen) == 0)
1883071d4279SBram Moolenaar 	{
1884071d4279SBram Moolenaar 	    /* Found the marker, include a digit if it's there. */
1885071d4279SBram Moolenaar 	    len = markerlen;
1886071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(p[len]))
1887071d4279SBram Moolenaar 		++len;
1888071d4279SBram Moolenaar 	    if (*cms != NUL)
1889071d4279SBram Moolenaar 	    {
1890071d4279SBram Moolenaar 		/* Also delete 'commentstring' if it matches. */
1891071d4279SBram Moolenaar 		cms2 = (char_u *)strstr((char *)cms, "%s");
1892071d4279SBram Moolenaar 		if (p - line >= cms2 - cms
1893071d4279SBram Moolenaar 			&& STRNCMP(p - (cms2 - cms), cms, cms2 - cms) == 0
1894071d4279SBram Moolenaar 			&& STRNCMP(p + len, cms2 + 2, STRLEN(cms2 + 2)) == 0)
1895071d4279SBram Moolenaar 		{
1896071d4279SBram Moolenaar 		    p -= cms2 - cms;
1897071d4279SBram Moolenaar 		    len += (int)STRLEN(cms) - 2;
1898071d4279SBram Moolenaar 		}
1899071d4279SBram Moolenaar 	    }
1900071d4279SBram Moolenaar 	    if (u_save(lnum - 1, lnum + 1) == OK)
1901071d4279SBram Moolenaar 	    {
1902071d4279SBram Moolenaar 		/* Make new line: text-before-marker + text-after-marker */
1903071d4279SBram Moolenaar 		newline = alloc((unsigned)(STRLEN(line) - len + 1));
1904071d4279SBram Moolenaar 		if (newline != NULL)
1905071d4279SBram Moolenaar 		{
1906071d4279SBram Moolenaar 		    STRNCPY(newline, line, p - line);
1907071d4279SBram Moolenaar 		    STRCPY(newline + (p - line), p + len);
1908071d4279SBram Moolenaar 		    ml_replace(lnum, newline, FALSE);
1909071d4279SBram Moolenaar 		}
1910071d4279SBram Moolenaar 	    }
1911071d4279SBram Moolenaar 	    break;
1912071d4279SBram Moolenaar 	}
1913071d4279SBram Moolenaar }
1914071d4279SBram Moolenaar 
19157b0294cbSBram Moolenaar /* get_foldtext() {{{2 */
19167b0294cbSBram Moolenaar /*
19177b0294cbSBram Moolenaar  * Return the text for a closed fold at line "lnum", with last line "lnume".
19187b0294cbSBram Moolenaar  * When 'foldtext' isn't set puts the result in "buf[51]".  Otherwise the
19197b0294cbSBram Moolenaar  * result is in allocated memory.
19207b0294cbSBram Moolenaar  */
19217b0294cbSBram Moolenaar     char_u *
19227b0294cbSBram Moolenaar get_foldtext(wp, lnum, lnume, foldinfo, buf)
19237b0294cbSBram Moolenaar     win_T	*wp;
19247b0294cbSBram Moolenaar     linenr_T	lnum, lnume;
19257b0294cbSBram Moolenaar     foldinfo_T	*foldinfo;
19267b0294cbSBram Moolenaar     char_u	*buf;
19277b0294cbSBram Moolenaar {
19287b0294cbSBram Moolenaar     char_u	*text = NULL;
19297b0294cbSBram Moolenaar #ifdef FEAT_EVAL
1930dab38d51SBram Moolenaar      /* an error occurred when evaluating 'fdt' setting */
1931dab38d51SBram Moolenaar     static int	    got_fdt_error = FALSE;
1932dab38d51SBram Moolenaar     int		    save_did_emsg = did_emsg;
1933dab38d51SBram Moolenaar     static win_T    *last_wp = NULL;
1934dab38d51SBram Moolenaar     static linenr_T last_lnum = 0;
1935dab38d51SBram Moolenaar 
1936dab38d51SBram Moolenaar     if (last_wp != wp || last_wp == NULL
1937dab38d51SBram Moolenaar 					|| last_lnum > lnum || last_lnum == 0)
1938dab38d51SBram Moolenaar 	/* window changed, try evaluating foldtext setting once again */
1939dab38d51SBram Moolenaar 	got_fdt_error = FALSE;
1940dab38d51SBram Moolenaar 
1941dab38d51SBram Moolenaar     if (!got_fdt_error)
1942dab38d51SBram Moolenaar 	/* a previous error should not abort evaluating 'foldexpr' */
1943dab38d51SBram Moolenaar 	did_emsg = FALSE;
1944dab38d51SBram Moolenaar 
19457b0294cbSBram Moolenaar     if (*wp->w_p_fdt != NUL)
19467b0294cbSBram Moolenaar     {
19475b88ba47SBram Moolenaar 	char_u	dashes[MAX_LEVEL + 2];
19487b0294cbSBram Moolenaar 	win_T	*save_curwin;
19497b0294cbSBram Moolenaar 	int	level;
19507b0294cbSBram Moolenaar 	char_u	*p;
19517b0294cbSBram Moolenaar 
19527b0294cbSBram Moolenaar 	/* Set "v:foldstart" and "v:foldend". */
19537b0294cbSBram Moolenaar 	set_vim_var_nr(VV_FOLDSTART, lnum);
19547b0294cbSBram Moolenaar 	set_vim_var_nr(VV_FOLDEND, lnume);
19557b0294cbSBram Moolenaar 
19567b0294cbSBram Moolenaar 	/* Set "v:folddashes" to a string of "level" dashes. */
19577b0294cbSBram Moolenaar 	/* Set "v:foldlevel" to "level". */
19587b0294cbSBram Moolenaar 	level = foldinfo->fi_level;
19595b88ba47SBram Moolenaar 	if (level > (int)sizeof(dashes) - 1)
19605b88ba47SBram Moolenaar 	    level = (int)sizeof(dashes) - 1;
19617b0294cbSBram Moolenaar 	vim_memset(dashes, '-', (size_t)level);
19627b0294cbSBram Moolenaar 	dashes[level] = NUL;
19637b0294cbSBram Moolenaar 	set_vim_var_string(VV_FOLDDASHES, dashes, -1);
19647b0294cbSBram Moolenaar 	set_vim_var_nr(VV_FOLDLEVEL, (long)level);
1965dab38d51SBram Moolenaar 
1966dab38d51SBram Moolenaar 	/* skip evaluating foldtext on errors */
1967dab38d51SBram Moolenaar 	if (!got_fdt_error)
1968dab38d51SBram Moolenaar 	{
19697b0294cbSBram Moolenaar 	    save_curwin = curwin;
19707b0294cbSBram Moolenaar 	    curwin = wp;
19717b0294cbSBram Moolenaar 	    curbuf = wp->w_buffer;
19727b0294cbSBram Moolenaar 
1973dab38d51SBram Moolenaar 	    ++emsg_silent; /* handle exceptions, but don't display errors */
1974b71eaaeaSBram Moolenaar 	    text = eval_to_string_safe(wp->w_p_fdt, NULL,
1975d1f56e68SBram Moolenaar 			 was_set_insecurely((char_u *)"foldtext", OPT_LOCAL));
1976dab38d51SBram Moolenaar 	    --emsg_silent;
1977dab38d51SBram Moolenaar 
1978dab38d51SBram Moolenaar 	    if (text == NULL || did_emsg)
1979dab38d51SBram Moolenaar 		got_fdt_error = TRUE;
19807b0294cbSBram Moolenaar 
19817b0294cbSBram Moolenaar 	    curwin = save_curwin;
19827b0294cbSBram Moolenaar 	    curbuf = curwin->w_buffer;
1983dab38d51SBram Moolenaar 	}
1984dab38d51SBram Moolenaar 	last_lnum = lnum;
1985dab38d51SBram Moolenaar 	last_wp   = wp;
19867b0294cbSBram Moolenaar 	set_vim_var_string(VV_FOLDDASHES, NULL, -1);
19877b0294cbSBram Moolenaar 
1988dab38d51SBram Moolenaar 	if (!did_emsg && save_did_emsg)
1989dab38d51SBram Moolenaar 	    did_emsg = save_did_emsg;
1990dab38d51SBram Moolenaar 
19917b0294cbSBram Moolenaar 	if (text != NULL)
19927b0294cbSBram Moolenaar 	{
19937b0294cbSBram Moolenaar 	    /* Replace unprintable characters, if there are any.  But
19947b0294cbSBram Moolenaar 	     * replace a TAB with a space. */
19957b0294cbSBram Moolenaar 	    for (p = text; *p != NUL; ++p)
19967b0294cbSBram Moolenaar 	    {
19977b0294cbSBram Moolenaar # ifdef FEAT_MBYTE
1998009b2592SBram Moolenaar 		int	len;
1999009b2592SBram Moolenaar 
20000fa313a7SBram Moolenaar 		if (has_mbyte && (len = (*mb_ptr2len)(p)) > 1)
20017b0294cbSBram Moolenaar 		{
20027b0294cbSBram Moolenaar 		    if (!vim_isprintc((*mb_ptr2char)(p)))
20037b0294cbSBram Moolenaar 			break;
20047b0294cbSBram Moolenaar 		    p += len - 1;
20057b0294cbSBram Moolenaar 		}
20067b0294cbSBram Moolenaar 		else
20077b0294cbSBram Moolenaar # endif
20087b0294cbSBram Moolenaar 		    if (*p == TAB)
20097b0294cbSBram Moolenaar 			*p = ' ';
20107b0294cbSBram Moolenaar 		    else if (ptr2cells(p) > 1)
20117b0294cbSBram Moolenaar 			break;
20127b0294cbSBram Moolenaar 	    }
20137b0294cbSBram Moolenaar 	    if (*p != NUL)
20147b0294cbSBram Moolenaar 	    {
20157b0294cbSBram Moolenaar 		p = transstr(text);
20167b0294cbSBram Moolenaar 		vim_free(text);
20177b0294cbSBram Moolenaar 		text = p;
20187b0294cbSBram Moolenaar 	    }
20197b0294cbSBram Moolenaar 	}
20207b0294cbSBram Moolenaar     }
20217b0294cbSBram Moolenaar     if (text == NULL)
20227b0294cbSBram Moolenaar #endif
20237b0294cbSBram Moolenaar     {
20247b0294cbSBram Moolenaar 	sprintf((char *)buf, _("+--%3ld lines folded "),
20257b0294cbSBram Moolenaar 						    (long)(lnume - lnum + 1));
20267b0294cbSBram Moolenaar 	text = buf;
20277b0294cbSBram Moolenaar     }
20287b0294cbSBram Moolenaar     return text;
20297b0294cbSBram Moolenaar }
20307b0294cbSBram Moolenaar 
2031071d4279SBram Moolenaar /* foldtext_cleanup() {{{2 */
2032071d4279SBram Moolenaar /*
2033071d4279SBram Moolenaar  * Remove 'foldmarker' and 'commentstring' from "str" (in-place).
2034071d4279SBram Moolenaar  */
2035071d4279SBram Moolenaar     void
2036071d4279SBram Moolenaar foldtext_cleanup(str)
2037071d4279SBram Moolenaar     char_u	*str;
2038071d4279SBram Moolenaar {
2039071d4279SBram Moolenaar     char_u	*cms_start;	/* first part or the whole comment */
2040071d4279SBram Moolenaar     int		cms_slen = 0;	/* length of cms_start */
2041071d4279SBram Moolenaar     char_u	*cms_end;	/* last part of the comment or NULL */
2042071d4279SBram Moolenaar     int		cms_elen = 0;	/* length of cms_end */
2043071d4279SBram Moolenaar     char_u	*s;
2044def9e829SBram Moolenaar     char_u	*p;
2045071d4279SBram Moolenaar     int		len;
2046071d4279SBram Moolenaar     int		did1 = FALSE;
2047071d4279SBram Moolenaar     int		did2 = FALSE;
2048071d4279SBram Moolenaar 
2049071d4279SBram Moolenaar     /* Ignore leading and trailing white space in 'commentstring'. */
2050071d4279SBram Moolenaar     cms_start = skipwhite(curbuf->b_p_cms);
2051a93fa7eeSBram Moolenaar     cms_slen = (int)STRLEN(cms_start);
2052071d4279SBram Moolenaar     while (cms_slen > 0 && vim_iswhite(cms_start[cms_slen - 1]))
2053071d4279SBram Moolenaar 	--cms_slen;
2054071d4279SBram Moolenaar 
2055071d4279SBram Moolenaar     /* locate "%s" in 'commentstring', use the part before and after it. */
2056071d4279SBram Moolenaar     cms_end = (char_u *)strstr((char *)cms_start, "%s");
2057071d4279SBram Moolenaar     if (cms_end != NULL)
2058071d4279SBram Moolenaar     {
2059a93fa7eeSBram Moolenaar 	cms_elen = cms_slen - (int)(cms_end - cms_start);
2060a93fa7eeSBram Moolenaar 	cms_slen = (int)(cms_end - cms_start);
2061071d4279SBram Moolenaar 
2062071d4279SBram Moolenaar 	/* exclude white space before "%s" */
2063071d4279SBram Moolenaar 	while (cms_slen > 0 && vim_iswhite(cms_start[cms_slen - 1]))
2064071d4279SBram Moolenaar 	    --cms_slen;
2065071d4279SBram Moolenaar 
2066071d4279SBram Moolenaar 	/* skip "%s" and white space after it */
2067071d4279SBram Moolenaar 	s = skipwhite(cms_end + 2);
2068a93fa7eeSBram Moolenaar 	cms_elen -= (int)(s - cms_end);
2069071d4279SBram Moolenaar 	cms_end = s;
2070071d4279SBram Moolenaar     }
2071071d4279SBram Moolenaar     parseMarker(curwin);
2072071d4279SBram Moolenaar 
2073071d4279SBram Moolenaar     for (s = str; *s != NUL; )
2074071d4279SBram Moolenaar     {
2075071d4279SBram Moolenaar 	len = 0;
2076071d4279SBram Moolenaar 	if (STRNCMP(s, curwin->w_p_fmr, foldstartmarkerlen) == 0)
2077071d4279SBram Moolenaar 	    len = foldstartmarkerlen;
2078071d4279SBram Moolenaar 	else if (STRNCMP(s, foldendmarker, foldendmarkerlen) == 0)
2079071d4279SBram Moolenaar 	    len = foldendmarkerlen;
2080def9e829SBram Moolenaar 	if (len > 0)
2081def9e829SBram Moolenaar 	{
2082071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(s[len]))
2083071d4279SBram Moolenaar 		++len;
2084def9e829SBram Moolenaar 
2085def9e829SBram Moolenaar 	    /* May remove 'commentstring' start.  Useful when it's a double
2086def9e829SBram Moolenaar 	     * quote and we already removed a double quote. */
2087def9e829SBram Moolenaar 	    for (p = s; p > str && vim_iswhite(p[-1]); --p)
2088def9e829SBram Moolenaar 		;
2089def9e829SBram Moolenaar 	    if (p >= str + cms_slen
2090def9e829SBram Moolenaar 			   && STRNCMP(p - cms_slen, cms_start, cms_slen) == 0)
2091def9e829SBram Moolenaar 	    {
2092a93fa7eeSBram Moolenaar 		len += (int)(s - p) + cms_slen;
2093def9e829SBram Moolenaar 		s = p - cms_slen;
2094def9e829SBram Moolenaar 	    }
2095071d4279SBram Moolenaar 	}
2096071d4279SBram Moolenaar 	else if (cms_end != NULL)
2097071d4279SBram Moolenaar 	{
2098def9e829SBram Moolenaar 	    if (!did1 && cms_slen > 0 && STRNCMP(s, cms_start, cms_slen) == 0)
2099071d4279SBram Moolenaar 	    {
2100071d4279SBram Moolenaar 		len = cms_slen;
2101071d4279SBram Moolenaar 		did1 = TRUE;
2102071d4279SBram Moolenaar 	    }
2103def9e829SBram Moolenaar 	    else if (!did2 && cms_elen > 0
2104def9e829SBram Moolenaar 					&& STRNCMP(s, cms_end, cms_elen) == 0)
2105071d4279SBram Moolenaar 	    {
2106071d4279SBram Moolenaar 		len = cms_elen;
2107071d4279SBram Moolenaar 		did2 = TRUE;
2108071d4279SBram Moolenaar 	    }
2109071d4279SBram Moolenaar 	}
2110071d4279SBram Moolenaar 	if (len != 0)
2111071d4279SBram Moolenaar 	{
2112071d4279SBram Moolenaar 	    while (vim_iswhite(s[len]))
2113071d4279SBram Moolenaar 		++len;
21148c8de839SBram Moolenaar 	    STRMOVE(s, s + len);
2115071d4279SBram Moolenaar 	}
2116071d4279SBram Moolenaar 	else
2117071d4279SBram Moolenaar 	{
21181cd871b5SBram Moolenaar 	    mb_ptr_adv(s);
2119071d4279SBram Moolenaar 	}
2120071d4279SBram Moolenaar     }
2121071d4279SBram Moolenaar }
2122071d4279SBram Moolenaar 
2123071d4279SBram Moolenaar /* Folding by indent, expr, marker and syntax. {{{1 */
2124071d4279SBram Moolenaar /* Define "fline_T", passed to get fold level for a line. {{{2 */
2125071d4279SBram Moolenaar typedef struct
2126071d4279SBram Moolenaar {
2127071d4279SBram Moolenaar     win_T	*wp;		/* window */
2128071d4279SBram Moolenaar     linenr_T	lnum;		/* current line number */
2129071d4279SBram Moolenaar     linenr_T	off;		/* offset between lnum and real line number */
2130071d4279SBram Moolenaar     linenr_T	lnum_save;	/* line nr used by foldUpdateIEMSRecurse() */
2131071d4279SBram Moolenaar     int		lvl;		/* current level (-1 for undefined) */
2132071d4279SBram Moolenaar     int		lvl_next;	/* level used for next line */
2133071d4279SBram Moolenaar     int		start;		/* number of folds that are forced to start at
2134071d4279SBram Moolenaar 				   this line. */
2135071d4279SBram Moolenaar     int		end;		/* level of fold that is forced to end below
2136071d4279SBram Moolenaar 				   this line */
2137071d4279SBram Moolenaar     int		had_end;	/* level of fold that is forced to end above
2138071d4279SBram Moolenaar 				   this line (copy of "end" of prev. line) */
2139071d4279SBram Moolenaar } fline_T;
2140071d4279SBram Moolenaar 
2141071d4279SBram Moolenaar /* Flag is set when redrawing is needed. */
2142071d4279SBram Moolenaar static int fold_changed;
2143071d4279SBram Moolenaar 
2144071d4279SBram Moolenaar /* Function declarations. {{{2 */
2145*d99df423SBram Moolenaar static linenr_T foldUpdateIEMSRecurse(garray_T *gap, int level, linenr_T startlnum, fline_T *flp, void (*getlevel)(fline_T *), linenr_T bot, int topflags);
2146d25c16e2SBram Moolenaar static int foldInsert(garray_T *gap, int i);
2147d25c16e2SBram Moolenaar static void foldSplit(garray_T *gap, int i, linenr_T top, linenr_T bot);
2148d25c16e2SBram Moolenaar static void foldRemove(garray_T *gap, linenr_T top, linenr_T bot);
2149d25c16e2SBram Moolenaar static void foldMerge(fold_T *fp1, garray_T *gap, fold_T *fp2);
2150d25c16e2SBram Moolenaar static void foldlevelIndent(fline_T *flp);
2151071d4279SBram Moolenaar #ifdef FEAT_DIFF
2152d25c16e2SBram Moolenaar static void foldlevelDiff(fline_T *flp);
2153071d4279SBram Moolenaar #endif
2154d25c16e2SBram Moolenaar static void foldlevelExpr(fline_T *flp);
2155d25c16e2SBram Moolenaar static void foldlevelMarker(fline_T *flp);
2156d25c16e2SBram Moolenaar static void foldlevelSyntax(fline_T *flp);
2157071d4279SBram Moolenaar 
2158071d4279SBram Moolenaar /* foldUpdateIEMS() {{{2 */
2159071d4279SBram Moolenaar /*
2160071d4279SBram Moolenaar  * Update the folding for window "wp", at least from lines "top" to "bot".
2161071d4279SBram Moolenaar  * Return TRUE if any folds did change.
2162071d4279SBram Moolenaar  */
2163071d4279SBram Moolenaar     static void
2164071d4279SBram Moolenaar foldUpdateIEMS(wp, top, bot)
2165071d4279SBram Moolenaar     win_T	*wp;
2166071d4279SBram Moolenaar     linenr_T	top;
2167071d4279SBram Moolenaar     linenr_T	bot;
2168071d4279SBram Moolenaar {
2169071d4279SBram Moolenaar     linenr_T	start;
2170071d4279SBram Moolenaar     linenr_T	end;
2171071d4279SBram Moolenaar     fline_T	fline;
2172*d99df423SBram Moolenaar     void	(*getlevel)(fline_T *);
2173071d4279SBram Moolenaar     int		level;
2174071d4279SBram Moolenaar     fold_T	*fp;
2175071d4279SBram Moolenaar 
2176071d4279SBram Moolenaar     /* Avoid problems when being called recursively. */
2177071d4279SBram Moolenaar     if (invalid_top != (linenr_T)0)
2178071d4279SBram Moolenaar 	return;
2179071d4279SBram Moolenaar 
2180071d4279SBram Moolenaar     if (wp->w_foldinvalid)
2181071d4279SBram Moolenaar     {
2182071d4279SBram Moolenaar 	/* Need to update all folds. */
2183071d4279SBram Moolenaar 	top = 1;
2184071d4279SBram Moolenaar 	bot = wp->w_buffer->b_ml.ml_line_count;
2185071d4279SBram Moolenaar 	wp->w_foldinvalid = FALSE;
2186071d4279SBram Moolenaar 
2187071d4279SBram Moolenaar 	/* Mark all folds a maybe-small. */
2188071d4279SBram Moolenaar 	setSmallMaybe(&wp->w_folds);
2189071d4279SBram Moolenaar     }
2190071d4279SBram Moolenaar 
2191071d4279SBram Moolenaar #ifdef FEAT_DIFF
2192071d4279SBram Moolenaar     /* add the context for "diff" folding */
2193071d4279SBram Moolenaar     if (foldmethodIsDiff(wp))
2194071d4279SBram Moolenaar     {
2195071d4279SBram Moolenaar 	if (top > diff_context)
2196071d4279SBram Moolenaar 	    top -= diff_context;
2197071d4279SBram Moolenaar 	else
2198071d4279SBram Moolenaar 	    top = 1;
2199071d4279SBram Moolenaar 	bot += diff_context;
2200071d4279SBram Moolenaar     }
2201071d4279SBram Moolenaar #endif
2202071d4279SBram Moolenaar 
2203071d4279SBram Moolenaar     /* When deleting lines at the end of the buffer "top" can be past the end
2204071d4279SBram Moolenaar      * of the buffer. */
2205071d4279SBram Moolenaar     if (top > wp->w_buffer->b_ml.ml_line_count)
2206071d4279SBram Moolenaar 	top = wp->w_buffer->b_ml.ml_line_count;
2207071d4279SBram Moolenaar 
2208071d4279SBram Moolenaar     fold_changed = FALSE;
2209071d4279SBram Moolenaar     fline.wp = wp;
2210071d4279SBram Moolenaar     fline.off = 0;
2211071d4279SBram Moolenaar     fline.lvl = 0;
2212071d4279SBram Moolenaar     fline.lvl_next = -1;
2213071d4279SBram Moolenaar     fline.start = 0;
2214071d4279SBram Moolenaar     fline.end = MAX_LEVEL + 1;
2215071d4279SBram Moolenaar     fline.had_end = MAX_LEVEL + 1;
2216071d4279SBram Moolenaar 
2217071d4279SBram Moolenaar     invalid_top = top;
2218071d4279SBram Moolenaar     invalid_bot = bot;
2219071d4279SBram Moolenaar 
2220071d4279SBram Moolenaar     if (foldmethodIsMarker(wp))
2221071d4279SBram Moolenaar     {
2222071d4279SBram Moolenaar 	getlevel = foldlevelMarker;
2223071d4279SBram Moolenaar 
2224071d4279SBram Moolenaar 	/* Init marker variables to speed up foldlevelMarker(). */
2225071d4279SBram Moolenaar 	parseMarker(wp);
2226071d4279SBram Moolenaar 
2227071d4279SBram Moolenaar 	/* Need to get the level of the line above top, it is used if there is
2228071d4279SBram Moolenaar 	 * no marker at the top. */
2229071d4279SBram Moolenaar 	if (top > 1)
2230071d4279SBram Moolenaar 	{
2231071d4279SBram Moolenaar 	    /* Get the fold level at top - 1. */
2232071d4279SBram Moolenaar 	    level = foldLevelWin(wp, top - 1);
2233071d4279SBram Moolenaar 
2234071d4279SBram Moolenaar 	    /* The fold may end just above the top, check for that. */
2235071d4279SBram Moolenaar 	    fline.lnum = top - 1;
2236071d4279SBram Moolenaar 	    fline.lvl = level;
2237071d4279SBram Moolenaar 	    getlevel(&fline);
2238071d4279SBram Moolenaar 
2239071d4279SBram Moolenaar 	    /* If a fold started here, we already had the level, if it stops
2240071d4279SBram Moolenaar 	     * here, we need to use lvl_next.  Could also start and end a fold
2241071d4279SBram Moolenaar 	     * in the same line. */
2242071d4279SBram Moolenaar 	    if (fline.lvl > level)
2243071d4279SBram Moolenaar 		fline.lvl = level - (fline.lvl - fline.lvl_next);
2244071d4279SBram Moolenaar 	    else
2245071d4279SBram Moolenaar 		fline.lvl = fline.lvl_next;
2246071d4279SBram Moolenaar 	}
2247071d4279SBram Moolenaar 	fline.lnum = top;
2248071d4279SBram Moolenaar 	getlevel(&fline);
2249071d4279SBram Moolenaar     }
2250071d4279SBram Moolenaar     else
2251071d4279SBram Moolenaar     {
2252071d4279SBram Moolenaar 	fline.lnum = top;
2253071d4279SBram Moolenaar 	if (foldmethodIsExpr(wp))
2254071d4279SBram Moolenaar 	{
2255071d4279SBram Moolenaar 	    getlevel = foldlevelExpr;
2256071d4279SBram Moolenaar 	    /* start one line back, because a "<1" may indicate the end of a
2257071d4279SBram Moolenaar 	     * fold in the topline */
2258071d4279SBram Moolenaar 	    if (top > 1)
2259071d4279SBram Moolenaar 		--fline.lnum;
2260071d4279SBram Moolenaar 	}
2261071d4279SBram Moolenaar 	else if (foldmethodIsSyntax(wp))
2262071d4279SBram Moolenaar 	    getlevel = foldlevelSyntax;
2263071d4279SBram Moolenaar #ifdef FEAT_DIFF
2264071d4279SBram Moolenaar 	else if (foldmethodIsDiff(wp))
2265071d4279SBram Moolenaar 	    getlevel = foldlevelDiff;
2266071d4279SBram Moolenaar #endif
2267071d4279SBram Moolenaar 	else
2268071d4279SBram Moolenaar 	    getlevel = foldlevelIndent;
2269071d4279SBram Moolenaar 
2270071d4279SBram Moolenaar 	/* Backup to a line for which the fold level is defined.  Since it's
2271071d4279SBram Moolenaar 	 * always defined for line one, we will stop there. */
2272071d4279SBram Moolenaar 	fline.lvl = -1;
2273071d4279SBram Moolenaar 	for ( ; !got_int; --fline.lnum)
2274071d4279SBram Moolenaar 	{
2275071d4279SBram Moolenaar 	    /* Reset lvl_next each time, because it will be set to a value for
2276071d4279SBram Moolenaar 	     * the next line, but we search backwards here. */
2277071d4279SBram Moolenaar 	    fline.lvl_next = -1;
2278071d4279SBram Moolenaar 	    getlevel(&fline);
2279071d4279SBram Moolenaar 	    if (fline.lvl >= 0)
2280071d4279SBram Moolenaar 		break;
2281071d4279SBram Moolenaar 	}
2282071d4279SBram Moolenaar     }
2283071d4279SBram Moolenaar 
2284ec98647bSBram Moolenaar     /*
2285ec98647bSBram Moolenaar      * If folding is defined by the syntax, it is possible that a change in
2286ec98647bSBram Moolenaar      * one line will cause all sub-folds of the current fold to change (e.g.,
2287ec98647bSBram Moolenaar      * closing a C-style comment can cause folds in the subsequent lines to
2288ec98647bSBram Moolenaar      * appear). To take that into account we should adjust the value of "bot"
2289ec98647bSBram Moolenaar      * to point to the end of the current fold:
2290ec98647bSBram Moolenaar      */
2291ec98647bSBram Moolenaar     if (foldlevelSyntax == getlevel)
2292ec98647bSBram Moolenaar     {
2293ec98647bSBram Moolenaar 	garray_T *gap = &wp->w_folds;
2294f506c5bbSBram Moolenaar 	fold_T	 *fpn = NULL;
2295ec98647bSBram Moolenaar 	int	  current_fdl = 0;
2296ec98647bSBram Moolenaar 	linenr_T  fold_start_lnum = 0;
2297ec98647bSBram Moolenaar 	linenr_T  lnum_rel = fline.lnum;
2298ec98647bSBram Moolenaar 
2299ec98647bSBram Moolenaar 	while (current_fdl < fline.lvl)
2300ec98647bSBram Moolenaar 	{
2301f506c5bbSBram Moolenaar 	    if (!foldFind(gap, lnum_rel, &fpn))
2302ec98647bSBram Moolenaar 		break;
2303ec98647bSBram Moolenaar 	    ++current_fdl;
2304ec98647bSBram Moolenaar 
2305f506c5bbSBram Moolenaar 	    fold_start_lnum += fpn->fd_top;
2306f506c5bbSBram Moolenaar 	    gap = &fpn->fd_nested;
2307f506c5bbSBram Moolenaar 	    lnum_rel -= fpn->fd_top;
2308ec98647bSBram Moolenaar 	}
2309f506c5bbSBram Moolenaar 	if (fpn != NULL && current_fdl == fline.lvl)
2310ec98647bSBram Moolenaar 	{
2311f506c5bbSBram Moolenaar 	    linenr_T fold_end_lnum = fold_start_lnum + fpn->fd_len;
2312ec98647bSBram Moolenaar 
2313ec98647bSBram Moolenaar 	    if (fold_end_lnum > bot)
2314ec98647bSBram Moolenaar 		bot = fold_end_lnum;
2315ec98647bSBram Moolenaar 	}
2316ec98647bSBram Moolenaar     }
2317ec98647bSBram Moolenaar 
2318071d4279SBram Moolenaar     start = fline.lnum;
2319071d4279SBram Moolenaar     end = bot;
2320071d4279SBram Moolenaar     /* Do at least one line. */
2321071d4279SBram Moolenaar     if (start > end && end < wp->w_buffer->b_ml.ml_line_count)
2322071d4279SBram Moolenaar 	end = start;
2323071d4279SBram Moolenaar     while (!got_int)
2324071d4279SBram Moolenaar     {
2325071d4279SBram Moolenaar 	/* Always stop at the end of the file ("end" can be past the end of
2326071d4279SBram Moolenaar 	 * the file). */
2327071d4279SBram Moolenaar 	if (fline.lnum > wp->w_buffer->b_ml.ml_line_count)
2328071d4279SBram Moolenaar 	    break;
2329071d4279SBram Moolenaar 	if (fline.lnum > end)
2330071d4279SBram Moolenaar 	{
2331071d4279SBram Moolenaar 	    /* For "marker", "expr"  and "syntax"  methods: If a change caused
2332071d4279SBram Moolenaar 	     * a fold to be removed, we need to continue at least until where
2333071d4279SBram Moolenaar 	     * it ended. */
2334071d4279SBram Moolenaar 	    if (getlevel != foldlevelMarker
2335071d4279SBram Moolenaar 		    && getlevel != foldlevelSyntax
2336071d4279SBram Moolenaar 		    && getlevel != foldlevelExpr)
2337071d4279SBram Moolenaar 		break;
2338071d4279SBram Moolenaar 	    if ((start <= end
2339071d4279SBram Moolenaar 			&& foldFind(&wp->w_folds, end, &fp)
2340071d4279SBram Moolenaar 			&& fp->fd_top + fp->fd_len - 1 > end)
2341071d4279SBram Moolenaar 		    || (fline.lvl == 0
2342071d4279SBram Moolenaar 			&& foldFind(&wp->w_folds, fline.lnum, &fp)
2343071d4279SBram Moolenaar 			&& fp->fd_top < fline.lnum))
2344071d4279SBram Moolenaar 		end = fp->fd_top + fp->fd_len - 1;
2345071d4279SBram Moolenaar 	    else if (getlevel == foldlevelSyntax
2346071d4279SBram Moolenaar 		    && foldLevelWin(wp, fline.lnum) != fline.lvl)
2347071d4279SBram Moolenaar 		/* For "syntax" method: Compare the foldlevel that the syntax
2348071d4279SBram Moolenaar 		 * tells us to the foldlevel from the existing folds.  If they
2349071d4279SBram Moolenaar 		 * don't match continue updating folds. */
2350071d4279SBram Moolenaar 		end = fline.lnum;
2351071d4279SBram Moolenaar 	    else
2352071d4279SBram Moolenaar 		break;
2353071d4279SBram Moolenaar 	}
2354071d4279SBram Moolenaar 
2355071d4279SBram Moolenaar 	/* A level 1 fold starts at a line with foldlevel > 0. */
2356071d4279SBram Moolenaar 	if (fline.lvl > 0)
2357071d4279SBram Moolenaar 	{
2358071d4279SBram Moolenaar 	    invalid_top = fline.lnum;
2359071d4279SBram Moolenaar 	    invalid_bot = end;
2360071d4279SBram Moolenaar 	    end = foldUpdateIEMSRecurse(&wp->w_folds,
2361071d4279SBram Moolenaar 				   1, start, &fline, getlevel, end, FD_LEVEL);
2362071d4279SBram Moolenaar 	    start = fline.lnum;
2363071d4279SBram Moolenaar 	}
2364071d4279SBram Moolenaar 	else
2365071d4279SBram Moolenaar 	{
2366071d4279SBram Moolenaar 	    if (fline.lnum == wp->w_buffer->b_ml.ml_line_count)
2367071d4279SBram Moolenaar 		break;
2368071d4279SBram Moolenaar 	    ++fline.lnum;
2369071d4279SBram Moolenaar 	    fline.lvl = fline.lvl_next;
2370071d4279SBram Moolenaar 	    getlevel(&fline);
2371071d4279SBram Moolenaar 	}
2372071d4279SBram Moolenaar     }
2373071d4279SBram Moolenaar 
2374071d4279SBram Moolenaar     /* There can't be any folds from start until end now. */
2375071d4279SBram Moolenaar     foldRemove(&wp->w_folds, start, end);
2376071d4279SBram Moolenaar 
2377071d4279SBram Moolenaar     /* If some fold changed, need to redraw and position cursor. */
2378071d4279SBram Moolenaar     if (fold_changed && wp->w_p_fen)
2379e3b3f28dSBram Moolenaar 	changed_window_setting_win(wp);
2380071d4279SBram Moolenaar 
2381071d4279SBram Moolenaar     /* If we updated folds past "bot", need to redraw more lines.  Don't do
2382071d4279SBram Moolenaar      * this in other situations, the changed lines will be redrawn anyway and
2383071d4279SBram Moolenaar      * this method can cause the whole window to be updated. */
2384071d4279SBram Moolenaar     if (end != bot)
2385071d4279SBram Moolenaar     {
2386071d4279SBram Moolenaar 	if (wp->w_redraw_top == 0 || wp->w_redraw_top > top)
2387071d4279SBram Moolenaar 	    wp->w_redraw_top = top;
2388071d4279SBram Moolenaar 	if (wp->w_redraw_bot < end)
2389071d4279SBram Moolenaar 	    wp->w_redraw_bot = end;
2390071d4279SBram Moolenaar     }
2391071d4279SBram Moolenaar 
2392071d4279SBram Moolenaar     invalid_top = (linenr_T)0;
2393071d4279SBram Moolenaar }
2394071d4279SBram Moolenaar 
2395071d4279SBram Moolenaar /* foldUpdateIEMSRecurse() {{{2 */
2396071d4279SBram Moolenaar /*
2397071d4279SBram Moolenaar  * Update a fold that starts at "flp->lnum".  At this line there is always a
2398071d4279SBram Moolenaar  * valid foldlevel, and its level >= "level".
2399071d4279SBram Moolenaar  * "flp" is valid for "flp->lnum" when called and it's valid when returning.
2400071d4279SBram Moolenaar  * "flp->lnum" is set to the lnum just below the fold, if it ends before
2401071d4279SBram Moolenaar  * "bot", it's "bot" plus one if the fold continues and it's bigger when using
2402071d4279SBram Moolenaar  * the marker method and a text change made following folds to change.
2403071d4279SBram Moolenaar  * When returning, "flp->lnum_save" is the line number that was used to get
2404071d4279SBram Moolenaar  * the level when the level at "flp->lnum" is invalid.
2405071d4279SBram Moolenaar  * Remove any folds from "startlnum" up to here at this level.
2406071d4279SBram Moolenaar  * Recursively update nested folds.
2407071d4279SBram Moolenaar  * Below line "bot" there are no changes in the text.
2408071d4279SBram Moolenaar  * "flp->lnum", "flp->lnum_save" and "bot" are relative to the start of the
2409071d4279SBram Moolenaar  * outer fold.
2410071d4279SBram Moolenaar  * "flp->off" is the offset to the real line number in the buffer.
2411071d4279SBram Moolenaar  *
2412071d4279SBram Moolenaar  * All this would be a lot simpler if all folds in the range would be deleted
2413e37d50a5SBram Moolenaar  * and then created again.  But we would lose all information about the
2414071d4279SBram Moolenaar  * folds, even when making changes that don't affect the folding (e.g. "vj~").
2415071d4279SBram Moolenaar  *
2416071d4279SBram Moolenaar  * Returns bot, which may have been increased for lines that also need to be
2417071d4279SBram Moolenaar  * updated as a result of a detected change in the fold.
2418071d4279SBram Moolenaar  */
2419071d4279SBram Moolenaar     static linenr_T
2420071d4279SBram Moolenaar foldUpdateIEMSRecurse(gap, level, startlnum, flp, getlevel, bot, topflags)
2421071d4279SBram Moolenaar     garray_T	*gap;
2422071d4279SBram Moolenaar     int		level;
2423071d4279SBram Moolenaar     linenr_T	startlnum;
2424071d4279SBram Moolenaar     fline_T	*flp;
2425*d99df423SBram Moolenaar     void	(*getlevel)(fline_T *);
2426071d4279SBram Moolenaar     linenr_T	bot;
2427071d4279SBram Moolenaar     int		topflags;	/* flags used by containing fold */
2428071d4279SBram Moolenaar {
2429071d4279SBram Moolenaar     linenr_T	ll;
2430071d4279SBram Moolenaar     fold_T	*fp = NULL;
2431071d4279SBram Moolenaar     fold_T	*fp2;
2432071d4279SBram Moolenaar     int		lvl = level;
2433071d4279SBram Moolenaar     linenr_T	startlnum2 = startlnum;
2434071d4279SBram Moolenaar     linenr_T	firstlnum = flp->lnum;	/* first lnum we got */
2435071d4279SBram Moolenaar     int		i;
2436071d4279SBram Moolenaar     int		finish = FALSE;
2437071d4279SBram Moolenaar     linenr_T	linecount = flp->wp->w_buffer->b_ml.ml_line_count - flp->off;
2438071d4279SBram Moolenaar     int		concat;
2439071d4279SBram Moolenaar 
2440071d4279SBram Moolenaar     /*
2441071d4279SBram Moolenaar      * If using the marker method, the start line is not the start of a fold
2442071d4279SBram Moolenaar      * at the level we're dealing with and the level is non-zero, we must use
2443071d4279SBram Moolenaar      * the previous fold.  But ignore a fold that starts at or below
2444071d4279SBram Moolenaar      * startlnum, it must be deleted.
2445071d4279SBram Moolenaar      */
2446071d4279SBram Moolenaar     if (getlevel == foldlevelMarker && flp->start <= flp->lvl - level
2447071d4279SBram Moolenaar 							      && flp->lvl > 0)
2448071d4279SBram Moolenaar     {
2449cde88547SBram Moolenaar 	(void)foldFind(gap, startlnum - 1, &fp);
2450071d4279SBram Moolenaar 	if (fp >= ((fold_T *)gap->ga_data) + gap->ga_len
2451071d4279SBram Moolenaar 						   || fp->fd_top >= startlnum)
2452071d4279SBram Moolenaar 	    fp = NULL;
2453071d4279SBram Moolenaar     }
2454071d4279SBram Moolenaar 
2455071d4279SBram Moolenaar     /*
2456071d4279SBram Moolenaar      * Loop over all lines in this fold, or until "bot" is hit.
2457071d4279SBram Moolenaar      * Handle nested folds inside of this fold.
2458071d4279SBram Moolenaar      * "flp->lnum" is the current line.  When finding the end of the fold, it
2459071d4279SBram Moolenaar      * is just below the end of the fold.
2460071d4279SBram Moolenaar      * "*flp" contains the level of the line "flp->lnum" or a following one if
2461071d4279SBram Moolenaar      * there are lines with an invalid fold level.  "flp->lnum_save" is the
2462071d4279SBram Moolenaar      * line number that was used to get the fold level (below "flp->lnum" when
2463071d4279SBram Moolenaar      * it has an invalid fold level).  When called the fold level is always
2464071d4279SBram Moolenaar      * valid, thus "flp->lnum_save" is equal to "flp->lnum".
2465071d4279SBram Moolenaar      */
2466071d4279SBram Moolenaar     flp->lnum_save = flp->lnum;
2467071d4279SBram Moolenaar     while (!got_int)
2468071d4279SBram Moolenaar     {
2469071d4279SBram Moolenaar 	/* Updating folds can be slow, check for CTRL-C. */
2470071d4279SBram Moolenaar 	line_breakcheck();
2471071d4279SBram Moolenaar 
2472071d4279SBram Moolenaar 	/* Set "lvl" to the level of line "flp->lnum".  When flp->start is set
2473071d4279SBram Moolenaar 	 * and after the first line of the fold, set the level to zero to
2474071d4279SBram Moolenaar 	 * force the fold to end.  Do the same when had_end is set: Previous
2475071d4279SBram Moolenaar 	 * line was marked as end of a fold. */
2476071d4279SBram Moolenaar 	lvl = flp->lvl;
2477071d4279SBram Moolenaar 	if (lvl > MAX_LEVEL)
2478071d4279SBram Moolenaar 	    lvl = MAX_LEVEL;
2479071d4279SBram Moolenaar 	if (flp->lnum > firstlnum
2480071d4279SBram Moolenaar 		&& (level > lvl - flp->start || level >= flp->had_end))
2481071d4279SBram Moolenaar 	    lvl = 0;
2482071d4279SBram Moolenaar 
2483071d4279SBram Moolenaar 	if (flp->lnum > bot && !finish && fp != NULL)
2484071d4279SBram Moolenaar 	{
2485071d4279SBram Moolenaar 	    /* For "marker" and "syntax" methods:
2486071d4279SBram Moolenaar 	     * - If a change caused a nested fold to be removed, we need to
2487071d4279SBram Moolenaar 	     *   delete it and continue at least until where it ended.
2488071d4279SBram Moolenaar 	     * - If a change caused a nested fold to be created, or this fold
2489071d4279SBram Moolenaar 	     *   to continue below its original end, need to finish this fold.
2490071d4279SBram Moolenaar 	     */
2491071d4279SBram Moolenaar 	    if (getlevel != foldlevelMarker
2492071d4279SBram Moolenaar 		    && getlevel != foldlevelExpr
2493071d4279SBram Moolenaar 		    && getlevel != foldlevelSyntax)
2494071d4279SBram Moolenaar 		break;
2495071d4279SBram Moolenaar 	    i = 0;
2496071d4279SBram Moolenaar 	    fp2 = fp;
2497071d4279SBram Moolenaar 	    if (lvl >= level)
2498071d4279SBram Moolenaar 	    {
2499071d4279SBram Moolenaar 		/* Compute how deep the folds currently are, if it's deeper
2500071d4279SBram Moolenaar 		 * than "lvl" then some must be deleted, need to update
2501071d4279SBram Moolenaar 		 * at least one nested fold. */
2502071d4279SBram Moolenaar 		ll = flp->lnum - fp->fd_top;
2503071d4279SBram Moolenaar 		while (foldFind(&fp2->fd_nested, ll, &fp2))
2504071d4279SBram Moolenaar 		{
2505071d4279SBram Moolenaar 		    ++i;
2506071d4279SBram Moolenaar 		    ll -= fp2->fd_top;
2507071d4279SBram Moolenaar 		}
2508071d4279SBram Moolenaar 	    }
2509071d4279SBram Moolenaar 	    if (lvl < level + i)
2510071d4279SBram Moolenaar 	    {
2511cde88547SBram Moolenaar 		(void)foldFind(&fp->fd_nested, flp->lnum - fp->fd_top, &fp2);
2512071d4279SBram Moolenaar 		if (fp2 != NULL)
2513071d4279SBram Moolenaar 		    bot = fp2->fd_top + fp2->fd_len - 1 + fp->fd_top;
2514071d4279SBram Moolenaar 	    }
2515071d4279SBram Moolenaar 	    else if (fp->fd_top + fp->fd_len <= flp->lnum && lvl >= level)
2516071d4279SBram Moolenaar 		finish = TRUE;
2517071d4279SBram Moolenaar 	    else
2518071d4279SBram Moolenaar 		break;
2519071d4279SBram Moolenaar 	}
2520071d4279SBram Moolenaar 
2521071d4279SBram Moolenaar 	/* At the start of the first nested fold and at the end of the current
2522071d4279SBram Moolenaar 	 * fold: check if existing folds at this level, before the current
2523071d4279SBram Moolenaar 	 * one, need to be deleted or truncated. */
2524071d4279SBram Moolenaar 	if (fp == NULL
2525071d4279SBram Moolenaar 		&& (lvl != level
2526071d4279SBram Moolenaar 		    || flp->lnum_save >= bot
2527071d4279SBram Moolenaar 		    || flp->start != 0
2528071d4279SBram Moolenaar 		    || flp->had_end <= MAX_LEVEL
2529071d4279SBram Moolenaar 		    || flp->lnum == linecount))
2530071d4279SBram Moolenaar 	{
2531071d4279SBram Moolenaar 	    /*
2532071d4279SBram Moolenaar 	     * Remove or update folds that have lines between startlnum and
2533071d4279SBram Moolenaar 	     * firstlnum.
2534071d4279SBram Moolenaar 	     */
2535071d4279SBram Moolenaar 	    while (!got_int)
2536071d4279SBram Moolenaar 	    {
2537071d4279SBram Moolenaar 		/* set concat to 1 if it's allowed to concatenated this fold
2538071d4279SBram Moolenaar 		 * with a previous one that touches it. */
2539071d4279SBram Moolenaar 		if (flp->start != 0 || flp->had_end <= MAX_LEVEL)
2540071d4279SBram Moolenaar 		    concat = 0;
2541071d4279SBram Moolenaar 		else
2542071d4279SBram Moolenaar 		    concat = 1;
2543071d4279SBram Moolenaar 
2544071d4279SBram Moolenaar 		/* Find an existing fold to re-use.  Preferably one that
2545071d4279SBram Moolenaar 		 * includes startlnum, otherwise one that ends just before
2546071d4279SBram Moolenaar 		 * startlnum or starts after it. */
2547071d4279SBram Moolenaar 		if (foldFind(gap, startlnum, &fp)
2548071d4279SBram Moolenaar 			|| (fp < ((fold_T *)gap->ga_data) + gap->ga_len
2549071d4279SBram Moolenaar 			    && fp->fd_top <= firstlnum)
2550071d4279SBram Moolenaar 			|| foldFind(gap, firstlnum - concat, &fp)
2551071d4279SBram Moolenaar 			|| (fp < ((fold_T *)gap->ga_data) + gap->ga_len
2552071d4279SBram Moolenaar 			    && ((lvl < level && fp->fd_top < flp->lnum)
2553071d4279SBram Moolenaar 				|| (lvl >= level
2554071d4279SBram Moolenaar 					   && fp->fd_top <= flp->lnum_save))))
2555071d4279SBram Moolenaar 		{
2556071d4279SBram Moolenaar 		    if (fp->fd_top + fp->fd_len + concat > firstlnum)
2557071d4279SBram Moolenaar 		    {
2558071d4279SBram Moolenaar 			/* Use existing fold for the new fold.  If it starts
2559071d4279SBram Moolenaar 			 * before where we started looking, extend it.  If it
2560071d4279SBram Moolenaar 			 * starts at another line, update nested folds to keep
2561071d4279SBram Moolenaar 			 * their position, compensating for the new fd_top. */
2562071d4279SBram Moolenaar 			if (fp->fd_top >= startlnum && fp->fd_top != firstlnum)
2563071d4279SBram Moolenaar 			{
2564071d4279SBram Moolenaar 			    if (fp->fd_top > firstlnum)
2565071d4279SBram Moolenaar 				/* like lines are inserted */
2566071d4279SBram Moolenaar 				foldMarkAdjustRecurse(&fp->fd_nested,
2567071d4279SBram Moolenaar 					(linenr_T)0, (linenr_T)MAXLNUM,
2568071d4279SBram Moolenaar 					(long)(fp->fd_top - firstlnum), 0L);
2569071d4279SBram Moolenaar 			    else
2570071d4279SBram Moolenaar 				/* like lines are deleted */
2571071d4279SBram Moolenaar 				foldMarkAdjustRecurse(&fp->fd_nested,
2572071d4279SBram Moolenaar 					(linenr_T)0,
2573071d4279SBram Moolenaar 					(long)(firstlnum - fp->fd_top - 1),
2574071d4279SBram Moolenaar 					(linenr_T)MAXLNUM,
2575071d4279SBram Moolenaar 					(long)(fp->fd_top - firstlnum));
2576071d4279SBram Moolenaar 			    fp->fd_len += fp->fd_top - firstlnum;
2577071d4279SBram Moolenaar 			    fp->fd_top = firstlnum;
2578071d4279SBram Moolenaar 			    fold_changed = TRUE;
2579071d4279SBram Moolenaar 			}
2580071d4279SBram Moolenaar 			else if (flp->start != 0 && lvl == level
2581071d4279SBram Moolenaar 						   && fp->fd_top != firstlnum)
2582071d4279SBram Moolenaar 			{
2583071d4279SBram Moolenaar 			    /* Existing fold that includes startlnum must stop
2584071d4279SBram Moolenaar 			     * if we find the start of a new fold at the same
2585071d4279SBram Moolenaar 			     * level.  Split it.  Delete contained folds at
2586071d4279SBram Moolenaar 			     * this point to split them too. */
2587071d4279SBram Moolenaar 			    foldRemove(&fp->fd_nested, flp->lnum - fp->fd_top,
2588071d4279SBram Moolenaar 						      flp->lnum - fp->fd_top);
2589071d4279SBram Moolenaar 			    i = (int)(fp - (fold_T *)gap->ga_data);
2590071d4279SBram Moolenaar 			    foldSplit(gap, i, flp->lnum, flp->lnum - 1);
2591071d4279SBram Moolenaar 			    fp = (fold_T *)gap->ga_data + i + 1;
2592071d4279SBram Moolenaar 			    /* If using the "marker" or "syntax" method, we
2593071d4279SBram Moolenaar 			     * need to continue until the end of the fold is
2594071d4279SBram Moolenaar 			     * found. */
2595071d4279SBram Moolenaar 			    if (getlevel == foldlevelMarker
2596071d4279SBram Moolenaar 				    || getlevel == foldlevelExpr
2597071d4279SBram Moolenaar 				    || getlevel == foldlevelSyntax)
2598071d4279SBram Moolenaar 				finish = TRUE;
2599071d4279SBram Moolenaar 			}
2600071d4279SBram Moolenaar 			break;
2601071d4279SBram Moolenaar 		    }
2602071d4279SBram Moolenaar 		    if (fp->fd_top >= startlnum)
2603071d4279SBram Moolenaar 		    {
2604071d4279SBram Moolenaar 			/* A fold that starts at or after startlnum and stops
2605071d4279SBram Moolenaar 			 * before the new fold must be deleted.  Continue
2606071d4279SBram Moolenaar 			 * looking for the next one. */
2607071d4279SBram Moolenaar 			deleteFoldEntry(gap,
2608071d4279SBram Moolenaar 				     (int)(fp - (fold_T *)gap->ga_data), TRUE);
2609071d4279SBram Moolenaar 		    }
2610071d4279SBram Moolenaar 		    else
2611071d4279SBram Moolenaar 		    {
2612071d4279SBram Moolenaar 			/* A fold has some lines above startlnum, truncate it
2613071d4279SBram Moolenaar 			 * to stop just above startlnum.  */
2614071d4279SBram Moolenaar 			fp->fd_len = startlnum - fp->fd_top;
2615071d4279SBram Moolenaar 			foldMarkAdjustRecurse(&fp->fd_nested,
2616071d4279SBram Moolenaar 				(linenr_T)fp->fd_len, (linenr_T)MAXLNUM,
2617071d4279SBram Moolenaar 						       (linenr_T)MAXLNUM, 0L);
2618071d4279SBram Moolenaar 			fold_changed = TRUE;
2619071d4279SBram Moolenaar 		    }
2620071d4279SBram Moolenaar 		}
2621071d4279SBram Moolenaar 		else
2622071d4279SBram Moolenaar 		{
2623071d4279SBram Moolenaar 		    /* Insert new fold.  Careful: ga_data may be NULL and it
2624071d4279SBram Moolenaar 		     * may change! */
2625071d4279SBram Moolenaar 		    i = (int)(fp - (fold_T *)gap->ga_data);
2626071d4279SBram Moolenaar 		    if (foldInsert(gap, i) != OK)
2627071d4279SBram Moolenaar 			return bot;
2628071d4279SBram Moolenaar 		    fp = (fold_T *)gap->ga_data + i;
2629071d4279SBram Moolenaar 		    /* The new fold continues until bot, unless we find the
2630071d4279SBram Moolenaar 		     * end earlier. */
2631071d4279SBram Moolenaar 		    fp->fd_top = firstlnum;
2632071d4279SBram Moolenaar 		    fp->fd_len = bot - firstlnum + 1;
2633071d4279SBram Moolenaar 		    /* When the containing fold is open, the new fold is open.
2634071d4279SBram Moolenaar 		     * The new fold is closed if the fold above it is closed.
2635071d4279SBram Moolenaar 		     * The first fold depends on the containing fold. */
2636071d4279SBram Moolenaar 		    if (topflags == FD_OPEN)
2637071d4279SBram Moolenaar 		    {
2638071d4279SBram Moolenaar 			flp->wp->w_fold_manual = TRUE;
2639071d4279SBram Moolenaar 			fp->fd_flags = FD_OPEN;
2640071d4279SBram Moolenaar 		    }
2641071d4279SBram Moolenaar 		    else if (i <= 0)
2642071d4279SBram Moolenaar 		    {
2643071d4279SBram Moolenaar 			fp->fd_flags = topflags;
2644071d4279SBram Moolenaar 			if (topflags != FD_LEVEL)
2645071d4279SBram Moolenaar 			    flp->wp->w_fold_manual = TRUE;
2646071d4279SBram Moolenaar 		    }
2647071d4279SBram Moolenaar 		    else
2648071d4279SBram Moolenaar 			fp->fd_flags = (fp - 1)->fd_flags;
2649071d4279SBram Moolenaar 		    fp->fd_small = MAYBE;
2650071d4279SBram Moolenaar 		    /* If using the "marker", "expr" or "syntax" method, we
2651071d4279SBram Moolenaar 		     * need to continue until the end of the fold is found. */
2652071d4279SBram Moolenaar 		    if (getlevel == foldlevelMarker
2653071d4279SBram Moolenaar 			    || getlevel == foldlevelExpr
2654071d4279SBram Moolenaar 			    || getlevel == foldlevelSyntax)
2655071d4279SBram Moolenaar 			finish = TRUE;
2656071d4279SBram Moolenaar 		    fold_changed = TRUE;
2657071d4279SBram Moolenaar 		    break;
2658071d4279SBram Moolenaar 		}
2659071d4279SBram Moolenaar 	    }
2660071d4279SBram Moolenaar 	}
2661071d4279SBram Moolenaar 
2662071d4279SBram Moolenaar 	if (lvl < level || flp->lnum > linecount)
2663071d4279SBram Moolenaar 	{
2664071d4279SBram Moolenaar 	    /*
2665071d4279SBram Moolenaar 	     * Found a line with a lower foldlevel, this fold ends just above
2666071d4279SBram Moolenaar 	     * "flp->lnum".
2667071d4279SBram Moolenaar 	     */
2668071d4279SBram Moolenaar 	    break;
2669071d4279SBram Moolenaar 	}
2670071d4279SBram Moolenaar 
2671071d4279SBram Moolenaar 	/*
2672071d4279SBram Moolenaar 	 * The fold includes the line "flp->lnum" and "flp->lnum_save".
2673eb3593b3SBram Moolenaar 	 * Check "fp" for safety.
2674071d4279SBram Moolenaar 	 */
2675eb3593b3SBram Moolenaar 	if (lvl > level && fp != NULL)
2676071d4279SBram Moolenaar 	{
2677071d4279SBram Moolenaar 	    /*
2678071d4279SBram Moolenaar 	     * There is a nested fold, handle it recursively.
2679071d4279SBram Moolenaar 	     */
2680071d4279SBram Moolenaar 	    /* At least do one line (can happen when finish is TRUE). */
2681071d4279SBram Moolenaar 	    if (bot < flp->lnum)
2682071d4279SBram Moolenaar 		bot = flp->lnum;
2683071d4279SBram Moolenaar 
2684071d4279SBram Moolenaar 	    /* Line numbers in the nested fold are relative to the start of
2685071d4279SBram Moolenaar 	     * this fold. */
2686071d4279SBram Moolenaar 	    flp->lnum = flp->lnum_save - fp->fd_top;
2687071d4279SBram Moolenaar 	    flp->off += fp->fd_top;
2688071d4279SBram Moolenaar 	    i = (int)(fp - (fold_T *)gap->ga_data);
2689071d4279SBram Moolenaar 	    bot = foldUpdateIEMSRecurse(&fp->fd_nested, level + 1,
2690071d4279SBram Moolenaar 				       startlnum2 - fp->fd_top, flp, getlevel,
2691071d4279SBram Moolenaar 					      bot - fp->fd_top, fp->fd_flags);
2692071d4279SBram Moolenaar 	    fp = (fold_T *)gap->ga_data + i;
2693071d4279SBram Moolenaar 	    flp->lnum += fp->fd_top;
2694071d4279SBram Moolenaar 	    flp->lnum_save += fp->fd_top;
2695071d4279SBram Moolenaar 	    flp->off -= fp->fd_top;
2696071d4279SBram Moolenaar 	    bot += fp->fd_top;
2697071d4279SBram Moolenaar 	    startlnum2 = flp->lnum;
2698071d4279SBram Moolenaar 
2699071d4279SBram Moolenaar 	    /* This fold may end at the same line, don't incr. flp->lnum. */
2700071d4279SBram Moolenaar 	}
2701071d4279SBram Moolenaar 	else
2702071d4279SBram Moolenaar 	{
2703071d4279SBram Moolenaar 	    /*
2704071d4279SBram Moolenaar 	     * Get the level of the next line, then continue the loop to check
2705071d4279SBram Moolenaar 	     * if it ends there.
2706071d4279SBram Moolenaar 	     * Skip over undefined lines, to find the foldlevel after it.
2707071d4279SBram Moolenaar 	     * For the last line in the file the foldlevel is always valid.
2708071d4279SBram Moolenaar 	     */
2709071d4279SBram Moolenaar 	    flp->lnum = flp->lnum_save;
2710071d4279SBram Moolenaar 	    ll = flp->lnum + 1;
2711071d4279SBram Moolenaar 	    while (!got_int)
2712071d4279SBram Moolenaar 	    {
2713071d4279SBram Moolenaar 		/* Make the previous level available to foldlevel(). */
2714071d4279SBram Moolenaar 		prev_lnum = flp->lnum;
2715071d4279SBram Moolenaar 		prev_lnum_lvl = flp->lvl;
2716071d4279SBram Moolenaar 
2717071d4279SBram Moolenaar 		if (++flp->lnum > linecount)
2718071d4279SBram Moolenaar 		    break;
2719071d4279SBram Moolenaar 		flp->lvl = flp->lvl_next;
2720071d4279SBram Moolenaar 		getlevel(flp);
2721071d4279SBram Moolenaar 		if (flp->lvl >= 0 || flp->had_end <= MAX_LEVEL)
2722071d4279SBram Moolenaar 		    break;
2723071d4279SBram Moolenaar 	    }
2724071d4279SBram Moolenaar 	    prev_lnum = 0;
2725071d4279SBram Moolenaar 	    if (flp->lnum > linecount)
2726071d4279SBram Moolenaar 		break;
2727071d4279SBram Moolenaar 
2728071d4279SBram Moolenaar 	    /* leave flp->lnum_save to lnum of the line that was used to get
2729071d4279SBram Moolenaar 	     * the level, flp->lnum to the lnum of the next line. */
2730071d4279SBram Moolenaar 	    flp->lnum_save = flp->lnum;
2731071d4279SBram Moolenaar 	    flp->lnum = ll;
2732071d4279SBram Moolenaar 	}
2733071d4279SBram Moolenaar     }
2734071d4279SBram Moolenaar 
2735071d4279SBram Moolenaar     if (fp == NULL)	/* only happens when got_int is set */
2736071d4279SBram Moolenaar 	return bot;
2737071d4279SBram Moolenaar 
2738071d4279SBram Moolenaar     /*
2739071d4279SBram Moolenaar      * Get here when:
2740071d4279SBram Moolenaar      * lvl < level: the folds ends just above "flp->lnum"
2741071d4279SBram Moolenaar      * lvl >= level: fold continues below "bot"
2742071d4279SBram Moolenaar      */
2743071d4279SBram Moolenaar 
2744071d4279SBram Moolenaar     /* Current fold at least extends until lnum. */
2745071d4279SBram Moolenaar     if (fp->fd_len < flp->lnum - fp->fd_top)
2746071d4279SBram Moolenaar     {
2747071d4279SBram Moolenaar 	fp->fd_len = flp->lnum - fp->fd_top;
2748ef6fc097SBram Moolenaar 	fp->fd_small = MAYBE;
2749071d4279SBram Moolenaar 	fold_changed = TRUE;
2750071d4279SBram Moolenaar     }
2751071d4279SBram Moolenaar 
2752071d4279SBram Moolenaar     /* Delete contained folds from the end of the last one found until where
2753071d4279SBram Moolenaar      * we stopped looking. */
2754071d4279SBram Moolenaar     foldRemove(&fp->fd_nested, startlnum2 - fp->fd_top,
2755071d4279SBram Moolenaar 						  flp->lnum - 1 - fp->fd_top);
2756071d4279SBram Moolenaar 
2757071d4279SBram Moolenaar     if (lvl < level)
2758071d4279SBram Moolenaar     {
2759071d4279SBram Moolenaar 	/* End of fold found, update the length when it got shorter. */
2760071d4279SBram Moolenaar 	if (fp->fd_len != flp->lnum - fp->fd_top)
2761071d4279SBram Moolenaar 	{
2762071d4279SBram Moolenaar 	    if (fp->fd_top + fp->fd_len > bot + 1)
2763071d4279SBram Moolenaar 	    {
276425394022SBram Moolenaar 		/* fold continued below bot */
2765071d4279SBram Moolenaar 		if (getlevel == foldlevelMarker
2766071d4279SBram Moolenaar 			|| getlevel == foldlevelExpr
2767071d4279SBram Moolenaar 			|| getlevel == foldlevelSyntax)
2768071d4279SBram Moolenaar 		{
2769071d4279SBram Moolenaar 		    /* marker method: truncate the fold and make sure the
2770071d4279SBram Moolenaar 		     * previously included lines are processed again */
2771071d4279SBram Moolenaar 		    bot = fp->fd_top + fp->fd_len - 1;
2772071d4279SBram Moolenaar 		    fp->fd_len = flp->lnum - fp->fd_top;
2773071d4279SBram Moolenaar 		}
2774071d4279SBram Moolenaar 		else
2775071d4279SBram Moolenaar 		{
2776071d4279SBram Moolenaar 		    /* indent or expr method: split fold to create a new one
2777071d4279SBram Moolenaar 		     * below bot */
2778071d4279SBram Moolenaar 		    i = (int)(fp - (fold_T *)gap->ga_data);
2779071d4279SBram Moolenaar 		    foldSplit(gap, i, flp->lnum, bot);
2780071d4279SBram Moolenaar 		    fp = (fold_T *)gap->ga_data + i;
2781071d4279SBram Moolenaar 		}
2782071d4279SBram Moolenaar 	    }
2783071d4279SBram Moolenaar 	    else
2784071d4279SBram Moolenaar 		fp->fd_len = flp->lnum - fp->fd_top;
2785071d4279SBram Moolenaar 	    fold_changed = TRUE;
2786071d4279SBram Moolenaar 	}
2787071d4279SBram Moolenaar     }
2788071d4279SBram Moolenaar 
2789071d4279SBram Moolenaar     /* delete following folds that end before the current line */
2790071d4279SBram Moolenaar     for (;;)
2791071d4279SBram Moolenaar     {
2792071d4279SBram Moolenaar 	fp2 = fp + 1;
2793071d4279SBram Moolenaar 	if (fp2 >= (fold_T *)gap->ga_data + gap->ga_len
2794071d4279SBram Moolenaar 						  || fp2->fd_top > flp->lnum)
2795071d4279SBram Moolenaar 	    break;
2796071d4279SBram Moolenaar 	if (fp2->fd_top + fp2->fd_len > flp->lnum)
2797071d4279SBram Moolenaar 	{
2798071d4279SBram Moolenaar 	    if (fp2->fd_top < flp->lnum)
2799071d4279SBram Moolenaar 	    {
2800071d4279SBram Moolenaar 		/* Make fold that includes lnum start at lnum. */
2801071d4279SBram Moolenaar 		foldMarkAdjustRecurse(&fp2->fd_nested,
2802071d4279SBram Moolenaar 			(linenr_T)0, (long)(flp->lnum - fp2->fd_top - 1),
2803071d4279SBram Moolenaar 			(linenr_T)MAXLNUM, (long)(fp2->fd_top - flp->lnum));
2804071d4279SBram Moolenaar 		fp2->fd_len -= flp->lnum - fp2->fd_top;
2805071d4279SBram Moolenaar 		fp2->fd_top = flp->lnum;
2806071d4279SBram Moolenaar 		fold_changed = TRUE;
2807071d4279SBram Moolenaar 	    }
2808071d4279SBram Moolenaar 
2809071d4279SBram Moolenaar 	    if (lvl >= level)
2810071d4279SBram Moolenaar 	    {
2811071d4279SBram Moolenaar 		/* merge new fold with existing fold that follows */
2812071d4279SBram Moolenaar 		foldMerge(fp, gap, fp2);
2813071d4279SBram Moolenaar 	    }
2814071d4279SBram Moolenaar 	    break;
2815071d4279SBram Moolenaar 	}
2816071d4279SBram Moolenaar 	fold_changed = TRUE;
2817071d4279SBram Moolenaar 	deleteFoldEntry(gap, (int)(fp2 - (fold_T *)gap->ga_data), TRUE);
2818071d4279SBram Moolenaar     }
2819071d4279SBram Moolenaar 
2820071d4279SBram Moolenaar     /* Need to redraw the lines we inspected, which might be further down than
2821071d4279SBram Moolenaar      * was asked for. */
2822071d4279SBram Moolenaar     if (bot < flp->lnum - 1)
2823071d4279SBram Moolenaar 	bot = flp->lnum - 1;
2824071d4279SBram Moolenaar 
2825071d4279SBram Moolenaar     return bot;
2826071d4279SBram Moolenaar }
2827071d4279SBram Moolenaar 
2828071d4279SBram Moolenaar /* foldInsert() {{{2 */
2829071d4279SBram Moolenaar /*
2830071d4279SBram Moolenaar  * Insert a new fold in "gap" at position "i".
2831071d4279SBram Moolenaar  * Returns OK for success, FAIL for failure.
2832071d4279SBram Moolenaar  */
2833071d4279SBram Moolenaar     static int
2834071d4279SBram Moolenaar foldInsert(gap, i)
2835071d4279SBram Moolenaar     garray_T	*gap;
2836071d4279SBram Moolenaar     int		i;
2837071d4279SBram Moolenaar {
2838071d4279SBram Moolenaar     fold_T	*fp;
2839071d4279SBram Moolenaar 
2840071d4279SBram Moolenaar     if (ga_grow(gap, 1) != OK)
2841071d4279SBram Moolenaar 	return FAIL;
2842071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data + i;
2843071d4279SBram Moolenaar     if (i < gap->ga_len)
2844071d4279SBram Moolenaar 	mch_memmove(fp + 1, fp, sizeof(fold_T) * (gap->ga_len - i));
2845071d4279SBram Moolenaar     ++gap->ga_len;
2846071d4279SBram Moolenaar     ga_init2(&fp->fd_nested, (int)sizeof(fold_T), 10);
2847071d4279SBram Moolenaar     return OK;
2848071d4279SBram Moolenaar }
2849071d4279SBram Moolenaar 
2850071d4279SBram Moolenaar /* foldSplit() {{{2 */
2851071d4279SBram Moolenaar /*
2852071d4279SBram Moolenaar  * Split the "i"th fold in "gap", which starts before "top" and ends below
2853071d4279SBram Moolenaar  * "bot" in two pieces, one ending above "top" and the other starting below
2854071d4279SBram Moolenaar  * "bot".
2855071d4279SBram Moolenaar  * The caller must first have taken care of any nested folds from "top" to
2856071d4279SBram Moolenaar  * "bot"!
2857071d4279SBram Moolenaar  */
2858071d4279SBram Moolenaar     static void
2859071d4279SBram Moolenaar foldSplit(gap, i, top, bot)
2860071d4279SBram Moolenaar     garray_T	*gap;
2861071d4279SBram Moolenaar     int		i;
2862071d4279SBram Moolenaar     linenr_T	top;
2863071d4279SBram Moolenaar     linenr_T	bot;
2864071d4279SBram Moolenaar {
2865071d4279SBram Moolenaar     fold_T	*fp;
2866071d4279SBram Moolenaar     fold_T	*fp2;
2867071d4279SBram Moolenaar     garray_T	*gap1;
2868071d4279SBram Moolenaar     garray_T	*gap2;
2869071d4279SBram Moolenaar     int		idx;
2870071d4279SBram Moolenaar     int		len;
2871071d4279SBram Moolenaar 
2872071d4279SBram Moolenaar     /* The fold continues below bot, need to split it. */
2873071d4279SBram Moolenaar     if (foldInsert(gap, i + 1) == FAIL)
2874071d4279SBram Moolenaar 	return;
2875071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data + i;
2876071d4279SBram Moolenaar     fp[1].fd_top = bot + 1;
2877071d4279SBram Moolenaar     fp[1].fd_len = fp->fd_len - (fp[1].fd_top - fp->fd_top);
2878071d4279SBram Moolenaar     fp[1].fd_flags = fp->fd_flags;
2879eda6eb9fSBram Moolenaar     fp[1].fd_small = MAYBE;
2880eda6eb9fSBram Moolenaar     fp->fd_small = MAYBE;
2881071d4279SBram Moolenaar 
2882071d4279SBram Moolenaar     /* Move nested folds below bot to new fold.  There can't be
2883071d4279SBram Moolenaar      * any between top and bot, they have been removed by the caller. */
2884071d4279SBram Moolenaar     gap1 = &fp->fd_nested;
2885071d4279SBram Moolenaar     gap2 = &fp[1].fd_nested;
2886071d4279SBram Moolenaar     (void)(foldFind(gap1, bot + 1 - fp->fd_top, &fp2));
2887071d4279SBram Moolenaar     len = (int)((fold_T *)gap1->ga_data + gap1->ga_len - fp2);
2888071d4279SBram Moolenaar     if (len > 0 && ga_grow(gap2, len) == OK)
2889071d4279SBram Moolenaar     {
2890071d4279SBram Moolenaar 	for (idx = 0; idx < len; ++idx)
2891071d4279SBram Moolenaar 	{
2892071d4279SBram Moolenaar 	    ((fold_T *)gap2->ga_data)[idx] = fp2[idx];
2893071d4279SBram Moolenaar 	    ((fold_T *)gap2->ga_data)[idx].fd_top
2894071d4279SBram Moolenaar 						 -= fp[1].fd_top - fp->fd_top;
2895071d4279SBram Moolenaar 	}
2896071d4279SBram Moolenaar 	gap2->ga_len = len;
2897071d4279SBram Moolenaar 	gap1->ga_len -= len;
2898071d4279SBram Moolenaar     }
2899071d4279SBram Moolenaar     fp->fd_len = top - fp->fd_top;
2900071d4279SBram Moolenaar     fold_changed = TRUE;
2901071d4279SBram Moolenaar }
2902071d4279SBram Moolenaar 
2903071d4279SBram Moolenaar /* foldRemove() {{{2 */
2904071d4279SBram Moolenaar /*
2905071d4279SBram Moolenaar  * Remove folds within the range "top" to and including "bot".
2906071d4279SBram Moolenaar  * Check for these situations:
2907071d4279SBram Moolenaar  *      1  2  3
2908071d4279SBram Moolenaar  *      1  2  3
2909071d4279SBram Moolenaar  * top     2  3  4  5
2910071d4279SBram Moolenaar  *	   2  3  4  5
2911071d4279SBram Moolenaar  * bot	   2  3  4  5
2912071d4279SBram Moolenaar  *	      3     5  6
2913071d4279SBram Moolenaar  *	      3     5  6
2914071d4279SBram Moolenaar  *
2915071d4279SBram Moolenaar  * 1: not changed
291625394022SBram Moolenaar  * 2: truncate to stop above "top"
2917071d4279SBram Moolenaar  * 3: split in two parts, one stops above "top", other starts below "bot".
2918071d4279SBram Moolenaar  * 4: deleted
2919071d4279SBram Moolenaar  * 5: made to start below "bot".
2920071d4279SBram Moolenaar  * 6: not changed
2921071d4279SBram Moolenaar  */
2922071d4279SBram Moolenaar     static void
2923071d4279SBram Moolenaar foldRemove(gap, top, bot)
2924071d4279SBram Moolenaar     garray_T	*gap;
2925071d4279SBram Moolenaar     linenr_T	top;
2926071d4279SBram Moolenaar     linenr_T	bot;
2927071d4279SBram Moolenaar {
2928071d4279SBram Moolenaar     fold_T	*fp = NULL;
2929071d4279SBram Moolenaar 
2930071d4279SBram Moolenaar     if (bot < top)
2931071d4279SBram Moolenaar 	return;		/* nothing to do */
2932071d4279SBram Moolenaar 
2933071d4279SBram Moolenaar     for (;;)
2934071d4279SBram Moolenaar     {
2935071d4279SBram Moolenaar 	/* Find fold that includes top or a following one. */
2936071d4279SBram Moolenaar 	if (foldFind(gap, top, &fp) && fp->fd_top < top)
2937071d4279SBram Moolenaar 	{
2938071d4279SBram Moolenaar 	    /* 2: or 3: need to delete nested folds */
2939071d4279SBram Moolenaar 	    foldRemove(&fp->fd_nested, top - fp->fd_top, bot - fp->fd_top);
2940071d4279SBram Moolenaar 	    if (fp->fd_top + fp->fd_len > bot + 1)
2941071d4279SBram Moolenaar 	    {
2942071d4279SBram Moolenaar 		/* 3: need to split it. */
2943071d4279SBram Moolenaar 		foldSplit(gap, (int)(fp - (fold_T *)gap->ga_data), top, bot);
2944071d4279SBram Moolenaar 	    }
2945071d4279SBram Moolenaar 	    else
2946071d4279SBram Moolenaar 	    {
2947071d4279SBram Moolenaar 		/* 2: truncate fold at "top". */
2948071d4279SBram Moolenaar 		fp->fd_len = top - fp->fd_top;
2949071d4279SBram Moolenaar 	    }
2950071d4279SBram Moolenaar 	    fold_changed = TRUE;
2951071d4279SBram Moolenaar 	    continue;
2952071d4279SBram Moolenaar 	}
2953071d4279SBram Moolenaar 	if (fp >= (fold_T *)(gap->ga_data) + gap->ga_len
2954071d4279SBram Moolenaar 		|| fp->fd_top > bot)
2955071d4279SBram Moolenaar 	{
2956071d4279SBram Moolenaar 	    /* 6: Found a fold below bot, can stop looking. */
2957071d4279SBram Moolenaar 	    break;
2958071d4279SBram Moolenaar 	}
2959071d4279SBram Moolenaar 	if (fp->fd_top >= top)
2960071d4279SBram Moolenaar 	{
2961071d4279SBram Moolenaar 	    /* Found an entry below top. */
2962071d4279SBram Moolenaar 	    fold_changed = TRUE;
2963293ee4d4SBram Moolenaar 	    if (fp->fd_top + fp->fd_len - 1 > bot)
2964071d4279SBram Moolenaar 	    {
2965071d4279SBram Moolenaar 		/* 5: Make fold that includes bot start below bot. */
2966071d4279SBram Moolenaar 		foldMarkAdjustRecurse(&fp->fd_nested,
2967071d4279SBram Moolenaar 			(linenr_T)0, (long)(bot - fp->fd_top),
2968071d4279SBram Moolenaar 			(linenr_T)MAXLNUM, (long)(fp->fd_top - bot - 1));
2969071d4279SBram Moolenaar 		fp->fd_len -= bot - fp->fd_top + 1;
2970071d4279SBram Moolenaar 		fp->fd_top = bot + 1;
2971071d4279SBram Moolenaar 		break;
2972071d4279SBram Moolenaar 	    }
2973071d4279SBram Moolenaar 
2974071d4279SBram Moolenaar 	    /* 4: Delete completely contained fold. */
2975071d4279SBram Moolenaar 	    deleteFoldEntry(gap, (int)(fp - (fold_T *)gap->ga_data), TRUE);
2976071d4279SBram Moolenaar 	}
2977071d4279SBram Moolenaar     }
2978071d4279SBram Moolenaar }
2979071d4279SBram Moolenaar 
2980071d4279SBram Moolenaar /* foldMerge() {{{2 */
2981071d4279SBram Moolenaar /*
298225394022SBram Moolenaar  * Merge two adjacent folds (and the nested ones in them).
298325394022SBram Moolenaar  * This only works correctly when the folds are really adjacent!  Thus "fp1"
2984071d4279SBram Moolenaar  * must end just above "fp2".
2985071d4279SBram Moolenaar  * The resulting fold is "fp1", nested folds are moved from "fp2" to "fp1".
2986071d4279SBram Moolenaar  * Fold entry "fp2" in "gap" is deleted.
2987071d4279SBram Moolenaar  */
2988071d4279SBram Moolenaar     static void
2989071d4279SBram Moolenaar foldMerge(fp1, gap, fp2)
2990071d4279SBram Moolenaar     fold_T	*fp1;
2991071d4279SBram Moolenaar     garray_T	*gap;
2992071d4279SBram Moolenaar     fold_T	*fp2;
2993071d4279SBram Moolenaar {
2994071d4279SBram Moolenaar     fold_T	*fp3;
2995071d4279SBram Moolenaar     fold_T	*fp4;
2996071d4279SBram Moolenaar     int		idx;
2997071d4279SBram Moolenaar     garray_T	*gap1 = &fp1->fd_nested;
2998071d4279SBram Moolenaar     garray_T	*gap2 = &fp2->fd_nested;
2999071d4279SBram Moolenaar 
3000071d4279SBram Moolenaar     /* If the last nested fold in fp1 touches the first nested fold in fp2,
3001071d4279SBram Moolenaar      * merge them recursively. */
3002071d4279SBram Moolenaar     if (foldFind(gap1, fp1->fd_len - 1L, &fp3) && foldFind(gap2, 0L, &fp4))
3003071d4279SBram Moolenaar 	foldMerge(fp3, gap2, fp4);
3004071d4279SBram Moolenaar 
3005071d4279SBram Moolenaar     /* Move nested folds in fp2 to the end of fp1. */
3006071d4279SBram Moolenaar     if (gap2->ga_len > 0 && ga_grow(gap1, gap2->ga_len) == OK)
3007071d4279SBram Moolenaar     {
3008071d4279SBram Moolenaar 	for (idx = 0; idx < gap2->ga_len; ++idx)
3009071d4279SBram Moolenaar 	{
3010071d4279SBram Moolenaar 	    ((fold_T *)gap1->ga_data)[gap1->ga_len]
3011071d4279SBram Moolenaar 					= ((fold_T *)gap2->ga_data)[idx];
3012071d4279SBram Moolenaar 	    ((fold_T *)gap1->ga_data)[gap1->ga_len].fd_top += fp1->fd_len;
3013071d4279SBram Moolenaar 	    ++gap1->ga_len;
3014071d4279SBram Moolenaar 	}
3015071d4279SBram Moolenaar 	gap2->ga_len = 0;
3016071d4279SBram Moolenaar     }
3017071d4279SBram Moolenaar 
3018071d4279SBram Moolenaar     fp1->fd_len += fp2->fd_len;
3019071d4279SBram Moolenaar     deleteFoldEntry(gap, (int)(fp2 - (fold_T *)gap->ga_data), TRUE);
3020071d4279SBram Moolenaar     fold_changed = TRUE;
3021071d4279SBram Moolenaar }
3022071d4279SBram Moolenaar 
3023071d4279SBram Moolenaar /* foldlevelIndent() {{{2 */
3024071d4279SBram Moolenaar /*
3025071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "indent" method.
3026071d4279SBram Moolenaar  * Doesn't use any caching.
3027071d4279SBram Moolenaar  * Returns a level of -1 if the foldlevel depends on surrounding lines.
3028071d4279SBram Moolenaar  */
3029071d4279SBram Moolenaar     static void
3030071d4279SBram Moolenaar foldlevelIndent(flp)
3031071d4279SBram Moolenaar     fline_T	*flp;
3032071d4279SBram Moolenaar {
3033071d4279SBram Moolenaar     char_u	*s;
3034071d4279SBram Moolenaar     buf_T	*buf;
3035071d4279SBram Moolenaar     linenr_T	lnum = flp->lnum + flp->off;
3036071d4279SBram Moolenaar 
3037071d4279SBram Moolenaar     buf = flp->wp->w_buffer;
3038071d4279SBram Moolenaar     s = skipwhite(ml_get_buf(buf, lnum, FALSE));
3039071d4279SBram Moolenaar 
3040071d4279SBram Moolenaar     /* empty line or lines starting with a character in 'foldignore': level
3041071d4279SBram Moolenaar      * depends on surrounding lines */
3042071d4279SBram Moolenaar     if (*s == NUL || vim_strchr(flp->wp->w_p_fdi, *s) != NULL)
3043071d4279SBram Moolenaar     {
3044071d4279SBram Moolenaar 	/* first and last line can't be undefined, use level 0 */
3045071d4279SBram Moolenaar 	if (lnum == 1 || lnum == buf->b_ml.ml_line_count)
3046071d4279SBram Moolenaar 	    flp->lvl = 0;
3047071d4279SBram Moolenaar 	else
3048071d4279SBram Moolenaar 	    flp->lvl = -1;
3049071d4279SBram Moolenaar     }
3050071d4279SBram Moolenaar     else
30516bcbcc59SBram Moolenaar 	flp->lvl = get_indent_buf(buf, lnum) / get_sw_value(curbuf);
3052071d4279SBram Moolenaar     if (flp->lvl > flp->wp->w_p_fdn)
305374c596b5SBram Moolenaar     {
3054071d4279SBram Moolenaar 	flp->lvl = flp->wp->w_p_fdn;
305574c596b5SBram Moolenaar 	if (flp->lvl < 0)
305674c596b5SBram Moolenaar 	    flp->lvl = 0;
305774c596b5SBram Moolenaar     }
3058071d4279SBram Moolenaar }
3059071d4279SBram Moolenaar 
3060071d4279SBram Moolenaar /* foldlevelDiff() {{{2 */
3061071d4279SBram Moolenaar #ifdef FEAT_DIFF
3062071d4279SBram Moolenaar /*
3063071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "diff" method.
3064071d4279SBram Moolenaar  * Doesn't use any caching.
3065071d4279SBram Moolenaar  */
3066071d4279SBram Moolenaar     static void
3067071d4279SBram Moolenaar foldlevelDiff(flp)
3068071d4279SBram Moolenaar     fline_T	*flp;
3069071d4279SBram Moolenaar {
3070071d4279SBram Moolenaar     if (diff_infold(flp->wp, flp->lnum + flp->off))
3071071d4279SBram Moolenaar 	flp->lvl = 1;
3072071d4279SBram Moolenaar     else
3073071d4279SBram Moolenaar 	flp->lvl = 0;
3074071d4279SBram Moolenaar }
3075071d4279SBram Moolenaar #endif
3076071d4279SBram Moolenaar 
3077071d4279SBram Moolenaar /* foldlevelExpr() {{{2 */
3078071d4279SBram Moolenaar /*
3079071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "expr" method.
3080071d4279SBram Moolenaar  * Doesn't use any caching.
3081071d4279SBram Moolenaar  * Returns a level of -1 if the foldlevel depends on surrounding lines.
3082071d4279SBram Moolenaar  */
3083071d4279SBram Moolenaar     static void
3084071d4279SBram Moolenaar foldlevelExpr(flp)
3085071d4279SBram Moolenaar     fline_T	*flp;
3086071d4279SBram Moolenaar {
3087071d4279SBram Moolenaar #ifndef FEAT_EVAL
3088071d4279SBram Moolenaar     flp->start = FALSE;
3089071d4279SBram Moolenaar     flp->lvl = 0;
3090071d4279SBram Moolenaar #else
3091071d4279SBram Moolenaar     win_T	*win;
3092071d4279SBram Moolenaar     int		n;
3093071d4279SBram Moolenaar     int		c;
3094071d4279SBram Moolenaar     linenr_T	lnum = flp->lnum + flp->off;
3095071d4279SBram Moolenaar     int		save_keytyped;
3096071d4279SBram Moolenaar 
3097071d4279SBram Moolenaar     win = curwin;
3098071d4279SBram Moolenaar     curwin = flp->wp;
3099071d4279SBram Moolenaar     curbuf = flp->wp->w_buffer;
3100071d4279SBram Moolenaar     set_vim_var_nr(VV_LNUM, lnum);
3101071d4279SBram Moolenaar 
3102071d4279SBram Moolenaar     flp->start = 0;
3103071d4279SBram Moolenaar     flp->had_end = flp->end;
3104071d4279SBram Moolenaar     flp->end = MAX_LEVEL + 1;
3105071d4279SBram Moolenaar     if (lnum <= 1)
3106071d4279SBram Moolenaar 	flp->lvl = 0;
3107071d4279SBram Moolenaar 
3108071d4279SBram Moolenaar     /* KeyTyped may be reset to 0 when calling a function which invokes
3109071d4279SBram Moolenaar      * do_cmdline().  To make 'foldopen' work correctly restore KeyTyped. */
3110071d4279SBram Moolenaar     save_keytyped = KeyTyped;
3111071d4279SBram Moolenaar     n = eval_foldexpr(flp->wp->w_p_fde, &c);
3112071d4279SBram Moolenaar     KeyTyped = save_keytyped;
3113071d4279SBram Moolenaar 
3114071d4279SBram Moolenaar     switch (c)
3115071d4279SBram Moolenaar     {
3116071d4279SBram Moolenaar 	/* "a1", "a2", .. : add to the fold level */
3117071d4279SBram Moolenaar 	case 'a': if (flp->lvl >= 0)
3118071d4279SBram Moolenaar 		  {
3119071d4279SBram Moolenaar 		      flp->lvl += n;
3120071d4279SBram Moolenaar 		      flp->lvl_next = flp->lvl;
3121071d4279SBram Moolenaar 		  }
3122071d4279SBram Moolenaar 		  flp->start = n;
3123071d4279SBram Moolenaar 		  break;
3124071d4279SBram Moolenaar 
3125071d4279SBram Moolenaar 	/* "s1", "s2", .. : subtract from the fold level */
3126071d4279SBram Moolenaar 	case 's': if (flp->lvl >= 0)
3127071d4279SBram Moolenaar 		  {
3128071d4279SBram Moolenaar 		      if (n > flp->lvl)
3129071d4279SBram Moolenaar 			  flp->lvl_next = 0;
3130071d4279SBram Moolenaar 		      else
3131071d4279SBram Moolenaar 			  flp->lvl_next = flp->lvl - n;
3132071d4279SBram Moolenaar 		      flp->end = flp->lvl_next + 1;
3133071d4279SBram Moolenaar 		  }
3134071d4279SBram Moolenaar 		  break;
3135071d4279SBram Moolenaar 
3136071d4279SBram Moolenaar 	/* ">1", ">2", .. : start a fold with a certain level */
3137071d4279SBram Moolenaar 	case '>': flp->lvl = n;
3138071d4279SBram Moolenaar 		  flp->lvl_next = n;
3139071d4279SBram Moolenaar 		  flp->start = 1;
3140071d4279SBram Moolenaar 		  break;
3141071d4279SBram Moolenaar 
3142071d4279SBram Moolenaar 	/* "<1", "<2", .. : end a fold with a certain level */
3143071d4279SBram Moolenaar 	case '<': flp->lvl_next = n - 1;
3144071d4279SBram Moolenaar 		  flp->end = n;
3145071d4279SBram Moolenaar 		  break;
3146071d4279SBram Moolenaar 
3147071d4279SBram Moolenaar 	/* "=": No change in level */
3148071d4279SBram Moolenaar 	case '=': flp->lvl_next = flp->lvl;
3149071d4279SBram Moolenaar 		  break;
3150071d4279SBram Moolenaar 
3151071d4279SBram Moolenaar 	/* "-1", "0", "1", ..: set fold level */
3152071d4279SBram Moolenaar 	default:  if (n < 0)
3153071d4279SBram Moolenaar 		      /* Use the current level for the next line, so that "a1"
3154071d4279SBram Moolenaar 		       * will work there. */
3155071d4279SBram Moolenaar 		      flp->lvl_next = flp->lvl;
3156071d4279SBram Moolenaar 		  else
3157071d4279SBram Moolenaar 		      flp->lvl_next = n;
3158071d4279SBram Moolenaar 		  flp->lvl = n;
3159071d4279SBram Moolenaar 		  break;
3160071d4279SBram Moolenaar     }
3161071d4279SBram Moolenaar 
3162071d4279SBram Moolenaar     /* If the level is unknown for the first or the last line in the file, use
3163071d4279SBram Moolenaar      * level 0. */
3164071d4279SBram Moolenaar     if (flp->lvl < 0)
3165071d4279SBram Moolenaar     {
3166071d4279SBram Moolenaar 	if (lnum <= 1)
3167071d4279SBram Moolenaar 	{
3168071d4279SBram Moolenaar 	    flp->lvl = 0;
3169071d4279SBram Moolenaar 	    flp->lvl_next = 0;
3170071d4279SBram Moolenaar 	}
3171071d4279SBram Moolenaar 	if (lnum == curbuf->b_ml.ml_line_count)
3172071d4279SBram Moolenaar 	    flp->lvl_next = 0;
3173071d4279SBram Moolenaar     }
3174071d4279SBram Moolenaar 
3175071d4279SBram Moolenaar     curwin = win;
3176071d4279SBram Moolenaar     curbuf = curwin->w_buffer;
3177071d4279SBram Moolenaar #endif
3178071d4279SBram Moolenaar }
3179071d4279SBram Moolenaar 
3180071d4279SBram Moolenaar /* parseMarker() {{{2 */
3181071d4279SBram Moolenaar /*
3182071d4279SBram Moolenaar  * Parse 'foldmarker' and set "foldendmarker", "foldstartmarkerlen" and
3183071d4279SBram Moolenaar  * "foldendmarkerlen".
3184071d4279SBram Moolenaar  * Relies on the option value to have been checked for correctness already.
3185071d4279SBram Moolenaar  */
3186071d4279SBram Moolenaar     static void
3187071d4279SBram Moolenaar parseMarker(wp)
3188071d4279SBram Moolenaar     win_T	*wp;
3189071d4279SBram Moolenaar {
3190071d4279SBram Moolenaar     foldendmarker = vim_strchr(wp->w_p_fmr, ',');
3191071d4279SBram Moolenaar     foldstartmarkerlen = (int)(foldendmarker++ - wp->w_p_fmr);
3192071d4279SBram Moolenaar     foldendmarkerlen = (int)STRLEN(foldendmarker);
3193071d4279SBram Moolenaar }
3194071d4279SBram Moolenaar 
3195071d4279SBram Moolenaar /* foldlevelMarker() {{{2 */
3196071d4279SBram Moolenaar /*
3197071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "marker" method.
3198071d4279SBram Moolenaar  * "foldendmarker", "foldstartmarkerlen" and "foldendmarkerlen" must have been
3199071d4279SBram Moolenaar  * set before calling this.
3200071d4279SBram Moolenaar  * Requires that flp->lvl is set to the fold level of the previous line!
3201071d4279SBram Moolenaar  * Careful: This means you can't call this function twice on the same line.
3202071d4279SBram Moolenaar  * Doesn't use any caching.
3203071d4279SBram Moolenaar  * Sets flp->start when a start marker was found.
3204071d4279SBram Moolenaar  */
3205071d4279SBram Moolenaar     static void
3206071d4279SBram Moolenaar foldlevelMarker(flp)
3207071d4279SBram Moolenaar     fline_T	*flp;
3208071d4279SBram Moolenaar {
3209071d4279SBram Moolenaar     char_u	*startmarker;
3210071d4279SBram Moolenaar     int		cstart;
3211071d4279SBram Moolenaar     int		cend;
3212071d4279SBram Moolenaar     int		start_lvl = flp->lvl;
3213071d4279SBram Moolenaar     char_u	*s;
3214071d4279SBram Moolenaar     int		n;
3215071d4279SBram Moolenaar 
3216071d4279SBram Moolenaar     /* cache a few values for speed */
3217071d4279SBram Moolenaar     startmarker = flp->wp->w_p_fmr;
3218071d4279SBram Moolenaar     cstart = *startmarker;
3219071d4279SBram Moolenaar     ++startmarker;
3220071d4279SBram Moolenaar     cend = *foldendmarker;
3221071d4279SBram Moolenaar 
3222071d4279SBram Moolenaar     /* Default: no start found, next level is same as current level */
3223071d4279SBram Moolenaar     flp->start = 0;
3224071d4279SBram Moolenaar     flp->lvl_next = flp->lvl;
3225071d4279SBram Moolenaar 
3226071d4279SBram Moolenaar     s = ml_get_buf(flp->wp->w_buffer, flp->lnum + flp->off, FALSE);
3227071d4279SBram Moolenaar     while (*s)
3228071d4279SBram Moolenaar     {
3229071d4279SBram Moolenaar 	if (*s == cstart
3230071d4279SBram Moolenaar 		  && STRNCMP(s + 1, startmarker, foldstartmarkerlen - 1) == 0)
3231071d4279SBram Moolenaar 	{
3232071d4279SBram Moolenaar 	    /* found startmarker: set flp->lvl */
3233071d4279SBram Moolenaar 	    s += foldstartmarkerlen;
3234071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(*s))
3235071d4279SBram Moolenaar 	    {
3236071d4279SBram Moolenaar 		n = atoi((char *)s);
3237071d4279SBram Moolenaar 		if (n > 0)
3238071d4279SBram Moolenaar 		{
3239071d4279SBram Moolenaar 		    flp->lvl = n;
3240071d4279SBram Moolenaar 		    flp->lvl_next = n;
3241071d4279SBram Moolenaar 		    if (n <= start_lvl)
3242071d4279SBram Moolenaar 			flp->start = 1;
3243071d4279SBram Moolenaar 		    else
3244071d4279SBram Moolenaar 			flp->start = n - start_lvl;
3245071d4279SBram Moolenaar 		}
3246071d4279SBram Moolenaar 	    }
3247071d4279SBram Moolenaar 	    else
3248071d4279SBram Moolenaar 	    {
3249071d4279SBram Moolenaar 		++flp->lvl;
3250071d4279SBram Moolenaar 		++flp->lvl_next;
3251071d4279SBram Moolenaar 		++flp->start;
3252071d4279SBram Moolenaar 	    }
3253071d4279SBram Moolenaar 	}
3254071d4279SBram Moolenaar 	else if (*s == cend
3255071d4279SBram Moolenaar 	      && STRNCMP(s + 1, foldendmarker + 1, foldendmarkerlen - 1) == 0)
3256071d4279SBram Moolenaar 	{
3257071d4279SBram Moolenaar 	    /* found endmarker: set flp->lvl_next */
3258071d4279SBram Moolenaar 	    s += foldendmarkerlen;
3259071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(*s))
3260071d4279SBram Moolenaar 	    {
3261071d4279SBram Moolenaar 		n = atoi((char *)s);
3262071d4279SBram Moolenaar 		if (n > 0)
3263071d4279SBram Moolenaar 		{
3264071d4279SBram Moolenaar 		    flp->lvl = n;
3265071d4279SBram Moolenaar 		    flp->lvl_next = n - 1;
3266071d4279SBram Moolenaar 		    /* never start a fold with an end marker */
32673ee0229fSBram Moolenaar 		    if (flp->lvl_next > start_lvl)
32683ee0229fSBram Moolenaar 			flp->lvl_next = start_lvl;
3269071d4279SBram Moolenaar 		}
3270071d4279SBram Moolenaar 	    }
3271071d4279SBram Moolenaar 	    else
3272071d4279SBram Moolenaar 		--flp->lvl_next;
3273071d4279SBram Moolenaar 	}
3274071d4279SBram Moolenaar 	else
32751ef15e30SBram Moolenaar 	    mb_ptr_adv(s);
3276071d4279SBram Moolenaar     }
3277071d4279SBram Moolenaar 
3278071d4279SBram Moolenaar     /* The level can't go negative, must be missing a start marker. */
3279071d4279SBram Moolenaar     if (flp->lvl_next < 0)
3280071d4279SBram Moolenaar 	flp->lvl_next = 0;
3281071d4279SBram Moolenaar }
3282071d4279SBram Moolenaar 
3283071d4279SBram Moolenaar /* foldlevelSyntax() {{{2 */
3284071d4279SBram Moolenaar /*
3285071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "syntax" method.
3286071d4279SBram Moolenaar  * Doesn't use any caching.
3287071d4279SBram Moolenaar  */
3288071d4279SBram Moolenaar     static void
3289071d4279SBram Moolenaar foldlevelSyntax(flp)
3290071d4279SBram Moolenaar     fline_T	*flp;
3291071d4279SBram Moolenaar {
3292071d4279SBram Moolenaar #ifndef FEAT_SYN_HL
3293071d4279SBram Moolenaar     flp->start = 0;
3294071d4279SBram Moolenaar     flp->lvl = 0;
3295071d4279SBram Moolenaar #else
3296071d4279SBram Moolenaar     linenr_T	lnum = flp->lnum + flp->off;
3297071d4279SBram Moolenaar     int		n;
3298071d4279SBram Moolenaar 
3299071d4279SBram Moolenaar     /* Use the maximum fold level at the start of this line and the next. */
3300071d4279SBram Moolenaar     flp->lvl = syn_get_foldlevel(flp->wp, lnum);
3301071d4279SBram Moolenaar     flp->start = 0;
3302071d4279SBram Moolenaar     if (lnum < flp->wp->w_buffer->b_ml.ml_line_count)
3303071d4279SBram Moolenaar     {
3304071d4279SBram Moolenaar 	n = syn_get_foldlevel(flp->wp, lnum + 1);
3305071d4279SBram Moolenaar 	if (n > flp->lvl)
3306071d4279SBram Moolenaar 	{
3307071d4279SBram Moolenaar 	    flp->start = n - flp->lvl;	/* fold(s) start here */
3308071d4279SBram Moolenaar 	    flp->lvl = n;
3309071d4279SBram Moolenaar 	}
3310071d4279SBram Moolenaar     }
3311071d4279SBram Moolenaar #endif
3312071d4279SBram Moolenaar }
3313071d4279SBram Moolenaar 
3314071d4279SBram Moolenaar /* functions for storing the fold state in a View {{{1 */
3315071d4279SBram Moolenaar /* put_folds() {{{2 */
3316071d4279SBram Moolenaar #if defined(FEAT_SESSION) || defined(PROTO)
3317d25c16e2SBram Moolenaar static int put_folds_recurse(FILE *fd, garray_T *gap, linenr_T off);
3318d25c16e2SBram Moolenaar static int put_foldopen_recurse(FILE *fd, win_T *wp, garray_T *gap, linenr_T off);
3319d25c16e2SBram Moolenaar static int put_fold_open_close(FILE *fd, fold_T *fp, linenr_T off);
3320071d4279SBram Moolenaar 
3321071d4279SBram Moolenaar /*
3322071d4279SBram Moolenaar  * Write commands to "fd" to restore the manual folds in window "wp".
3323071d4279SBram Moolenaar  * Return FAIL if writing fails.
3324071d4279SBram Moolenaar  */
3325071d4279SBram Moolenaar     int
3326071d4279SBram Moolenaar put_folds(fd, wp)
3327071d4279SBram Moolenaar     FILE	*fd;
3328071d4279SBram Moolenaar     win_T	*wp;
3329071d4279SBram Moolenaar {
3330071d4279SBram Moolenaar     if (foldmethodIsManual(wp))
3331071d4279SBram Moolenaar     {
3332071d4279SBram Moolenaar 	if (put_line(fd, "silent! normal! zE") == FAIL
3333071d4279SBram Moolenaar 		|| put_folds_recurse(fd, &wp->w_folds, (linenr_T)0) == FAIL)
3334071d4279SBram Moolenaar 	    return FAIL;
3335071d4279SBram Moolenaar     }
3336071d4279SBram Moolenaar 
3337071d4279SBram Moolenaar     /* If some folds are manually opened/closed, need to restore that. */
3338071d4279SBram Moolenaar     if (wp->w_fold_manual)
3339d25ad65aSBram Moolenaar 	return put_foldopen_recurse(fd, wp, &wp->w_folds, (linenr_T)0);
3340071d4279SBram Moolenaar 
3341071d4279SBram Moolenaar     return OK;
3342071d4279SBram Moolenaar }
3343071d4279SBram Moolenaar 
3344071d4279SBram Moolenaar /* put_folds_recurse() {{{2 */
3345071d4279SBram Moolenaar /*
3346071d4279SBram Moolenaar  * Write commands to "fd" to recreate manually created folds.
3347071d4279SBram Moolenaar  * Returns FAIL when writing failed.
3348071d4279SBram Moolenaar  */
3349071d4279SBram Moolenaar     static int
3350071d4279SBram Moolenaar put_folds_recurse(fd, gap, off)
3351071d4279SBram Moolenaar     FILE	*fd;
3352071d4279SBram Moolenaar     garray_T	*gap;
3353071d4279SBram Moolenaar     linenr_T	off;
3354071d4279SBram Moolenaar {
3355071d4279SBram Moolenaar     int		i;
3356071d4279SBram Moolenaar     fold_T	*fp;
3357071d4279SBram Moolenaar 
3358071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
3359071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; i++)
3360071d4279SBram Moolenaar     {
3361071d4279SBram Moolenaar 	/* Do nested folds first, they will be created closed. */
3362071d4279SBram Moolenaar 	if (put_folds_recurse(fd, &fp->fd_nested, off + fp->fd_top) == FAIL)
3363071d4279SBram Moolenaar 	    return FAIL;
3364071d4279SBram Moolenaar 	if (fprintf(fd, "%ld,%ldfold", fp->fd_top + off,
3365071d4279SBram Moolenaar 					fp->fd_top + off + fp->fd_len - 1) < 0
3366071d4279SBram Moolenaar 		|| put_eol(fd) == FAIL)
3367071d4279SBram Moolenaar 	    return FAIL;
3368071d4279SBram Moolenaar 	++fp;
3369071d4279SBram Moolenaar     }
3370071d4279SBram Moolenaar     return OK;
3371071d4279SBram Moolenaar }
3372071d4279SBram Moolenaar 
3373071d4279SBram Moolenaar /* put_foldopen_recurse() {{{2 */
3374071d4279SBram Moolenaar /*
3375071d4279SBram Moolenaar  * Write commands to "fd" to open and close manually opened/closed folds.
3376071d4279SBram Moolenaar  * Returns FAIL when writing failed.
3377071d4279SBram Moolenaar  */
3378071d4279SBram Moolenaar     static int
3379d25ad65aSBram Moolenaar put_foldopen_recurse(fd, wp, gap, off)
3380071d4279SBram Moolenaar     FILE	*fd;
3381d25ad65aSBram Moolenaar     win_T	*wp;
3382071d4279SBram Moolenaar     garray_T	*gap;
3383071d4279SBram Moolenaar     linenr_T	off;
3384071d4279SBram Moolenaar {
3385071d4279SBram Moolenaar     int		i;
3386d25ad65aSBram Moolenaar     int		level;
3387071d4279SBram Moolenaar     fold_T	*fp;
3388071d4279SBram Moolenaar 
3389071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
3390071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; i++)
3391071d4279SBram Moolenaar     {
3392071d4279SBram Moolenaar 	if (fp->fd_flags != FD_LEVEL)
3393071d4279SBram Moolenaar 	{
3394071d4279SBram Moolenaar 	    if (fp->fd_nested.ga_len > 0)
3395071d4279SBram Moolenaar 	    {
3396d25ad65aSBram Moolenaar 		/* open nested folds while this fold is open */
3397071d4279SBram Moolenaar 		if (fprintf(fd, "%ld", fp->fd_top + off) < 0
3398071d4279SBram Moolenaar 			|| put_eol(fd) == FAIL
3399002a4edcSBram Moolenaar 			|| put_line(fd, "normal! zo") == FAIL)
3400071d4279SBram Moolenaar 		    return FAIL;
3401d25ad65aSBram Moolenaar 		if (put_foldopen_recurse(fd, wp, &fp->fd_nested,
3402d25ad65aSBram Moolenaar 							     off + fp->fd_top)
3403071d4279SBram Moolenaar 			== FAIL)
3404071d4279SBram Moolenaar 		    return FAIL;
3405d25ad65aSBram Moolenaar 		/* close the parent when needed */
3406d25ad65aSBram Moolenaar 		if (fp->fd_flags == FD_CLOSED)
3407d25ad65aSBram Moolenaar 		{
3408d25ad65aSBram Moolenaar 		    if (put_fold_open_close(fd, fp, off) == FAIL)
3409d25ad65aSBram Moolenaar 			return FAIL;
3410071d4279SBram Moolenaar 		}
3411d25ad65aSBram Moolenaar 	    }
3412d25ad65aSBram Moolenaar 	    else
3413d25ad65aSBram Moolenaar 	    {
3414d25ad65aSBram Moolenaar 		/* Open or close the leaf according to the window foldlevel.
3415d25ad65aSBram Moolenaar 		 * Do not close a leaf that is already closed, as it will close
3416d25ad65aSBram Moolenaar 		 * the parent. */
3417d25ad65aSBram Moolenaar 		level = foldLevelWin(wp, off + fp->fd_top);
3418d25ad65aSBram Moolenaar 		if ((fp->fd_flags == FD_CLOSED && wp->w_p_fdl >= level)
3419d25ad65aSBram Moolenaar 			|| (fp->fd_flags != FD_CLOSED && wp->w_p_fdl < level))
3420d25ad65aSBram Moolenaar 		if (put_fold_open_close(fd, fp, off) == FAIL)
3421d25ad65aSBram Moolenaar 		    return FAIL;
3422d25ad65aSBram Moolenaar 	    }
3423d25ad65aSBram Moolenaar 	}
3424d25ad65aSBram Moolenaar 	++fp;
3425d25ad65aSBram Moolenaar     }
3426d25ad65aSBram Moolenaar 
3427d25ad65aSBram Moolenaar     return OK;
3428d25ad65aSBram Moolenaar }
3429d25ad65aSBram Moolenaar 
3430d25ad65aSBram Moolenaar /* put_fold_open_close() {{{2 */
3431d25ad65aSBram Moolenaar /*
3432d25ad65aSBram Moolenaar  * Write the open or close command to "fd".
3433d25ad65aSBram Moolenaar  * Returns FAIL when writing failed.
3434d25ad65aSBram Moolenaar  */
3435d25ad65aSBram Moolenaar     static int
3436d25ad65aSBram Moolenaar put_fold_open_close(fd, fp, off)
3437d25ad65aSBram Moolenaar     FILE	*fd;
3438d25ad65aSBram Moolenaar     fold_T	*fp;
3439d25ad65aSBram Moolenaar     linenr_T	off;
3440d25ad65aSBram Moolenaar {
3441071d4279SBram Moolenaar     if (fprintf(fd, "%ld", fp->fd_top + off) < 0
3442071d4279SBram Moolenaar 	    || put_eol(fd) == FAIL
3443002a4edcSBram Moolenaar 	    || fprintf(fd, "normal! z%c",
3444071d4279SBram Moolenaar 			   fp->fd_flags == FD_CLOSED ? 'c' : 'o') < 0
3445071d4279SBram Moolenaar 	    || put_eol(fd) == FAIL)
3446071d4279SBram Moolenaar 	return FAIL;
3447071d4279SBram Moolenaar 
3448071d4279SBram Moolenaar     return OK;
3449071d4279SBram Moolenaar }
3450071d4279SBram Moolenaar #endif /* FEAT_SESSION */
3451071d4279SBram Moolenaar 
3452071d4279SBram Moolenaar /* }}}1 */
3453071d4279SBram Moolenaar #endif /* defined(FEAT_FOLDING) || defined(PROTO) */
3454