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
copyFoldingState(win_T * wp_from,win_T * wp_to)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
hasAnyFolding(win_T * win)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
hasFolding(linenr_T lnum,linenr_T * firstp,linenr_T * lastp)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
hasFoldingWin(win_T * win,linenr_T lnum,linenr_T * firstp,linenr_T * lastp,int cache,foldinfo_T * infop)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
foldLevel(linenr_T lnum)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
lineFolded(win_T * win,linenr_T lnum)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
foldedCount(win_T * win,linenr_T lnum,foldinfo_T * infop)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
foldmethodIsManual(win_T * wp)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
foldmethodIsIndent(win_T * wp)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
foldmethodIsExpr(win_T * wp)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
foldmethodIsMarker(win_T * wp)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
foldmethodIsSyntax(win_T * wp)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
foldmethodIsDiff(win_T * wp)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
closeFold(linenr_T lnum,long count)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
closeFoldRecurse(linenr_T lnum)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
opFoldRange(linenr_T first,linenr_T last,int opening,int recurse,int had_visual)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
openFold(linenr_T lnum,long count)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
openFoldRecurse(linenr_T lnum)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
foldOpenCursor(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
newFoldLevel(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
newFoldLevelWin(win_T * wp)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
foldCheckClose(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
checkCloseRec(garray_T * gap,linenr_T lnum,int level)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
foldManualAllowed(int create)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
foldCreate(linenr_T start,linenr_T end)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;
6112c93c685SBram Moolenaar if (gap->ga_len == 0)
6122c93c685SBram Moolenaar i = 0;
6132c93c685SBram Moolenaar else
6142c93c685SBram Moolenaar {
615071d4279SBram Moolenaar for (;;)
616071d4279SBram Moolenaar {
617071d4279SBram Moolenaar if (!foldFind(gap, start_rel, &fp))
618071d4279SBram Moolenaar break;
619071d4279SBram Moolenaar if (fp->fd_top + fp->fd_len > end_rel)
620071d4279SBram Moolenaar {
6212c93c685SBram Moolenaar // New fold is completely inside this fold: Go one level
6222c93c685SBram Moolenaar // deeper.
623071d4279SBram Moolenaar gap = &fp->fd_nested;
624071d4279SBram Moolenaar start_rel -= fp->fd_top;
625071d4279SBram Moolenaar end_rel -= fp->fd_top;
626071d4279SBram Moolenaar if (use_level || fp->fd_flags == FD_LEVEL)
627071d4279SBram Moolenaar {
628071d4279SBram Moolenaar use_level = TRUE;
629071d4279SBram Moolenaar if (level >= curwin->w_p_fdl)
630071d4279SBram Moolenaar closed = TRUE;
631071d4279SBram Moolenaar }
632071d4279SBram Moolenaar else if (fp->fd_flags == FD_CLOSED)
633071d4279SBram Moolenaar closed = TRUE;
634071d4279SBram Moolenaar ++level;
635071d4279SBram Moolenaar }
636071d4279SBram Moolenaar else
637071d4279SBram Moolenaar {
6382c93c685SBram Moolenaar // This fold and new fold overlap: Insert here and move some
6392c93c685SBram Moolenaar // folds inside the new fold.
640071d4279SBram Moolenaar break;
641071d4279SBram Moolenaar }
642071d4279SBram Moolenaar }
6435e1f22ffSBram Moolenaar if (gap->ga_len == 0)
6445e1f22ffSBram Moolenaar i = 0;
6455e1f22ffSBram Moolenaar else
646071d4279SBram Moolenaar i = (int)(fp - (fold_T *)gap->ga_data);
6472c93c685SBram Moolenaar }
6482c93c685SBram Moolenaar
649071d4279SBram Moolenaar if (ga_grow(gap, 1) == OK)
650071d4279SBram Moolenaar {
651071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i;
652071d4279SBram Moolenaar ga_init2(&fold_ga, (int)sizeof(fold_T), 10);
653071d4279SBram Moolenaar
654217e1b83SBram Moolenaar // Count number of folds that will be contained in the new fold.
655071d4279SBram Moolenaar for (cont = 0; i + cont < gap->ga_len; ++cont)
656071d4279SBram Moolenaar if (fp[cont].fd_top > end_rel)
657071d4279SBram Moolenaar break;
658071d4279SBram Moolenaar if (cont > 0 && ga_grow(&fold_ga, cont) == OK)
659071d4279SBram Moolenaar {
660217e1b83SBram Moolenaar // If the first fold starts before the new fold, let the new fold
661217e1b83SBram Moolenaar // start there. Otherwise the existing fold would change.
662071d4279SBram Moolenaar if (start_rel > fp->fd_top)
663071d4279SBram Moolenaar start_rel = fp->fd_top;
664071d4279SBram Moolenaar
665217e1b83SBram Moolenaar // When last contained fold isn't completely contained, adjust end
666217e1b83SBram Moolenaar // of new fold.
667071d4279SBram Moolenaar if (end_rel < fp[cont - 1].fd_top + fp[cont - 1].fd_len - 1)
668071d4279SBram Moolenaar end_rel = fp[cont - 1].fd_top + fp[cont - 1].fd_len - 1;
669217e1b83SBram Moolenaar // Move contained folds to inside new fold.
670071d4279SBram Moolenaar mch_memmove(fold_ga.ga_data, fp, sizeof(fold_T) * cont);
671071d4279SBram Moolenaar fold_ga.ga_len += cont;
672071d4279SBram Moolenaar i += cont;
673071d4279SBram Moolenaar
674217e1b83SBram Moolenaar // Adjust line numbers in contained folds to be relative to the
675217e1b83SBram Moolenaar // new fold.
676071d4279SBram Moolenaar for (j = 0; j < cont; ++j)
677071d4279SBram Moolenaar ((fold_T *)fold_ga.ga_data)[j].fd_top -= start_rel;
678071d4279SBram Moolenaar }
679217e1b83SBram Moolenaar // Move remaining entries to after the new fold.
680071d4279SBram Moolenaar if (i < gap->ga_len)
681071d4279SBram Moolenaar mch_memmove(fp + 1, (fold_T *)gap->ga_data + i,
682071d4279SBram Moolenaar sizeof(fold_T) * (gap->ga_len - i));
683071d4279SBram Moolenaar gap->ga_len = gap->ga_len + 1 - cont;
684071d4279SBram Moolenaar
685217e1b83SBram Moolenaar // insert new fold
686071d4279SBram Moolenaar fp->fd_nested = fold_ga;
687071d4279SBram Moolenaar fp->fd_top = start_rel;
688071d4279SBram Moolenaar fp->fd_len = end_rel - start_rel + 1;
689071d4279SBram Moolenaar
690217e1b83SBram Moolenaar // We want the new fold to be closed. If it would remain open because
691217e1b83SBram Moolenaar // of using 'foldlevel', need to adjust fd_flags of containing folds.
692071d4279SBram Moolenaar if (use_level && !closed && level < curwin->w_p_fdl)
693071d4279SBram Moolenaar closeFold(start, 1L);
694071d4279SBram Moolenaar if (!use_level)
695071d4279SBram Moolenaar curwin->w_fold_manual = TRUE;
696071d4279SBram Moolenaar fp->fd_flags = FD_CLOSED;
697071d4279SBram Moolenaar fp->fd_small = MAYBE;
698071d4279SBram Moolenaar
699217e1b83SBram Moolenaar // redraw
700071d4279SBram Moolenaar changed_window_setting();
701071d4279SBram Moolenaar }
702071d4279SBram Moolenaar }
703071d4279SBram Moolenaar
704217e1b83SBram Moolenaar // deleteFold() {{{2
705071d4279SBram Moolenaar /*
706071d4279SBram Moolenaar * Delete a fold at line "start" in the current window.
707071d4279SBram Moolenaar * When "end" is not 0, delete all folds from "start" to "end".
708071d4279SBram Moolenaar * When "recursive" is TRUE delete recursively.
709071d4279SBram Moolenaar */
710071d4279SBram Moolenaar void
deleteFold(linenr_T start,linenr_T end,int recursive,int had_visual)71166f948e9SBram Moolenaar deleteFold(
71266f948e9SBram Moolenaar linenr_T start,
71366f948e9SBram Moolenaar linenr_T end,
71466f948e9SBram Moolenaar int recursive,
715217e1b83SBram Moolenaar int had_visual) // TRUE when Visual selection used
716071d4279SBram Moolenaar {
717071d4279SBram Moolenaar garray_T *gap;
718071d4279SBram Moolenaar fold_T *fp;
719071d4279SBram Moolenaar garray_T *found_ga;
720071d4279SBram Moolenaar fold_T *found_fp = NULL;
721071d4279SBram Moolenaar linenr_T found_off = 0;
722238b8e20SBram Moolenaar int use_level;
723071d4279SBram Moolenaar int maybe_small = FALSE;
724071d4279SBram Moolenaar int level = 0;
725071d4279SBram Moolenaar linenr_T lnum = start;
726071d4279SBram Moolenaar linenr_T lnum_off;
727071d4279SBram Moolenaar int did_one = FALSE;
728071d4279SBram Moolenaar linenr_T first_lnum = MAXLNUM;
729071d4279SBram Moolenaar linenr_T last_lnum = 0;
730071d4279SBram Moolenaar
731071d4279SBram Moolenaar checkupdate(curwin);
732071d4279SBram Moolenaar
733071d4279SBram Moolenaar while (lnum <= end)
734071d4279SBram Moolenaar {
735217e1b83SBram Moolenaar // Find the deepest fold for "start".
736071d4279SBram Moolenaar gap = &curwin->w_folds;
737071d4279SBram Moolenaar found_ga = NULL;
738071d4279SBram Moolenaar lnum_off = 0;
739238b8e20SBram Moolenaar use_level = FALSE;
740071d4279SBram Moolenaar for (;;)
741071d4279SBram Moolenaar {
742071d4279SBram Moolenaar if (!foldFind(gap, lnum - lnum_off, &fp))
743071d4279SBram Moolenaar break;
744217e1b83SBram Moolenaar // lnum is inside this fold, remember info
745071d4279SBram Moolenaar found_ga = gap;
746071d4279SBram Moolenaar found_fp = fp;
747071d4279SBram Moolenaar found_off = lnum_off;
748071d4279SBram Moolenaar
749217e1b83SBram Moolenaar // if "lnum" is folded, don't check nesting
750071d4279SBram Moolenaar if (check_closed(curwin, fp, &use_level, level,
751071d4279SBram Moolenaar &maybe_small, lnum_off))
752071d4279SBram Moolenaar break;
753071d4279SBram Moolenaar
754217e1b83SBram Moolenaar // check nested folds
755071d4279SBram Moolenaar gap = &fp->fd_nested;
756071d4279SBram Moolenaar lnum_off += fp->fd_top;
757071d4279SBram Moolenaar ++level;
758071d4279SBram Moolenaar }
759071d4279SBram Moolenaar if (found_ga == NULL)
760071d4279SBram Moolenaar {
761071d4279SBram Moolenaar ++lnum;
762071d4279SBram Moolenaar }
763071d4279SBram Moolenaar else
764071d4279SBram Moolenaar {
765071d4279SBram Moolenaar lnum = found_fp->fd_top + found_fp->fd_len + found_off;
766071d4279SBram Moolenaar
767071d4279SBram Moolenaar if (foldmethodIsManual(curwin))
768071d4279SBram Moolenaar deleteFoldEntry(found_ga,
769071d4279SBram Moolenaar (int)(found_fp - (fold_T *)found_ga->ga_data), recursive);
770071d4279SBram Moolenaar else
771071d4279SBram Moolenaar {
772238b8e20SBram Moolenaar if (first_lnum > found_fp->fd_top + found_off)
773238b8e20SBram Moolenaar first_lnum = found_fp->fd_top + found_off;
774238b8e20SBram Moolenaar if (last_lnum < lnum)
775071d4279SBram Moolenaar last_lnum = lnum;
776238b8e20SBram Moolenaar if (!did_one)
777071d4279SBram Moolenaar parseMarker(curwin);
778071d4279SBram Moolenaar deleteFoldMarkers(found_fp, recursive, found_off);
779071d4279SBram Moolenaar }
780238b8e20SBram Moolenaar did_one = TRUE;
781071d4279SBram Moolenaar
782217e1b83SBram Moolenaar // redraw window
783071d4279SBram Moolenaar changed_window_setting();
784071d4279SBram Moolenaar }
785071d4279SBram Moolenaar }
786071d4279SBram Moolenaar if (!did_one)
787071d4279SBram Moolenaar {
788f9e3e09fSBram Moolenaar emsg(_(e_nofold));
789217e1b83SBram Moolenaar // Force a redraw to remove the Visual highlighting.
790071d4279SBram Moolenaar if (had_visual)
791071d4279SBram Moolenaar redraw_curbuf_later(INVERTED);
792071d4279SBram Moolenaar }
793238b8e20SBram Moolenaar else
794217e1b83SBram Moolenaar // Deleting markers may make cursor column invalid.
795238b8e20SBram Moolenaar check_cursor_col();
796238b8e20SBram Moolenaar
797071d4279SBram Moolenaar if (last_lnum > 0)
798071d4279SBram Moolenaar changed_lines(first_lnum, (colnr_T)0, last_lnum, 0L);
799071d4279SBram Moolenaar }
800071d4279SBram Moolenaar
801217e1b83SBram Moolenaar // clearFolding() {{{2
802071d4279SBram Moolenaar /*
803071d4279SBram Moolenaar * Remove all folding for window "win".
804071d4279SBram Moolenaar */
805071d4279SBram Moolenaar void
clearFolding(win_T * win)80666f948e9SBram Moolenaar clearFolding(win_T *win)
807071d4279SBram Moolenaar {
808071d4279SBram Moolenaar deleteFoldRecurse(&win->w_folds);
809071d4279SBram Moolenaar win->w_foldinvalid = FALSE;
810071d4279SBram Moolenaar }
811071d4279SBram Moolenaar
812217e1b83SBram Moolenaar // foldUpdate() {{{2
813071d4279SBram Moolenaar /*
814071d4279SBram Moolenaar * Update folds for changes in the buffer of a window.
815071d4279SBram Moolenaar * Note that inserted/deleted lines must have already been taken care of by
816071d4279SBram Moolenaar * calling foldMarkAdjust().
817071d4279SBram Moolenaar * The changes in lines from top to bot (inclusive).
818071d4279SBram Moolenaar */
819071d4279SBram Moolenaar void
foldUpdate(win_T * wp,linenr_T top,linenr_T bot)82066f948e9SBram Moolenaar foldUpdate(win_T *wp, linenr_T top, linenr_T bot)
821071d4279SBram Moolenaar {
822071d4279SBram Moolenaar fold_T *fp;
823071d4279SBram Moolenaar
824429fcfbfSBram Moolenaar if (disable_fold_update > 0)
825429fcfbfSBram Moolenaar return;
8264f57eefeSBram Moolenaar #ifdef FEAT_DIFF
8274f57eefeSBram Moolenaar if (need_diff_redraw)
8284f57eefeSBram Moolenaar // will update later
8294f57eefeSBram Moolenaar return;
8304f57eefeSBram Moolenaar #endif
831429fcfbfSBram Moolenaar
83264f37d30SBram Moolenaar if (wp->w_folds.ga_len > 0)
83364f37d30SBram Moolenaar {
834217e1b83SBram Moolenaar // Mark all folds from top to bot as maybe-small.
835d5f6933dSBram Moolenaar (void)foldFind(&wp->w_folds, top, &fp);
836d5f6933dSBram Moolenaar while (fp < (fold_T *)wp->w_folds.ga_data + wp->w_folds.ga_len
837071d4279SBram Moolenaar && fp->fd_top < bot)
838071d4279SBram Moolenaar {
839071d4279SBram Moolenaar fp->fd_small = MAYBE;
840071d4279SBram Moolenaar ++fp;
841071d4279SBram Moolenaar }
84264f37d30SBram Moolenaar }
843071d4279SBram Moolenaar
844071d4279SBram Moolenaar if (foldmethodIsIndent(wp)
845071d4279SBram Moolenaar || foldmethodIsExpr(wp)
846071d4279SBram Moolenaar || foldmethodIsMarker(wp)
847071d4279SBram Moolenaar #ifdef FEAT_DIFF
848071d4279SBram Moolenaar || foldmethodIsDiff(wp)
849071d4279SBram Moolenaar #endif
850071d4279SBram Moolenaar || foldmethodIsSyntax(wp))
851fa6d5afdSBram Moolenaar {
852fa6d5afdSBram Moolenaar int save_got_int = got_int;
853fa6d5afdSBram Moolenaar
854217e1b83SBram Moolenaar // reset got_int here, otherwise it won't work
855fa6d5afdSBram Moolenaar got_int = FALSE;
856071d4279SBram Moolenaar foldUpdateIEMS(wp, top, bot);
857fa6d5afdSBram Moolenaar got_int |= save_got_int;
858fa6d5afdSBram Moolenaar }
859071d4279SBram Moolenaar }
860071d4279SBram Moolenaar
861217e1b83SBram Moolenaar // foldUpdateAll() {{{2
862071d4279SBram Moolenaar /*
863071d4279SBram Moolenaar * Update all lines in a window for folding.
864071d4279SBram Moolenaar * Used when a fold setting changes or after reloading the buffer.
865071d4279SBram Moolenaar * The actual updating is postponed until fold info is used, to avoid doing
866071d4279SBram Moolenaar * every time a setting is changed or a syntax item is added.
867071d4279SBram Moolenaar */
868071d4279SBram Moolenaar void
foldUpdateAll(win_T * win)86966f948e9SBram Moolenaar foldUpdateAll(win_T *win)
870071d4279SBram Moolenaar {
871071d4279SBram Moolenaar win->w_foldinvalid = TRUE;
872071d4279SBram Moolenaar redraw_win_later(win, NOT_VALID);
873071d4279SBram Moolenaar }
874071d4279SBram Moolenaar
875217e1b83SBram Moolenaar // foldMoveTo() {{{2
876071d4279SBram Moolenaar /*
877071d4279SBram Moolenaar * If "updown" is FALSE: Move to the start or end of the fold.
878071d4279SBram Moolenaar * If "updown" is TRUE: move to fold at the same level.
879071d4279SBram Moolenaar * If not moved return FAIL.
880071d4279SBram Moolenaar */
881071d4279SBram Moolenaar int
foldMoveTo(int updown,int dir,long count)88266f948e9SBram Moolenaar foldMoveTo(
88366f948e9SBram Moolenaar int updown,
884217e1b83SBram Moolenaar int dir, // FORWARD or BACKWARD
88566f948e9SBram Moolenaar long count)
886071d4279SBram Moolenaar {
887071d4279SBram Moolenaar long n;
888071d4279SBram Moolenaar int retval = FAIL;
889071d4279SBram Moolenaar linenr_T lnum_off;
890071d4279SBram Moolenaar linenr_T lnum_found;
891071d4279SBram Moolenaar linenr_T lnum;
892071d4279SBram Moolenaar int use_level;
893071d4279SBram Moolenaar int maybe_small;
894071d4279SBram Moolenaar garray_T *gap;
895071d4279SBram Moolenaar fold_T *fp;
896071d4279SBram Moolenaar int level;
897071d4279SBram Moolenaar int last;
898071d4279SBram Moolenaar
899293ee4d4SBram Moolenaar checkupdate(curwin);
900293ee4d4SBram Moolenaar
901217e1b83SBram Moolenaar // Repeat "count" times.
902071d4279SBram Moolenaar for (n = 0; n < count; ++n)
903071d4279SBram Moolenaar {
904217e1b83SBram Moolenaar // Find nested folds. Stop when a fold is closed. The deepest fold
905217e1b83SBram Moolenaar // that moves the cursor is used.
906071d4279SBram Moolenaar lnum_off = 0;
907071d4279SBram Moolenaar gap = &curwin->w_folds;
908c136a352SBram Moolenaar if (gap->ga_len == 0)
909c136a352SBram Moolenaar break;
910071d4279SBram Moolenaar use_level = FALSE;
911071d4279SBram Moolenaar maybe_small = FALSE;
912071d4279SBram Moolenaar lnum_found = curwin->w_cursor.lnum;
913071d4279SBram Moolenaar level = 0;
914071d4279SBram Moolenaar last = FALSE;
915071d4279SBram Moolenaar for (;;)
916071d4279SBram Moolenaar {
917071d4279SBram Moolenaar if (!foldFind(gap, curwin->w_cursor.lnum - lnum_off, &fp))
918071d4279SBram Moolenaar {
9196a78f328SBram Moolenaar if (!updown || gap->ga_len == 0)
920071d4279SBram Moolenaar break;
921071d4279SBram Moolenaar
922217e1b83SBram Moolenaar // When moving up, consider a fold above the cursor; when
923217e1b83SBram Moolenaar // moving down consider a fold below the cursor.
924071d4279SBram Moolenaar if (dir == FORWARD)
925071d4279SBram Moolenaar {
926071d4279SBram Moolenaar if (fp - (fold_T *)gap->ga_data >= gap->ga_len)
927071d4279SBram Moolenaar break;
928071d4279SBram Moolenaar --fp;
929071d4279SBram Moolenaar }
930071d4279SBram Moolenaar else
931071d4279SBram Moolenaar {
932071d4279SBram Moolenaar if (fp == (fold_T *)gap->ga_data)
933071d4279SBram Moolenaar break;
934071d4279SBram Moolenaar }
935217e1b83SBram Moolenaar // don't look for contained folds, they will always move
936217e1b83SBram Moolenaar // the cursor too far.
937071d4279SBram Moolenaar last = TRUE;
938071d4279SBram Moolenaar }
939071d4279SBram Moolenaar
940071d4279SBram Moolenaar if (!last)
941071d4279SBram Moolenaar {
942217e1b83SBram Moolenaar // Check if this fold is closed.
943071d4279SBram Moolenaar if (check_closed(curwin, fp, &use_level, level,
944071d4279SBram Moolenaar &maybe_small, lnum_off))
945071d4279SBram Moolenaar last = TRUE;
946071d4279SBram Moolenaar
947217e1b83SBram Moolenaar // "[z" and "]z" stop at closed fold
948071d4279SBram Moolenaar if (last && !updown)
949071d4279SBram Moolenaar break;
950071d4279SBram Moolenaar }
951071d4279SBram Moolenaar
952071d4279SBram Moolenaar if (updown)
953071d4279SBram Moolenaar {
954071d4279SBram Moolenaar if (dir == FORWARD)
955071d4279SBram Moolenaar {
956217e1b83SBram Moolenaar // to start of next fold if there is one
957071d4279SBram Moolenaar if (fp + 1 - (fold_T *)gap->ga_data < gap->ga_len)
958071d4279SBram Moolenaar {
959071d4279SBram Moolenaar lnum = fp[1].fd_top + lnum_off;
960071d4279SBram Moolenaar if (lnum > curwin->w_cursor.lnum)
961071d4279SBram Moolenaar lnum_found = lnum;
962071d4279SBram Moolenaar }
963071d4279SBram Moolenaar }
964071d4279SBram Moolenaar else
965071d4279SBram Moolenaar {
966217e1b83SBram Moolenaar // to end of previous fold if there is one
967071d4279SBram Moolenaar if (fp > (fold_T *)gap->ga_data)
968071d4279SBram Moolenaar {
969071d4279SBram Moolenaar lnum = fp[-1].fd_top + lnum_off + fp[-1].fd_len - 1;
970071d4279SBram Moolenaar if (lnum < curwin->w_cursor.lnum)
971071d4279SBram Moolenaar lnum_found = lnum;
972071d4279SBram Moolenaar }
973071d4279SBram Moolenaar }
974071d4279SBram Moolenaar }
975071d4279SBram Moolenaar else
976071d4279SBram Moolenaar {
977217e1b83SBram Moolenaar // Open fold found, set cursor to its start/end and then check
978217e1b83SBram Moolenaar // nested folds.
979071d4279SBram Moolenaar if (dir == FORWARD)
980071d4279SBram Moolenaar {
981071d4279SBram Moolenaar lnum = fp->fd_top + lnum_off + fp->fd_len - 1;
982071d4279SBram Moolenaar if (lnum > curwin->w_cursor.lnum)
983071d4279SBram Moolenaar lnum_found = lnum;
984071d4279SBram Moolenaar }
985071d4279SBram Moolenaar else
986071d4279SBram Moolenaar {
987071d4279SBram Moolenaar lnum = fp->fd_top + lnum_off;
988071d4279SBram Moolenaar if (lnum < curwin->w_cursor.lnum)
989071d4279SBram Moolenaar lnum_found = lnum;
990071d4279SBram Moolenaar }
991071d4279SBram Moolenaar }
992071d4279SBram Moolenaar
993071d4279SBram Moolenaar if (last)
994071d4279SBram Moolenaar break;
995071d4279SBram Moolenaar
996217e1b83SBram Moolenaar // Check nested folds (if any).
997071d4279SBram Moolenaar gap = &fp->fd_nested;
998071d4279SBram Moolenaar lnum_off += fp->fd_top;
999071d4279SBram Moolenaar ++level;
1000071d4279SBram Moolenaar }
1001071d4279SBram Moolenaar if (lnum_found != curwin->w_cursor.lnum)
1002071d4279SBram Moolenaar {
1003071d4279SBram Moolenaar if (retval == FAIL)
1004071d4279SBram Moolenaar setpcmark();
1005071d4279SBram Moolenaar curwin->w_cursor.lnum = lnum_found;
1006071d4279SBram Moolenaar curwin->w_cursor.col = 0;
1007071d4279SBram Moolenaar retval = OK;
1008071d4279SBram Moolenaar }
1009071d4279SBram Moolenaar else
1010071d4279SBram Moolenaar break;
1011071d4279SBram Moolenaar }
1012071d4279SBram Moolenaar
1013071d4279SBram Moolenaar return retval;
1014071d4279SBram Moolenaar }
1015071d4279SBram Moolenaar
1016217e1b83SBram Moolenaar // foldInitWin() {{{2
1017071d4279SBram Moolenaar /*
1018071d4279SBram Moolenaar * Init the fold info in a new window.
1019071d4279SBram Moolenaar */
1020071d4279SBram Moolenaar void
foldInitWin(win_T * new_win)102166f948e9SBram Moolenaar foldInitWin(win_T *new_win)
1022071d4279SBram Moolenaar {
102370b2a56dSBram Moolenaar ga_init2(&new_win->w_folds, (int)sizeof(fold_T), 10);
1024071d4279SBram Moolenaar }
1025071d4279SBram Moolenaar
1026217e1b83SBram Moolenaar // find_wl_entry() {{{2
1027071d4279SBram Moolenaar /*
1028071d4279SBram Moolenaar * Find an entry in the win->w_lines[] array for buffer line "lnum".
1029071d4279SBram Moolenaar * Only valid entries are considered (for entries where wl_valid is FALSE the
1030071d4279SBram Moolenaar * line number can be wrong).
1031071d4279SBram Moolenaar * Returns index of entry or -1 if not found.
1032071d4279SBram Moolenaar */
1033071d4279SBram Moolenaar int
find_wl_entry(win_T * win,linenr_T lnum)103466f948e9SBram Moolenaar find_wl_entry(win_T *win, linenr_T lnum)
1035071d4279SBram Moolenaar {
1036071d4279SBram Moolenaar int i;
1037071d4279SBram Moolenaar
1038071d4279SBram Moolenaar for (i = 0; i < win->w_lines_valid; ++i)
1039071d4279SBram Moolenaar if (win->w_lines[i].wl_valid)
1040071d4279SBram Moolenaar {
1041071d4279SBram Moolenaar if (lnum < win->w_lines[i].wl_lnum)
1042071d4279SBram Moolenaar return -1;
1043071d4279SBram Moolenaar if (lnum <= win->w_lines[i].wl_lastlnum)
1044071d4279SBram Moolenaar return i;
1045071d4279SBram Moolenaar }
1046071d4279SBram Moolenaar return -1;
1047071d4279SBram Moolenaar }
1048071d4279SBram Moolenaar
1049217e1b83SBram Moolenaar // foldAdjustVisual() {{{2
1050071d4279SBram Moolenaar /*
1051071d4279SBram Moolenaar * Adjust the Visual area to include any fold at the start or end completely.
1052071d4279SBram Moolenaar */
1053071d4279SBram Moolenaar void
foldAdjustVisual(void)105466f948e9SBram Moolenaar foldAdjustVisual(void)
1055071d4279SBram Moolenaar {
1056071d4279SBram Moolenaar pos_T *start, *end;
1057071d4279SBram Moolenaar char_u *ptr;
1058071d4279SBram Moolenaar
1059071d4279SBram Moolenaar if (!VIsual_active || !hasAnyFolding(curwin))
1060071d4279SBram Moolenaar return;
1061071d4279SBram Moolenaar
1062b5aedf3eSBram Moolenaar if (LTOREQ_POS(VIsual, curwin->w_cursor))
1063071d4279SBram Moolenaar {
1064071d4279SBram Moolenaar start = &VIsual;
1065071d4279SBram Moolenaar end = &curwin->w_cursor;
1066071d4279SBram Moolenaar }
1067071d4279SBram Moolenaar else
1068071d4279SBram Moolenaar {
1069071d4279SBram Moolenaar start = &curwin->w_cursor;
1070071d4279SBram Moolenaar end = &VIsual;
1071071d4279SBram Moolenaar }
1072071d4279SBram Moolenaar if (hasFolding(start->lnum, &start->lnum, NULL))
1073071d4279SBram Moolenaar start->col = 0;
1074071d4279SBram Moolenaar if (hasFolding(end->lnum, NULL, &end->lnum))
1075071d4279SBram Moolenaar {
1076071d4279SBram Moolenaar ptr = ml_get(end->lnum);
1077071d4279SBram Moolenaar end->col = (colnr_T)STRLEN(ptr);
1078071d4279SBram Moolenaar if (end->col > 0 && *p_sel == 'o')
1079071d4279SBram Moolenaar --end->col;
1080217e1b83SBram Moolenaar // prevent cursor from moving on the trail byte
1081071d4279SBram Moolenaar if (has_mbyte)
1082071d4279SBram Moolenaar mb_adjust_cursor();
1083071d4279SBram Moolenaar }
1084071d4279SBram Moolenaar }
1085071d4279SBram Moolenaar
1086217e1b83SBram Moolenaar // cursor_foldstart() {{{2
1087071d4279SBram Moolenaar /*
1088071d4279SBram Moolenaar * Move the cursor to the first line of a closed fold.
1089071d4279SBram Moolenaar */
1090071d4279SBram Moolenaar void
foldAdjustCursor(void)109166f948e9SBram Moolenaar foldAdjustCursor(void)
1092071d4279SBram Moolenaar {
1093071d4279SBram Moolenaar (void)hasFolding(curwin->w_cursor.lnum, &curwin->w_cursor.lnum, NULL);
1094071d4279SBram Moolenaar }
1095071d4279SBram Moolenaar
1096217e1b83SBram Moolenaar // Internal functions for "fold_T" {{{1
1097217e1b83SBram Moolenaar // cloneFoldGrowArray() {{{2
1098071d4279SBram Moolenaar /*
1099071d4279SBram Moolenaar * Will "clone" (i.e deep copy) a garray_T of folds.
1100071d4279SBram Moolenaar *
1101071d4279SBram Moolenaar * Return FAIL if the operation cannot be completed, otherwise OK.
1102071d4279SBram Moolenaar */
1103071d4279SBram Moolenaar void
cloneFoldGrowArray(garray_T * from,garray_T * to)110466f948e9SBram Moolenaar cloneFoldGrowArray(garray_T *from, garray_T *to)
1105071d4279SBram Moolenaar {
1106071d4279SBram Moolenaar int i;
1107071d4279SBram Moolenaar fold_T *from_p;
1108071d4279SBram Moolenaar fold_T *to_p;
1109071d4279SBram Moolenaar
1110071d4279SBram Moolenaar ga_init2(to, from->ga_itemsize, from->ga_growsize);
1111071d4279SBram Moolenaar if (from->ga_len == 0 || ga_grow(to, from->ga_len) == FAIL)
1112071d4279SBram Moolenaar return;
1113071d4279SBram Moolenaar
1114071d4279SBram Moolenaar from_p = (fold_T *)from->ga_data;
1115071d4279SBram Moolenaar to_p = (fold_T *)to->ga_data;
1116071d4279SBram Moolenaar
1117071d4279SBram Moolenaar for (i = 0; i < from->ga_len; i++)
1118071d4279SBram Moolenaar {
1119071d4279SBram Moolenaar to_p->fd_top = from_p->fd_top;
1120071d4279SBram Moolenaar to_p->fd_len = from_p->fd_len;
1121071d4279SBram Moolenaar to_p->fd_flags = from_p->fd_flags;
1122071d4279SBram Moolenaar to_p->fd_small = from_p->fd_small;
1123071d4279SBram Moolenaar cloneFoldGrowArray(&from_p->fd_nested, &to_p->fd_nested);
1124071d4279SBram Moolenaar ++to->ga_len;
1125071d4279SBram Moolenaar ++from_p;
1126071d4279SBram Moolenaar ++to_p;
1127071d4279SBram Moolenaar }
1128071d4279SBram Moolenaar }
1129071d4279SBram Moolenaar
1130217e1b83SBram Moolenaar // foldFind() {{{2
1131071d4279SBram Moolenaar /*
1132071d4279SBram Moolenaar * Search for line "lnum" in folds of growarray "gap".
1133071d4279SBram Moolenaar * Set *fpp to the fold struct for the fold that contains "lnum" or
1134071d4279SBram Moolenaar * the first fold below it (careful: it can be beyond the end of the array!).
1135071d4279SBram Moolenaar * Returns FALSE when there is no fold that contains "lnum".
1136071d4279SBram Moolenaar */
1137071d4279SBram Moolenaar static int
foldFind(garray_T * gap,linenr_T lnum,fold_T ** fpp)113866f948e9SBram Moolenaar foldFind(garray_T *gap, linenr_T lnum, fold_T **fpp)
1139071d4279SBram Moolenaar {
1140071d4279SBram Moolenaar linenr_T low, high;
1141071d4279SBram Moolenaar fold_T *fp;
1142071d4279SBram Moolenaar int i;
1143071d4279SBram Moolenaar
114464f37d30SBram Moolenaar if (gap->ga_len == 0)
114564f37d30SBram Moolenaar {
114664f37d30SBram Moolenaar *fpp = NULL;
114764f37d30SBram Moolenaar return FALSE;
114864f37d30SBram Moolenaar }
114964f37d30SBram Moolenaar
1150071d4279SBram Moolenaar /*
1151071d4279SBram Moolenaar * Perform a binary search.
1152071d4279SBram Moolenaar * "low" is lowest index of possible match.
1153071d4279SBram Moolenaar * "high" is highest index of possible match.
1154071d4279SBram Moolenaar */
1155071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data;
1156071d4279SBram Moolenaar low = 0;
1157071d4279SBram Moolenaar high = gap->ga_len - 1;
1158071d4279SBram Moolenaar while (low <= high)
1159071d4279SBram Moolenaar {
1160071d4279SBram Moolenaar i = (low + high) / 2;
1161071d4279SBram Moolenaar if (fp[i].fd_top > lnum)
1162217e1b83SBram Moolenaar // fold below lnum, adjust high
1163071d4279SBram Moolenaar high = i - 1;
1164071d4279SBram Moolenaar else if (fp[i].fd_top + fp[i].fd_len <= lnum)
1165217e1b83SBram Moolenaar // fold above lnum, adjust low
1166071d4279SBram Moolenaar low = i + 1;
1167071d4279SBram Moolenaar else
1168071d4279SBram Moolenaar {
1169217e1b83SBram Moolenaar // lnum is inside this fold
1170071d4279SBram Moolenaar *fpp = fp + i;
1171071d4279SBram Moolenaar return TRUE;
1172071d4279SBram Moolenaar }
1173071d4279SBram Moolenaar }
1174071d4279SBram Moolenaar *fpp = fp + low;
1175071d4279SBram Moolenaar return FALSE;
1176071d4279SBram Moolenaar }
1177071d4279SBram Moolenaar
1178217e1b83SBram Moolenaar // foldLevelWin() {{{2
1179071d4279SBram Moolenaar /*
1180071d4279SBram Moolenaar * Return fold level at line number "lnum" in window "wp".
1181071d4279SBram Moolenaar */
1182071d4279SBram Moolenaar static int
foldLevelWin(win_T * wp,linenr_T lnum)118366f948e9SBram Moolenaar foldLevelWin(win_T *wp, linenr_T lnum)
1184071d4279SBram Moolenaar {
1185071d4279SBram Moolenaar fold_T *fp;
1186071d4279SBram Moolenaar linenr_T lnum_rel = lnum;
1187071d4279SBram Moolenaar int level = 0;
1188071d4279SBram Moolenaar garray_T *gap;
1189071d4279SBram Moolenaar
1190217e1b83SBram Moolenaar // Recursively search for a fold that contains "lnum".
1191071d4279SBram Moolenaar gap = &wp->w_folds;
1192071d4279SBram Moolenaar for (;;)
1193071d4279SBram Moolenaar {
1194071d4279SBram Moolenaar if (!foldFind(gap, lnum_rel, &fp))
1195071d4279SBram Moolenaar break;
1196217e1b83SBram Moolenaar // Check nested folds. Line number is relative to containing fold.
1197071d4279SBram Moolenaar gap = &fp->fd_nested;
1198071d4279SBram Moolenaar lnum_rel -= fp->fd_top;
1199071d4279SBram Moolenaar ++level;
1200071d4279SBram Moolenaar }
1201071d4279SBram Moolenaar
1202071d4279SBram Moolenaar return level;
1203071d4279SBram Moolenaar }
1204071d4279SBram Moolenaar
1205217e1b83SBram Moolenaar // checkupdate() {{{2
1206071d4279SBram Moolenaar /*
1207071d4279SBram Moolenaar * Check if the folds in window "wp" are invalid and update them if needed.
1208071d4279SBram Moolenaar */
1209071d4279SBram Moolenaar static void
checkupdate(win_T * wp)121066f948e9SBram Moolenaar checkupdate(win_T *wp)
1211071d4279SBram Moolenaar {
1212071d4279SBram Moolenaar if (wp->w_foldinvalid)
1213071d4279SBram Moolenaar {
1214217e1b83SBram Moolenaar foldUpdate(wp, (linenr_T)1, (linenr_T)MAXLNUM); // will update all
1215071d4279SBram Moolenaar wp->w_foldinvalid = FALSE;
1216071d4279SBram Moolenaar }
1217071d4279SBram Moolenaar }
1218071d4279SBram Moolenaar
1219217e1b83SBram Moolenaar // setFoldRepeat() {{{2
1220071d4279SBram Moolenaar /*
1221071d4279SBram Moolenaar * Open or close fold for current window at line "lnum".
1222071d4279SBram Moolenaar * Repeat "count" times.
1223071d4279SBram Moolenaar */
1224071d4279SBram Moolenaar static void
setFoldRepeat(linenr_T lnum,long count,int do_open)122566f948e9SBram Moolenaar setFoldRepeat(linenr_T lnum, long count, int do_open)
1226071d4279SBram Moolenaar {
1227071d4279SBram Moolenaar int done;
1228071d4279SBram Moolenaar long n;
1229071d4279SBram Moolenaar
1230071d4279SBram Moolenaar for (n = 0; n < count; ++n)
1231071d4279SBram Moolenaar {
1232071d4279SBram Moolenaar done = DONE_NOTHING;
1233fe86f2d7SBram Moolenaar (void)setManualFold(lnum, do_open, FALSE, &done);
1234071d4279SBram Moolenaar if (!(done & DONE_ACTION))
1235071d4279SBram Moolenaar {
1236217e1b83SBram Moolenaar // Only give an error message when no fold could be opened.
1237071d4279SBram Moolenaar if (n == 0 && !(done & DONE_FOLD))
1238f9e3e09fSBram Moolenaar emsg(_(e_nofold));
1239071d4279SBram Moolenaar break;
1240071d4279SBram Moolenaar }
1241071d4279SBram Moolenaar }
1242071d4279SBram Moolenaar }
1243071d4279SBram Moolenaar
1244217e1b83SBram Moolenaar // setManualFold() {{{2
1245071d4279SBram Moolenaar /*
1246071d4279SBram Moolenaar * Open or close the fold in the current window which contains "lnum".
1247071d4279SBram Moolenaar * Also does this for other windows in diff mode when needed.
1248071d4279SBram Moolenaar */
1249071d4279SBram Moolenaar static linenr_T
setManualFold(linenr_T lnum,int opening,int recurse,int * donep)125066f948e9SBram Moolenaar setManualFold(
125166f948e9SBram Moolenaar linenr_T lnum,
1252217e1b83SBram Moolenaar int opening, // TRUE when opening, FALSE when closing
1253217e1b83SBram Moolenaar int recurse, // TRUE when closing/opening recursive
125466f948e9SBram Moolenaar int *donep)
1255071d4279SBram Moolenaar {
1256071d4279SBram Moolenaar #ifdef FEAT_DIFF
1257071d4279SBram Moolenaar if (foldmethodIsDiff(curwin) && curwin->w_p_scb)
1258071d4279SBram Moolenaar {
1259071d4279SBram Moolenaar win_T *wp;
1260071d4279SBram Moolenaar linenr_T dlnum;
1261071d4279SBram Moolenaar
1262071d4279SBram Moolenaar /*
1263071d4279SBram Moolenaar * Do the same operation in other windows in diff mode. Calculate the
1264071d4279SBram Moolenaar * line number from the diffs.
1265071d4279SBram Moolenaar */
1266071d4279SBram Moolenaar FOR_ALL_WINDOWS(wp)
1267071d4279SBram Moolenaar {
1268071d4279SBram Moolenaar if (wp != curwin && foldmethodIsDiff(wp) && wp->w_p_scb)
1269071d4279SBram Moolenaar {
1270071d4279SBram Moolenaar dlnum = diff_lnum_win(curwin->w_cursor.lnum, wp);
1271071d4279SBram Moolenaar if (dlnum != 0)
1272071d4279SBram Moolenaar (void)setManualFoldWin(wp, dlnum, opening, recurse, NULL);
1273071d4279SBram Moolenaar }
1274071d4279SBram Moolenaar }
1275071d4279SBram Moolenaar }
1276071d4279SBram Moolenaar #endif
1277071d4279SBram Moolenaar
1278071d4279SBram Moolenaar return setManualFoldWin(curwin, lnum, opening, recurse, donep);
1279071d4279SBram Moolenaar }
1280071d4279SBram Moolenaar
1281217e1b83SBram Moolenaar // setManualFoldWin() {{{2
1282071d4279SBram Moolenaar /*
1283071d4279SBram Moolenaar * Open or close the fold in window "wp" which contains "lnum".
1284071d4279SBram Moolenaar * "donep", when not NULL, points to flag that is set to DONE_FOLD when some
1285071d4279SBram Moolenaar * fold was found and to DONE_ACTION when some fold was opened or closed.
1286071d4279SBram Moolenaar * When "donep" is NULL give an error message when no fold was found for
1287071d4279SBram Moolenaar * "lnum", but only if "wp" is "curwin".
1288071d4279SBram Moolenaar * Return the line number of the next line that could be closed.
1289071d4279SBram Moolenaar * It's only valid when "opening" is TRUE!
1290071d4279SBram Moolenaar */
1291071d4279SBram Moolenaar static linenr_T
setManualFoldWin(win_T * wp,linenr_T lnum,int opening,int recurse,int * donep)129266f948e9SBram Moolenaar setManualFoldWin(
129366f948e9SBram Moolenaar win_T *wp,
129466f948e9SBram Moolenaar linenr_T lnum,
1295217e1b83SBram Moolenaar int opening, // TRUE when opening, FALSE when closing
1296217e1b83SBram Moolenaar int recurse, // TRUE when closing/opening recursive
129766f948e9SBram Moolenaar int *donep)
1298071d4279SBram Moolenaar {
1299071d4279SBram Moolenaar fold_T *fp;
1300071d4279SBram Moolenaar fold_T *fp2;
1301071d4279SBram Moolenaar fold_T *found = NULL;
1302071d4279SBram Moolenaar int j;
1303071d4279SBram Moolenaar int level = 0;
1304071d4279SBram Moolenaar int use_level = FALSE;
1305071d4279SBram Moolenaar int found_fold = FALSE;
1306071d4279SBram Moolenaar garray_T *gap;
1307071d4279SBram Moolenaar linenr_T next = MAXLNUM;
1308071d4279SBram Moolenaar linenr_T off = 0;
1309071d4279SBram Moolenaar int done = 0;
1310071d4279SBram Moolenaar
1311071d4279SBram Moolenaar checkupdate(wp);
1312071d4279SBram Moolenaar
1313071d4279SBram Moolenaar /*
1314071d4279SBram Moolenaar * Find the fold, open or close it.
1315071d4279SBram Moolenaar */
1316071d4279SBram Moolenaar gap = &wp->w_folds;
1317071d4279SBram Moolenaar for (;;)
1318071d4279SBram Moolenaar {
1319071d4279SBram Moolenaar if (!foldFind(gap, lnum, &fp))
1320071d4279SBram Moolenaar {
1321217e1b83SBram Moolenaar // If there is a following fold, continue there next time.
13229c2b0663SBram Moolenaar if (fp != NULL && fp < (fold_T *)gap->ga_data + gap->ga_len)
1323071d4279SBram Moolenaar next = fp->fd_top + off;
1324071d4279SBram Moolenaar break;
1325071d4279SBram Moolenaar }
1326071d4279SBram Moolenaar
1327217e1b83SBram Moolenaar // lnum is inside this fold
1328071d4279SBram Moolenaar found_fold = TRUE;
1329071d4279SBram Moolenaar
1330217e1b83SBram Moolenaar // If there is a following fold, continue there next time.
1331071d4279SBram Moolenaar if (fp + 1 < (fold_T *)gap->ga_data + gap->ga_len)
1332071d4279SBram Moolenaar next = fp[1].fd_top + off;
1333071d4279SBram Moolenaar
1334217e1b83SBram Moolenaar // Change from level-dependent folding to manual.
1335071d4279SBram Moolenaar if (use_level || fp->fd_flags == FD_LEVEL)
1336071d4279SBram Moolenaar {
1337071d4279SBram Moolenaar use_level = TRUE;
1338071d4279SBram Moolenaar if (level >= wp->w_p_fdl)
1339071d4279SBram Moolenaar fp->fd_flags = FD_CLOSED;
1340071d4279SBram Moolenaar else
1341071d4279SBram Moolenaar fp->fd_flags = FD_OPEN;
1342071d4279SBram Moolenaar fp2 = (fold_T *)fp->fd_nested.ga_data;
1343071d4279SBram Moolenaar for (j = 0; j < fp->fd_nested.ga_len; ++j)
1344071d4279SBram Moolenaar fp2[j].fd_flags = FD_LEVEL;
1345071d4279SBram Moolenaar }
1346071d4279SBram Moolenaar
1347217e1b83SBram Moolenaar // Simple case: Close recursively means closing the fold.
1348071d4279SBram Moolenaar if (!opening && recurse)
1349071d4279SBram Moolenaar {
1350071d4279SBram Moolenaar if (fp->fd_flags != FD_CLOSED)
1351071d4279SBram Moolenaar {
1352071d4279SBram Moolenaar done |= DONE_ACTION;
1353071d4279SBram Moolenaar fp->fd_flags = FD_CLOSED;
1354071d4279SBram Moolenaar }
1355071d4279SBram Moolenaar }
1356071d4279SBram Moolenaar else if (fp->fd_flags == FD_CLOSED)
1357071d4279SBram Moolenaar {
1358217e1b83SBram Moolenaar // When opening, open topmost closed fold.
1359071d4279SBram Moolenaar if (opening)
1360071d4279SBram Moolenaar {
1361071d4279SBram Moolenaar fp->fd_flags = FD_OPEN;
1362071d4279SBram Moolenaar done |= DONE_ACTION;
1363071d4279SBram Moolenaar if (recurse)
1364071d4279SBram Moolenaar foldOpenNested(fp);
1365071d4279SBram Moolenaar }
1366071d4279SBram Moolenaar break;
1367071d4279SBram Moolenaar }
1368071d4279SBram Moolenaar
1369217e1b83SBram Moolenaar // fold is open, check nested folds
1370071d4279SBram Moolenaar found = fp;
1371071d4279SBram Moolenaar gap = &fp->fd_nested;
1372071d4279SBram Moolenaar lnum -= fp->fd_top;
1373071d4279SBram Moolenaar off += fp->fd_top;
1374071d4279SBram Moolenaar ++level;
1375071d4279SBram Moolenaar }
1376071d4279SBram Moolenaar if (found_fold)
1377071d4279SBram Moolenaar {
1378217e1b83SBram Moolenaar // When closing and not recurse, close deepest open fold.
1379071d4279SBram Moolenaar if (!opening && found != NULL)
1380071d4279SBram Moolenaar {
1381071d4279SBram Moolenaar found->fd_flags = FD_CLOSED;
1382071d4279SBram Moolenaar done |= DONE_ACTION;
1383071d4279SBram Moolenaar }
1384071d4279SBram Moolenaar wp->w_fold_manual = TRUE;
1385071d4279SBram Moolenaar if (done & DONE_ACTION)
1386071d4279SBram Moolenaar changed_window_setting_win(wp);
1387071d4279SBram Moolenaar done |= DONE_FOLD;
1388071d4279SBram Moolenaar }
1389071d4279SBram Moolenaar else if (donep == NULL && wp == curwin)
1390f9e3e09fSBram Moolenaar emsg(_(e_nofold));
1391071d4279SBram Moolenaar
1392071d4279SBram Moolenaar if (donep != NULL)
1393071d4279SBram Moolenaar *donep |= done;
1394071d4279SBram Moolenaar
1395071d4279SBram Moolenaar return next;
1396071d4279SBram Moolenaar }
1397071d4279SBram Moolenaar
1398217e1b83SBram Moolenaar // foldOpenNested() {{{2
1399071d4279SBram Moolenaar /*
1400071d4279SBram Moolenaar * Open all nested folds in fold "fpr" recursively.
1401071d4279SBram Moolenaar */
1402071d4279SBram Moolenaar static void
foldOpenNested(fold_T * fpr)140366f948e9SBram Moolenaar foldOpenNested(fold_T *fpr)
1404071d4279SBram Moolenaar {
1405071d4279SBram Moolenaar int i;
1406071d4279SBram Moolenaar fold_T *fp;
1407071d4279SBram Moolenaar
1408071d4279SBram Moolenaar fp = (fold_T *)fpr->fd_nested.ga_data;
1409071d4279SBram Moolenaar for (i = 0; i < fpr->fd_nested.ga_len; ++i)
1410071d4279SBram Moolenaar {
1411071d4279SBram Moolenaar foldOpenNested(&fp[i]);
1412071d4279SBram Moolenaar fp[i].fd_flags = FD_OPEN;
1413071d4279SBram Moolenaar }
1414071d4279SBram Moolenaar }
1415071d4279SBram Moolenaar
1416217e1b83SBram Moolenaar // deleteFoldEntry() {{{2
1417071d4279SBram Moolenaar /*
1418071d4279SBram Moolenaar * Delete fold "idx" from growarray "gap".
1419071d4279SBram Moolenaar * When "recursive" is TRUE also delete all the folds contained in it.
1420071d4279SBram Moolenaar * When "recursive" is FALSE contained folds are moved one level up.
1421071d4279SBram Moolenaar */
1422071d4279SBram Moolenaar static void
deleteFoldEntry(garray_T * gap,int idx,int recursive)142366f948e9SBram Moolenaar deleteFoldEntry(garray_T *gap, int idx, int recursive)
1424071d4279SBram Moolenaar {
1425071d4279SBram Moolenaar fold_T *fp;
1426071d4279SBram Moolenaar int i;
1427071d4279SBram Moolenaar long moved;
1428071d4279SBram Moolenaar fold_T *nfp;
1429071d4279SBram Moolenaar
1430071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + idx;
1431071d4279SBram Moolenaar if (recursive || fp->fd_nested.ga_len == 0)
1432071d4279SBram Moolenaar {
1433217e1b83SBram Moolenaar // recursively delete the contained folds
1434071d4279SBram Moolenaar deleteFoldRecurse(&fp->fd_nested);
1435071d4279SBram Moolenaar --gap->ga_len;
1436071d4279SBram Moolenaar if (idx < gap->ga_len)
1437071d4279SBram Moolenaar mch_memmove(fp, fp + 1, sizeof(fold_T) * (gap->ga_len - idx));
1438071d4279SBram Moolenaar }
1439071d4279SBram Moolenaar else
1440071d4279SBram Moolenaar {
1441217e1b83SBram Moolenaar // Move nested folds one level up, to overwrite the fold that is
1442217e1b83SBram Moolenaar // deleted.
1443071d4279SBram Moolenaar moved = fp->fd_nested.ga_len;
1444071d4279SBram Moolenaar if (ga_grow(gap, (int)(moved - 1)) == OK)
1445071d4279SBram Moolenaar {
1446217e1b83SBram Moolenaar // Get "fp" again, the array may have been reallocated.
1447c9927c13SBram Moolenaar fp = (fold_T *)gap->ga_data + idx;
1448c9927c13SBram Moolenaar
1449217e1b83SBram Moolenaar // adjust fd_top and fd_flags for the moved folds
1450071d4279SBram Moolenaar nfp = (fold_T *)fp->fd_nested.ga_data;
1451071d4279SBram Moolenaar for (i = 0; i < moved; ++i)
1452071d4279SBram Moolenaar {
1453071d4279SBram Moolenaar nfp[i].fd_top += fp->fd_top;
1454071d4279SBram Moolenaar if (fp->fd_flags == FD_LEVEL)
1455071d4279SBram Moolenaar nfp[i].fd_flags = FD_LEVEL;
1456071d4279SBram Moolenaar if (fp->fd_small == MAYBE)
1457071d4279SBram Moolenaar nfp[i].fd_small = MAYBE;
1458071d4279SBram Moolenaar }
1459071d4279SBram Moolenaar
1460217e1b83SBram Moolenaar // move the existing folds down to make room
1461c9927c13SBram Moolenaar if (idx + 1 < gap->ga_len)
1462071d4279SBram Moolenaar mch_memmove(fp + moved, fp + 1,
1463c9927c13SBram Moolenaar sizeof(fold_T) * (gap->ga_len - (idx + 1)));
1464217e1b83SBram Moolenaar // move the contained folds one level up
1465071d4279SBram Moolenaar mch_memmove(fp, nfp, (size_t)(sizeof(fold_T) * moved));
1466071d4279SBram Moolenaar vim_free(nfp);
1467071d4279SBram Moolenaar gap->ga_len += moved - 1;
1468071d4279SBram Moolenaar }
1469071d4279SBram Moolenaar }
1470071d4279SBram Moolenaar }
1471071d4279SBram Moolenaar
1472217e1b83SBram Moolenaar // deleteFoldRecurse() {{{2
1473071d4279SBram Moolenaar /*
1474071d4279SBram Moolenaar * Delete nested folds in a fold.
1475071d4279SBram Moolenaar */
1476071d4279SBram Moolenaar void
deleteFoldRecurse(garray_T * gap)147766f948e9SBram Moolenaar deleteFoldRecurse(garray_T *gap)
1478071d4279SBram Moolenaar {
1479071d4279SBram Moolenaar int i;
1480071d4279SBram Moolenaar
1481071d4279SBram Moolenaar for (i = 0; i < gap->ga_len; ++i)
1482071d4279SBram Moolenaar deleteFoldRecurse(&(((fold_T *)(gap->ga_data))[i].fd_nested));
1483071d4279SBram Moolenaar ga_clear(gap);
1484071d4279SBram Moolenaar }
1485071d4279SBram Moolenaar
1486217e1b83SBram Moolenaar // foldMarkAdjust() {{{2
1487071d4279SBram Moolenaar /*
1488071d4279SBram Moolenaar * Update line numbers of folds for inserted/deleted lines.
1489071d4279SBram Moolenaar */
1490071d4279SBram Moolenaar void
foldMarkAdjust(win_T * wp,linenr_T line1,linenr_T line2,long amount,long amount_after)149166f948e9SBram Moolenaar foldMarkAdjust(
149266f948e9SBram Moolenaar win_T *wp,
149366f948e9SBram Moolenaar linenr_T line1,
149466f948e9SBram Moolenaar linenr_T line2,
149566f948e9SBram Moolenaar long amount,
149666f948e9SBram Moolenaar long amount_after)
1497071d4279SBram Moolenaar {
1498217e1b83SBram Moolenaar // If deleting marks from line1 to line2, but not deleting all those
1499217e1b83SBram Moolenaar // lines, set line2 so that only deleted lines have their folds removed.
1500071d4279SBram Moolenaar if (amount == MAXLNUM && line2 >= line1 && line2 - line1 >= -amount_after)
1501071d4279SBram Moolenaar line2 = line1 - amount_after - 1;
1502217e1b83SBram Moolenaar // If appending a line in Insert mode, it should be included in the fold
1503217e1b83SBram Moolenaar // just above the line.
1504071d4279SBram Moolenaar if ((State & INSERT) && amount == (linenr_T)1 && line2 == MAXLNUM)
1505071d4279SBram Moolenaar --line1;
1506071d4279SBram Moolenaar foldMarkAdjustRecurse(&wp->w_folds, line1, line2, amount, amount_after);
1507071d4279SBram Moolenaar }
1508071d4279SBram Moolenaar
1509217e1b83SBram Moolenaar // foldMarkAdjustRecurse() {{{2
1510071d4279SBram Moolenaar static void
foldMarkAdjustRecurse(garray_T * gap,linenr_T line1,linenr_T line2,long amount,long amount_after)151166f948e9SBram Moolenaar foldMarkAdjustRecurse(
151266f948e9SBram Moolenaar garray_T *gap,
151366f948e9SBram Moolenaar linenr_T line1,
151466f948e9SBram Moolenaar linenr_T line2,
151566f948e9SBram Moolenaar long amount,
151666f948e9SBram Moolenaar long amount_after)
1517071d4279SBram Moolenaar {
1518071d4279SBram Moolenaar fold_T *fp;
1519071d4279SBram Moolenaar int i;
1520071d4279SBram Moolenaar linenr_T last;
1521071d4279SBram Moolenaar linenr_T top;
1522071d4279SBram Moolenaar
152307e87e9eSBram Moolenaar if (gap->ga_len == 0)
152407e87e9eSBram Moolenaar return;
152507e87e9eSBram Moolenaar
1526217e1b83SBram Moolenaar // In Insert mode an inserted line at the top of a fold is considered part
1527217e1b83SBram Moolenaar // of the fold, otherwise it isn't.
1528071d4279SBram Moolenaar if ((State & INSERT) && amount == (linenr_T)1 && line2 == MAXLNUM)
1529071d4279SBram Moolenaar top = line1 + 1;
1530071d4279SBram Moolenaar else
1531071d4279SBram Moolenaar top = line1;
1532071d4279SBram Moolenaar
1533217e1b83SBram Moolenaar // Find the fold containing or just below "line1".
1534071d4279SBram Moolenaar (void)foldFind(gap, line1, &fp);
1535071d4279SBram Moolenaar
1536071d4279SBram Moolenaar /*
1537071d4279SBram Moolenaar * Adjust all folds below "line1" that are affected.
1538071d4279SBram Moolenaar */
1539071d4279SBram Moolenaar for (i = (int)(fp - (fold_T *)gap->ga_data); i < gap->ga_len; ++i, ++fp)
1540071d4279SBram Moolenaar {
1541071d4279SBram Moolenaar /*
1542071d4279SBram Moolenaar * Check for these situations:
1543071d4279SBram Moolenaar * 1 2 3
1544071d4279SBram Moolenaar * 1 2 3
1545071d4279SBram Moolenaar * line1 2 3 4 5
1546071d4279SBram Moolenaar * 2 3 4 5
1547071d4279SBram Moolenaar * 2 3 4 5
1548071d4279SBram Moolenaar * line2 2 3 4 5
1549071d4279SBram Moolenaar * 3 5 6
1550071d4279SBram Moolenaar * 3 5 6
1551071d4279SBram Moolenaar */
1552071d4279SBram Moolenaar
1553217e1b83SBram Moolenaar last = fp->fd_top + fp->fd_len - 1; // last line of fold
1554071d4279SBram Moolenaar
1555217e1b83SBram Moolenaar // 1. fold completely above line1: nothing to do
1556071d4279SBram Moolenaar if (last < line1)
1557071d4279SBram Moolenaar continue;
1558071d4279SBram Moolenaar
1559217e1b83SBram Moolenaar // 6. fold below line2: only adjust for amount_after
1560071d4279SBram Moolenaar if (fp->fd_top > line2)
1561071d4279SBram Moolenaar {
1562071d4279SBram Moolenaar if (amount_after == 0)
1563071d4279SBram Moolenaar break;
1564071d4279SBram Moolenaar fp->fd_top += amount_after;
1565071d4279SBram Moolenaar }
1566071d4279SBram Moolenaar else
1567071d4279SBram Moolenaar {
1568071d4279SBram Moolenaar if (fp->fd_top >= top && last <= line2)
1569071d4279SBram Moolenaar {
1570217e1b83SBram Moolenaar // 4. fold completely contained in range
1571071d4279SBram Moolenaar if (amount == MAXLNUM)
1572071d4279SBram Moolenaar {
1573217e1b83SBram Moolenaar // Deleting lines: delete the fold completely
1574071d4279SBram Moolenaar deleteFoldEntry(gap, i, TRUE);
1575217e1b83SBram Moolenaar --i; // adjust index for deletion
1576071d4279SBram Moolenaar --fp;
1577071d4279SBram Moolenaar }
1578071d4279SBram Moolenaar else
1579071d4279SBram Moolenaar fp->fd_top += amount;
1580071d4279SBram Moolenaar }
1581071d4279SBram Moolenaar else
1582071d4279SBram Moolenaar {
1583071d4279SBram Moolenaar if (fp->fd_top < top)
1584071d4279SBram Moolenaar {
1585217e1b83SBram Moolenaar // 2 or 3: need to correct nested folds too
1586194b94c5SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested, line1 - fp->fd_top,
1587194b94c5SBram Moolenaar line2 - fp->fd_top, amount, amount_after);
1588071d4279SBram Moolenaar if (last <= line2)
1589071d4279SBram Moolenaar {
1590217e1b83SBram Moolenaar // 2. fold contains line1, line2 is below fold
1591071d4279SBram Moolenaar if (amount == MAXLNUM)
1592071d4279SBram Moolenaar fp->fd_len = line1 - fp->fd_top;
1593071d4279SBram Moolenaar else
1594071d4279SBram Moolenaar fp->fd_len += amount;
1595071d4279SBram Moolenaar }
1596071d4279SBram Moolenaar else
1597071d4279SBram Moolenaar {
1598217e1b83SBram Moolenaar // 3. fold contains line1 and line2
1599071d4279SBram Moolenaar fp->fd_len += amount_after;
1600071d4279SBram Moolenaar }
1601071d4279SBram Moolenaar }
1602071d4279SBram Moolenaar else
1603071d4279SBram Moolenaar {
1604217e1b83SBram Moolenaar // 5. fold is below line1 and contains line2; need to
1605217e1b83SBram Moolenaar // correct nested folds too
1606071d4279SBram Moolenaar if (amount == MAXLNUM)
1607071d4279SBram Moolenaar {
16081159b168SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested,
16091159b168SBram Moolenaar line1 - fp->fd_top,
16101159b168SBram Moolenaar line2 - fp->fd_top,
16111159b168SBram Moolenaar amount,
16121159b168SBram Moolenaar amount_after + (fp->fd_top - top));
1613071d4279SBram Moolenaar fp->fd_len -= line2 - fp->fd_top + 1;
1614071d4279SBram Moolenaar fp->fd_top = line1;
1615071d4279SBram Moolenaar }
1616071d4279SBram Moolenaar else
1617071d4279SBram Moolenaar {
16181159b168SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested,
16191159b168SBram Moolenaar line1 - fp->fd_top,
16201159b168SBram Moolenaar line2 - fp->fd_top,
16211159b168SBram Moolenaar amount,
16221159b168SBram Moolenaar amount_after - amount);
1623071d4279SBram Moolenaar fp->fd_len += amount_after - amount;
1624071d4279SBram Moolenaar fp->fd_top += amount;
1625071d4279SBram Moolenaar }
1626071d4279SBram Moolenaar }
1627071d4279SBram Moolenaar }
1628071d4279SBram Moolenaar }
1629071d4279SBram Moolenaar }
1630071d4279SBram Moolenaar }
1631071d4279SBram Moolenaar
1632217e1b83SBram Moolenaar // getDeepestNesting() {{{2
1633071d4279SBram Moolenaar /*
1634071d4279SBram Moolenaar * Get the lowest 'foldlevel' value that makes the deepest nested fold in the
1635071d4279SBram Moolenaar * current window open.
1636071d4279SBram Moolenaar */
1637071d4279SBram Moolenaar int
getDeepestNesting(void)163866f948e9SBram Moolenaar getDeepestNesting(void)
1639071d4279SBram Moolenaar {
1640071d4279SBram Moolenaar checkupdate(curwin);
1641071d4279SBram Moolenaar return getDeepestNestingRecurse(&curwin->w_folds);
1642071d4279SBram Moolenaar }
1643071d4279SBram Moolenaar
1644071d4279SBram Moolenaar static int
getDeepestNestingRecurse(garray_T * gap)164566f948e9SBram Moolenaar getDeepestNestingRecurse(garray_T *gap)
1646071d4279SBram Moolenaar {
1647071d4279SBram Moolenaar int i;
1648071d4279SBram Moolenaar int level;
1649071d4279SBram Moolenaar int maxlevel = 0;
1650071d4279SBram Moolenaar fold_T *fp;
1651071d4279SBram Moolenaar
1652071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data;
1653071d4279SBram Moolenaar for (i = 0; i < gap->ga_len; ++i)
1654071d4279SBram Moolenaar {
1655071d4279SBram Moolenaar level = getDeepestNestingRecurse(&fp[i].fd_nested) + 1;
1656071d4279SBram Moolenaar if (level > maxlevel)
1657071d4279SBram Moolenaar maxlevel = level;
1658071d4279SBram Moolenaar }
1659071d4279SBram Moolenaar
1660071d4279SBram Moolenaar return maxlevel;
1661071d4279SBram Moolenaar }
1662071d4279SBram Moolenaar
1663217e1b83SBram Moolenaar // check_closed() {{{2
1664071d4279SBram Moolenaar /*
1665071d4279SBram Moolenaar * Check if a fold is closed and update the info needed to check nested folds.
1666071d4279SBram Moolenaar */
1667071d4279SBram Moolenaar static int
check_closed(win_T * win,fold_T * fp,int * use_levelp,int level,int * maybe_smallp,linenr_T lnum_off)166866f948e9SBram Moolenaar check_closed(
166966f948e9SBram Moolenaar win_T *win,
167066f948e9SBram Moolenaar fold_T *fp,
1671217e1b83SBram Moolenaar int *use_levelp, // TRUE: outer fold had FD_LEVEL
1672217e1b83SBram Moolenaar int level, // folding depth
1673217e1b83SBram Moolenaar int *maybe_smallp, // TRUE: outer this had fd_small == MAYBE
1674217e1b83SBram Moolenaar linenr_T lnum_off) // line number offset for fp->fd_top
1675071d4279SBram Moolenaar {
1676071d4279SBram Moolenaar int closed = FALSE;
1677071d4279SBram Moolenaar
1678217e1b83SBram Moolenaar // Check if this fold is closed. If the flag is FD_LEVEL this
1679217e1b83SBram Moolenaar // fold and all folds it contains depend on 'foldlevel'.
1680071d4279SBram Moolenaar if (*use_levelp || fp->fd_flags == FD_LEVEL)
1681071d4279SBram Moolenaar {
1682071d4279SBram Moolenaar *use_levelp = TRUE;
1683071d4279SBram Moolenaar if (level >= win->w_p_fdl)
1684071d4279SBram Moolenaar closed = TRUE;
1685071d4279SBram Moolenaar }
1686071d4279SBram Moolenaar else if (fp->fd_flags == FD_CLOSED)
1687071d4279SBram Moolenaar closed = TRUE;
1688071d4279SBram Moolenaar
1689217e1b83SBram Moolenaar // Small fold isn't closed anyway.
1690071d4279SBram Moolenaar if (fp->fd_small == MAYBE)
1691071d4279SBram Moolenaar *maybe_smallp = TRUE;
1692071d4279SBram Moolenaar if (closed)
1693071d4279SBram Moolenaar {
1694071d4279SBram Moolenaar if (*maybe_smallp)
1695071d4279SBram Moolenaar fp->fd_small = MAYBE;
1696071d4279SBram Moolenaar checkSmall(win, fp, lnum_off);
1697071d4279SBram Moolenaar if (fp->fd_small == TRUE)
1698071d4279SBram Moolenaar closed = FALSE;
1699071d4279SBram Moolenaar }
1700071d4279SBram Moolenaar return closed;
1701071d4279SBram Moolenaar }
1702071d4279SBram Moolenaar
1703217e1b83SBram Moolenaar // checkSmall() {{{2
1704071d4279SBram Moolenaar /*
1705071d4279SBram Moolenaar * Update fd_small field of fold "fp".
1706071d4279SBram Moolenaar */
1707071d4279SBram Moolenaar static void
checkSmall(win_T * wp,fold_T * fp,linenr_T lnum_off)170866f948e9SBram Moolenaar checkSmall(
170966f948e9SBram Moolenaar win_T *wp,
171066f948e9SBram Moolenaar fold_T *fp,
1711217e1b83SBram Moolenaar linenr_T lnum_off) // offset for fp->fd_top
1712071d4279SBram Moolenaar {
1713071d4279SBram Moolenaar int count;
1714071d4279SBram Moolenaar int n;
1715071d4279SBram Moolenaar
1716071d4279SBram Moolenaar if (fp->fd_small == MAYBE)
1717071d4279SBram Moolenaar {
1718217e1b83SBram Moolenaar // Mark any nested folds to maybe-small
1719071d4279SBram Moolenaar setSmallMaybe(&fp->fd_nested);
1720071d4279SBram Moolenaar
1721071d4279SBram Moolenaar if (fp->fd_len > curwin->w_p_fml)
1722071d4279SBram Moolenaar fp->fd_small = FALSE;
1723071d4279SBram Moolenaar else
1724071d4279SBram Moolenaar {
1725071d4279SBram Moolenaar count = 0;
1726071d4279SBram Moolenaar for (n = 0; n < fp->fd_len; ++n)
1727071d4279SBram Moolenaar {
1728071d4279SBram Moolenaar count += plines_win_nofold(wp, fp->fd_top + lnum_off + n);
1729071d4279SBram Moolenaar if (count > curwin->w_p_fml)
1730071d4279SBram Moolenaar {
1731071d4279SBram Moolenaar fp->fd_small = FALSE;
1732071d4279SBram Moolenaar return;
1733071d4279SBram Moolenaar }
1734071d4279SBram Moolenaar }
1735071d4279SBram Moolenaar fp->fd_small = TRUE;
1736071d4279SBram Moolenaar }
1737071d4279SBram Moolenaar }
1738071d4279SBram Moolenaar }
1739071d4279SBram Moolenaar
1740217e1b83SBram Moolenaar // setSmallMaybe() {{{2
1741071d4279SBram Moolenaar /*
1742071d4279SBram Moolenaar * Set small flags in "gap" to MAYBE.
1743071d4279SBram Moolenaar */
1744071d4279SBram Moolenaar static void
setSmallMaybe(garray_T * gap)174566f948e9SBram Moolenaar setSmallMaybe(garray_T *gap)
1746071d4279SBram Moolenaar {
1747071d4279SBram Moolenaar int i;
1748071d4279SBram Moolenaar fold_T *fp;
1749071d4279SBram Moolenaar
1750071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data;
1751071d4279SBram Moolenaar for (i = 0; i < gap->ga_len; ++i)
1752071d4279SBram Moolenaar fp[i].fd_small = MAYBE;
1753071d4279SBram Moolenaar }
1754071d4279SBram Moolenaar
1755217e1b83SBram Moolenaar // foldCreateMarkers() {{{2
1756071d4279SBram Moolenaar /*
1757071d4279SBram Moolenaar * Create a fold from line "start" to line "end" (inclusive) in the current
1758071d4279SBram Moolenaar * window by adding markers.
1759071d4279SBram Moolenaar */
1760071d4279SBram Moolenaar static void
foldCreateMarkers(linenr_T start,linenr_T end)176166f948e9SBram Moolenaar foldCreateMarkers(linenr_T start, linenr_T end)
1762071d4279SBram Moolenaar {
1763071d4279SBram Moolenaar if (!curbuf->b_p_ma)
1764071d4279SBram Moolenaar {
1765108010aaSBram Moolenaar emsg(_(e_cannot_make_changes_modifiable_is_off));
1766071d4279SBram Moolenaar return;
1767071d4279SBram Moolenaar }
1768071d4279SBram Moolenaar parseMarker(curwin);
1769071d4279SBram Moolenaar
1770071d4279SBram Moolenaar foldAddMarker(start, curwin->w_p_fmr, foldstartmarkerlen);
1771071d4279SBram Moolenaar foldAddMarker(end, foldendmarker, foldendmarkerlen);
1772071d4279SBram Moolenaar
1773217e1b83SBram Moolenaar // Update both changes here, to avoid all folds after the start are
1774217e1b83SBram Moolenaar // changed when the start marker is inserted and the end isn't.
1775071d4279SBram Moolenaar changed_lines(start, (colnr_T)0, end, 0L);
1776071d4279SBram Moolenaar }
1777071d4279SBram Moolenaar
1778217e1b83SBram Moolenaar // foldAddMarker() {{{2
1779071d4279SBram Moolenaar /*
1780071d4279SBram Moolenaar * Add "marker[markerlen]" in 'commentstring' to line "lnum".
1781071d4279SBram Moolenaar */
1782071d4279SBram Moolenaar static void
foldAddMarker(linenr_T lnum,char_u * marker,int markerlen)178366f948e9SBram Moolenaar foldAddMarker(linenr_T lnum, char_u *marker, int markerlen)
1784071d4279SBram Moolenaar {
1785071d4279SBram Moolenaar char_u *cms = curbuf->b_p_cms;
1786071d4279SBram Moolenaar char_u *line;
1787071d4279SBram Moolenaar int line_len;
1788071d4279SBram Moolenaar char_u *newline;
1789071d4279SBram Moolenaar char_u *p = (char_u *)strstr((char *)curbuf->b_p_cms, "%s");
1790025a6b70SBram Moolenaar int line_is_comment = FALSE;
1791071d4279SBram Moolenaar
1792217e1b83SBram Moolenaar // Allocate a new line: old-line + 'cms'-start + marker + 'cms'-end
1793071d4279SBram Moolenaar line = ml_get(lnum);
1794071d4279SBram Moolenaar line_len = (int)STRLEN(line);
1795071d4279SBram Moolenaar
1796071d4279SBram Moolenaar if (u_save(lnum - 1, lnum + 1) == OK)
1797071d4279SBram Moolenaar {
1798217e1b83SBram Moolenaar // Check if the line ends with an unclosed comment
1799025a6b70SBram Moolenaar (void)skip_comment(line, FALSE, FALSE, &line_is_comment);
1800964b3746SBram Moolenaar newline = alloc(line_len + markerlen + STRLEN(cms) + 1);
1801071d4279SBram Moolenaar if (newline == NULL)
1802071d4279SBram Moolenaar return;
1803071d4279SBram Moolenaar STRCPY(newline, line);
1804217e1b83SBram Moolenaar // Append the marker to the end of the line
1805025a6b70SBram Moolenaar if (p == NULL || line_is_comment)
1806ce0842a6SBram Moolenaar vim_strncpy(newline + line_len, marker, markerlen);
1807071d4279SBram Moolenaar else
1808071d4279SBram Moolenaar {
1809071d4279SBram Moolenaar STRCPY(newline + line_len, cms);
1810071d4279SBram Moolenaar STRNCPY(newline + line_len + (p - cms), marker, markerlen);
1811071d4279SBram Moolenaar STRCPY(newline + line_len + (p - cms) + markerlen, p + 2);
1812071d4279SBram Moolenaar }
1813071d4279SBram Moolenaar
1814071d4279SBram Moolenaar ml_replace(lnum, newline, FALSE);
1815071d4279SBram Moolenaar }
1816071d4279SBram Moolenaar }
1817071d4279SBram Moolenaar
1818217e1b83SBram Moolenaar // deleteFoldMarkers() {{{2
1819071d4279SBram Moolenaar /*
1820071d4279SBram Moolenaar * Delete the markers for a fold, causing it to be deleted.
1821071d4279SBram Moolenaar */
1822071d4279SBram Moolenaar static void
deleteFoldMarkers(fold_T * fp,int recursive,linenr_T lnum_off)182366f948e9SBram Moolenaar deleteFoldMarkers(
182466f948e9SBram Moolenaar fold_T *fp,
182566f948e9SBram Moolenaar int recursive,
1826217e1b83SBram Moolenaar linenr_T lnum_off) // offset for fp->fd_top
1827071d4279SBram Moolenaar {
1828071d4279SBram Moolenaar int i;
1829071d4279SBram Moolenaar
1830071d4279SBram Moolenaar if (recursive)
1831071d4279SBram Moolenaar for (i = 0; i < fp->fd_nested.ga_len; ++i)
1832071d4279SBram Moolenaar deleteFoldMarkers((fold_T *)fp->fd_nested.ga_data + i, TRUE,
1833071d4279SBram Moolenaar lnum_off + fp->fd_top);
1834071d4279SBram Moolenaar foldDelMarker(fp->fd_top + lnum_off, curwin->w_p_fmr, foldstartmarkerlen);
1835071d4279SBram Moolenaar foldDelMarker(fp->fd_top + lnum_off + fp->fd_len - 1,
1836071d4279SBram Moolenaar foldendmarker, foldendmarkerlen);
1837071d4279SBram Moolenaar }
1838071d4279SBram Moolenaar
1839217e1b83SBram Moolenaar // foldDelMarker() {{{2
1840071d4279SBram Moolenaar /*
1841071d4279SBram Moolenaar * Delete marker "marker[markerlen]" at the end of line "lnum".
1842071d4279SBram Moolenaar * Delete 'commentstring' if it matches.
18439a4a8c4dSBram Moolenaar * If the marker is not found, there is no error message. Could be a missing
1844071d4279SBram Moolenaar * close-marker.
1845071d4279SBram Moolenaar */
1846071d4279SBram Moolenaar static void
foldDelMarker(linenr_T lnum,char_u * marker,int markerlen)184766f948e9SBram Moolenaar foldDelMarker(linenr_T lnum, char_u *marker, int markerlen)
1848071d4279SBram Moolenaar {
1849071d4279SBram Moolenaar char_u *line;
1850071d4279SBram Moolenaar char_u *newline;
1851071d4279SBram Moolenaar char_u *p;
1852071d4279SBram Moolenaar int len;
1853071d4279SBram Moolenaar char_u *cms = curbuf->b_p_cms;
1854071d4279SBram Moolenaar char_u *cms2;
1855071d4279SBram Moolenaar
18569a4a8c4dSBram Moolenaar // end marker may be missing and fold extends below the last line
18579a4a8c4dSBram Moolenaar if (lnum > curbuf->b_ml.ml_line_count)
18589a4a8c4dSBram Moolenaar return;
1859071d4279SBram Moolenaar line = ml_get(lnum);
1860071d4279SBram Moolenaar for (p = line; *p != NUL; ++p)
1861071d4279SBram Moolenaar if (STRNCMP(p, marker, markerlen) == 0)
1862071d4279SBram Moolenaar {
1863217e1b83SBram Moolenaar // Found the marker, include a digit if it's there.
1864071d4279SBram Moolenaar len = markerlen;
1865071d4279SBram Moolenaar if (VIM_ISDIGIT(p[len]))
1866071d4279SBram Moolenaar ++len;
1867071d4279SBram Moolenaar if (*cms != NUL)
1868071d4279SBram Moolenaar {
1869217e1b83SBram Moolenaar // Also delete 'commentstring' if it matches.
1870071d4279SBram Moolenaar cms2 = (char_u *)strstr((char *)cms, "%s");
1871071d4279SBram Moolenaar if (p - line >= cms2 - cms
1872071d4279SBram Moolenaar && STRNCMP(p - (cms2 - cms), cms, cms2 - cms) == 0
1873071d4279SBram Moolenaar && STRNCMP(p + len, cms2 + 2, STRLEN(cms2 + 2)) == 0)
1874071d4279SBram Moolenaar {
1875071d4279SBram Moolenaar p -= cms2 - cms;
1876071d4279SBram Moolenaar len += (int)STRLEN(cms) - 2;
1877071d4279SBram Moolenaar }
1878071d4279SBram Moolenaar }
1879071d4279SBram Moolenaar if (u_save(lnum - 1, lnum + 1) == OK)
1880071d4279SBram Moolenaar {
1881217e1b83SBram Moolenaar // Make new line: text-before-marker + text-after-marker
1882964b3746SBram Moolenaar newline = alloc(STRLEN(line) - len + 1);
1883071d4279SBram Moolenaar if (newline != NULL)
1884071d4279SBram Moolenaar {
1885071d4279SBram Moolenaar STRNCPY(newline, line, p - line);
1886071d4279SBram Moolenaar STRCPY(newline + (p - line), p + len);
1887071d4279SBram Moolenaar ml_replace(lnum, newline, FALSE);
1888071d4279SBram Moolenaar }
1889071d4279SBram Moolenaar }
1890071d4279SBram Moolenaar break;
1891071d4279SBram Moolenaar }
1892071d4279SBram Moolenaar }
1893071d4279SBram Moolenaar
1894217e1b83SBram Moolenaar // get_foldtext() {{{2
18957b0294cbSBram Moolenaar /*
18967b0294cbSBram Moolenaar * Return the text for a closed fold at line "lnum", with last line "lnume".
1897ee695f78SBram Moolenaar * When 'foldtext' isn't set puts the result in "buf[FOLD_TEXT_LEN]".
1898ee695f78SBram Moolenaar * Otherwise the result is in allocated memory.
18997b0294cbSBram Moolenaar */
19007b0294cbSBram Moolenaar char_u *
get_foldtext(win_T * wp,linenr_T lnum,linenr_T lnume,foldinfo_T * foldinfo,char_u * buf)190166f948e9SBram Moolenaar get_foldtext(
190266f948e9SBram Moolenaar win_T *wp,
190366f948e9SBram Moolenaar linenr_T lnum,
190466f948e9SBram Moolenaar linenr_T lnume,
190566f948e9SBram Moolenaar foldinfo_T *foldinfo,
190666f948e9SBram Moolenaar char_u *buf)
19077b0294cbSBram Moolenaar {
19087b0294cbSBram Moolenaar char_u *text = NULL;
19097b0294cbSBram Moolenaar #ifdef FEAT_EVAL
1910217e1b83SBram Moolenaar // an error occurred when evaluating 'fdt' setting
1911dab38d51SBram Moolenaar static int got_fdt_error = FALSE;
1912dab38d51SBram Moolenaar int save_did_emsg = did_emsg;
1913dab38d51SBram Moolenaar static win_T *last_wp = NULL;
1914dab38d51SBram Moolenaar static linenr_T last_lnum = 0;
1915dab38d51SBram Moolenaar
1916dab38d51SBram Moolenaar if (last_wp != wp || last_wp == NULL
1917dab38d51SBram Moolenaar || last_lnum > lnum || last_lnum == 0)
1918217e1b83SBram Moolenaar // window changed, try evaluating foldtext setting once again
1919dab38d51SBram Moolenaar got_fdt_error = FALSE;
1920dab38d51SBram Moolenaar
1921dab38d51SBram Moolenaar if (!got_fdt_error)
1922217e1b83SBram Moolenaar // a previous error should not abort evaluating 'foldexpr'
1923dab38d51SBram Moolenaar did_emsg = FALSE;
1924dab38d51SBram Moolenaar
19257b0294cbSBram Moolenaar if (*wp->w_p_fdt != NUL)
19267b0294cbSBram Moolenaar {
19275b88ba47SBram Moolenaar char_u dashes[MAX_LEVEL + 2];
19287b0294cbSBram Moolenaar win_T *save_curwin;
19297b0294cbSBram Moolenaar int level;
19307b0294cbSBram Moolenaar char_u *p;
19317b0294cbSBram Moolenaar
1932217e1b83SBram Moolenaar // Set "v:foldstart" and "v:foldend".
19337b0294cbSBram Moolenaar set_vim_var_nr(VV_FOLDSTART, lnum);
19347b0294cbSBram Moolenaar set_vim_var_nr(VV_FOLDEND, lnume);
19357b0294cbSBram Moolenaar
1936217e1b83SBram Moolenaar // Set "v:folddashes" to a string of "level" dashes.
1937217e1b83SBram Moolenaar // Set "v:foldlevel" to "level".
19387b0294cbSBram Moolenaar level = foldinfo->fi_level;
19395b88ba47SBram Moolenaar if (level > (int)sizeof(dashes) - 1)
19405b88ba47SBram Moolenaar level = (int)sizeof(dashes) - 1;
19417b0294cbSBram Moolenaar vim_memset(dashes, '-', (size_t)level);
19427b0294cbSBram Moolenaar dashes[level] = NUL;
19437b0294cbSBram Moolenaar set_vim_var_string(VV_FOLDDASHES, dashes, -1);
19447b0294cbSBram Moolenaar set_vim_var_nr(VV_FOLDLEVEL, (long)level);
1945dab38d51SBram Moolenaar
1946217e1b83SBram Moolenaar // skip evaluating foldtext on errors
1947dab38d51SBram Moolenaar if (!got_fdt_error)
1948dab38d51SBram Moolenaar {
19497b0294cbSBram Moolenaar save_curwin = curwin;
19507b0294cbSBram Moolenaar curwin = wp;
19517b0294cbSBram Moolenaar curbuf = wp->w_buffer;
19527b0294cbSBram Moolenaar
1953217e1b83SBram Moolenaar ++emsg_silent; // handle exceptions, but don't display errors
1954b171fb17SBram Moolenaar text = eval_to_string_safe(wp->w_p_fdt,
1955d1f56e68SBram Moolenaar was_set_insecurely((char_u *)"foldtext", OPT_LOCAL));
1956dab38d51SBram Moolenaar --emsg_silent;
1957dab38d51SBram Moolenaar
1958dab38d51SBram Moolenaar if (text == NULL || did_emsg)
1959dab38d51SBram Moolenaar got_fdt_error = TRUE;
19607b0294cbSBram Moolenaar
19617b0294cbSBram Moolenaar curwin = save_curwin;
19627b0294cbSBram Moolenaar curbuf = curwin->w_buffer;
1963dab38d51SBram Moolenaar }
1964dab38d51SBram Moolenaar last_lnum = lnum;
1965dab38d51SBram Moolenaar last_wp = wp;
19667b0294cbSBram Moolenaar set_vim_var_string(VV_FOLDDASHES, NULL, -1);
19677b0294cbSBram Moolenaar
1968dab38d51SBram Moolenaar if (!did_emsg && save_did_emsg)
1969dab38d51SBram Moolenaar did_emsg = save_did_emsg;
1970dab38d51SBram Moolenaar
19717b0294cbSBram Moolenaar if (text != NULL)
19727b0294cbSBram Moolenaar {
1973217e1b83SBram Moolenaar // Replace unprintable characters, if there are any. But
1974217e1b83SBram Moolenaar // replace a TAB with a space.
19757b0294cbSBram Moolenaar for (p = text; *p != NUL; ++p)
19767b0294cbSBram Moolenaar {
1977009b2592SBram Moolenaar int len;
1978009b2592SBram Moolenaar
19790fa313a7SBram Moolenaar if (has_mbyte && (len = (*mb_ptr2len)(p)) > 1)
19807b0294cbSBram Moolenaar {
19817b0294cbSBram Moolenaar if (!vim_isprintc((*mb_ptr2char)(p)))
19827b0294cbSBram Moolenaar break;
19837b0294cbSBram Moolenaar p += len - 1;
19847b0294cbSBram Moolenaar }
19857b0294cbSBram Moolenaar else
19867b0294cbSBram Moolenaar if (*p == TAB)
19877b0294cbSBram Moolenaar *p = ' ';
19887b0294cbSBram Moolenaar else if (ptr2cells(p) > 1)
19897b0294cbSBram Moolenaar break;
19907b0294cbSBram Moolenaar }
19917b0294cbSBram Moolenaar if (*p != NUL)
19927b0294cbSBram Moolenaar {
19937b0294cbSBram Moolenaar p = transstr(text);
19947b0294cbSBram Moolenaar vim_free(text);
19957b0294cbSBram Moolenaar text = p;
19967b0294cbSBram Moolenaar }
19977b0294cbSBram Moolenaar }
19987b0294cbSBram Moolenaar }
19997b0294cbSBram Moolenaar if (text == NULL)
20007b0294cbSBram Moolenaar #endif
20017b0294cbSBram Moolenaar {
2002ee695f78SBram Moolenaar long count = (long)(lnume - lnum + 1);
2003ee695f78SBram Moolenaar
2004ee695f78SBram Moolenaar vim_snprintf((char *)buf, FOLD_TEXT_LEN,
20051c465444SBram Moolenaar NGETTEXT("+--%3ld line folded ",
2006ee695f78SBram Moolenaar "+--%3ld lines folded ", count),
2007ee695f78SBram Moolenaar count);
20087b0294cbSBram Moolenaar text = buf;
20097b0294cbSBram Moolenaar }
20107b0294cbSBram Moolenaar return text;
20117b0294cbSBram Moolenaar }
20127b0294cbSBram Moolenaar
2013217e1b83SBram Moolenaar // foldtext_cleanup() {{{2
2014db022f3fSBram Moolenaar #ifdef FEAT_EVAL
2015071d4279SBram Moolenaar /*
2016071d4279SBram Moolenaar * Remove 'foldmarker' and 'commentstring' from "str" (in-place).
2017071d4279SBram Moolenaar */
2018db022f3fSBram Moolenaar static void
foldtext_cleanup(char_u * str)201966f948e9SBram Moolenaar foldtext_cleanup(char_u *str)
2020071d4279SBram Moolenaar {
2021217e1b83SBram Moolenaar char_u *cms_start; // first part or the whole comment
2022217e1b83SBram Moolenaar int cms_slen = 0; // length of cms_start
2023217e1b83SBram Moolenaar char_u *cms_end; // last part of the comment or NULL
2024217e1b83SBram Moolenaar int cms_elen = 0; // length of cms_end
2025071d4279SBram Moolenaar char_u *s;
2026def9e829SBram Moolenaar char_u *p;
2027071d4279SBram Moolenaar int len;
2028071d4279SBram Moolenaar int did1 = FALSE;
2029071d4279SBram Moolenaar int did2 = FALSE;
2030071d4279SBram Moolenaar
2031217e1b83SBram Moolenaar // Ignore leading and trailing white space in 'commentstring'.
2032071d4279SBram Moolenaar cms_start = skipwhite(curbuf->b_p_cms);
2033a93fa7eeSBram Moolenaar cms_slen = (int)STRLEN(cms_start);
20341c465444SBram Moolenaar while (cms_slen > 0 && VIM_ISWHITE(cms_start[cms_slen - 1]))
2035071d4279SBram Moolenaar --cms_slen;
2036071d4279SBram Moolenaar
2037217e1b83SBram Moolenaar // locate "%s" in 'commentstring', use the part before and after it.
2038071d4279SBram Moolenaar cms_end = (char_u *)strstr((char *)cms_start, "%s");
2039071d4279SBram Moolenaar if (cms_end != NULL)
2040071d4279SBram Moolenaar {
2041a93fa7eeSBram Moolenaar cms_elen = cms_slen - (int)(cms_end - cms_start);
2042a93fa7eeSBram Moolenaar cms_slen = (int)(cms_end - cms_start);
2043071d4279SBram Moolenaar
2044217e1b83SBram Moolenaar // exclude white space before "%s"
20451c465444SBram Moolenaar while (cms_slen > 0 && VIM_ISWHITE(cms_start[cms_slen - 1]))
2046071d4279SBram Moolenaar --cms_slen;
2047071d4279SBram Moolenaar
2048217e1b83SBram Moolenaar // skip "%s" and white space after it
2049071d4279SBram Moolenaar s = skipwhite(cms_end + 2);
2050a93fa7eeSBram Moolenaar cms_elen -= (int)(s - cms_end);
2051071d4279SBram Moolenaar cms_end = s;
2052071d4279SBram Moolenaar }
2053071d4279SBram Moolenaar parseMarker(curwin);
2054071d4279SBram Moolenaar
2055071d4279SBram Moolenaar for (s = str; *s != NUL; )
2056071d4279SBram Moolenaar {
2057071d4279SBram Moolenaar len = 0;
2058071d4279SBram Moolenaar if (STRNCMP(s, curwin->w_p_fmr, foldstartmarkerlen) == 0)
2059071d4279SBram Moolenaar len = foldstartmarkerlen;
2060071d4279SBram Moolenaar else if (STRNCMP(s, foldendmarker, foldendmarkerlen) == 0)
2061071d4279SBram Moolenaar len = foldendmarkerlen;
2062def9e829SBram Moolenaar if (len > 0)
2063def9e829SBram Moolenaar {
2064071d4279SBram Moolenaar if (VIM_ISDIGIT(s[len]))
2065071d4279SBram Moolenaar ++len;
2066def9e829SBram Moolenaar
2067217e1b83SBram Moolenaar // May remove 'commentstring' start. Useful when it's a double
2068217e1b83SBram Moolenaar // quote and we already removed a double quote.
20691c465444SBram Moolenaar for (p = s; p > str && VIM_ISWHITE(p[-1]); --p)
2070def9e829SBram Moolenaar ;
2071def9e829SBram Moolenaar if (p >= str + cms_slen
2072def9e829SBram Moolenaar && STRNCMP(p - cms_slen, cms_start, cms_slen) == 0)
2073def9e829SBram Moolenaar {
2074a93fa7eeSBram Moolenaar len += (int)(s - p) + cms_slen;
2075def9e829SBram Moolenaar s = p - cms_slen;
2076def9e829SBram Moolenaar }
2077071d4279SBram Moolenaar }
2078071d4279SBram Moolenaar else if (cms_end != NULL)
2079071d4279SBram Moolenaar {
2080def9e829SBram Moolenaar if (!did1 && cms_slen > 0 && STRNCMP(s, cms_start, cms_slen) == 0)
2081071d4279SBram Moolenaar {
2082071d4279SBram Moolenaar len = cms_slen;
2083071d4279SBram Moolenaar did1 = TRUE;
2084071d4279SBram Moolenaar }
2085def9e829SBram Moolenaar else if (!did2 && cms_elen > 0
2086def9e829SBram Moolenaar && STRNCMP(s, cms_end, cms_elen) == 0)
2087071d4279SBram Moolenaar {
2088071d4279SBram Moolenaar len = cms_elen;
2089071d4279SBram Moolenaar did2 = TRUE;
2090071d4279SBram Moolenaar }
2091071d4279SBram Moolenaar }
2092071d4279SBram Moolenaar if (len != 0)
2093071d4279SBram Moolenaar {
20941c465444SBram Moolenaar while (VIM_ISWHITE(s[len]))
2095071d4279SBram Moolenaar ++len;
20968c8de839SBram Moolenaar STRMOVE(s, s + len);
2097071d4279SBram Moolenaar }
2098071d4279SBram Moolenaar else
2099071d4279SBram Moolenaar {
210091acfffcSBram Moolenaar MB_PTR_ADV(s);
2101071d4279SBram Moolenaar }
2102071d4279SBram Moolenaar }
2103071d4279SBram Moolenaar }
2104db022f3fSBram Moolenaar #endif
2105071d4279SBram Moolenaar
2106217e1b83SBram Moolenaar // Folding by indent, expr, marker and syntax. {{{1
2107217e1b83SBram Moolenaar // Define "fline_T", passed to get fold level for a line. {{{2
2108071d4279SBram Moolenaar typedef struct
2109071d4279SBram Moolenaar {
2110217e1b83SBram Moolenaar win_T *wp; // window
2111217e1b83SBram Moolenaar linenr_T lnum; // current line number
2112217e1b83SBram Moolenaar linenr_T off; // offset between lnum and real line number
2113217e1b83SBram Moolenaar linenr_T lnum_save; // line nr used by foldUpdateIEMSRecurse()
2114217e1b83SBram Moolenaar int lvl; // current level (-1 for undefined)
2115217e1b83SBram Moolenaar int lvl_next; // level used for next line
2116217e1b83SBram Moolenaar int start; // number of folds that are forced to start at
2117217e1b83SBram Moolenaar // this line.
2118217e1b83SBram Moolenaar int end; // level of fold that is forced to end below
2119217e1b83SBram Moolenaar // this line
2120217e1b83SBram Moolenaar int had_end; // level of fold that is forced to end above
2121217e1b83SBram Moolenaar // this line (copy of "end" of prev. line)
2122071d4279SBram Moolenaar } fline_T;
2123071d4279SBram Moolenaar
2124217e1b83SBram Moolenaar // Flag is set when redrawing is needed.
2125071d4279SBram Moolenaar static int fold_changed;
2126071d4279SBram Moolenaar
2127217e1b83SBram Moolenaar // Function declarations. {{{2
2128d99df423SBram 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);
2129d25c16e2SBram Moolenaar static int foldInsert(garray_T *gap, int i);
2130d25c16e2SBram Moolenaar static void foldSplit(garray_T *gap, int i, linenr_T top, linenr_T bot);
2131d25c16e2SBram Moolenaar static void foldRemove(garray_T *gap, linenr_T top, linenr_T bot);
2132d25c16e2SBram Moolenaar static void foldMerge(fold_T *fp1, garray_T *gap, fold_T *fp2);
2133d25c16e2SBram Moolenaar static void foldlevelIndent(fline_T *flp);
2134071d4279SBram Moolenaar #ifdef FEAT_DIFF
2135d25c16e2SBram Moolenaar static void foldlevelDiff(fline_T *flp);
2136071d4279SBram Moolenaar #endif
2137d25c16e2SBram Moolenaar static void foldlevelExpr(fline_T *flp);
2138d25c16e2SBram Moolenaar static void foldlevelMarker(fline_T *flp);
2139d25c16e2SBram Moolenaar static void foldlevelSyntax(fline_T *flp);
2140071d4279SBram Moolenaar
2141217e1b83SBram Moolenaar // foldUpdateIEMS() {{{2
2142071d4279SBram Moolenaar /*
2143071d4279SBram Moolenaar * Update the folding for window "wp", at least from lines "top" to "bot".
2144071d4279SBram Moolenaar * Return TRUE if any folds did change.
2145071d4279SBram Moolenaar */
2146071d4279SBram Moolenaar static void
foldUpdateIEMS(win_T * wp,linenr_T top,linenr_T bot)214766f948e9SBram Moolenaar foldUpdateIEMS(win_T *wp, linenr_T top, linenr_T bot)
2148071d4279SBram Moolenaar {
2149071d4279SBram Moolenaar linenr_T start;
2150071d4279SBram Moolenaar linenr_T end;
2151071d4279SBram Moolenaar fline_T fline;
2152d99df423SBram Moolenaar void (*getlevel)(fline_T *);
2153071d4279SBram Moolenaar int level;
2154071d4279SBram Moolenaar fold_T *fp;
2155071d4279SBram Moolenaar
2156217e1b83SBram Moolenaar // Avoid problems when being called recursively.
2157071d4279SBram Moolenaar if (invalid_top != (linenr_T)0)
2158071d4279SBram Moolenaar return;
2159071d4279SBram Moolenaar
2160071d4279SBram Moolenaar if (wp->w_foldinvalid)
2161071d4279SBram Moolenaar {
2162217e1b83SBram Moolenaar // Need to update all folds.
2163071d4279SBram Moolenaar top = 1;
2164071d4279SBram Moolenaar bot = wp->w_buffer->b_ml.ml_line_count;
2165071d4279SBram Moolenaar wp->w_foldinvalid = FALSE;
2166071d4279SBram Moolenaar
2167217e1b83SBram Moolenaar // Mark all folds a maybe-small.
2168071d4279SBram Moolenaar setSmallMaybe(&wp->w_folds);
2169071d4279SBram Moolenaar }
2170071d4279SBram Moolenaar
2171071d4279SBram Moolenaar #ifdef FEAT_DIFF
2172217e1b83SBram Moolenaar // add the context for "diff" folding
2173071d4279SBram Moolenaar if (foldmethodIsDiff(wp))
2174071d4279SBram Moolenaar {
2175071d4279SBram Moolenaar if (top > diff_context)
2176071d4279SBram Moolenaar top -= diff_context;
2177071d4279SBram Moolenaar else
2178071d4279SBram Moolenaar top = 1;
2179071d4279SBram Moolenaar bot += diff_context;
2180071d4279SBram Moolenaar }
2181071d4279SBram Moolenaar #endif
2182071d4279SBram Moolenaar
2183217e1b83SBram Moolenaar // When deleting lines at the end of the buffer "top" can be past the end
2184217e1b83SBram Moolenaar // of the buffer.
2185071d4279SBram Moolenaar if (top > wp->w_buffer->b_ml.ml_line_count)
2186071d4279SBram Moolenaar top = wp->w_buffer->b_ml.ml_line_count;
2187071d4279SBram Moolenaar
2188071d4279SBram Moolenaar fold_changed = FALSE;
2189071d4279SBram Moolenaar fline.wp = wp;
2190071d4279SBram Moolenaar fline.off = 0;
2191071d4279SBram Moolenaar fline.lvl = 0;
2192071d4279SBram Moolenaar fline.lvl_next = -1;
2193071d4279SBram Moolenaar fline.start = 0;
2194071d4279SBram Moolenaar fline.end = MAX_LEVEL + 1;
2195071d4279SBram Moolenaar fline.had_end = MAX_LEVEL + 1;
2196071d4279SBram Moolenaar
2197071d4279SBram Moolenaar invalid_top = top;
2198071d4279SBram Moolenaar invalid_bot = bot;
2199071d4279SBram Moolenaar
2200071d4279SBram Moolenaar if (foldmethodIsMarker(wp))
2201071d4279SBram Moolenaar {
2202071d4279SBram Moolenaar getlevel = foldlevelMarker;
2203071d4279SBram Moolenaar
2204217e1b83SBram Moolenaar // Init marker variables to speed up foldlevelMarker().
2205071d4279SBram Moolenaar parseMarker(wp);
2206071d4279SBram Moolenaar
2207217e1b83SBram Moolenaar // Need to get the level of the line above top, it is used if there is
2208217e1b83SBram Moolenaar // no marker at the top.
2209071d4279SBram Moolenaar if (top > 1)
2210071d4279SBram Moolenaar {
2211217e1b83SBram Moolenaar // Get the fold level at top - 1.
2212071d4279SBram Moolenaar level = foldLevelWin(wp, top - 1);
2213071d4279SBram Moolenaar
2214217e1b83SBram Moolenaar // The fold may end just above the top, check for that.
2215071d4279SBram Moolenaar fline.lnum = top - 1;
2216071d4279SBram Moolenaar fline.lvl = level;
2217071d4279SBram Moolenaar getlevel(&fline);
2218071d4279SBram Moolenaar
2219217e1b83SBram Moolenaar // If a fold started here, we already had the level, if it stops
2220217e1b83SBram Moolenaar // here, we need to use lvl_next. Could also start and end a fold
2221217e1b83SBram Moolenaar // in the same line.
2222071d4279SBram Moolenaar if (fline.lvl > level)
2223071d4279SBram Moolenaar fline.lvl = level - (fline.lvl - fline.lvl_next);
2224071d4279SBram Moolenaar else
2225071d4279SBram Moolenaar fline.lvl = fline.lvl_next;
2226071d4279SBram Moolenaar }
2227071d4279SBram Moolenaar fline.lnum = top;
2228071d4279SBram Moolenaar getlevel(&fline);
2229071d4279SBram Moolenaar }
2230071d4279SBram Moolenaar else
2231071d4279SBram Moolenaar {
2232071d4279SBram Moolenaar fline.lnum = top;
2233071d4279SBram Moolenaar if (foldmethodIsExpr(wp))
2234071d4279SBram Moolenaar {
2235071d4279SBram Moolenaar getlevel = foldlevelExpr;
2236217e1b83SBram Moolenaar // start one line back, because a "<1" may indicate the end of a
2237217e1b83SBram Moolenaar // fold in the topline
2238071d4279SBram Moolenaar if (top > 1)
2239071d4279SBram Moolenaar --fline.lnum;
2240071d4279SBram Moolenaar }
2241071d4279SBram Moolenaar else if (foldmethodIsSyntax(wp))
2242071d4279SBram Moolenaar getlevel = foldlevelSyntax;
2243071d4279SBram Moolenaar #ifdef FEAT_DIFF
2244071d4279SBram Moolenaar else if (foldmethodIsDiff(wp))
2245071d4279SBram Moolenaar getlevel = foldlevelDiff;
2246071d4279SBram Moolenaar #endif
2247071d4279SBram Moolenaar else
2248071d4279SBram Moolenaar getlevel = foldlevelIndent;
2249071d4279SBram Moolenaar
2250217e1b83SBram Moolenaar // Backup to a line for which the fold level is defined. Since it's
2251217e1b83SBram Moolenaar // always defined for line one, we will stop there.
2252071d4279SBram Moolenaar fline.lvl = -1;
2253071d4279SBram Moolenaar for ( ; !got_int; --fline.lnum)
2254071d4279SBram Moolenaar {
2255217e1b83SBram Moolenaar // Reset lvl_next each time, because it will be set to a value for
2256217e1b83SBram Moolenaar // the next line, but we search backwards here.
2257071d4279SBram Moolenaar fline.lvl_next = -1;
2258071d4279SBram Moolenaar getlevel(&fline);
2259071d4279SBram Moolenaar if (fline.lvl >= 0)
2260071d4279SBram Moolenaar break;
2261071d4279SBram Moolenaar }
2262071d4279SBram Moolenaar }
2263071d4279SBram Moolenaar
2264ec98647bSBram Moolenaar /*
2265ec98647bSBram Moolenaar * If folding is defined by the syntax, it is possible that a change in
2266ec98647bSBram Moolenaar * one line will cause all sub-folds of the current fold to change (e.g.,
2267ec98647bSBram Moolenaar * closing a C-style comment can cause folds in the subsequent lines to
2268ec98647bSBram Moolenaar * appear). To take that into account we should adjust the value of "bot"
2269ec98647bSBram Moolenaar * to point to the end of the current fold:
2270ec98647bSBram Moolenaar */
2271ec98647bSBram Moolenaar if (foldlevelSyntax == getlevel)
2272ec98647bSBram Moolenaar {
2273ec98647bSBram Moolenaar garray_T *gap = &wp->w_folds;
2274f506c5bbSBram Moolenaar fold_T *fpn = NULL;
2275ec98647bSBram Moolenaar int current_fdl = 0;
2276ec98647bSBram Moolenaar linenr_T fold_start_lnum = 0;
2277ec98647bSBram Moolenaar linenr_T lnum_rel = fline.lnum;
2278ec98647bSBram Moolenaar
2279ec98647bSBram Moolenaar while (current_fdl < fline.lvl)
2280ec98647bSBram Moolenaar {
2281f506c5bbSBram Moolenaar if (!foldFind(gap, lnum_rel, &fpn))
2282ec98647bSBram Moolenaar break;
2283ec98647bSBram Moolenaar ++current_fdl;
2284ec98647bSBram Moolenaar
2285f506c5bbSBram Moolenaar fold_start_lnum += fpn->fd_top;
2286f506c5bbSBram Moolenaar gap = &fpn->fd_nested;
2287f506c5bbSBram Moolenaar lnum_rel -= fpn->fd_top;
2288ec98647bSBram Moolenaar }
2289f506c5bbSBram Moolenaar if (fpn != NULL && current_fdl == fline.lvl)
2290ec98647bSBram Moolenaar {
2291f506c5bbSBram Moolenaar linenr_T fold_end_lnum = fold_start_lnum + fpn->fd_len;
2292ec98647bSBram Moolenaar
2293ec98647bSBram Moolenaar if (fold_end_lnum > bot)
2294ec98647bSBram Moolenaar bot = fold_end_lnum;
2295ec98647bSBram Moolenaar }
2296ec98647bSBram Moolenaar }
2297ec98647bSBram Moolenaar
2298071d4279SBram Moolenaar start = fline.lnum;
2299071d4279SBram Moolenaar end = bot;
2300217e1b83SBram Moolenaar // Do at least one line.
2301071d4279SBram Moolenaar if (start > end && end < wp->w_buffer->b_ml.ml_line_count)
2302071d4279SBram Moolenaar end = start;
2303071d4279SBram Moolenaar while (!got_int)
2304071d4279SBram Moolenaar {
2305217e1b83SBram Moolenaar // Always stop at the end of the file ("end" can be past the end of
2306217e1b83SBram Moolenaar // the file).
2307071d4279SBram Moolenaar if (fline.lnum > wp->w_buffer->b_ml.ml_line_count)
2308071d4279SBram Moolenaar break;
2309071d4279SBram Moolenaar if (fline.lnum > end)
2310071d4279SBram Moolenaar {
2311217e1b83SBram Moolenaar // For "marker", "expr" and "syntax" methods: If a change caused
2312217e1b83SBram Moolenaar // a fold to be removed, we need to continue at least until where
2313217e1b83SBram Moolenaar // it ended.
2314071d4279SBram Moolenaar if (getlevel != foldlevelMarker
2315071d4279SBram Moolenaar && getlevel != foldlevelSyntax
2316071d4279SBram Moolenaar && getlevel != foldlevelExpr)
2317071d4279SBram Moolenaar break;
2318071d4279SBram Moolenaar if ((start <= end
2319071d4279SBram Moolenaar && foldFind(&wp->w_folds, end, &fp)
2320071d4279SBram Moolenaar && fp->fd_top + fp->fd_len - 1 > end)
2321071d4279SBram Moolenaar || (fline.lvl == 0
2322071d4279SBram Moolenaar && foldFind(&wp->w_folds, fline.lnum, &fp)
2323071d4279SBram Moolenaar && fp->fd_top < fline.lnum))
2324071d4279SBram Moolenaar end = fp->fd_top + fp->fd_len - 1;
2325071d4279SBram Moolenaar else if (getlevel == foldlevelSyntax
2326071d4279SBram Moolenaar && foldLevelWin(wp, fline.lnum) != fline.lvl)
2327217e1b83SBram Moolenaar // For "syntax" method: Compare the foldlevel that the syntax
2328217e1b83SBram Moolenaar // tells us to the foldlevel from the existing folds. If they
2329217e1b83SBram Moolenaar // don't match continue updating folds.
2330071d4279SBram Moolenaar end = fline.lnum;
2331071d4279SBram Moolenaar else
2332071d4279SBram Moolenaar break;
2333071d4279SBram Moolenaar }
2334071d4279SBram Moolenaar
2335217e1b83SBram Moolenaar // A level 1 fold starts at a line with foldlevel > 0.
2336071d4279SBram Moolenaar if (fline.lvl > 0)
2337071d4279SBram Moolenaar {
2338071d4279SBram Moolenaar invalid_top = fline.lnum;
2339071d4279SBram Moolenaar invalid_bot = end;
2340071d4279SBram Moolenaar end = foldUpdateIEMSRecurse(&wp->w_folds,
2341071d4279SBram Moolenaar 1, start, &fline, getlevel, end, FD_LEVEL);
2342071d4279SBram Moolenaar start = fline.lnum;
2343071d4279SBram Moolenaar }
2344071d4279SBram Moolenaar else
2345071d4279SBram Moolenaar {
2346071d4279SBram Moolenaar if (fline.lnum == wp->w_buffer->b_ml.ml_line_count)
2347071d4279SBram Moolenaar break;
2348071d4279SBram Moolenaar ++fline.lnum;
2349071d4279SBram Moolenaar fline.lvl = fline.lvl_next;
2350071d4279SBram Moolenaar getlevel(&fline);
2351071d4279SBram Moolenaar }
2352071d4279SBram Moolenaar }
2353071d4279SBram Moolenaar
2354217e1b83SBram Moolenaar // There can't be any folds from start until end now.
2355071d4279SBram Moolenaar foldRemove(&wp->w_folds, start, end);
2356071d4279SBram Moolenaar
2357217e1b83SBram Moolenaar // If some fold changed, need to redraw and position cursor.
2358071d4279SBram Moolenaar if (fold_changed && wp->w_p_fen)
2359e3b3f28dSBram Moolenaar changed_window_setting_win(wp);
2360071d4279SBram Moolenaar
2361217e1b83SBram Moolenaar // If we updated folds past "bot", need to redraw more lines. Don't do
2362217e1b83SBram Moolenaar // this in other situations, the changed lines will be redrawn anyway and
2363217e1b83SBram Moolenaar // this method can cause the whole window to be updated.
2364071d4279SBram Moolenaar if (end != bot)
2365071d4279SBram Moolenaar {
2366071d4279SBram Moolenaar if (wp->w_redraw_top == 0 || wp->w_redraw_top > top)
2367071d4279SBram Moolenaar wp->w_redraw_top = top;
2368071d4279SBram Moolenaar if (wp->w_redraw_bot < end)
2369071d4279SBram Moolenaar wp->w_redraw_bot = end;
2370071d4279SBram Moolenaar }
2371071d4279SBram Moolenaar
2372071d4279SBram Moolenaar invalid_top = (linenr_T)0;
2373071d4279SBram Moolenaar }
2374071d4279SBram Moolenaar
2375217e1b83SBram Moolenaar // foldUpdateIEMSRecurse() {{{2
2376071d4279SBram Moolenaar /*
2377071d4279SBram Moolenaar * Update a fold that starts at "flp->lnum". At this line there is always a
2378071d4279SBram Moolenaar * valid foldlevel, and its level >= "level".
2379071d4279SBram Moolenaar * "flp" is valid for "flp->lnum" when called and it's valid when returning.
2380071d4279SBram Moolenaar * "flp->lnum" is set to the lnum just below the fold, if it ends before
2381071d4279SBram Moolenaar * "bot", it's "bot" plus one if the fold continues and it's bigger when using
2382071d4279SBram Moolenaar * the marker method and a text change made following folds to change.
2383071d4279SBram Moolenaar * When returning, "flp->lnum_save" is the line number that was used to get
2384071d4279SBram Moolenaar * the level when the level at "flp->lnum" is invalid.
2385071d4279SBram Moolenaar * Remove any folds from "startlnum" up to here at this level.
2386071d4279SBram Moolenaar * Recursively update nested folds.
2387071d4279SBram Moolenaar * Below line "bot" there are no changes in the text.
2388071d4279SBram Moolenaar * "flp->lnum", "flp->lnum_save" and "bot" are relative to the start of the
2389071d4279SBram Moolenaar * outer fold.
2390071d4279SBram Moolenaar * "flp->off" is the offset to the real line number in the buffer.
2391071d4279SBram Moolenaar *
2392071d4279SBram Moolenaar * All this would be a lot simpler if all folds in the range would be deleted
2393e37d50a5SBram Moolenaar * and then created again. But we would lose all information about the
2394071d4279SBram Moolenaar * folds, even when making changes that don't affect the folding (e.g. "vj~").
2395071d4279SBram Moolenaar *
2396071d4279SBram Moolenaar * Returns bot, which may have been increased for lines that also need to be
2397071d4279SBram Moolenaar * updated as a result of a detected change in the fold.
2398071d4279SBram Moolenaar */
2399071d4279SBram 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)240066f948e9SBram Moolenaar foldUpdateIEMSRecurse(
240166f948e9SBram Moolenaar garray_T *gap,
240266f948e9SBram Moolenaar int level,
240366f948e9SBram Moolenaar linenr_T startlnum,
240466f948e9SBram Moolenaar fline_T *flp,
240566f948e9SBram Moolenaar void (*getlevel)(fline_T *),
240666f948e9SBram Moolenaar linenr_T bot,
2407217e1b83SBram Moolenaar int topflags) // flags used by containing fold
2408071d4279SBram Moolenaar {
2409071d4279SBram Moolenaar linenr_T ll;
2410071d4279SBram Moolenaar fold_T *fp = NULL;
2411071d4279SBram Moolenaar fold_T *fp2;
2412071d4279SBram Moolenaar int lvl = level;
2413071d4279SBram Moolenaar linenr_T startlnum2 = startlnum;
2414217e1b83SBram Moolenaar linenr_T firstlnum = flp->lnum; // first lnum we got
2415071d4279SBram Moolenaar int i;
2416071d4279SBram Moolenaar int finish = FALSE;
2417071d4279SBram Moolenaar linenr_T linecount = flp->wp->w_buffer->b_ml.ml_line_count - flp->off;
2418071d4279SBram Moolenaar int concat;
2419071d4279SBram Moolenaar
2420071d4279SBram Moolenaar /*
2421071d4279SBram Moolenaar * If using the marker method, the start line is not the start of a fold
2422071d4279SBram Moolenaar * at the level we're dealing with and the level is non-zero, we must use
2423071d4279SBram Moolenaar * the previous fold. But ignore a fold that starts at or below
2424071d4279SBram Moolenaar * startlnum, it must be deleted.
2425071d4279SBram Moolenaar */
2426071d4279SBram Moolenaar if (getlevel == foldlevelMarker && flp->start <= flp->lvl - level
2427071d4279SBram Moolenaar && flp->lvl > 0)
2428071d4279SBram Moolenaar {
2429cde88547SBram Moolenaar (void)foldFind(gap, startlnum - 1, &fp);
2430da697645SBram Moolenaar if (fp != NULL && (fp >= ((fold_T *)gap->ga_data) + gap->ga_len
2431da697645SBram Moolenaar || fp->fd_top >= startlnum))
2432071d4279SBram Moolenaar fp = NULL;
2433071d4279SBram Moolenaar }
2434071d4279SBram Moolenaar
2435071d4279SBram Moolenaar /*
2436071d4279SBram Moolenaar * Loop over all lines in this fold, or until "bot" is hit.
2437071d4279SBram Moolenaar * Handle nested folds inside of this fold.
2438071d4279SBram Moolenaar * "flp->lnum" is the current line. When finding the end of the fold, it
2439071d4279SBram Moolenaar * is just below the end of the fold.
2440071d4279SBram Moolenaar * "*flp" contains the level of the line "flp->lnum" or a following one if
2441071d4279SBram Moolenaar * there are lines with an invalid fold level. "flp->lnum_save" is the
2442071d4279SBram Moolenaar * line number that was used to get the fold level (below "flp->lnum" when
2443071d4279SBram Moolenaar * it has an invalid fold level). When called the fold level is always
2444071d4279SBram Moolenaar * valid, thus "flp->lnum_save" is equal to "flp->lnum".
2445071d4279SBram Moolenaar */
2446071d4279SBram Moolenaar flp->lnum_save = flp->lnum;
2447071d4279SBram Moolenaar while (!got_int)
2448071d4279SBram Moolenaar {
2449217e1b83SBram Moolenaar // Updating folds can be slow, check for CTRL-C.
2450071d4279SBram Moolenaar line_breakcheck();
2451071d4279SBram Moolenaar
2452217e1b83SBram Moolenaar // Set "lvl" to the level of line "flp->lnum". When flp->start is set
2453217e1b83SBram Moolenaar // and after the first line of the fold, set the level to zero to
2454217e1b83SBram Moolenaar // force the fold to end. Do the same when had_end is set: Previous
2455217e1b83SBram Moolenaar // line was marked as end of a fold.
2456071d4279SBram Moolenaar lvl = flp->lvl;
2457071d4279SBram Moolenaar if (lvl > MAX_LEVEL)
2458071d4279SBram Moolenaar lvl = MAX_LEVEL;
2459071d4279SBram Moolenaar if (flp->lnum > firstlnum
2460071d4279SBram Moolenaar && (level > lvl - flp->start || level >= flp->had_end))
2461071d4279SBram Moolenaar lvl = 0;
2462071d4279SBram Moolenaar
2463071d4279SBram Moolenaar if (flp->lnum > bot && !finish && fp != NULL)
2464071d4279SBram Moolenaar {
2465217e1b83SBram Moolenaar // For "marker" and "syntax" methods:
2466217e1b83SBram Moolenaar // - If a change caused a nested fold to be removed, we need to
2467217e1b83SBram Moolenaar // delete it and continue at least until where it ended.
2468217e1b83SBram Moolenaar // - If a change caused a nested fold to be created, or this fold
2469217e1b83SBram Moolenaar // to continue below its original end, need to finish this fold.
2470071d4279SBram Moolenaar if (getlevel != foldlevelMarker
2471071d4279SBram Moolenaar && getlevel != foldlevelExpr
2472071d4279SBram Moolenaar && getlevel != foldlevelSyntax)
2473071d4279SBram Moolenaar break;
2474071d4279SBram Moolenaar i = 0;
2475071d4279SBram Moolenaar fp2 = fp;
2476071d4279SBram Moolenaar if (lvl >= level)
2477071d4279SBram Moolenaar {
2478217e1b83SBram Moolenaar // Compute how deep the folds currently are, if it's deeper
2479217e1b83SBram Moolenaar // than "lvl" then some must be deleted, need to update
2480217e1b83SBram Moolenaar // at least one nested fold.
2481071d4279SBram Moolenaar ll = flp->lnum - fp->fd_top;
2482071d4279SBram Moolenaar while (foldFind(&fp2->fd_nested, ll, &fp2))
2483071d4279SBram Moolenaar {
2484071d4279SBram Moolenaar ++i;
2485071d4279SBram Moolenaar ll -= fp2->fd_top;
2486071d4279SBram Moolenaar }
2487071d4279SBram Moolenaar }
2488071d4279SBram Moolenaar if (lvl < level + i)
2489071d4279SBram Moolenaar {
2490cde88547SBram Moolenaar (void)foldFind(&fp->fd_nested, flp->lnum - fp->fd_top, &fp2);
2491071d4279SBram Moolenaar if (fp2 != NULL)
2492071d4279SBram Moolenaar bot = fp2->fd_top + fp2->fd_len - 1 + fp->fd_top;
2493071d4279SBram Moolenaar }
2494071d4279SBram Moolenaar else if (fp->fd_top + fp->fd_len <= flp->lnum && lvl >= level)
2495071d4279SBram Moolenaar finish = TRUE;
2496071d4279SBram Moolenaar else
2497071d4279SBram Moolenaar break;
2498071d4279SBram Moolenaar }
2499071d4279SBram Moolenaar
2500217e1b83SBram Moolenaar // At the start of the first nested fold and at the end of the current
2501217e1b83SBram Moolenaar // fold: check if existing folds at this level, before the current
2502217e1b83SBram Moolenaar // one, need to be deleted or truncated.
2503071d4279SBram Moolenaar if (fp == NULL
2504071d4279SBram Moolenaar && (lvl != level
2505071d4279SBram Moolenaar || flp->lnum_save >= bot
2506071d4279SBram Moolenaar || flp->start != 0
2507071d4279SBram Moolenaar || flp->had_end <= MAX_LEVEL
2508071d4279SBram Moolenaar || flp->lnum == linecount))
2509071d4279SBram Moolenaar {
2510071d4279SBram Moolenaar /*
2511071d4279SBram Moolenaar * Remove or update folds that have lines between startlnum and
2512071d4279SBram Moolenaar * firstlnum.
2513071d4279SBram Moolenaar */
2514071d4279SBram Moolenaar while (!got_int)
2515071d4279SBram Moolenaar {
2516217e1b83SBram Moolenaar // set concat to 1 if it's allowed to concatenated this fold
2517217e1b83SBram Moolenaar // with a previous one that touches it.
2518071d4279SBram Moolenaar if (flp->start != 0 || flp->had_end <= MAX_LEVEL)
2519071d4279SBram Moolenaar concat = 0;
2520071d4279SBram Moolenaar else
2521071d4279SBram Moolenaar concat = 1;
2522071d4279SBram Moolenaar
2523217e1b83SBram Moolenaar // Find an existing fold to re-use. Preferably one that
2524217e1b83SBram Moolenaar // includes startlnum, otherwise one that ends just before
2525217e1b83SBram Moolenaar // startlnum or starts after it.
252664f37d30SBram Moolenaar if (gap->ga_len > 0 && (foldFind(gap, startlnum, &fp)
2527071d4279SBram Moolenaar || (fp < ((fold_T *)gap->ga_data) + gap->ga_len
2528071d4279SBram Moolenaar && fp->fd_top <= firstlnum)
2529071d4279SBram Moolenaar || foldFind(gap, firstlnum - concat, &fp)
2530071d4279SBram Moolenaar || (fp < ((fold_T *)gap->ga_data) + gap->ga_len
2531071d4279SBram Moolenaar && ((lvl < level && fp->fd_top < flp->lnum)
2532071d4279SBram Moolenaar || (lvl >= level
253364f37d30SBram Moolenaar && fp->fd_top <= flp->lnum_save)))))
2534071d4279SBram Moolenaar {
2535071d4279SBram Moolenaar if (fp->fd_top + fp->fd_len + concat > firstlnum)
2536071d4279SBram Moolenaar {
2537217e1b83SBram Moolenaar // Use existing fold for the new fold. If it starts
2538217e1b83SBram Moolenaar // before where we started looking, extend it. If it
2539217e1b83SBram Moolenaar // starts at another line, update nested folds to keep
2540217e1b83SBram Moolenaar // their position, compensating for the new fd_top.
2541eadbc2b4SBram Moolenaar if (fp->fd_top == firstlnum)
2542eadbc2b4SBram Moolenaar {
2543217e1b83SBram Moolenaar // have found a fold beginning where we want
2544eadbc2b4SBram Moolenaar }
2545eadbc2b4SBram Moolenaar else if (fp->fd_top >= startlnum)
2546071d4279SBram Moolenaar {
2547071d4279SBram Moolenaar if (fp->fd_top > firstlnum)
2548217e1b83SBram Moolenaar // like lines are inserted
2549071d4279SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested,
2550071d4279SBram Moolenaar (linenr_T)0, (linenr_T)MAXLNUM,
2551071d4279SBram Moolenaar (long)(fp->fd_top - firstlnum), 0L);
2552071d4279SBram Moolenaar else
2553217e1b83SBram Moolenaar // like lines are deleted
2554071d4279SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested,
2555071d4279SBram Moolenaar (linenr_T)0,
2556071d4279SBram Moolenaar (long)(firstlnum - fp->fd_top - 1),
2557071d4279SBram Moolenaar (linenr_T)MAXLNUM,
2558071d4279SBram Moolenaar (long)(fp->fd_top - firstlnum));
2559071d4279SBram Moolenaar fp->fd_len += fp->fd_top - firstlnum;
2560071d4279SBram Moolenaar fp->fd_top = firstlnum;
2561071d4279SBram Moolenaar fold_changed = TRUE;
2562071d4279SBram Moolenaar }
2563eadbc2b4SBram Moolenaar else if ((flp->start != 0 && lvl == level)
2564eadbc2b4SBram Moolenaar || firstlnum != startlnum)
2565071d4279SBram Moolenaar {
2566eadbc2b4SBram Moolenaar linenr_T breakstart;
2567eadbc2b4SBram Moolenaar linenr_T breakend;
2568eadbc2b4SBram Moolenaar
2569eadbc2b4SBram Moolenaar /*
2570eadbc2b4SBram Moolenaar * Before there was a fold spanning from above
2571eadbc2b4SBram Moolenaar * startlnum to below firstlnum. This fold is valid
2572eadbc2b4SBram Moolenaar * above startlnum (because we are not updating
2573eadbc2b4SBram Moolenaar * that range), but there should now be a break in
2574eadbc2b4SBram Moolenaar * it.
2575eadbc2b4SBram Moolenaar * If the break is because we are now forced to
2576eadbc2b4SBram Moolenaar * start a new fold at the level "level" at line
2577eadbc2b4SBram Moolenaar * fline->lnum, then we need to split the fold at
2578eadbc2b4SBram Moolenaar * fline->lnum.
2579eadbc2b4SBram Moolenaar * If the break is because the range
2580eadbc2b4SBram Moolenaar * [startlnum, firstlnum) is now at a lower indent
2581eadbc2b4SBram Moolenaar * than "level", we need to split the fold in this
2582eadbc2b4SBram Moolenaar * range.
2583eadbc2b4SBram Moolenaar * Any splits have to be done recursively.
2584eadbc2b4SBram Moolenaar */
2585eadbc2b4SBram Moolenaar if (firstlnum != startlnum)
2586eadbc2b4SBram Moolenaar {
2587eadbc2b4SBram Moolenaar breakstart = startlnum;
2588eadbc2b4SBram Moolenaar breakend = firstlnum;
2589eadbc2b4SBram Moolenaar }
2590eadbc2b4SBram Moolenaar else
2591eadbc2b4SBram Moolenaar {
2592eadbc2b4SBram Moolenaar breakstart = flp->lnum;
2593eadbc2b4SBram Moolenaar breakend = flp->lnum;
2594eadbc2b4SBram Moolenaar }
2595eadbc2b4SBram Moolenaar foldRemove(&fp->fd_nested, breakstart - fp->fd_top,
2596eadbc2b4SBram Moolenaar breakend - fp->fd_top);
2597071d4279SBram Moolenaar i = (int)(fp - (fold_T *)gap->ga_data);
2598eadbc2b4SBram Moolenaar foldSplit(gap, i, breakstart, breakend - 1);
2599071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i + 1;
2600eadbc2b4SBram Moolenaar
2601217e1b83SBram Moolenaar // If using the "marker" or "syntax" method, we
2602217e1b83SBram Moolenaar // need to continue until the end of the fold is
2603217e1b83SBram Moolenaar // found.
2604071d4279SBram Moolenaar if (getlevel == foldlevelMarker
2605071d4279SBram Moolenaar || getlevel == foldlevelExpr
2606071d4279SBram Moolenaar || getlevel == foldlevelSyntax)
2607071d4279SBram Moolenaar finish = TRUE;
2608071d4279SBram Moolenaar }
2609eadbc2b4SBram Moolenaar
2610eadbc2b4SBram Moolenaar if (fp->fd_top == startlnum && concat)
2611eadbc2b4SBram Moolenaar {
2612eadbc2b4SBram Moolenaar i = (int)(fp - (fold_T *)gap->ga_data);
2613eadbc2b4SBram Moolenaar if (i != 0)
2614eadbc2b4SBram Moolenaar {
2615eadbc2b4SBram Moolenaar fp2 = fp - 1;
2616eadbc2b4SBram Moolenaar if (fp2->fd_top + fp2->fd_len == fp->fd_top)
2617eadbc2b4SBram Moolenaar {
2618eadbc2b4SBram Moolenaar foldMerge(fp2, gap, fp);
2619eadbc2b4SBram Moolenaar fp = fp2;
2620eadbc2b4SBram Moolenaar }
2621eadbc2b4SBram Moolenaar }
2622eadbc2b4SBram Moolenaar }
2623071d4279SBram Moolenaar break;
2624071d4279SBram Moolenaar }
2625071d4279SBram Moolenaar if (fp->fd_top >= startlnum)
2626071d4279SBram Moolenaar {
2627217e1b83SBram Moolenaar // A fold that starts at or after startlnum and stops
2628217e1b83SBram Moolenaar // before the new fold must be deleted. Continue
2629217e1b83SBram Moolenaar // looking for the next one.
2630071d4279SBram Moolenaar deleteFoldEntry(gap,
2631071d4279SBram Moolenaar (int)(fp - (fold_T *)gap->ga_data), TRUE);
2632071d4279SBram Moolenaar }
2633071d4279SBram Moolenaar else
2634071d4279SBram Moolenaar {
2635217e1b83SBram Moolenaar // A fold has some lines above startlnum, truncate it
2636217e1b83SBram Moolenaar // to stop just above startlnum.
2637071d4279SBram Moolenaar fp->fd_len = startlnum - fp->fd_top;
2638071d4279SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested,
2639071d4279SBram Moolenaar (linenr_T)fp->fd_len, (linenr_T)MAXLNUM,
2640071d4279SBram Moolenaar (linenr_T)MAXLNUM, 0L);
2641071d4279SBram Moolenaar fold_changed = TRUE;
2642071d4279SBram Moolenaar }
2643071d4279SBram Moolenaar }
2644071d4279SBram Moolenaar else
2645071d4279SBram Moolenaar {
2646217e1b83SBram Moolenaar // Insert new fold. Careful: ga_data may be NULL and it
2647217e1b83SBram Moolenaar // may change!
264864f37d30SBram Moolenaar if (gap->ga_len == 0)
264964f37d30SBram Moolenaar i = 0;
265064f37d30SBram Moolenaar else
2651071d4279SBram Moolenaar i = (int)(fp - (fold_T *)gap->ga_data);
2652071d4279SBram Moolenaar if (foldInsert(gap, i) != OK)
2653071d4279SBram Moolenaar return bot;
2654071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i;
2655217e1b83SBram Moolenaar // The new fold continues until bot, unless we find the
2656217e1b83SBram Moolenaar // end earlier.
2657071d4279SBram Moolenaar fp->fd_top = firstlnum;
2658071d4279SBram Moolenaar fp->fd_len = bot - firstlnum + 1;
2659217e1b83SBram Moolenaar // When the containing fold is open, the new fold is open.
2660217e1b83SBram Moolenaar // The new fold is closed if the fold above it is closed.
2661217e1b83SBram Moolenaar // The first fold depends on the containing fold.
2662071d4279SBram Moolenaar if (topflags == FD_OPEN)
2663071d4279SBram Moolenaar {
2664071d4279SBram Moolenaar flp->wp->w_fold_manual = TRUE;
2665071d4279SBram Moolenaar fp->fd_flags = FD_OPEN;
2666071d4279SBram Moolenaar }
2667071d4279SBram Moolenaar else if (i <= 0)
2668071d4279SBram Moolenaar {
2669071d4279SBram Moolenaar fp->fd_flags = topflags;
2670071d4279SBram Moolenaar if (topflags != FD_LEVEL)
2671071d4279SBram Moolenaar flp->wp->w_fold_manual = TRUE;
2672071d4279SBram Moolenaar }
2673071d4279SBram Moolenaar else
2674071d4279SBram Moolenaar fp->fd_flags = (fp - 1)->fd_flags;
2675071d4279SBram Moolenaar fp->fd_small = MAYBE;
2676217e1b83SBram Moolenaar // If using the "marker", "expr" or "syntax" method, we
2677217e1b83SBram Moolenaar // need to continue until the end of the fold is found.
2678071d4279SBram Moolenaar if (getlevel == foldlevelMarker
2679071d4279SBram Moolenaar || getlevel == foldlevelExpr
2680071d4279SBram Moolenaar || getlevel == foldlevelSyntax)
2681071d4279SBram Moolenaar finish = TRUE;
2682071d4279SBram Moolenaar fold_changed = TRUE;
2683071d4279SBram Moolenaar break;
2684071d4279SBram Moolenaar }
2685071d4279SBram Moolenaar }
2686071d4279SBram Moolenaar }
2687071d4279SBram Moolenaar
2688071d4279SBram Moolenaar if (lvl < level || flp->lnum > linecount)
2689071d4279SBram Moolenaar {
2690071d4279SBram Moolenaar /*
2691071d4279SBram Moolenaar * Found a line with a lower foldlevel, this fold ends just above
2692071d4279SBram Moolenaar * "flp->lnum".
2693071d4279SBram Moolenaar */
2694071d4279SBram Moolenaar break;
2695071d4279SBram Moolenaar }
2696071d4279SBram Moolenaar
2697071d4279SBram Moolenaar /*
2698071d4279SBram Moolenaar * The fold includes the line "flp->lnum" and "flp->lnum_save".
2699eb3593b3SBram Moolenaar * Check "fp" for safety.
2700071d4279SBram Moolenaar */
2701eb3593b3SBram Moolenaar if (lvl > level && fp != NULL)
2702071d4279SBram Moolenaar {
2703071d4279SBram Moolenaar /*
2704071d4279SBram Moolenaar * There is a nested fold, handle it recursively.
2705071d4279SBram Moolenaar */
2706217e1b83SBram Moolenaar // At least do one line (can happen when finish is TRUE).
2707071d4279SBram Moolenaar if (bot < flp->lnum)
2708071d4279SBram Moolenaar bot = flp->lnum;
2709071d4279SBram Moolenaar
2710217e1b83SBram Moolenaar // Line numbers in the nested fold are relative to the start of
2711217e1b83SBram Moolenaar // this fold.
2712071d4279SBram Moolenaar flp->lnum = flp->lnum_save - fp->fd_top;
2713071d4279SBram Moolenaar flp->off += fp->fd_top;
2714071d4279SBram Moolenaar i = (int)(fp - (fold_T *)gap->ga_data);
2715071d4279SBram Moolenaar bot = foldUpdateIEMSRecurse(&fp->fd_nested, level + 1,
2716071d4279SBram Moolenaar startlnum2 - fp->fd_top, flp, getlevel,
2717071d4279SBram Moolenaar bot - fp->fd_top, fp->fd_flags);
2718071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i;
2719071d4279SBram Moolenaar flp->lnum += fp->fd_top;
2720071d4279SBram Moolenaar flp->lnum_save += fp->fd_top;
2721071d4279SBram Moolenaar flp->off -= fp->fd_top;
2722071d4279SBram Moolenaar bot += fp->fd_top;
2723071d4279SBram Moolenaar startlnum2 = flp->lnum;
2724071d4279SBram Moolenaar
2725217e1b83SBram Moolenaar // This fold may end at the same line, don't incr. flp->lnum.
2726071d4279SBram Moolenaar }
2727071d4279SBram Moolenaar else
2728071d4279SBram Moolenaar {
2729071d4279SBram Moolenaar /*
2730071d4279SBram Moolenaar * Get the level of the next line, then continue the loop to check
2731071d4279SBram Moolenaar * if it ends there.
2732071d4279SBram Moolenaar * Skip over undefined lines, to find the foldlevel after it.
2733071d4279SBram Moolenaar * For the last line in the file the foldlevel is always valid.
2734071d4279SBram Moolenaar */
2735071d4279SBram Moolenaar flp->lnum = flp->lnum_save;
2736071d4279SBram Moolenaar ll = flp->lnum + 1;
2737071d4279SBram Moolenaar while (!got_int)
2738071d4279SBram Moolenaar {
2739217e1b83SBram Moolenaar // Make the previous level available to foldlevel().
2740071d4279SBram Moolenaar prev_lnum = flp->lnum;
2741071d4279SBram Moolenaar prev_lnum_lvl = flp->lvl;
2742071d4279SBram Moolenaar
2743071d4279SBram Moolenaar if (++flp->lnum > linecount)
2744071d4279SBram Moolenaar break;
2745071d4279SBram Moolenaar flp->lvl = flp->lvl_next;
2746071d4279SBram Moolenaar getlevel(flp);
2747071d4279SBram Moolenaar if (flp->lvl >= 0 || flp->had_end <= MAX_LEVEL)
2748071d4279SBram Moolenaar break;
2749071d4279SBram Moolenaar }
2750071d4279SBram Moolenaar prev_lnum = 0;
2751071d4279SBram Moolenaar if (flp->lnum > linecount)
2752071d4279SBram Moolenaar break;
2753071d4279SBram Moolenaar
2754217e1b83SBram Moolenaar // leave flp->lnum_save to lnum of the line that was used to get
2755217e1b83SBram Moolenaar // the level, flp->lnum to the lnum of the next line.
2756071d4279SBram Moolenaar flp->lnum_save = flp->lnum;
2757071d4279SBram Moolenaar flp->lnum = ll;
2758071d4279SBram Moolenaar }
2759071d4279SBram Moolenaar }
2760071d4279SBram Moolenaar
2761217e1b83SBram Moolenaar if (fp == NULL) // only happens when got_int is set
2762071d4279SBram Moolenaar return bot;
2763071d4279SBram Moolenaar
2764071d4279SBram Moolenaar /*
2765071d4279SBram Moolenaar * Get here when:
2766071d4279SBram Moolenaar * lvl < level: the folds ends just above "flp->lnum"
2767071d4279SBram Moolenaar * lvl >= level: fold continues below "bot"
2768071d4279SBram Moolenaar */
2769071d4279SBram Moolenaar
27709a4a8c4dSBram Moolenaar // Current fold at least extends until lnum.
2771071d4279SBram Moolenaar if (fp->fd_len < flp->lnum - fp->fd_top)
2772071d4279SBram Moolenaar {
2773071d4279SBram Moolenaar fp->fd_len = flp->lnum - fp->fd_top;
2774ef6fc097SBram Moolenaar fp->fd_small = MAYBE;
2775071d4279SBram Moolenaar fold_changed = TRUE;
2776071d4279SBram Moolenaar }
27779a4a8c4dSBram Moolenaar else if (fp->fd_top + fp->fd_len > linecount)
27789a4a8c4dSBram Moolenaar // running into the end of the buffer (deleted last line)
27799a4a8c4dSBram Moolenaar fp->fd_len = linecount - fp->fd_top + 1;
2780071d4279SBram Moolenaar
27819a4a8c4dSBram Moolenaar // Delete contained folds from the end of the last one found until where
27829a4a8c4dSBram Moolenaar // we stopped looking.
2783071d4279SBram Moolenaar foldRemove(&fp->fd_nested, startlnum2 - fp->fd_top,
2784071d4279SBram Moolenaar flp->lnum - 1 - fp->fd_top);
2785071d4279SBram Moolenaar
2786071d4279SBram Moolenaar if (lvl < level)
2787071d4279SBram Moolenaar {
2788217e1b83SBram Moolenaar // End of fold found, update the length when it got shorter.
2789071d4279SBram Moolenaar if (fp->fd_len != flp->lnum - fp->fd_top)
2790071d4279SBram Moolenaar {
27919d20ce69SBram Moolenaar if (fp->fd_top + fp->fd_len - 1 > bot)
2792071d4279SBram Moolenaar {
2793217e1b83SBram Moolenaar // fold continued below bot
2794071d4279SBram Moolenaar if (getlevel == foldlevelMarker
2795071d4279SBram Moolenaar || getlevel == foldlevelExpr
2796071d4279SBram Moolenaar || getlevel == foldlevelSyntax)
2797071d4279SBram Moolenaar {
2798217e1b83SBram Moolenaar // marker method: truncate the fold and make sure the
2799217e1b83SBram Moolenaar // previously included lines are processed again
2800071d4279SBram Moolenaar bot = fp->fd_top + fp->fd_len - 1;
2801071d4279SBram Moolenaar fp->fd_len = flp->lnum - fp->fd_top;
2802071d4279SBram Moolenaar }
2803071d4279SBram Moolenaar else
2804071d4279SBram Moolenaar {
2805217e1b83SBram Moolenaar // indent or expr method: split fold to create a new one
2806217e1b83SBram Moolenaar // below bot
2807071d4279SBram Moolenaar i = (int)(fp - (fold_T *)gap->ga_data);
2808071d4279SBram Moolenaar foldSplit(gap, i, flp->lnum, bot);
2809071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i;
2810071d4279SBram Moolenaar }
2811071d4279SBram Moolenaar }
2812071d4279SBram Moolenaar else
2813071d4279SBram Moolenaar fp->fd_len = flp->lnum - fp->fd_top;
2814071d4279SBram Moolenaar fold_changed = TRUE;
2815071d4279SBram Moolenaar }
2816071d4279SBram Moolenaar }
2817071d4279SBram Moolenaar
2818217e1b83SBram Moolenaar // delete following folds that end before the current line
2819071d4279SBram Moolenaar for (;;)
2820071d4279SBram Moolenaar {
2821071d4279SBram Moolenaar fp2 = fp + 1;
2822071d4279SBram Moolenaar if (fp2 >= (fold_T *)gap->ga_data + gap->ga_len
2823071d4279SBram Moolenaar || fp2->fd_top > flp->lnum)
2824071d4279SBram Moolenaar break;
2825071d4279SBram Moolenaar if (fp2->fd_top + fp2->fd_len > flp->lnum)
2826071d4279SBram Moolenaar {
2827071d4279SBram Moolenaar if (fp2->fd_top < flp->lnum)
2828071d4279SBram Moolenaar {
2829217e1b83SBram Moolenaar // Make fold that includes lnum start at lnum.
2830071d4279SBram Moolenaar foldMarkAdjustRecurse(&fp2->fd_nested,
2831071d4279SBram Moolenaar (linenr_T)0, (long)(flp->lnum - fp2->fd_top - 1),
2832071d4279SBram Moolenaar (linenr_T)MAXLNUM, (long)(fp2->fd_top - flp->lnum));
2833071d4279SBram Moolenaar fp2->fd_len -= flp->lnum - fp2->fd_top;
2834071d4279SBram Moolenaar fp2->fd_top = flp->lnum;
2835071d4279SBram Moolenaar fold_changed = TRUE;
2836071d4279SBram Moolenaar }
2837071d4279SBram Moolenaar
2838071d4279SBram Moolenaar if (lvl >= level)
2839071d4279SBram Moolenaar {
2840217e1b83SBram Moolenaar // merge new fold with existing fold that follows
2841071d4279SBram Moolenaar foldMerge(fp, gap, fp2);
2842071d4279SBram Moolenaar }
2843071d4279SBram Moolenaar break;
2844071d4279SBram Moolenaar }
2845071d4279SBram Moolenaar fold_changed = TRUE;
2846071d4279SBram Moolenaar deleteFoldEntry(gap, (int)(fp2 - (fold_T *)gap->ga_data), TRUE);
2847071d4279SBram Moolenaar }
2848071d4279SBram Moolenaar
2849217e1b83SBram Moolenaar // Need to redraw the lines we inspected, which might be further down than
2850217e1b83SBram Moolenaar // was asked for.
2851071d4279SBram Moolenaar if (bot < flp->lnum - 1)
2852071d4279SBram Moolenaar bot = flp->lnum - 1;
2853071d4279SBram Moolenaar
2854071d4279SBram Moolenaar return bot;
2855071d4279SBram Moolenaar }
2856071d4279SBram Moolenaar
2857217e1b83SBram Moolenaar // foldInsert() {{{2
2858071d4279SBram Moolenaar /*
2859071d4279SBram Moolenaar * Insert a new fold in "gap" at position "i".
2860071d4279SBram Moolenaar * Returns OK for success, FAIL for failure.
2861071d4279SBram Moolenaar */
2862071d4279SBram Moolenaar static int
foldInsert(garray_T * gap,int i)286366f948e9SBram Moolenaar foldInsert(garray_T *gap, int i)
2864071d4279SBram Moolenaar {
2865071d4279SBram Moolenaar fold_T *fp;
2866071d4279SBram Moolenaar
2867071d4279SBram Moolenaar if (ga_grow(gap, 1) != OK)
2868071d4279SBram Moolenaar return FAIL;
2869071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i;
287064f37d30SBram Moolenaar if (gap->ga_len > 0 && i < gap->ga_len)
2871071d4279SBram Moolenaar mch_memmove(fp + 1, fp, sizeof(fold_T) * (gap->ga_len - i));
2872071d4279SBram Moolenaar ++gap->ga_len;
2873071d4279SBram Moolenaar ga_init2(&fp->fd_nested, (int)sizeof(fold_T), 10);
2874071d4279SBram Moolenaar return OK;
2875071d4279SBram Moolenaar }
2876071d4279SBram Moolenaar
2877217e1b83SBram Moolenaar // foldSplit() {{{2
2878071d4279SBram Moolenaar /*
2879071d4279SBram Moolenaar * Split the "i"th fold in "gap", which starts before "top" and ends below
2880071d4279SBram Moolenaar * "bot" in two pieces, one ending above "top" and the other starting below
2881071d4279SBram Moolenaar * "bot".
2882071d4279SBram Moolenaar * The caller must first have taken care of any nested folds from "top" to
2883071d4279SBram Moolenaar * "bot"!
2884071d4279SBram Moolenaar */
2885071d4279SBram Moolenaar static void
foldSplit(garray_T * gap,int i,linenr_T top,linenr_T bot)288666f948e9SBram Moolenaar foldSplit(
288766f948e9SBram Moolenaar garray_T *gap,
288866f948e9SBram Moolenaar int i,
288966f948e9SBram Moolenaar linenr_T top,
289066f948e9SBram Moolenaar linenr_T bot)
2891071d4279SBram Moolenaar {
2892071d4279SBram Moolenaar fold_T *fp;
2893071d4279SBram Moolenaar fold_T *fp2;
2894071d4279SBram Moolenaar garray_T *gap1;
2895071d4279SBram Moolenaar garray_T *gap2;
2896071d4279SBram Moolenaar int idx;
2897071d4279SBram Moolenaar int len;
2898071d4279SBram Moolenaar
2899217e1b83SBram Moolenaar // The fold continues below bot, need to split it.
2900071d4279SBram Moolenaar if (foldInsert(gap, i + 1) == FAIL)
2901071d4279SBram Moolenaar return;
2902071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data + i;
2903071d4279SBram Moolenaar fp[1].fd_top = bot + 1;
2904071d4279SBram Moolenaar fp[1].fd_len = fp->fd_len - (fp[1].fd_top - fp->fd_top);
2905071d4279SBram Moolenaar fp[1].fd_flags = fp->fd_flags;
2906eda6eb9fSBram Moolenaar fp[1].fd_small = MAYBE;
2907eda6eb9fSBram Moolenaar fp->fd_small = MAYBE;
2908071d4279SBram Moolenaar
2909217e1b83SBram Moolenaar // Move nested folds below bot to new fold. There can't be
2910217e1b83SBram Moolenaar // any between top and bot, they have been removed by the caller.
2911071d4279SBram Moolenaar gap1 = &fp->fd_nested;
2912071d4279SBram Moolenaar gap2 = &fp[1].fd_nested;
29139c2b0663SBram Moolenaar if (foldFind(gap1, bot + 1 - fp->fd_top, &fp2))
29149c2b0663SBram Moolenaar {
2915071d4279SBram Moolenaar len = (int)((fold_T *)gap1->ga_data + gap1->ga_len - fp2);
2916071d4279SBram Moolenaar if (len > 0 && ga_grow(gap2, len) == OK)
2917071d4279SBram Moolenaar {
2918071d4279SBram Moolenaar for (idx = 0; idx < len; ++idx)
2919071d4279SBram Moolenaar {
2920071d4279SBram Moolenaar ((fold_T *)gap2->ga_data)[idx] = fp2[idx];
2921071d4279SBram Moolenaar ((fold_T *)gap2->ga_data)[idx].fd_top
2922071d4279SBram Moolenaar -= fp[1].fd_top - fp->fd_top;
2923071d4279SBram Moolenaar }
2924071d4279SBram Moolenaar gap2->ga_len = len;
2925071d4279SBram Moolenaar gap1->ga_len -= len;
2926071d4279SBram Moolenaar }
29279c2b0663SBram Moolenaar }
2928071d4279SBram Moolenaar fp->fd_len = top - fp->fd_top;
2929071d4279SBram Moolenaar fold_changed = TRUE;
2930071d4279SBram Moolenaar }
2931071d4279SBram Moolenaar
2932217e1b83SBram Moolenaar // foldRemove() {{{2
2933071d4279SBram Moolenaar /*
2934071d4279SBram Moolenaar * Remove folds within the range "top" to and including "bot".
2935071d4279SBram Moolenaar * Check for these situations:
2936071d4279SBram Moolenaar * 1 2 3
2937071d4279SBram Moolenaar * 1 2 3
2938071d4279SBram Moolenaar * top 2 3 4 5
2939071d4279SBram Moolenaar * 2 3 4 5
2940071d4279SBram Moolenaar * bot 2 3 4 5
2941071d4279SBram Moolenaar * 3 5 6
2942071d4279SBram Moolenaar * 3 5 6
2943071d4279SBram Moolenaar *
2944071d4279SBram Moolenaar * 1: not changed
294525394022SBram Moolenaar * 2: truncate to stop above "top"
2946071d4279SBram Moolenaar * 3: split in two parts, one stops above "top", other starts below "bot".
2947071d4279SBram Moolenaar * 4: deleted
2948071d4279SBram Moolenaar * 5: made to start below "bot".
2949071d4279SBram Moolenaar * 6: not changed
2950071d4279SBram Moolenaar */
2951071d4279SBram Moolenaar static void
foldRemove(garray_T * gap,linenr_T top,linenr_T bot)295266f948e9SBram Moolenaar foldRemove(garray_T *gap, linenr_T top, linenr_T bot)
2953071d4279SBram Moolenaar {
2954071d4279SBram Moolenaar fold_T *fp = NULL;
2955071d4279SBram Moolenaar
2956071d4279SBram Moolenaar if (bot < top)
2957217e1b83SBram Moolenaar return; // nothing to do
2958071d4279SBram Moolenaar
295964f37d30SBram Moolenaar while (gap->ga_len > 0)
2960071d4279SBram Moolenaar {
2961217e1b83SBram Moolenaar // Find fold that includes top or a following one.
2962071d4279SBram Moolenaar if (foldFind(gap, top, &fp) && fp->fd_top < top)
2963071d4279SBram Moolenaar {
2964217e1b83SBram Moolenaar // 2: or 3: need to delete nested folds
2965071d4279SBram Moolenaar foldRemove(&fp->fd_nested, top - fp->fd_top, bot - fp->fd_top);
2966b11c826dSBram Moolenaar if (fp->fd_top + fp->fd_len - 1 > bot)
2967071d4279SBram Moolenaar {
2968217e1b83SBram Moolenaar // 3: need to split it.
2969071d4279SBram Moolenaar foldSplit(gap, (int)(fp - (fold_T *)gap->ga_data), top, bot);
2970071d4279SBram Moolenaar }
2971071d4279SBram Moolenaar else
2972071d4279SBram Moolenaar {
2973217e1b83SBram Moolenaar // 2: truncate fold at "top".
2974071d4279SBram Moolenaar fp->fd_len = top - fp->fd_top;
2975071d4279SBram Moolenaar }
2976071d4279SBram Moolenaar fold_changed = TRUE;
2977071d4279SBram Moolenaar continue;
2978071d4279SBram Moolenaar }
2979071d4279SBram Moolenaar if (fp >= (fold_T *)(gap->ga_data) + gap->ga_len
2980071d4279SBram Moolenaar || fp->fd_top > bot)
2981071d4279SBram Moolenaar {
2982217e1b83SBram Moolenaar // 6: Found a fold below bot, can stop looking.
2983071d4279SBram Moolenaar break;
2984071d4279SBram Moolenaar }
2985071d4279SBram Moolenaar if (fp->fd_top >= top)
2986071d4279SBram Moolenaar {
2987217e1b83SBram Moolenaar // Found an entry below top.
2988071d4279SBram Moolenaar fold_changed = TRUE;
2989293ee4d4SBram Moolenaar if (fp->fd_top + fp->fd_len - 1 > bot)
2990071d4279SBram Moolenaar {
2991217e1b83SBram Moolenaar // 5: Make fold that includes bot start below bot.
2992071d4279SBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested,
2993071d4279SBram Moolenaar (linenr_T)0, (long)(bot - fp->fd_top),
2994071d4279SBram Moolenaar (linenr_T)MAXLNUM, (long)(fp->fd_top - bot - 1));
2995071d4279SBram Moolenaar fp->fd_len -= bot - fp->fd_top + 1;
2996071d4279SBram Moolenaar fp->fd_top = bot + 1;
2997071d4279SBram Moolenaar break;
2998071d4279SBram Moolenaar }
2999071d4279SBram Moolenaar
3000217e1b83SBram Moolenaar // 4: Delete completely contained fold.
3001071d4279SBram Moolenaar deleteFoldEntry(gap, (int)(fp - (fold_T *)gap->ga_data), TRUE);
3002071d4279SBram Moolenaar }
3003071d4279SBram Moolenaar }
3004071d4279SBram Moolenaar }
3005071d4279SBram Moolenaar
3006217e1b83SBram Moolenaar // foldReverseOrder() {{{2
300788d298aeSBram Moolenaar static void
foldReverseOrder(garray_T * gap,linenr_T start_arg,linenr_T end_arg)3008b11c826dSBram Moolenaar foldReverseOrder(garray_T *gap, linenr_T start_arg, linenr_T end_arg)
300988d298aeSBram Moolenaar {
301088d298aeSBram Moolenaar fold_T *left, *right;
301188d298aeSBram Moolenaar fold_T tmp;
3012b11c826dSBram Moolenaar linenr_T start = start_arg;
3013b11c826dSBram Moolenaar linenr_T end = end_arg;
301488d298aeSBram Moolenaar
301588d298aeSBram Moolenaar for (; start < end; start++, end--)
301688d298aeSBram Moolenaar {
301788d298aeSBram Moolenaar left = (fold_T *)gap->ga_data + start;
301888d298aeSBram Moolenaar right = (fold_T *)gap->ga_data + end;
301988d298aeSBram Moolenaar tmp = *left;
302088d298aeSBram Moolenaar *left = *right;
302188d298aeSBram Moolenaar *right = tmp;
302288d298aeSBram Moolenaar }
302388d298aeSBram Moolenaar }
302488d298aeSBram Moolenaar
3025217e1b83SBram Moolenaar // foldMoveRange() {{{2
302688d298aeSBram Moolenaar /*
302788d298aeSBram Moolenaar * Move folds within the inclusive range "line1" to "line2" to after "dest"
302888d298aeSBram Moolenaar * requires "line1" <= "line2" <= "dest"
302988d298aeSBram Moolenaar *
303088d298aeSBram Moolenaar * There are the following situations for the first fold at or below line1 - 1.
303188d298aeSBram Moolenaar * 1 2 3 4
303288d298aeSBram Moolenaar * 1 2 3 4
303388d298aeSBram Moolenaar * line1 2 3 4
303488d298aeSBram Moolenaar * 2 3 4 5 6 7
303588d298aeSBram Moolenaar * line2 3 4 5 6 7
303688d298aeSBram Moolenaar * 3 4 6 7 8 9
303788d298aeSBram Moolenaar * dest 4 7 8 9
303888d298aeSBram Moolenaar * 4 7 8 10
303988d298aeSBram Moolenaar * 4 7 8 10
304088d298aeSBram Moolenaar *
304188d298aeSBram Moolenaar * In the following descriptions, "moved" means moving in the buffer, *and* in
304288d298aeSBram Moolenaar * the fold array.
304388d298aeSBram Moolenaar * Meanwhile, "shifted" just means moving in the buffer.
304488d298aeSBram Moolenaar * 1. not changed
304588d298aeSBram Moolenaar * 2. truncated above line1
304688d298aeSBram Moolenaar * 3. length reduced by line2 - line1, folds starting between the end of 3 and
304788d298aeSBram Moolenaar * dest are truncated and shifted up
304888d298aeSBram Moolenaar * 4. internal folds moved (from [line1, line2] to dest)
304988d298aeSBram Moolenaar * 5. moved to dest.
305088d298aeSBram Moolenaar * 6. truncated below line2 and moved.
305188d298aeSBram Moolenaar * 7. length reduced by line2 - dest, folds starting between line2 and dest are
305288d298aeSBram Moolenaar * removed, top is moved down by move_len.
305388d298aeSBram Moolenaar * 8. truncated below dest and shifted up.
305488d298aeSBram Moolenaar * 9. shifted up
305588d298aeSBram Moolenaar * 10. not changed
305688d298aeSBram Moolenaar */
305788d298aeSBram Moolenaar
305888d298aeSBram Moolenaar static void
truncate_fold(fold_T * fp,linenr_T end)305988d298aeSBram Moolenaar truncate_fold(fold_T *fp, linenr_T end)
306088d298aeSBram Moolenaar {
306140ebc0afSBram Moolenaar end += 1;
306288d298aeSBram Moolenaar foldRemove(&fp->fd_nested, end - fp->fd_top, MAXLNUM);
306340ebc0afSBram Moolenaar fp->fd_len = end - fp->fd_top;
306488d298aeSBram Moolenaar }
306588d298aeSBram Moolenaar
306688d298aeSBram Moolenaar #define fold_end(fp) ((fp)->fd_top + (fp)->fd_len - 1)
306781fcb67fSBram Moolenaar #define valid_fold(fp, gap) ((gap)->ga_len > 0 && (fp) < ((fold_T *)(gap)->ga_data + (gap)->ga_len))
306888d298aeSBram Moolenaar #define fold_index(fp, gap) ((size_t)(fp - ((fold_T *)(gap)->ga_data)))
306988d298aeSBram Moolenaar
30704a5a8dd6SBram Moolenaar void
foldMoveRange(garray_T * gap,linenr_T line1,linenr_T line2,linenr_T dest)30714a5a8dd6SBram Moolenaar foldMoveRange(garray_T *gap, linenr_T line1, linenr_T line2, linenr_T dest)
307288d298aeSBram Moolenaar {
307388d298aeSBram Moolenaar fold_T *fp;
307488d298aeSBram Moolenaar linenr_T range_len = line2 - line1 + 1;
307588d298aeSBram Moolenaar linenr_T move_len = dest - line2;
307688d298aeSBram Moolenaar int at_start = foldFind(gap, line1 - 1, &fp);
307788d298aeSBram Moolenaar size_t move_start = 0, move_end = 0, dest_index = 0;
307888d298aeSBram Moolenaar
307988d298aeSBram Moolenaar if (at_start)
308088d298aeSBram Moolenaar {
308188d298aeSBram Moolenaar if (fold_end(fp) > dest)
308288d298aeSBram Moolenaar {
3083217e1b83SBram Moolenaar // Case 4
3084217e1b83SBram Moolenaar // don't have to change this fold, but have to move nested folds.
308588d298aeSBram Moolenaar foldMoveRange(&fp->fd_nested, line1 - fp->fd_top, line2 -
308688d298aeSBram Moolenaar fp->fd_top, dest - fp->fd_top);
308788d298aeSBram Moolenaar return;
308888d298aeSBram Moolenaar }
308988d298aeSBram Moolenaar else if (fold_end(fp) > line2)
309088d298aeSBram Moolenaar {
3091217e1b83SBram Moolenaar // Case 3
3092217e1b83SBram Moolenaar // Remove nested folds between line1 and line2 & reduce the
3093217e1b83SBram Moolenaar // length of fold by "range_len".
3094217e1b83SBram Moolenaar // Folds after this one must be dealt with.
309588d298aeSBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested, line1 - fp->fd_top, line2 -
309688d298aeSBram Moolenaar fp->fd_top, MAXLNUM, -range_len);
309788d298aeSBram Moolenaar fp->fd_len -= range_len;
309888d298aeSBram Moolenaar }
309988d298aeSBram Moolenaar else
3100217e1b83SBram Moolenaar // Case 2 truncate fold, folds after this one must be dealt with.
310140ebc0afSBram Moolenaar truncate_fold(fp, line1 - 1);
310288d298aeSBram Moolenaar
3103217e1b83SBram Moolenaar // Look at the next fold, and treat that one as if it were the first
3104217e1b83SBram Moolenaar // after "line1" (because now it is).
310588d298aeSBram Moolenaar fp = fp + 1;
310688d298aeSBram Moolenaar }
310788d298aeSBram Moolenaar
310888d298aeSBram Moolenaar if (!valid_fold(fp, gap) || fp->fd_top > dest)
310988d298aeSBram Moolenaar {
3110217e1b83SBram Moolenaar // Case 10
3111217e1b83SBram Moolenaar // No folds after "line1" and before "dest"
311288d298aeSBram Moolenaar return;
311388d298aeSBram Moolenaar }
311488d298aeSBram Moolenaar else if (fp->fd_top > line2)
311588d298aeSBram Moolenaar {
311640ebc0afSBram Moolenaar for (; valid_fold(fp, gap) && fold_end(fp) <= dest; fp++)
3117217e1b83SBram Moolenaar // Case 9. (for all case 9's) -- shift up.
311888d298aeSBram Moolenaar fp->fd_top -= range_len;
311988d298aeSBram Moolenaar
312040ebc0afSBram Moolenaar if (valid_fold(fp, gap) && fp->fd_top <= dest)
312188d298aeSBram Moolenaar {
3122217e1b83SBram Moolenaar // Case 8. -- ensure truncated at dest, shift up
312388d298aeSBram Moolenaar truncate_fold(fp, dest);
312488d298aeSBram Moolenaar fp->fd_top -= range_len;
312588d298aeSBram Moolenaar }
312688d298aeSBram Moolenaar return;
312788d298aeSBram Moolenaar }
312888d298aeSBram Moolenaar else if (fold_end(fp) > dest)
312988d298aeSBram Moolenaar {
3130217e1b83SBram Moolenaar // Case 7 -- remove nested folds and shrink
313188d298aeSBram Moolenaar foldMarkAdjustRecurse(&fp->fd_nested, line2 + 1 - fp->fd_top, dest -
313288d298aeSBram Moolenaar fp->fd_top, MAXLNUM, -move_len);
313388d298aeSBram Moolenaar fp->fd_len -= move_len;
313488d298aeSBram Moolenaar fp->fd_top += move_len;
313588d298aeSBram Moolenaar return;
313688d298aeSBram Moolenaar }
313788d298aeSBram Moolenaar
3138217e1b83SBram Moolenaar // Case 5 or 6
3139217e1b83SBram Moolenaar // changes rely on whether there are folds between the end of
3140217e1b83SBram Moolenaar // this fold and "dest".
314188d298aeSBram Moolenaar move_start = fold_index(fp, gap);
314288d298aeSBram Moolenaar
314388d298aeSBram Moolenaar for (; valid_fold(fp, gap) && fp->fd_top <= dest; fp++)
314488d298aeSBram Moolenaar {
314588d298aeSBram Moolenaar if (fp->fd_top <= line2)
314688d298aeSBram Moolenaar {
3147217e1b83SBram Moolenaar // 1. 2. or 3.
314888d298aeSBram Moolenaar if (fold_end(fp) > line2)
3149217e1b83SBram Moolenaar // 2. or 3., truncate before moving
315088d298aeSBram Moolenaar truncate_fold(fp, line2);
315188d298aeSBram Moolenaar
315288d298aeSBram Moolenaar fp->fd_top += move_len;
315388d298aeSBram Moolenaar continue;
315488d298aeSBram Moolenaar }
315588d298aeSBram Moolenaar
3156217e1b83SBram Moolenaar // Record index of the first fold after the moved range.
315788d298aeSBram Moolenaar if (move_end == 0)
315888d298aeSBram Moolenaar move_end = fold_index(fp, gap);
315988d298aeSBram Moolenaar
316088d298aeSBram Moolenaar if (fold_end(fp) > dest)
316188d298aeSBram Moolenaar truncate_fold(fp, dest);
316288d298aeSBram Moolenaar
316388d298aeSBram Moolenaar fp->fd_top -= range_len;
316488d298aeSBram Moolenaar }
316588d298aeSBram Moolenaar
316688d298aeSBram Moolenaar dest_index = fold_index(fp, gap);
316788d298aeSBram Moolenaar
316888d298aeSBram Moolenaar /*
316994be619eSBram Moolenaar * All folds are now correct, but not necessarily in the correct order. We
317094be619eSBram Moolenaar * must swap folds in the range [move_end, dest_index) with those in the
317194be619eSBram Moolenaar * range [move_start, move_end).
317288d298aeSBram Moolenaar */
317394be619eSBram Moolenaar if (move_end == 0)
3174217e1b83SBram Moolenaar // There are no folds after those moved, hence no folds have been moved
3175217e1b83SBram Moolenaar // out of order.
317694be619eSBram Moolenaar return;
31770c0d4ecaSBram Moolenaar foldReverseOrder(gap, (linenr_T)move_start, (linenr_T)dest_index - 1);
31780c0d4ecaSBram Moolenaar foldReverseOrder(gap, (linenr_T)move_start,
31790c0d4ecaSBram Moolenaar (linenr_T)(move_start + dest_index - move_end - 1));
31800c0d4ecaSBram Moolenaar foldReverseOrder(gap, (linenr_T)(move_start + dest_index - move_end),
31810c0d4ecaSBram Moolenaar (linenr_T)(dest_index - 1));
318288d298aeSBram Moolenaar }
318388d298aeSBram Moolenaar #undef fold_end
318488d298aeSBram Moolenaar #undef valid_fold
318588d298aeSBram Moolenaar #undef fold_index
318688d298aeSBram Moolenaar
3187217e1b83SBram Moolenaar // foldMerge() {{{2
3188071d4279SBram Moolenaar /*
318925394022SBram Moolenaar * Merge two adjacent folds (and the nested ones in them).
319025394022SBram Moolenaar * This only works correctly when the folds are really adjacent! Thus "fp1"
3191071d4279SBram Moolenaar * must end just above "fp2".
3192071d4279SBram Moolenaar * The resulting fold is "fp1", nested folds are moved from "fp2" to "fp1".
3193071d4279SBram Moolenaar * Fold entry "fp2" in "gap" is deleted.
3194071d4279SBram Moolenaar */
3195071d4279SBram Moolenaar static void
foldMerge(fold_T * fp1,garray_T * gap,fold_T * fp2)319666f948e9SBram Moolenaar foldMerge(fold_T *fp1, garray_T *gap, fold_T *fp2)
3197071d4279SBram Moolenaar {
3198071d4279SBram Moolenaar fold_T *fp3;
3199071d4279SBram Moolenaar fold_T *fp4;
3200071d4279SBram Moolenaar int idx;
3201071d4279SBram Moolenaar garray_T *gap1 = &fp1->fd_nested;
3202071d4279SBram Moolenaar garray_T *gap2 = &fp2->fd_nested;
3203071d4279SBram Moolenaar
3204217e1b83SBram Moolenaar // If the last nested fold in fp1 touches the first nested fold in fp2,
3205217e1b83SBram Moolenaar // merge them recursively.
3206071d4279SBram Moolenaar if (foldFind(gap1, fp1->fd_len - 1L, &fp3) && foldFind(gap2, 0L, &fp4))
3207071d4279SBram Moolenaar foldMerge(fp3, gap2, fp4);
3208071d4279SBram Moolenaar
3209217e1b83SBram Moolenaar // Move nested folds in fp2 to the end of fp1.
3210071d4279SBram Moolenaar if (gap2->ga_len > 0 && ga_grow(gap1, gap2->ga_len) == OK)
3211071d4279SBram Moolenaar {
3212071d4279SBram Moolenaar for (idx = 0; idx < gap2->ga_len; ++idx)
3213071d4279SBram Moolenaar {
3214071d4279SBram Moolenaar ((fold_T *)gap1->ga_data)[gap1->ga_len]
3215071d4279SBram Moolenaar = ((fold_T *)gap2->ga_data)[idx];
3216071d4279SBram Moolenaar ((fold_T *)gap1->ga_data)[gap1->ga_len].fd_top += fp1->fd_len;
3217071d4279SBram Moolenaar ++gap1->ga_len;
3218071d4279SBram Moolenaar }
3219071d4279SBram Moolenaar gap2->ga_len = 0;
3220071d4279SBram Moolenaar }
3221071d4279SBram Moolenaar
3222071d4279SBram Moolenaar fp1->fd_len += fp2->fd_len;
3223071d4279SBram Moolenaar deleteFoldEntry(gap, (int)(fp2 - (fold_T *)gap->ga_data), TRUE);
3224071d4279SBram Moolenaar fold_changed = TRUE;
3225071d4279SBram Moolenaar }
3226071d4279SBram Moolenaar
3227217e1b83SBram Moolenaar // foldlevelIndent() {{{2
3228071d4279SBram Moolenaar /*
3229071d4279SBram Moolenaar * Low level function to get the foldlevel for the "indent" method.
3230071d4279SBram Moolenaar * Doesn't use any caching.
3231071d4279SBram Moolenaar * Returns a level of -1 if the foldlevel depends on surrounding lines.
3232071d4279SBram Moolenaar */
3233071d4279SBram Moolenaar static void
foldlevelIndent(fline_T * flp)323466f948e9SBram Moolenaar foldlevelIndent(fline_T *flp)
3235071d4279SBram Moolenaar {
3236071d4279SBram Moolenaar char_u *s;
3237071d4279SBram Moolenaar buf_T *buf;
3238071d4279SBram Moolenaar linenr_T lnum = flp->lnum + flp->off;
3239071d4279SBram Moolenaar
3240071d4279SBram Moolenaar buf = flp->wp->w_buffer;
3241071d4279SBram Moolenaar s = skipwhite(ml_get_buf(buf, lnum, FALSE));
3242071d4279SBram Moolenaar
3243217e1b83SBram Moolenaar // empty line or lines starting with a character in 'foldignore': level
3244217e1b83SBram Moolenaar // depends on surrounding lines
3245071d4279SBram Moolenaar if (*s == NUL || vim_strchr(flp->wp->w_p_fdi, *s) != NULL)
3246071d4279SBram Moolenaar {
3247217e1b83SBram Moolenaar // first and last line can't be undefined, use level 0
3248071d4279SBram Moolenaar if (lnum == 1 || lnum == buf->b_ml.ml_line_count)
3249071d4279SBram Moolenaar flp->lvl = 0;
3250071d4279SBram Moolenaar else
3251071d4279SBram Moolenaar flp->lvl = -1;
3252071d4279SBram Moolenaar }
3253071d4279SBram Moolenaar else
32540c27cbcaSBram Moolenaar flp->lvl = get_indent_buf(buf, lnum) / get_sw_value(buf);
3255071d4279SBram Moolenaar if (flp->lvl > flp->wp->w_p_fdn)
325674c596b5SBram Moolenaar {
3257071d4279SBram Moolenaar flp->lvl = flp->wp->w_p_fdn;
325874c596b5SBram Moolenaar if (flp->lvl < 0)
325974c596b5SBram Moolenaar flp->lvl = 0;
326074c596b5SBram Moolenaar }
3261071d4279SBram Moolenaar }
3262071d4279SBram Moolenaar
3263217e1b83SBram Moolenaar // foldlevelDiff() {{{2
3264071d4279SBram Moolenaar #ifdef FEAT_DIFF
3265071d4279SBram Moolenaar /*
3266071d4279SBram Moolenaar * Low level function to get the foldlevel for the "diff" method.
3267071d4279SBram Moolenaar * Doesn't use any caching.
3268071d4279SBram Moolenaar */
3269071d4279SBram Moolenaar static void
foldlevelDiff(fline_T * flp)327066f948e9SBram Moolenaar foldlevelDiff(fline_T *flp)
3271071d4279SBram Moolenaar {
3272071d4279SBram Moolenaar if (diff_infold(flp->wp, flp->lnum + flp->off))
3273071d4279SBram Moolenaar flp->lvl = 1;
3274071d4279SBram Moolenaar else
3275071d4279SBram Moolenaar flp->lvl = 0;
3276071d4279SBram Moolenaar }
3277071d4279SBram Moolenaar #endif
3278071d4279SBram Moolenaar
3279217e1b83SBram Moolenaar // foldlevelExpr() {{{2
3280071d4279SBram Moolenaar /*
3281071d4279SBram Moolenaar * Low level function to get the foldlevel for the "expr" method.
3282071d4279SBram Moolenaar * Doesn't use any caching.
3283071d4279SBram Moolenaar * Returns a level of -1 if the foldlevel depends on surrounding lines.
3284071d4279SBram Moolenaar */
3285071d4279SBram Moolenaar static void
foldlevelExpr(fline_T * flp)328666f948e9SBram Moolenaar foldlevelExpr(fline_T *flp)
3287071d4279SBram Moolenaar {
3288071d4279SBram Moolenaar #ifndef FEAT_EVAL
3289071d4279SBram Moolenaar flp->start = FALSE;
3290071d4279SBram Moolenaar flp->lvl = 0;
3291071d4279SBram Moolenaar #else
3292071d4279SBram Moolenaar win_T *win;
3293071d4279SBram Moolenaar int n;
3294071d4279SBram Moolenaar int c;
3295071d4279SBram Moolenaar linenr_T lnum = flp->lnum + flp->off;
3296071d4279SBram Moolenaar int save_keytyped;
3297071d4279SBram Moolenaar
3298071d4279SBram Moolenaar win = curwin;
3299071d4279SBram Moolenaar curwin = flp->wp;
3300071d4279SBram Moolenaar curbuf = flp->wp->w_buffer;
3301071d4279SBram Moolenaar set_vim_var_nr(VV_LNUM, lnum);
3302071d4279SBram Moolenaar
3303071d4279SBram Moolenaar flp->start = 0;
3304071d4279SBram Moolenaar flp->had_end = flp->end;
3305071d4279SBram Moolenaar flp->end = MAX_LEVEL + 1;
3306071d4279SBram Moolenaar if (lnum <= 1)
3307071d4279SBram Moolenaar flp->lvl = 0;
3308071d4279SBram Moolenaar
3309217e1b83SBram Moolenaar // KeyTyped may be reset to 0 when calling a function which invokes
3310217e1b83SBram Moolenaar // do_cmdline(). To make 'foldopen' work correctly restore KeyTyped.
3311071d4279SBram Moolenaar save_keytyped = KeyTyped;
3312f9547eb6SBram Moolenaar n = eval_foldexpr(flp->wp->w_p_fde, &c);
3313071d4279SBram Moolenaar KeyTyped = save_keytyped;
3314071d4279SBram Moolenaar
3315071d4279SBram Moolenaar switch (c)
3316071d4279SBram Moolenaar {
3317217e1b83SBram Moolenaar // "a1", "a2", .. : add to the fold level
3318071d4279SBram Moolenaar case 'a': if (flp->lvl >= 0)
3319071d4279SBram Moolenaar {
3320071d4279SBram Moolenaar flp->lvl += n;
3321071d4279SBram Moolenaar flp->lvl_next = flp->lvl;
3322071d4279SBram Moolenaar }
3323071d4279SBram Moolenaar flp->start = n;
3324071d4279SBram Moolenaar break;
3325071d4279SBram Moolenaar
3326217e1b83SBram Moolenaar // "s1", "s2", .. : subtract from the fold level
3327071d4279SBram Moolenaar case 's': if (flp->lvl >= 0)
3328071d4279SBram Moolenaar {
3329071d4279SBram Moolenaar if (n > flp->lvl)
3330071d4279SBram Moolenaar flp->lvl_next = 0;
3331071d4279SBram Moolenaar else
3332071d4279SBram Moolenaar flp->lvl_next = flp->lvl - n;
3333071d4279SBram Moolenaar flp->end = flp->lvl_next + 1;
3334071d4279SBram Moolenaar }
3335071d4279SBram Moolenaar break;
3336071d4279SBram Moolenaar
3337217e1b83SBram Moolenaar // ">1", ">2", .. : start a fold with a certain level
3338071d4279SBram Moolenaar case '>': flp->lvl = n;
3339071d4279SBram Moolenaar flp->lvl_next = n;
3340071d4279SBram Moolenaar flp->start = 1;
3341071d4279SBram Moolenaar break;
3342071d4279SBram Moolenaar
3343217e1b83SBram Moolenaar // "<1", "<2", .. : end a fold with a certain level
3344071d4279SBram Moolenaar case '<': flp->lvl_next = n - 1;
3345071d4279SBram Moolenaar flp->end = n;
3346071d4279SBram Moolenaar break;
3347071d4279SBram Moolenaar
3348217e1b83SBram Moolenaar // "=": No change in level
3349071d4279SBram Moolenaar case '=': flp->lvl_next = flp->lvl;
3350071d4279SBram Moolenaar break;
3351071d4279SBram Moolenaar
3352217e1b83SBram Moolenaar // "-1", "0", "1", ..: set fold level
3353071d4279SBram Moolenaar default: if (n < 0)
3354217e1b83SBram Moolenaar // Use the current level for the next line, so that "a1"
3355217e1b83SBram Moolenaar // will work there.
3356071d4279SBram Moolenaar flp->lvl_next = flp->lvl;
3357071d4279SBram Moolenaar else
3358071d4279SBram Moolenaar flp->lvl_next = n;
3359071d4279SBram Moolenaar flp->lvl = n;
3360071d4279SBram Moolenaar break;
3361071d4279SBram Moolenaar }
3362071d4279SBram Moolenaar
3363217e1b83SBram Moolenaar // If the level is unknown for the first or the last line in the file, use
3364217e1b83SBram Moolenaar // level 0.
3365071d4279SBram Moolenaar if (flp->lvl < 0)
3366071d4279SBram Moolenaar {
3367071d4279SBram Moolenaar if (lnum <= 1)
3368071d4279SBram Moolenaar {
3369071d4279SBram Moolenaar flp->lvl = 0;
3370071d4279SBram Moolenaar flp->lvl_next = 0;
3371071d4279SBram Moolenaar }
3372071d4279SBram Moolenaar if (lnum == curbuf->b_ml.ml_line_count)
3373071d4279SBram Moolenaar flp->lvl_next = 0;
3374071d4279SBram Moolenaar }
3375071d4279SBram Moolenaar
3376071d4279SBram Moolenaar curwin = win;
3377071d4279SBram Moolenaar curbuf = curwin->w_buffer;
3378071d4279SBram Moolenaar #endif
3379071d4279SBram Moolenaar }
3380071d4279SBram Moolenaar
3381217e1b83SBram Moolenaar // parseMarker() {{{2
3382071d4279SBram Moolenaar /*
3383071d4279SBram Moolenaar * Parse 'foldmarker' and set "foldendmarker", "foldstartmarkerlen" and
3384071d4279SBram Moolenaar * "foldendmarkerlen".
3385071d4279SBram Moolenaar * Relies on the option value to have been checked for correctness already.
3386071d4279SBram Moolenaar */
3387071d4279SBram Moolenaar static void
parseMarker(win_T * wp)338866f948e9SBram Moolenaar parseMarker(win_T *wp)
3389071d4279SBram Moolenaar {
3390071d4279SBram Moolenaar foldendmarker = vim_strchr(wp->w_p_fmr, ',');
3391071d4279SBram Moolenaar foldstartmarkerlen = (int)(foldendmarker++ - wp->w_p_fmr);
3392071d4279SBram Moolenaar foldendmarkerlen = (int)STRLEN(foldendmarker);
3393071d4279SBram Moolenaar }
3394071d4279SBram Moolenaar
3395217e1b83SBram Moolenaar // foldlevelMarker() {{{2
3396071d4279SBram Moolenaar /*
3397071d4279SBram Moolenaar * Low level function to get the foldlevel for the "marker" method.
3398071d4279SBram Moolenaar * "foldendmarker", "foldstartmarkerlen" and "foldendmarkerlen" must have been
3399071d4279SBram Moolenaar * set before calling this.
3400071d4279SBram Moolenaar * Requires that flp->lvl is set to the fold level of the previous line!
3401071d4279SBram Moolenaar * Careful: This means you can't call this function twice on the same line.
3402071d4279SBram Moolenaar * Doesn't use any caching.
3403071d4279SBram Moolenaar * Sets flp->start when a start marker was found.
3404071d4279SBram Moolenaar */
3405071d4279SBram Moolenaar static void
foldlevelMarker(fline_T * flp)340666f948e9SBram Moolenaar foldlevelMarker(fline_T *flp)
3407071d4279SBram Moolenaar {
3408071d4279SBram Moolenaar char_u *startmarker;
3409071d4279SBram Moolenaar int cstart;
3410071d4279SBram Moolenaar int cend;
3411071d4279SBram Moolenaar int start_lvl = flp->lvl;
3412071d4279SBram Moolenaar char_u *s;
3413071d4279SBram Moolenaar int n;
3414071d4279SBram Moolenaar
3415217e1b83SBram Moolenaar // cache a few values for speed
3416071d4279SBram Moolenaar startmarker = flp->wp->w_p_fmr;
3417071d4279SBram Moolenaar cstart = *startmarker;
3418071d4279SBram Moolenaar ++startmarker;
3419071d4279SBram Moolenaar cend = *foldendmarker;
3420071d4279SBram Moolenaar
3421217e1b83SBram Moolenaar // Default: no start found, next level is same as current level
3422071d4279SBram Moolenaar flp->start = 0;
3423071d4279SBram Moolenaar flp->lvl_next = flp->lvl;
3424071d4279SBram Moolenaar
3425071d4279SBram Moolenaar s = ml_get_buf(flp->wp->w_buffer, flp->lnum + flp->off, FALSE);
3426071d4279SBram Moolenaar while (*s)
3427071d4279SBram Moolenaar {
3428071d4279SBram Moolenaar if (*s == cstart
3429071d4279SBram Moolenaar && STRNCMP(s + 1, startmarker, foldstartmarkerlen - 1) == 0)
3430071d4279SBram Moolenaar {
3431217e1b83SBram Moolenaar // found startmarker: set flp->lvl
3432071d4279SBram Moolenaar s += foldstartmarkerlen;
3433071d4279SBram Moolenaar if (VIM_ISDIGIT(*s))
3434071d4279SBram Moolenaar {
3435071d4279SBram Moolenaar n = atoi((char *)s);
3436071d4279SBram Moolenaar if (n > 0)
3437071d4279SBram Moolenaar {
3438071d4279SBram Moolenaar flp->lvl = n;
3439071d4279SBram Moolenaar flp->lvl_next = n;
3440071d4279SBram Moolenaar if (n <= start_lvl)
3441071d4279SBram Moolenaar flp->start = 1;
3442071d4279SBram Moolenaar else
3443071d4279SBram Moolenaar flp->start = n - start_lvl;
3444071d4279SBram Moolenaar }
3445071d4279SBram Moolenaar }
3446071d4279SBram Moolenaar else
3447071d4279SBram Moolenaar {
3448071d4279SBram Moolenaar ++flp->lvl;
3449071d4279SBram Moolenaar ++flp->lvl_next;
3450071d4279SBram Moolenaar ++flp->start;
3451071d4279SBram Moolenaar }
3452071d4279SBram Moolenaar }
3453071d4279SBram Moolenaar else if (*s == cend
3454071d4279SBram Moolenaar && STRNCMP(s + 1, foldendmarker + 1, foldendmarkerlen - 1) == 0)
3455071d4279SBram Moolenaar {
3456217e1b83SBram Moolenaar // found endmarker: set flp->lvl_next
3457071d4279SBram Moolenaar s += foldendmarkerlen;
3458071d4279SBram Moolenaar if (VIM_ISDIGIT(*s))
3459071d4279SBram Moolenaar {
3460071d4279SBram Moolenaar n = atoi((char *)s);
3461071d4279SBram Moolenaar if (n > 0)
3462071d4279SBram Moolenaar {
3463071d4279SBram Moolenaar flp->lvl = n;
3464071d4279SBram Moolenaar flp->lvl_next = n - 1;
3465217e1b83SBram Moolenaar // never start a fold with an end marker
34663ee0229fSBram Moolenaar if (flp->lvl_next > start_lvl)
34673ee0229fSBram Moolenaar flp->lvl_next = start_lvl;
3468071d4279SBram Moolenaar }
3469071d4279SBram Moolenaar }
3470071d4279SBram Moolenaar else
3471071d4279SBram Moolenaar --flp->lvl_next;
3472071d4279SBram Moolenaar }
3473071d4279SBram Moolenaar else
347491acfffcSBram Moolenaar MB_PTR_ADV(s);
3475071d4279SBram Moolenaar }
3476071d4279SBram Moolenaar
3477217e1b83SBram Moolenaar // The level can't go negative, must be missing a start marker.
3478071d4279SBram Moolenaar if (flp->lvl_next < 0)
3479071d4279SBram Moolenaar flp->lvl_next = 0;
3480071d4279SBram Moolenaar }
3481071d4279SBram Moolenaar
3482217e1b83SBram Moolenaar // foldlevelSyntax() {{{2
3483071d4279SBram Moolenaar /*
3484071d4279SBram Moolenaar * Low level function to get the foldlevel for the "syntax" method.
3485071d4279SBram Moolenaar * Doesn't use any caching.
3486071d4279SBram Moolenaar */
3487071d4279SBram Moolenaar static void
foldlevelSyntax(fline_T * flp)348866f948e9SBram Moolenaar foldlevelSyntax(fline_T *flp)
3489071d4279SBram Moolenaar {
3490071d4279SBram Moolenaar #ifndef FEAT_SYN_HL
3491071d4279SBram Moolenaar flp->start = 0;
3492071d4279SBram Moolenaar flp->lvl = 0;
3493071d4279SBram Moolenaar #else
3494071d4279SBram Moolenaar linenr_T lnum = flp->lnum + flp->off;
3495071d4279SBram Moolenaar int n;
3496071d4279SBram Moolenaar
3497217e1b83SBram Moolenaar // Use the maximum fold level at the start of this line and the next.
3498071d4279SBram Moolenaar flp->lvl = syn_get_foldlevel(flp->wp, lnum);
3499071d4279SBram Moolenaar flp->start = 0;
3500071d4279SBram Moolenaar if (lnum < flp->wp->w_buffer->b_ml.ml_line_count)
3501071d4279SBram Moolenaar {
3502071d4279SBram Moolenaar n = syn_get_foldlevel(flp->wp, lnum + 1);
3503071d4279SBram Moolenaar if (n > flp->lvl)
3504071d4279SBram Moolenaar {
3505217e1b83SBram Moolenaar flp->start = n - flp->lvl; // fold(s) start here
3506071d4279SBram Moolenaar flp->lvl = n;
3507071d4279SBram Moolenaar }
3508071d4279SBram Moolenaar }
3509071d4279SBram Moolenaar #endif
3510071d4279SBram Moolenaar }
3511071d4279SBram Moolenaar
3512217e1b83SBram Moolenaar // functions for storing the fold state in a View {{{1
3513217e1b83SBram Moolenaar // put_folds() {{{2
3514071d4279SBram Moolenaar #if defined(FEAT_SESSION) || defined(PROTO)
3515d25c16e2SBram Moolenaar static int put_folds_recurse(FILE *fd, garray_T *gap, linenr_T off);
3516d25c16e2SBram Moolenaar static int put_foldopen_recurse(FILE *fd, win_T *wp, garray_T *gap, linenr_T off);
3517d25c16e2SBram Moolenaar static int put_fold_open_close(FILE *fd, fold_T *fp, linenr_T off);
3518071d4279SBram Moolenaar
3519071d4279SBram Moolenaar /*
3520071d4279SBram Moolenaar * Write commands to "fd" to restore the manual folds in window "wp".
3521071d4279SBram Moolenaar * Return FAIL if writing fails.
3522071d4279SBram Moolenaar */
3523071d4279SBram Moolenaar int
put_folds(FILE * fd,win_T * wp)352466f948e9SBram Moolenaar put_folds(FILE *fd, win_T *wp)
3525071d4279SBram Moolenaar {
3526071d4279SBram Moolenaar if (foldmethodIsManual(wp))
3527071d4279SBram Moolenaar {
3528071d4279SBram Moolenaar if (put_line(fd, "silent! normal! zE") == FAIL
3529f9547eb6SBram Moolenaar || put_folds_recurse(fd, &wp->w_folds, (linenr_T)0) == FAIL
3530f9547eb6SBram Moolenaar || put_line(fd, "let &fdl = &fdl") == FAIL)
3531071d4279SBram Moolenaar return FAIL;
3532071d4279SBram Moolenaar }
3533071d4279SBram Moolenaar
3534217e1b83SBram Moolenaar // If some folds are manually opened/closed, need to restore that.
3535071d4279SBram Moolenaar if (wp->w_fold_manual)
3536d25ad65aSBram Moolenaar return put_foldopen_recurse(fd, wp, &wp->w_folds, (linenr_T)0);
3537071d4279SBram Moolenaar
3538071d4279SBram Moolenaar return OK;
3539071d4279SBram Moolenaar }
3540071d4279SBram Moolenaar
3541217e1b83SBram Moolenaar // put_folds_recurse() {{{2
3542071d4279SBram Moolenaar /*
3543071d4279SBram Moolenaar * Write commands to "fd" to recreate manually created folds.
3544071d4279SBram Moolenaar * Returns FAIL when writing failed.
3545071d4279SBram Moolenaar */
3546071d4279SBram Moolenaar static int
put_folds_recurse(FILE * fd,garray_T * gap,linenr_T off)354766f948e9SBram Moolenaar put_folds_recurse(FILE *fd, garray_T *gap, linenr_T off)
3548071d4279SBram Moolenaar {
3549071d4279SBram Moolenaar int i;
3550071d4279SBram Moolenaar fold_T *fp;
3551071d4279SBram Moolenaar
3552071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data;
3553071d4279SBram Moolenaar for (i = 0; i < gap->ga_len; i++)
3554071d4279SBram Moolenaar {
3555217e1b83SBram Moolenaar // Do nested folds first, they will be created closed.
3556071d4279SBram Moolenaar if (put_folds_recurse(fd, &fp->fd_nested, off + fp->fd_top) == FAIL)
3557071d4279SBram Moolenaar return FAIL;
3558071d4279SBram Moolenaar if (fprintf(fd, "%ld,%ldfold", fp->fd_top + off,
3559071d4279SBram Moolenaar fp->fd_top + off + fp->fd_len - 1) < 0
3560071d4279SBram Moolenaar || put_eol(fd) == FAIL)
3561071d4279SBram Moolenaar return FAIL;
3562071d4279SBram Moolenaar ++fp;
3563071d4279SBram Moolenaar }
3564071d4279SBram Moolenaar return OK;
3565071d4279SBram Moolenaar }
3566071d4279SBram Moolenaar
3567217e1b83SBram Moolenaar // put_foldopen_recurse() {{{2
3568071d4279SBram Moolenaar /*
3569071d4279SBram Moolenaar * Write commands to "fd" to open and close manually opened/closed folds.
3570071d4279SBram Moolenaar * Returns FAIL when writing failed.
3571071d4279SBram Moolenaar */
3572071d4279SBram Moolenaar static int
put_foldopen_recurse(FILE * fd,win_T * wp,garray_T * gap,linenr_T off)357366f948e9SBram Moolenaar put_foldopen_recurse(
357466f948e9SBram Moolenaar FILE *fd,
357566f948e9SBram Moolenaar win_T *wp,
357666f948e9SBram Moolenaar garray_T *gap,
357766f948e9SBram Moolenaar linenr_T off)
3578071d4279SBram Moolenaar {
3579071d4279SBram Moolenaar int i;
3580d25ad65aSBram Moolenaar int level;
3581071d4279SBram Moolenaar fold_T *fp;
3582071d4279SBram Moolenaar
3583071d4279SBram Moolenaar fp = (fold_T *)gap->ga_data;
3584071d4279SBram Moolenaar for (i = 0; i < gap->ga_len; i++)
3585071d4279SBram Moolenaar {
3586071d4279SBram Moolenaar if (fp->fd_flags != FD_LEVEL)
3587071d4279SBram Moolenaar {
3588071d4279SBram Moolenaar if (fp->fd_nested.ga_len > 0)
3589071d4279SBram Moolenaar {
3590217e1b83SBram Moolenaar // open nested folds while this fold is open
3591071d4279SBram Moolenaar if (fprintf(fd, "%ld", fp->fd_top + off) < 0
3592071d4279SBram Moolenaar || put_eol(fd) == FAIL
3593002a4edcSBram Moolenaar || put_line(fd, "normal! zo") == FAIL)
3594071d4279SBram Moolenaar return FAIL;
3595d25ad65aSBram Moolenaar if (put_foldopen_recurse(fd, wp, &fp->fd_nested,
3596d25ad65aSBram Moolenaar off + fp->fd_top)
3597071d4279SBram Moolenaar == FAIL)
3598071d4279SBram Moolenaar return FAIL;
3599217e1b83SBram Moolenaar // close the parent when needed
3600d25ad65aSBram Moolenaar if (fp->fd_flags == FD_CLOSED)
3601d25ad65aSBram Moolenaar {
3602d25ad65aSBram Moolenaar if (put_fold_open_close(fd, fp, off) == FAIL)
3603d25ad65aSBram Moolenaar return FAIL;
3604071d4279SBram Moolenaar }
3605d25ad65aSBram Moolenaar }
3606d25ad65aSBram Moolenaar else
3607d25ad65aSBram Moolenaar {
3608217e1b83SBram Moolenaar // Open or close the leaf according to the window foldlevel.
3609217e1b83SBram Moolenaar // Do not close a leaf that is already closed, as it will close
3610217e1b83SBram Moolenaar // the parent.
3611d25ad65aSBram Moolenaar level = foldLevelWin(wp, off + fp->fd_top);
3612d25ad65aSBram Moolenaar if ((fp->fd_flags == FD_CLOSED && wp->w_p_fdl >= level)
3613d25ad65aSBram Moolenaar || (fp->fd_flags != FD_CLOSED && wp->w_p_fdl < level))
3614d25ad65aSBram Moolenaar if (put_fold_open_close(fd, fp, off) == FAIL)
3615d25ad65aSBram Moolenaar return FAIL;
3616d25ad65aSBram Moolenaar }
3617d25ad65aSBram Moolenaar }
3618d25ad65aSBram Moolenaar ++fp;
3619d25ad65aSBram Moolenaar }
3620d25ad65aSBram Moolenaar
3621d25ad65aSBram Moolenaar return OK;
3622d25ad65aSBram Moolenaar }
3623d25ad65aSBram Moolenaar
3624217e1b83SBram Moolenaar // put_fold_open_close() {{{2
3625d25ad65aSBram Moolenaar /*
3626d25ad65aSBram Moolenaar * Write the open or close command to "fd".
3627d25ad65aSBram Moolenaar * Returns FAIL when writing failed.
3628d25ad65aSBram Moolenaar */
3629d25ad65aSBram Moolenaar static int
put_fold_open_close(FILE * fd,fold_T * fp,linenr_T off)363066f948e9SBram Moolenaar put_fold_open_close(FILE *fd, fold_T *fp, linenr_T off)
3631d25ad65aSBram Moolenaar {
3632071d4279SBram Moolenaar if (fprintf(fd, "%ld", fp->fd_top + off) < 0
3633071d4279SBram Moolenaar || put_eol(fd) == FAIL
3634002a4edcSBram Moolenaar || fprintf(fd, "normal! z%c",
3635071d4279SBram Moolenaar fp->fd_flags == FD_CLOSED ? 'c' : 'o') < 0
3636071d4279SBram Moolenaar || put_eol(fd) == FAIL)
3637071d4279SBram Moolenaar return FAIL;
3638071d4279SBram Moolenaar
3639071d4279SBram Moolenaar return OK;
3640071d4279SBram Moolenaar }
3641217e1b83SBram Moolenaar #endif // FEAT_SESSION
3642071d4279SBram Moolenaar
3643217e1b83SBram Moolenaar // }}}1
3644db022f3fSBram Moolenaar #endif // defined(FEAT_FOLDING) || defined(PROTO)
3645db022f3fSBram Moolenaar
3646db022f3fSBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO)
3647db022f3fSBram Moolenaar
3648db022f3fSBram Moolenaar /*
3649db022f3fSBram Moolenaar * "foldclosed()" and "foldclosedend()" functions
3650db022f3fSBram Moolenaar */
3651db022f3fSBram Moolenaar static void
foldclosed_both(typval_T * argvars UNUSED,typval_T * rettv,int end UNUSED)3652db022f3fSBram Moolenaar foldclosed_both(
3653db022f3fSBram Moolenaar typval_T *argvars UNUSED,
3654db022f3fSBram Moolenaar typval_T *rettv,
3655db022f3fSBram Moolenaar int end UNUSED)
3656db022f3fSBram Moolenaar {
3657db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3658db022f3fSBram Moolenaar linenr_T lnum;
3659db022f3fSBram Moolenaar linenr_T first, last;
3660db022f3fSBram Moolenaar
3661*4490ec4eSYegappan Lakshmanan if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
3662*4490ec4eSYegappan Lakshmanan return;
3663*4490ec4eSYegappan Lakshmanan
3664db022f3fSBram Moolenaar lnum = tv_get_lnum(argvars);
3665db022f3fSBram Moolenaar if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
3666db022f3fSBram Moolenaar {
3667db022f3fSBram Moolenaar if (hasFoldingWin(curwin, lnum, &first, &last, FALSE, NULL))
3668db022f3fSBram Moolenaar {
3669db022f3fSBram Moolenaar if (end)
3670db022f3fSBram Moolenaar rettv->vval.v_number = (varnumber_T)last;
3671db022f3fSBram Moolenaar else
3672db022f3fSBram Moolenaar rettv->vval.v_number = (varnumber_T)first;
3673db022f3fSBram Moolenaar return;
3674db022f3fSBram Moolenaar }
3675db022f3fSBram Moolenaar }
3676db022f3fSBram Moolenaar #endif
3677db022f3fSBram Moolenaar rettv->vval.v_number = -1;
3678db022f3fSBram Moolenaar }
3679db022f3fSBram Moolenaar
3680db022f3fSBram Moolenaar /*
3681db022f3fSBram Moolenaar * "foldclosed()" function
3682db022f3fSBram Moolenaar */
3683db022f3fSBram Moolenaar void
f_foldclosed(typval_T * argvars,typval_T * rettv)3684db022f3fSBram Moolenaar f_foldclosed(typval_T *argvars, typval_T *rettv)
3685db022f3fSBram Moolenaar {
3686db022f3fSBram Moolenaar foldclosed_both(argvars, rettv, FALSE);
3687db022f3fSBram Moolenaar }
3688db022f3fSBram Moolenaar
3689db022f3fSBram Moolenaar /*
3690db022f3fSBram Moolenaar * "foldclosedend()" function
3691db022f3fSBram Moolenaar */
3692db022f3fSBram Moolenaar void
f_foldclosedend(typval_T * argvars,typval_T * rettv)3693db022f3fSBram Moolenaar f_foldclosedend(typval_T *argvars, typval_T *rettv)
3694db022f3fSBram Moolenaar {
3695db022f3fSBram Moolenaar foldclosed_both(argvars, rettv, TRUE);
3696db022f3fSBram Moolenaar }
3697db022f3fSBram Moolenaar
3698db022f3fSBram Moolenaar /*
3699db022f3fSBram Moolenaar * "foldlevel()" function
3700db022f3fSBram Moolenaar */
3701db022f3fSBram Moolenaar void
f_foldlevel(typval_T * argvars UNUSED,typval_T * rettv UNUSED)3702db022f3fSBram Moolenaar f_foldlevel(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3703db022f3fSBram Moolenaar {
3704db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3705db022f3fSBram Moolenaar linenr_T lnum;
3706db022f3fSBram Moolenaar
3707*4490ec4eSYegappan Lakshmanan if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
3708*4490ec4eSYegappan Lakshmanan return;
3709*4490ec4eSYegappan Lakshmanan
3710db022f3fSBram Moolenaar lnum = tv_get_lnum(argvars);
3711db022f3fSBram Moolenaar if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
3712db022f3fSBram Moolenaar rettv->vval.v_number = foldLevel(lnum);
3713db022f3fSBram Moolenaar # endif
3714db022f3fSBram Moolenaar }
3715db022f3fSBram Moolenaar
3716db022f3fSBram Moolenaar /*
3717db022f3fSBram Moolenaar * "foldtext()" function
3718db022f3fSBram Moolenaar */
3719db022f3fSBram Moolenaar void
f_foldtext(typval_T * argvars UNUSED,typval_T * rettv)3720db022f3fSBram Moolenaar f_foldtext(typval_T *argvars UNUSED, typval_T *rettv)
3721db022f3fSBram Moolenaar {
3722db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3723db022f3fSBram Moolenaar linenr_T foldstart;
3724db022f3fSBram Moolenaar linenr_T foldend;
3725db022f3fSBram Moolenaar char_u *dashes;
3726db022f3fSBram Moolenaar linenr_T lnum;
3727db022f3fSBram Moolenaar char_u *s;
3728db022f3fSBram Moolenaar char_u *r;
3729db022f3fSBram Moolenaar int len;
3730db022f3fSBram Moolenaar char *txt;
3731db022f3fSBram Moolenaar long count;
3732db022f3fSBram Moolenaar # endif
3733db022f3fSBram Moolenaar
3734db022f3fSBram Moolenaar rettv->v_type = VAR_STRING;
3735db022f3fSBram Moolenaar rettv->vval.v_string = NULL;
3736db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3737db022f3fSBram Moolenaar foldstart = (linenr_T)get_vim_var_nr(VV_FOLDSTART);
3738db022f3fSBram Moolenaar foldend = (linenr_T)get_vim_var_nr(VV_FOLDEND);
3739db022f3fSBram Moolenaar dashes = get_vim_var_str(VV_FOLDDASHES);
3740db022f3fSBram Moolenaar if (foldstart > 0 && foldend <= curbuf->b_ml.ml_line_count
3741db022f3fSBram Moolenaar && dashes != NULL)
3742db022f3fSBram Moolenaar {
3743db022f3fSBram Moolenaar // Find first non-empty line in the fold.
3744db022f3fSBram Moolenaar for (lnum = foldstart; lnum < foldend; ++lnum)
3745db022f3fSBram Moolenaar if (!linewhite(lnum))
3746db022f3fSBram Moolenaar break;
3747db022f3fSBram Moolenaar
3748db022f3fSBram Moolenaar // Find interesting text in this line.
3749db022f3fSBram Moolenaar s = skipwhite(ml_get(lnum));
3750db022f3fSBram Moolenaar // skip C comment-start
3751db022f3fSBram Moolenaar if (s[0] == '/' && (s[1] == '*' || s[1] == '/'))
3752db022f3fSBram Moolenaar {
3753db022f3fSBram Moolenaar s = skipwhite(s + 2);
3754db022f3fSBram Moolenaar if (*skipwhite(s) == NUL
3755db022f3fSBram Moolenaar && lnum + 1 < (linenr_T)get_vim_var_nr(VV_FOLDEND))
3756db022f3fSBram Moolenaar {
3757db022f3fSBram Moolenaar s = skipwhite(ml_get(lnum + 1));
3758db022f3fSBram Moolenaar if (*s == '*')
3759db022f3fSBram Moolenaar s = skipwhite(s + 1);
3760db022f3fSBram Moolenaar }
3761db022f3fSBram Moolenaar }
3762db022f3fSBram Moolenaar count = (long)(foldend - foldstart + 1);
3763db022f3fSBram Moolenaar txt = NGETTEXT("+-%s%3ld line: ", "+-%s%3ld lines: ", count);
3764db022f3fSBram Moolenaar r = alloc(STRLEN(txt)
3765db022f3fSBram Moolenaar + STRLEN(dashes) // for %s
3766db022f3fSBram Moolenaar + 20 // for %3ld
3767db022f3fSBram Moolenaar + STRLEN(s)); // concatenated
3768db022f3fSBram Moolenaar if (r != NULL)
3769db022f3fSBram Moolenaar {
3770db022f3fSBram Moolenaar sprintf((char *)r, txt, dashes, count);
3771db022f3fSBram Moolenaar len = (int)STRLEN(r);
3772db022f3fSBram Moolenaar STRCAT(r, s);
3773db022f3fSBram Moolenaar // remove 'foldmarker' and 'commentstring'
3774db022f3fSBram Moolenaar foldtext_cleanup(r + len);
3775db022f3fSBram Moolenaar rettv->vval.v_string = r;
3776db022f3fSBram Moolenaar }
3777db022f3fSBram Moolenaar }
3778db022f3fSBram Moolenaar # endif
3779db022f3fSBram Moolenaar }
3780db022f3fSBram Moolenaar
3781db022f3fSBram Moolenaar /*
3782db022f3fSBram Moolenaar * "foldtextresult(lnum)" function
3783db022f3fSBram Moolenaar */
3784db022f3fSBram Moolenaar void
f_foldtextresult(typval_T * argvars UNUSED,typval_T * rettv)3785db022f3fSBram Moolenaar f_foldtextresult(typval_T *argvars UNUSED, typval_T *rettv)
3786db022f3fSBram Moolenaar {
3787db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3788db022f3fSBram Moolenaar linenr_T lnum;
3789db022f3fSBram Moolenaar char_u *text;
3790db022f3fSBram Moolenaar char_u buf[FOLD_TEXT_LEN];
3791db022f3fSBram Moolenaar foldinfo_T foldinfo;
3792db022f3fSBram Moolenaar int fold_count;
3793db022f3fSBram Moolenaar static int entered = FALSE;
3794db022f3fSBram Moolenaar # endif
3795db022f3fSBram Moolenaar
3796db022f3fSBram Moolenaar rettv->v_type = VAR_STRING;
3797db022f3fSBram Moolenaar rettv->vval.v_string = NULL;
3798*4490ec4eSYegappan Lakshmanan
3799*4490ec4eSYegappan Lakshmanan if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
3800*4490ec4eSYegappan Lakshmanan return;
3801*4490ec4eSYegappan Lakshmanan
3802db022f3fSBram Moolenaar # ifdef FEAT_FOLDING
3803db022f3fSBram Moolenaar if (entered)
3804db022f3fSBram Moolenaar return; // reject recursive use
3805db022f3fSBram Moolenaar entered = TRUE;
3806db022f3fSBram Moolenaar
3807db022f3fSBram Moolenaar lnum = tv_get_lnum(argvars);
3808db022f3fSBram Moolenaar // treat illegal types and illegal string values for {lnum} the same
3809db022f3fSBram Moolenaar if (lnum < 0)
3810db022f3fSBram Moolenaar lnum = 0;
3811db022f3fSBram Moolenaar fold_count = foldedCount(curwin, lnum, &foldinfo);
3812db022f3fSBram Moolenaar if (fold_count > 0)
3813db022f3fSBram Moolenaar {
3814db022f3fSBram Moolenaar text = get_foldtext(curwin, lnum, lnum + fold_count - 1,
3815db022f3fSBram Moolenaar &foldinfo, buf);
3816db022f3fSBram Moolenaar if (text == buf)
3817db022f3fSBram Moolenaar text = vim_strsave(text);
3818db022f3fSBram Moolenaar rettv->vval.v_string = text;
3819db022f3fSBram Moolenaar }
3820db022f3fSBram Moolenaar
3821db022f3fSBram Moolenaar entered = FALSE;
3822db022f3fSBram Moolenaar # endif
3823db022f3fSBram Moolenaar }
3824db022f3fSBram Moolenaar
3825db022f3fSBram Moolenaar #endif // FEAT_EVAL
3826