blob: 7b81feba7aa59359ea00137b12c3a83aa27e6d30 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +000011 * diff.c: code for diff'ing two, three or four buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +000012 */
13
14#include "vim.h"
15
16#if defined(FEAT_DIFF) || defined(PROTO)
17
Bram Moolenaar071d4272004-06-13 20:20:40 +000018static int diff_busy = FALSE; /* ex_diffgetput() is busy */
19
20/* flags obtained from the 'diffopt' option */
21#define DIFF_FILLER 1 /* display filler lines */
22#define DIFF_ICASE 2 /* ignore case */
23#define DIFF_IWHITE 4 /* ignore change in white space */
Bram Moolenaarc4675a12006-03-15 22:50:30 +000024#define DIFF_HORIZONTAL 8 /* horizontal splits */
25#define DIFF_VERTICAL 16 /* vertical splits */
Bram Moolenaar071d4272004-06-13 20:20:40 +000026static int diff_flags = DIFF_FILLER;
27
28#define LBUFLEN 50 /* length of line in diff file */
29
30static int diff_a_works = MAYBE; /* TRUE when "diff -a" works, FALSE when it
31 doesn't work, MAYBE when not checked yet */
Bram Moolenaar48e330a2016-02-23 14:53:34 +010032#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +000033static int diff_bin_works = MAYBE; /* TRUE when "diff --binary" works, FALSE
34 when it doesn't work, MAYBE when not
35 checked yet */
36#endif
37
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010038static int diff_buf_idx(buf_T *buf);
39static int diff_buf_idx_tp(buf_T *buf, tabpage_T *tp);
40static void diff_mark_adjust_tp(tabpage_T *tp, int idx, linenr_T line1, linenr_T line2, long amount, long amount_after);
41static void diff_check_unchanged(tabpage_T *tp, diff_T *dp);
42static int diff_check_sanity(tabpage_T *tp, diff_T *dp);
43static void diff_redraw(int dofold);
44static int diff_write(buf_T *buf, char_u *fname);
45static void diff_file(char_u *tmp_orig, char_u *tmp_new, char_u *tmp_diff);
46static int diff_equal_entry(diff_T *dp, int idx1, int idx2);
47static int diff_cmp(char_u *s1, char_u *s2);
Bram Moolenaar071d4272004-06-13 20:20:40 +000048#ifdef FEAT_FOLDING
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010049static void diff_fold_update(diff_T *dp, int skip_idx);
Bram Moolenaar071d4272004-06-13 20:20:40 +000050#endif
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010051static void diff_read(int idx_orig, int idx_new, char_u *fname);
52static void diff_copy_entry(diff_T *dprev, diff_T *dp, int idx_orig, int idx_new);
53static diff_T *diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +000054
55#ifndef USE_CR
56# define tag_fgets vim_fgets
57#endif
58
59/*
Bram Moolenaar071d4272004-06-13 20:20:40 +000060 * Called when deleting or unloading a buffer: No longer make a diff with it.
Bram Moolenaar071d4272004-06-13 20:20:40 +000061 */
62 void
Bram Moolenaar7454a062016-01-30 15:14:10 +010063diff_buf_delete(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +000064{
65 int i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000066 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +000067
Bram Moolenaar29323592016-07-24 22:04:11 +020068 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +000069 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000070 i = diff_buf_idx_tp(buf, tp);
71 if (i != DB_COUNT)
72 {
73 tp->tp_diffbuf[i] = NULL;
74 tp->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +000075 if (tp == curtab)
76 diff_redraw(TRUE);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000077 }
Bram Moolenaar071d4272004-06-13 20:20:40 +000078 }
79}
80
81/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +000082 * Check if the current buffer should be added to or removed from the list of
83 * diff buffers.
84 */
85 void
Bram Moolenaar7454a062016-01-30 15:14:10 +010086diff_buf_adjust(win_T *win)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +000087{
88 win_T *wp;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000089 int i;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +000090
91 if (!win->w_p_diff)
92 {
93 /* When there is no window showing a diff for this buffer, remove
94 * it from the diffs. */
Bram Moolenaar29323592016-07-24 22:04:11 +020095 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +000096 if (wp->w_buffer == win->w_buffer && wp->w_p_diff)
97 break;
98 if (wp == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000099 {
100 i = diff_buf_idx(win->w_buffer);
101 if (i != DB_COUNT)
102 {
103 curtab->tp_diffbuf[i] = NULL;
104 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000105 diff_redraw(TRUE);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000106 }
107 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000108 }
109 else
110 diff_buf_add(win->w_buffer);
111}
112
113/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000114 * Add a buffer to make diffs for.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000115 * Call this when a new buffer is being edited in the current window where
116 * 'diff' is set.
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000117 * Marks the current buffer as being part of the diff and requiring updating.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000118 * This must be done before any autocmd, because a command may use info
119 * about the screen contents.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000120 */
121 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100122diff_buf_add(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000123{
124 int i;
125
126 if (diff_buf_idx(buf) != DB_COUNT)
127 return; /* It's already there. */
128
129 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000130 if (curtab->tp_diffbuf[i] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000131 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000132 curtab->tp_diffbuf[i] = buf;
133 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000134 diff_redraw(TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000135 return;
136 }
137
Bram Moolenaar89eaa412016-07-31 14:17:27 +0200138 EMSGN(_("E96: Cannot diff more than %ld buffers"), DB_COUNT);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000139}
140
141/*
Bram Moolenaar25ea0542017-02-03 23:16:28 +0100142 * Remove all buffers to make diffs for.
143 */
144 static void
145diff_buf_clear(void)
146{
147 int i;
148
149 for (i = 0; i < DB_COUNT; ++i)
150 if (curtab->tp_diffbuf[i] != NULL)
151 {
152 curtab->tp_diffbuf[i] = NULL;
153 curtab->tp_diff_invalid = TRUE;
154 diff_redraw(TRUE);
155 }
156}
157
158/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000159 * Find buffer "buf" in the list of diff buffers for the current tab page.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160 * Return its index or DB_COUNT if not found.
161 */
162 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100163diff_buf_idx(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000164{
165 int idx;
166
167 for (idx = 0; idx < DB_COUNT; ++idx)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000168 if (curtab->tp_diffbuf[idx] == buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169 break;
170 return idx;
171}
172
173/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000174 * Find buffer "buf" in the list of diff buffers for tab page "tp".
175 * Return its index or DB_COUNT if not found.
176 */
177 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100178diff_buf_idx_tp(buf_T *buf, tabpage_T *tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000179{
180 int idx;
181
182 for (idx = 0; idx < DB_COUNT; ++idx)
183 if (tp->tp_diffbuf[idx] == buf)
184 break;
185 return idx;
186}
187
188/*
189 * Mark the diff info involving buffer "buf" as invalid, it will be updated
190 * when info is requested.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000191 */
192 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100193diff_invalidate(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000194{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000195 tabpage_T *tp;
196 int i;
197
Bram Moolenaar29323592016-07-24 22:04:11 +0200198 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000199 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000200 i = diff_buf_idx_tp(buf, tp);
201 if (i != DB_COUNT)
202 {
203 tp->tp_diff_invalid = TRUE;
204 if (tp == curtab)
205 diff_redraw(TRUE);
206 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000207 }
208}
209
210/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000211 * Called by mark_adjust(): update line numbers in "curbuf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000212 */
213 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100214diff_mark_adjust(
215 linenr_T line1,
216 linenr_T line2,
217 long amount,
218 long amount_after)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000219{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000220 int idx;
221 tabpage_T *tp;
222
223 /* Handle all tab pages that use the current buffer in a diff. */
Bram Moolenaar29323592016-07-24 22:04:11 +0200224 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000225 {
226 idx = diff_buf_idx_tp(curbuf, tp);
227 if (idx != DB_COUNT)
228 diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after);
229 }
230}
231
232/*
233 * Update line numbers in tab page "tp" for "curbuf" with index "idx".
234 * This attempts to update the changes as much as possible:
235 * When inserting/deleting lines outside of existing change blocks, create a
236 * new change block and update the line numbers in following blocks.
237 * When inserting/deleting lines in existing change blocks, update them.
238 */
239 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100240diff_mark_adjust_tp(
241 tabpage_T *tp,
242 int idx,
243 linenr_T line1,
244 linenr_T line2,
245 long amount,
246 long amount_after)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000247{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000248 diff_T *dp;
249 diff_T *dprev;
250 diff_T *dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000251 int i;
252 int inserted, deleted;
253 int n, off;
254 linenr_T last;
255 linenr_T lnum_deleted = line1; /* lnum of remaining deletion */
256 int check_unchanged;
257
Bram Moolenaar071d4272004-06-13 20:20:40 +0000258 if (line2 == MAXLNUM)
259 {
260 /* mark_adjust(99, MAXLNUM, 9, 0): insert lines */
261 inserted = amount;
262 deleted = 0;
263 }
264 else if (amount_after > 0)
265 {
266 /* mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines*/
267 inserted = amount_after;
268 deleted = 0;
269 }
270 else
271 {
272 /* mark_adjust(98, 99, MAXLNUM, -2): delete lines */
273 inserted = 0;
274 deleted = -amount_after;
275 }
276
277 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000278 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000279 for (;;)
280 {
281 /* If the change is after the previous diff block and before the next
282 * diff block, thus not touching an existing change, create a new diff
283 * block. Don't do this when ex_diffgetput() is busy. */
284 if ((dp == NULL || dp->df_lnum[idx] - 1 > line2
285 || (line2 == MAXLNUM && dp->df_lnum[idx] > line1))
286 && (dprev == NULL
287 || dprev->df_lnum[idx] + dprev->df_count[idx] < line1)
288 && !diff_busy)
289 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000290 dnext = diff_alloc_new(tp, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000291 if (dnext == NULL)
292 return;
293
294 dnext->df_lnum[idx] = line1;
295 dnext->df_count[idx] = inserted;
296 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000297 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 {
299 if (dprev == NULL)
300 dnext->df_lnum[i] = line1;
301 else
302 dnext->df_lnum[i] = line1
303 + (dprev->df_lnum[i] + dprev->df_count[i])
304 - (dprev->df_lnum[idx] + dprev->df_count[idx]);
305 dnext->df_count[i] = deleted;
306 }
307 }
308
309 /* if at end of the list, quit */
310 if (dp == NULL)
311 break;
312
313 /*
314 * Check for these situations:
315 * 1 2 3
316 * 1 2 3
317 * line1 2 3 4 5
318 * 2 3 4 5
319 * 2 3 4 5
320 * line2 2 3 4 5
321 * 3 5 6
322 * 3 5 6
323 */
324 /* compute last line of this change */
325 last = dp->df_lnum[idx] + dp->df_count[idx] - 1;
326
327 /* 1. change completely above line1: nothing to do */
328 if (last >= line1 - 1)
329 {
330 /* 6. change below line2: only adjust for amount_after; also when
331 * "deleted" became zero when deleted all lines between two diffs */
332 if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2)
333 {
334 if (amount_after == 0)
335 break; /* nothing left to change */
336 dp->df_lnum[idx] += amount_after;
337 }
338 else
339 {
340 check_unchanged = FALSE;
341
342 /* 2. 3. 4. 5.: inserted/deleted lines touching this diff. */
343 if (deleted > 0)
344 {
345 if (dp->df_lnum[idx] >= line1)
346 {
347 off = dp->df_lnum[idx] - lnum_deleted;
348 if (last <= line2)
349 {
350 /* 4. delete all lines of diff */
351 if (dp->df_next != NULL
352 && dp->df_next->df_lnum[idx] - 1 <= line2)
353 {
354 /* delete continues in next diff, only do
355 * lines until that one */
356 n = dp->df_next->df_lnum[idx] - lnum_deleted;
357 deleted -= n;
358 n -= dp->df_count[idx];
359 lnum_deleted = dp->df_next->df_lnum[idx];
360 }
361 else
362 n = deleted - dp->df_count[idx];
363 dp->df_count[idx] = 0;
364 }
365 else
366 {
367 /* 5. delete lines at or just before top of diff */
368 n = off;
369 dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1;
370 check_unchanged = TRUE;
371 }
372 dp->df_lnum[idx] = line1;
373 }
374 else
375 {
376 off = 0;
377 if (last < line2)
378 {
379 /* 2. delete at end of of diff */
380 dp->df_count[idx] -= last - lnum_deleted + 1;
381 if (dp->df_next != NULL
382 && dp->df_next->df_lnum[idx] - 1 <= line2)
383 {
384 /* delete continues in next diff, only do
385 * lines until that one */
386 n = dp->df_next->df_lnum[idx] - 1 - last;
387 deleted -= dp->df_next->df_lnum[idx]
388 - lnum_deleted;
389 lnum_deleted = dp->df_next->df_lnum[idx];
390 }
391 else
392 n = line2 - last;
393 check_unchanged = TRUE;
394 }
395 else
396 {
397 /* 3. delete lines inside the diff */
398 n = 0;
399 dp->df_count[idx] -= deleted;
400 }
401 }
402
403 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000404 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000405 {
406 dp->df_lnum[i] -= off;
407 dp->df_count[i] += n;
408 }
409 }
410 else
411 {
412 if (dp->df_lnum[idx] <= line1)
413 {
414 /* inserted lines somewhere in this diff */
415 dp->df_count[idx] += inserted;
416 check_unchanged = TRUE;
417 }
418 else
419 /* inserted lines somewhere above this diff */
420 dp->df_lnum[idx] += inserted;
421 }
422
423 if (check_unchanged)
424 /* Check if inserted lines are equal, may reduce the
425 * size of the diff. TODO: also check for equal lines
426 * in the middle and perhaps split the block. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000427 diff_check_unchanged(tp, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000428 }
429 }
430
431 /* check if this block touches the previous one, may merge them. */
432 if (dprev != NULL && dprev->df_lnum[idx] + dprev->df_count[idx]
433 == dp->df_lnum[idx])
434 {
435 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000436 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000437 dprev->df_count[i] += dp->df_count[i];
438 dprev->df_next = dp->df_next;
439 vim_free(dp);
440 dp = dprev->df_next;
441 }
442 else
443 {
444 /* Advance to next entry. */
445 dprev = dp;
446 dp = dp->df_next;
447 }
448 }
449
450 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000451 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000452 while (dp != NULL)
453 {
454 /* All counts are zero, remove this entry. */
455 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000456 if (tp->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000457 break;
458 if (i == DB_COUNT)
459 {
460 dnext = dp->df_next;
461 vim_free(dp);
462 dp = dnext;
463 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000464 tp->tp_first_diff = dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000465 else
466 dprev->df_next = dnext;
467 }
468 else
469 {
470 /* Advance to next entry. */
471 dprev = dp;
472 dp = dp->df_next;
473 }
474
475 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000476
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000477 if (tp == curtab)
478 {
479 diff_redraw(TRUE);
480
481 /* Need to recompute the scroll binding, may remove or add filler
482 * lines (e.g., when adding lines above w_topline). But it's slow when
483 * making many changes, postpone until redrawing. */
484 diff_need_scrollbind = TRUE;
485 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000486}
487
488/*
489 * Allocate a new diff block and link it between "dprev" and "dp".
490 */
491 static diff_T *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100492diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000493{
494 diff_T *dnew;
495
496 dnew = (diff_T *)alloc((unsigned)sizeof(diff_T));
497 if (dnew != NULL)
498 {
499 dnew->df_next = dp;
500 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000501 tp->tp_first_diff = dnew;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000502 else
503 dprev->df_next = dnew;
504 }
505 return dnew;
506}
507
508/*
509 * Check if the diff block "dp" can be made smaller for lines at the start and
510 * end that are equal. Called after inserting lines.
511 * This may result in a change where all buffers have zero lines, the caller
512 * must take care of removing it.
513 */
514 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100515diff_check_unchanged(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000516{
517 int i_org;
518 int i_new;
519 int off_org, off_new;
520 char_u *line_org;
521 int dir = FORWARD;
522
523 /* Find the first buffers, use it as the original, compare the other
524 * buffer lines against this one. */
525 for (i_org = 0; i_org < DB_COUNT; ++i_org)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000526 if (tp->tp_diffbuf[i_org] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000527 break;
528 if (i_org == DB_COUNT) /* safety check */
529 return;
530
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000531 if (diff_check_sanity(tp, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000532 return;
533
534 /* First check lines at the top, then at the bottom. */
535 off_org = 0;
536 off_new = 0;
537 for (;;)
538 {
539 /* Repeat until a line is found which is different or the number of
540 * lines has become zero. */
541 while (dp->df_count[i_org] > 0)
542 {
543 /* Copy the line, the next ml_get() will invalidate it. */
544 if (dir == BACKWARD)
545 off_org = dp->df_count[i_org] - 1;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000546 line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000547 dp->df_lnum[i_org] + off_org, FALSE));
548 if (line_org == NULL)
549 return;
550 for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new)
551 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000552 if (tp->tp_diffbuf[i_new] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000553 continue;
554 if (dir == BACKWARD)
555 off_new = dp->df_count[i_new] - 1;
556 /* if other buffer doesn't have this line, it was inserted */
557 if (off_new < 0 || off_new >= dp->df_count[i_new])
558 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000559 if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000560 dp->df_lnum[i_new] + off_new, FALSE)) != 0)
561 break;
562 }
563 vim_free(line_org);
564
565 /* Stop when a line isn't equal in all diff buffers. */
566 if (i_new != DB_COUNT)
567 break;
568
569 /* Line matched in all buffers, remove it from the diff. */
570 for (i_new = i_org; i_new < DB_COUNT; ++i_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000571 if (tp->tp_diffbuf[i_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000572 {
573 if (dir == FORWARD)
574 ++dp->df_lnum[i_new];
575 --dp->df_count[i_new];
576 }
577 }
578 if (dir == BACKWARD)
579 break;
580 dir = BACKWARD;
581 }
582}
583
584/*
585 * Check if a diff block doesn't contain invalid line numbers.
586 * This can happen when the diff program returns invalid results.
587 */
588 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100589diff_check_sanity(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590{
591 int i;
592
593 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000594 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000595 if (dp->df_lnum[i] + dp->df_count[i] - 1
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000596 > tp->tp_diffbuf[i]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000597 return FAIL;
598 return OK;
599}
600
601/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000602 * Mark all diff buffers in the current tab page for redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000603 */
604 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100605diff_redraw(
606 int dofold) /* also recompute the folds */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000607{
608 win_T *wp;
609 int n;
610
Bram Moolenaar29323592016-07-24 22:04:11 +0200611 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000612 if (wp->w_p_diff)
613 {
Bram Moolenaar60f83772006-03-12 21:52:47 +0000614 redraw_win_later(wp, SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615#ifdef FEAT_FOLDING
616 if (dofold && foldmethodIsDiff(wp))
617 foldUpdateAll(wp);
618#endif
619 /* A change may have made filler lines invalid, need to take care
620 * of that for other windows. */
Bram Moolenaara80888d2012-10-21 22:18:21 +0200621 n = diff_check(wp, wp->w_topline);
622 if ((wp != curwin && wp->w_topfill > 0) || n > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000623 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000624 if (wp->w_topfill > n)
625 wp->w_topfill = (n < 0 ? 0 : n);
Bram Moolenaara80888d2012-10-21 22:18:21 +0200626 else if (n > 0 && n > wp->w_topfill)
627 wp->w_topfill = n;
Bram Moolenaar846a2ff2014-05-28 11:35:37 +0200628 check_topfill(wp, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000629 }
630 }
631}
632
633/*
634 * Write buffer "buf" to file "name".
635 * Always use 'fileformat' set to "unix".
636 * Return FAIL for failure
637 */
638 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100639diff_write(buf_T *buf, char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000640{
641 int r;
642 char_u *save_ff;
643
644 save_ff = buf->b_p_ff;
645 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX);
646 r = buf_write(buf, fname, NULL, (linenr_T)1, buf->b_ml.ml_line_count,
647 NULL, FALSE, FALSE, FALSE, TRUE);
648 free_string_option(buf->b_p_ff);
649 buf->b_p_ff = save_ff;
650 return r;
651}
652
653/*
654 * Completely update the diffs for the buffers involved.
655 * This uses the ordinary "diff" command.
656 * The buffers are written to a file, also for unmodified buffers (the file
657 * could have been produced by autocommands, e.g. the netrw plugin).
658 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100660ex_diffupdate(
Bram Moolenaarf1d25012016-03-03 12:22:53 +0100661 exarg_T *eap) /* can be NULL */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000662{
663 buf_T *buf;
664 int idx_orig;
665 int idx_new;
666 char_u *tmp_orig;
667 char_u *tmp_new;
668 char_u *tmp_diff;
669 FILE *fd;
670 int ok;
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000671 int io_error = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672
673 /* Delete all diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000674 diff_clear(curtab);
675 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000676
677 /* Use the first buffer as the original text. */
678 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000679 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000680 break;
681 if (idx_orig == DB_COUNT)
682 return;
683
684 /* Only need to do something when there is another buffer. */
685 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000686 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000687 break;
688 if (idx_new == DB_COUNT)
689 return;
690
691 /* We need three temp file names. */
Bram Moolenaare5c421c2015-03-31 13:33:08 +0200692 tmp_orig = vim_tempname('o', TRUE);
693 tmp_new = vim_tempname('n', TRUE);
694 tmp_diff = vim_tempname('d', TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000695 if (tmp_orig == NULL || tmp_new == NULL || tmp_diff == NULL)
696 goto theend;
697
698 /*
699 * Do a quick test if "diff" really works. Otherwise it looks like there
700 * are no differences. Can't use the return value, it's non-zero when
701 * there are differences.
702 * May try twice, first with "-a" and then without.
703 */
704 for (;;)
705 {
706 ok = FALSE;
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000707 fd = mch_fopen((char *)tmp_orig, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000708 if (fd == NULL)
709 io_error = TRUE;
710 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000711 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000712 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
713 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000714 fclose(fd);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000715 fd = mch_fopen((char *)tmp_new, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000716 if (fd == NULL)
717 io_error = TRUE;
718 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000719 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000720 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
721 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000722 fclose(fd);
723 diff_file(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000724 fd = mch_fopen((char *)tmp_diff, "r");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000725 if (fd == NULL)
726 io_error = TRUE;
727 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000728 {
729 char_u linebuf[LBUFLEN];
730
731 for (;;)
732 {
733 /* There must be a line that contains "1c1". */
734 if (tag_fgets(linebuf, LBUFLEN, fd))
735 break;
736 if (STRNCMP(linebuf, "1c1", 3) == 0)
737 ok = TRUE;
738 }
739 fclose(fd);
740 }
741 mch_remove(tmp_diff);
742 mch_remove(tmp_new);
743 }
744 mch_remove(tmp_orig);
745 }
746
747#ifdef FEAT_EVAL
748 /* When using 'diffexpr' break here. */
749 if (*p_dex != NUL)
750 break;
751#endif
752
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100753#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754 /* If the "-a" argument works, also check if "--binary" works. */
755 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
756 {
757 diff_a_works = TRUE;
758 diff_bin_works = TRUE;
759 continue;
760 }
761 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
762 {
763 /* Tried --binary, but it failed. "-a" works though. */
764 diff_bin_works = FALSE;
765 ok = TRUE;
766 }
767#endif
768
769 /* If we checked if "-a" works already, break here. */
770 if (diff_a_works != MAYBE)
771 break;
772 diff_a_works = ok;
773
774 /* If "-a" works break here, otherwise retry without "-a". */
775 if (ok)
776 break;
777 }
778 if (!ok)
779 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000780 if (io_error)
781 EMSG(_("E810: Cannot read or write temp files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000782 EMSG(_("E97: Cannot create diffs"));
783 diff_a_works = MAYBE;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100784#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000785 diff_bin_works = MAYBE;
786#endif
787 goto theend;
788 }
789
Bram Moolenaarbd1d5602012-05-18 18:47:17 +0200790 /* :diffupdate! */
791 if (eap != NULL && eap->forceit)
792 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new)
793 {
794 buf = curtab->tp_diffbuf[idx_new];
795 if (buf_valid(buf))
796 buf_check_timestamp(buf, FALSE);
797 }
798
Bram Moolenaar071d4272004-06-13 20:20:40 +0000799 /* Write the first buffer to a tempfile. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000800 buf = curtab->tp_diffbuf[idx_orig];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 if (diff_write(buf, tmp_orig) == FAIL)
802 goto theend;
803
804 /* Make a difference between the first buffer and every other. */
805 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
806 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000807 buf = curtab->tp_diffbuf[idx_new];
Bram Moolenaar9dd33af2015-08-04 21:51:25 +0200808 if (buf == NULL || buf->b_ml.ml_mfp == NULL)
809 continue; /* skip buffer that isn't loaded */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000810 if (diff_write(buf, tmp_new) == FAIL)
811 continue;
812 diff_file(tmp_orig, tmp_new, tmp_diff);
813
814 /* Read the diff output and add each entry to the diff list. */
815 diff_read(idx_orig, idx_new, tmp_diff);
816 mch_remove(tmp_diff);
817 mch_remove(tmp_new);
818 }
819 mch_remove(tmp_orig);
820
Bram Moolenaar60a44dc2007-10-19 16:58:12 +0000821 /* force updating cursor position on screen */
822 curwin->w_valid_cursor.lnum = 0;
823
Bram Moolenaar071d4272004-06-13 20:20:40 +0000824 diff_redraw(TRUE);
825
826theend:
827 vim_free(tmp_orig);
828 vim_free(tmp_new);
829 vim_free(tmp_diff);
830}
831
832/*
833 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
834 */
835 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100836diff_file(
837 char_u *tmp_orig,
838 char_u *tmp_new,
839 char_u *tmp_diff)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000840{
841 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000842 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000843
844#ifdef FEAT_EVAL
845 if (*p_dex != NUL)
846 /* Use 'diffexpr' to generate the diff file. */
847 eval_diff(tmp_orig, tmp_new, tmp_diff);
848 else
849#endif
850 {
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000851 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
852 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
853 cmd = alloc((unsigned)len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000854 if (cmd != NULL)
855 {
Bram Moolenaar95fb60a2005-03-08 22:29:20 +0000856 /* We don't want $DIFF_OPTIONS to get in the way. */
857 if (getenv("DIFF_OPTIONS"))
858 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
859
Bram Moolenaar071d4272004-06-13 20:20:40 +0000860 /* Build the diff command and execute it. Always use -a, binary
861 * differences are of no use. Ignore errors, diff returns
862 * non-zero when differences have been found. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000863 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +0000864 diff_a_works == FALSE ? "" : "-a ",
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100865#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000866 diff_bin_works == TRUE ? "--binary " : "",
867#else
868 "",
869#endif
870 (diff_flags & DIFF_IWHITE) ? "-b " : "",
871 (diff_flags & DIFF_ICASE) ? "-i " : "",
872 tmp_orig, tmp_new);
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000873 append_redir(cmd, (int)len, p_srr, tmp_diff);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000874#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000875 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000876#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000877 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000878#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000879 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000880#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000881 vim_free(cmd);
882 }
883 }
884}
885
886/*
887 * Create a new version of a file from the current buffer and a diff file.
888 * The buffer is written to a file, also for unmodified buffers (the file
889 * could have been produced by autocommands, e.g. the netrw plugin).
890 */
891 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100892ex_diffpatch(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000893{
894 char_u *tmp_orig; /* name of original temp file */
895 char_u *tmp_new; /* name of patched temp file */
896 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000897 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000898 win_T *old_curwin = curwin;
899 char_u *newname = NULL; /* name of patched file buffer */
900#ifdef UNIX
901 char_u dirbuf[MAXPATHL];
902 char_u *fullname = NULL;
903#endif
904#ifdef FEAT_BROWSE
905 char_u *browseFile = NULL;
906 int browse_flag = cmdmod.browse;
907#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +0200908 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909
910#ifdef FEAT_BROWSE
911 if (cmdmod.browse)
912 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +0000913 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaar071d4272004-06-13 20:20:40 +0000914 eap->arg, NULL, NULL, BROWSE_FILTER_ALL_FILES, NULL);
915 if (browseFile == NULL)
916 return; /* operation cancelled */
917 eap->arg = browseFile;
918 cmdmod.browse = FALSE; /* don't let do_ecmd() browse again */
919 }
920#endif
921
922 /* We need two temp file names. */
Bram Moolenaare5c421c2015-03-31 13:33:08 +0200923 tmp_orig = vim_tempname('o', FALSE);
924 tmp_new = vim_tempname('n', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000925 if (tmp_orig == NULL || tmp_new == NULL)
926 goto theend;
927
928 /* Write the current buffer to "tmp_orig". */
929 if (buf_write(curbuf, tmp_orig, NULL,
930 (linenr_T)1, curbuf->b_ml.ml_line_count,
931 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
932 goto theend;
933
934#ifdef UNIX
935 /* Get the absolute path of the patchfile, changing directory below. */
936 fullname = FullName_save(eap->arg, FALSE);
937#endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000938 buflen = STRLEN(tmp_orig) + (
Bram Moolenaar071d4272004-06-13 20:20:40 +0000939# ifdef UNIX
940 fullname != NULL ? STRLEN(fullname) :
941# endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000942 STRLEN(eap->arg)) + STRLEN(tmp_new) + 16;
943 buf = alloc((unsigned)buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000944 if (buf == NULL)
945 goto theend;
946
947#ifdef UNIX
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000948 /* Temporarily chdir to /tmp, to avoid patching files in the current
Bram Moolenaar071d4272004-06-13 20:20:40 +0000949 * directory when the patch file contains more than one patch. When we
950 * have our own temp dir use that instead, it will be cleaned up when we
951 * exit (any .rej files created). Don't change directory if we can't
952 * return to the current. */
953 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
954 dirbuf[0] = NUL;
955 else
956 {
957# ifdef TEMPDIRNAMES
958 if (vim_tempdir != NULL)
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000959 ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000960 else
961# endif
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000962 ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +0000963 shorten_fnames(TRUE);
964 }
965#endif
966
967#ifdef FEAT_EVAL
968 if (*p_pex != NUL)
969 /* Use 'patchexpr' to generate the new file. */
970 eval_patch(tmp_orig,
971# ifdef UNIX
972 fullname != NULL ? fullname :
973# endif
974 eap->arg, tmp_new);
975 else
976#endif
977 {
978 /* Build the patch command and execute it. Ignore errors. Switch to
979 * cooked mode to allow the user to respond to prompts. */
Bram Moolenaar1ef73e32017-03-09 19:21:30 +0100980 vim_snprintf((char *)buf, buflen,
981#ifdef UNIX
982 "patch -o %s %s < '%s'",
983#else
984 "patch -o %s %s < \"%s\"",
985#endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000986 tmp_new, tmp_orig,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000987# ifdef UNIX
988 fullname != NULL ? fullname :
989# endif
990 eap->arg);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000991#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000992 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000993#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000994 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000995#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000996 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000997#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000998 }
999
1000#ifdef UNIX
1001 if (dirbuf[0] != NUL)
1002 {
1003 if (mch_chdir((char *)dirbuf) != 0)
1004 EMSG(_(e_prev_dir));
1005 shorten_fnames(TRUE);
1006 }
1007#endif
1008
1009 /* patch probably has written over the screen */
1010 redraw_later(CLEAR);
1011
1012 /* Delete any .orig or .rej file created. */
1013 STRCPY(buf, tmp_new);
1014 STRCAT(buf, ".orig");
1015 mch_remove(buf);
1016 STRCPY(buf, tmp_new);
1017 STRCAT(buf, ".rej");
1018 mch_remove(buf);
1019
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001020 /* Only continue if the output file was created. */
1021 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
1022 EMSG(_("E816: Cannot read patch output"));
1023 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001024 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001025 if (curbuf->b_fname != NULL)
1026 {
1027 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001028 (int)(STRLEN(curbuf->b_fname) + 4));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001029 if (newname != NULL)
1030 STRCAT(newname, ".new");
1031 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001032
1033#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001034 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001035#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001036 /* don't use a new tab page, each tab page has its own diffs */
1037 cmdmod.tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001038
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001039 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001040 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001041 /* Pretend it was a ":split fname" command */
1042 eap->cmdidx = CMD_split;
1043 eap->arg = tmp_new;
1044 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001045
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001046 /* check that split worked and editing tmp_new */
1047 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001048 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001049 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1050 diff_win_options(curwin, TRUE);
1051 diff_win_options(old_curwin, TRUE);
1052
1053 if (newname != NULL)
1054 {
1055 /* do a ":file filename.new" on the patched buffer */
1056 eap->arg = newname;
1057 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001058
1059#ifdef FEAT_AUTOCMD
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001060 /* Do filetype detection with the new name. */
1061 if (au_has_group((char_u *)"filetypedetect"))
1062 do_cmdline_cmd((char_u *)":doau filetypedetect BufRead");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001063#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001064 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001065 }
1066 }
1067 }
1068
1069theend:
1070 if (tmp_orig != NULL)
1071 mch_remove(tmp_orig);
1072 vim_free(tmp_orig);
1073 if (tmp_new != NULL)
1074 mch_remove(tmp_new);
1075 vim_free(tmp_new);
1076 vim_free(newname);
1077 vim_free(buf);
1078#ifdef UNIX
1079 vim_free(fullname);
1080#endif
1081#ifdef FEAT_BROWSE
1082 vim_free(browseFile);
1083 cmdmod.browse = browse_flag;
1084#endif
1085}
1086
1087/*
1088 * Split the window and edit another file, setting options to show the diffs.
1089 */
1090 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001091ex_diffsplit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001092{
1093 win_T *old_curwin = curwin;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001094 bufref_T old_curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001095
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001096 set_bufref(&old_curbuf, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001097#ifdef FEAT_GUI
1098 need_mouse_correct = TRUE;
1099#endif
Bram Moolenaar46328f92016-08-28 15:39:57 +02001100 /* Need to compute w_fraction when no redraw happened yet. */
1101 validate_cursor();
1102 set_fraction(curwin);
1103
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001104 /* don't use a new tab page, each tab page has its own diffs */
1105 cmdmod.tab = 0;
1106
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001107 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001108 {
1109 /* Pretend it was a ":split fname" command */
1110 eap->cmdidx = CMD_split;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001111 curwin->w_p_diff = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001112 do_exedit(eap, old_curwin);
1113
1114 if (curwin != old_curwin) /* split must have worked */
1115 {
1116 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1117 diff_win_options(curwin, TRUE);
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001118 if (win_valid(old_curwin))
1119 {
1120 diff_win_options(old_curwin, TRUE);
1121
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001122 if (bufref_valid(&old_curbuf))
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001123 /* Move the cursor position to that of the old window. */
1124 curwin->w_cursor.lnum = diff_get_corresponding_line(
Bram Moolenaar025e3e02016-10-18 14:50:18 +02001125 old_curbuf.br_buf, old_curwin->w_cursor.lnum);
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001126 }
Bram Moolenaar46328f92016-08-28 15:39:57 +02001127 /* Now that lines are folded scroll to show the cursor at the same
1128 * relative position. */
1129 scroll_to_fraction(curwin, curwin->w_height);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001130 }
1131 }
1132}
1133
1134/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001135 * Set options to show diffs for the current window.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001136 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001137 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001138ex_diffthis(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001139{
1140 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1141 diff_win_options(curwin, TRUE);
1142}
1143
1144/*
1145 * Set options in window "wp" for diff mode.
1146 */
1147 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001148diff_win_options(
1149 win_T *wp,
1150 int addbuf) /* Add buffer to diff. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001151{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001152# ifdef FEAT_FOLDING
1153 win_T *old_curwin = curwin;
1154
1155 /* close the manually opened folds */
1156 curwin = wp;
1157 newFoldLevel();
1158 curwin = old_curwin;
1159# endif
1160
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001161 /* Use 'scrollbind' and 'cursorbind' when available */
1162#ifdef FEAT_SCROLLBIND
Bram Moolenaar43929962015-07-03 15:06:56 +02001163 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001164 wp->w_p_scb_save = wp->w_p_scb;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001165 wp->w_p_scb = TRUE;
1166#endif
Bram Moolenaar860cae12010-06-05 23:22:07 +02001167#ifdef FEAT_CURSORBIND
Bram Moolenaar43929962015-07-03 15:06:56 +02001168 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001169 wp->w_p_crb_save = wp->w_p_crb;
Bram Moolenaar860cae12010-06-05 23:22:07 +02001170 wp->w_p_crb = TRUE;
1171#endif
Bram Moolenaar43929962015-07-03 15:06:56 +02001172 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001173 wp->w_p_wrap_save = wp->w_p_wrap;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001174 wp->w_p_wrap = FALSE;
1175# ifdef FEAT_FOLDING
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001176 curwin = wp;
1177 curbuf = curwin->w_buffer;
Bram Moolenaar43929962015-07-03 15:06:56 +02001178 if (!wp->w_p_diff)
1179 {
1180 if (wp->w_p_diff_saved)
1181 free_string_option(wp->w_p_fdm_save);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001182 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm);
Bram Moolenaar43929962015-07-03 15:06:56 +02001183 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001184 set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001185 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001186 curwin = old_curwin;
1187 curbuf = curwin->w_buffer;
Bram Moolenaar43929962015-07-03 15:06:56 +02001188 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001189 {
1190 wp->w_p_fdc_save = wp->w_p_fdc;
1191 wp->w_p_fen_save = wp->w_p_fen;
1192 wp->w_p_fdl_save = wp->w_p_fdl;
1193 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001194 wp->w_p_fdc = diff_foldcolumn;
1195 wp->w_p_fen = TRUE;
1196 wp->w_p_fdl = 0;
1197 foldUpdateAll(wp);
1198 /* make sure topline is not halfway a fold */
1199 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001200# endif
1201#ifdef FEAT_SCROLLBIND
1202 if (vim_strchr(p_sbo, 'h') == NULL)
1203 do_cmdline_cmd((char_u *)"set sbo+=hor");
1204#endif
Bram Moolenaara87aa802013-07-03 15:47:03 +02001205 /* Saved the current values, to be restored in ex_diffoff(). */
1206 wp->w_p_diff_saved = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001207
Bram Moolenaar43929962015-07-03 15:06:56 +02001208 wp->w_p_diff = TRUE;
1209
Bram Moolenaar071d4272004-06-13 20:20:40 +00001210 if (addbuf)
1211 diff_buf_add(wp->w_buffer);
1212 redraw_win_later(wp, NOT_VALID);
1213}
1214
1215/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001216 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001217 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001218 */
1219 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001220ex_diffoff(exarg_T *eap)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001221{
1222 win_T *wp;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001223#ifdef FEAT_SCROLLBIND
1224 int diffwin = FALSE;
1225#endif
1226
Bram Moolenaar29323592016-07-24 22:04:11 +02001227 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001228 {
Bram Moolenaar00462ff2013-09-20 20:13:53 +02001229 if (eap->forceit ? wp->w_p_diff : wp == curwin)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001230 {
Bram Moolenaar43929962015-07-03 15:06:56 +02001231 /* Set 'diff' off. If option values were saved in
1232 * diff_win_options(), restore the ones whose settings seem to have
1233 * been left over from diff mode. */
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001234 wp->w_p_diff = FALSE;
Bram Moolenaara87aa802013-07-03 15:47:03 +02001235
Bram Moolenaara87aa802013-07-03 15:47:03 +02001236 if (wp->w_p_diff_saved)
1237 {
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001238
Bram Moolenaar43929962015-07-03 15:06:56 +02001239#ifdef FEAT_SCROLLBIND
1240 if (wp->w_p_scb)
1241 wp->w_p_scb = wp->w_p_scb_save;
1242#endif
1243#ifdef FEAT_CURSORBIND
1244 if (wp->w_p_crb)
1245 wp->w_p_crb = wp->w_p_crb_save;
1246#endif
1247 if (!wp->w_p_wrap)
1248 wp->w_p_wrap = wp->w_p_wrap_save;
1249#ifdef FEAT_FOLDING
1250 free_string_option(wp->w_p_fdm);
1251 wp->w_p_fdm = vim_strsave(wp->w_p_fdm_save);
1252
1253 if (wp->w_p_fdc == diff_foldcolumn)
1254 wp->w_p_fdc = wp->w_p_fdc_save;
1255 if (wp->w_p_fdl == 0)
1256 wp->w_p_fdl = wp->w_p_fdl_save;
1257
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001258 /* Only restore 'foldenable' when 'foldmethod' is not
1259 * "manual", otherwise we continue to show the diff folds. */
Bram Moolenaar43929962015-07-03 15:06:56 +02001260 if (wp->w_p_fen)
1261 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE
1262 : wp->w_p_fen_save;
1263
1264 foldUpdateAll(wp);
Bram Moolenaar43929962015-07-03 15:06:56 +02001265#endif
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001266 }
Bram Moolenaare67d5462016-08-27 22:40:42 +02001267 /* remove filler lines */
1268 wp->w_topfill = 0;
1269
1270 /* make sure topline is not halfway a fold and cursor is
1271 * invalidated */
1272 changed_window_setting_win(wp);
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001273
Bram Moolenaara87aa802013-07-03 15:47:03 +02001274 /* Note: 'sbo' is not restored, it's a global option. */
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001275 diff_buf_adjust(wp);
1276 }
1277#ifdef FEAT_SCROLLBIND
1278 diffwin |= wp->w_p_diff;
1279#endif
1280 }
1281
Bram Moolenaar25ea0542017-02-03 23:16:28 +01001282 /* Also remove hidden buffers from the list. */
1283 if (eap->forceit)
1284 diff_buf_clear();
1285
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001286#ifdef FEAT_SCROLLBIND
1287 /* Remove "hor" from from 'scrollopt' if there are no diff windows left. */
1288 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1289 do_cmdline_cmd((char_u *)"set sbo-=hor");
1290#endif
1291}
1292
1293/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001294 * Read the diff output and add each entry to the diff list.
1295 */
1296 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001297diff_read(
1298 int idx_orig, /* idx of original file */
1299 int idx_new, /* idx of new file */
1300 char_u *fname) /* name of diff output file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001301{
1302 FILE *fd;
1303 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001304 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001305 diff_T *dn, *dpl;
1306 long f1, l1, f2, l2;
1307 char_u linebuf[LBUFLEN]; /* only need to hold the diff line */
1308 int difftype;
1309 char_u *p;
1310 long off;
1311 int i;
1312 linenr_T lnum_orig, lnum_new;
1313 long count_orig, count_new;
1314 int notset = TRUE; /* block "*dp" not set yet */
1315
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001316 fd = mch_fopen((char *)fname, "r");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001317 if (fd == NULL)
1318 {
1319 EMSG(_("E98: Cannot read diff output"));
1320 return;
1321 }
1322
1323 for (;;)
1324 {
1325 if (tag_fgets(linebuf, LBUFLEN, fd))
1326 break; /* end of file */
1327 if (!isdigit(*linebuf))
1328 continue; /* not the start of a diff block */
1329
1330 /* This line must be one of three formats:
1331 * {first}[,{last}]c{first}[,{last}]
1332 * {first}a{first}[,{last}]
1333 * {first}[,{last}]d{first}
1334 */
1335 p = linebuf;
1336 f1 = getdigits(&p);
1337 if (*p == ',')
1338 {
1339 ++p;
1340 l1 = getdigits(&p);
1341 }
1342 else
1343 l1 = f1;
1344 if (*p != 'a' && *p != 'c' && *p != 'd')
1345 continue; /* invalid diff format */
1346 difftype = *p++;
1347 f2 = getdigits(&p);
1348 if (*p == ',')
1349 {
1350 ++p;
1351 l2 = getdigits(&p);
1352 }
1353 else
1354 l2 = f2;
1355 if (l1 < f1 || l2 < f2)
1356 continue; /* invalid line range */
1357
1358 if (difftype == 'a')
1359 {
1360 lnum_orig = f1 + 1;
1361 count_orig = 0;
1362 }
1363 else
1364 {
1365 lnum_orig = f1;
1366 count_orig = l1 - f1 + 1;
1367 }
1368 if (difftype == 'd')
1369 {
1370 lnum_new = f2 + 1;
1371 count_new = 0;
1372 }
1373 else
1374 {
1375 lnum_new = f2;
1376 count_new = l2 - f2 + 1;
1377 }
1378
1379 /* Go over blocks before the change, for which orig and new are equal.
1380 * Copy blocks from orig to new. */
1381 while (dp != NULL
1382 && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig])
1383 {
1384 if (notset)
1385 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1386 dprev = dp;
1387 dp = dp->df_next;
1388 notset = TRUE;
1389 }
1390
1391 if (dp != NULL
1392 && lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig]
1393 && lnum_orig + count_orig >= dp->df_lnum[idx_orig])
1394 {
1395 /* New block overlaps with existing block(s).
1396 * First find last block that overlaps. */
1397 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
1398 if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig])
1399 break;
1400
1401 /* If the newly found block starts before the old one, set the
1402 * start back a number of lines. */
1403 off = dp->df_lnum[idx_orig] - lnum_orig;
1404 if (off > 0)
1405 {
1406 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001407 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001408 dp->df_lnum[i] -= off;
1409 dp->df_lnum[idx_new] = lnum_new;
1410 dp->df_count[idx_new] = count_new;
1411 }
1412 else if (notset)
1413 {
1414 /* new block inside existing one, adjust new block */
1415 dp->df_lnum[idx_new] = lnum_new + off;
1416 dp->df_count[idx_new] = count_new - off;
1417 }
1418 else
1419 /* second overlap of new block with existing block */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001420 dp->df_count[idx_new] += count_new - count_orig
1421 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1422 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001423
1424 /* Adjust the size of the block to include all the lines to the
1425 * end of the existing block or the new diff, whatever ends last. */
1426 off = (lnum_orig + count_orig)
1427 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1428 if (off < 0)
1429 {
1430 /* new change ends in existing block, adjust the end if not
1431 * done already */
1432 if (notset)
1433 dp->df_count[idx_new] += -off;
1434 off = 0;
1435 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001436 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001437 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001438 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1439 - dp->df_lnum[i] + off;
1440
1441 /* Delete the diff blocks that have been merged into one. */
1442 dn = dp->df_next;
1443 dp->df_next = dpl->df_next;
1444 while (dn != dp->df_next)
1445 {
1446 dpl = dn->df_next;
1447 vim_free(dn);
1448 dn = dpl;
1449 }
1450 }
1451 else
1452 {
1453 /* Allocate a new diffblock. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001454 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001455 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001456 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001457
1458 dp->df_lnum[idx_orig] = lnum_orig;
1459 dp->df_count[idx_orig] = count_orig;
1460 dp->df_lnum[idx_new] = lnum_new;
1461 dp->df_count[idx_new] = count_new;
1462
1463 /* Set values for other buffers, these must be equal to the
1464 * original buffer, otherwise there would have been a change
1465 * already. */
1466 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001467 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001468 diff_copy_entry(dprev, dp, idx_orig, i);
1469 }
1470 notset = FALSE; /* "*dp" has been set */
1471 }
1472
1473 /* for remaining diff blocks orig and new are equal */
1474 while (dp != NULL)
1475 {
1476 if (notset)
1477 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1478 dprev = dp;
1479 dp = dp->df_next;
1480 notset = TRUE;
1481 }
1482
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001483done:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001484 fclose(fd);
1485}
1486
1487/*
1488 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1489 */
1490 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001491diff_copy_entry(
1492 diff_T *dprev,
1493 diff_T *dp,
1494 int idx_orig,
1495 int idx_new)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001496{
1497 long off;
1498
1499 if (dprev == NULL)
1500 off = 0;
1501 else
1502 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1503 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1504 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1505 dp->df_count[idx_new] = dp->df_count[idx_orig];
1506}
1507
1508/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001509 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001510 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001511 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001512diff_clear(tabpage_T *tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001513{
1514 diff_T *p, *next_p;
1515
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001516 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001517 {
1518 next_p = p->df_next;
1519 vim_free(p);
1520 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001521 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001522}
1523
1524/*
1525 * Check diff status for line "lnum" in buffer "buf":
1526 * Returns 0 for nothing special
1527 * Returns -1 for a line that should be highlighted as changed.
1528 * Returns -2 for a line that should be highlighted as added/deleted.
1529 * Returns > 0 for inserting that many filler lines above it (never happens
1530 * when 'diffopt' doesn't contain "filler").
1531 * This should only be used for windows where 'diff' is set.
1532 */
1533 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001534diff_check(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001535{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001536 int idx; /* index in tp_diffbuf[] for this buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001537 diff_T *dp;
1538 int maxcount;
1539 int i;
1540 buf_T *buf = wp->w_buffer;
1541 int cmp;
1542
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001543 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001544 ex_diffupdate(NULL); /* update after a big change */
1545
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001546 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) /* no diffs at all */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001547 return 0;
1548
1549 /* safety check: "lnum" must be a buffer line */
1550 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1551 return 0;
1552
1553 idx = diff_buf_idx(buf);
1554 if (idx == DB_COUNT)
1555 return 0; /* no diffs for buffer "buf" */
1556
1557#ifdef FEAT_FOLDING
1558 /* A closed fold never has filler lines. */
1559 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1560 return 0;
1561#endif
1562
1563 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001564 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001565 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1566 break;
1567 if (dp == NULL || lnum < dp->df_lnum[idx])
1568 return 0;
1569
1570 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1571 {
1572 int zero = FALSE;
1573
1574 /* Changed or inserted line. If the other buffers have a count of
1575 * zero, the lines were inserted. If the other buffers have the same
1576 * count, check if the lines are identical. */
1577 cmp = FALSE;
1578 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001579 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001580 {
1581 if (dp->df_count[i] == 0)
1582 zero = TRUE;
1583 else
1584 {
1585 if (dp->df_count[i] != dp->df_count[idx])
1586 return -1; /* nr of lines changed. */
1587 cmp = TRUE;
1588 }
1589 }
1590 if (cmp)
1591 {
1592 /* Compare all lines. If they are equal the lines were inserted
1593 * in some buffers, deleted in others, but not changed. */
1594 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001595 if (i != idx && curtab->tp_diffbuf[i] != NULL
1596 && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001597 if (!diff_equal_entry(dp, idx, i))
1598 return -1;
1599 }
1600 /* If there is no buffer with zero lines then there is no difference
1601 * any longer. Happens when making a change (or undo) that removes
1602 * the difference. Can't remove the entry here, we might be halfway
1603 * updating the window. Just report the text as unchanged. Other
1604 * windows might still show the change though. */
1605 if (zero == FALSE)
1606 return 0;
1607 return -2;
1608 }
1609
1610 /* If 'diffopt' doesn't contain "filler", return 0. */
1611 if (!(diff_flags & DIFF_FILLER))
1612 return 0;
1613
1614 /* Insert filler lines above the line just below the change. Will return
1615 * 0 when this buf had the max count. */
1616 maxcount = 0;
1617 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001618 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001619 maxcount = dp->df_count[i];
1620 return maxcount - dp->df_count[idx];
1621}
1622
1623/*
1624 * Compare two entries in diff "*dp" and return TRUE if they are equal.
1625 */
1626 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001627diff_equal_entry(diff_T *dp, int idx1, int idx2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001628{
1629 int i;
1630 char_u *line;
1631 int cmp;
1632
1633 if (dp->df_count[idx1] != dp->df_count[idx2])
1634 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001635 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001636 return FALSE;
1637 for (i = 0; i < dp->df_count[idx1]; ++i)
1638 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001639 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001640 dp->df_lnum[idx1] + i, FALSE));
1641 if (line == NULL)
1642 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001643 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001644 dp->df_lnum[idx2] + i, FALSE));
1645 vim_free(line);
1646 if (cmp != 0)
1647 return FALSE;
1648 }
1649 return TRUE;
1650}
1651
1652/*
1653 * Compare strings "s1" and "s2" according to 'diffopt'.
1654 * Return non-zero when they are different.
1655 */
1656 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001657diff_cmp(char_u *s1, char_u *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001658{
1659 char_u *p1, *p2;
1660#ifdef FEAT_MBYTE
1661 int l;
1662#endif
1663
1664 if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0)
1665 return STRCMP(s1, s2);
1666 if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE))
1667 return MB_STRICMP(s1, s2);
1668
1669 /* Ignore white space changes and possibly ignore case. */
1670 p1 = s1;
1671 p2 = s2;
1672 while (*p1 != NUL && *p2 != NUL)
1673 {
1674 if (vim_iswhite(*p1) && vim_iswhite(*p2))
1675 {
1676 p1 = skipwhite(p1);
1677 p2 = skipwhite(p2);
1678 }
1679 else
1680 {
1681#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001682 l = (*mb_ptr2len)(p1);
1683 if (l != (*mb_ptr2len)(p2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001684 break;
1685 if (l > 1)
1686 {
1687 if (STRNCMP(p1, p2, l) != 0
1688 && (!enc_utf8
1689 || !(diff_flags & DIFF_ICASE)
1690 || utf_fold(utf_ptr2char(p1))
1691 != utf_fold(utf_ptr2char(p2))))
1692 break;
1693 p1 += l;
1694 p2 += l;
1695 }
1696 else
1697#endif
1698 {
1699 if (*p1 != *p2 && (!(diff_flags & DIFF_ICASE)
1700 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
1701 break;
1702 ++p1;
1703 ++p2;
1704 }
1705 }
1706 }
1707
1708 /* Ignore trailing white space. */
1709 p1 = skipwhite(p1);
1710 p2 = skipwhite(p2);
1711 if (*p1 != NUL || *p2 != NUL)
1712 return 1;
1713 return 0;
1714}
1715
1716/*
1717 * Return the number of filler lines above "lnum".
1718 */
1719 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001720diff_check_fill(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001721{
1722 int n;
1723
1724 /* be quick when there are no filler lines */
1725 if (!(diff_flags & DIFF_FILLER))
1726 return 0;
1727 n = diff_check(wp, lnum);
1728 if (n <= 0)
1729 return 0;
1730 return n;
1731}
1732
1733/*
1734 * Set the topline of "towin" to match the position in "fromwin", so that they
1735 * show the same diff'ed lines.
1736 */
1737 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001738diff_set_topline(win_T *fromwin, win_T *towin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001739{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001740 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001741 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001742 int fromidx;
1743 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001744 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001745 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001746 int i;
1747
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001748 fromidx = diff_buf_idx(frombuf);
1749 if (fromidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001750 return; /* safety check */
1751
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001752 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001753 ex_diffupdate(NULL); /* update after a big change */
1754
1755 towin->w_topfill = 0;
1756
1757 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001758 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001759 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001760 break;
1761 if (dp == NULL)
1762 {
1763 /* After last change, compute topline relative to end of file; no
1764 * filler lines. */
1765 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001766 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001767 }
1768 else
1769 {
1770 /* Find index for "towin". */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001771 toidx = diff_buf_idx(towin->w_buffer);
1772 if (toidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001773 return; /* safety check */
1774
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001775 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
1776 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001777 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001778 /* Inside a change: compute filler lines. With three or more
1779 * buffers we need to know the largest count. */
1780 max_count = 0;
1781 for (i = 0; i < DB_COUNT; ++i)
1782 if (curtab->tp_diffbuf[i] != NULL
1783 && max_count < dp->df_count[i])
1784 max_count = dp->df_count[i];
1785
1786 if (dp->df_count[toidx] == dp->df_count[fromidx])
1787 {
1788 /* same number of lines: use same filler count */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001789 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001790 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001791 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001792 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001793 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001794 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001795 /* more lines in towin and fromwin doesn't show diff
1796 * lines, only filler lines */
1797 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
1798 {
1799 /* towin also only shows filler lines */
1800 towin->w_topline = dp->df_lnum[toidx]
1801 + dp->df_count[toidx];
1802 towin->w_topfill = fromwin->w_topfill;
1803 }
1804 else
1805 /* towin still has some diff lines to show */
1806 towin->w_topline = dp->df_lnum[toidx]
1807 + max_count - fromwin->w_topfill;
1808 }
1809 }
1810 else if (towin->w_topline >= dp->df_lnum[toidx]
1811 + dp->df_count[toidx])
1812 {
1813 /* less lines in towin and no diff lines to show: compute
1814 * filler lines */
1815 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
1816 if (diff_flags & DIFF_FILLER)
1817 {
1818 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
1819 /* fromwin is also out of diff lines */
1820 towin->w_topfill = fromwin->w_topfill;
1821 else
1822 /* fromwin has some diff lines */
1823 towin->w_topfill = dp->df_lnum[fromidx]
1824 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001825 }
1826 }
1827 }
1828 }
1829
1830 /* safety check (if diff info gets outdated strange things may happen) */
1831 towin->w_botfill = FALSE;
1832 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
1833 {
1834 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
1835 towin->w_botfill = TRUE;
1836 }
1837 if (towin->w_topline < 1)
1838 {
1839 towin->w_topline = 1;
1840 towin->w_topfill = 0;
1841 }
1842
1843 /* When w_topline changes need to recompute w_botline and cursor position */
1844 invalidate_botline_win(towin);
1845 changed_line_abv_curs_win(towin);
1846
1847 check_topfill(towin, FALSE);
1848#ifdef FEAT_FOLDING
1849 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
1850 NULL, TRUE, NULL);
1851#endif
1852}
1853
1854/*
1855 * This is called when 'diffopt' is changed.
1856 */
1857 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001858diffopt_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859{
1860 char_u *p;
1861 int diff_context_new = 6;
1862 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001863 int diff_foldcolumn_new = 2;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001864 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865
1866 p = p_dip;
1867 while (*p != NUL)
1868 {
1869 if (STRNCMP(p, "filler", 6) == 0)
1870 {
1871 p += 6;
1872 diff_flags_new |= DIFF_FILLER;
1873 }
1874 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
1875 {
1876 p += 8;
1877 diff_context_new = getdigits(&p);
1878 }
1879 else if (STRNCMP(p, "icase", 5) == 0)
1880 {
1881 p += 5;
1882 diff_flags_new |= DIFF_ICASE;
1883 }
1884 else if (STRNCMP(p, "iwhite", 6) == 0)
1885 {
1886 p += 6;
1887 diff_flags_new |= DIFF_IWHITE;
1888 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001889 else if (STRNCMP(p, "horizontal", 10) == 0)
1890 {
1891 p += 10;
1892 diff_flags_new |= DIFF_HORIZONTAL;
1893 }
1894 else if (STRNCMP(p, "vertical", 8) == 0)
1895 {
1896 p += 8;
1897 diff_flags_new |= DIFF_VERTICAL;
1898 }
1899 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
1900 {
1901 p += 11;
1902 diff_foldcolumn_new = getdigits(&p);
1903 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904 if (*p != ',' && *p != NUL)
1905 return FAIL;
1906 if (*p == ',')
1907 ++p;
1908 }
1909
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001910 /* Can't have both "horizontal" and "vertical". */
1911 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
1912 return FAIL;
1913
Bram Moolenaar071d4272004-06-13 20:20:40 +00001914 /* If "icase" or "iwhite" was added or removed, need to update the diff. */
1915 if (diff_flags != diff_flags_new)
Bram Moolenaar29323592016-07-24 22:04:11 +02001916 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001917 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001918
1919 diff_flags = diff_flags_new;
1920 diff_context = diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001921 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001922
1923 diff_redraw(TRUE);
1924
1925 /* recompute the scroll binding with the new option value, may
1926 * remove or add filler lines */
1927 check_scrollbind((linenr_T)0, 0L);
1928
1929 return OK;
1930}
1931
1932/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001933 * Return TRUE if 'diffopt' contains "horizontal".
1934 */
1935 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001936diffopt_horizontal(void)
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001937{
1938 return (diff_flags & DIFF_HORIZONTAL) != 0;
1939}
1940
1941/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001942 * Find the difference within a changed line.
1943 * Returns TRUE if the line was added, no other buffer has it.
1944 */
1945 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001946diff_find_change(
1947 win_T *wp,
1948 linenr_T lnum,
1949 int *startp, /* first char of the change */
1950 int *endp) /* last char of the change */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951{
1952 char_u *line_org;
1953 char_u *line_new;
1954 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001955 int si_org, si_new;
1956 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001957 diff_T *dp;
1958 int idx;
1959 int off;
1960 int added = TRUE;
1961
1962 /* Make a copy of the line, the next ml_get() will invalidate it. */
1963 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
1964 if (line_org == NULL)
1965 return FALSE;
1966
1967 idx = diff_buf_idx(wp->w_buffer);
1968 if (idx == DB_COUNT) /* cannot happen */
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001969 {
1970 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001971 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001972 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001973
1974 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001975 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001976 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1977 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001978 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001979 {
1980 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001981 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001982 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001983
1984 off = lnum - dp->df_lnum[idx];
1985
1986 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001987 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001988 {
1989 /* Skip lines that are not in the other change (filler lines). */
1990 if (off >= dp->df_count[i])
1991 continue;
1992 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001993 line_new = ml_get_buf(curtab->tp_diffbuf[i],
1994 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001995
1996 /* Search for start of difference */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001997 si_org = si_new = 0;
1998 while (line_org[si_org] != NUL)
1999 {
2000 if ((diff_flags & DIFF_IWHITE)
2001 && vim_iswhite(line_org[si_org])
2002 && vim_iswhite(line_new[si_new]))
2003 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002004 si_org = (int)(skipwhite(line_org + si_org) - line_org);
2005 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002006 }
2007 else
2008 {
2009 if (line_org[si_org] != line_new[si_new])
2010 break;
2011 ++si_org;
2012 ++si_new;
2013 }
2014 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002015#ifdef FEAT_MBYTE
2016 if (has_mbyte)
2017 {
2018 /* Move back to first byte of character in both lines (may
2019 * have "nn^" in line_org and "n^ in line_new). */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002020 si_org -= (*mb_head_off)(line_org, line_org + si_org);
2021 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002022 }
2023#endif
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002024 if (*startp > si_org)
2025 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002026
2027 /* Search for end of difference, if any. */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002028 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002029 {
2030 ei_org = (int)STRLEN(line_org);
2031 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002032 while (ei_org >= *startp && ei_new >= si_new
2033 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002034 {
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002035 if ((diff_flags & DIFF_IWHITE)
2036 && vim_iswhite(line_org[ei_org])
2037 && vim_iswhite(line_new[ei_new]))
2038 {
2039 while (ei_org >= *startp
2040 && vim_iswhite(line_org[ei_org]))
2041 --ei_org;
2042 while (ei_new >= si_new
2043 && vim_iswhite(line_new[ei_new]))
2044 --ei_new;
2045 }
2046 else
2047 {
2048 if (line_org[ei_org] != line_new[ei_new])
2049 break;
2050 --ei_org;
2051 --ei_new;
2052 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002053 }
2054 if (*endp < ei_org)
2055 *endp = ei_org;
2056 }
2057 }
2058
2059 vim_free(line_org);
2060 return added;
2061}
2062
2063#if defined(FEAT_FOLDING) || defined(PROTO)
2064/*
2065 * Return TRUE if line "lnum" is not close to a diff block, this line should
2066 * be in a fold.
2067 * Return FALSE if there are no diff blocks at all in this window.
2068 */
2069 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002070diff_infold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071{
2072 int i;
2073 int idx = -1;
2074 int other = FALSE;
2075 diff_T *dp;
2076
2077 /* Return if 'diff' isn't set. */
2078 if (!wp->w_p_diff)
2079 return FALSE;
2080
2081 for (i = 0; i < DB_COUNT; ++i)
2082 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002083 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002085 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002086 other = TRUE;
2087 }
2088
2089 /* return here if there are no diffs in the window */
2090 if (idx == -1 || !other)
2091 return FALSE;
2092
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002093 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002094 ex_diffupdate(NULL); /* update after a big change */
2095
2096 /* Return if there are no diff blocks. All lines will be folded. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002097 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002098 return TRUE;
2099
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002100 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002101 {
2102 /* If this change is below the line there can't be any further match. */
2103 if (dp->df_lnum[idx] - diff_context > lnum)
2104 break;
2105 /* If this change ends before the line we have a match. */
2106 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2107 return FALSE;
2108 }
2109 return TRUE;
2110}
2111#endif
2112
2113/*
2114 * "dp" and "do" commands.
2115 */
2116 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002117nv_diffgetput(int put, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002118{
2119 exarg_T ea;
Bram Moolenaar6a643652014-10-31 13:54:25 +01002120 char_u buf[30];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002121
Bram Moolenaar6a643652014-10-31 13:54:25 +01002122 if (count == 0)
2123 ea.arg = (char_u *)"";
2124 else
2125 {
2126 vim_snprintf((char *)buf, 30, "%ld", count);
2127 ea.arg = buf;
2128 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002129 if (put)
2130 ea.cmdidx = CMD_diffput;
2131 else
2132 ea.cmdidx = CMD_diffget;
2133 ea.addr_count = 0;
2134 ea.line1 = curwin->w_cursor.lnum;
2135 ea.line2 = curwin->w_cursor.lnum;
2136 ex_diffgetput(&ea);
2137}
2138
2139/*
2140 * ":diffget"
2141 * ":diffput"
2142 */
2143 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002144ex_diffgetput(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002145{
2146 linenr_T lnum;
2147 int count;
2148 linenr_T off = 0;
2149 diff_T *dp;
2150 diff_T *dprev;
2151 diff_T *dfree;
2152 int idx_cur;
2153 int idx_other;
2154 int idx_from;
2155 int idx_to;
2156 int i;
2157 int added;
2158 char_u *p;
2159 aco_save_T aco;
2160 buf_T *buf;
2161 int start_skip, end_skip;
2162 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002163 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002164 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002165
2166 /* Find the current buffer in the list of diff buffers. */
2167 idx_cur = diff_buf_idx(curbuf);
2168 if (idx_cur == DB_COUNT)
2169 {
2170 EMSG(_("E99: Current buffer is not in diff mode"));
2171 return;
2172 }
2173
2174 if (*eap->arg == NUL)
2175 {
2176 /* No argument: Find the other buffer in the list of diff buffers. */
2177 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002178 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002179 && curtab->tp_diffbuf[idx_other] != NULL)
2180 {
2181 if (eap->cmdidx != CMD_diffput
2182 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2183 break;
2184 found_not_ma = TRUE;
2185 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002186 if (idx_other == DB_COUNT)
2187 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002188 if (found_not_ma)
2189 EMSG(_("E793: No other buffer in diff mode is modifiable"));
2190 else
2191 EMSG(_("E100: No other buffer in diff mode"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002192 return;
2193 }
2194
2195 /* Check that there isn't a third buffer in the list */
2196 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002197 if (curtab->tp_diffbuf[i] != curbuf
2198 && curtab->tp_diffbuf[i] != NULL
2199 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002200 {
2201 EMSG(_("E101: More than two buffers in diff mode, don't know which one to use"));
2202 return;
2203 }
2204 }
2205 else
2206 {
2207 /* Buffer number or pattern given. Ignore trailing white space. */
2208 p = eap->arg + STRLEN(eap->arg);
2209 while (p > eap->arg && vim_iswhite(p[-1]))
2210 --p;
2211 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2212 ;
2213 if (eap->arg + i == p) /* digits only */
2214 i = atol((char *)eap->arg);
2215 else
2216 {
Bram Moolenaar0c279bb2013-03-19 14:25:54 +01002217 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002218 if (i < 0)
2219 return; /* error message already given */
2220 }
2221 buf = buflist_findnr(i);
2222 if (buf == NULL)
2223 {
2224 EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg);
2225 return;
2226 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002227 if (buf == curbuf)
2228 return; /* nothing to do */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002229 idx_other = diff_buf_idx(buf);
2230 if (idx_other == DB_COUNT)
2231 {
2232 EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg);
2233 return;
2234 }
2235 }
2236
2237 diff_busy = TRUE;
2238
2239 /* When no range given include the line above or below the cursor. */
2240 if (eap->addr_count == 0)
2241 {
2242 /* Make it possible that ":diffget" on the last line gets line below
2243 * the cursor line when there is no difference above the cursor. */
2244 if (eap->cmdidx == CMD_diffget
2245 && eap->line1 == curbuf->b_ml.ml_line_count
2246 && diff_check(curwin, eap->line1) == 0
2247 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2248 ++eap->line2;
2249 else if (eap->line1 > 0)
2250 --eap->line1;
2251 }
2252
2253 if (eap->cmdidx == CMD_diffget)
2254 {
2255 idx_from = idx_other;
2256 idx_to = idx_cur;
2257 }
2258 else
2259 {
2260 idx_from = idx_cur;
2261 idx_to = idx_other;
2262 /* Need to make the other buffer the current buffer to be able to make
2263 * changes in it. */
2264 /* set curwin/curbuf to buf and save a few things */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002265 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002266 }
2267
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002268 /* May give the warning for a changed buffer here, which can trigger the
2269 * FileChangedRO autocommand, which may do nasty things and mess
2270 * everything up. */
2271 if (!curbuf->b_changed)
2272 {
2273 change_warning(0);
2274 if (diff_buf_idx(curbuf) != idx_to)
2275 {
2276 EMSG(_("E787: Buffer changed unexpectedly"));
2277 return;
2278 }
2279 }
2280
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002282 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002283 {
2284 if (dp->df_lnum[idx_cur] > eap->line2 + off)
2285 break; /* past the range that was specified */
2286
2287 dfree = NULL;
2288 lnum = dp->df_lnum[idx_to];
2289 count = dp->df_count[idx_to];
2290 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2291 && u_save(lnum - 1, lnum + count) != FAIL)
2292 {
2293 /* Inside the specified range and saving for undo worked. */
2294 start_skip = 0;
2295 end_skip = 0;
2296 if (eap->addr_count > 0)
2297 {
2298 /* A range was specified: check if lines need to be skipped. */
2299 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2300 if (start_skip > 0)
2301 {
2302 /* range starts below start of current diff block */
2303 if (start_skip > count)
2304 {
2305 lnum += count;
2306 count = 0;
2307 }
2308 else
2309 {
2310 count -= start_skip;
2311 lnum += start_skip;
2312 }
2313 }
2314 else
2315 start_skip = 0;
2316
2317 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2318 - (eap->line2 + off);
2319 if (end_skip > 0)
2320 {
2321 /* range ends above end of current/from diff block */
2322 if (idx_cur == idx_from) /* :diffput */
2323 {
2324 i = dp->df_count[idx_cur] - start_skip - end_skip;
2325 if (count > i)
2326 count = i;
2327 }
2328 else /* :diffget */
2329 {
2330 count -= end_skip;
2331 end_skip = dp->df_count[idx_from] - start_skip - count;
2332 if (end_skip < 0)
2333 end_skip = 0;
2334 }
2335 }
2336 else
2337 end_skip = 0;
2338 }
2339
Bram Moolenaare962c672014-10-15 12:56:49 +02002340 buf_empty = bufempty();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002341 added = 0;
2342 for (i = 0; i < count; ++i)
2343 {
Bram Moolenaar280f1262006-01-30 00:14:18 +00002344 /* remember deleting the last line of the buffer */
2345 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002346 ml_delete(lnum, FALSE);
2347 --added;
2348 }
2349 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2350 {
2351 linenr_T nr;
2352
2353 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002354 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002355 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002356 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2357 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002358 if (p != NULL)
2359 {
2360 ml_append(lnum + i - 1, p, 0, FALSE);
2361 vim_free(p);
2362 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002363 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2364 {
2365 /* Added the first line into an empty buffer, need to
2366 * delete the dummy empty line. */
2367 buf_empty = FALSE;
2368 ml_delete((linenr_T)2, FALSE);
2369 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002370 }
2371 }
2372 new_count = dp->df_count[idx_to] + added;
2373 dp->df_count[idx_to] = new_count;
2374
2375 if (start_skip == 0 && end_skip == 0)
2376 {
2377 /* Check if there are any other buffers and if the diff is
2378 * equal in them. */
2379 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002380 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2381 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002382 && !diff_equal_entry(dp, idx_from, i))
2383 break;
2384 if (i == DB_COUNT)
2385 {
2386 /* delete the diff entry, the buffers are now equal here */
2387 dfree = dp;
2388 dp = dp->df_next;
2389 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002390 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002391 else
2392 dprev->df_next = dp;
2393 }
2394 }
2395
2396 /* Adjust marks. This will change the following entries! */
2397 if (added != 0)
2398 {
2399 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2400 if (curwin->w_cursor.lnum >= lnum)
2401 {
2402 /* Adjust the cursor position if it's in/after the changed
2403 * lines. */
2404 if (curwin->w_cursor.lnum >= lnum + count)
2405 curwin->w_cursor.lnum += added;
2406 else if (added < 0)
2407 curwin->w_cursor.lnum = lnum;
2408 }
2409 }
2410 changed_lines(lnum, 0, lnum + count, (long)added);
2411
2412 if (dfree != NULL)
2413 {
2414 /* Diff is deleted, update folds in other windows. */
2415#ifdef FEAT_FOLDING
2416 diff_fold_update(dfree, idx_to);
2417#endif
2418 vim_free(dfree);
2419 }
2420 else
2421 /* mark_adjust() may have changed the count in a wrong way */
2422 dp->df_count[idx_to] = new_count;
2423
2424 /* When changing the current buffer, keep track of line numbers */
2425 if (idx_cur == idx_to)
2426 off += added;
2427 }
2428
2429 /* If before the range or not deleted, go to next diff. */
2430 if (dfree == NULL)
2431 {
2432 dprev = dp;
2433 dp = dp->df_next;
2434 }
2435 }
2436
2437 /* restore curwin/curbuf and a few other things */
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002438 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002439 {
2440 /* Syncing undo only works for the current buffer, but we change
2441 * another buffer. Sync undo if the command was typed. This isn't
2442 * 100% right when ":diffput" is used in a function or mapping. */
2443 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002444 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002445 aucmd_restbuf(&aco);
2446 }
2447
2448 diff_busy = FALSE;
2449
2450 /* Check that the cursor is on a valid character and update it's position.
2451 * When there were filler lines the topline has become invalid. */
2452 check_cursor();
2453 changed_line_abv_curs();
2454
2455 /* Also need to redraw the other buffers. */
2456 diff_redraw(FALSE);
2457}
2458
2459#ifdef FEAT_FOLDING
2460/*
2461 * Update folds for all diff buffers for entry "dp".
2462 * Skip buffer with index "skip_idx".
2463 * When there are no diffs, all folds are removed.
2464 */
2465 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002466diff_fold_update(diff_T *dp, int skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002467{
2468 int i;
2469 win_T *wp;
2470
Bram Moolenaar29323592016-07-24 22:04:11 +02002471 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002472 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002473 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002474 foldUpdate(wp, dp->df_lnum[i],
2475 dp->df_lnum[i] + dp->df_count[i]);
2476}
2477#endif
2478
2479/*
2480 * Return TRUE if buffer "buf" is in diff-mode.
2481 */
2482 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002483diff_mode_buf(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002485 tabpage_T *tp;
2486
Bram Moolenaar29323592016-07-24 22:04:11 +02002487 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002488 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
2489 return TRUE;
2490 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002491}
2492
2493/*
2494 * Move "count" times in direction "dir" to the next diff block.
2495 * Return FAIL if there isn't such a diff block.
2496 */
2497 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002498diff_move_to(int dir, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002499{
2500 int idx;
2501 linenr_T lnum = curwin->w_cursor.lnum;
2502 diff_T *dp;
2503
2504 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002505 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002506 return FAIL;
2507
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002508 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002509 ex_diffupdate(NULL); /* update after a big change */
2510
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002511 if (curtab->tp_first_diff == NULL) /* no diffs today */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002512 return FAIL;
2513
2514 while (--count >= 0)
2515 {
2516 /* Check if already before first diff. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002517 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002518 break;
2519
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002520 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002521 {
2522 if (dp == NULL)
2523 break;
2524 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
2525 || (dir == BACKWARD
2526 && (dp->df_next == NULL
2527 || lnum <= dp->df_next->df_lnum[idx])))
2528 {
2529 lnum = dp->df_lnum[idx];
2530 break;
2531 }
2532 }
2533 }
2534
2535 /* don't end up past the end of the file */
2536 if (lnum > curbuf->b_ml.ml_line_count)
2537 lnum = curbuf->b_ml.ml_line_count;
2538
2539 /* When the cursor didn't move at all we fail. */
2540 if (lnum == curwin->w_cursor.lnum)
2541 return FAIL;
2542
2543 setpcmark();
2544 curwin->w_cursor.lnum = lnum;
2545 curwin->w_cursor.col = 0;
2546
2547 return OK;
2548}
2549
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002550/*
2551 * Return the line number in the current window that is closest to "lnum1" in
2552 * "buf1" in diff mode.
2553 */
2554 static linenr_T
2555diff_get_corresponding_line_int(
Bram Moolenaar7454a062016-01-30 15:14:10 +01002556 buf_T *buf1,
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002557 linenr_T lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02002558{
2559 int idx1;
2560 int idx2;
2561 diff_T *dp;
2562 int baseline = 0;
Bram Moolenaar860cae12010-06-05 23:22:07 +02002563
2564 idx1 = diff_buf_idx(buf1);
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002565 idx2 = diff_buf_idx(curbuf);
Bram Moolenaar860cae12010-06-05 23:22:07 +02002566 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
2567 return lnum1;
2568
2569 if (curtab->tp_diff_invalid)
2570 ex_diffupdate(NULL); /* update after a big change */
2571
2572 if (curtab->tp_first_diff == NULL) /* no diffs today */
2573 return lnum1;
2574
2575 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
2576 {
2577 if (dp->df_lnum[idx1] > lnum1)
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002578 return lnum1 - baseline;
2579 if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02002580 {
2581 /* Inside the diffblock */
2582 baseline = lnum1 - dp->df_lnum[idx1];
2583 if (baseline > dp->df_count[idx2])
2584 baseline = dp->df_count[idx2];
2585
2586 return dp->df_lnum[idx2] + baseline;
2587 }
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002588 if ( (dp->df_lnum[idx1] == lnum1)
2589 && (dp->df_count[idx1] == 0)
2590 && (dp->df_lnum[idx2] <= curwin->w_cursor.lnum)
2591 && ((dp->df_lnum[idx2] + dp->df_count[idx2])
2592 > curwin->w_cursor.lnum))
Bram Moolenaar860cae12010-06-05 23:22:07 +02002593 /*
2594 * Special case: if the cursor is just after a zero-count
2595 * block (i.e. all filler) and the target cursor is already
2596 * inside the corresponding block, leave the target cursor
2597 * unmoved. This makes repeated CTRL-W W operations work
2598 * as expected.
2599 */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002600 return curwin->w_cursor.lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02002601 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
2602 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
2603 }
2604
2605 /* If we get here then the cursor is after the last diff */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002606 return lnum1 - baseline;
2607}
Bram Moolenaar860cae12010-06-05 23:22:07 +02002608
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002609/*
2610 * Return the line number in the current window that is closest to "lnum1" in
2611 * "buf1" in diff mode. Checks the line number to be valid.
2612 */
2613 linenr_T
2614diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1)
2615{
2616 linenr_T lnum = diff_get_corresponding_line_int(buf1, lnum1);
2617
2618 /* don't end up past the end of the file */
2619 if (lnum > curbuf->b_ml.ml_line_count)
2620 return curbuf->b_ml.ml_line_count;
2621 return lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02002622}
Bram Moolenaar860cae12010-06-05 23:22:07 +02002623
Bram Moolenaar071d4272004-06-13 20:20:40 +00002624/*
2625 * For line "lnum" in the current window find the equivalent lnum in window
2626 * "wp", compensating for inserted/deleted lines.
2627 */
2628 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01002629diff_lnum_win(linenr_T lnum, win_T *wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002630{
2631 diff_T *dp;
2632 int idx;
2633 int i;
2634 linenr_T n;
2635
2636 idx = diff_buf_idx(curbuf);
2637 if (idx == DB_COUNT) /* safety check */
2638 return (linenr_T)0;
2639
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002640 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002641 ex_diffupdate(NULL); /* update after a big change */
2642
2643 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002644 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002645 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2646 break;
2647
2648 /* When after the last change, compute relative to the last line number. */
2649 if (dp == NULL)
2650 return wp->w_buffer->b_ml.ml_line_count
2651 - (curbuf->b_ml.ml_line_count - lnum);
2652
2653 /* Find index for "wp". */
2654 i = diff_buf_idx(wp->w_buffer);
2655 if (i == DB_COUNT) /* safety check */
2656 return (linenr_T)0;
2657
2658 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
2659 if (n > dp->df_lnum[i] + dp->df_count[i])
2660 n = dp->df_lnum[i] + dp->df_count[i];
2661 return n;
2662}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002663
2664#endif /* FEAT_DIFF */