xref: /vim-8.2.3635/src/fold.c (revision 64f37d30)
1edf3f97aSBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet:
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 
19217e1b83SBram Moolenaar // local declarations. {{{1
20217e1b83SBram 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 {
29217e1b83SBram Moolenaar     linenr_T	fd_top;		// first line of fold; for nested fold
30217e1b83SBram Moolenaar 				// relative to parent
31217e1b83SBram Moolenaar     linenr_T	fd_len;		// number of lines in the fold
32217e1b83SBram Moolenaar     garray_T	fd_nested;	// array of nested folds
33217e1b83SBram Moolenaar     char	fd_flags;	// see below
34217e1b83SBram Moolenaar     char	fd_small;	// TRUE, FALSE or MAYBE: fold smaller than
35217e1b83SBram Moolenaar 				// 'foldminlines'; MAYBE applies to nested
36217e1b83SBram Moolenaar 				// folds too
37071d4279SBram Moolenaar } fold_T;
38071d4279SBram Moolenaar 
39217e1b83SBram Moolenaar #define FD_OPEN		0	// fold is open (nested ones can be closed)
40217e1b83SBram Moolenaar #define FD_CLOSED	1	// fold is closed
41217e1b83SBram Moolenaar #define FD_LEVEL	2	// depends on 'foldlevel' (nested folds too)
42071d4279SBram Moolenaar 
43217e1b83SBram Moolenaar #define MAX_LEVEL	20	// maximum fold depth
44071d4279SBram Moolenaar 
45217e1b83SBram 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 
87217e1b83SBram Moolenaar // Flags used for "done" argument of setManualFold.
88071d4279SBram Moolenaar #define DONE_NOTHING	0
89217e1b83SBram Moolenaar #define DONE_ACTION	1	// did close or open a fold
90217e1b83SBram 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 
96217e1b83SBram Moolenaar // Exported folding functions. {{{1
97217e1b83SBram Moolenaar // copyFoldingState() {{{2
984033c55eSBram Moolenaar 
99071d4279SBram Moolenaar /*
100071d4279SBram Moolenaar  * Copy that folding state from window "wp_from" to window "wp_to".
101071d4279SBram Moolenaar  */
102071d4279SBram Moolenaar     void
10366f948e9SBram Moolenaar copyFoldingState(win_T *wp_from, win_T *wp_to)
104071d4279SBram Moolenaar {
105071d4279SBram Moolenaar     wp_to->w_fold_manual = wp_from->w_fold_manual;
106071d4279SBram Moolenaar     wp_to->w_foldinvalid = wp_from->w_foldinvalid;
107071d4279SBram Moolenaar     cloneFoldGrowArray(&wp_from->w_folds, &wp_to->w_folds);
108071d4279SBram Moolenaar }
109071d4279SBram Moolenaar 
110217e1b83SBram Moolenaar // hasAnyFolding() {{{2
111071d4279SBram Moolenaar /*
112071d4279SBram Moolenaar  * Return TRUE if there may be folded lines in the current window.
113071d4279SBram Moolenaar  */
114071d4279SBram Moolenaar     int
11566f948e9SBram Moolenaar hasAnyFolding(win_T *win)
116071d4279SBram Moolenaar {
117217e1b83SBram Moolenaar     // very simple now, but can become more complex later
118071d4279SBram Moolenaar     return (win->w_p_fen
119071d4279SBram Moolenaar 	    && (!foldmethodIsManual(win) || win->w_folds.ga_len > 0));
120071d4279SBram Moolenaar }
121071d4279SBram Moolenaar 
122217e1b83SBram Moolenaar // hasFolding() {{{2
123071d4279SBram Moolenaar /*
124071d4279SBram Moolenaar  * Return TRUE if line "lnum" in the current window is part of a closed
125071d4279SBram Moolenaar  * fold.
126071d4279SBram Moolenaar  * When returning TRUE, *firstp and *lastp are set to the first and last
127071d4279SBram Moolenaar  * lnum of the sequence of folded lines (skipped when NULL).
128071d4279SBram Moolenaar  */
129071d4279SBram Moolenaar     int
13066f948e9SBram Moolenaar hasFolding(linenr_T lnum, linenr_T *firstp, linenr_T *lastp)
131071d4279SBram Moolenaar {
132071d4279SBram Moolenaar     return hasFoldingWin(curwin, lnum, firstp, lastp, TRUE, NULL);
133071d4279SBram Moolenaar }
134071d4279SBram Moolenaar 
135217e1b83SBram Moolenaar // hasFoldingWin() {{{2
136071d4279SBram Moolenaar     int
13766f948e9SBram Moolenaar hasFoldingWin(
13866f948e9SBram Moolenaar     win_T	*win,
13966f948e9SBram Moolenaar     linenr_T	lnum,
14066f948e9SBram Moolenaar     linenr_T	*firstp,
14166f948e9SBram Moolenaar     linenr_T	*lastp,
142217e1b83SBram Moolenaar     int		cache,		// when TRUE: use cached values of window
143217e1b83SBram Moolenaar     foldinfo_T	*infop)		// where to store fold info
144071d4279SBram Moolenaar {
145071d4279SBram Moolenaar     int		had_folded = FALSE;
146071d4279SBram Moolenaar     linenr_T	first = 0;
147071d4279SBram Moolenaar     linenr_T	last = 0;
148071d4279SBram Moolenaar     linenr_T	lnum_rel = lnum;
149071d4279SBram Moolenaar     int		x;
150071d4279SBram Moolenaar     fold_T	*fp;
151071d4279SBram Moolenaar     int		level = 0;
152071d4279SBram Moolenaar     int		use_level = FALSE;
153071d4279SBram Moolenaar     int		maybe_small = FALSE;
154071d4279SBram Moolenaar     garray_T	*gap;
155945ec093SBram Moolenaar     int		low_level = 0;
156071d4279SBram Moolenaar 
157071d4279SBram Moolenaar     checkupdate(win);
1580c27cbcaSBram Moolenaar 
159071d4279SBram Moolenaar     /*
160071d4279SBram Moolenaar      * Return quickly when there is no folding at all in this window.
161071d4279SBram Moolenaar      */
162071d4279SBram Moolenaar     if (!hasAnyFolding(win))
163071d4279SBram Moolenaar     {
164071d4279SBram Moolenaar 	if (infop != NULL)
165071d4279SBram Moolenaar 	    infop->fi_level = 0;
166071d4279SBram Moolenaar 	return FALSE;
167071d4279SBram Moolenaar     }
168071d4279SBram Moolenaar 
169071d4279SBram Moolenaar     if (cache)
170071d4279SBram Moolenaar     {
171071d4279SBram Moolenaar 	/*
172071d4279SBram Moolenaar 	 * First look in cached info for displayed lines.  This is probably
173071d4279SBram Moolenaar 	 * the fastest, but it can only be used if the entry is still valid.
174071d4279SBram Moolenaar 	 */
175071d4279SBram Moolenaar 	x = find_wl_entry(win, lnum);
176071d4279SBram Moolenaar 	if (x >= 0)
177071d4279SBram Moolenaar 	{
178071d4279SBram Moolenaar 	    first = win->w_lines[x].wl_lnum;
179071d4279SBram Moolenaar 	    last = win->w_lines[x].wl_lastlnum;
180071d4279SBram Moolenaar 	    had_folded = win->w_lines[x].wl_folded;
181071d4279SBram Moolenaar 	}
182071d4279SBram Moolenaar     }
183071d4279SBram Moolenaar 
184071d4279SBram Moolenaar     if (first == 0)
185071d4279SBram Moolenaar     {
186071d4279SBram Moolenaar 	/*
187071d4279SBram Moolenaar 	 * Recursively search for a fold that contains "lnum".
188071d4279SBram Moolenaar 	 */
189071d4279SBram Moolenaar 	gap = &win->w_folds;
190071d4279SBram Moolenaar 	for (;;)
191071d4279SBram Moolenaar 	{
192071d4279SBram Moolenaar 	    if (!foldFind(gap, lnum_rel, &fp))
193071d4279SBram Moolenaar 		break;
194071d4279SBram Moolenaar 
195217e1b83SBram Moolenaar 	    // Remember lowest level of fold that starts in "lnum".
196071d4279SBram Moolenaar 	    if (lnum_rel == fp->fd_top && low_level == 0)
197071d4279SBram Moolenaar 		low_level = level + 1;
198071d4279SBram Moolenaar 
199071d4279SBram Moolenaar 	    first += fp->fd_top;
200071d4279SBram Moolenaar 	    last += fp->fd_top;
201071d4279SBram Moolenaar 
202217e1b83SBram Moolenaar 	    // is this fold closed?
203071d4279SBram Moolenaar 	    had_folded = check_closed(win, fp, &use_level, level,
204071d4279SBram Moolenaar 					       &maybe_small, lnum - lnum_rel);
205071d4279SBram Moolenaar 	    if (had_folded)
206071d4279SBram Moolenaar 	    {
207217e1b83SBram Moolenaar 		// Fold closed: Set last and quit loop.
208071d4279SBram Moolenaar 		last += fp->fd_len - 1;
209071d4279SBram Moolenaar 		break;
210071d4279SBram Moolenaar 	    }
211071d4279SBram Moolenaar 
212217e1b83SBram Moolenaar 	    // Fold found, but it's open: Check nested folds.  Line number is
213217e1b83SBram Moolenaar 	    // relative to containing fold.
214071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
215071d4279SBram Moolenaar 	    lnum_rel -= fp->fd_top;
216071d4279SBram Moolenaar 	    ++level;
217071d4279SBram Moolenaar 	}
218071d4279SBram Moolenaar     }
219071d4279SBram Moolenaar 
220071d4279SBram Moolenaar     if (!had_folded)
221071d4279SBram Moolenaar     {
222071d4279SBram Moolenaar 	if (infop != NULL)
223071d4279SBram Moolenaar 	{
224071d4279SBram Moolenaar 	    infop->fi_level = level;
225071d4279SBram Moolenaar 	    infop->fi_lnum = lnum - lnum_rel;
226071d4279SBram Moolenaar 	    infop->fi_low_level = low_level == 0 ? level : low_level;
227071d4279SBram Moolenaar 	}
228071d4279SBram Moolenaar 	return FALSE;
229071d4279SBram Moolenaar     }
230071d4279SBram Moolenaar 
23105b20fb1SBram Moolenaar     if (last > win->w_buffer->b_ml.ml_line_count)
23205b20fb1SBram Moolenaar 	last = win->w_buffer->b_ml.ml_line_count;
233071d4279SBram Moolenaar     if (lastp != NULL)
234071d4279SBram Moolenaar 	*lastp = last;
235071d4279SBram Moolenaar     if (firstp != NULL)
236071d4279SBram Moolenaar 	*firstp = first;
237071d4279SBram Moolenaar     if (infop != NULL)
238071d4279SBram Moolenaar     {
239071d4279SBram Moolenaar 	infop->fi_level = level + 1;
240071d4279SBram Moolenaar 	infop->fi_lnum = first;
241071d4279SBram Moolenaar 	infop->fi_low_level = low_level == 0 ? level + 1 : low_level;
242071d4279SBram Moolenaar     }
243071d4279SBram Moolenaar     return TRUE;
244071d4279SBram Moolenaar }
245071d4279SBram Moolenaar 
246217e1b83SBram Moolenaar // foldLevel() {{{2
247db022f3fSBram Moolenaar #ifdef FEAT_EVAL
248071d4279SBram Moolenaar /*
249071d4279SBram Moolenaar  * Return fold level at line number "lnum" in the current window.
250071d4279SBram Moolenaar  */
251db022f3fSBram Moolenaar     static int
25266f948e9SBram Moolenaar foldLevel(linenr_T lnum)
253071d4279SBram Moolenaar {
254217e1b83SBram Moolenaar     // While updating the folds lines between invalid_top and invalid_bot have
255217e1b83SBram Moolenaar     // an undefined fold level.  Otherwise update the folds first.
256071d4279SBram Moolenaar     if (invalid_top == (linenr_T)0)
257071d4279SBram Moolenaar 	checkupdate(curwin);
258071d4279SBram Moolenaar     else if (lnum == prev_lnum && prev_lnum_lvl >= 0)
259071d4279SBram Moolenaar 	return prev_lnum_lvl;
260071d4279SBram Moolenaar     else if (lnum >= invalid_top && lnum <= invalid_bot)
261071d4279SBram Moolenaar 	return -1;
262071d4279SBram Moolenaar 
263217e1b83SBram Moolenaar     // Return quickly when there is no folding at all in this window.
264071d4279SBram Moolenaar     if (!hasAnyFolding(curwin))
265071d4279SBram Moolenaar 	return 0;
266071d4279SBram Moolenaar 
267071d4279SBram Moolenaar     return foldLevelWin(curwin, lnum);
268071d4279SBram Moolenaar }
269db022f3fSBram Moolenaar #endif
270071d4279SBram Moolenaar 
271217e1b83SBram Moolenaar // lineFolded()	{{{2
272071d4279SBram Moolenaar /*
273071d4279SBram Moolenaar  * Low level function to check if a line is folded.  Doesn't use any caching.
274071d4279SBram Moolenaar  * Return TRUE if line is folded.
275071d4279SBram Moolenaar  * Return FALSE if line is not folded.
276071d4279SBram Moolenaar  * Return MAYBE if the line is folded when next to a folded line.
277071d4279SBram Moolenaar  */
278071d4279SBram Moolenaar     int
27966f948e9SBram Moolenaar lineFolded(win_T *win, linenr_T lnum)
280071d4279SBram Moolenaar {
281071d4279SBram Moolenaar     return foldedCount(win, lnum, NULL) != 0;
282071d4279SBram Moolenaar }
283071d4279SBram Moolenaar 
284217e1b83SBram Moolenaar // foldedCount() {{{2
285071d4279SBram Moolenaar /*
286071d4279SBram Moolenaar  * Count the number of lines that are folded at line number "lnum".
287071d4279SBram Moolenaar  * Normally "lnum" is the first line of a possible fold, and the returned
288071d4279SBram Moolenaar  * number is the number of lines in the fold.
289071d4279SBram Moolenaar  * Doesn't use caching from the displayed window.
290071d4279SBram Moolenaar  * Returns number of folded lines from "lnum", or 0 if line is not folded.
291071d4279SBram Moolenaar  * When "infop" is not NULL, fills *infop with the fold level info.
292071d4279SBram Moolenaar  */
293071d4279SBram Moolenaar     long
29466f948e9SBram Moolenaar foldedCount(win_T *win, linenr_T lnum, foldinfo_T *infop)
295071d4279SBram Moolenaar {
296071d4279SBram Moolenaar     linenr_T	last;
297071d4279SBram Moolenaar 
298071d4279SBram Moolenaar     if (hasFoldingWin(win, lnum, NULL, &last, FALSE, infop))
299071d4279SBram Moolenaar 	return (long)(last - lnum + 1);
300071d4279SBram Moolenaar     return 0;
301071d4279SBram Moolenaar }
302071d4279SBram Moolenaar 
303217e1b83SBram Moolenaar // foldmethodIsManual() {{{2
304071d4279SBram Moolenaar /*
305071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "manual"
306071d4279SBram Moolenaar  */
307071d4279SBram Moolenaar     int
30866f948e9SBram Moolenaar foldmethodIsManual(win_T *wp)
309071d4279SBram Moolenaar {
310071d4279SBram Moolenaar     return (wp->w_p_fdm[3] == 'u');
311071d4279SBram Moolenaar }
312071d4279SBram Moolenaar 
313217e1b83SBram Moolenaar // foldmethodIsIndent() {{{2
314071d4279SBram Moolenaar /*
315071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "indent"
316071d4279SBram Moolenaar  */
317071d4279SBram Moolenaar     int
31866f948e9SBram Moolenaar foldmethodIsIndent(win_T *wp)
319071d4279SBram Moolenaar {
320071d4279SBram Moolenaar     return (wp->w_p_fdm[0] == 'i');
321071d4279SBram Moolenaar }
322071d4279SBram Moolenaar 
323217e1b83SBram Moolenaar // foldmethodIsExpr() {{{2
324071d4279SBram Moolenaar /*
325071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "expr"
326071d4279SBram Moolenaar  */
327071d4279SBram Moolenaar     int
32866f948e9SBram Moolenaar foldmethodIsExpr(win_T *wp)
329071d4279SBram Moolenaar {
330071d4279SBram Moolenaar     return (wp->w_p_fdm[1] == 'x');
331071d4279SBram Moolenaar }
332071d4279SBram Moolenaar 
333217e1b83SBram Moolenaar // foldmethodIsMarker() {{{2
334071d4279SBram Moolenaar /*
335071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "marker"
336071d4279SBram Moolenaar  */
337071d4279SBram Moolenaar     int
33866f948e9SBram Moolenaar foldmethodIsMarker(win_T *wp)
339071d4279SBram Moolenaar {
340071d4279SBram Moolenaar     return (wp->w_p_fdm[2] == 'r');
341071d4279SBram Moolenaar }
342071d4279SBram Moolenaar 
343217e1b83SBram Moolenaar // foldmethodIsSyntax() {{{2
344071d4279SBram Moolenaar /*
345071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "syntax"
346071d4279SBram Moolenaar  */
347071d4279SBram Moolenaar     int
34866f948e9SBram Moolenaar foldmethodIsSyntax(win_T *wp)
349071d4279SBram Moolenaar {
350071d4279SBram Moolenaar     return (wp->w_p_fdm[0] == 's');
351071d4279SBram Moolenaar }
352071d4279SBram Moolenaar 
353217e1b83SBram Moolenaar // foldmethodIsDiff() {{{2
354071d4279SBram Moolenaar /*
355071d4279SBram Moolenaar  * Return TRUE if 'foldmethod' is "diff"
356071d4279SBram Moolenaar  */
357071d4279SBram Moolenaar     int
35866f948e9SBram Moolenaar foldmethodIsDiff(win_T *wp)
359071d4279SBram Moolenaar {
360071d4279SBram Moolenaar     return (wp->w_p_fdm[0] == 'd');
361071d4279SBram Moolenaar }
362071d4279SBram Moolenaar 
363217e1b83SBram Moolenaar // closeFold() {{{2
364071d4279SBram Moolenaar /*
365071d4279SBram Moolenaar  * Close fold for current window at line "lnum".
366071d4279SBram Moolenaar  * Repeat "count" times.
367071d4279SBram Moolenaar  */
368071d4279SBram Moolenaar     void
36966f948e9SBram Moolenaar closeFold(linenr_T lnum, long count)
370071d4279SBram Moolenaar {
371071d4279SBram Moolenaar     setFoldRepeat(lnum, count, FALSE);
372071d4279SBram Moolenaar }
373071d4279SBram Moolenaar 
374217e1b83SBram Moolenaar // closeFoldRecurse() {{{2
375071d4279SBram Moolenaar /*
376071d4279SBram Moolenaar  * Close fold for current window at line "lnum" recursively.
377071d4279SBram Moolenaar  */
378071d4279SBram Moolenaar     void
37966f948e9SBram Moolenaar closeFoldRecurse(linenr_T lnum)
380071d4279SBram Moolenaar {
381071d4279SBram Moolenaar     (void)setManualFold(lnum, FALSE, TRUE, NULL);
382071d4279SBram Moolenaar }
383071d4279SBram Moolenaar 
384217e1b83SBram Moolenaar // opFoldRange() {{{2
385071d4279SBram Moolenaar /*
386071d4279SBram Moolenaar  * Open or Close folds for current window in lines "first" to "last".
387071d4279SBram Moolenaar  * Used for "zo", "zO", "zc" and "zC" in Visual mode.
388071d4279SBram Moolenaar  */
389071d4279SBram Moolenaar     void
39066f948e9SBram Moolenaar opFoldRange(
39166f948e9SBram Moolenaar     linenr_T	first,
39266f948e9SBram Moolenaar     linenr_T	last,
393217e1b83SBram Moolenaar     int		opening,	// TRUE to open, FALSE to close
394217e1b83SBram Moolenaar     int		recurse,	// TRUE to do it recursively
395217e1b83SBram Moolenaar     int		had_visual)	// TRUE when Visual selection used
396071d4279SBram Moolenaar {
397217e1b83SBram Moolenaar     int		done = DONE_NOTHING;	// avoid error messages
398071d4279SBram Moolenaar     linenr_T	lnum;
399071d4279SBram Moolenaar     linenr_T	lnum_next;
400071d4279SBram Moolenaar 
401071d4279SBram Moolenaar     for (lnum = first; lnum <= last; lnum = lnum_next + 1)
402071d4279SBram Moolenaar     {
403071d4279SBram Moolenaar 	lnum_next = lnum;
404217e1b83SBram Moolenaar 	// Opening one level only: next fold to open is after the one going to
405217e1b83SBram Moolenaar 	// be opened.
406071d4279SBram Moolenaar 	if (opening && !recurse)
407071d4279SBram Moolenaar 	    (void)hasFolding(lnum, NULL, &lnum_next);
408071d4279SBram Moolenaar 	(void)setManualFold(lnum, opening, recurse, &done);
409217e1b83SBram Moolenaar 	// Closing one level only: next line to close a fold is after just
410217e1b83SBram Moolenaar 	// closed fold.
411071d4279SBram Moolenaar 	if (!opening && !recurse)
412071d4279SBram Moolenaar 	    (void)hasFolding(lnum, NULL, &lnum_next);
413071d4279SBram Moolenaar     }
414071d4279SBram Moolenaar     if (done == DONE_NOTHING)
415f9e3e09fSBram Moolenaar 	emsg(_(e_nofold));
416217e1b83SBram Moolenaar     // Force a redraw to remove the Visual highlighting.
417071d4279SBram Moolenaar     if (had_visual)
418071d4279SBram Moolenaar 	redraw_curbuf_later(INVERTED);
419071d4279SBram Moolenaar }
420071d4279SBram Moolenaar 
421217e1b83SBram Moolenaar // openFold() {{{2
422071d4279SBram Moolenaar /*
423071d4279SBram Moolenaar  * Open fold for current window at line "lnum".
424071d4279SBram Moolenaar  * Repeat "count" times.
425071d4279SBram Moolenaar  */
426071d4279SBram Moolenaar     void
42766f948e9SBram Moolenaar openFold(linenr_T lnum, long count)
428071d4279SBram Moolenaar {
429071d4279SBram Moolenaar     setFoldRepeat(lnum, count, TRUE);
430071d4279SBram Moolenaar }
431071d4279SBram Moolenaar 
432217e1b83SBram Moolenaar // openFoldRecurse() {{{2
433071d4279SBram Moolenaar /*
434071d4279SBram Moolenaar  * Open fold for current window at line "lnum" recursively.
435071d4279SBram Moolenaar  */
436071d4279SBram Moolenaar     void
43766f948e9SBram Moolenaar openFoldRecurse(linenr_T lnum)
438071d4279SBram Moolenaar {
439071d4279SBram Moolenaar     (void)setManualFold(lnum, TRUE, TRUE, NULL);
440071d4279SBram Moolenaar }
441071d4279SBram Moolenaar 
442217e1b83SBram Moolenaar // foldOpenCursor() {{{2
443071d4279SBram Moolenaar /*
444071d4279SBram Moolenaar  * Open folds until the cursor line is not in a closed fold.
445071d4279SBram Moolenaar  */
446071d4279SBram Moolenaar     void
44766f948e9SBram Moolenaar foldOpenCursor(void)
448071d4279SBram Moolenaar {
449071d4279SBram Moolenaar     int		done;
450071d4279SBram Moolenaar 
451071d4279SBram Moolenaar     checkupdate(curwin);
452071d4279SBram Moolenaar     if (hasAnyFolding(curwin))
453071d4279SBram Moolenaar 	for (;;)
454071d4279SBram Moolenaar 	{
455071d4279SBram Moolenaar 	    done = DONE_NOTHING;
456071d4279SBram Moolenaar 	    (void)setManualFold(curwin->w_cursor.lnum, TRUE, FALSE, &done);
457071d4279SBram Moolenaar 	    if (!(done & DONE_ACTION))
458071d4279SBram Moolenaar 		break;
459071d4279SBram Moolenaar 	}
460071d4279SBram Moolenaar }
461071d4279SBram Moolenaar 
462217e1b83SBram Moolenaar // newFoldLevel() {{{2
463071d4279SBram Moolenaar /*
464071d4279SBram Moolenaar  * Set new foldlevel for current window.
465071d4279SBram Moolenaar  */
466071d4279SBram Moolenaar     void
46766f948e9SBram Moolenaar newFoldLevel(void)
468071d4279SBram Moolenaar {
469071d4279SBram Moolenaar     newFoldLevelWin(curwin);
470071d4279SBram Moolenaar 
471071d4279SBram Moolenaar #ifdef FEAT_DIFF
472071d4279SBram Moolenaar     if (foldmethodIsDiff(curwin) && curwin->w_p_scb)
473071d4279SBram Moolenaar     {
474071d4279SBram Moolenaar 	win_T	    *wp;
475071d4279SBram Moolenaar 
476071d4279SBram Moolenaar 	/*
477071d4279SBram Moolenaar 	 * Set the same foldlevel in other windows in diff mode.
478071d4279SBram Moolenaar 	 */
479071d4279SBram Moolenaar 	FOR_ALL_WINDOWS(wp)
480071d4279SBram Moolenaar 	{
481071d4279SBram Moolenaar 	    if (wp != curwin && foldmethodIsDiff(wp) && wp->w_p_scb)
482071d4279SBram Moolenaar 	    {
483071d4279SBram Moolenaar 		wp->w_p_fdl = curwin->w_p_fdl;
484071d4279SBram Moolenaar 		newFoldLevelWin(wp);
485071d4279SBram Moolenaar 	    }
486071d4279SBram Moolenaar 	}
487071d4279SBram Moolenaar     }
488071d4279SBram Moolenaar #endif
489071d4279SBram Moolenaar }
490071d4279SBram Moolenaar 
491071d4279SBram Moolenaar     static void
49266f948e9SBram Moolenaar newFoldLevelWin(win_T *wp)
493071d4279SBram Moolenaar {
494071d4279SBram Moolenaar     fold_T	*fp;
495071d4279SBram Moolenaar     int		i;
496071d4279SBram Moolenaar 
497071d4279SBram Moolenaar     checkupdate(wp);
498071d4279SBram Moolenaar     if (wp->w_fold_manual)
499071d4279SBram Moolenaar     {
500217e1b83SBram Moolenaar 	// Set all flags for the first level of folds to FD_LEVEL.  Following
501217e1b83SBram Moolenaar 	// manual open/close will then change the flags to FD_OPEN or
502217e1b83SBram Moolenaar 	// FD_CLOSED for those folds that don't use 'foldlevel'.
503071d4279SBram Moolenaar 	fp = (fold_T *)wp->w_folds.ga_data;
504071d4279SBram Moolenaar 	for (i = 0; i < wp->w_folds.ga_len; ++i)
505071d4279SBram Moolenaar 	    fp[i].fd_flags = FD_LEVEL;
506071d4279SBram Moolenaar 	wp->w_fold_manual = FALSE;
507071d4279SBram Moolenaar     }
508071d4279SBram Moolenaar     changed_window_setting_win(wp);
509071d4279SBram Moolenaar }
510071d4279SBram Moolenaar 
511217e1b83SBram Moolenaar // foldCheckClose() {{{2
512071d4279SBram Moolenaar /*
513071d4279SBram Moolenaar  * Apply 'foldlevel' to all folds that don't contain the cursor.
514071d4279SBram Moolenaar  */
515071d4279SBram Moolenaar     void
51666f948e9SBram Moolenaar foldCheckClose(void)
517071d4279SBram Moolenaar {
518217e1b83SBram Moolenaar     if (*p_fcl != NUL)	// can only be "all" right now
519071d4279SBram Moolenaar     {
520071d4279SBram Moolenaar 	checkupdate(curwin);
521071d4279SBram Moolenaar 	if (checkCloseRec(&curwin->w_folds, curwin->w_cursor.lnum,
522071d4279SBram Moolenaar 							(int)curwin->w_p_fdl))
523071d4279SBram Moolenaar 	    changed_window_setting();
524071d4279SBram Moolenaar     }
525071d4279SBram Moolenaar }
526071d4279SBram Moolenaar 
527217e1b83SBram Moolenaar // checkCloseRec() {{{2
528071d4279SBram Moolenaar     static int
52966f948e9SBram Moolenaar checkCloseRec(garray_T *gap, linenr_T lnum, int level)
530071d4279SBram Moolenaar {
531071d4279SBram Moolenaar     fold_T	*fp;
532071d4279SBram Moolenaar     int		retval = FALSE;
533071d4279SBram Moolenaar     int		i;
534071d4279SBram Moolenaar 
535071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
536071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
537071d4279SBram Moolenaar     {
538217e1b83SBram Moolenaar 	// Only manually opened folds may need to be closed.
539071d4279SBram Moolenaar 	if (fp[i].fd_flags == FD_OPEN)
540071d4279SBram Moolenaar 	{
541071d4279SBram Moolenaar 	    if (level <= 0 && (lnum < fp[i].fd_top
542071d4279SBram Moolenaar 				      || lnum >= fp[i].fd_top + fp[i].fd_len))
543071d4279SBram Moolenaar 	    {
544071d4279SBram Moolenaar 		fp[i].fd_flags = FD_LEVEL;
545071d4279SBram Moolenaar 		retval = TRUE;
546071d4279SBram Moolenaar 	    }
547071d4279SBram Moolenaar 	    else
548071d4279SBram Moolenaar 		retval |= checkCloseRec(&fp[i].fd_nested, lnum - fp[i].fd_top,
549071d4279SBram Moolenaar 								   level - 1);
550071d4279SBram Moolenaar 	}
551071d4279SBram Moolenaar     }
552071d4279SBram Moolenaar     return retval;
553071d4279SBram Moolenaar }
554071d4279SBram Moolenaar 
555217e1b83SBram Moolenaar // foldCreateAllowed() {{{2
556071d4279SBram Moolenaar /*
557071d4279SBram Moolenaar  * Return TRUE if it's allowed to manually create or delete a fold.
558071d4279SBram Moolenaar  * Give an error message and return FALSE if not.
559071d4279SBram Moolenaar  */
560071d4279SBram Moolenaar     int
56166f948e9SBram Moolenaar foldManualAllowed(int create)
562071d4279SBram Moolenaar {
563071d4279SBram Moolenaar     if (foldmethodIsManual(curwin) || foldmethodIsMarker(curwin))
564071d4279SBram Moolenaar 	return TRUE;
565071d4279SBram Moolenaar     if (create)
566f9e3e09fSBram Moolenaar 	emsg(_("E350: Cannot create fold with current 'foldmethod'"));
567071d4279SBram Moolenaar     else
568f9e3e09fSBram Moolenaar 	emsg(_("E351: Cannot delete fold with current 'foldmethod'"));
569071d4279SBram Moolenaar     return FALSE;
570071d4279SBram Moolenaar }
571071d4279SBram Moolenaar 
572217e1b83SBram Moolenaar // foldCreate() {{{2
573071d4279SBram Moolenaar /*
574071d4279SBram Moolenaar  * Create a fold from line "start" to line "end" (inclusive) in the current
575071d4279SBram Moolenaar  * window.
576071d4279SBram Moolenaar  */
577071d4279SBram Moolenaar     void
57866f948e9SBram Moolenaar foldCreate(linenr_T start, linenr_T end)
579071d4279SBram Moolenaar {
580071d4279SBram Moolenaar     fold_T	*fp;
581071d4279SBram Moolenaar     garray_T	*gap;
582071d4279SBram Moolenaar     garray_T	fold_ga;
583071d4279SBram Moolenaar     int		i, j;
584071d4279SBram Moolenaar     int		cont;
585071d4279SBram Moolenaar     int		use_level = FALSE;
586071d4279SBram Moolenaar     int		closed = FALSE;
587071d4279SBram Moolenaar     int		level = 0;
588071d4279SBram Moolenaar     linenr_T	start_rel = start;
589071d4279SBram Moolenaar     linenr_T	end_rel = end;
590071d4279SBram Moolenaar 
591071d4279SBram Moolenaar     if (start > end)
592071d4279SBram Moolenaar     {
593217e1b83SBram Moolenaar 	// reverse the range
594071d4279SBram Moolenaar 	end = start_rel;
595071d4279SBram Moolenaar 	start = end_rel;
596071d4279SBram Moolenaar 	start_rel = start;
597071d4279SBram Moolenaar 	end_rel = end;
598071d4279SBram Moolenaar     }
599071d4279SBram Moolenaar 
600217e1b83SBram Moolenaar     // When 'foldmethod' is "marker" add markers, which creates the folds.
601071d4279SBram Moolenaar     if (foldmethodIsMarker(curwin))
602071d4279SBram Moolenaar     {
603071d4279SBram Moolenaar 	foldCreateMarkers(start, end);
604071d4279SBram Moolenaar 	return;
605071d4279SBram Moolenaar     }
606071d4279SBram Moolenaar 
607071d4279SBram Moolenaar     checkupdate(curwin);
608071d4279SBram Moolenaar 
609217e1b83SBram Moolenaar     // Find the place to insert the new fold.
610071d4279SBram Moolenaar     gap = &curwin->w_folds;
611071d4279SBram Moolenaar     for (;;)
612071d4279SBram Moolenaar     {
613071d4279SBram Moolenaar 	if (!foldFind(gap, start_rel, &fp))
614071d4279SBram Moolenaar 	    break;
615071d4279SBram Moolenaar 	if (fp->fd_top + fp->fd_len > end_rel)
616071d4279SBram Moolenaar 	{
617217e1b83SBram Moolenaar 	    // New fold is completely inside this fold: Go one level deeper.
618071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
619071d4279SBram Moolenaar 	    start_rel -= fp->fd_top;
620071d4279SBram Moolenaar 	    end_rel -= fp->fd_top;
621071d4279SBram Moolenaar 	    if (use_level || fp->fd_flags == FD_LEVEL)
622071d4279SBram Moolenaar 	    {
623071d4279SBram Moolenaar 		use_level = TRUE;
624071d4279SBram Moolenaar 		if (level >= curwin->w_p_fdl)
625071d4279SBram Moolenaar 		    closed = TRUE;
626071d4279SBram Moolenaar 	    }
627071d4279SBram Moolenaar 	    else if (fp->fd_flags == FD_CLOSED)
628071d4279SBram Moolenaar 		closed = TRUE;
629071d4279SBram Moolenaar 	    ++level;
630071d4279SBram Moolenaar 	}
631071d4279SBram Moolenaar 	else
632071d4279SBram Moolenaar 	{
633217e1b83SBram Moolenaar 	    // This fold and new fold overlap: Insert here and move some folds
634217e1b83SBram Moolenaar 	    // inside the new fold.
635071d4279SBram Moolenaar 	    break;
636071d4279SBram Moolenaar 	}
637071d4279SBram Moolenaar     }
638071d4279SBram Moolenaar 
639071d4279SBram Moolenaar     i = (int)(fp - (fold_T *)gap->ga_data);
640071d4279SBram Moolenaar     if (ga_grow(gap, 1) == OK)
641071d4279SBram Moolenaar     {
642071d4279SBram Moolenaar 	fp = (fold_T *)gap->ga_data + i;
643071d4279SBram Moolenaar 	ga_init2(&fold_ga, (int)sizeof(fold_T), 10);
644071d4279SBram Moolenaar 
645217e1b83SBram Moolenaar 	// Count number of folds that will be contained in the new fold.
646071d4279SBram Moolenaar 	for (cont = 0; i + cont < gap->ga_len; ++cont)
647071d4279SBram Moolenaar 	    if (fp[cont].fd_top > end_rel)
648071d4279SBram Moolenaar 		break;
649071d4279SBram Moolenaar 	if (cont > 0 && ga_grow(&fold_ga, cont) == OK)
650071d4279SBram Moolenaar 	{
651217e1b83SBram Moolenaar 	    // If the first fold starts before the new fold, let the new fold
652217e1b83SBram Moolenaar 	    // start there.  Otherwise the existing fold would change.
653071d4279SBram Moolenaar 	    if (start_rel > fp->fd_top)
654071d4279SBram Moolenaar 		start_rel = fp->fd_top;
655071d4279SBram Moolenaar 
656217e1b83SBram Moolenaar 	    // When last contained fold isn't completely contained, adjust end
657217e1b83SBram Moolenaar 	    // of new fold.
658071d4279SBram Moolenaar 	    if (end_rel < fp[cont - 1].fd_top + fp[cont - 1].fd_len - 1)
659071d4279SBram Moolenaar 		end_rel = fp[cont - 1].fd_top + fp[cont - 1].fd_len - 1;
660217e1b83SBram Moolenaar 	    // Move contained folds to inside new fold.
661071d4279SBram Moolenaar 	    mch_memmove(fold_ga.ga_data, fp, sizeof(fold_T) * cont);
662071d4279SBram Moolenaar 	    fold_ga.ga_len += cont;
663071d4279SBram Moolenaar 	    i += cont;
664071d4279SBram Moolenaar 
665217e1b83SBram Moolenaar 	    // Adjust line numbers in contained folds to be relative to the
666217e1b83SBram Moolenaar 	    // new fold.
667071d4279SBram Moolenaar 	    for (j = 0; j < cont; ++j)
668071d4279SBram Moolenaar 		((fold_T *)fold_ga.ga_data)[j].fd_top -= start_rel;
669071d4279SBram Moolenaar 	}
670217e1b83SBram Moolenaar 	// Move remaining entries to after the new fold.
671071d4279SBram Moolenaar 	if (i < gap->ga_len)
672071d4279SBram Moolenaar 	    mch_memmove(fp + 1, (fold_T *)gap->ga_data + i,
673071d4279SBram Moolenaar 				     sizeof(fold_T) * (gap->ga_len - i));
674071d4279SBram Moolenaar 	gap->ga_len = gap->ga_len + 1 - cont;
675071d4279SBram Moolenaar 
676217e1b83SBram Moolenaar 	// insert new fold
677071d4279SBram Moolenaar 	fp->fd_nested = fold_ga;
678071d4279SBram Moolenaar 	fp->fd_top = start_rel;
679071d4279SBram Moolenaar 	fp->fd_len = end_rel - start_rel + 1;
680071d4279SBram Moolenaar 
681217e1b83SBram Moolenaar 	// We want the new fold to be closed.  If it would remain open because
682217e1b83SBram Moolenaar 	// of using 'foldlevel', need to adjust fd_flags of containing folds.
683071d4279SBram Moolenaar 	if (use_level && !closed && level < curwin->w_p_fdl)
684071d4279SBram Moolenaar 	    closeFold(start, 1L);
685071d4279SBram Moolenaar 	if (!use_level)
686071d4279SBram Moolenaar 	    curwin->w_fold_manual = TRUE;
687071d4279SBram Moolenaar 	fp->fd_flags = FD_CLOSED;
688071d4279SBram Moolenaar 	fp->fd_small = MAYBE;
689071d4279SBram Moolenaar 
690217e1b83SBram Moolenaar 	// redraw
691071d4279SBram Moolenaar 	changed_window_setting();
692071d4279SBram Moolenaar     }
693071d4279SBram Moolenaar }
694071d4279SBram Moolenaar 
695217e1b83SBram Moolenaar // deleteFold() {{{2
696071d4279SBram Moolenaar /*
697071d4279SBram Moolenaar  * Delete a fold at line "start" in the current window.
698071d4279SBram Moolenaar  * When "end" is not 0, delete all folds from "start" to "end".
699071d4279SBram Moolenaar  * When "recursive" is TRUE delete recursively.
700071d4279SBram Moolenaar  */
701071d4279SBram Moolenaar     void
70266f948e9SBram Moolenaar deleteFold(
70366f948e9SBram Moolenaar     linenr_T	start,
70466f948e9SBram Moolenaar     linenr_T	end,
70566f948e9SBram Moolenaar     int		recursive,
706217e1b83SBram Moolenaar     int		had_visual)	// TRUE when Visual selection used
707071d4279SBram Moolenaar {
708071d4279SBram Moolenaar     garray_T	*gap;
709071d4279SBram Moolenaar     fold_T	*fp;
710071d4279SBram Moolenaar     garray_T	*found_ga;
711071d4279SBram Moolenaar     fold_T	*found_fp = NULL;
712071d4279SBram Moolenaar     linenr_T	found_off = 0;
713238b8e20SBram Moolenaar     int		use_level;
714071d4279SBram Moolenaar     int		maybe_small = FALSE;
715071d4279SBram Moolenaar     int		level = 0;
716071d4279SBram Moolenaar     linenr_T	lnum = start;
717071d4279SBram Moolenaar     linenr_T	lnum_off;
718071d4279SBram Moolenaar     int		did_one = FALSE;
719071d4279SBram Moolenaar     linenr_T	first_lnum = MAXLNUM;
720071d4279SBram Moolenaar     linenr_T	last_lnum = 0;
721071d4279SBram Moolenaar 
722071d4279SBram Moolenaar     checkupdate(curwin);
723071d4279SBram Moolenaar 
724071d4279SBram Moolenaar     while (lnum <= end)
725071d4279SBram Moolenaar     {
726217e1b83SBram Moolenaar 	// Find the deepest fold for "start".
727071d4279SBram Moolenaar 	gap = &curwin->w_folds;
728071d4279SBram Moolenaar 	found_ga = NULL;
729071d4279SBram Moolenaar 	lnum_off = 0;
730238b8e20SBram Moolenaar 	use_level = FALSE;
731071d4279SBram Moolenaar 	for (;;)
732071d4279SBram Moolenaar 	{
733071d4279SBram Moolenaar 	    if (!foldFind(gap, lnum - lnum_off, &fp))
734071d4279SBram Moolenaar 		break;
735217e1b83SBram Moolenaar 	    // lnum is inside this fold, remember info
736071d4279SBram Moolenaar 	    found_ga = gap;
737071d4279SBram Moolenaar 	    found_fp = fp;
738071d4279SBram Moolenaar 	    found_off = lnum_off;
739071d4279SBram Moolenaar 
740217e1b83SBram Moolenaar 	    // if "lnum" is folded, don't check nesting
741071d4279SBram Moolenaar 	    if (check_closed(curwin, fp, &use_level, level,
742071d4279SBram Moolenaar 						      &maybe_small, lnum_off))
743071d4279SBram Moolenaar 		break;
744071d4279SBram Moolenaar 
745217e1b83SBram Moolenaar 	    // check nested folds
746071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
747071d4279SBram Moolenaar 	    lnum_off += fp->fd_top;
748071d4279SBram Moolenaar 	    ++level;
749071d4279SBram Moolenaar 	}
750071d4279SBram Moolenaar 	if (found_ga == NULL)
751071d4279SBram Moolenaar 	{
752071d4279SBram Moolenaar 	    ++lnum;
753071d4279SBram Moolenaar 	}
754071d4279SBram Moolenaar 	else
755071d4279SBram Moolenaar 	{
756071d4279SBram Moolenaar 	    lnum = found_fp->fd_top + found_fp->fd_len + found_off;
757071d4279SBram Moolenaar 
758071d4279SBram Moolenaar 	    if (foldmethodIsManual(curwin))
759071d4279SBram Moolenaar 		deleteFoldEntry(found_ga,
760071d4279SBram Moolenaar 		    (int)(found_fp - (fold_T *)found_ga->ga_data), recursive);
761071d4279SBram Moolenaar 	    else
762071d4279SBram Moolenaar 	    {
763238b8e20SBram Moolenaar 		if (first_lnum > found_fp->fd_top + found_off)
764238b8e20SBram Moolenaar 		    first_lnum = found_fp->fd_top + found_off;
765238b8e20SBram Moolenaar 		if (last_lnum < lnum)
766071d4279SBram Moolenaar 		    last_lnum = lnum;
767238b8e20SBram Moolenaar 		if (!did_one)
768071d4279SBram Moolenaar 		    parseMarker(curwin);
769071d4279SBram Moolenaar 		deleteFoldMarkers(found_fp, recursive, found_off);
770071d4279SBram Moolenaar 	    }
771238b8e20SBram Moolenaar 	    did_one = TRUE;
772071d4279SBram Moolenaar 
773217e1b83SBram Moolenaar 	    // redraw window
774071d4279SBram Moolenaar 	    changed_window_setting();
775071d4279SBram Moolenaar 	}
776071d4279SBram Moolenaar     }
777071d4279SBram Moolenaar     if (!did_one)
778071d4279SBram Moolenaar     {
779f9e3e09fSBram Moolenaar 	emsg(_(e_nofold));
780217e1b83SBram Moolenaar 	// Force a redraw to remove the Visual highlighting.
781071d4279SBram Moolenaar 	if (had_visual)
782071d4279SBram Moolenaar 	    redraw_curbuf_later(INVERTED);
783071d4279SBram Moolenaar     }
784238b8e20SBram Moolenaar     else
785217e1b83SBram Moolenaar 	// Deleting markers may make cursor column invalid.
786238b8e20SBram Moolenaar 	check_cursor_col();
787238b8e20SBram Moolenaar 
788071d4279SBram Moolenaar     if (last_lnum > 0)
789071d4279SBram Moolenaar 	changed_lines(first_lnum, (colnr_T)0, last_lnum, 0L);
790071d4279SBram Moolenaar }
791071d4279SBram Moolenaar 
792217e1b83SBram Moolenaar // clearFolding() {{{2
793071d4279SBram Moolenaar /*
794071d4279SBram Moolenaar  * Remove all folding for window "win".
795071d4279SBram Moolenaar  */
796071d4279SBram Moolenaar     void
79766f948e9SBram Moolenaar clearFolding(win_T *win)
798071d4279SBram Moolenaar {
799071d4279SBram Moolenaar     deleteFoldRecurse(&win->w_folds);
800071d4279SBram Moolenaar     win->w_foldinvalid = FALSE;
801071d4279SBram Moolenaar }
802071d4279SBram Moolenaar 
803217e1b83SBram Moolenaar // foldUpdate() {{{2
804071d4279SBram Moolenaar /*
805071d4279SBram Moolenaar  * Update folds for changes in the buffer of a window.
806071d4279SBram Moolenaar  * Note that inserted/deleted lines must have already been taken care of by
807071d4279SBram Moolenaar  * calling foldMarkAdjust().
808071d4279SBram Moolenaar  * The changes in lines from top to bot (inclusive).
809071d4279SBram Moolenaar  */
810071d4279SBram Moolenaar     void
81166f948e9SBram Moolenaar foldUpdate(win_T *wp, linenr_T top, linenr_T bot)
812071d4279SBram Moolenaar {
813071d4279SBram Moolenaar     fold_T	*fp;
814071d4279SBram Moolenaar 
815429fcfbfSBram Moolenaar     if (disable_fold_update > 0)
816429fcfbfSBram Moolenaar 	return;
8174f57eefeSBram Moolenaar #ifdef FEAT_DIFF
8184f57eefeSBram Moolenaar     if (need_diff_redraw)
8194f57eefeSBram Moolenaar 	// will update later
8204f57eefeSBram Moolenaar 	return;
8214f57eefeSBram Moolenaar #endif
822429fcfbfSBram Moolenaar 
823*64f37d30SBram Moolenaar     if (wp->w_folds.ga_len > 0)
824*64f37d30SBram Moolenaar     {
825217e1b83SBram Moolenaar 	// Mark all folds from top to bot as maybe-small.
826d5f6933dSBram Moolenaar 	(void)foldFind(&wp->w_folds, top, &fp);
827d5f6933dSBram Moolenaar 	while (fp < (fold_T *)wp->w_folds.ga_data + wp->w_folds.ga_len
828071d4279SBram Moolenaar 		&& fp->fd_top < bot)
829071d4279SBram Moolenaar 	{
830071d4279SBram Moolenaar 	    fp->fd_small = MAYBE;
831071d4279SBram Moolenaar 	    ++fp;
832071d4279SBram Moolenaar 	}
833*64f37d30SBram Moolenaar     }
834071d4279SBram Moolenaar 
835071d4279SBram Moolenaar     if (foldmethodIsIndent(wp)
836071d4279SBram Moolenaar 	    || foldmethodIsExpr(wp)
837071d4279SBram Moolenaar 	    || foldmethodIsMarker(wp)
838071d4279SBram Moolenaar #ifdef FEAT_DIFF
839071d4279SBram Moolenaar 	    || foldmethodIsDiff(wp)
840071d4279SBram Moolenaar #endif
841071d4279SBram Moolenaar 	    || foldmethodIsSyntax(wp))
842fa6d5afdSBram Moolenaar     {
843fa6d5afdSBram Moolenaar 	int save_got_int = got_int;
844fa6d5afdSBram Moolenaar 
845217e1b83SBram Moolenaar 	// reset got_int here, otherwise it won't work
846fa6d5afdSBram Moolenaar 	got_int = FALSE;
847071d4279SBram Moolenaar 	foldUpdateIEMS(wp, top, bot);
848fa6d5afdSBram Moolenaar 	got_int |= save_got_int;
849fa6d5afdSBram Moolenaar     }
850071d4279SBram Moolenaar }
851071d4279SBram Moolenaar 
852217e1b83SBram Moolenaar // foldUpdateAll() {{{2
853071d4279SBram Moolenaar /*
854071d4279SBram Moolenaar  * Update all lines in a window for folding.
855071d4279SBram Moolenaar  * Used when a fold setting changes or after reloading the buffer.
856071d4279SBram Moolenaar  * The actual updating is postponed until fold info is used, to avoid doing
857071d4279SBram Moolenaar  * every time a setting is changed or a syntax item is added.
858071d4279SBram Moolenaar  */
859071d4279SBram Moolenaar     void
86066f948e9SBram Moolenaar foldUpdateAll(win_T *win)
861071d4279SBram Moolenaar {
862071d4279SBram Moolenaar     win->w_foldinvalid = TRUE;
863071d4279SBram Moolenaar     redraw_win_later(win, NOT_VALID);
864071d4279SBram Moolenaar }
865071d4279SBram Moolenaar 
866217e1b83SBram Moolenaar // foldMoveTo() {{{2
867071d4279SBram Moolenaar /*
868071d4279SBram Moolenaar  * If "updown" is FALSE: Move to the start or end of the fold.
869071d4279SBram Moolenaar  * If "updown" is TRUE: move to fold at the same level.
870071d4279SBram Moolenaar  * If not moved return FAIL.
871071d4279SBram Moolenaar  */
872071d4279SBram Moolenaar     int
87366f948e9SBram Moolenaar foldMoveTo(
87466f948e9SBram Moolenaar     int		updown,
875217e1b83SBram Moolenaar     int		dir,	    // FORWARD or BACKWARD
87666f948e9SBram Moolenaar     long	count)
877071d4279SBram Moolenaar {
878071d4279SBram Moolenaar     long	n;
879071d4279SBram Moolenaar     int		retval = FAIL;
880071d4279SBram Moolenaar     linenr_T	lnum_off;
881071d4279SBram Moolenaar     linenr_T	lnum_found;
882071d4279SBram Moolenaar     linenr_T	lnum;
883071d4279SBram Moolenaar     int		use_level;
884071d4279SBram Moolenaar     int		maybe_small;
885071d4279SBram Moolenaar     garray_T	*gap;
886071d4279SBram Moolenaar     fold_T	*fp;
887071d4279SBram Moolenaar     int		level;
888071d4279SBram Moolenaar     int		last;
889071d4279SBram Moolenaar 
890293ee4d4SBram Moolenaar     checkupdate(curwin);
891293ee4d4SBram Moolenaar 
892217e1b83SBram Moolenaar     // Repeat "count" times.
893071d4279SBram Moolenaar     for (n = 0; n < count; ++n)
894071d4279SBram Moolenaar     {
895217e1b83SBram Moolenaar 	// Find nested folds.  Stop when a fold is closed.  The deepest fold
896217e1b83SBram Moolenaar 	// that moves the cursor is used.
897071d4279SBram Moolenaar 	lnum_off = 0;
898071d4279SBram Moolenaar 	gap = &curwin->w_folds;
899071d4279SBram Moolenaar 	use_level = FALSE;
900071d4279SBram Moolenaar 	maybe_small = FALSE;
901071d4279SBram Moolenaar 	lnum_found = curwin->w_cursor.lnum;
902071d4279SBram Moolenaar 	level = 0;
903071d4279SBram Moolenaar 	last = FALSE;
904071d4279SBram Moolenaar 	for (;;)
905071d4279SBram Moolenaar 	{
906071d4279SBram Moolenaar 	    if (!foldFind(gap, curwin->w_cursor.lnum - lnum_off, &fp))
907071d4279SBram Moolenaar 	    {
908071d4279SBram Moolenaar 		if (!updown)
909071d4279SBram Moolenaar 		    break;
910071d4279SBram Moolenaar 
911217e1b83SBram Moolenaar 		// When moving up, consider a fold above the cursor; when
912217e1b83SBram Moolenaar 		// moving down consider a fold below the cursor.
913071d4279SBram Moolenaar 		if (dir == FORWARD)
914071d4279SBram Moolenaar 		{
915071d4279SBram Moolenaar 		    if (fp - (fold_T *)gap->ga_data >= gap->ga_len)
916071d4279SBram Moolenaar 			break;
917071d4279SBram Moolenaar 		    --fp;
918071d4279SBram Moolenaar 		}
919071d4279SBram Moolenaar 		else
920071d4279SBram Moolenaar 		{
921071d4279SBram Moolenaar 		    if (fp == (fold_T *)gap->ga_data)
922071d4279SBram Moolenaar 			break;
923071d4279SBram Moolenaar 		}
924217e1b83SBram Moolenaar 		// don't look for contained folds, they will always move
925217e1b83SBram Moolenaar 		// the cursor too far.
926071d4279SBram Moolenaar 		last = TRUE;
927071d4279SBram Moolenaar 	    }
928071d4279SBram Moolenaar 
929071d4279SBram Moolenaar 	    if (!last)
930071d4279SBram Moolenaar 	    {
931217e1b83SBram Moolenaar 		// Check if this fold is closed.
932071d4279SBram Moolenaar 		if (check_closed(curwin, fp, &use_level, level,
933071d4279SBram Moolenaar 						      &maybe_small, lnum_off))
934071d4279SBram Moolenaar 		    last = TRUE;
935071d4279SBram Moolenaar 
936217e1b83SBram Moolenaar 		// "[z" and "]z" stop at closed fold
937071d4279SBram Moolenaar 		if (last && !updown)
938071d4279SBram Moolenaar 		    break;
939071d4279SBram Moolenaar 	    }
940071d4279SBram Moolenaar 
941071d4279SBram Moolenaar 	    if (updown)
942071d4279SBram Moolenaar 	    {
943071d4279SBram Moolenaar 		if (dir == FORWARD)
944071d4279SBram Moolenaar 		{
945217e1b83SBram Moolenaar 		    // to start of next fold if there is one
946071d4279SBram Moolenaar 		    if (fp + 1 - (fold_T *)gap->ga_data < gap->ga_len)
947071d4279SBram Moolenaar 		    {
948071d4279SBram Moolenaar 			lnum = fp[1].fd_top + lnum_off;
949071d4279SBram Moolenaar 			if (lnum > curwin->w_cursor.lnum)
950071d4279SBram Moolenaar 			    lnum_found = lnum;
951071d4279SBram Moolenaar 		    }
952071d4279SBram Moolenaar 		}
953071d4279SBram Moolenaar 		else
954071d4279SBram Moolenaar 		{
955217e1b83SBram Moolenaar 		    // to end of previous fold if there is one
956071d4279SBram Moolenaar 		    if (fp > (fold_T *)gap->ga_data)
957071d4279SBram Moolenaar 		    {
958071d4279SBram Moolenaar 			lnum = fp[-1].fd_top + lnum_off + fp[-1].fd_len - 1;
959071d4279SBram Moolenaar 			if (lnum < curwin->w_cursor.lnum)
960071d4279SBram Moolenaar 			    lnum_found = lnum;
961071d4279SBram Moolenaar 		    }
962071d4279SBram Moolenaar 		}
963071d4279SBram Moolenaar 	    }
964071d4279SBram Moolenaar 	    else
965071d4279SBram Moolenaar 	    {
966217e1b83SBram Moolenaar 		// Open fold found, set cursor to its start/end and then check
967217e1b83SBram Moolenaar 		// nested folds.
968071d4279SBram Moolenaar 		if (dir == FORWARD)
969071d4279SBram Moolenaar 		{
970071d4279SBram Moolenaar 		    lnum = fp->fd_top + lnum_off + fp->fd_len - 1;
971071d4279SBram Moolenaar 		    if (lnum > curwin->w_cursor.lnum)
972071d4279SBram Moolenaar 			lnum_found = lnum;
973071d4279SBram Moolenaar 		}
974071d4279SBram Moolenaar 		else
975071d4279SBram Moolenaar 		{
976071d4279SBram Moolenaar 		    lnum = fp->fd_top + lnum_off;
977071d4279SBram Moolenaar 		    if (lnum < curwin->w_cursor.lnum)
978071d4279SBram Moolenaar 			lnum_found = lnum;
979071d4279SBram Moolenaar 		}
980071d4279SBram Moolenaar 	    }
981071d4279SBram Moolenaar 
982071d4279SBram Moolenaar 	    if (last)
983071d4279SBram Moolenaar 		break;
984071d4279SBram Moolenaar 
985217e1b83SBram Moolenaar 	    // Check nested folds (if any).
986071d4279SBram Moolenaar 	    gap = &fp->fd_nested;
987071d4279SBram Moolenaar 	    lnum_off += fp->fd_top;
988071d4279SBram Moolenaar 	    ++level;
989071d4279SBram Moolenaar 	}
990071d4279SBram Moolenaar 	if (lnum_found != curwin->w_cursor.lnum)
991071d4279SBram Moolenaar 	{
992071d4279SBram Moolenaar 	    if (retval == FAIL)
993071d4279SBram Moolenaar 		setpcmark();
994071d4279SBram Moolenaar 	    curwin->w_cursor.lnum = lnum_found;
995071d4279SBram Moolenaar 	    curwin->w_cursor.col = 0;
996071d4279SBram Moolenaar 	    retval = OK;
997071d4279SBram Moolenaar 	}
998071d4279SBram Moolenaar 	else
999071d4279SBram Moolenaar 	    break;
1000071d4279SBram Moolenaar     }
1001071d4279SBram Moolenaar 
1002071d4279SBram Moolenaar     return retval;
1003071d4279SBram Moolenaar }
1004071d4279SBram Moolenaar 
1005217e1b83SBram Moolenaar // foldInitWin() {{{2
1006071d4279SBram Moolenaar /*
1007071d4279SBram Moolenaar  * Init the fold info in a new window.
1008071d4279SBram Moolenaar  */
1009071d4279SBram Moolenaar     void
101066f948e9SBram Moolenaar foldInitWin(win_T *new_win)
1011071d4279SBram Moolenaar {
101270b2a56dSBram Moolenaar     ga_init2(&new_win->w_folds, (int)sizeof(fold_T), 10);
1013071d4279SBram Moolenaar }
1014071d4279SBram Moolenaar 
1015217e1b83SBram Moolenaar // find_wl_entry() {{{2
1016071d4279SBram Moolenaar /*
1017071d4279SBram Moolenaar  * Find an entry in the win->w_lines[] array for buffer line "lnum".
1018071d4279SBram Moolenaar  * Only valid entries are considered (for entries where wl_valid is FALSE the
1019071d4279SBram Moolenaar  * line number can be wrong).
1020071d4279SBram Moolenaar  * Returns index of entry or -1 if not found.
1021071d4279SBram Moolenaar  */
1022071d4279SBram Moolenaar     int
102366f948e9SBram Moolenaar find_wl_entry(win_T *win, linenr_T lnum)
1024071d4279SBram Moolenaar {
1025071d4279SBram Moolenaar     int		i;
1026071d4279SBram Moolenaar 
1027071d4279SBram Moolenaar     for (i = 0; i < win->w_lines_valid; ++i)
1028071d4279SBram Moolenaar 	if (win->w_lines[i].wl_valid)
1029071d4279SBram Moolenaar 	{
1030071d4279SBram Moolenaar 	    if (lnum < win->w_lines[i].wl_lnum)
1031071d4279SBram Moolenaar 		return -1;
1032071d4279SBram Moolenaar 	    if (lnum <= win->w_lines[i].wl_lastlnum)
1033071d4279SBram Moolenaar 		return i;
1034071d4279SBram Moolenaar 	}
1035071d4279SBram Moolenaar     return -1;
1036071d4279SBram Moolenaar }
1037071d4279SBram Moolenaar 
1038217e1b83SBram Moolenaar // foldAdjustVisual() {{{2
1039071d4279SBram Moolenaar /*
1040071d4279SBram Moolenaar  * Adjust the Visual area to include any fold at the start or end completely.
1041071d4279SBram Moolenaar  */
1042071d4279SBram Moolenaar     void
104366f948e9SBram Moolenaar foldAdjustVisual(void)
1044071d4279SBram Moolenaar {
1045071d4279SBram Moolenaar     pos_T	*start, *end;
1046071d4279SBram Moolenaar     char_u	*ptr;
1047071d4279SBram Moolenaar 
1048071d4279SBram Moolenaar     if (!VIsual_active || !hasAnyFolding(curwin))
1049071d4279SBram Moolenaar 	return;
1050071d4279SBram Moolenaar 
1051b5aedf3eSBram Moolenaar     if (LTOREQ_POS(VIsual, curwin->w_cursor))
1052071d4279SBram Moolenaar     {
1053071d4279SBram Moolenaar 	start = &VIsual;
1054071d4279SBram Moolenaar 	end = &curwin->w_cursor;
1055071d4279SBram Moolenaar     }
1056071d4279SBram Moolenaar     else
1057071d4279SBram Moolenaar     {
1058071d4279SBram Moolenaar 	start = &curwin->w_cursor;
1059071d4279SBram Moolenaar 	end = &VIsual;
1060071d4279SBram Moolenaar     }
1061071d4279SBram Moolenaar     if (hasFolding(start->lnum, &start->lnum, NULL))
1062071d4279SBram Moolenaar 	start->col = 0;
1063071d4279SBram Moolenaar     if (hasFolding(end->lnum, NULL, &end->lnum))
1064071d4279SBram Moolenaar     {
1065071d4279SBram Moolenaar 	ptr = ml_get(end->lnum);
1066071d4279SBram Moolenaar 	end->col = (colnr_T)STRLEN(ptr);
1067071d4279SBram Moolenaar 	if (end->col > 0 && *p_sel == 'o')
1068071d4279SBram Moolenaar 	    --end->col;
1069217e1b83SBram Moolenaar 	// prevent cursor from moving on the trail byte
1070071d4279SBram Moolenaar 	if (has_mbyte)
1071071d4279SBram Moolenaar 	    mb_adjust_cursor();
1072071d4279SBram Moolenaar     }
1073071d4279SBram Moolenaar }
1074071d4279SBram Moolenaar 
1075217e1b83SBram Moolenaar // cursor_foldstart() {{{2
1076071d4279SBram Moolenaar /*
1077071d4279SBram Moolenaar  * Move the cursor to the first line of a closed fold.
1078071d4279SBram Moolenaar  */
1079071d4279SBram Moolenaar     void
108066f948e9SBram Moolenaar foldAdjustCursor(void)
1081071d4279SBram Moolenaar {
1082071d4279SBram Moolenaar     (void)hasFolding(curwin->w_cursor.lnum, &curwin->w_cursor.lnum, NULL);
1083071d4279SBram Moolenaar }
1084071d4279SBram Moolenaar 
1085217e1b83SBram Moolenaar // Internal functions for "fold_T" {{{1
1086217e1b83SBram Moolenaar // cloneFoldGrowArray() {{{2
1087071d4279SBram Moolenaar /*
1088071d4279SBram Moolenaar  * Will "clone" (i.e deep copy) a garray_T of folds.
1089071d4279SBram Moolenaar  *
1090071d4279SBram Moolenaar  * Return FAIL if the operation cannot be completed, otherwise OK.
1091071d4279SBram Moolenaar  */
1092071d4279SBram Moolenaar     void
109366f948e9SBram Moolenaar cloneFoldGrowArray(garray_T *from, garray_T *to)
1094071d4279SBram Moolenaar {
1095071d4279SBram Moolenaar     int		i;
1096071d4279SBram Moolenaar     fold_T	*from_p;
1097071d4279SBram Moolenaar     fold_T	*to_p;
1098071d4279SBram Moolenaar 
1099071d4279SBram Moolenaar     ga_init2(to, from->ga_itemsize, from->ga_growsize);
1100071d4279SBram Moolenaar     if (from->ga_len == 0 || ga_grow(to, from->ga_len) == FAIL)
1101071d4279SBram Moolenaar 	return;
1102071d4279SBram Moolenaar 
1103071d4279SBram Moolenaar     from_p = (fold_T *)from->ga_data;
1104071d4279SBram Moolenaar     to_p = (fold_T *)to->ga_data;
1105071d4279SBram Moolenaar 
1106071d4279SBram Moolenaar     for (i = 0; i < from->ga_len; i++)
1107071d4279SBram Moolenaar     {
1108071d4279SBram Moolenaar 	to_p->fd_top = from_p->fd_top;
1109071d4279SBram Moolenaar 	to_p->fd_len = from_p->fd_len;
1110071d4279SBram Moolenaar 	to_p->fd_flags = from_p->fd_flags;
1111071d4279SBram Moolenaar 	to_p->fd_small = from_p->fd_small;
1112071d4279SBram Moolenaar 	cloneFoldGrowArray(&from_p->fd_nested, &to_p->fd_nested);
1113071d4279SBram Moolenaar 	++to->ga_len;
1114071d4279SBram Moolenaar 	++from_p;
1115071d4279SBram Moolenaar 	++to_p;
1116071d4279SBram Moolenaar     }
1117071d4279SBram Moolenaar }
1118071d4279SBram Moolenaar 
1119217e1b83SBram Moolenaar // foldFind() {{{2
1120071d4279SBram Moolenaar /*
1121071d4279SBram Moolenaar  * Search for line "lnum" in folds of growarray "gap".
1122071d4279SBram Moolenaar  * Set *fpp to the fold struct for the fold that contains "lnum" or
1123071d4279SBram Moolenaar  * the first fold below it (careful: it can be beyond the end of the array!).
1124071d4279SBram Moolenaar  * Returns FALSE when there is no fold that contains "lnum".
1125071d4279SBram Moolenaar  */
1126071d4279SBram Moolenaar     static int
112766f948e9SBram Moolenaar foldFind(garray_T *gap, linenr_T lnum, fold_T **fpp)
1128071d4279SBram Moolenaar {
1129071d4279SBram Moolenaar     linenr_T	low, high;
1130071d4279SBram Moolenaar     fold_T	*fp;
1131071d4279SBram Moolenaar     int		i;
1132071d4279SBram Moolenaar 
1133*64f37d30SBram Moolenaar     if (gap->ga_len == 0)
1134*64f37d30SBram Moolenaar     {
1135*64f37d30SBram Moolenaar 	*fpp = NULL;
1136*64f37d30SBram Moolenaar 	return FALSE;
1137*64f37d30SBram Moolenaar     }
1138*64f37d30SBram Moolenaar 
1139071d4279SBram Moolenaar     /*
1140071d4279SBram Moolenaar      * Perform a binary search.
1141071d4279SBram Moolenaar      * "low" is lowest index of possible match.
1142071d4279SBram Moolenaar      * "high" is highest index of possible match.
1143071d4279SBram Moolenaar      */
1144071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
1145071d4279SBram Moolenaar     low = 0;
1146071d4279SBram Moolenaar     high = gap->ga_len - 1;
1147071d4279SBram Moolenaar     while (low <= high)
1148071d4279SBram Moolenaar     {
1149071d4279SBram Moolenaar 	i = (low + high) / 2;
1150071d4279SBram Moolenaar 	if (fp[i].fd_top > lnum)
1151217e1b83SBram Moolenaar 	    // fold below lnum, adjust high
1152071d4279SBram Moolenaar 	    high = i - 1;
1153071d4279SBram Moolenaar 	else if (fp[i].fd_top + fp[i].fd_len <= lnum)
1154217e1b83SBram Moolenaar 	    // fold above lnum, adjust low
1155071d4279SBram Moolenaar 	    low = i + 1;
1156071d4279SBram Moolenaar 	else
1157071d4279SBram Moolenaar 	{
1158217e1b83SBram Moolenaar 	    // lnum is inside this fold
1159071d4279SBram Moolenaar 	    *fpp = fp + i;
1160071d4279SBram Moolenaar 	    return TRUE;
1161071d4279SBram Moolenaar 	}
1162071d4279SBram Moolenaar     }
1163071d4279SBram Moolenaar     *fpp = fp + low;
1164071d4279SBram Moolenaar     return FALSE;
1165071d4279SBram Moolenaar }
1166071d4279SBram Moolenaar 
1167217e1b83SBram Moolenaar // foldLevelWin() {{{2
1168071d4279SBram Moolenaar /*
1169071d4279SBram Moolenaar  * Return fold level at line number "lnum" in window "wp".
1170071d4279SBram Moolenaar  */
1171071d4279SBram Moolenaar     static int
117266f948e9SBram Moolenaar foldLevelWin(win_T *wp, linenr_T lnum)
1173071d4279SBram Moolenaar {
1174071d4279SBram Moolenaar     fold_T	*fp;
1175071d4279SBram Moolenaar     linenr_T	lnum_rel = lnum;
1176071d4279SBram Moolenaar     int		level =  0;
1177071d4279SBram Moolenaar     garray_T	*gap;
1178071d4279SBram Moolenaar 
1179217e1b83SBram Moolenaar     // Recursively search for a fold that contains "lnum".
1180071d4279SBram Moolenaar     gap = &wp->w_folds;
1181071d4279SBram Moolenaar     for (;;)
1182071d4279SBram Moolenaar     {
1183071d4279SBram Moolenaar 	if (!foldFind(gap, lnum_rel, &fp))
1184071d4279SBram Moolenaar 	    break;
1185217e1b83SBram Moolenaar 	// Check nested folds.  Line number is relative to containing fold.
1186071d4279SBram Moolenaar 	gap = &fp->fd_nested;
1187071d4279SBram Moolenaar 	lnum_rel -= fp->fd_top;
1188071d4279SBram Moolenaar 	++level;
1189071d4279SBram Moolenaar     }
1190071d4279SBram Moolenaar 
1191071d4279SBram Moolenaar     return level;
1192071d4279SBram Moolenaar }
1193071d4279SBram Moolenaar 
1194217e1b83SBram Moolenaar // checkupdate() {{{2
1195071d4279SBram Moolenaar /*
1196071d4279SBram Moolenaar  * Check if the folds in window "wp" are invalid and update them if needed.
1197071d4279SBram Moolenaar  */
1198071d4279SBram Moolenaar     static void
119966f948e9SBram Moolenaar checkupdate(win_T *wp)
1200071d4279SBram Moolenaar {
1201071d4279SBram Moolenaar     if (wp->w_foldinvalid)
1202071d4279SBram Moolenaar     {
1203217e1b83SBram Moolenaar 	foldUpdate(wp, (linenr_T)1, (linenr_T)MAXLNUM); // will update all
1204071d4279SBram Moolenaar 	wp->w_foldinvalid = FALSE;
1205071d4279SBram Moolenaar     }
1206071d4279SBram Moolenaar }
1207071d4279SBram Moolenaar 
1208217e1b83SBram Moolenaar // setFoldRepeat() {{{2
1209071d4279SBram Moolenaar /*
1210071d4279SBram Moolenaar  * Open or close fold for current window at line "lnum".
1211071d4279SBram Moolenaar  * Repeat "count" times.
1212071d4279SBram Moolenaar  */
1213071d4279SBram Moolenaar     static void
121466f948e9SBram Moolenaar setFoldRepeat(linenr_T lnum, long count, int do_open)
1215071d4279SBram Moolenaar {
1216071d4279SBram Moolenaar     int		done;
1217071d4279SBram Moolenaar     long	n;
1218071d4279SBram Moolenaar 
1219071d4279SBram Moolenaar     for (n = 0; n < count; ++n)
1220071d4279SBram Moolenaar     {
1221071d4279SBram Moolenaar 	done = DONE_NOTHING;
1222fe86f2d7SBram Moolenaar 	(void)setManualFold(lnum, do_open, FALSE, &done);
1223071d4279SBram Moolenaar 	if (!(done & DONE_ACTION))
1224071d4279SBram Moolenaar 	{
1225217e1b83SBram Moolenaar 	    // Only give an error message when no fold could be opened.
1226071d4279SBram Moolenaar 	    if (n == 0 && !(done & DONE_FOLD))
1227f9e3e09fSBram Moolenaar 		emsg(_(e_nofold));
1228071d4279SBram Moolenaar 	    break;
1229071d4279SBram Moolenaar 	}
1230071d4279SBram Moolenaar     }
1231071d4279SBram Moolenaar }
1232071d4279SBram Moolenaar 
1233217e1b83SBram Moolenaar // setManualFold() {{{2
1234071d4279SBram Moolenaar /*
1235071d4279SBram Moolenaar  * Open or close the fold in the current window which contains "lnum".
1236071d4279SBram Moolenaar  * Also does this for other windows in diff mode when needed.
1237071d4279SBram Moolenaar  */
1238071d4279SBram Moolenaar     static linenr_T
123966f948e9SBram Moolenaar setManualFold(
124066f948e9SBram Moolenaar     linenr_T	lnum,
1241217e1b83SBram Moolenaar     int		opening,    // TRUE when opening, FALSE when closing
1242217e1b83SBram Moolenaar     int		recurse,    // TRUE when closing/opening recursive
124366f948e9SBram Moolenaar     int		*donep)
1244071d4279SBram Moolenaar {
1245071d4279SBram Moolenaar #ifdef FEAT_DIFF
1246071d4279SBram Moolenaar     if (foldmethodIsDiff(curwin) && curwin->w_p_scb)
1247071d4279SBram Moolenaar     {
1248071d4279SBram Moolenaar 	win_T	    *wp;
1249071d4279SBram Moolenaar 	linenr_T    dlnum;
1250071d4279SBram Moolenaar 
1251071d4279SBram Moolenaar 	/*
1252071d4279SBram Moolenaar 	 * Do the same operation in other windows in diff mode.  Calculate the
1253071d4279SBram Moolenaar 	 * line number from the diffs.
1254071d4279SBram Moolenaar 	 */
1255071d4279SBram Moolenaar 	FOR_ALL_WINDOWS(wp)
1256071d4279SBram Moolenaar 	{
1257071d4279SBram Moolenaar 	    if (wp != curwin && foldmethodIsDiff(wp) && wp->w_p_scb)
1258071d4279SBram Moolenaar 	    {
1259071d4279SBram Moolenaar 		dlnum = diff_lnum_win(curwin->w_cursor.lnum, wp);
1260071d4279SBram Moolenaar 		if (dlnum != 0)
1261071d4279SBram Moolenaar 		    (void)setManualFoldWin(wp, dlnum, opening, recurse, NULL);
1262071d4279SBram Moolenaar 	    }
1263071d4279SBram Moolenaar 	}
1264071d4279SBram Moolenaar     }
1265071d4279SBram Moolenaar #endif
1266071d4279SBram Moolenaar 
1267071d4279SBram Moolenaar     return setManualFoldWin(curwin, lnum, opening, recurse, donep);
1268071d4279SBram Moolenaar }
1269071d4279SBram Moolenaar 
1270217e1b83SBram Moolenaar // setManualFoldWin() {{{2
1271071d4279SBram Moolenaar /*
1272071d4279SBram Moolenaar  * Open or close the fold in window "wp" which contains "lnum".
1273071d4279SBram Moolenaar  * "donep", when not NULL, points to flag that is set to DONE_FOLD when some
1274071d4279SBram Moolenaar  * fold was found and to DONE_ACTION when some fold was opened or closed.
1275071d4279SBram Moolenaar  * When "donep" is NULL give an error message when no fold was found for
1276071d4279SBram Moolenaar  * "lnum", but only if "wp" is "curwin".
1277071d4279SBram Moolenaar  * Return the line number of the next line that could be closed.
1278071d4279SBram Moolenaar  * It's only valid when "opening" is TRUE!
1279071d4279SBram Moolenaar  */
1280071d4279SBram Moolenaar     static linenr_T
128166f948e9SBram Moolenaar setManualFoldWin(
128266f948e9SBram Moolenaar     win_T	*wp,
128366f948e9SBram Moolenaar     linenr_T	lnum,
1284217e1b83SBram Moolenaar     int		opening,    // TRUE when opening, FALSE when closing
1285217e1b83SBram Moolenaar     int		recurse,    // TRUE when closing/opening recursive
128666f948e9SBram Moolenaar     int		*donep)
1287071d4279SBram Moolenaar {
1288071d4279SBram Moolenaar     fold_T	*fp;
1289071d4279SBram Moolenaar     fold_T	*fp2;
1290071d4279SBram Moolenaar     fold_T	*found = NULL;
1291071d4279SBram Moolenaar     int		j;
1292071d4279SBram Moolenaar     int		level = 0;
1293071d4279SBram Moolenaar     int		use_level = FALSE;
1294071d4279SBram Moolenaar     int		found_fold = FALSE;
1295071d4279SBram Moolenaar     garray_T	*gap;
1296071d4279SBram Moolenaar     linenr_T	next = MAXLNUM;
1297071d4279SBram Moolenaar     linenr_T	off = 0;
1298071d4279SBram Moolenaar     int		done = 0;
1299071d4279SBram Moolenaar 
1300071d4279SBram Moolenaar     checkupdate(wp);
1301071d4279SBram Moolenaar 
1302071d4279SBram Moolenaar     /*
1303071d4279SBram Moolenaar      * Find the fold, open or close it.
1304071d4279SBram Moolenaar      */
1305071d4279SBram Moolenaar     gap = &wp->w_folds;
1306071d4279SBram Moolenaar     for (;;)
1307071d4279SBram Moolenaar     {
1308071d4279SBram Moolenaar 	if (!foldFind(gap, lnum, &fp))
1309071d4279SBram Moolenaar 	{
1310217e1b83SBram Moolenaar 	    // If there is a following fold, continue there next time.
1311071d4279SBram Moolenaar 	    if (fp < (fold_T *)gap->ga_data + gap->ga_len)
1312071d4279SBram Moolenaar 		next = fp->fd_top + off;
1313071d4279SBram Moolenaar 	    break;
1314071d4279SBram Moolenaar 	}
1315071d4279SBram Moolenaar 
1316217e1b83SBram Moolenaar 	// lnum is inside this fold
1317071d4279SBram Moolenaar 	found_fold = TRUE;
1318071d4279SBram Moolenaar 
1319217e1b83SBram Moolenaar 	// If there is a following fold, continue there next time.
1320071d4279SBram Moolenaar 	if (fp + 1 < (fold_T *)gap->ga_data + gap->ga_len)
1321071d4279SBram Moolenaar 	    next = fp[1].fd_top + off;
1322071d4279SBram Moolenaar 
1323217e1b83SBram Moolenaar 	// Change from level-dependent folding to manual.
1324071d4279SBram Moolenaar 	if (use_level || fp->fd_flags == FD_LEVEL)
1325071d4279SBram Moolenaar 	{
1326071d4279SBram Moolenaar 	    use_level = TRUE;
1327071d4279SBram Moolenaar 	    if (level >= wp->w_p_fdl)
1328071d4279SBram Moolenaar 		fp->fd_flags = FD_CLOSED;
1329071d4279SBram Moolenaar 	    else
1330071d4279SBram Moolenaar 		fp->fd_flags = FD_OPEN;
1331071d4279SBram Moolenaar 	    fp2 = (fold_T *)fp->fd_nested.ga_data;
1332071d4279SBram Moolenaar 	    for (j = 0; j < fp->fd_nested.ga_len; ++j)
1333071d4279SBram Moolenaar 		fp2[j].fd_flags = FD_LEVEL;
1334071d4279SBram Moolenaar 	}
1335071d4279SBram Moolenaar 
1336217e1b83SBram Moolenaar 	// Simple case: Close recursively means closing the fold.
1337071d4279SBram Moolenaar 	if (!opening && recurse)
1338071d4279SBram Moolenaar 	{
1339071d4279SBram Moolenaar 	    if (fp->fd_flags != FD_CLOSED)
1340071d4279SBram Moolenaar 	    {
1341071d4279SBram Moolenaar 		done |= DONE_ACTION;
1342071d4279SBram Moolenaar 		fp->fd_flags = FD_CLOSED;
1343071d4279SBram Moolenaar 	    }
1344071d4279SBram Moolenaar 	}
1345071d4279SBram Moolenaar 	else if (fp->fd_flags == FD_CLOSED)
1346071d4279SBram Moolenaar 	{
1347217e1b83SBram Moolenaar 	    // When opening, open topmost closed fold.
1348071d4279SBram Moolenaar 	    if (opening)
1349071d4279SBram Moolenaar 	    {
1350071d4279SBram Moolenaar 		fp->fd_flags = FD_OPEN;
1351071d4279SBram Moolenaar 		done |= DONE_ACTION;
1352071d4279SBram Moolenaar 		if (recurse)
1353071d4279SBram Moolenaar 		    foldOpenNested(fp);
1354071d4279SBram Moolenaar 	    }
1355071d4279SBram Moolenaar 	    break;
1356071d4279SBram Moolenaar 	}
1357071d4279SBram Moolenaar 
1358217e1b83SBram Moolenaar 	// fold is open, check nested folds
1359071d4279SBram Moolenaar 	found = fp;
1360071d4279SBram Moolenaar 	gap = &fp->fd_nested;
1361071d4279SBram Moolenaar 	lnum -= fp->fd_top;
1362071d4279SBram Moolenaar 	off += fp->fd_top;
1363071d4279SBram Moolenaar 	++level;
1364071d4279SBram Moolenaar     }
1365071d4279SBram Moolenaar     if (found_fold)
1366071d4279SBram Moolenaar     {
1367217e1b83SBram Moolenaar 	// When closing and not recurse, close deepest open fold.
1368071d4279SBram Moolenaar 	if (!opening && found != NULL)
1369071d4279SBram Moolenaar 	{
1370071d4279SBram Moolenaar 	    found->fd_flags = FD_CLOSED;
1371071d4279SBram Moolenaar 	    done |= DONE_ACTION;
1372071d4279SBram Moolenaar 	}
1373071d4279SBram Moolenaar 	wp->w_fold_manual = TRUE;
1374071d4279SBram Moolenaar 	if (done & DONE_ACTION)
1375071d4279SBram Moolenaar 	    changed_window_setting_win(wp);
1376071d4279SBram Moolenaar 	done |= DONE_FOLD;
1377071d4279SBram Moolenaar     }
1378071d4279SBram Moolenaar     else if (donep == NULL && wp == curwin)
1379f9e3e09fSBram Moolenaar 	emsg(_(e_nofold));
1380071d4279SBram Moolenaar 
1381071d4279SBram Moolenaar     if (donep != NULL)
1382071d4279SBram Moolenaar 	*donep |= done;
1383071d4279SBram Moolenaar 
1384071d4279SBram Moolenaar     return next;
1385071d4279SBram Moolenaar }
1386071d4279SBram Moolenaar 
1387217e1b83SBram Moolenaar // foldOpenNested() {{{2
1388071d4279SBram Moolenaar /*
1389071d4279SBram Moolenaar  * Open all nested folds in fold "fpr" recursively.
1390071d4279SBram Moolenaar  */
1391071d4279SBram Moolenaar     static void
139266f948e9SBram Moolenaar foldOpenNested(fold_T *fpr)
1393071d4279SBram Moolenaar {
1394071d4279SBram Moolenaar     int		i;
1395071d4279SBram Moolenaar     fold_T	*fp;
1396071d4279SBram Moolenaar 
1397071d4279SBram Moolenaar     fp = (fold_T *)fpr->fd_nested.ga_data;
1398071d4279SBram Moolenaar     for (i = 0; i < fpr->fd_nested.ga_len; ++i)
1399071d4279SBram Moolenaar     {
1400071d4279SBram Moolenaar 	foldOpenNested(&fp[i]);
1401071d4279SBram Moolenaar 	fp[i].fd_flags = FD_OPEN;
1402071d4279SBram Moolenaar     }
1403071d4279SBram Moolenaar }
1404071d4279SBram Moolenaar 
1405217e1b83SBram Moolenaar // deleteFoldEntry() {{{2
1406071d4279SBram Moolenaar /*
1407071d4279SBram Moolenaar  * Delete fold "idx" from growarray "gap".
1408071d4279SBram Moolenaar  * When "recursive" is TRUE also delete all the folds contained in it.
1409071d4279SBram Moolenaar  * When "recursive" is FALSE contained folds are moved one level up.
1410071d4279SBram Moolenaar  */
1411071d4279SBram Moolenaar     static void
141266f948e9SBram Moolenaar deleteFoldEntry(garray_T *gap, int idx, int recursive)
1413071d4279SBram Moolenaar {
1414071d4279SBram Moolenaar     fold_T	*fp;
1415071d4279SBram Moolenaar     int		i;
1416071d4279SBram Moolenaar     long	moved;
1417071d4279SBram Moolenaar     fold_T	*nfp;
1418071d4279SBram Moolenaar 
1419071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data + idx;
1420071d4279SBram Moolenaar     if (recursive || fp->fd_nested.ga_len == 0)
1421071d4279SBram Moolenaar     {
1422217e1b83SBram Moolenaar 	// recursively delete the contained folds
1423071d4279SBram Moolenaar 	deleteFoldRecurse(&fp->fd_nested);
1424071d4279SBram Moolenaar 	--gap->ga_len;
1425071d4279SBram Moolenaar 	if (idx < gap->ga_len)
1426071d4279SBram Moolenaar 	    mch_memmove(fp, fp + 1, sizeof(fold_T) * (gap->ga_len - idx));
1427071d4279SBram Moolenaar     }
1428071d4279SBram Moolenaar     else
1429071d4279SBram Moolenaar     {
1430217e1b83SBram Moolenaar 	// Move nested folds one level up, to overwrite the fold that is
1431217e1b83SBram Moolenaar 	// deleted.
1432071d4279SBram Moolenaar 	moved = fp->fd_nested.ga_len;
1433071d4279SBram Moolenaar 	if (ga_grow(gap, (int)(moved - 1)) == OK)
1434071d4279SBram Moolenaar 	{
1435217e1b83SBram Moolenaar 	    // Get "fp" again, the array may have been reallocated.
1436c9927c13SBram Moolenaar 	    fp = (fold_T *)gap->ga_data + idx;
1437c9927c13SBram Moolenaar 
1438217e1b83SBram Moolenaar 	    // adjust fd_top and fd_flags for the moved folds
1439071d4279SBram Moolenaar 	    nfp = (fold_T *)fp->fd_nested.ga_data;
1440071d4279SBram Moolenaar 	    for (i = 0; i < moved; ++i)
1441071d4279SBram Moolenaar 	    {
1442071d4279SBram Moolenaar 		nfp[i].fd_top += fp->fd_top;
1443071d4279SBram Moolenaar 		if (fp->fd_flags == FD_LEVEL)
1444071d4279SBram Moolenaar 		    nfp[i].fd_flags = FD_LEVEL;
1445071d4279SBram Moolenaar 		if (fp->fd_small == MAYBE)
1446071d4279SBram Moolenaar 		    nfp[i].fd_small = MAYBE;
1447071d4279SBram Moolenaar 	    }
1448071d4279SBram Moolenaar 
1449217e1b83SBram Moolenaar 	    // move the existing folds down to make room
1450c9927c13SBram Moolenaar 	    if (idx + 1 < gap->ga_len)
1451071d4279SBram Moolenaar 		mch_memmove(fp + moved, fp + 1,
1452c9927c13SBram Moolenaar 				  sizeof(fold_T) * (gap->ga_len - (idx + 1)));
1453217e1b83SBram Moolenaar 	    // move the contained folds one level up
1454071d4279SBram Moolenaar 	    mch_memmove(fp, nfp, (size_t)(sizeof(fold_T) * moved));
1455071d4279SBram Moolenaar 	    vim_free(nfp);
1456071d4279SBram Moolenaar 	    gap->ga_len += moved - 1;
1457071d4279SBram Moolenaar 	}
1458071d4279SBram Moolenaar     }
1459071d4279SBram Moolenaar }
1460071d4279SBram Moolenaar 
1461217e1b83SBram Moolenaar // deleteFoldRecurse() {{{2
1462071d4279SBram Moolenaar /*
1463071d4279SBram Moolenaar  * Delete nested folds in a fold.
1464071d4279SBram Moolenaar  */
1465071d4279SBram Moolenaar     void
146666f948e9SBram Moolenaar deleteFoldRecurse(garray_T *gap)
1467071d4279SBram Moolenaar {
1468071d4279SBram Moolenaar     int		i;
1469071d4279SBram Moolenaar 
1470071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
1471071d4279SBram Moolenaar 	deleteFoldRecurse(&(((fold_T *)(gap->ga_data))[i].fd_nested));
1472071d4279SBram Moolenaar     ga_clear(gap);
1473071d4279SBram Moolenaar }
1474071d4279SBram Moolenaar 
1475217e1b83SBram Moolenaar // foldMarkAdjust() {{{2
1476071d4279SBram Moolenaar /*
1477071d4279SBram Moolenaar  * Update line numbers of folds for inserted/deleted lines.
1478071d4279SBram Moolenaar  */
1479071d4279SBram Moolenaar     void
148066f948e9SBram Moolenaar foldMarkAdjust(
148166f948e9SBram Moolenaar     win_T	*wp,
148266f948e9SBram Moolenaar     linenr_T	line1,
148366f948e9SBram Moolenaar     linenr_T	line2,
148466f948e9SBram Moolenaar     long	amount,
148566f948e9SBram Moolenaar     long	amount_after)
1486071d4279SBram Moolenaar {
1487217e1b83SBram Moolenaar     // If deleting marks from line1 to line2, but not deleting all those
1488217e1b83SBram Moolenaar     // lines, set line2 so that only deleted lines have their folds removed.
1489071d4279SBram Moolenaar     if (amount == MAXLNUM && line2 >= line1 && line2 - line1 >= -amount_after)
1490071d4279SBram Moolenaar 	line2 = line1 - amount_after - 1;
1491217e1b83SBram Moolenaar     // If appending a line in Insert mode, it should be included in the fold
1492217e1b83SBram Moolenaar     // just above the line.
1493071d4279SBram Moolenaar     if ((State & INSERT) && amount == (linenr_T)1 && line2 == MAXLNUM)
1494071d4279SBram Moolenaar 	--line1;
1495071d4279SBram Moolenaar     foldMarkAdjustRecurse(&wp->w_folds, line1, line2, amount, amount_after);
1496071d4279SBram Moolenaar }
1497071d4279SBram Moolenaar 
1498217e1b83SBram Moolenaar // foldMarkAdjustRecurse() {{{2
1499071d4279SBram Moolenaar     static void
150066f948e9SBram Moolenaar foldMarkAdjustRecurse(
150166f948e9SBram Moolenaar     garray_T	*gap,
150266f948e9SBram Moolenaar     linenr_T	line1,
150366f948e9SBram Moolenaar     linenr_T	line2,
150466f948e9SBram Moolenaar     long	amount,
150566f948e9SBram Moolenaar     long	amount_after)
1506071d4279SBram Moolenaar {
1507071d4279SBram Moolenaar     fold_T	*fp;
1508071d4279SBram Moolenaar     int		i;
1509071d4279SBram Moolenaar     linenr_T	last;
1510071d4279SBram Moolenaar     linenr_T	top;
1511071d4279SBram Moolenaar 
1512217e1b83SBram Moolenaar     // In Insert mode an inserted line at the top of a fold is considered part
1513217e1b83SBram Moolenaar     // of the fold, otherwise it isn't.
1514071d4279SBram Moolenaar     if ((State & INSERT) && amount == (linenr_T)1 && line2 == MAXLNUM)
1515071d4279SBram Moolenaar 	top = line1 + 1;
1516071d4279SBram Moolenaar     else
1517071d4279SBram Moolenaar 	top = line1;
1518071d4279SBram Moolenaar 
1519217e1b83SBram Moolenaar     // Find the fold containing or just below "line1".
1520071d4279SBram Moolenaar     (void)foldFind(gap, line1, &fp);
1521071d4279SBram Moolenaar 
1522071d4279SBram Moolenaar     /*
1523071d4279SBram Moolenaar      * Adjust all folds below "line1" that are affected.
1524071d4279SBram Moolenaar      */
1525071d4279SBram Moolenaar     for (i = (int)(fp - (fold_T *)gap->ga_data); i < gap->ga_len; ++i, ++fp)
1526071d4279SBram Moolenaar     {
1527071d4279SBram Moolenaar 	/*
1528071d4279SBram Moolenaar 	 * Check for these situations:
1529071d4279SBram Moolenaar 	 *	  1  2	3
1530071d4279SBram Moolenaar 	 *	  1  2	3
1531071d4279SBram Moolenaar 	 * line1     2	3  4  5
1532071d4279SBram Moolenaar 	 *	     2	3  4  5
1533071d4279SBram Moolenaar 	 *	     2	3  4  5
1534071d4279SBram Moolenaar 	 * line2     2	3  4  5
1535071d4279SBram Moolenaar 	 *		3     5  6
1536071d4279SBram Moolenaar 	 *		3     5  6
1537071d4279SBram Moolenaar 	 */
1538071d4279SBram Moolenaar 
1539217e1b83SBram Moolenaar 	last = fp->fd_top + fp->fd_len - 1; // last line of fold
1540071d4279SBram Moolenaar 
1541217e1b83SBram Moolenaar 	// 1. fold completely above line1: nothing to do
1542071d4279SBram Moolenaar 	if (last < line1)
1543071d4279SBram Moolenaar 	    continue;
1544071d4279SBram Moolenaar 
1545217e1b83SBram Moolenaar 	// 6. fold below line2: only adjust for amount_after
1546071d4279SBram Moolenaar 	if (fp->fd_top > line2)
1547071d4279SBram Moolenaar 	{
1548071d4279SBram Moolenaar 	    if (amount_after == 0)
1549071d4279SBram Moolenaar 		break;
1550071d4279SBram Moolenaar 	    fp->fd_top += amount_after;
1551071d4279SBram Moolenaar 	}
1552071d4279SBram Moolenaar 	else
1553071d4279SBram Moolenaar 	{
1554071d4279SBram Moolenaar 	    if (fp->fd_top >= top && last <= line2)
1555071d4279SBram Moolenaar 	    {
1556217e1b83SBram Moolenaar 		// 4. fold completely contained in range
1557071d4279SBram Moolenaar 		if (amount == MAXLNUM)
1558071d4279SBram Moolenaar 		{
1559217e1b83SBram Moolenaar 		    // Deleting lines: delete the fold completely
1560071d4279SBram Moolenaar 		    deleteFoldEntry(gap, i, TRUE);
1561217e1b83SBram Moolenaar 		    --i;    // adjust index for deletion
1562071d4279SBram Moolenaar 		    --fp;
1563071d4279SBram Moolenaar 		}
1564071d4279SBram Moolenaar 		else
1565071d4279SBram Moolenaar 		    fp->fd_top += amount;
1566071d4279SBram Moolenaar 	    }
1567071d4279SBram Moolenaar 	    else
1568071d4279SBram Moolenaar 	    {
1569071d4279SBram Moolenaar 		if (fp->fd_top < top)
1570071d4279SBram Moolenaar 		{
1571217e1b83SBram Moolenaar 		    // 2 or 3: need to correct nested folds too
1572194b94c5SBram Moolenaar 		    foldMarkAdjustRecurse(&fp->fd_nested, line1 - fp->fd_top,
1573194b94c5SBram Moolenaar 				  line2 - fp->fd_top, amount, amount_after);
1574071d4279SBram Moolenaar 		    if (last <= line2)
1575071d4279SBram Moolenaar 		    {
1576217e1b83SBram Moolenaar 			// 2. fold contains line1, line2 is below fold
1577071d4279SBram Moolenaar 			if (amount == MAXLNUM)
1578071d4279SBram Moolenaar 			    fp->fd_len = line1 - fp->fd_top;
1579071d4279SBram Moolenaar 			else
1580071d4279SBram Moolenaar 			    fp->fd_len += amount;
1581071d4279SBram Moolenaar 		    }
1582071d4279SBram Moolenaar 		    else
1583071d4279SBram Moolenaar 		    {
1584217e1b83SBram Moolenaar 			// 3. fold contains line1 and line2
1585071d4279SBram Moolenaar 			fp->fd_len += amount_after;
1586071d4279SBram Moolenaar 		    }
1587071d4279SBram Moolenaar 		}
1588071d4279SBram Moolenaar 		else
1589071d4279SBram Moolenaar 		{
1590217e1b83SBram Moolenaar 		    // 5. fold is below line1 and contains line2; need to
1591217e1b83SBram Moolenaar 		    // correct nested folds too
1592071d4279SBram Moolenaar 		    if (amount == MAXLNUM)
1593071d4279SBram Moolenaar 		    {
15941159b168SBram Moolenaar 			foldMarkAdjustRecurse(&fp->fd_nested,
15951159b168SBram Moolenaar 				  line1 - fp->fd_top,
15961159b168SBram Moolenaar 				  line2 - fp->fd_top,
15971159b168SBram Moolenaar 				  amount,
15981159b168SBram Moolenaar 				  amount_after + (fp->fd_top - top));
1599071d4279SBram Moolenaar 			fp->fd_len -= line2 - fp->fd_top + 1;
1600071d4279SBram Moolenaar 			fp->fd_top = line1;
1601071d4279SBram Moolenaar 		    }
1602071d4279SBram Moolenaar 		    else
1603071d4279SBram Moolenaar 		    {
16041159b168SBram Moolenaar 			foldMarkAdjustRecurse(&fp->fd_nested,
16051159b168SBram Moolenaar 				  line1 - fp->fd_top,
16061159b168SBram Moolenaar 				  line2 - fp->fd_top,
16071159b168SBram Moolenaar 				  amount,
16081159b168SBram Moolenaar 				  amount_after - amount);
1609071d4279SBram Moolenaar 			fp->fd_len += amount_after - amount;
1610071d4279SBram Moolenaar 			fp->fd_top += amount;
1611071d4279SBram Moolenaar 		    }
1612071d4279SBram Moolenaar 		}
1613071d4279SBram Moolenaar 	    }
1614071d4279SBram Moolenaar 	}
1615071d4279SBram Moolenaar     }
1616071d4279SBram Moolenaar }
1617071d4279SBram Moolenaar 
1618217e1b83SBram Moolenaar // getDeepestNesting() {{{2
1619071d4279SBram Moolenaar /*
1620071d4279SBram Moolenaar  * Get the lowest 'foldlevel' value that makes the deepest nested fold in the
1621071d4279SBram Moolenaar  * current window open.
1622071d4279SBram Moolenaar  */
1623071d4279SBram Moolenaar     int
162466f948e9SBram Moolenaar getDeepestNesting(void)
1625071d4279SBram Moolenaar {
1626071d4279SBram Moolenaar     checkupdate(curwin);
1627071d4279SBram Moolenaar     return getDeepestNestingRecurse(&curwin->w_folds);
1628071d4279SBram Moolenaar }
1629071d4279SBram Moolenaar 
1630071d4279SBram Moolenaar     static int
163166f948e9SBram Moolenaar getDeepestNestingRecurse(garray_T *gap)
1632071d4279SBram Moolenaar {
1633071d4279SBram Moolenaar     int		i;
1634071d4279SBram Moolenaar     int		level;
1635071d4279SBram Moolenaar     int		maxlevel = 0;
1636071d4279SBram Moolenaar     fold_T	*fp;
1637071d4279SBram Moolenaar 
1638071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
1639071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
1640071d4279SBram Moolenaar     {
1641071d4279SBram Moolenaar 	level = getDeepestNestingRecurse(&fp[i].fd_nested) + 1;
1642071d4279SBram Moolenaar 	if (level > maxlevel)
1643071d4279SBram Moolenaar 	    maxlevel = level;
1644071d4279SBram Moolenaar     }
1645071d4279SBram Moolenaar 
1646071d4279SBram Moolenaar     return maxlevel;
1647071d4279SBram Moolenaar }
1648071d4279SBram Moolenaar 
1649217e1b83SBram Moolenaar // check_closed() {{{2
1650071d4279SBram Moolenaar /*
1651071d4279SBram Moolenaar  * Check if a fold is closed and update the info needed to check nested folds.
1652071d4279SBram Moolenaar  */
1653071d4279SBram Moolenaar     static int
165466f948e9SBram Moolenaar check_closed(
165566f948e9SBram Moolenaar     win_T	*win,
165666f948e9SBram Moolenaar     fold_T	*fp,
1657217e1b83SBram Moolenaar     int		*use_levelp,	    // TRUE: outer fold had FD_LEVEL
1658217e1b83SBram Moolenaar     int		level,		    // folding depth
1659217e1b83SBram Moolenaar     int		*maybe_smallp,	    // TRUE: outer this had fd_small == MAYBE
1660217e1b83SBram Moolenaar     linenr_T	lnum_off)	    // line number offset for fp->fd_top
1661071d4279SBram Moolenaar {
1662071d4279SBram Moolenaar     int		closed = FALSE;
1663071d4279SBram Moolenaar 
1664217e1b83SBram Moolenaar     // Check if this fold is closed.  If the flag is FD_LEVEL this
1665217e1b83SBram Moolenaar     // fold and all folds it contains depend on 'foldlevel'.
1666071d4279SBram Moolenaar     if (*use_levelp || fp->fd_flags == FD_LEVEL)
1667071d4279SBram Moolenaar     {
1668071d4279SBram Moolenaar 	*use_levelp = TRUE;
1669071d4279SBram Moolenaar 	if (level >= win->w_p_fdl)
1670071d4279SBram Moolenaar 	    closed = TRUE;
1671071d4279SBram Moolenaar     }
1672071d4279SBram Moolenaar     else if (fp->fd_flags == FD_CLOSED)
1673071d4279SBram Moolenaar 	closed = TRUE;
1674071d4279SBram Moolenaar 
1675217e1b83SBram Moolenaar     // Small fold isn't closed anyway.
1676071d4279SBram Moolenaar     if (fp->fd_small == MAYBE)
1677071d4279SBram Moolenaar 	*maybe_smallp = TRUE;
1678071d4279SBram Moolenaar     if (closed)
1679071d4279SBram Moolenaar     {
1680071d4279SBram Moolenaar 	if (*maybe_smallp)
1681071d4279SBram Moolenaar 	    fp->fd_small = MAYBE;
1682071d4279SBram Moolenaar 	checkSmall(win, fp, lnum_off);
1683071d4279SBram Moolenaar 	if (fp->fd_small == TRUE)
1684071d4279SBram Moolenaar 	    closed = FALSE;
1685071d4279SBram Moolenaar     }
1686071d4279SBram Moolenaar     return closed;
1687071d4279SBram Moolenaar }
1688071d4279SBram Moolenaar 
1689217e1b83SBram Moolenaar // checkSmall() {{{2
1690071d4279SBram Moolenaar /*
1691071d4279SBram Moolenaar  * Update fd_small field of fold "fp".
1692071d4279SBram Moolenaar  */
1693071d4279SBram Moolenaar     static void
169466f948e9SBram Moolenaar checkSmall(
169566f948e9SBram Moolenaar     win_T	*wp,
169666f948e9SBram Moolenaar     fold_T	*fp,
1697217e1b83SBram Moolenaar     linenr_T	lnum_off)	// offset for fp->fd_top
1698071d4279SBram Moolenaar {
1699071d4279SBram Moolenaar     int		count;
1700071d4279SBram Moolenaar     int		n;
1701071d4279SBram Moolenaar 
1702071d4279SBram Moolenaar     if (fp->fd_small == MAYBE)
1703071d4279SBram Moolenaar     {
1704217e1b83SBram Moolenaar 	// Mark any nested folds to maybe-small
1705071d4279SBram Moolenaar 	setSmallMaybe(&fp->fd_nested);
1706071d4279SBram Moolenaar 
1707071d4279SBram Moolenaar 	if (fp->fd_len > curwin->w_p_fml)
1708071d4279SBram Moolenaar 	    fp->fd_small = FALSE;
1709071d4279SBram Moolenaar 	else
1710071d4279SBram Moolenaar 	{
1711071d4279SBram Moolenaar 	    count = 0;
1712071d4279SBram Moolenaar 	    for (n = 0; n < fp->fd_len; ++n)
1713071d4279SBram Moolenaar 	    {
1714071d4279SBram Moolenaar 		count += plines_win_nofold(wp, fp->fd_top + lnum_off + n);
1715071d4279SBram Moolenaar 		if (count > curwin->w_p_fml)
1716071d4279SBram Moolenaar 		{
1717071d4279SBram Moolenaar 		    fp->fd_small = FALSE;
1718071d4279SBram Moolenaar 		    return;
1719071d4279SBram Moolenaar 		}
1720071d4279SBram Moolenaar 	    }
1721071d4279SBram Moolenaar 	    fp->fd_small = TRUE;
1722071d4279SBram Moolenaar 	}
1723071d4279SBram Moolenaar     }
1724071d4279SBram Moolenaar }
1725071d4279SBram Moolenaar 
1726217e1b83SBram Moolenaar // setSmallMaybe() {{{2
1727071d4279SBram Moolenaar /*
1728071d4279SBram Moolenaar  * Set small flags in "gap" to MAYBE.
1729071d4279SBram Moolenaar  */
1730071d4279SBram Moolenaar     static void
173166f948e9SBram Moolenaar setSmallMaybe(garray_T *gap)
1732071d4279SBram Moolenaar {
1733071d4279SBram Moolenaar     int		i;
1734071d4279SBram Moolenaar     fold_T	*fp;
1735071d4279SBram Moolenaar 
1736071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
1737071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
1738071d4279SBram Moolenaar 	fp[i].fd_small = MAYBE;
1739071d4279SBram Moolenaar }
1740071d4279SBram Moolenaar 
1741217e1b83SBram Moolenaar // foldCreateMarkers() {{{2
1742071d4279SBram Moolenaar /*
1743071d4279SBram Moolenaar  * Create a fold from line "start" to line "end" (inclusive) in the current
1744071d4279SBram Moolenaar  * window by adding markers.
1745071d4279SBram Moolenaar  */
1746071d4279SBram Moolenaar     static void
174766f948e9SBram Moolenaar foldCreateMarkers(linenr_T start, linenr_T end)
1748071d4279SBram Moolenaar {
1749071d4279SBram Moolenaar     if (!curbuf->b_p_ma)
1750071d4279SBram Moolenaar     {
1751f9e3e09fSBram Moolenaar 	emsg(_(e_modifiable));
1752071d4279SBram Moolenaar 	return;
1753071d4279SBram Moolenaar     }
1754071d4279SBram Moolenaar     parseMarker(curwin);
1755071d4279SBram Moolenaar 
1756071d4279SBram Moolenaar     foldAddMarker(start, curwin->w_p_fmr, foldstartmarkerlen);
1757071d4279SBram Moolenaar     foldAddMarker(end, foldendmarker, foldendmarkerlen);
1758071d4279SBram Moolenaar 
1759217e1b83SBram Moolenaar     // Update both changes here, to avoid all folds after the start are
1760217e1b83SBram Moolenaar     // changed when the start marker is inserted and the end isn't.
1761071d4279SBram Moolenaar     changed_lines(start, (colnr_T)0, end, 0L);
1762071d4279SBram Moolenaar }
1763071d4279SBram Moolenaar 
1764217e1b83SBram Moolenaar // foldAddMarker() {{{2
1765071d4279SBram Moolenaar /*
1766071d4279SBram Moolenaar  * Add "marker[markerlen]" in 'commentstring' to line "lnum".
1767071d4279SBram Moolenaar  */
1768071d4279SBram Moolenaar     static void
176966f948e9SBram Moolenaar foldAddMarker(linenr_T lnum, char_u *marker, int markerlen)
1770071d4279SBram Moolenaar {
1771071d4279SBram Moolenaar     char_u	*cms = curbuf->b_p_cms;
1772071d4279SBram Moolenaar     char_u	*line;
1773071d4279SBram Moolenaar     int		line_len;
1774071d4279SBram Moolenaar     char_u	*newline;
1775071d4279SBram Moolenaar     char_u	*p = (char_u *)strstr((char *)curbuf->b_p_cms, "%s");
1776025a6b70SBram Moolenaar     int		line_is_comment = FALSE;
1777071d4279SBram Moolenaar 
1778217e1b83SBram Moolenaar     // Allocate a new line: old-line + 'cms'-start + marker + 'cms'-end
1779071d4279SBram Moolenaar     line = ml_get(lnum);
1780071d4279SBram Moolenaar     line_len = (int)STRLEN(line);
1781071d4279SBram Moolenaar 
1782071d4279SBram Moolenaar     if (u_save(lnum - 1, lnum + 1) == OK)
1783071d4279SBram Moolenaar     {
1784217e1b83SBram Moolenaar 	// Check if the line ends with an unclosed comment
1785025a6b70SBram Moolenaar 	(void)skip_comment(line, FALSE, FALSE, &line_is_comment);
1786964b3746SBram Moolenaar 	newline = alloc(line_len + markerlen + STRLEN(cms) + 1);
1787071d4279SBram Moolenaar 	if (newline == NULL)
1788071d4279SBram Moolenaar 	    return;
1789071d4279SBram Moolenaar 	STRCPY(newline, line);
1790217e1b83SBram Moolenaar 	// Append the marker to the end of the line
1791025a6b70SBram Moolenaar 	if (p == NULL || line_is_comment)
1792ce0842a6SBram Moolenaar 	    vim_strncpy(newline + line_len, marker, markerlen);
1793071d4279SBram Moolenaar 	else
1794071d4279SBram Moolenaar 	{
1795071d4279SBram Moolenaar 	    STRCPY(newline + line_len, cms);
1796071d4279SBram Moolenaar 	    STRNCPY(newline + line_len + (p - cms), marker, markerlen);
1797071d4279SBram Moolenaar 	    STRCPY(newline + line_len + (p - cms) + markerlen, p + 2);
1798071d4279SBram Moolenaar 	}
1799071d4279SBram Moolenaar 
1800071d4279SBram Moolenaar 	ml_replace(lnum, newline, FALSE);
1801071d4279SBram Moolenaar     }
1802071d4279SBram Moolenaar }
1803071d4279SBram Moolenaar 
1804217e1b83SBram Moolenaar // deleteFoldMarkers() {{{2
1805071d4279SBram Moolenaar /*
1806071d4279SBram Moolenaar  * Delete the markers for a fold, causing it to be deleted.
1807071d4279SBram Moolenaar  */
1808071d4279SBram Moolenaar     static void
180966f948e9SBram Moolenaar deleteFoldMarkers(
181066f948e9SBram Moolenaar     fold_T	*fp,
181166f948e9SBram Moolenaar     int		recursive,
1812217e1b83SBram Moolenaar     linenr_T	lnum_off)	// offset for fp->fd_top
1813071d4279SBram Moolenaar {
1814071d4279SBram Moolenaar     int		i;
1815071d4279SBram Moolenaar 
1816071d4279SBram Moolenaar     if (recursive)
1817071d4279SBram Moolenaar 	for (i = 0; i < fp->fd_nested.ga_len; ++i)
1818071d4279SBram Moolenaar 	    deleteFoldMarkers((fold_T *)fp->fd_nested.ga_data + i, TRUE,
1819071d4279SBram Moolenaar 						       lnum_off + fp->fd_top);
1820071d4279SBram Moolenaar     foldDelMarker(fp->fd_top + lnum_off, curwin->w_p_fmr, foldstartmarkerlen);
1821071d4279SBram Moolenaar     foldDelMarker(fp->fd_top + lnum_off + fp->fd_len - 1,
1822071d4279SBram Moolenaar 					     foldendmarker, foldendmarkerlen);
1823071d4279SBram Moolenaar }
1824071d4279SBram Moolenaar 
1825217e1b83SBram Moolenaar // foldDelMarker() {{{2
1826071d4279SBram Moolenaar /*
1827071d4279SBram Moolenaar  * Delete marker "marker[markerlen]" at the end of line "lnum".
1828071d4279SBram Moolenaar  * Delete 'commentstring' if it matches.
18299a4a8c4dSBram Moolenaar  * If the marker is not found, there is no error message.  Could be a missing
1830071d4279SBram Moolenaar  * close-marker.
1831071d4279SBram Moolenaar  */
1832071d4279SBram Moolenaar     static void
183366f948e9SBram Moolenaar foldDelMarker(linenr_T lnum, char_u *marker, int markerlen)
1834071d4279SBram Moolenaar {
1835071d4279SBram Moolenaar     char_u	*line;
1836071d4279SBram Moolenaar     char_u	*newline;
1837071d4279SBram Moolenaar     char_u	*p;
1838071d4279SBram Moolenaar     int		len;
1839071d4279SBram Moolenaar     char_u	*cms = curbuf->b_p_cms;
1840071d4279SBram Moolenaar     char_u	*cms2;
1841071d4279SBram Moolenaar 
18429a4a8c4dSBram Moolenaar     // end marker may be missing and fold extends below the last line
18439a4a8c4dSBram Moolenaar     if (lnum > curbuf->b_ml.ml_line_count)
18449a4a8c4dSBram Moolenaar 	return;
1845071d4279SBram Moolenaar     line = ml_get(lnum);
1846071d4279SBram Moolenaar     for (p = line; *p != NUL; ++p)
1847071d4279SBram Moolenaar 	if (STRNCMP(p, marker, markerlen) == 0)
1848071d4279SBram Moolenaar 	{
1849217e1b83SBram Moolenaar 	    // Found the marker, include a digit if it's there.
1850071d4279SBram Moolenaar 	    len = markerlen;
1851071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(p[len]))
1852071d4279SBram Moolenaar 		++len;
1853071d4279SBram Moolenaar 	    if (*cms != NUL)
1854071d4279SBram Moolenaar 	    {
1855217e1b83SBram Moolenaar 		// Also delete 'commentstring' if it matches.
1856071d4279SBram Moolenaar 		cms2 = (char_u *)strstr((char *)cms, "%s");
1857071d4279SBram Moolenaar 		if (p - line >= cms2 - cms
1858071d4279SBram Moolenaar 			&& STRNCMP(p - (cms2 - cms), cms, cms2 - cms) == 0
1859071d4279SBram Moolenaar 			&& STRNCMP(p + len, cms2 + 2, STRLEN(cms2 + 2)) == 0)
1860071d4279SBram Moolenaar 		{
1861071d4279SBram Moolenaar 		    p -= cms2 - cms;
1862071d4279SBram Moolenaar 		    len += (int)STRLEN(cms) - 2;
1863071d4279SBram Moolenaar 		}
1864071d4279SBram Moolenaar 	    }
1865071d4279SBram Moolenaar 	    if (u_save(lnum - 1, lnum + 1) == OK)
1866071d4279SBram Moolenaar 	    {
1867217e1b83SBram Moolenaar 		// Make new line: text-before-marker + text-after-marker
1868964b3746SBram Moolenaar 		newline = alloc(STRLEN(line) - len + 1);
1869071d4279SBram Moolenaar 		if (newline != NULL)
1870071d4279SBram Moolenaar 		{
1871071d4279SBram Moolenaar 		    STRNCPY(newline, line, p - line);
1872071d4279SBram Moolenaar 		    STRCPY(newline + (p - line), p + len);
1873071d4279SBram Moolenaar 		    ml_replace(lnum, newline, FALSE);
1874071d4279SBram Moolenaar 		}
1875071d4279SBram Moolenaar 	    }
1876071d4279SBram Moolenaar 	    break;
1877071d4279SBram Moolenaar 	}
1878071d4279SBram Moolenaar }
1879071d4279SBram Moolenaar 
1880217e1b83SBram Moolenaar // get_foldtext() {{{2
18817b0294cbSBram Moolenaar /*
18827b0294cbSBram Moolenaar  * Return the text for a closed fold at line "lnum", with last line "lnume".
1883ee695f78SBram Moolenaar  * When 'foldtext' isn't set puts the result in "buf[FOLD_TEXT_LEN]".
1884ee695f78SBram Moolenaar  * Otherwise the result is in allocated memory.
18857b0294cbSBram Moolenaar  */
18867b0294cbSBram Moolenaar     char_u *
188766f948e9SBram Moolenaar get_foldtext(
188866f948e9SBram Moolenaar     win_T	*wp,
188966f948e9SBram Moolenaar     linenr_T	lnum,
189066f948e9SBram Moolenaar     linenr_T	lnume,
189166f948e9SBram Moolenaar     foldinfo_T	*foldinfo,
189266f948e9SBram Moolenaar     char_u	*buf)
18937b0294cbSBram Moolenaar {
18947b0294cbSBram Moolenaar     char_u	*text = NULL;
18957b0294cbSBram Moolenaar #ifdef FEAT_EVAL
1896217e1b83SBram Moolenaar      // an error occurred when evaluating 'fdt' setting
1897dab38d51SBram Moolenaar     static int	    got_fdt_error = FALSE;
1898dab38d51SBram Moolenaar     int		    save_did_emsg = did_emsg;
1899dab38d51SBram Moolenaar     static win_T    *last_wp = NULL;
1900dab38d51SBram Moolenaar     static linenr_T last_lnum = 0;
1901dab38d51SBram Moolenaar 
1902dab38d51SBram Moolenaar     if (last_wp != wp || last_wp == NULL
1903dab38d51SBram Moolenaar 					|| last_lnum > lnum || last_lnum == 0)
1904217e1b83SBram Moolenaar 	// window changed, try evaluating foldtext setting once again
1905dab38d51SBram Moolenaar 	got_fdt_error = FALSE;
1906dab38d51SBram Moolenaar 
1907dab38d51SBram Moolenaar     if (!got_fdt_error)
1908217e1b83SBram Moolenaar 	// a previous error should not abort evaluating 'foldexpr'
1909dab38d51SBram Moolenaar 	did_emsg = FALSE;
1910dab38d51SBram Moolenaar 
19117b0294cbSBram Moolenaar     if (*wp->w_p_fdt != NUL)
19127b0294cbSBram Moolenaar     {
19135b88ba47SBram Moolenaar 	char_u	dashes[MAX_LEVEL + 2];
19147b0294cbSBram Moolenaar 	win_T	*save_curwin;
19157b0294cbSBram Moolenaar 	int	level;
19167b0294cbSBram Moolenaar 	char_u	*p;
19177b0294cbSBram Moolenaar 
1918217e1b83SBram Moolenaar 	// Set "v:foldstart" and "v:foldend".
19197b0294cbSBram Moolenaar 	set_vim_var_nr(VV_FOLDSTART, lnum);
19207b0294cbSBram Moolenaar 	set_vim_var_nr(VV_FOLDEND, lnume);
19217b0294cbSBram Moolenaar 
1922217e1b83SBram Moolenaar 	// Set "v:folddashes" to a string of "level" dashes.
1923217e1b83SBram Moolenaar 	// Set "v:foldlevel" to "level".
19247b0294cbSBram Moolenaar 	level = foldinfo->fi_level;
19255b88ba47SBram Moolenaar 	if (level > (int)sizeof(dashes) - 1)
19265b88ba47SBram Moolenaar 	    level = (int)sizeof(dashes) - 1;
19277b0294cbSBram Moolenaar 	vim_memset(dashes, '-', (size_t)level);
19287b0294cbSBram Moolenaar 	dashes[level] = NUL;
19297b0294cbSBram Moolenaar 	set_vim_var_string(VV_FOLDDASHES, dashes, -1);
19307b0294cbSBram Moolenaar 	set_vim_var_nr(VV_FOLDLEVEL, (long)level);
1931dab38d51SBram Moolenaar 
1932217e1b83SBram Moolenaar 	// skip evaluating foldtext on errors
1933dab38d51SBram Moolenaar 	if (!got_fdt_error)
1934dab38d51SBram Moolenaar 	{
19357b0294cbSBram Moolenaar 	    save_curwin = curwin;
19367b0294cbSBram Moolenaar 	    curwin = wp;
19377b0294cbSBram Moolenaar 	    curbuf = wp->w_buffer;
19387b0294cbSBram Moolenaar 
1939217e1b83SBram Moolenaar 	    ++emsg_silent; // handle exceptions, but don't display errors
1940b171fb17SBram Moolenaar 	    text = eval_to_string_safe(wp->w_p_fdt,
1941d1f56e68SBram Moolenaar 			 was_set_insecurely((char_u *)"foldtext", OPT_LOCAL));
1942dab38d51SBram Moolenaar 	    --emsg_silent;
1943dab38d51SBram Moolenaar 
1944dab38d51SBram Moolenaar 	    if (text == NULL || did_emsg)
1945dab38d51SBram Moolenaar 		got_fdt_error = TRUE;
19467b0294cbSBram Moolenaar 
19477b0294cbSBram Moolenaar 	    curwin = save_curwin;
19487b0294cbSBram Moolenaar 	    curbuf = curwin->w_buffer;
1949dab38d51SBram Moolenaar 	}
1950dab38d51SBram Moolenaar 	last_lnum = lnum;
1951dab38d51SBram Moolenaar 	last_wp   = wp;
19527b0294cbSBram Moolenaar 	set_vim_var_string(VV_FOLDDASHES, NULL, -1);
19537b0294cbSBram Moolenaar 
1954dab38d51SBram Moolenaar 	if (!did_emsg && save_did_emsg)
1955dab38d51SBram Moolenaar 	    did_emsg = save_did_emsg;
1956dab38d51SBram Moolenaar 
19577b0294cbSBram Moolenaar 	if (text != NULL)
19587b0294cbSBram Moolenaar 	{
1959217e1b83SBram Moolenaar 	    // Replace unprintable characters, if there are any.  But
1960217e1b83SBram Moolenaar 	    // replace a TAB with a space.
19617b0294cbSBram Moolenaar 	    for (p = text; *p != NUL; ++p)
19627b0294cbSBram Moolenaar 	    {
1963009b2592SBram Moolenaar 		int	len;
1964009b2592SBram Moolenaar 
19650fa313a7SBram Moolenaar 		if (has_mbyte && (len = (*mb_ptr2len)(p)) > 1)
19667b0294cbSBram Moolenaar 		{
19677b0294cbSBram Moolenaar 		    if (!vim_isprintc((*mb_ptr2char)(p)))
19687b0294cbSBram Moolenaar 			break;
19697b0294cbSBram Moolenaar 		    p += len - 1;
19707b0294cbSBram Moolenaar 		}
19717b0294cbSBram Moolenaar 		else
19727b0294cbSBram Moolenaar 		    if (*p == TAB)
19737b0294cbSBram Moolenaar 			*p = ' ';
19747b0294cbSBram Moolenaar 		    else if (ptr2cells(p) > 1)
19757b0294cbSBram Moolenaar 			break;
19767b0294cbSBram Moolenaar 	    }
19777b0294cbSBram Moolenaar 	    if (*p != NUL)
19787b0294cbSBram Moolenaar 	    {
19797b0294cbSBram Moolenaar 		p = transstr(text);
19807b0294cbSBram Moolenaar 		vim_free(text);
19817b0294cbSBram Moolenaar 		text = p;
19827b0294cbSBram Moolenaar 	    }
19837b0294cbSBram Moolenaar 	}
19847b0294cbSBram Moolenaar     }
19857b0294cbSBram Moolenaar     if (text == NULL)
19867b0294cbSBram Moolenaar #endif
19877b0294cbSBram Moolenaar     {
1988ee695f78SBram Moolenaar 	long count = (long)(lnume - lnum + 1);
1989ee695f78SBram Moolenaar 
1990ee695f78SBram Moolenaar 	vim_snprintf((char *)buf, FOLD_TEXT_LEN,
19911c465444SBram Moolenaar 		     NGETTEXT("+--%3ld line folded ",
1992ee695f78SBram Moolenaar 					       "+--%3ld lines folded ", count),
1993ee695f78SBram Moolenaar 		     count);
19947b0294cbSBram Moolenaar 	text = buf;
19957b0294cbSBram Moolenaar     }
19967b0294cbSBram Moolenaar     return text;
19977b0294cbSBram Moolenaar }
19987b0294cbSBram Moolenaar 
1999217e1b83SBram Moolenaar // foldtext_cleanup() {{{2
2000db022f3fSBram Moolenaar #ifdef FEAT_EVAL
2001071d4279SBram Moolenaar /*
2002071d4279SBram Moolenaar  * Remove 'foldmarker' and 'commentstring' from "str" (in-place).
2003071d4279SBram Moolenaar  */
2004db022f3fSBram Moolenaar     static void
200566f948e9SBram Moolenaar foldtext_cleanup(char_u *str)
2006071d4279SBram Moolenaar {
2007217e1b83SBram Moolenaar     char_u	*cms_start;	// first part or the whole comment
2008217e1b83SBram Moolenaar     int		cms_slen = 0;	// length of cms_start
2009217e1b83SBram Moolenaar     char_u	*cms_end;	// last part of the comment or NULL
2010217e1b83SBram Moolenaar     int		cms_elen = 0;	// length of cms_end
2011071d4279SBram Moolenaar     char_u	*s;
2012def9e829SBram Moolenaar     char_u	*p;
2013071d4279SBram Moolenaar     int		len;
2014071d4279SBram Moolenaar     int		did1 = FALSE;
2015071d4279SBram Moolenaar     int		did2 = FALSE;
2016071d4279SBram Moolenaar 
2017217e1b83SBram Moolenaar     // Ignore leading and trailing white space in 'commentstring'.
2018071d4279SBram Moolenaar     cms_start = skipwhite(curbuf->b_p_cms);
2019a93fa7eeSBram Moolenaar     cms_slen = (int)STRLEN(cms_start);
20201c465444SBram Moolenaar     while (cms_slen > 0 && VIM_ISWHITE(cms_start[cms_slen - 1]))
2021071d4279SBram Moolenaar 	--cms_slen;
2022071d4279SBram Moolenaar 
2023217e1b83SBram Moolenaar     // locate "%s" in 'commentstring', use the part before and after it.
2024071d4279SBram Moolenaar     cms_end = (char_u *)strstr((char *)cms_start, "%s");
2025071d4279SBram Moolenaar     if (cms_end != NULL)
2026071d4279SBram Moolenaar     {
2027a93fa7eeSBram Moolenaar 	cms_elen = cms_slen - (int)(cms_end - cms_start);
2028a93fa7eeSBram Moolenaar 	cms_slen = (int)(cms_end - cms_start);
2029071d4279SBram Moolenaar 
2030217e1b83SBram Moolenaar 	// exclude white space before "%s"
20311c465444SBram Moolenaar 	while (cms_slen > 0 && VIM_ISWHITE(cms_start[cms_slen - 1]))
2032071d4279SBram Moolenaar 	    --cms_slen;
2033071d4279SBram Moolenaar 
2034217e1b83SBram Moolenaar 	// skip "%s" and white space after it
2035071d4279SBram Moolenaar 	s = skipwhite(cms_end + 2);
2036a93fa7eeSBram Moolenaar 	cms_elen -= (int)(s - cms_end);
2037071d4279SBram Moolenaar 	cms_end = s;
2038071d4279SBram Moolenaar     }
2039071d4279SBram Moolenaar     parseMarker(curwin);
2040071d4279SBram Moolenaar 
2041071d4279SBram Moolenaar     for (s = str; *s != NUL; )
2042071d4279SBram Moolenaar     {
2043071d4279SBram Moolenaar 	len = 0;
2044071d4279SBram Moolenaar 	if (STRNCMP(s, curwin->w_p_fmr, foldstartmarkerlen) == 0)
2045071d4279SBram Moolenaar 	    len = foldstartmarkerlen;
2046071d4279SBram Moolenaar 	else if (STRNCMP(s, foldendmarker, foldendmarkerlen) == 0)
2047071d4279SBram Moolenaar 	    len = foldendmarkerlen;
2048def9e829SBram Moolenaar 	if (len > 0)
2049def9e829SBram Moolenaar 	{
2050071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(s[len]))
2051071d4279SBram Moolenaar 		++len;
2052def9e829SBram Moolenaar 
2053217e1b83SBram Moolenaar 	    // May remove 'commentstring' start.  Useful when it's a double
2054217e1b83SBram Moolenaar 	    // quote and we already removed a double quote.
20551c465444SBram Moolenaar 	    for (p = s; p > str && VIM_ISWHITE(p[-1]); --p)
2056def9e829SBram Moolenaar 		;
2057def9e829SBram Moolenaar 	    if (p >= str + cms_slen
2058def9e829SBram Moolenaar 			   && STRNCMP(p - cms_slen, cms_start, cms_slen) == 0)
2059def9e829SBram Moolenaar 	    {
2060a93fa7eeSBram Moolenaar 		len += (int)(s - p) + cms_slen;
2061def9e829SBram Moolenaar 		s = p - cms_slen;
2062def9e829SBram Moolenaar 	    }
2063071d4279SBram Moolenaar 	}
2064071d4279SBram Moolenaar 	else if (cms_end != NULL)
2065071d4279SBram Moolenaar 	{
2066def9e829SBram Moolenaar 	    if (!did1 && cms_slen > 0 && STRNCMP(s, cms_start, cms_slen) == 0)
2067071d4279SBram Moolenaar 	    {
2068071d4279SBram Moolenaar 		len = cms_slen;
2069071d4279SBram Moolenaar 		did1 = TRUE;
2070071d4279SBram Moolenaar 	    }
2071def9e829SBram Moolenaar 	    else if (!did2 && cms_elen > 0
2072def9e829SBram Moolenaar 					&& STRNCMP(s, cms_end, cms_elen) == 0)
2073071d4279SBram Moolenaar 	    {
2074071d4279SBram Moolenaar 		len = cms_elen;
2075071d4279SBram Moolenaar 		did2 = TRUE;
2076071d4279SBram Moolenaar 	    }
2077071d4279SBram Moolenaar 	}
2078071d4279SBram Moolenaar 	if (len != 0)
2079071d4279SBram Moolenaar 	{
20801c465444SBram Moolenaar 	    while (VIM_ISWHITE(s[len]))
2081071d4279SBram Moolenaar 		++len;
20828c8de839SBram Moolenaar 	    STRMOVE(s, s + len);
2083071d4279SBram Moolenaar 	}
2084071d4279SBram Moolenaar 	else
2085071d4279SBram Moolenaar 	{
208691acfffcSBram Moolenaar 	    MB_PTR_ADV(s);
2087071d4279SBram Moolenaar 	}
2088071d4279SBram Moolenaar     }
2089071d4279SBram Moolenaar }
2090db022f3fSBram Moolenaar #endif
2091071d4279SBram Moolenaar 
2092217e1b83SBram Moolenaar // Folding by indent, expr, marker and syntax. {{{1
2093217e1b83SBram Moolenaar // Define "fline_T", passed to get fold level for a line. {{{2
2094071d4279SBram Moolenaar typedef struct
2095071d4279SBram Moolenaar {
2096217e1b83SBram Moolenaar     win_T	*wp;		// window
2097217e1b83SBram Moolenaar     linenr_T	lnum;		// current line number
2098217e1b83SBram Moolenaar     linenr_T	off;		// offset between lnum and real line number
2099217e1b83SBram Moolenaar     linenr_T	lnum_save;	// line nr used by foldUpdateIEMSRecurse()
2100217e1b83SBram Moolenaar     int		lvl;		// current level (-1 for undefined)
2101217e1b83SBram Moolenaar     int		lvl_next;	// level used for next line
2102217e1b83SBram Moolenaar     int		start;		// number of folds that are forced to start at
2103217e1b83SBram Moolenaar 				// this line.
2104217e1b83SBram Moolenaar     int		end;		// level of fold that is forced to end below
2105217e1b83SBram Moolenaar 				// this line
2106217e1b83SBram Moolenaar     int		had_end;	// level of fold that is forced to end above
2107217e1b83SBram Moolenaar 				// this line (copy of "end" of prev. line)
2108071d4279SBram Moolenaar } fline_T;
2109071d4279SBram Moolenaar 
2110217e1b83SBram Moolenaar // Flag is set when redrawing is needed.
2111071d4279SBram Moolenaar static int fold_changed;
2112071d4279SBram Moolenaar 
2113217e1b83SBram Moolenaar // Function declarations. {{{2
2114d99df423SBram 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);
2115d25c16e2SBram Moolenaar static int foldInsert(garray_T *gap, int i);
2116d25c16e2SBram Moolenaar static void foldSplit(garray_T *gap, int i, linenr_T top, linenr_T bot);
2117d25c16e2SBram Moolenaar static void foldRemove(garray_T *gap, linenr_T top, linenr_T bot);
2118d25c16e2SBram Moolenaar static void foldMerge(fold_T *fp1, garray_T *gap, fold_T *fp2);
2119d25c16e2SBram Moolenaar static void foldlevelIndent(fline_T *flp);
2120071d4279SBram Moolenaar #ifdef FEAT_DIFF
2121d25c16e2SBram Moolenaar static void foldlevelDiff(fline_T *flp);
2122071d4279SBram Moolenaar #endif
2123d25c16e2SBram Moolenaar static void foldlevelExpr(fline_T *flp);
2124d25c16e2SBram Moolenaar static void foldlevelMarker(fline_T *flp);
2125d25c16e2SBram Moolenaar static void foldlevelSyntax(fline_T *flp);
2126071d4279SBram Moolenaar 
2127217e1b83SBram Moolenaar // foldUpdateIEMS() {{{2
2128071d4279SBram Moolenaar /*
2129071d4279SBram Moolenaar  * Update the folding for window "wp", at least from lines "top" to "bot".
2130071d4279SBram Moolenaar  * Return TRUE if any folds did change.
2131071d4279SBram Moolenaar  */
2132071d4279SBram Moolenaar     static void
213366f948e9SBram Moolenaar foldUpdateIEMS(win_T *wp, linenr_T top, linenr_T bot)
2134071d4279SBram Moolenaar {
2135071d4279SBram Moolenaar     linenr_T	start;
2136071d4279SBram Moolenaar     linenr_T	end;
2137071d4279SBram Moolenaar     fline_T	fline;
2138d99df423SBram Moolenaar     void	(*getlevel)(fline_T *);
2139071d4279SBram Moolenaar     int		level;
2140071d4279SBram Moolenaar     fold_T	*fp;
2141071d4279SBram Moolenaar 
2142217e1b83SBram Moolenaar     // Avoid problems when being called recursively.
2143071d4279SBram Moolenaar     if (invalid_top != (linenr_T)0)
2144071d4279SBram Moolenaar 	return;
2145071d4279SBram Moolenaar 
2146071d4279SBram Moolenaar     if (wp->w_foldinvalid)
2147071d4279SBram Moolenaar     {
2148217e1b83SBram Moolenaar 	// Need to update all folds.
2149071d4279SBram Moolenaar 	top = 1;
2150071d4279SBram Moolenaar 	bot = wp->w_buffer->b_ml.ml_line_count;
2151071d4279SBram Moolenaar 	wp->w_foldinvalid = FALSE;
2152071d4279SBram Moolenaar 
2153217e1b83SBram Moolenaar 	// Mark all folds a maybe-small.
2154071d4279SBram Moolenaar 	setSmallMaybe(&wp->w_folds);
2155071d4279SBram Moolenaar     }
2156071d4279SBram Moolenaar 
2157071d4279SBram Moolenaar #ifdef FEAT_DIFF
2158217e1b83SBram Moolenaar     // add the context for "diff" folding
2159071d4279SBram Moolenaar     if (foldmethodIsDiff(wp))
2160071d4279SBram Moolenaar     {
2161071d4279SBram Moolenaar 	if (top > diff_context)
2162071d4279SBram Moolenaar 	    top -= diff_context;
2163071d4279SBram Moolenaar 	else
2164071d4279SBram Moolenaar 	    top = 1;
2165071d4279SBram Moolenaar 	bot += diff_context;
2166071d4279SBram Moolenaar     }
2167071d4279SBram Moolenaar #endif
2168071d4279SBram Moolenaar 
2169217e1b83SBram Moolenaar     // When deleting lines at the end of the buffer "top" can be past the end
2170217e1b83SBram Moolenaar     // of the buffer.
2171071d4279SBram Moolenaar     if (top > wp->w_buffer->b_ml.ml_line_count)
2172071d4279SBram Moolenaar 	top = wp->w_buffer->b_ml.ml_line_count;
2173071d4279SBram Moolenaar 
2174071d4279SBram Moolenaar     fold_changed = FALSE;
2175071d4279SBram Moolenaar     fline.wp = wp;
2176071d4279SBram Moolenaar     fline.off = 0;
2177071d4279SBram Moolenaar     fline.lvl = 0;
2178071d4279SBram Moolenaar     fline.lvl_next = -1;
2179071d4279SBram Moolenaar     fline.start = 0;
2180071d4279SBram Moolenaar     fline.end = MAX_LEVEL + 1;
2181071d4279SBram Moolenaar     fline.had_end = MAX_LEVEL + 1;
2182071d4279SBram Moolenaar 
2183071d4279SBram Moolenaar     invalid_top = top;
2184071d4279SBram Moolenaar     invalid_bot = bot;
2185071d4279SBram Moolenaar 
2186071d4279SBram Moolenaar     if (foldmethodIsMarker(wp))
2187071d4279SBram Moolenaar     {
2188071d4279SBram Moolenaar 	getlevel = foldlevelMarker;
2189071d4279SBram Moolenaar 
2190217e1b83SBram Moolenaar 	// Init marker variables to speed up foldlevelMarker().
2191071d4279SBram Moolenaar 	parseMarker(wp);
2192071d4279SBram Moolenaar 
2193217e1b83SBram Moolenaar 	// Need to get the level of the line above top, it is used if there is
2194217e1b83SBram Moolenaar 	// no marker at the top.
2195071d4279SBram Moolenaar 	if (top > 1)
2196071d4279SBram Moolenaar 	{
2197217e1b83SBram Moolenaar 	    // Get the fold level at top - 1.
2198071d4279SBram Moolenaar 	    level = foldLevelWin(wp, top - 1);
2199071d4279SBram Moolenaar 
2200217e1b83SBram Moolenaar 	    // The fold may end just above the top, check for that.
2201071d4279SBram Moolenaar 	    fline.lnum = top - 1;
2202071d4279SBram Moolenaar 	    fline.lvl = level;
2203071d4279SBram Moolenaar 	    getlevel(&fline);
2204071d4279SBram Moolenaar 
2205217e1b83SBram Moolenaar 	    // If a fold started here, we already had the level, if it stops
2206217e1b83SBram Moolenaar 	    // here, we need to use lvl_next.  Could also start and end a fold
2207217e1b83SBram Moolenaar 	    // in the same line.
2208071d4279SBram Moolenaar 	    if (fline.lvl > level)
2209071d4279SBram Moolenaar 		fline.lvl = level - (fline.lvl - fline.lvl_next);
2210071d4279SBram Moolenaar 	    else
2211071d4279SBram Moolenaar 		fline.lvl = fline.lvl_next;
2212071d4279SBram Moolenaar 	}
2213071d4279SBram Moolenaar 	fline.lnum = top;
2214071d4279SBram Moolenaar 	getlevel(&fline);
2215071d4279SBram Moolenaar     }
2216071d4279SBram Moolenaar     else
2217071d4279SBram Moolenaar     {
2218071d4279SBram Moolenaar 	fline.lnum = top;
2219071d4279SBram Moolenaar 	if (foldmethodIsExpr(wp))
2220071d4279SBram Moolenaar 	{
2221071d4279SBram Moolenaar 	    getlevel = foldlevelExpr;
2222217e1b83SBram Moolenaar 	    // start one line back, because a "<1" may indicate the end of a
2223217e1b83SBram Moolenaar 	    // fold in the topline
2224071d4279SBram Moolenaar 	    if (top > 1)
2225071d4279SBram Moolenaar 		--fline.lnum;
2226071d4279SBram Moolenaar 	}
2227071d4279SBram Moolenaar 	else if (foldmethodIsSyntax(wp))
2228071d4279SBram Moolenaar 	    getlevel = foldlevelSyntax;
2229071d4279SBram Moolenaar #ifdef FEAT_DIFF
2230071d4279SBram Moolenaar 	else if (foldmethodIsDiff(wp))
2231071d4279SBram Moolenaar 	    getlevel = foldlevelDiff;
2232071d4279SBram Moolenaar #endif
2233071d4279SBram Moolenaar 	else
2234071d4279SBram Moolenaar 	    getlevel = foldlevelIndent;
2235071d4279SBram Moolenaar 
2236217e1b83SBram Moolenaar 	// Backup to a line for which the fold level is defined.  Since it's
2237217e1b83SBram Moolenaar 	// always defined for line one, we will stop there.
2238071d4279SBram Moolenaar 	fline.lvl = -1;
2239071d4279SBram Moolenaar 	for ( ; !got_int; --fline.lnum)
2240071d4279SBram Moolenaar 	{
2241217e1b83SBram Moolenaar 	    // Reset lvl_next each time, because it will be set to a value for
2242217e1b83SBram Moolenaar 	    // the next line, but we search backwards here.
2243071d4279SBram Moolenaar 	    fline.lvl_next = -1;
2244071d4279SBram Moolenaar 	    getlevel(&fline);
2245071d4279SBram Moolenaar 	    if (fline.lvl >= 0)
2246071d4279SBram Moolenaar 		break;
2247071d4279SBram Moolenaar 	}
2248071d4279SBram Moolenaar     }
2249071d4279SBram Moolenaar 
2250ec98647bSBram Moolenaar     /*
2251ec98647bSBram Moolenaar      * If folding is defined by the syntax, it is possible that a change in
2252ec98647bSBram Moolenaar      * one line will cause all sub-folds of the current fold to change (e.g.,
2253ec98647bSBram Moolenaar      * closing a C-style comment can cause folds in the subsequent lines to
2254ec98647bSBram Moolenaar      * appear). To take that into account we should adjust the value of "bot"
2255ec98647bSBram Moolenaar      * to point to the end of the current fold:
2256ec98647bSBram Moolenaar      */
2257ec98647bSBram Moolenaar     if (foldlevelSyntax == getlevel)
2258ec98647bSBram Moolenaar     {
2259ec98647bSBram Moolenaar 	garray_T *gap = &wp->w_folds;
2260f506c5bbSBram Moolenaar 	fold_T	 *fpn = NULL;
2261ec98647bSBram Moolenaar 	int	  current_fdl = 0;
2262ec98647bSBram Moolenaar 	linenr_T  fold_start_lnum = 0;
2263ec98647bSBram Moolenaar 	linenr_T  lnum_rel = fline.lnum;
2264ec98647bSBram Moolenaar 
2265ec98647bSBram Moolenaar 	while (current_fdl < fline.lvl)
2266ec98647bSBram Moolenaar 	{
2267f506c5bbSBram Moolenaar 	    if (!foldFind(gap, lnum_rel, &fpn))
2268ec98647bSBram Moolenaar 		break;
2269ec98647bSBram Moolenaar 	    ++current_fdl;
2270ec98647bSBram Moolenaar 
2271f506c5bbSBram Moolenaar 	    fold_start_lnum += fpn->fd_top;
2272f506c5bbSBram Moolenaar 	    gap = &fpn->fd_nested;
2273f506c5bbSBram Moolenaar 	    lnum_rel -= fpn->fd_top;
2274ec98647bSBram Moolenaar 	}
2275f506c5bbSBram Moolenaar 	if (fpn != NULL && current_fdl == fline.lvl)
2276ec98647bSBram Moolenaar 	{
2277f506c5bbSBram Moolenaar 	    linenr_T fold_end_lnum = fold_start_lnum + fpn->fd_len;
2278ec98647bSBram Moolenaar 
2279ec98647bSBram Moolenaar 	    if (fold_end_lnum > bot)
2280ec98647bSBram Moolenaar 		bot = fold_end_lnum;
2281ec98647bSBram Moolenaar 	}
2282ec98647bSBram Moolenaar     }
2283ec98647bSBram Moolenaar 
2284071d4279SBram Moolenaar     start = fline.lnum;
2285071d4279SBram Moolenaar     end = bot;
2286217e1b83SBram Moolenaar     // Do at least one line.
2287071d4279SBram Moolenaar     if (start > end && end < wp->w_buffer->b_ml.ml_line_count)
2288071d4279SBram Moolenaar 	end = start;
2289071d4279SBram Moolenaar     while (!got_int)
2290071d4279SBram Moolenaar     {
2291217e1b83SBram Moolenaar 	// Always stop at the end of the file ("end" can be past the end of
2292217e1b83SBram Moolenaar 	// the file).
2293071d4279SBram Moolenaar 	if (fline.lnum > wp->w_buffer->b_ml.ml_line_count)
2294071d4279SBram Moolenaar 	    break;
2295071d4279SBram Moolenaar 	if (fline.lnum > end)
2296071d4279SBram Moolenaar 	{
2297217e1b83SBram Moolenaar 	    // For "marker", "expr"  and "syntax"  methods: If a change caused
2298217e1b83SBram Moolenaar 	    // a fold to be removed, we need to continue at least until where
2299217e1b83SBram Moolenaar 	    // it ended.
2300071d4279SBram Moolenaar 	    if (getlevel != foldlevelMarker
2301071d4279SBram Moolenaar 		    && getlevel != foldlevelSyntax
2302071d4279SBram Moolenaar 		    && getlevel != foldlevelExpr)
2303071d4279SBram Moolenaar 		break;
2304071d4279SBram Moolenaar 	    if ((start <= end
2305071d4279SBram Moolenaar 			&& foldFind(&wp->w_folds, end, &fp)
2306071d4279SBram Moolenaar 			&& fp->fd_top + fp->fd_len - 1 > end)
2307071d4279SBram Moolenaar 		    || (fline.lvl == 0
2308071d4279SBram Moolenaar 			&& foldFind(&wp->w_folds, fline.lnum, &fp)
2309071d4279SBram Moolenaar 			&& fp->fd_top < fline.lnum))
2310071d4279SBram Moolenaar 		end = fp->fd_top + fp->fd_len - 1;
2311071d4279SBram Moolenaar 	    else if (getlevel == foldlevelSyntax
2312071d4279SBram Moolenaar 		    && foldLevelWin(wp, fline.lnum) != fline.lvl)
2313217e1b83SBram Moolenaar 		// For "syntax" method: Compare the foldlevel that the syntax
2314217e1b83SBram Moolenaar 		// tells us to the foldlevel from the existing folds.  If they
2315217e1b83SBram Moolenaar 		// don't match continue updating folds.
2316071d4279SBram Moolenaar 		end = fline.lnum;
2317071d4279SBram Moolenaar 	    else
2318071d4279SBram Moolenaar 		break;
2319071d4279SBram Moolenaar 	}
2320071d4279SBram Moolenaar 
2321217e1b83SBram Moolenaar 	// A level 1 fold starts at a line with foldlevel > 0.
2322071d4279SBram Moolenaar 	if (fline.lvl > 0)
2323071d4279SBram Moolenaar 	{
2324071d4279SBram Moolenaar 	    invalid_top = fline.lnum;
2325071d4279SBram Moolenaar 	    invalid_bot = end;
2326071d4279SBram Moolenaar 	    end = foldUpdateIEMSRecurse(&wp->w_folds,
2327071d4279SBram Moolenaar 				   1, start, &fline, getlevel, end, FD_LEVEL);
2328071d4279SBram Moolenaar 	    start = fline.lnum;
2329071d4279SBram Moolenaar 	}
2330071d4279SBram Moolenaar 	else
2331071d4279SBram Moolenaar 	{
2332071d4279SBram Moolenaar 	    if (fline.lnum == wp->w_buffer->b_ml.ml_line_count)
2333071d4279SBram Moolenaar 		break;
2334071d4279SBram Moolenaar 	    ++fline.lnum;
2335071d4279SBram Moolenaar 	    fline.lvl = fline.lvl_next;
2336071d4279SBram Moolenaar 	    getlevel(&fline);
2337071d4279SBram Moolenaar 	}
2338071d4279SBram Moolenaar     }
2339071d4279SBram Moolenaar 
2340217e1b83SBram Moolenaar     // There can't be any folds from start until end now.
2341071d4279SBram Moolenaar     foldRemove(&wp->w_folds, start, end);
2342071d4279SBram Moolenaar 
2343217e1b83SBram Moolenaar     // If some fold changed, need to redraw and position cursor.
2344071d4279SBram Moolenaar     if (fold_changed && wp->w_p_fen)
2345e3b3f28dSBram Moolenaar 	changed_window_setting_win(wp);
2346071d4279SBram Moolenaar 
2347217e1b83SBram Moolenaar     // If we updated folds past "bot", need to redraw more lines.  Don't do
2348217e1b83SBram Moolenaar     // this in other situations, the changed lines will be redrawn anyway and
2349217e1b83SBram Moolenaar     // this method can cause the whole window to be updated.
2350071d4279SBram Moolenaar     if (end != bot)
2351071d4279SBram Moolenaar     {
2352071d4279SBram Moolenaar 	if (wp->w_redraw_top == 0 || wp->w_redraw_top > top)
2353071d4279SBram Moolenaar 	    wp->w_redraw_top = top;
2354071d4279SBram Moolenaar 	if (wp->w_redraw_bot < end)
2355071d4279SBram Moolenaar 	    wp->w_redraw_bot = end;
2356071d4279SBram Moolenaar     }
2357071d4279SBram Moolenaar 
2358071d4279SBram Moolenaar     invalid_top = (linenr_T)0;
2359071d4279SBram Moolenaar }
2360071d4279SBram Moolenaar 
2361217e1b83SBram Moolenaar // foldUpdateIEMSRecurse() {{{2
2362071d4279SBram Moolenaar /*
2363071d4279SBram Moolenaar  * Update a fold that starts at "flp->lnum".  At this line there is always a
2364071d4279SBram Moolenaar  * valid foldlevel, and its level >= "level".
2365071d4279SBram Moolenaar  * "flp" is valid for "flp->lnum" when called and it's valid when returning.
2366071d4279SBram Moolenaar  * "flp->lnum" is set to the lnum just below the fold, if it ends before
2367071d4279SBram Moolenaar  * "bot", it's "bot" plus one if the fold continues and it's bigger when using
2368071d4279SBram Moolenaar  * the marker method and a text change made following folds to change.
2369071d4279SBram Moolenaar  * When returning, "flp->lnum_save" is the line number that was used to get
2370071d4279SBram Moolenaar  * the level when the level at "flp->lnum" is invalid.
2371071d4279SBram Moolenaar  * Remove any folds from "startlnum" up to here at this level.
2372071d4279SBram Moolenaar  * Recursively update nested folds.
2373071d4279SBram Moolenaar  * Below line "bot" there are no changes in the text.
2374071d4279SBram Moolenaar  * "flp->lnum", "flp->lnum_save" and "bot" are relative to the start of the
2375071d4279SBram Moolenaar  * outer fold.
2376071d4279SBram Moolenaar  * "flp->off" is the offset to the real line number in the buffer.
2377071d4279SBram Moolenaar  *
2378071d4279SBram Moolenaar  * All this would be a lot simpler if all folds in the range would be deleted
2379e37d50a5SBram Moolenaar  * and then created again.  But we would lose all information about the
2380071d4279SBram Moolenaar  * folds, even when making changes that don't affect the folding (e.g. "vj~").
2381071d4279SBram Moolenaar  *
2382071d4279SBram Moolenaar  * Returns bot, which may have been increased for lines that also need to be
2383071d4279SBram Moolenaar  * updated as a result of a detected change in the fold.
2384071d4279SBram Moolenaar  */
2385071d4279SBram Moolenaar     static linenr_T
238666f948e9SBram Moolenaar foldUpdateIEMSRecurse(
238766f948e9SBram Moolenaar     garray_T	*gap,
238866f948e9SBram Moolenaar     int		level,
238966f948e9SBram Moolenaar     linenr_T	startlnum,
239066f948e9SBram Moolenaar     fline_T	*flp,
239166f948e9SBram Moolenaar     void	(*getlevel)(fline_T *),
239266f948e9SBram Moolenaar     linenr_T	bot,
2393217e1b83SBram Moolenaar     int		topflags)	// flags used by containing fold
2394071d4279SBram Moolenaar {
2395071d4279SBram Moolenaar     linenr_T	ll;
2396071d4279SBram Moolenaar     fold_T	*fp = NULL;
2397071d4279SBram Moolenaar     fold_T	*fp2;
2398071d4279SBram Moolenaar     int		lvl = level;
2399071d4279SBram Moolenaar     linenr_T	startlnum2 = startlnum;
2400217e1b83SBram Moolenaar     linenr_T	firstlnum = flp->lnum;	// first lnum we got
2401071d4279SBram Moolenaar     int		i;
2402071d4279SBram Moolenaar     int		finish = FALSE;
2403071d4279SBram Moolenaar     linenr_T	linecount = flp->wp->w_buffer->b_ml.ml_line_count - flp->off;
2404071d4279SBram Moolenaar     int		concat;
2405071d4279SBram Moolenaar 
2406071d4279SBram Moolenaar     /*
2407071d4279SBram Moolenaar      * If using the marker method, the start line is not the start of a fold
2408071d4279SBram Moolenaar      * at the level we're dealing with and the level is non-zero, we must use
2409071d4279SBram Moolenaar      * the previous fold.  But ignore a fold that starts at or below
2410071d4279SBram Moolenaar      * startlnum, it must be deleted.
2411071d4279SBram Moolenaar      */
2412071d4279SBram Moolenaar     if (getlevel == foldlevelMarker && flp->start <= flp->lvl - level
2413071d4279SBram Moolenaar 							      && flp->lvl > 0)
2414071d4279SBram Moolenaar     {
2415cde88547SBram Moolenaar 	(void)foldFind(gap, startlnum - 1, &fp);
2416071d4279SBram Moolenaar 	if (fp >= ((fold_T *)gap->ga_data) + gap->ga_len
2417071d4279SBram Moolenaar 						   || fp->fd_top >= startlnum)
2418071d4279SBram Moolenaar 	    fp = NULL;
2419071d4279SBram Moolenaar     }
2420071d4279SBram Moolenaar 
2421071d4279SBram Moolenaar     /*
2422071d4279SBram Moolenaar      * Loop over all lines in this fold, or until "bot" is hit.
2423071d4279SBram Moolenaar      * Handle nested folds inside of this fold.
2424071d4279SBram Moolenaar      * "flp->lnum" is the current line.  When finding the end of the fold, it
2425071d4279SBram Moolenaar      * is just below the end of the fold.
2426071d4279SBram Moolenaar      * "*flp" contains the level of the line "flp->lnum" or a following one if
2427071d4279SBram Moolenaar      * there are lines with an invalid fold level.  "flp->lnum_save" is the
2428071d4279SBram Moolenaar      * line number that was used to get the fold level (below "flp->lnum" when
2429071d4279SBram Moolenaar      * it has an invalid fold level).  When called the fold level is always
2430071d4279SBram Moolenaar      * valid, thus "flp->lnum_save" is equal to "flp->lnum".
2431071d4279SBram Moolenaar      */
2432071d4279SBram Moolenaar     flp->lnum_save = flp->lnum;
2433071d4279SBram Moolenaar     while (!got_int)
2434071d4279SBram Moolenaar     {
2435217e1b83SBram Moolenaar 	// Updating folds can be slow, check for CTRL-C.
2436071d4279SBram Moolenaar 	line_breakcheck();
2437071d4279SBram Moolenaar 
2438217e1b83SBram Moolenaar 	// Set "lvl" to the level of line "flp->lnum".  When flp->start is set
2439217e1b83SBram Moolenaar 	// and after the first line of the fold, set the level to zero to
2440217e1b83SBram Moolenaar 	// force the fold to end.  Do the same when had_end is set: Previous
2441217e1b83SBram Moolenaar 	// line was marked as end of a fold.
2442071d4279SBram Moolenaar 	lvl = flp->lvl;
2443071d4279SBram Moolenaar 	if (lvl > MAX_LEVEL)
2444071d4279SBram Moolenaar 	    lvl = MAX_LEVEL;
2445071d4279SBram Moolenaar 	if (flp->lnum > firstlnum
2446071d4279SBram Moolenaar 		&& (level > lvl - flp->start || level >= flp->had_end))
2447071d4279SBram Moolenaar 	    lvl = 0;
2448071d4279SBram Moolenaar 
2449071d4279SBram Moolenaar 	if (flp->lnum > bot && !finish && fp != NULL)
2450071d4279SBram Moolenaar 	{
2451217e1b83SBram Moolenaar 	    // For "marker" and "syntax" methods:
2452217e1b83SBram Moolenaar 	    // - If a change caused a nested fold to be removed, we need to
2453217e1b83SBram Moolenaar 	    //   delete it and continue at least until where it ended.
2454217e1b83SBram Moolenaar 	    // - If a change caused a nested fold to be created, or this fold
2455217e1b83SBram Moolenaar 	    //   to continue below its original end, need to finish this fold.
2456071d4279SBram Moolenaar 	    if (getlevel != foldlevelMarker
2457071d4279SBram Moolenaar 		    && getlevel != foldlevelExpr
2458071d4279SBram Moolenaar 		    && getlevel != foldlevelSyntax)
2459071d4279SBram Moolenaar 		break;
2460071d4279SBram Moolenaar 	    i = 0;
2461071d4279SBram Moolenaar 	    fp2 = fp;
2462071d4279SBram Moolenaar 	    if (lvl >= level)
2463071d4279SBram Moolenaar 	    {
2464217e1b83SBram Moolenaar 		// Compute how deep the folds currently are, if it's deeper
2465217e1b83SBram Moolenaar 		// than "lvl" then some must be deleted, need to update
2466217e1b83SBram Moolenaar 		// at least one nested fold.
2467071d4279SBram Moolenaar 		ll = flp->lnum - fp->fd_top;
2468071d4279SBram Moolenaar 		while (foldFind(&fp2->fd_nested, ll, &fp2))
2469071d4279SBram Moolenaar 		{
2470071d4279SBram Moolenaar 		    ++i;
2471071d4279SBram Moolenaar 		    ll -= fp2->fd_top;
2472071d4279SBram Moolenaar 		}
2473071d4279SBram Moolenaar 	    }
2474071d4279SBram Moolenaar 	    if (lvl < level + i)
2475071d4279SBram Moolenaar 	    {
2476cde88547SBram Moolenaar 		(void)foldFind(&fp->fd_nested, flp->lnum - fp->fd_top, &fp2);
2477071d4279SBram Moolenaar 		if (fp2 != NULL)
2478071d4279SBram Moolenaar 		    bot = fp2->fd_top + fp2->fd_len - 1 + fp->fd_top;
2479071d4279SBram Moolenaar 	    }
2480071d4279SBram Moolenaar 	    else if (fp->fd_top + fp->fd_len <= flp->lnum && lvl >= level)
2481071d4279SBram Moolenaar 		finish = TRUE;
2482071d4279SBram Moolenaar 	    else
2483071d4279SBram Moolenaar 		break;
2484071d4279SBram Moolenaar 	}
2485071d4279SBram Moolenaar 
2486217e1b83SBram Moolenaar 	// At the start of the first nested fold and at the end of the current
2487217e1b83SBram Moolenaar 	// fold: check if existing folds at this level, before the current
2488217e1b83SBram Moolenaar 	// one, need to be deleted or truncated.
2489071d4279SBram Moolenaar 	if (fp == NULL
2490071d4279SBram Moolenaar 		&& (lvl != level
2491071d4279SBram Moolenaar 		    || flp->lnum_save >= bot
2492071d4279SBram Moolenaar 		    || flp->start != 0
2493071d4279SBram Moolenaar 		    || flp->had_end <= MAX_LEVEL
2494071d4279SBram Moolenaar 		    || flp->lnum == linecount))
2495071d4279SBram Moolenaar 	{
2496071d4279SBram Moolenaar 	    /*
2497071d4279SBram Moolenaar 	     * Remove or update folds that have lines between startlnum and
2498071d4279SBram Moolenaar 	     * firstlnum.
2499071d4279SBram Moolenaar 	     */
2500071d4279SBram Moolenaar 	    while (!got_int)
2501071d4279SBram Moolenaar 	    {
2502217e1b83SBram Moolenaar 		// set concat to 1 if it's allowed to concatenated this fold
2503217e1b83SBram Moolenaar 		// with a previous one that touches it.
2504071d4279SBram Moolenaar 		if (flp->start != 0 || flp->had_end <= MAX_LEVEL)
2505071d4279SBram Moolenaar 		    concat = 0;
2506071d4279SBram Moolenaar 		else
2507071d4279SBram Moolenaar 		    concat = 1;
2508071d4279SBram Moolenaar 
2509217e1b83SBram Moolenaar 		// Find an existing fold to re-use.  Preferably one that
2510217e1b83SBram Moolenaar 		// includes startlnum, otherwise one that ends just before
2511217e1b83SBram Moolenaar 		// startlnum or starts after it.
2512*64f37d30SBram Moolenaar 		if (gap->ga_len > 0 && (foldFind(gap, startlnum, &fp)
2513071d4279SBram Moolenaar 			|| (fp < ((fold_T *)gap->ga_data) + gap->ga_len
2514071d4279SBram Moolenaar 			    && fp->fd_top <= firstlnum)
2515071d4279SBram Moolenaar 			|| foldFind(gap, firstlnum - concat, &fp)
2516071d4279SBram Moolenaar 			|| (fp < ((fold_T *)gap->ga_data) + gap->ga_len
2517071d4279SBram Moolenaar 			    && ((lvl < level && fp->fd_top < flp->lnum)
2518071d4279SBram Moolenaar 				|| (lvl >= level
2519*64f37d30SBram Moolenaar 					   && fp->fd_top <= flp->lnum_save)))))
2520071d4279SBram Moolenaar 		{
2521071d4279SBram Moolenaar 		    if (fp->fd_top + fp->fd_len + concat > firstlnum)
2522071d4279SBram Moolenaar 		    {
2523217e1b83SBram Moolenaar 			// Use existing fold for the new fold.  If it starts
2524217e1b83SBram Moolenaar 			// before where we started looking, extend it.  If it
2525217e1b83SBram Moolenaar 			// starts at another line, update nested folds to keep
2526217e1b83SBram Moolenaar 			// their position, compensating for the new fd_top.
2527eadbc2b4SBram Moolenaar 			if (fp->fd_top == firstlnum)
2528eadbc2b4SBram Moolenaar 			{
2529217e1b83SBram Moolenaar 			    // have found a fold beginning where we want
2530eadbc2b4SBram Moolenaar 			}
2531eadbc2b4SBram Moolenaar 			else if (fp->fd_top >= startlnum)
2532071d4279SBram Moolenaar 			{
2533071d4279SBram Moolenaar 			    if (fp->fd_top > firstlnum)
2534217e1b83SBram Moolenaar 				// like lines are inserted
2535071d4279SBram Moolenaar 				foldMarkAdjustRecurse(&fp->fd_nested,
2536071d4279SBram Moolenaar 					(linenr_T)0, (linenr_T)MAXLNUM,
2537071d4279SBram Moolenaar 					(long)(fp->fd_top - firstlnum), 0L);
2538071d4279SBram Moolenaar 			    else
2539217e1b83SBram Moolenaar 				// like lines are deleted
2540071d4279SBram Moolenaar 				foldMarkAdjustRecurse(&fp->fd_nested,
2541071d4279SBram Moolenaar 					(linenr_T)0,
2542071d4279SBram Moolenaar 					(long)(firstlnum - fp->fd_top - 1),
2543071d4279SBram Moolenaar 					(linenr_T)MAXLNUM,
2544071d4279SBram Moolenaar 					(long)(fp->fd_top - firstlnum));
2545071d4279SBram Moolenaar 			    fp->fd_len += fp->fd_top - firstlnum;
2546071d4279SBram Moolenaar 			    fp->fd_top = firstlnum;
2547071d4279SBram Moolenaar 			    fold_changed = TRUE;
2548071d4279SBram Moolenaar 			}
2549eadbc2b4SBram Moolenaar 			else if ((flp->start != 0 && lvl == level)
2550eadbc2b4SBram Moolenaar 						     || firstlnum != startlnum)
2551071d4279SBram Moolenaar 			{
2552eadbc2b4SBram Moolenaar 			    linenr_T breakstart;
2553eadbc2b4SBram Moolenaar 			    linenr_T breakend;
2554eadbc2b4SBram Moolenaar 
2555eadbc2b4SBram Moolenaar 			    /*
2556eadbc2b4SBram Moolenaar 			     * Before there was a fold spanning from above
2557eadbc2b4SBram Moolenaar 			     * startlnum to below firstlnum. This fold is valid
2558eadbc2b4SBram Moolenaar 			     * above startlnum (because we are not updating
2559eadbc2b4SBram Moolenaar 			     * that range), but there should now be a break in
2560eadbc2b4SBram Moolenaar 			     * it.
2561eadbc2b4SBram Moolenaar 			     * If the break is because we are now forced to
2562eadbc2b4SBram Moolenaar 			     * start a new fold at the level "level" at line
2563eadbc2b4SBram Moolenaar 			     * fline->lnum, then we need to split the fold at
2564eadbc2b4SBram Moolenaar 			     * fline->lnum.
2565eadbc2b4SBram Moolenaar 			     * If the break is because the range
2566eadbc2b4SBram Moolenaar 			     * [startlnum, firstlnum) is now at a lower indent
2567eadbc2b4SBram Moolenaar 			     * than "level", we need to split the fold in this
2568eadbc2b4SBram Moolenaar 			     * range.
2569eadbc2b4SBram Moolenaar 			     * Any splits have to be done recursively.
2570eadbc2b4SBram Moolenaar 			     */
2571eadbc2b4SBram Moolenaar 			    if (firstlnum != startlnum)
2572eadbc2b4SBram Moolenaar 			    {
2573eadbc2b4SBram Moolenaar 				breakstart = startlnum;
2574eadbc2b4SBram Moolenaar 				breakend = firstlnum;
2575eadbc2b4SBram Moolenaar 			    }
2576eadbc2b4SBram Moolenaar 			    else
2577eadbc2b4SBram Moolenaar 			    {
2578eadbc2b4SBram Moolenaar 				breakstart = flp->lnum;
2579eadbc2b4SBram Moolenaar 				breakend = flp->lnum;
2580eadbc2b4SBram Moolenaar 			    }
2581eadbc2b4SBram Moolenaar 			    foldRemove(&fp->fd_nested, breakstart - fp->fd_top,
2582eadbc2b4SBram Moolenaar 						      breakend - fp->fd_top);
2583071d4279SBram Moolenaar 			    i = (int)(fp - (fold_T *)gap->ga_data);
2584eadbc2b4SBram Moolenaar 			    foldSplit(gap, i, breakstart, breakend - 1);
2585071d4279SBram Moolenaar 			    fp = (fold_T *)gap->ga_data + i + 1;
2586eadbc2b4SBram Moolenaar 
2587217e1b83SBram Moolenaar 			    // If using the "marker" or "syntax" method, we
2588217e1b83SBram Moolenaar 			    // need to continue until the end of the fold is
2589217e1b83SBram Moolenaar 			    // found.
2590071d4279SBram Moolenaar 			    if (getlevel == foldlevelMarker
2591071d4279SBram Moolenaar 				    || getlevel == foldlevelExpr
2592071d4279SBram Moolenaar 				    || getlevel == foldlevelSyntax)
2593071d4279SBram Moolenaar 				finish = TRUE;
2594071d4279SBram Moolenaar 			}
2595eadbc2b4SBram Moolenaar 
2596eadbc2b4SBram Moolenaar 			if (fp->fd_top == startlnum && concat)
2597eadbc2b4SBram Moolenaar 			{
2598eadbc2b4SBram Moolenaar 			    i = (int)(fp - (fold_T *)gap->ga_data);
2599eadbc2b4SBram Moolenaar 			    if (i != 0)
2600eadbc2b4SBram Moolenaar 			    {
2601eadbc2b4SBram Moolenaar 				fp2 = fp - 1;
2602eadbc2b4SBram Moolenaar 				if (fp2->fd_top + fp2->fd_len == fp->fd_top)
2603eadbc2b4SBram Moolenaar 				{
2604eadbc2b4SBram Moolenaar 				    foldMerge(fp2, gap, fp);
2605eadbc2b4SBram Moolenaar 				    fp = fp2;
2606eadbc2b4SBram Moolenaar 				}
2607eadbc2b4SBram Moolenaar 			    }
2608eadbc2b4SBram Moolenaar 			}
2609071d4279SBram Moolenaar 			break;
2610071d4279SBram Moolenaar 		    }
2611071d4279SBram Moolenaar 		    if (fp->fd_top >= startlnum)
2612071d4279SBram Moolenaar 		    {
2613217e1b83SBram Moolenaar 			// A fold that starts at or after startlnum and stops
2614217e1b83SBram Moolenaar 			// before the new fold must be deleted.  Continue
2615217e1b83SBram Moolenaar 			// looking for the next one.
2616071d4279SBram Moolenaar 			deleteFoldEntry(gap,
2617071d4279SBram Moolenaar 				     (int)(fp - (fold_T *)gap->ga_data), TRUE);
2618071d4279SBram Moolenaar 		    }
2619071d4279SBram Moolenaar 		    else
2620071d4279SBram Moolenaar 		    {
2621217e1b83SBram Moolenaar 			// A fold has some lines above startlnum, truncate it
2622217e1b83SBram Moolenaar 			// to stop just above startlnum.
2623071d4279SBram Moolenaar 			fp->fd_len = startlnum - fp->fd_top;
2624071d4279SBram Moolenaar 			foldMarkAdjustRecurse(&fp->fd_nested,
2625071d4279SBram Moolenaar 				(linenr_T)fp->fd_len, (linenr_T)MAXLNUM,
2626071d4279SBram Moolenaar 						       (linenr_T)MAXLNUM, 0L);
2627071d4279SBram Moolenaar 			fold_changed = TRUE;
2628071d4279SBram Moolenaar 		    }
2629071d4279SBram Moolenaar 		}
2630071d4279SBram Moolenaar 		else
2631071d4279SBram Moolenaar 		{
2632217e1b83SBram Moolenaar 		    // Insert new fold.  Careful: ga_data may be NULL and it
2633217e1b83SBram Moolenaar 		    // may change!
2634*64f37d30SBram Moolenaar 		    if (gap->ga_len == 0)
2635*64f37d30SBram Moolenaar 			i = 0;
2636*64f37d30SBram Moolenaar 		    else
2637071d4279SBram Moolenaar 			i = (int)(fp - (fold_T *)gap->ga_data);
2638071d4279SBram Moolenaar 		    if (foldInsert(gap, i) != OK)
2639071d4279SBram Moolenaar 			return bot;
2640071d4279SBram Moolenaar 		    fp = (fold_T *)gap->ga_data + i;
2641217e1b83SBram Moolenaar 		    // The new fold continues until bot, unless we find the
2642217e1b83SBram Moolenaar 		    // end earlier.
2643071d4279SBram Moolenaar 		    fp->fd_top = firstlnum;
2644071d4279SBram Moolenaar 		    fp->fd_len = bot - firstlnum + 1;
2645217e1b83SBram Moolenaar 		    // When the containing fold is open, the new fold is open.
2646217e1b83SBram Moolenaar 		    // The new fold is closed if the fold above it is closed.
2647217e1b83SBram Moolenaar 		    // The first fold depends on the containing fold.
2648071d4279SBram Moolenaar 		    if (topflags == FD_OPEN)
2649071d4279SBram Moolenaar 		    {
2650071d4279SBram Moolenaar 			flp->wp->w_fold_manual = TRUE;
2651071d4279SBram Moolenaar 			fp->fd_flags = FD_OPEN;
2652071d4279SBram Moolenaar 		    }
2653071d4279SBram Moolenaar 		    else if (i <= 0)
2654071d4279SBram Moolenaar 		    {
2655071d4279SBram Moolenaar 			fp->fd_flags = topflags;
2656071d4279SBram Moolenaar 			if (topflags != FD_LEVEL)
2657071d4279SBram Moolenaar 			    flp->wp->w_fold_manual = TRUE;
2658071d4279SBram Moolenaar 		    }
2659071d4279SBram Moolenaar 		    else
2660071d4279SBram Moolenaar 			fp->fd_flags = (fp - 1)->fd_flags;
2661071d4279SBram Moolenaar 		    fp->fd_small = MAYBE;
2662217e1b83SBram Moolenaar 		    // If using the "marker", "expr" or "syntax" method, we
2663217e1b83SBram Moolenaar 		    // need to continue until the end of the fold is found.
2664071d4279SBram Moolenaar 		    if (getlevel == foldlevelMarker
2665071d4279SBram Moolenaar 			    || getlevel == foldlevelExpr
2666071d4279SBram Moolenaar 			    || getlevel == foldlevelSyntax)
2667071d4279SBram Moolenaar 			finish = TRUE;
2668071d4279SBram Moolenaar 		    fold_changed = TRUE;
2669071d4279SBram Moolenaar 		    break;
2670071d4279SBram Moolenaar 		}
2671071d4279SBram Moolenaar 	    }
2672071d4279SBram Moolenaar 	}
2673071d4279SBram Moolenaar 
2674071d4279SBram Moolenaar 	if (lvl < level || flp->lnum > linecount)
2675071d4279SBram Moolenaar 	{
2676071d4279SBram Moolenaar 	    /*
2677071d4279SBram Moolenaar 	     * Found a line with a lower foldlevel, this fold ends just above
2678071d4279SBram Moolenaar 	     * "flp->lnum".
2679071d4279SBram Moolenaar 	     */
2680071d4279SBram Moolenaar 	    break;
2681071d4279SBram Moolenaar 	}
2682071d4279SBram Moolenaar 
2683071d4279SBram Moolenaar 	/*
2684071d4279SBram Moolenaar 	 * The fold includes the line "flp->lnum" and "flp->lnum_save".
2685eb3593b3SBram Moolenaar 	 * Check "fp" for safety.
2686071d4279SBram Moolenaar 	 */
2687eb3593b3SBram Moolenaar 	if (lvl > level && fp != NULL)
2688071d4279SBram Moolenaar 	{
2689071d4279SBram Moolenaar 	    /*
2690071d4279SBram Moolenaar 	     * There is a nested fold, handle it recursively.
2691071d4279SBram Moolenaar 	     */
2692217e1b83SBram Moolenaar 	    // At least do one line (can happen when finish is TRUE).
2693071d4279SBram Moolenaar 	    if (bot < flp->lnum)
2694071d4279SBram Moolenaar 		bot = flp->lnum;
2695071d4279SBram Moolenaar 
2696217e1b83SBram Moolenaar 	    // Line numbers in the nested fold are relative to the start of
2697217e1b83SBram Moolenaar 	    // this fold.
2698071d4279SBram Moolenaar 	    flp->lnum = flp->lnum_save - fp->fd_top;
2699071d4279SBram Moolenaar 	    flp->off += fp->fd_top;
2700071d4279SBram Moolenaar 	    i = (int)(fp - (fold_T *)gap->ga_data);
2701071d4279SBram Moolenaar 	    bot = foldUpdateIEMSRecurse(&fp->fd_nested, level + 1,
2702071d4279SBram Moolenaar 				       startlnum2 - fp->fd_top, flp, getlevel,
2703071d4279SBram Moolenaar 					      bot - fp->fd_top, fp->fd_flags);
2704071d4279SBram Moolenaar 	    fp = (fold_T *)gap->ga_data + i;
2705071d4279SBram Moolenaar 	    flp->lnum += fp->fd_top;
2706071d4279SBram Moolenaar 	    flp->lnum_save += fp->fd_top;
2707071d4279SBram Moolenaar 	    flp->off -= fp->fd_top;
2708071d4279SBram Moolenaar 	    bot += fp->fd_top;
2709071d4279SBram Moolenaar 	    startlnum2 = flp->lnum;
2710071d4279SBram Moolenaar 
2711217e1b83SBram Moolenaar 	    // This fold may end at the same line, don't incr. flp->lnum.
2712071d4279SBram Moolenaar 	}
2713071d4279SBram Moolenaar 	else
2714071d4279SBram Moolenaar 	{
2715071d4279SBram Moolenaar 	    /*
2716071d4279SBram Moolenaar 	     * Get the level of the next line, then continue the loop to check
2717071d4279SBram Moolenaar 	     * if it ends there.
2718071d4279SBram Moolenaar 	     * Skip over undefined lines, to find the foldlevel after it.
2719071d4279SBram Moolenaar 	     * For the last line in the file the foldlevel is always valid.
2720071d4279SBram Moolenaar 	     */
2721071d4279SBram Moolenaar 	    flp->lnum = flp->lnum_save;
2722071d4279SBram Moolenaar 	    ll = flp->lnum + 1;
2723071d4279SBram Moolenaar 	    while (!got_int)
2724071d4279SBram Moolenaar 	    {
2725217e1b83SBram Moolenaar 		// Make the previous level available to foldlevel().
2726071d4279SBram Moolenaar 		prev_lnum = flp->lnum;
2727071d4279SBram Moolenaar 		prev_lnum_lvl = flp->lvl;
2728071d4279SBram Moolenaar 
2729071d4279SBram Moolenaar 		if (++flp->lnum > linecount)
2730071d4279SBram Moolenaar 		    break;
2731071d4279SBram Moolenaar 		flp->lvl = flp->lvl_next;
2732071d4279SBram Moolenaar 		getlevel(flp);
2733071d4279SBram Moolenaar 		if (flp->lvl >= 0 || flp->had_end <= MAX_LEVEL)
2734071d4279SBram Moolenaar 		    break;
2735071d4279SBram Moolenaar 	    }
2736071d4279SBram Moolenaar 	    prev_lnum = 0;
2737071d4279SBram Moolenaar 	    if (flp->lnum > linecount)
2738071d4279SBram Moolenaar 		break;
2739071d4279SBram Moolenaar 
2740217e1b83SBram Moolenaar 	    // leave flp->lnum_save to lnum of the line that was used to get
2741217e1b83SBram Moolenaar 	    // the level, flp->lnum to the lnum of the next line.
2742071d4279SBram Moolenaar 	    flp->lnum_save = flp->lnum;
2743071d4279SBram Moolenaar 	    flp->lnum = ll;
2744071d4279SBram Moolenaar 	}
2745071d4279SBram Moolenaar     }
2746071d4279SBram Moolenaar 
2747217e1b83SBram Moolenaar     if (fp == NULL)	// only happens when got_int is set
2748071d4279SBram Moolenaar 	return bot;
2749071d4279SBram Moolenaar 
2750071d4279SBram Moolenaar     /*
2751071d4279SBram Moolenaar      * Get here when:
2752071d4279SBram Moolenaar      * lvl < level: the folds ends just above "flp->lnum"
2753071d4279SBram Moolenaar      * lvl >= level: fold continues below "bot"
2754071d4279SBram Moolenaar      */
2755071d4279SBram Moolenaar 
27569a4a8c4dSBram Moolenaar     // Current fold at least extends until lnum.
2757071d4279SBram Moolenaar     if (fp->fd_len < flp->lnum - fp->fd_top)
2758071d4279SBram Moolenaar     {
2759071d4279SBram Moolenaar 	fp->fd_len = flp->lnum - fp->fd_top;
2760ef6fc097SBram Moolenaar 	fp->fd_small = MAYBE;
2761071d4279SBram Moolenaar 	fold_changed = TRUE;
2762071d4279SBram Moolenaar     }
27639a4a8c4dSBram Moolenaar     else if (fp->fd_top + fp->fd_len > linecount)
27649a4a8c4dSBram Moolenaar 	// running into the end of the buffer (deleted last line)
27659a4a8c4dSBram Moolenaar 	fp->fd_len = linecount - fp->fd_top + 1;
2766071d4279SBram Moolenaar 
27679a4a8c4dSBram Moolenaar     // Delete contained folds from the end of the last one found until where
27689a4a8c4dSBram Moolenaar     // we stopped looking.
2769071d4279SBram Moolenaar     foldRemove(&fp->fd_nested, startlnum2 - fp->fd_top,
2770071d4279SBram Moolenaar 						  flp->lnum - 1 - fp->fd_top);
2771071d4279SBram Moolenaar 
2772071d4279SBram Moolenaar     if (lvl < level)
2773071d4279SBram Moolenaar     {
2774217e1b83SBram Moolenaar 	// End of fold found, update the length when it got shorter.
2775071d4279SBram Moolenaar 	if (fp->fd_len != flp->lnum - fp->fd_top)
2776071d4279SBram Moolenaar 	{
27779d20ce69SBram Moolenaar 	    if (fp->fd_top + fp->fd_len - 1 > bot)
2778071d4279SBram Moolenaar 	    {
2779217e1b83SBram Moolenaar 		// fold continued below bot
2780071d4279SBram Moolenaar 		if (getlevel == foldlevelMarker
2781071d4279SBram Moolenaar 			|| getlevel == foldlevelExpr
2782071d4279SBram Moolenaar 			|| getlevel == foldlevelSyntax)
2783071d4279SBram Moolenaar 		{
2784217e1b83SBram Moolenaar 		    // marker method: truncate the fold and make sure the
2785217e1b83SBram Moolenaar 		    // previously included lines are processed again
2786071d4279SBram Moolenaar 		    bot = fp->fd_top + fp->fd_len - 1;
2787071d4279SBram Moolenaar 		    fp->fd_len = flp->lnum - fp->fd_top;
2788071d4279SBram Moolenaar 		}
2789071d4279SBram Moolenaar 		else
2790071d4279SBram Moolenaar 		{
2791217e1b83SBram Moolenaar 		    // indent or expr method: split fold to create a new one
2792217e1b83SBram Moolenaar 		    // below bot
2793071d4279SBram Moolenaar 		    i = (int)(fp - (fold_T *)gap->ga_data);
2794071d4279SBram Moolenaar 		    foldSplit(gap, i, flp->lnum, bot);
2795071d4279SBram Moolenaar 		    fp = (fold_T *)gap->ga_data + i;
2796071d4279SBram Moolenaar 		}
2797071d4279SBram Moolenaar 	    }
2798071d4279SBram Moolenaar 	    else
2799071d4279SBram Moolenaar 		fp->fd_len = flp->lnum - fp->fd_top;
2800071d4279SBram Moolenaar 	    fold_changed = TRUE;
2801071d4279SBram Moolenaar 	}
2802071d4279SBram Moolenaar     }
2803071d4279SBram Moolenaar 
2804217e1b83SBram Moolenaar     // delete following folds that end before the current line
2805071d4279SBram Moolenaar     for (;;)
2806071d4279SBram Moolenaar     {
2807071d4279SBram Moolenaar 	fp2 = fp + 1;
2808071d4279SBram Moolenaar 	if (fp2 >= (fold_T *)gap->ga_data + gap->ga_len
2809071d4279SBram Moolenaar 						  || fp2->fd_top > flp->lnum)
2810071d4279SBram Moolenaar 	    break;
2811071d4279SBram Moolenaar 	if (fp2->fd_top + fp2->fd_len > flp->lnum)
2812071d4279SBram Moolenaar 	{
2813071d4279SBram Moolenaar 	    if (fp2->fd_top < flp->lnum)
2814071d4279SBram Moolenaar 	    {
2815217e1b83SBram Moolenaar 		// Make fold that includes lnum start at lnum.
2816071d4279SBram Moolenaar 		foldMarkAdjustRecurse(&fp2->fd_nested,
2817071d4279SBram Moolenaar 			(linenr_T)0, (long)(flp->lnum - fp2->fd_top - 1),
2818071d4279SBram Moolenaar 			(linenr_T)MAXLNUM, (long)(fp2->fd_top - flp->lnum));
2819071d4279SBram Moolenaar 		fp2->fd_len -= flp->lnum - fp2->fd_top;
2820071d4279SBram Moolenaar 		fp2->fd_top = flp->lnum;
2821071d4279SBram Moolenaar 		fold_changed = TRUE;
2822071d4279SBram Moolenaar 	    }
2823071d4279SBram Moolenaar 
2824071d4279SBram Moolenaar 	    if (lvl >= level)
2825071d4279SBram Moolenaar 	    {
2826217e1b83SBram Moolenaar 		// merge new fold with existing fold that follows
2827071d4279SBram Moolenaar 		foldMerge(fp, gap, fp2);
2828071d4279SBram Moolenaar 	    }
2829071d4279SBram Moolenaar 	    break;
2830071d4279SBram Moolenaar 	}
2831071d4279SBram Moolenaar 	fold_changed = TRUE;
2832071d4279SBram Moolenaar 	deleteFoldEntry(gap, (int)(fp2 - (fold_T *)gap->ga_data), TRUE);
2833071d4279SBram Moolenaar     }
2834071d4279SBram Moolenaar 
2835217e1b83SBram Moolenaar     // Need to redraw the lines we inspected, which might be further down than
2836217e1b83SBram Moolenaar     // was asked for.
2837071d4279SBram Moolenaar     if (bot < flp->lnum - 1)
2838071d4279SBram Moolenaar 	bot = flp->lnum - 1;
2839071d4279SBram Moolenaar 
2840071d4279SBram Moolenaar     return bot;
2841071d4279SBram Moolenaar }
2842071d4279SBram Moolenaar 
2843217e1b83SBram Moolenaar // foldInsert() {{{2
2844071d4279SBram Moolenaar /*
2845071d4279SBram Moolenaar  * Insert a new fold in "gap" at position "i".
2846071d4279SBram Moolenaar  * Returns OK for success, FAIL for failure.
2847071d4279SBram Moolenaar  */
2848071d4279SBram Moolenaar     static int
284966f948e9SBram Moolenaar foldInsert(garray_T *gap, int i)
2850071d4279SBram Moolenaar {
2851071d4279SBram Moolenaar     fold_T	*fp;
2852071d4279SBram Moolenaar 
2853071d4279SBram Moolenaar     if (ga_grow(gap, 1) != OK)
2854071d4279SBram Moolenaar 	return FAIL;
2855071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data + i;
2856*64f37d30SBram Moolenaar     if (gap->ga_len > 0 && i < gap->ga_len)
2857071d4279SBram Moolenaar 	mch_memmove(fp + 1, fp, sizeof(fold_T) * (gap->ga_len - i));
2858071d4279SBram Moolenaar     ++gap->ga_len;
2859071d4279SBram Moolenaar     ga_init2(&fp->fd_nested, (int)sizeof(fold_T), 10);
2860071d4279SBram Moolenaar     return OK;
2861071d4279SBram Moolenaar }
2862071d4279SBram Moolenaar 
2863217e1b83SBram Moolenaar // foldSplit() {{{2
2864071d4279SBram Moolenaar /*
2865071d4279SBram Moolenaar  * Split the "i"th fold in "gap", which starts before "top" and ends below
2866071d4279SBram Moolenaar  * "bot" in two pieces, one ending above "top" and the other starting below
2867071d4279SBram Moolenaar  * "bot".
2868071d4279SBram Moolenaar  * The caller must first have taken care of any nested folds from "top" to
2869071d4279SBram Moolenaar  * "bot"!
2870071d4279SBram Moolenaar  */
2871071d4279SBram Moolenaar     static void
287266f948e9SBram Moolenaar foldSplit(
287366f948e9SBram Moolenaar     garray_T	*gap,
287466f948e9SBram Moolenaar     int		i,
287566f948e9SBram Moolenaar     linenr_T	top,
287666f948e9SBram Moolenaar     linenr_T	bot)
2877071d4279SBram Moolenaar {
2878071d4279SBram Moolenaar     fold_T	*fp;
2879071d4279SBram Moolenaar     fold_T	*fp2;
2880071d4279SBram Moolenaar     garray_T	*gap1;
2881071d4279SBram Moolenaar     garray_T	*gap2;
2882071d4279SBram Moolenaar     int		idx;
2883071d4279SBram Moolenaar     int		len;
2884071d4279SBram Moolenaar 
2885217e1b83SBram Moolenaar     // The fold continues below bot, need to split it.
2886071d4279SBram Moolenaar     if (foldInsert(gap, i + 1) == FAIL)
2887071d4279SBram Moolenaar 	return;
2888071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data + i;
2889071d4279SBram Moolenaar     fp[1].fd_top = bot + 1;
2890071d4279SBram Moolenaar     fp[1].fd_len = fp->fd_len - (fp[1].fd_top - fp->fd_top);
2891071d4279SBram Moolenaar     fp[1].fd_flags = fp->fd_flags;
2892eda6eb9fSBram Moolenaar     fp[1].fd_small = MAYBE;
2893eda6eb9fSBram Moolenaar     fp->fd_small = MAYBE;
2894071d4279SBram Moolenaar 
2895217e1b83SBram Moolenaar     // Move nested folds below bot to new fold.  There can't be
2896217e1b83SBram Moolenaar     // any between top and bot, they have been removed by the caller.
2897071d4279SBram Moolenaar     gap1 = &fp->fd_nested;
2898071d4279SBram Moolenaar     gap2 = &fp[1].fd_nested;
2899071d4279SBram Moolenaar     (void)(foldFind(gap1, bot + 1 - fp->fd_top, &fp2));
2900071d4279SBram Moolenaar     len = (int)((fold_T *)gap1->ga_data + gap1->ga_len - fp2);
2901071d4279SBram Moolenaar     if (len > 0 && ga_grow(gap2, len) == OK)
2902071d4279SBram Moolenaar     {
2903071d4279SBram Moolenaar 	for (idx = 0; idx < len; ++idx)
2904071d4279SBram Moolenaar 	{
2905071d4279SBram Moolenaar 	    ((fold_T *)gap2->ga_data)[idx] = fp2[idx];
2906071d4279SBram Moolenaar 	    ((fold_T *)gap2->ga_data)[idx].fd_top
2907071d4279SBram Moolenaar 						 -= fp[1].fd_top - fp->fd_top;
2908071d4279SBram Moolenaar 	}
2909071d4279SBram Moolenaar 	gap2->ga_len = len;
2910071d4279SBram Moolenaar 	gap1->ga_len -= len;
2911071d4279SBram Moolenaar     }
2912071d4279SBram Moolenaar     fp->fd_len = top - fp->fd_top;
2913071d4279SBram Moolenaar     fold_changed = TRUE;
2914071d4279SBram Moolenaar }
2915071d4279SBram Moolenaar 
2916217e1b83SBram Moolenaar // foldRemove() {{{2
2917071d4279SBram Moolenaar /*
2918071d4279SBram Moolenaar  * Remove folds within the range "top" to and including "bot".
2919071d4279SBram Moolenaar  * Check for these situations:
2920071d4279SBram Moolenaar  *      1  2  3
2921071d4279SBram Moolenaar  *      1  2  3
2922071d4279SBram Moolenaar  * top     2  3  4  5
2923071d4279SBram Moolenaar  *	   2  3  4  5
2924071d4279SBram Moolenaar  * bot	   2  3  4  5
2925071d4279SBram Moolenaar  *	      3     5  6
2926071d4279SBram Moolenaar  *	      3     5  6
2927071d4279SBram Moolenaar  *
2928071d4279SBram Moolenaar  * 1: not changed
292925394022SBram Moolenaar  * 2: truncate to stop above "top"
2930071d4279SBram Moolenaar  * 3: split in two parts, one stops above "top", other starts below "bot".
2931071d4279SBram Moolenaar  * 4: deleted
2932071d4279SBram Moolenaar  * 5: made to start below "bot".
2933071d4279SBram Moolenaar  * 6: not changed
2934071d4279SBram Moolenaar  */
2935071d4279SBram Moolenaar     static void
293666f948e9SBram Moolenaar foldRemove(garray_T *gap, linenr_T top, linenr_T bot)
2937071d4279SBram Moolenaar {
2938071d4279SBram Moolenaar     fold_T	*fp = NULL;
2939071d4279SBram Moolenaar 
2940071d4279SBram Moolenaar     if (bot < top)
2941217e1b83SBram Moolenaar 	return;		// nothing to do
2942071d4279SBram Moolenaar 
2943*64f37d30SBram Moolenaar     while (gap->ga_len > 0)
2944071d4279SBram Moolenaar     {
2945217e1b83SBram Moolenaar 	// Find fold that includes top or a following one.
2946071d4279SBram Moolenaar 	if (foldFind(gap, top, &fp) && fp->fd_top < top)
2947071d4279SBram Moolenaar 	{
2948217e1b83SBram Moolenaar 	    // 2: or 3: need to delete nested folds
2949071d4279SBram Moolenaar 	    foldRemove(&fp->fd_nested, top - fp->fd_top, bot - fp->fd_top);
2950b11c826dSBram Moolenaar 	    if (fp->fd_top + fp->fd_len - 1 > bot)
2951071d4279SBram Moolenaar 	    {
2952217e1b83SBram Moolenaar 		// 3: need to split it.
2953071d4279SBram Moolenaar 		foldSplit(gap, (int)(fp - (fold_T *)gap->ga_data), top, bot);
2954071d4279SBram Moolenaar 	    }
2955071d4279SBram Moolenaar 	    else
2956071d4279SBram Moolenaar 	    {
2957217e1b83SBram Moolenaar 		// 2: truncate fold at "top".
2958071d4279SBram Moolenaar 		fp->fd_len = top - fp->fd_top;
2959071d4279SBram Moolenaar 	    }
2960071d4279SBram Moolenaar 	    fold_changed = TRUE;
2961071d4279SBram Moolenaar 	    continue;
2962071d4279SBram Moolenaar 	}
2963071d4279SBram Moolenaar 	if (fp >= (fold_T *)(gap->ga_data) + gap->ga_len
2964071d4279SBram Moolenaar 		|| fp->fd_top > bot)
2965071d4279SBram Moolenaar 	{
2966217e1b83SBram Moolenaar 	    // 6: Found a fold below bot, can stop looking.
2967071d4279SBram Moolenaar 	    break;
2968071d4279SBram Moolenaar 	}
2969071d4279SBram Moolenaar 	if (fp->fd_top >= top)
2970071d4279SBram Moolenaar 	{
2971217e1b83SBram Moolenaar 	    // Found an entry below top.
2972071d4279SBram Moolenaar 	    fold_changed = TRUE;
2973293ee4d4SBram Moolenaar 	    if (fp->fd_top + fp->fd_len - 1 > bot)
2974071d4279SBram Moolenaar 	    {
2975217e1b83SBram Moolenaar 		// 5: Make fold that includes bot start below bot.
2976071d4279SBram Moolenaar 		foldMarkAdjustRecurse(&fp->fd_nested,
2977071d4279SBram Moolenaar 			(linenr_T)0, (long)(bot - fp->fd_top),
2978071d4279SBram Moolenaar 			(linenr_T)MAXLNUM, (long)(fp->fd_top - bot - 1));
2979071d4279SBram Moolenaar 		fp->fd_len -= bot - fp->fd_top + 1;
2980071d4279SBram Moolenaar 		fp->fd_top = bot + 1;
2981071d4279SBram Moolenaar 		break;
2982071d4279SBram Moolenaar 	    }
2983071d4279SBram Moolenaar 
2984217e1b83SBram Moolenaar 	    // 4: Delete completely contained fold.
2985071d4279SBram Moolenaar 	    deleteFoldEntry(gap, (int)(fp - (fold_T *)gap->ga_data), TRUE);
2986071d4279SBram Moolenaar 	}
2987071d4279SBram Moolenaar     }
2988071d4279SBram Moolenaar }
2989071d4279SBram Moolenaar 
2990217e1b83SBram Moolenaar // foldReverseOrder() {{{2
299188d298aeSBram Moolenaar     static void
2992b11c826dSBram Moolenaar foldReverseOrder(garray_T *gap, linenr_T start_arg, linenr_T end_arg)
299388d298aeSBram Moolenaar {
299488d298aeSBram Moolenaar     fold_T *left, *right;
299588d298aeSBram Moolenaar     fold_T tmp;
2996b11c826dSBram Moolenaar     linenr_T start = start_arg;
2997b11c826dSBram Moolenaar     linenr_T end = end_arg;
299888d298aeSBram Moolenaar 
299988d298aeSBram Moolenaar     for (; start < end; start++, end--)
300088d298aeSBram Moolenaar     {
300188d298aeSBram Moolenaar 	left = (fold_T *)gap->ga_data + start;
300288d298aeSBram Moolenaar 	right = (fold_T *)gap->ga_data + end;
300388d298aeSBram Moolenaar 	tmp  = *left;
300488d298aeSBram Moolenaar 	*left = *right;
300588d298aeSBram Moolenaar 	*right = tmp;
300688d298aeSBram Moolenaar     }
300788d298aeSBram Moolenaar }
300888d298aeSBram Moolenaar 
3009217e1b83SBram Moolenaar // foldMoveRange() {{{2
301088d298aeSBram Moolenaar /*
301188d298aeSBram Moolenaar  * Move folds within the inclusive range "line1" to "line2" to after "dest"
301288d298aeSBram Moolenaar  * requires "line1" <= "line2" <= "dest"
301388d298aeSBram Moolenaar  *
301488d298aeSBram Moolenaar  * There are the following situations for the first fold at or below line1 - 1.
301588d298aeSBram Moolenaar  *       1  2  3  4
301688d298aeSBram Moolenaar  *       1  2  3  4
301788d298aeSBram Moolenaar  * line1    2  3  4
301888d298aeSBram Moolenaar  *          2  3  4  5  6  7
301988d298aeSBram Moolenaar  * line2       3  4  5  6  7
302088d298aeSBram Moolenaar  *             3  4     6  7  8  9
302188d298aeSBram Moolenaar  * dest           4        7  8  9
302288d298aeSBram Moolenaar  *                4        7  8    10
302388d298aeSBram Moolenaar  *                4        7  8    10
302488d298aeSBram Moolenaar  *
302588d298aeSBram Moolenaar  * In the following descriptions, "moved" means moving in the buffer, *and* in
302688d298aeSBram Moolenaar  * the fold array.
302788d298aeSBram Moolenaar  * Meanwhile, "shifted" just means moving in the buffer.
302888d298aeSBram Moolenaar  * 1. not changed
302988d298aeSBram Moolenaar  * 2. truncated above line1
303088d298aeSBram Moolenaar  * 3. length reduced by  line2 - line1, folds starting between the end of 3 and
303188d298aeSBram Moolenaar  *    dest are truncated and shifted up
303288d298aeSBram Moolenaar  * 4. internal folds moved (from [line1, line2] to dest)
303388d298aeSBram Moolenaar  * 5. moved to dest.
303488d298aeSBram Moolenaar  * 6. truncated below line2 and moved.
303588d298aeSBram Moolenaar  * 7. length reduced by line2 - dest, folds starting between line2 and dest are
303688d298aeSBram Moolenaar  *    removed, top is moved down by move_len.
303788d298aeSBram Moolenaar  * 8. truncated below dest and shifted up.
303888d298aeSBram Moolenaar  * 9. shifted up
303988d298aeSBram Moolenaar  * 10. not changed
304088d298aeSBram Moolenaar  */
304188d298aeSBram Moolenaar 
304288d298aeSBram Moolenaar     static void
304388d298aeSBram Moolenaar truncate_fold(fold_T *fp, linenr_T end)
304488d298aeSBram Moolenaar {
304540ebc0afSBram Moolenaar     end += 1;
304688d298aeSBram Moolenaar     foldRemove(&fp->fd_nested, end - fp->fd_top, MAXLNUM);
304740ebc0afSBram Moolenaar     fp->fd_len = end - fp->fd_top;
304888d298aeSBram Moolenaar }
304988d298aeSBram Moolenaar 
305088d298aeSBram Moolenaar #define fold_end(fp) ((fp)->fd_top + (fp)->fd_len - 1)
305188d298aeSBram Moolenaar #define valid_fold(fp, gap) ((fp) < ((fold_T *)(gap)->ga_data + (gap)->ga_len))
305288d298aeSBram Moolenaar #define fold_index(fp, gap) ((size_t)(fp - ((fold_T *)(gap)->ga_data)))
305388d298aeSBram Moolenaar 
30544a5a8dd6SBram Moolenaar     void
30554a5a8dd6SBram Moolenaar foldMoveRange(garray_T *gap, linenr_T line1, linenr_T line2, linenr_T dest)
305688d298aeSBram Moolenaar {
305788d298aeSBram Moolenaar     fold_T *fp;
305888d298aeSBram Moolenaar     linenr_T range_len = line2 - line1 + 1;
305988d298aeSBram Moolenaar     linenr_T move_len = dest - line2;
306088d298aeSBram Moolenaar     int at_start = foldFind(gap, line1 - 1, &fp);
306188d298aeSBram Moolenaar     size_t move_start = 0, move_end = 0, dest_index = 0;
306288d298aeSBram Moolenaar 
306388d298aeSBram Moolenaar     if (at_start)
306488d298aeSBram Moolenaar     {
306588d298aeSBram Moolenaar 	if (fold_end(fp) > dest)
306688d298aeSBram Moolenaar 	{
3067217e1b83SBram Moolenaar 	    // Case 4
3068217e1b83SBram Moolenaar 	   // don't have to change this fold, but have to move nested folds.
306988d298aeSBram Moolenaar 	    foldMoveRange(&fp->fd_nested, line1 - fp->fd_top, line2 -
307088d298aeSBram Moolenaar 		    fp->fd_top, dest - fp->fd_top);
307188d298aeSBram Moolenaar 	    return;
307288d298aeSBram Moolenaar 	}
307388d298aeSBram Moolenaar 	else if (fold_end(fp) > line2)
307488d298aeSBram Moolenaar 	{
3075217e1b83SBram Moolenaar 	    // Case 3
3076217e1b83SBram Moolenaar 	    // Remove nested folds between line1 and line2 & reduce the
3077217e1b83SBram Moolenaar 	    // length of fold by "range_len".
3078217e1b83SBram Moolenaar 	    // Folds after this one must be dealt with.
307988d298aeSBram Moolenaar 	    foldMarkAdjustRecurse(&fp->fd_nested, line1 - fp->fd_top, line2 -
308088d298aeSBram Moolenaar 		    fp->fd_top, MAXLNUM, -range_len);
308188d298aeSBram Moolenaar 	    fp->fd_len -= range_len;
308288d298aeSBram Moolenaar 	}
308388d298aeSBram Moolenaar 	else
3084217e1b83SBram Moolenaar 	    // Case 2 truncate fold, folds after this one must be dealt with.
308540ebc0afSBram Moolenaar 	    truncate_fold(fp, line1 - 1);
308688d298aeSBram Moolenaar 
3087217e1b83SBram Moolenaar 	// Look at the next fold, and treat that one as if it were the first
3088217e1b83SBram Moolenaar 	// after  "line1" (because now it is).
308988d298aeSBram Moolenaar 	fp = fp + 1;
309088d298aeSBram Moolenaar     }
309188d298aeSBram Moolenaar 
309288d298aeSBram Moolenaar     if (!valid_fold(fp, gap) || fp->fd_top > dest)
309388d298aeSBram Moolenaar     {
3094217e1b83SBram Moolenaar 	// Case 10
3095217e1b83SBram Moolenaar 	// No folds after "line1" and before "dest"
309688d298aeSBram Moolenaar 	return;
309788d298aeSBram Moolenaar     }
309888d298aeSBram Moolenaar     else if (fp->fd_top > line2)
309988d298aeSBram Moolenaar     {
310040ebc0afSBram Moolenaar 	for (; valid_fold(fp, gap) && fold_end(fp) <= dest; fp++)
3101217e1b83SBram Moolenaar 	// Case 9. (for all case 9's) -- shift up.
310288d298aeSBram Moolenaar 	    fp->fd_top -= range_len;
310388d298aeSBram Moolenaar 
310440ebc0afSBram Moolenaar 	if (valid_fold(fp, gap) && fp->fd_top <= dest)
310588d298aeSBram Moolenaar 	{
3106217e1b83SBram Moolenaar 	    // Case 8. -- ensure truncated at dest, shift up
310788d298aeSBram Moolenaar 	    truncate_fold(fp, dest);
310888d298aeSBram Moolenaar 	    fp->fd_top -= range_len;
310988d298aeSBram Moolenaar 	}
311088d298aeSBram Moolenaar 	return;
311188d298aeSBram Moolenaar     }
311288d298aeSBram Moolenaar     else if (fold_end(fp) > dest)
311388d298aeSBram Moolenaar     {
3114217e1b83SBram Moolenaar 	// Case 7 -- remove nested folds and shrink
311588d298aeSBram Moolenaar 	foldMarkAdjustRecurse(&fp->fd_nested, line2 + 1 - fp->fd_top, dest -
311688d298aeSBram Moolenaar 		fp->fd_top, MAXLNUM, -move_len);
311788d298aeSBram Moolenaar 	fp->fd_len -= move_len;
311888d298aeSBram Moolenaar 	fp->fd_top += move_len;
311988d298aeSBram Moolenaar 	return;
312088d298aeSBram Moolenaar     }
312188d298aeSBram Moolenaar 
3122217e1b83SBram Moolenaar     // Case 5 or 6
3123217e1b83SBram Moolenaar     // changes rely on whether there are folds between the end of
3124217e1b83SBram Moolenaar     // this fold and "dest".
312588d298aeSBram Moolenaar     move_start = fold_index(fp, gap);
312688d298aeSBram Moolenaar 
312788d298aeSBram Moolenaar     for (; valid_fold(fp, gap) && fp->fd_top <= dest; fp++)
312888d298aeSBram Moolenaar     {
312988d298aeSBram Moolenaar 	if (fp->fd_top <= line2)
313088d298aeSBram Moolenaar 	{
3131217e1b83SBram Moolenaar 	    // 1. 2. or 3.
313288d298aeSBram Moolenaar 	    if (fold_end(fp) > line2)
3133217e1b83SBram Moolenaar 		// 2. or 3., truncate before moving
313488d298aeSBram Moolenaar 		truncate_fold(fp, line2);
313588d298aeSBram Moolenaar 
313688d298aeSBram Moolenaar 	    fp->fd_top += move_len;
313788d298aeSBram Moolenaar 	    continue;
313888d298aeSBram Moolenaar 	}
313988d298aeSBram Moolenaar 
3140217e1b83SBram Moolenaar 	// Record index of the first fold after the moved range.
314188d298aeSBram Moolenaar 	if (move_end == 0)
314288d298aeSBram Moolenaar 	    move_end = fold_index(fp, gap);
314388d298aeSBram Moolenaar 
314488d298aeSBram Moolenaar 	if (fold_end(fp) > dest)
314588d298aeSBram Moolenaar 	    truncate_fold(fp, dest);
314688d298aeSBram Moolenaar 
314788d298aeSBram Moolenaar 	fp->fd_top -= range_len;
314888d298aeSBram Moolenaar     }
314988d298aeSBram Moolenaar 
315088d298aeSBram Moolenaar     dest_index = fold_index(fp, gap);
315188d298aeSBram Moolenaar 
315288d298aeSBram Moolenaar     /*
315394be619eSBram Moolenaar      * All folds are now correct, but not necessarily in the correct order.  We
315494be619eSBram Moolenaar      * must swap folds in the range [move_end, dest_index) with those in the
315594be619eSBram Moolenaar      * range [move_start, move_end).
315688d298aeSBram Moolenaar      */
315794be619eSBram Moolenaar     if (move_end == 0)
3158217e1b83SBram Moolenaar 	// There are no folds after those moved, hence no folds have been moved
3159217e1b83SBram Moolenaar 	// out of order.
316094be619eSBram Moolenaar 	return;
31610c0d4ecaSBram Moolenaar     foldReverseOrder(gap, (linenr_T)move_start, (linenr_T)dest_index - 1);
31620c0d4ecaSBram Moolenaar     foldReverseOrder(gap, (linenr_T)move_start,
31630c0d4ecaSBram Moolenaar 			   (linenr_T)(move_start + dest_index - move_end - 1));
31640c0d4ecaSBram Moolenaar     foldReverseOrder(gap, (linenr_T)(move_start + dest_index - move_end),
31650c0d4ecaSBram Moolenaar 						   (linenr_T)(dest_index - 1));
316688d298aeSBram Moolenaar }
316788d298aeSBram Moolenaar #undef fold_end
316888d298aeSBram Moolenaar #undef valid_fold
316988d298aeSBram Moolenaar #undef fold_index
317088d298aeSBram Moolenaar 
3171217e1b83SBram Moolenaar // foldMerge() {{{2
3172071d4279SBram Moolenaar /*
317325394022SBram Moolenaar  * Merge two adjacent folds (and the nested ones in them).
317425394022SBram Moolenaar  * This only works correctly when the folds are really adjacent!  Thus "fp1"
3175071d4279SBram Moolenaar  * must end just above "fp2".
3176071d4279SBram Moolenaar  * The resulting fold is "fp1", nested folds are moved from "fp2" to "fp1".
3177071d4279SBram Moolenaar  * Fold entry "fp2" in "gap" is deleted.
3178071d4279SBram Moolenaar  */
3179071d4279SBram Moolenaar     static void
318066f948e9SBram Moolenaar foldMerge(fold_T *fp1, garray_T *gap, fold_T *fp2)
3181071d4279SBram Moolenaar {
3182071d4279SBram Moolenaar     fold_T	*fp3;
3183071d4279SBram Moolenaar     fold_T	*fp4;
3184071d4279SBram Moolenaar     int		idx;
3185071d4279SBram Moolenaar     garray_T	*gap1 = &fp1->fd_nested;
3186071d4279SBram Moolenaar     garray_T	*gap2 = &fp2->fd_nested;
3187071d4279SBram Moolenaar 
3188217e1b83SBram Moolenaar     // If the last nested fold in fp1 touches the first nested fold in fp2,
3189217e1b83SBram Moolenaar     // merge them recursively.
3190071d4279SBram Moolenaar     if (foldFind(gap1, fp1->fd_len - 1L, &fp3) && foldFind(gap2, 0L, &fp4))
3191071d4279SBram Moolenaar 	foldMerge(fp3, gap2, fp4);
3192071d4279SBram Moolenaar 
3193217e1b83SBram Moolenaar     // Move nested folds in fp2 to the end of fp1.
3194071d4279SBram Moolenaar     if (gap2->ga_len > 0 && ga_grow(gap1, gap2->ga_len) == OK)
3195071d4279SBram Moolenaar     {
3196071d4279SBram Moolenaar 	for (idx = 0; idx < gap2->ga_len; ++idx)
3197071d4279SBram Moolenaar 	{
3198071d4279SBram Moolenaar 	    ((fold_T *)gap1->ga_data)[gap1->ga_len]
3199071d4279SBram Moolenaar 					= ((fold_T *)gap2->ga_data)[idx];
3200071d4279SBram Moolenaar 	    ((fold_T *)gap1->ga_data)[gap1->ga_len].fd_top += fp1->fd_len;
3201071d4279SBram Moolenaar 	    ++gap1->ga_len;
3202071d4279SBram Moolenaar 	}
3203071d4279SBram Moolenaar 	gap2->ga_len = 0;
3204071d4279SBram Moolenaar     }
3205071d4279SBram Moolenaar 
3206071d4279SBram Moolenaar     fp1->fd_len += fp2->fd_len;
3207071d4279SBram Moolenaar     deleteFoldEntry(gap, (int)(fp2 - (fold_T *)gap->ga_data), TRUE);
3208071d4279SBram Moolenaar     fold_changed = TRUE;
3209071d4279SBram Moolenaar }
3210071d4279SBram Moolenaar 
3211217e1b83SBram Moolenaar // foldlevelIndent() {{{2
3212071d4279SBram Moolenaar /*
3213071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "indent" method.
3214071d4279SBram Moolenaar  * Doesn't use any caching.
3215071d4279SBram Moolenaar  * Returns a level of -1 if the foldlevel depends on surrounding lines.
3216071d4279SBram Moolenaar  */
3217071d4279SBram Moolenaar     static void
321866f948e9SBram Moolenaar foldlevelIndent(fline_T *flp)
3219071d4279SBram Moolenaar {
3220071d4279SBram Moolenaar     char_u	*s;
3221071d4279SBram Moolenaar     buf_T	*buf;
3222071d4279SBram Moolenaar     linenr_T	lnum = flp->lnum + flp->off;
3223071d4279SBram Moolenaar 
3224071d4279SBram Moolenaar     buf = flp->wp->w_buffer;
3225071d4279SBram Moolenaar     s = skipwhite(ml_get_buf(buf, lnum, FALSE));
3226071d4279SBram Moolenaar 
3227217e1b83SBram Moolenaar     // empty line or lines starting with a character in 'foldignore': level
3228217e1b83SBram Moolenaar     // depends on surrounding lines
3229071d4279SBram Moolenaar     if (*s == NUL || vim_strchr(flp->wp->w_p_fdi, *s) != NULL)
3230071d4279SBram Moolenaar     {
3231217e1b83SBram Moolenaar 	// first and last line can't be undefined, use level 0
3232071d4279SBram Moolenaar 	if (lnum == 1 || lnum == buf->b_ml.ml_line_count)
3233071d4279SBram Moolenaar 	    flp->lvl = 0;
3234071d4279SBram Moolenaar 	else
3235071d4279SBram Moolenaar 	    flp->lvl = -1;
3236071d4279SBram Moolenaar     }
3237071d4279SBram Moolenaar     else
32380c27cbcaSBram Moolenaar 	flp->lvl = get_indent_buf(buf, lnum) / get_sw_value(buf);
3239071d4279SBram Moolenaar     if (flp->lvl > flp->wp->w_p_fdn)
324074c596b5SBram Moolenaar     {
3241071d4279SBram Moolenaar 	flp->lvl = flp->wp->w_p_fdn;
324274c596b5SBram Moolenaar 	if (flp->lvl < 0)
324374c596b5SBram Moolenaar 	    flp->lvl = 0;
324474c596b5SBram Moolenaar     }
3245071d4279SBram Moolenaar }
3246071d4279SBram Moolenaar 
3247217e1b83SBram Moolenaar // foldlevelDiff() {{{2
3248071d4279SBram Moolenaar #ifdef FEAT_DIFF
3249071d4279SBram Moolenaar /*
3250071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "diff" method.
3251071d4279SBram Moolenaar  * Doesn't use any caching.
3252071d4279SBram Moolenaar  */
3253071d4279SBram Moolenaar     static void
325466f948e9SBram Moolenaar foldlevelDiff(fline_T *flp)
3255071d4279SBram Moolenaar {
3256071d4279SBram Moolenaar     if (diff_infold(flp->wp, flp->lnum + flp->off))
3257071d4279SBram Moolenaar 	flp->lvl = 1;
3258071d4279SBram Moolenaar     else
3259071d4279SBram Moolenaar 	flp->lvl = 0;
3260071d4279SBram Moolenaar }
3261071d4279SBram Moolenaar #endif
3262071d4279SBram Moolenaar 
3263217e1b83SBram Moolenaar // foldlevelExpr() {{{2
3264071d4279SBram Moolenaar /*
3265071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "expr" method.
3266071d4279SBram Moolenaar  * Doesn't use any caching.
3267071d4279SBram Moolenaar  * Returns a level of -1 if the foldlevel depends on surrounding lines.
3268071d4279SBram Moolenaar  */
3269071d4279SBram Moolenaar     static void
327066f948e9SBram Moolenaar foldlevelExpr(fline_T *flp)
3271071d4279SBram Moolenaar {
3272071d4279SBram Moolenaar #ifndef FEAT_EVAL
3273071d4279SBram Moolenaar     flp->start = FALSE;
3274071d4279SBram Moolenaar     flp->lvl = 0;
3275071d4279SBram Moolenaar #else
3276071d4279SBram Moolenaar     win_T	*win;
3277071d4279SBram Moolenaar     int		n;
3278071d4279SBram Moolenaar     int		c;
3279071d4279SBram Moolenaar     linenr_T	lnum = flp->lnum + flp->off;
3280071d4279SBram Moolenaar     int		save_keytyped;
3281071d4279SBram Moolenaar 
3282071d4279SBram Moolenaar     win = curwin;
3283071d4279SBram Moolenaar     curwin = flp->wp;
3284071d4279SBram Moolenaar     curbuf = flp->wp->w_buffer;
3285071d4279SBram Moolenaar     set_vim_var_nr(VV_LNUM, lnum);
3286071d4279SBram Moolenaar 
3287071d4279SBram Moolenaar     flp->start = 0;
3288071d4279SBram Moolenaar     flp->had_end = flp->end;
3289071d4279SBram Moolenaar     flp->end = MAX_LEVEL + 1;
3290071d4279SBram Moolenaar     if (lnum <= 1)
3291071d4279SBram Moolenaar 	flp->lvl = 0;
3292071d4279SBram Moolenaar 
3293217e1b83SBram Moolenaar     // KeyTyped may be reset to 0 when calling a function which invokes
3294217e1b83SBram Moolenaar     // do_cmdline().  To make 'foldopen' work correctly restore KeyTyped.
3295071d4279SBram Moolenaar     save_keytyped = KeyTyped;
329622fcfad2SBram Moolenaar     n = (int)eval_foldexpr(flp->wp->w_p_fde, &c);
3297071d4279SBram Moolenaar     KeyTyped = save_keytyped;
3298071d4279SBram Moolenaar 
3299071d4279SBram Moolenaar     switch (c)
3300071d4279SBram Moolenaar     {
3301217e1b83SBram Moolenaar 	// "a1", "a2", .. : add to the fold level
3302071d4279SBram Moolenaar 	case 'a': if (flp->lvl >= 0)
3303071d4279SBram Moolenaar 		  {
3304071d4279SBram Moolenaar 		      flp->lvl += n;
3305071d4279SBram Moolenaar 		      flp->lvl_next = flp->lvl;
3306071d4279SBram Moolenaar 		  }
3307071d4279SBram Moolenaar 		  flp->start = n;
3308071d4279SBram Moolenaar 		  break;
3309071d4279SBram Moolenaar 
3310217e1b83SBram Moolenaar 	// "s1", "s2", .. : subtract from the fold level
3311071d4279SBram Moolenaar 	case 's': if (flp->lvl >= 0)
3312071d4279SBram Moolenaar 		  {
3313071d4279SBram Moolenaar 		      if (n > flp->lvl)
3314071d4279SBram Moolenaar 			  flp->lvl_next = 0;
3315071d4279SBram Moolenaar 		      else
3316071d4279SBram Moolenaar 			  flp->lvl_next = flp->lvl - n;
3317071d4279SBram Moolenaar 		      flp->end = flp->lvl_next + 1;
3318071d4279SBram Moolenaar 		  }
3319071d4279SBram Moolenaar 		  break;
3320071d4279SBram Moolenaar 
3321217e1b83SBram Moolenaar 	// ">1", ">2", .. : start a fold with a certain level
3322071d4279SBram Moolenaar 	case '>': flp->lvl = n;
3323071d4279SBram Moolenaar 		  flp->lvl_next = n;
3324071d4279SBram Moolenaar 		  flp->start = 1;
3325071d4279SBram Moolenaar 		  break;
3326071d4279SBram Moolenaar 
3327217e1b83SBram Moolenaar 	// "<1", "<2", .. : end a fold with a certain level
3328071d4279SBram Moolenaar 	case '<': flp->lvl_next = n - 1;
3329071d4279SBram Moolenaar 		  flp->end = n;
3330071d4279SBram Moolenaar 		  break;
3331071d4279SBram Moolenaar 
3332217e1b83SBram Moolenaar 	// "=": No change in level
3333071d4279SBram Moolenaar 	case '=': flp->lvl_next = flp->lvl;
3334071d4279SBram Moolenaar 		  break;
3335071d4279SBram Moolenaar 
3336217e1b83SBram Moolenaar 	// "-1", "0", "1", ..: set fold level
3337071d4279SBram Moolenaar 	default:  if (n < 0)
3338217e1b83SBram Moolenaar 		      // Use the current level for the next line, so that "a1"
3339217e1b83SBram Moolenaar 		      // will work there.
3340071d4279SBram Moolenaar 		      flp->lvl_next = flp->lvl;
3341071d4279SBram Moolenaar 		  else
3342071d4279SBram Moolenaar 		      flp->lvl_next = n;
3343071d4279SBram Moolenaar 		  flp->lvl = n;
3344071d4279SBram Moolenaar 		  break;
3345071d4279SBram Moolenaar     }
3346071d4279SBram Moolenaar 
3347217e1b83SBram Moolenaar     // If the level is unknown for the first or the last line in the file, use
3348217e1b83SBram Moolenaar     // level 0.
3349071d4279SBram Moolenaar     if (flp->lvl < 0)
3350071d4279SBram Moolenaar     {
3351071d4279SBram Moolenaar 	if (lnum <= 1)
3352071d4279SBram Moolenaar 	{
3353071d4279SBram Moolenaar 	    flp->lvl = 0;
3354071d4279SBram Moolenaar 	    flp->lvl_next = 0;
3355071d4279SBram Moolenaar 	}
3356071d4279SBram Moolenaar 	if (lnum == curbuf->b_ml.ml_line_count)
3357071d4279SBram Moolenaar 	    flp->lvl_next = 0;
3358071d4279SBram Moolenaar     }
3359071d4279SBram Moolenaar 
3360071d4279SBram Moolenaar     curwin = win;
3361071d4279SBram Moolenaar     curbuf = curwin->w_buffer;
3362071d4279SBram Moolenaar #endif
3363071d4279SBram Moolenaar }
3364071d4279SBram Moolenaar 
3365217e1b83SBram Moolenaar // parseMarker() {{{2
3366071d4279SBram Moolenaar /*
3367071d4279SBram Moolenaar  * Parse 'foldmarker' and set "foldendmarker", "foldstartmarkerlen" and
3368071d4279SBram Moolenaar  * "foldendmarkerlen".
3369071d4279SBram Moolenaar  * Relies on the option value to have been checked for correctness already.
3370071d4279SBram Moolenaar  */
3371071d4279SBram Moolenaar     static void
337266f948e9SBram Moolenaar parseMarker(win_T *wp)
3373071d4279SBram Moolenaar {
3374071d4279SBram Moolenaar     foldendmarker = vim_strchr(wp->w_p_fmr, ',');
3375071d4279SBram Moolenaar     foldstartmarkerlen = (int)(foldendmarker++ - wp->w_p_fmr);
3376071d4279SBram Moolenaar     foldendmarkerlen = (int)STRLEN(foldendmarker);
3377071d4279SBram Moolenaar }
3378071d4279SBram Moolenaar 
3379217e1b83SBram Moolenaar // foldlevelMarker() {{{2
3380071d4279SBram Moolenaar /*
3381071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "marker" method.
3382071d4279SBram Moolenaar  * "foldendmarker", "foldstartmarkerlen" and "foldendmarkerlen" must have been
3383071d4279SBram Moolenaar  * set before calling this.
3384071d4279SBram Moolenaar  * Requires that flp->lvl is set to the fold level of the previous line!
3385071d4279SBram Moolenaar  * Careful: This means you can't call this function twice on the same line.
3386071d4279SBram Moolenaar  * Doesn't use any caching.
3387071d4279SBram Moolenaar  * Sets flp->start when a start marker was found.
3388071d4279SBram Moolenaar  */
3389071d4279SBram Moolenaar     static void
339066f948e9SBram Moolenaar foldlevelMarker(fline_T *flp)
3391071d4279SBram Moolenaar {
3392071d4279SBram Moolenaar     char_u	*startmarker;
3393071d4279SBram Moolenaar     int		cstart;
3394071d4279SBram Moolenaar     int		cend;
3395071d4279SBram Moolenaar     int		start_lvl = flp->lvl;
3396071d4279SBram Moolenaar     char_u	*s;
3397071d4279SBram Moolenaar     int		n;
3398071d4279SBram Moolenaar 
3399217e1b83SBram Moolenaar     // cache a few values for speed
3400071d4279SBram Moolenaar     startmarker = flp->wp->w_p_fmr;
3401071d4279SBram Moolenaar     cstart = *startmarker;
3402071d4279SBram Moolenaar     ++startmarker;
3403071d4279SBram Moolenaar     cend = *foldendmarker;
3404071d4279SBram Moolenaar 
3405217e1b83SBram Moolenaar     // Default: no start found, next level is same as current level
3406071d4279SBram Moolenaar     flp->start = 0;
3407071d4279SBram Moolenaar     flp->lvl_next = flp->lvl;
3408071d4279SBram Moolenaar 
3409071d4279SBram Moolenaar     s = ml_get_buf(flp->wp->w_buffer, flp->lnum + flp->off, FALSE);
3410071d4279SBram Moolenaar     while (*s)
3411071d4279SBram Moolenaar     {
3412071d4279SBram Moolenaar 	if (*s == cstart
3413071d4279SBram Moolenaar 		  && STRNCMP(s + 1, startmarker, foldstartmarkerlen - 1) == 0)
3414071d4279SBram Moolenaar 	{
3415217e1b83SBram Moolenaar 	    // found startmarker: set flp->lvl
3416071d4279SBram Moolenaar 	    s += foldstartmarkerlen;
3417071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(*s))
3418071d4279SBram Moolenaar 	    {
3419071d4279SBram Moolenaar 		n = atoi((char *)s);
3420071d4279SBram Moolenaar 		if (n > 0)
3421071d4279SBram Moolenaar 		{
3422071d4279SBram Moolenaar 		    flp->lvl = n;
3423071d4279SBram Moolenaar 		    flp->lvl_next = n;
3424071d4279SBram Moolenaar 		    if (n <= start_lvl)
3425071d4279SBram Moolenaar 			flp->start = 1;
3426071d4279SBram Moolenaar 		    else
3427071d4279SBram Moolenaar 			flp->start = n - start_lvl;
3428071d4279SBram Moolenaar 		}
3429071d4279SBram Moolenaar 	    }
3430071d4279SBram Moolenaar 	    else
3431071d4279SBram Moolenaar 	    {
3432071d4279SBram Moolenaar 		++flp->lvl;
3433071d4279SBram Moolenaar 		++flp->lvl_next;
3434071d4279SBram Moolenaar 		++flp->start;
3435071d4279SBram Moolenaar 	    }
3436071d4279SBram Moolenaar 	}
3437071d4279SBram Moolenaar 	else if (*s == cend
3438071d4279SBram Moolenaar 	      && STRNCMP(s + 1, foldendmarker + 1, foldendmarkerlen - 1) == 0)
3439071d4279SBram Moolenaar 	{
3440217e1b83SBram Moolenaar 	    // found endmarker: set flp->lvl_next
3441071d4279SBram Moolenaar 	    s += foldendmarkerlen;
3442071d4279SBram Moolenaar 	    if (VIM_ISDIGIT(*s))
3443071d4279SBram Moolenaar 	    {
3444071d4279SBram Moolenaar 		n = atoi((char *)s);
3445071d4279SBram Moolenaar 		if (n > 0)
3446071d4279SBram Moolenaar 		{
3447071d4279SBram Moolenaar 		    flp->lvl = n;
3448071d4279SBram Moolenaar 		    flp->lvl_next = n - 1;
3449217e1b83SBram Moolenaar 		    // never start a fold with an end marker
34503ee0229fSBram Moolenaar 		    if (flp->lvl_next > start_lvl)
34513ee0229fSBram Moolenaar 			flp->lvl_next = start_lvl;
3452071d4279SBram Moolenaar 		}
3453071d4279SBram Moolenaar 	    }
3454071d4279SBram Moolenaar 	    else
3455071d4279SBram Moolenaar 		--flp->lvl_next;
3456071d4279SBram Moolenaar 	}
3457071d4279SBram Moolenaar 	else
345891acfffcSBram Moolenaar 	    MB_PTR_ADV(s);
3459071d4279SBram Moolenaar     }
3460071d4279SBram Moolenaar 
3461217e1b83SBram Moolenaar     // The level can't go negative, must be missing a start marker.
3462071d4279SBram Moolenaar     if (flp->lvl_next < 0)
3463071d4279SBram Moolenaar 	flp->lvl_next = 0;
3464071d4279SBram Moolenaar }
3465071d4279SBram Moolenaar 
3466217e1b83SBram Moolenaar // foldlevelSyntax() {{{2
3467071d4279SBram Moolenaar /*
3468071d4279SBram Moolenaar  * Low level function to get the foldlevel for the "syntax" method.
3469071d4279SBram Moolenaar  * Doesn't use any caching.
3470071d4279SBram Moolenaar  */
3471071d4279SBram Moolenaar     static void
347266f948e9SBram Moolenaar foldlevelSyntax(fline_T *flp)
3473071d4279SBram Moolenaar {
3474071d4279SBram Moolenaar #ifndef FEAT_SYN_HL
3475071d4279SBram Moolenaar     flp->start = 0;
3476071d4279SBram Moolenaar     flp->lvl = 0;
3477071d4279SBram Moolenaar #else
3478071d4279SBram Moolenaar     linenr_T	lnum = flp->lnum + flp->off;
3479071d4279SBram Moolenaar     int		n;
3480071d4279SBram Moolenaar 
3481217e1b83SBram Moolenaar     // Use the maximum fold level at the start of this line and the next.
3482071d4279SBram Moolenaar     flp->lvl = syn_get_foldlevel(flp->wp, lnum);
3483071d4279SBram Moolenaar     flp->start = 0;
3484071d4279SBram Moolenaar     if (lnum < flp->wp->w_buffer->b_ml.ml_line_count)
3485071d4279SBram Moolenaar     {
3486071d4279SBram Moolenaar 	n = syn_get_foldlevel(flp->wp, lnum + 1);
3487071d4279SBram Moolenaar 	if (n > flp->lvl)
3488071d4279SBram Moolenaar 	{
3489217e1b83SBram Moolenaar 	    flp->start = n - flp->lvl;	// fold(s) start here
3490071d4279SBram Moolenaar 	    flp->lvl = n;
3491071d4279SBram Moolenaar 	}
3492071d4279SBram Moolenaar     }
3493071d4279SBram Moolenaar #endif
3494071d4279SBram Moolenaar }
3495071d4279SBram Moolenaar 
3496217e1b83SBram Moolenaar // functions for storing the fold state in a View {{{1
3497217e1b83SBram Moolenaar // put_folds() {{{2
3498071d4279SBram Moolenaar #if defined(FEAT_SESSION) || defined(PROTO)
3499d25c16e2SBram Moolenaar static int put_folds_recurse(FILE *fd, garray_T *gap, linenr_T off);
3500d25c16e2SBram Moolenaar static int put_foldopen_recurse(FILE *fd, win_T *wp, garray_T *gap, linenr_T off);
3501d25c16e2SBram Moolenaar static int put_fold_open_close(FILE *fd, fold_T *fp, linenr_T off);
3502071d4279SBram Moolenaar 
3503071d4279SBram Moolenaar /*
3504071d4279SBram Moolenaar  * Write commands to "fd" to restore the manual folds in window "wp".
3505071d4279SBram Moolenaar  * Return FAIL if writing fails.
3506071d4279SBram Moolenaar  */
3507071d4279SBram Moolenaar     int
350866f948e9SBram Moolenaar put_folds(FILE *fd, win_T *wp)
3509071d4279SBram Moolenaar {
3510071d4279SBram Moolenaar     if (foldmethodIsManual(wp))
3511071d4279SBram Moolenaar     {
3512071d4279SBram Moolenaar 	if (put_line(fd, "silent! normal! zE") == FAIL
3513071d4279SBram Moolenaar 		|| put_folds_recurse(fd, &wp->w_folds, (linenr_T)0) == FAIL)
3514071d4279SBram Moolenaar 	    return FAIL;
3515071d4279SBram Moolenaar     }
3516071d4279SBram Moolenaar 
3517217e1b83SBram Moolenaar     // If some folds are manually opened/closed, need to restore that.
3518071d4279SBram Moolenaar     if (wp->w_fold_manual)
3519d25ad65aSBram Moolenaar 	return put_foldopen_recurse(fd, wp, &wp->w_folds, (linenr_T)0);
3520071d4279SBram Moolenaar 
3521071d4279SBram Moolenaar     return OK;
3522071d4279SBram Moolenaar }
3523071d4279SBram Moolenaar 
3524217e1b83SBram Moolenaar // put_folds_recurse() {{{2
3525071d4279SBram Moolenaar /*
3526071d4279SBram Moolenaar  * Write commands to "fd" to recreate manually created folds.
3527071d4279SBram Moolenaar  * Returns FAIL when writing failed.
3528071d4279SBram Moolenaar  */
3529071d4279SBram Moolenaar     static int
353066f948e9SBram Moolenaar put_folds_recurse(FILE *fd, garray_T *gap, linenr_T off)
3531071d4279SBram Moolenaar {
3532071d4279SBram Moolenaar     int		i;
3533071d4279SBram Moolenaar     fold_T	*fp;
3534071d4279SBram Moolenaar 
3535071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
3536071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; i++)
3537071d4279SBram Moolenaar     {
3538217e1b83SBram Moolenaar 	// Do nested folds first, they will be created closed.
3539071d4279SBram Moolenaar 	if (put_folds_recurse(fd, &fp->fd_nested, off + fp->fd_top) == FAIL)
3540071d4279SBram Moolenaar 	    return FAIL;
3541071d4279SBram Moolenaar 	if (fprintf(fd, "%ld,%ldfold", fp->fd_top + off,
3542071d4279SBram Moolenaar 					fp->fd_top + off + fp->fd_len - 1) < 0
3543071d4279SBram Moolenaar 		|| put_eol(fd) == FAIL)
3544071d4279SBram Moolenaar 	    return FAIL;
3545071d4279SBram Moolenaar 	++fp;
3546071d4279SBram Moolenaar     }
3547071d4279SBram Moolenaar     return OK;
3548071d4279SBram Moolenaar }
3549071d4279SBram Moolenaar 
3550217e1b83SBram Moolenaar // put_foldopen_recurse() {{{2
3551071d4279SBram Moolenaar /*
3552071d4279SBram Moolenaar  * Write commands to "fd" to open and close manually opened/closed folds.
3553071d4279SBram Moolenaar  * Returns FAIL when writing failed.
3554071d4279SBram Moolenaar  */
3555071d4279SBram Moolenaar     static int
355666f948e9SBram Moolenaar put_foldopen_recurse(
355766f948e9SBram Moolenaar     FILE	*fd,
355866f948e9SBram Moolenaar     win_T	*wp,
355966f948e9SBram Moolenaar     garray_T	*gap,
356066f948e9SBram Moolenaar     linenr_T	off)
3561071d4279SBram Moolenaar {
3562071d4279SBram Moolenaar     int		i;
3563d25ad65aSBram Moolenaar     int		level;
3564071d4279SBram Moolenaar     fold_T	*fp;
3565071d4279SBram Moolenaar 
3566071d4279SBram Moolenaar     fp = (fold_T *)gap->ga_data;
3567071d4279SBram Moolenaar     for (i = 0; i < gap->ga_len; i++)
3568071d4279SBram Moolenaar     {
3569071d4279SBram Moolenaar 	if (fp->fd_flags != FD_LEVEL)
3570071d4279SBram Moolenaar 	{
3571071d4279SBram Moolenaar 	    if (fp->fd_nested.ga_len > 0)
3572071d4279SBram Moolenaar 	    {
3573217e1b83SBram Moolenaar 		// open nested folds while this fold is open
3574071d4279SBram Moolenaar 		if (fprintf(fd, "%ld", fp->fd_top + off) < 0
3575071d4279SBram Moolenaar 			|| put_eol(fd) == FAIL
3576002a4edcSBram Moolenaar 			|| put_line(fd, "normal! zo") == FAIL)
3577071d4279SBram Moolenaar 		    return FAIL;
3578d25ad65aSBram Moolenaar 		if (put_foldopen_recurse(fd, wp, &fp->fd_nested,
3579d25ad65aSBram Moolenaar 							     off + fp->fd_top)
3580071d4279SBram Moolenaar 			== FAIL)
3581071d4279SBram Moolenaar 		    return FAIL;
3582217e1b83SBram Moolenaar 		// close the parent when needed
3583d25ad65aSBram Moolenaar 		if (fp->fd_flags == FD_CLOSED)
3584d25ad65aSBram Moolenaar 		{
3585d25ad65aSBram Moolenaar 		    if (put_fold_open_close(fd, fp, off) == FAIL)
3586d25ad65aSBram Moolenaar 			return FAIL;
3587071d4279SBram Moolenaar 		}
3588d25ad65aSBram Moolenaar 	    }
3589d25ad65aSBram Moolenaar 	    else
3590d25ad65aSBram Moolenaar 	    {
3591217e1b83SBram Moolenaar 		// Open or close the leaf according to the window foldlevel.
3592217e1b83SBram Moolenaar 		// Do not close a leaf that is already closed, as it will close
3593217e1b83SBram Moolenaar 		// the parent.
3594d25ad65aSBram Moolenaar 		level = foldLevelWin(wp, off + fp->fd_top);
3595d25ad65aSBram Moolenaar 		if ((fp->fd_flags == FD_CLOSED && wp->w_p_fdl >= level)
3596d25ad65aSBram Moolenaar 			|| (fp->fd_flags != FD_CLOSED && wp->w_p_fdl < level))
3597d25ad65aSBram Moolenaar 		if (put_fold_open_close(fd, fp, off) == FAIL)
3598d25ad65aSBram Moolenaar 		    return FAIL;
3599d25ad65aSBram Moolenaar 	    }
3600d25ad65aSBram Moolenaar 	}
3601d25ad65aSBram Moolenaar 	++fp;
3602d25ad65aSBram Moolenaar     }
3603d25ad65aSBram Moolenaar 
3604d25ad65aSBram Moolenaar     return OK;
3605d25ad65aSBram Moolenaar }
3606d25ad65aSBram Moolenaar 
3607217e1b83SBram Moolenaar // put_fold_open_close() {{{2
3608d25ad65aSBram Moolenaar /*
3609d25ad65aSBram Moolenaar  * Write the open or close command to "fd".
3610d25ad65aSBram Moolenaar  * Returns FAIL when writing failed.
3611d25ad65aSBram Moolenaar  */
3612d25ad65aSBram Moolenaar     static int
361366f948e9SBram Moolenaar put_fold_open_close(FILE *fd, fold_T *fp, linenr_T off)
3614d25ad65aSBram Moolenaar {
3615071d4279SBram Moolenaar     if (fprintf(fd, "%ld", fp->fd_top + off) < 0
3616071d4279SBram Moolenaar 	    || put_eol(fd) == FAIL
3617002a4edcSBram Moolenaar 	    || fprintf(fd, "normal! z%c",
3618071d4279SBram Moolenaar 			   fp->fd_flags == FD_CLOSED ? 'c' : 'o') < 0
3619071d4279SBram Moolenaar 	    || put_eol(fd) == FAIL)
3620071d4279SBram Moolenaar 	return FAIL;
3621071d4279SBram Moolenaar 
3622071d4279SBram Moolenaar     return OK;
3623071d4279SBram Moolenaar }
3624217e1b83SBram Moolenaar #endif // FEAT_SESSION
3625071d4279SBram Moolenaar 
3626217e1b83SBram Moolenaar // }}}1
3627db022f3fSBram Moolenaar #endif // defined(FEAT_FOLDING) || defined(PROTO)
3628db022f3fSBram Moolenaar 
3629db022f3fSBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO)
3630db022f3fSBram Moolenaar 
3631db022f3fSBram Moolenaar /*
3632db022f3fSBram Moolenaar  * "foldclosed()" and "foldclosedend()" functions
3633db022f3fSBram Moolenaar  */
3634db022f3fSBram Moolenaar     static void
3635db022f3fSBram Moolenaar foldclosed_both(
3636db022f3fSBram Moolenaar     typval_T	*argvars UNUSED,
3637db022f3fSBram Moolenaar     typval_T	*rettv,
3638db022f3fSBram Moolenaar     int		end UNUSED)
3639db022f3fSBram Moolenaar {
3640db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3641db022f3fSBram Moolenaar     linenr_T	lnum;
3642db022f3fSBram Moolenaar     linenr_T	first, last;
3643db022f3fSBram Moolenaar 
3644db022f3fSBram Moolenaar     lnum = tv_get_lnum(argvars);
3645db022f3fSBram Moolenaar     if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
3646db022f3fSBram Moolenaar     {
3647db022f3fSBram Moolenaar 	if (hasFoldingWin(curwin, lnum, &first, &last, FALSE, NULL))
3648db022f3fSBram Moolenaar 	{
3649db022f3fSBram Moolenaar 	    if (end)
3650db022f3fSBram Moolenaar 		rettv->vval.v_number = (varnumber_T)last;
3651db022f3fSBram Moolenaar 	    else
3652db022f3fSBram Moolenaar 		rettv->vval.v_number = (varnumber_T)first;
3653db022f3fSBram Moolenaar 	    return;
3654db022f3fSBram Moolenaar 	}
3655db022f3fSBram Moolenaar     }
3656db022f3fSBram Moolenaar #endif
3657db022f3fSBram Moolenaar     rettv->vval.v_number = -1;
3658db022f3fSBram Moolenaar }
3659db022f3fSBram Moolenaar 
3660db022f3fSBram Moolenaar /*
3661db022f3fSBram Moolenaar  * "foldclosed()" function
3662db022f3fSBram Moolenaar  */
3663db022f3fSBram Moolenaar     void
3664db022f3fSBram Moolenaar f_foldclosed(typval_T *argvars, typval_T *rettv)
3665db022f3fSBram Moolenaar {
3666db022f3fSBram Moolenaar     foldclosed_both(argvars, rettv, FALSE);
3667db022f3fSBram Moolenaar }
3668db022f3fSBram Moolenaar 
3669db022f3fSBram Moolenaar /*
3670db022f3fSBram Moolenaar  * "foldclosedend()" function
3671db022f3fSBram Moolenaar  */
3672db022f3fSBram Moolenaar     void
3673db022f3fSBram Moolenaar f_foldclosedend(typval_T *argvars, typval_T *rettv)
3674db022f3fSBram Moolenaar {
3675db022f3fSBram Moolenaar     foldclosed_both(argvars, rettv, TRUE);
3676db022f3fSBram Moolenaar }
3677db022f3fSBram Moolenaar 
3678db022f3fSBram Moolenaar /*
3679db022f3fSBram Moolenaar  * "foldlevel()" function
3680db022f3fSBram Moolenaar  */
3681db022f3fSBram Moolenaar     void
3682db022f3fSBram Moolenaar f_foldlevel(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3683db022f3fSBram Moolenaar {
3684db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3685db022f3fSBram Moolenaar     linenr_T	lnum;
3686db022f3fSBram Moolenaar 
3687db022f3fSBram Moolenaar     lnum = tv_get_lnum(argvars);
3688db022f3fSBram Moolenaar     if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
3689db022f3fSBram Moolenaar 	rettv->vval.v_number = foldLevel(lnum);
3690db022f3fSBram Moolenaar # endif
3691db022f3fSBram Moolenaar }
3692db022f3fSBram Moolenaar 
3693db022f3fSBram Moolenaar /*
3694db022f3fSBram Moolenaar  * "foldtext()" function
3695db022f3fSBram Moolenaar  */
3696db022f3fSBram Moolenaar     void
3697db022f3fSBram Moolenaar f_foldtext(typval_T *argvars UNUSED, typval_T *rettv)
3698db022f3fSBram Moolenaar {
3699db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3700db022f3fSBram Moolenaar     linenr_T	foldstart;
3701db022f3fSBram Moolenaar     linenr_T	foldend;
3702db022f3fSBram Moolenaar     char_u	*dashes;
3703db022f3fSBram Moolenaar     linenr_T	lnum;
3704db022f3fSBram Moolenaar     char_u	*s;
3705db022f3fSBram Moolenaar     char_u	*r;
3706db022f3fSBram Moolenaar     int		len;
3707db022f3fSBram Moolenaar     char	*txt;
3708db022f3fSBram Moolenaar     long	count;
3709db022f3fSBram Moolenaar # endif
3710db022f3fSBram Moolenaar 
3711db022f3fSBram Moolenaar     rettv->v_type = VAR_STRING;
3712db022f3fSBram Moolenaar     rettv->vval.v_string = NULL;
3713db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3714db022f3fSBram Moolenaar     foldstart = (linenr_T)get_vim_var_nr(VV_FOLDSTART);
3715db022f3fSBram Moolenaar     foldend = (linenr_T)get_vim_var_nr(VV_FOLDEND);
3716db022f3fSBram Moolenaar     dashes = get_vim_var_str(VV_FOLDDASHES);
3717db022f3fSBram Moolenaar     if (foldstart > 0 && foldend <= curbuf->b_ml.ml_line_count
3718db022f3fSBram Moolenaar 	    && dashes != NULL)
3719db022f3fSBram Moolenaar     {
3720db022f3fSBram Moolenaar 	// Find first non-empty line in the fold.
3721db022f3fSBram Moolenaar 	for (lnum = foldstart; lnum < foldend; ++lnum)
3722db022f3fSBram Moolenaar 	    if (!linewhite(lnum))
3723db022f3fSBram Moolenaar 		break;
3724db022f3fSBram Moolenaar 
3725db022f3fSBram Moolenaar 	// Find interesting text in this line.
3726db022f3fSBram Moolenaar 	s = skipwhite(ml_get(lnum));
3727db022f3fSBram Moolenaar 	// skip C comment-start
3728db022f3fSBram Moolenaar 	if (s[0] == '/' && (s[1] == '*' || s[1] == '/'))
3729db022f3fSBram Moolenaar 	{
3730db022f3fSBram Moolenaar 	    s = skipwhite(s + 2);
3731db022f3fSBram Moolenaar 	    if (*skipwhite(s) == NUL
3732db022f3fSBram Moolenaar 			    && lnum + 1 < (linenr_T)get_vim_var_nr(VV_FOLDEND))
3733db022f3fSBram Moolenaar 	    {
3734db022f3fSBram Moolenaar 		s = skipwhite(ml_get(lnum + 1));
3735db022f3fSBram Moolenaar 		if (*s == '*')
3736db022f3fSBram Moolenaar 		    s = skipwhite(s + 1);
3737db022f3fSBram Moolenaar 	    }
3738db022f3fSBram Moolenaar 	}
3739db022f3fSBram Moolenaar 	count = (long)(foldend - foldstart + 1);
3740db022f3fSBram Moolenaar 	txt = NGETTEXT("+-%s%3ld line: ", "+-%s%3ld lines: ", count);
3741db022f3fSBram Moolenaar 	r = alloc(STRLEN(txt)
3742db022f3fSBram Moolenaar 		    + STRLEN(dashes)	    // for %s
3743db022f3fSBram Moolenaar 		    + 20		    // for %3ld
3744db022f3fSBram Moolenaar 		    + STRLEN(s));	    // concatenated
3745db022f3fSBram Moolenaar 	if (r != NULL)
3746db022f3fSBram Moolenaar 	{
3747db022f3fSBram Moolenaar 	    sprintf((char *)r, txt, dashes, count);
3748db022f3fSBram Moolenaar 	    len = (int)STRLEN(r);
3749db022f3fSBram Moolenaar 	    STRCAT(r, s);
3750db022f3fSBram Moolenaar 	    // remove 'foldmarker' and 'commentstring'
3751db022f3fSBram Moolenaar 	    foldtext_cleanup(r + len);
3752db022f3fSBram Moolenaar 	    rettv->vval.v_string = r;
3753db022f3fSBram Moolenaar 	}
3754db022f3fSBram Moolenaar     }
3755db022f3fSBram Moolenaar # endif
3756db022f3fSBram Moolenaar }
3757db022f3fSBram Moolenaar 
3758db022f3fSBram Moolenaar /*
3759db022f3fSBram Moolenaar  * "foldtextresult(lnum)" function
3760db022f3fSBram Moolenaar  */
3761db022f3fSBram Moolenaar     void
3762db022f3fSBram Moolenaar f_foldtextresult(typval_T *argvars UNUSED, typval_T *rettv)
3763db022f3fSBram Moolenaar {
3764db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3765db022f3fSBram Moolenaar     linenr_T	lnum;
3766db022f3fSBram Moolenaar     char_u	*text;
3767db022f3fSBram Moolenaar     char_u	buf[FOLD_TEXT_LEN];
3768db022f3fSBram Moolenaar     foldinfo_T  foldinfo;
3769db022f3fSBram Moolenaar     int		fold_count;
3770db022f3fSBram Moolenaar     static int	entered = FALSE;
3771db022f3fSBram Moolenaar # endif
3772db022f3fSBram Moolenaar 
3773db022f3fSBram Moolenaar     rettv->v_type = VAR_STRING;
3774db022f3fSBram Moolenaar     rettv->vval.v_string = NULL;
3775db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3776db022f3fSBram Moolenaar     if (entered)
3777db022f3fSBram Moolenaar 	return; // reject recursive use
3778db022f3fSBram Moolenaar     entered = TRUE;
3779db022f3fSBram Moolenaar 
3780db022f3fSBram Moolenaar     lnum = tv_get_lnum(argvars);
3781db022f3fSBram Moolenaar     // treat illegal types and illegal string values for {lnum} the same
3782db022f3fSBram Moolenaar     if (lnum < 0)
3783db022f3fSBram Moolenaar 	lnum = 0;
3784db022f3fSBram Moolenaar     fold_count = foldedCount(curwin, lnum, &foldinfo);
3785db022f3fSBram Moolenaar     if (fold_count > 0)
3786db022f3fSBram Moolenaar     {
3787db022f3fSBram Moolenaar 	text = get_foldtext(curwin, lnum, lnum + fold_count - 1,
3788db022f3fSBram Moolenaar 							       &foldinfo, buf);
3789db022f3fSBram Moolenaar 	if (text == buf)
3790db022f3fSBram Moolenaar 	    text = vim_strsave(text);
3791db022f3fSBram Moolenaar 	rettv->vval.v_string = text;
3792db022f3fSBram Moolenaar     }
3793db022f3fSBram Moolenaar 
3794db022f3fSBram Moolenaar     entered = FALSE;
3795db022f3fSBram Moolenaar # endif
3796db022f3fSBram Moolenaar }
3797db022f3fSBram Moolenaar 
3798db022f3fSBram Moolenaar #endif // FEAT_EVAL
3799