blob: b64b45ce9607feabe48a9cbd5da3d6cd17fe2ee0 [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 Moolenaar5fd0ca72009-05-13 16:56:33 +0000980 vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"",
981 tmp_new, tmp_orig,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000982# ifdef UNIX
983 fullname != NULL ? fullname :
984# endif
985 eap->arg);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000986#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000987 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000988#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000990#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000991 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000992#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000993 }
994
995#ifdef UNIX
996 if (dirbuf[0] != NUL)
997 {
998 if (mch_chdir((char *)dirbuf) != 0)
999 EMSG(_(e_prev_dir));
1000 shorten_fnames(TRUE);
1001 }
1002#endif
1003
1004 /* patch probably has written over the screen */
1005 redraw_later(CLEAR);
1006
1007 /* Delete any .orig or .rej file created. */
1008 STRCPY(buf, tmp_new);
1009 STRCAT(buf, ".orig");
1010 mch_remove(buf);
1011 STRCPY(buf, tmp_new);
1012 STRCAT(buf, ".rej");
1013 mch_remove(buf);
1014
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001015 /* Only continue if the output file was created. */
1016 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
1017 EMSG(_("E816: Cannot read patch output"));
1018 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001019 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001020 if (curbuf->b_fname != NULL)
1021 {
1022 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001023 (int)(STRLEN(curbuf->b_fname) + 4));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001024 if (newname != NULL)
1025 STRCAT(newname, ".new");
1026 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001027
1028#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001029 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001030#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001031 /* don't use a new tab page, each tab page has its own diffs */
1032 cmdmod.tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001033
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001034 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001035 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001036 /* Pretend it was a ":split fname" command */
1037 eap->cmdidx = CMD_split;
1038 eap->arg = tmp_new;
1039 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001040
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001041 /* check that split worked and editing tmp_new */
1042 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001043 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001044 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1045 diff_win_options(curwin, TRUE);
1046 diff_win_options(old_curwin, TRUE);
1047
1048 if (newname != NULL)
1049 {
1050 /* do a ":file filename.new" on the patched buffer */
1051 eap->arg = newname;
1052 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001053
1054#ifdef FEAT_AUTOCMD
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001055 /* Do filetype detection with the new name. */
1056 if (au_has_group((char_u *)"filetypedetect"))
1057 do_cmdline_cmd((char_u *)":doau filetypedetect BufRead");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001058#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001059 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001060 }
1061 }
1062 }
1063
1064theend:
1065 if (tmp_orig != NULL)
1066 mch_remove(tmp_orig);
1067 vim_free(tmp_orig);
1068 if (tmp_new != NULL)
1069 mch_remove(tmp_new);
1070 vim_free(tmp_new);
1071 vim_free(newname);
1072 vim_free(buf);
1073#ifdef UNIX
1074 vim_free(fullname);
1075#endif
1076#ifdef FEAT_BROWSE
1077 vim_free(browseFile);
1078 cmdmod.browse = browse_flag;
1079#endif
1080}
1081
1082/*
1083 * Split the window and edit another file, setting options to show the diffs.
1084 */
1085 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001086ex_diffsplit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001087{
1088 win_T *old_curwin = curwin;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001089 bufref_T old_curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001090
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001091 set_bufref(&old_curbuf, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001092#ifdef FEAT_GUI
1093 need_mouse_correct = TRUE;
1094#endif
Bram Moolenaar46328f92016-08-28 15:39:57 +02001095 /* Need to compute w_fraction when no redraw happened yet. */
1096 validate_cursor();
1097 set_fraction(curwin);
1098
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001099 /* don't use a new tab page, each tab page has its own diffs */
1100 cmdmod.tab = 0;
1101
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001102 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001103 {
1104 /* Pretend it was a ":split fname" command */
1105 eap->cmdidx = CMD_split;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001106 curwin->w_p_diff = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107 do_exedit(eap, old_curwin);
1108
1109 if (curwin != old_curwin) /* split must have worked */
1110 {
1111 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1112 diff_win_options(curwin, TRUE);
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001113 if (win_valid(old_curwin))
1114 {
1115 diff_win_options(old_curwin, TRUE);
1116
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001117 if (bufref_valid(&old_curbuf))
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001118 /* Move the cursor position to that of the old window. */
1119 curwin->w_cursor.lnum = diff_get_corresponding_line(
Bram Moolenaar025e3e02016-10-18 14:50:18 +02001120 old_curbuf.br_buf, old_curwin->w_cursor.lnum);
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001121 }
Bram Moolenaar46328f92016-08-28 15:39:57 +02001122 /* Now that lines are folded scroll to show the cursor at the same
1123 * relative position. */
1124 scroll_to_fraction(curwin, curwin->w_height);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001125 }
1126 }
1127}
1128
1129/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001130 * Set options to show diffs for the current window.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001131 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001133ex_diffthis(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001134{
1135 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1136 diff_win_options(curwin, TRUE);
1137}
1138
1139/*
1140 * Set options in window "wp" for diff mode.
1141 */
1142 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001143diff_win_options(
1144 win_T *wp,
1145 int addbuf) /* Add buffer to diff. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001146{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001147# ifdef FEAT_FOLDING
1148 win_T *old_curwin = curwin;
1149
1150 /* close the manually opened folds */
1151 curwin = wp;
1152 newFoldLevel();
1153 curwin = old_curwin;
1154# endif
1155
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001156 /* Use 'scrollbind' and 'cursorbind' when available */
1157#ifdef FEAT_SCROLLBIND
Bram Moolenaar43929962015-07-03 15:06:56 +02001158 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001159 wp->w_p_scb_save = wp->w_p_scb;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001160 wp->w_p_scb = TRUE;
1161#endif
Bram Moolenaar860cae12010-06-05 23:22:07 +02001162#ifdef FEAT_CURSORBIND
Bram Moolenaar43929962015-07-03 15:06:56 +02001163 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001164 wp->w_p_crb_save = wp->w_p_crb;
Bram Moolenaar860cae12010-06-05 23:22:07 +02001165 wp->w_p_crb = TRUE;
1166#endif
Bram Moolenaar43929962015-07-03 15:06:56 +02001167 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001168 wp->w_p_wrap_save = wp->w_p_wrap;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001169 wp->w_p_wrap = FALSE;
1170# ifdef FEAT_FOLDING
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001171 curwin = wp;
1172 curbuf = curwin->w_buffer;
Bram Moolenaar43929962015-07-03 15:06:56 +02001173 if (!wp->w_p_diff)
1174 {
1175 if (wp->w_p_diff_saved)
1176 free_string_option(wp->w_p_fdm_save);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001177 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm);
Bram Moolenaar43929962015-07-03 15:06:56 +02001178 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001179 set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001180 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001181 curwin = old_curwin;
1182 curbuf = curwin->w_buffer;
Bram Moolenaar43929962015-07-03 15:06:56 +02001183 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001184 {
1185 wp->w_p_fdc_save = wp->w_p_fdc;
1186 wp->w_p_fen_save = wp->w_p_fen;
1187 wp->w_p_fdl_save = wp->w_p_fdl;
1188 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001189 wp->w_p_fdc = diff_foldcolumn;
1190 wp->w_p_fen = TRUE;
1191 wp->w_p_fdl = 0;
1192 foldUpdateAll(wp);
1193 /* make sure topline is not halfway a fold */
1194 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001195# endif
1196#ifdef FEAT_SCROLLBIND
1197 if (vim_strchr(p_sbo, 'h') == NULL)
1198 do_cmdline_cmd((char_u *)"set sbo+=hor");
1199#endif
Bram Moolenaara87aa802013-07-03 15:47:03 +02001200 /* Saved the current values, to be restored in ex_diffoff(). */
1201 wp->w_p_diff_saved = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001202
Bram Moolenaar43929962015-07-03 15:06:56 +02001203 wp->w_p_diff = TRUE;
1204
Bram Moolenaar071d4272004-06-13 20:20:40 +00001205 if (addbuf)
1206 diff_buf_add(wp->w_buffer);
1207 redraw_win_later(wp, NOT_VALID);
1208}
1209
1210/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001211 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001212 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001213 */
1214 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001215ex_diffoff(exarg_T *eap)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001216{
1217 win_T *wp;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001218#ifdef FEAT_SCROLLBIND
1219 int diffwin = FALSE;
1220#endif
1221
Bram Moolenaar29323592016-07-24 22:04:11 +02001222 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001223 {
Bram Moolenaar00462ff2013-09-20 20:13:53 +02001224 if (eap->forceit ? wp->w_p_diff : wp == curwin)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001225 {
Bram Moolenaar43929962015-07-03 15:06:56 +02001226 /* Set 'diff' off. If option values were saved in
1227 * diff_win_options(), restore the ones whose settings seem to have
1228 * been left over from diff mode. */
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001229 wp->w_p_diff = FALSE;
Bram Moolenaara87aa802013-07-03 15:47:03 +02001230
Bram Moolenaara87aa802013-07-03 15:47:03 +02001231 if (wp->w_p_diff_saved)
1232 {
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001233
Bram Moolenaar43929962015-07-03 15:06:56 +02001234#ifdef FEAT_SCROLLBIND
1235 if (wp->w_p_scb)
1236 wp->w_p_scb = wp->w_p_scb_save;
1237#endif
1238#ifdef FEAT_CURSORBIND
1239 if (wp->w_p_crb)
1240 wp->w_p_crb = wp->w_p_crb_save;
1241#endif
1242 if (!wp->w_p_wrap)
1243 wp->w_p_wrap = wp->w_p_wrap_save;
1244#ifdef FEAT_FOLDING
1245 free_string_option(wp->w_p_fdm);
1246 wp->w_p_fdm = vim_strsave(wp->w_p_fdm_save);
1247
1248 if (wp->w_p_fdc == diff_foldcolumn)
1249 wp->w_p_fdc = wp->w_p_fdc_save;
1250 if (wp->w_p_fdl == 0)
1251 wp->w_p_fdl = wp->w_p_fdl_save;
1252
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001253 /* Only restore 'foldenable' when 'foldmethod' is not
1254 * "manual", otherwise we continue to show the diff folds. */
Bram Moolenaar43929962015-07-03 15:06:56 +02001255 if (wp->w_p_fen)
1256 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE
1257 : wp->w_p_fen_save;
1258
1259 foldUpdateAll(wp);
Bram Moolenaar43929962015-07-03 15:06:56 +02001260#endif
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001261 }
Bram Moolenaare67d5462016-08-27 22:40:42 +02001262 /* remove filler lines */
1263 wp->w_topfill = 0;
1264
1265 /* make sure topline is not halfway a fold and cursor is
1266 * invalidated */
1267 changed_window_setting_win(wp);
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001268
Bram Moolenaara87aa802013-07-03 15:47:03 +02001269 /* Note: 'sbo' is not restored, it's a global option. */
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001270 diff_buf_adjust(wp);
1271 }
1272#ifdef FEAT_SCROLLBIND
1273 diffwin |= wp->w_p_diff;
1274#endif
1275 }
1276
Bram Moolenaar25ea0542017-02-03 23:16:28 +01001277 /* Also remove hidden buffers from the list. */
1278 if (eap->forceit)
1279 diff_buf_clear();
1280
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001281#ifdef FEAT_SCROLLBIND
1282 /* Remove "hor" from from 'scrollopt' if there are no diff windows left. */
1283 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1284 do_cmdline_cmd((char_u *)"set sbo-=hor");
1285#endif
1286}
1287
1288/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001289 * Read the diff output and add each entry to the diff list.
1290 */
1291 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001292diff_read(
1293 int idx_orig, /* idx of original file */
1294 int idx_new, /* idx of new file */
1295 char_u *fname) /* name of diff output file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001296{
1297 FILE *fd;
1298 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001299 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001300 diff_T *dn, *dpl;
1301 long f1, l1, f2, l2;
1302 char_u linebuf[LBUFLEN]; /* only need to hold the diff line */
1303 int difftype;
1304 char_u *p;
1305 long off;
1306 int i;
1307 linenr_T lnum_orig, lnum_new;
1308 long count_orig, count_new;
1309 int notset = TRUE; /* block "*dp" not set yet */
1310
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001311 fd = mch_fopen((char *)fname, "r");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001312 if (fd == NULL)
1313 {
1314 EMSG(_("E98: Cannot read diff output"));
1315 return;
1316 }
1317
1318 for (;;)
1319 {
1320 if (tag_fgets(linebuf, LBUFLEN, fd))
1321 break; /* end of file */
1322 if (!isdigit(*linebuf))
1323 continue; /* not the start of a diff block */
1324
1325 /* This line must be one of three formats:
1326 * {first}[,{last}]c{first}[,{last}]
1327 * {first}a{first}[,{last}]
1328 * {first}[,{last}]d{first}
1329 */
1330 p = linebuf;
1331 f1 = getdigits(&p);
1332 if (*p == ',')
1333 {
1334 ++p;
1335 l1 = getdigits(&p);
1336 }
1337 else
1338 l1 = f1;
1339 if (*p != 'a' && *p != 'c' && *p != 'd')
1340 continue; /* invalid diff format */
1341 difftype = *p++;
1342 f2 = getdigits(&p);
1343 if (*p == ',')
1344 {
1345 ++p;
1346 l2 = getdigits(&p);
1347 }
1348 else
1349 l2 = f2;
1350 if (l1 < f1 || l2 < f2)
1351 continue; /* invalid line range */
1352
1353 if (difftype == 'a')
1354 {
1355 lnum_orig = f1 + 1;
1356 count_orig = 0;
1357 }
1358 else
1359 {
1360 lnum_orig = f1;
1361 count_orig = l1 - f1 + 1;
1362 }
1363 if (difftype == 'd')
1364 {
1365 lnum_new = f2 + 1;
1366 count_new = 0;
1367 }
1368 else
1369 {
1370 lnum_new = f2;
1371 count_new = l2 - f2 + 1;
1372 }
1373
1374 /* Go over blocks before the change, for which orig and new are equal.
1375 * Copy blocks from orig to new. */
1376 while (dp != NULL
1377 && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig])
1378 {
1379 if (notset)
1380 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1381 dprev = dp;
1382 dp = dp->df_next;
1383 notset = TRUE;
1384 }
1385
1386 if (dp != NULL
1387 && lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig]
1388 && lnum_orig + count_orig >= dp->df_lnum[idx_orig])
1389 {
1390 /* New block overlaps with existing block(s).
1391 * First find last block that overlaps. */
1392 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
1393 if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig])
1394 break;
1395
1396 /* If the newly found block starts before the old one, set the
1397 * start back a number of lines. */
1398 off = dp->df_lnum[idx_orig] - lnum_orig;
1399 if (off > 0)
1400 {
1401 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001402 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001403 dp->df_lnum[i] -= off;
1404 dp->df_lnum[idx_new] = lnum_new;
1405 dp->df_count[idx_new] = count_new;
1406 }
1407 else if (notset)
1408 {
1409 /* new block inside existing one, adjust new block */
1410 dp->df_lnum[idx_new] = lnum_new + off;
1411 dp->df_count[idx_new] = count_new - off;
1412 }
1413 else
1414 /* second overlap of new block with existing block */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001415 dp->df_count[idx_new] += count_new - count_orig
1416 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1417 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001418
1419 /* Adjust the size of the block to include all the lines to the
1420 * end of the existing block or the new diff, whatever ends last. */
1421 off = (lnum_orig + count_orig)
1422 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1423 if (off < 0)
1424 {
1425 /* new change ends in existing block, adjust the end if not
1426 * done already */
1427 if (notset)
1428 dp->df_count[idx_new] += -off;
1429 off = 0;
1430 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001431 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001432 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001433 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1434 - dp->df_lnum[i] + off;
1435
1436 /* Delete the diff blocks that have been merged into one. */
1437 dn = dp->df_next;
1438 dp->df_next = dpl->df_next;
1439 while (dn != dp->df_next)
1440 {
1441 dpl = dn->df_next;
1442 vim_free(dn);
1443 dn = dpl;
1444 }
1445 }
1446 else
1447 {
1448 /* Allocate a new diffblock. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001449 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001450 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001451 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001452
1453 dp->df_lnum[idx_orig] = lnum_orig;
1454 dp->df_count[idx_orig] = count_orig;
1455 dp->df_lnum[idx_new] = lnum_new;
1456 dp->df_count[idx_new] = count_new;
1457
1458 /* Set values for other buffers, these must be equal to the
1459 * original buffer, otherwise there would have been a change
1460 * already. */
1461 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001462 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001463 diff_copy_entry(dprev, dp, idx_orig, i);
1464 }
1465 notset = FALSE; /* "*dp" has been set */
1466 }
1467
1468 /* for remaining diff blocks orig and new are equal */
1469 while (dp != NULL)
1470 {
1471 if (notset)
1472 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1473 dprev = dp;
1474 dp = dp->df_next;
1475 notset = TRUE;
1476 }
1477
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001478done:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001479 fclose(fd);
1480}
1481
1482/*
1483 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1484 */
1485 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001486diff_copy_entry(
1487 diff_T *dprev,
1488 diff_T *dp,
1489 int idx_orig,
1490 int idx_new)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001491{
1492 long off;
1493
1494 if (dprev == NULL)
1495 off = 0;
1496 else
1497 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1498 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1499 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1500 dp->df_count[idx_new] = dp->df_count[idx_orig];
1501}
1502
1503/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001504 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001505 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001506 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001507diff_clear(tabpage_T *tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001508{
1509 diff_T *p, *next_p;
1510
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001511 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001512 {
1513 next_p = p->df_next;
1514 vim_free(p);
1515 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001516 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001517}
1518
1519/*
1520 * Check diff status for line "lnum" in buffer "buf":
1521 * Returns 0 for nothing special
1522 * Returns -1 for a line that should be highlighted as changed.
1523 * Returns -2 for a line that should be highlighted as added/deleted.
1524 * Returns > 0 for inserting that many filler lines above it (never happens
1525 * when 'diffopt' doesn't contain "filler").
1526 * This should only be used for windows where 'diff' is set.
1527 */
1528 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001529diff_check(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001530{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001531 int idx; /* index in tp_diffbuf[] for this buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001532 diff_T *dp;
1533 int maxcount;
1534 int i;
1535 buf_T *buf = wp->w_buffer;
1536 int cmp;
1537
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001538 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001539 ex_diffupdate(NULL); /* update after a big change */
1540
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001541 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) /* no diffs at all */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001542 return 0;
1543
1544 /* safety check: "lnum" must be a buffer line */
1545 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1546 return 0;
1547
1548 idx = diff_buf_idx(buf);
1549 if (idx == DB_COUNT)
1550 return 0; /* no diffs for buffer "buf" */
1551
1552#ifdef FEAT_FOLDING
1553 /* A closed fold never has filler lines. */
1554 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1555 return 0;
1556#endif
1557
1558 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001559 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001560 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1561 break;
1562 if (dp == NULL || lnum < dp->df_lnum[idx])
1563 return 0;
1564
1565 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1566 {
1567 int zero = FALSE;
1568
1569 /* Changed or inserted line. If the other buffers have a count of
1570 * zero, the lines were inserted. If the other buffers have the same
1571 * count, check if the lines are identical. */
1572 cmp = FALSE;
1573 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001574 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001575 {
1576 if (dp->df_count[i] == 0)
1577 zero = TRUE;
1578 else
1579 {
1580 if (dp->df_count[i] != dp->df_count[idx])
1581 return -1; /* nr of lines changed. */
1582 cmp = TRUE;
1583 }
1584 }
1585 if (cmp)
1586 {
1587 /* Compare all lines. If they are equal the lines were inserted
1588 * in some buffers, deleted in others, but not changed. */
1589 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001590 if (i != idx && curtab->tp_diffbuf[i] != NULL
1591 && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001592 if (!diff_equal_entry(dp, idx, i))
1593 return -1;
1594 }
1595 /* If there is no buffer with zero lines then there is no difference
1596 * any longer. Happens when making a change (or undo) that removes
1597 * the difference. Can't remove the entry here, we might be halfway
1598 * updating the window. Just report the text as unchanged. Other
1599 * windows might still show the change though. */
1600 if (zero == FALSE)
1601 return 0;
1602 return -2;
1603 }
1604
1605 /* If 'diffopt' doesn't contain "filler", return 0. */
1606 if (!(diff_flags & DIFF_FILLER))
1607 return 0;
1608
1609 /* Insert filler lines above the line just below the change. Will return
1610 * 0 when this buf had the max count. */
1611 maxcount = 0;
1612 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001613 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001614 maxcount = dp->df_count[i];
1615 return maxcount - dp->df_count[idx];
1616}
1617
1618/*
1619 * Compare two entries in diff "*dp" and return TRUE if they are equal.
1620 */
1621 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001622diff_equal_entry(diff_T *dp, int idx1, int idx2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001623{
1624 int i;
1625 char_u *line;
1626 int cmp;
1627
1628 if (dp->df_count[idx1] != dp->df_count[idx2])
1629 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001630 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001631 return FALSE;
1632 for (i = 0; i < dp->df_count[idx1]; ++i)
1633 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001634 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001635 dp->df_lnum[idx1] + i, FALSE));
1636 if (line == NULL)
1637 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001638 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001639 dp->df_lnum[idx2] + i, FALSE));
1640 vim_free(line);
1641 if (cmp != 0)
1642 return FALSE;
1643 }
1644 return TRUE;
1645}
1646
1647/*
1648 * Compare strings "s1" and "s2" according to 'diffopt'.
1649 * Return non-zero when they are different.
1650 */
1651 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001652diff_cmp(char_u *s1, char_u *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001653{
1654 char_u *p1, *p2;
1655#ifdef FEAT_MBYTE
1656 int l;
1657#endif
1658
1659 if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0)
1660 return STRCMP(s1, s2);
1661 if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE))
1662 return MB_STRICMP(s1, s2);
1663
1664 /* Ignore white space changes and possibly ignore case. */
1665 p1 = s1;
1666 p2 = s2;
1667 while (*p1 != NUL && *p2 != NUL)
1668 {
1669 if (vim_iswhite(*p1) && vim_iswhite(*p2))
1670 {
1671 p1 = skipwhite(p1);
1672 p2 = skipwhite(p2);
1673 }
1674 else
1675 {
1676#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001677 l = (*mb_ptr2len)(p1);
1678 if (l != (*mb_ptr2len)(p2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001679 break;
1680 if (l > 1)
1681 {
1682 if (STRNCMP(p1, p2, l) != 0
1683 && (!enc_utf8
1684 || !(diff_flags & DIFF_ICASE)
1685 || utf_fold(utf_ptr2char(p1))
1686 != utf_fold(utf_ptr2char(p2))))
1687 break;
1688 p1 += l;
1689 p2 += l;
1690 }
1691 else
1692#endif
1693 {
1694 if (*p1 != *p2 && (!(diff_flags & DIFF_ICASE)
1695 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
1696 break;
1697 ++p1;
1698 ++p2;
1699 }
1700 }
1701 }
1702
1703 /* Ignore trailing white space. */
1704 p1 = skipwhite(p1);
1705 p2 = skipwhite(p2);
1706 if (*p1 != NUL || *p2 != NUL)
1707 return 1;
1708 return 0;
1709}
1710
1711/*
1712 * Return the number of filler lines above "lnum".
1713 */
1714 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001715diff_check_fill(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001716{
1717 int n;
1718
1719 /* be quick when there are no filler lines */
1720 if (!(diff_flags & DIFF_FILLER))
1721 return 0;
1722 n = diff_check(wp, lnum);
1723 if (n <= 0)
1724 return 0;
1725 return n;
1726}
1727
1728/*
1729 * Set the topline of "towin" to match the position in "fromwin", so that they
1730 * show the same diff'ed lines.
1731 */
1732 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001733diff_set_topline(win_T *fromwin, win_T *towin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001734{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001735 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001736 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001737 int fromidx;
1738 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001739 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001740 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001741 int i;
1742
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001743 fromidx = diff_buf_idx(frombuf);
1744 if (fromidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001745 return; /* safety check */
1746
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001747 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001748 ex_diffupdate(NULL); /* update after a big change */
1749
1750 towin->w_topfill = 0;
1751
1752 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001753 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001754 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001755 break;
1756 if (dp == NULL)
1757 {
1758 /* After last change, compute topline relative to end of file; no
1759 * filler lines. */
1760 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001761 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001762 }
1763 else
1764 {
1765 /* Find index for "towin". */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001766 toidx = diff_buf_idx(towin->w_buffer);
1767 if (toidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001768 return; /* safety check */
1769
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001770 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
1771 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001772 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001773 /* Inside a change: compute filler lines. With three or more
1774 * buffers we need to know the largest count. */
1775 max_count = 0;
1776 for (i = 0; i < DB_COUNT; ++i)
1777 if (curtab->tp_diffbuf[i] != NULL
1778 && max_count < dp->df_count[i])
1779 max_count = dp->df_count[i];
1780
1781 if (dp->df_count[toidx] == dp->df_count[fromidx])
1782 {
1783 /* same number of lines: use same filler count */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001784 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001785 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001786 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001787 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001788 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001789 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001790 /* more lines in towin and fromwin doesn't show diff
1791 * lines, only filler lines */
1792 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
1793 {
1794 /* towin also only shows filler lines */
1795 towin->w_topline = dp->df_lnum[toidx]
1796 + dp->df_count[toidx];
1797 towin->w_topfill = fromwin->w_topfill;
1798 }
1799 else
1800 /* towin still has some diff lines to show */
1801 towin->w_topline = dp->df_lnum[toidx]
1802 + max_count - fromwin->w_topfill;
1803 }
1804 }
1805 else if (towin->w_topline >= dp->df_lnum[toidx]
1806 + dp->df_count[toidx])
1807 {
1808 /* less lines in towin and no diff lines to show: compute
1809 * filler lines */
1810 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
1811 if (diff_flags & DIFF_FILLER)
1812 {
1813 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
1814 /* fromwin is also out of diff lines */
1815 towin->w_topfill = fromwin->w_topfill;
1816 else
1817 /* fromwin has some diff lines */
1818 towin->w_topfill = dp->df_lnum[fromidx]
1819 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001820 }
1821 }
1822 }
1823 }
1824
1825 /* safety check (if diff info gets outdated strange things may happen) */
1826 towin->w_botfill = FALSE;
1827 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
1828 {
1829 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
1830 towin->w_botfill = TRUE;
1831 }
1832 if (towin->w_topline < 1)
1833 {
1834 towin->w_topline = 1;
1835 towin->w_topfill = 0;
1836 }
1837
1838 /* When w_topline changes need to recompute w_botline and cursor position */
1839 invalidate_botline_win(towin);
1840 changed_line_abv_curs_win(towin);
1841
1842 check_topfill(towin, FALSE);
1843#ifdef FEAT_FOLDING
1844 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
1845 NULL, TRUE, NULL);
1846#endif
1847}
1848
1849/*
1850 * This is called when 'diffopt' is changed.
1851 */
1852 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001853diffopt_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854{
1855 char_u *p;
1856 int diff_context_new = 6;
1857 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001858 int diff_foldcolumn_new = 2;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001859 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001860
1861 p = p_dip;
1862 while (*p != NUL)
1863 {
1864 if (STRNCMP(p, "filler", 6) == 0)
1865 {
1866 p += 6;
1867 diff_flags_new |= DIFF_FILLER;
1868 }
1869 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
1870 {
1871 p += 8;
1872 diff_context_new = getdigits(&p);
1873 }
1874 else if (STRNCMP(p, "icase", 5) == 0)
1875 {
1876 p += 5;
1877 diff_flags_new |= DIFF_ICASE;
1878 }
1879 else if (STRNCMP(p, "iwhite", 6) == 0)
1880 {
1881 p += 6;
1882 diff_flags_new |= DIFF_IWHITE;
1883 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001884 else if (STRNCMP(p, "horizontal", 10) == 0)
1885 {
1886 p += 10;
1887 diff_flags_new |= DIFF_HORIZONTAL;
1888 }
1889 else if (STRNCMP(p, "vertical", 8) == 0)
1890 {
1891 p += 8;
1892 diff_flags_new |= DIFF_VERTICAL;
1893 }
1894 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
1895 {
1896 p += 11;
1897 diff_foldcolumn_new = getdigits(&p);
1898 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899 if (*p != ',' && *p != NUL)
1900 return FAIL;
1901 if (*p == ',')
1902 ++p;
1903 }
1904
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001905 /* Can't have both "horizontal" and "vertical". */
1906 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
1907 return FAIL;
1908
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 /* If "icase" or "iwhite" was added or removed, need to update the diff. */
1910 if (diff_flags != diff_flags_new)
Bram Moolenaar29323592016-07-24 22:04:11 +02001911 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001912 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913
1914 diff_flags = diff_flags_new;
1915 diff_context = diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001916 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001917
1918 diff_redraw(TRUE);
1919
1920 /* recompute the scroll binding with the new option value, may
1921 * remove or add filler lines */
1922 check_scrollbind((linenr_T)0, 0L);
1923
1924 return OK;
1925}
1926
1927/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001928 * Return TRUE if 'diffopt' contains "horizontal".
1929 */
1930 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001931diffopt_horizontal(void)
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001932{
1933 return (diff_flags & DIFF_HORIZONTAL) != 0;
1934}
1935
1936/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001937 * Find the difference within a changed line.
1938 * Returns TRUE if the line was added, no other buffer has it.
1939 */
1940 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001941diff_find_change(
1942 win_T *wp,
1943 linenr_T lnum,
1944 int *startp, /* first char of the change */
1945 int *endp) /* last char of the change */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001946{
1947 char_u *line_org;
1948 char_u *line_new;
1949 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001950 int si_org, si_new;
1951 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952 diff_T *dp;
1953 int idx;
1954 int off;
1955 int added = TRUE;
1956
1957 /* Make a copy of the line, the next ml_get() will invalidate it. */
1958 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
1959 if (line_org == NULL)
1960 return FALSE;
1961
1962 idx = diff_buf_idx(wp->w_buffer);
1963 if (idx == DB_COUNT) /* cannot happen */
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001964 {
1965 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001966 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001967 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001968
1969 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001970 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001971 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1972 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001973 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001974 {
1975 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001976 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001977 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978
1979 off = lnum - dp->df_lnum[idx];
1980
1981 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001982 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001983 {
1984 /* Skip lines that are not in the other change (filler lines). */
1985 if (off >= dp->df_count[i])
1986 continue;
1987 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001988 line_new = ml_get_buf(curtab->tp_diffbuf[i],
1989 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001990
1991 /* Search for start of difference */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001992 si_org = si_new = 0;
1993 while (line_org[si_org] != NUL)
1994 {
1995 if ((diff_flags & DIFF_IWHITE)
1996 && vim_iswhite(line_org[si_org])
1997 && vim_iswhite(line_new[si_new]))
1998 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001999 si_org = (int)(skipwhite(line_org + si_org) - line_org);
2000 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002001 }
2002 else
2003 {
2004 if (line_org[si_org] != line_new[si_new])
2005 break;
2006 ++si_org;
2007 ++si_new;
2008 }
2009 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002010#ifdef FEAT_MBYTE
2011 if (has_mbyte)
2012 {
2013 /* Move back to first byte of character in both lines (may
2014 * have "nn^" in line_org and "n^ in line_new). */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002015 si_org -= (*mb_head_off)(line_org, line_org + si_org);
2016 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 }
2018#endif
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002019 if (*startp > si_org)
2020 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002021
2022 /* Search for end of difference, if any. */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002023 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002024 {
2025 ei_org = (int)STRLEN(line_org);
2026 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002027 while (ei_org >= *startp && ei_new >= si_new
2028 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002029 {
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002030 if ((diff_flags & DIFF_IWHITE)
2031 && vim_iswhite(line_org[ei_org])
2032 && vim_iswhite(line_new[ei_new]))
2033 {
2034 while (ei_org >= *startp
2035 && vim_iswhite(line_org[ei_org]))
2036 --ei_org;
2037 while (ei_new >= si_new
2038 && vim_iswhite(line_new[ei_new]))
2039 --ei_new;
2040 }
2041 else
2042 {
2043 if (line_org[ei_org] != line_new[ei_new])
2044 break;
2045 --ei_org;
2046 --ei_new;
2047 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002048 }
2049 if (*endp < ei_org)
2050 *endp = ei_org;
2051 }
2052 }
2053
2054 vim_free(line_org);
2055 return added;
2056}
2057
2058#if defined(FEAT_FOLDING) || defined(PROTO)
2059/*
2060 * Return TRUE if line "lnum" is not close to a diff block, this line should
2061 * be in a fold.
2062 * Return FALSE if there are no diff blocks at all in this window.
2063 */
2064 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002065diff_infold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002066{
2067 int i;
2068 int idx = -1;
2069 int other = FALSE;
2070 diff_T *dp;
2071
2072 /* Return if 'diff' isn't set. */
2073 if (!wp->w_p_diff)
2074 return FALSE;
2075
2076 for (i = 0; i < DB_COUNT; ++i)
2077 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002078 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002079 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002080 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002081 other = TRUE;
2082 }
2083
2084 /* return here if there are no diffs in the window */
2085 if (idx == -1 || !other)
2086 return FALSE;
2087
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002088 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002089 ex_diffupdate(NULL); /* update after a big change */
2090
2091 /* Return if there are no diff blocks. All lines will be folded. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002092 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002093 return TRUE;
2094
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002095 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096 {
2097 /* If this change is below the line there can't be any further match. */
2098 if (dp->df_lnum[idx] - diff_context > lnum)
2099 break;
2100 /* If this change ends before the line we have a match. */
2101 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2102 return FALSE;
2103 }
2104 return TRUE;
2105}
2106#endif
2107
2108/*
2109 * "dp" and "do" commands.
2110 */
2111 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002112nv_diffgetput(int put, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002113{
2114 exarg_T ea;
Bram Moolenaar6a643652014-10-31 13:54:25 +01002115 char_u buf[30];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002116
Bram Moolenaar6a643652014-10-31 13:54:25 +01002117 if (count == 0)
2118 ea.arg = (char_u *)"";
2119 else
2120 {
2121 vim_snprintf((char *)buf, 30, "%ld", count);
2122 ea.arg = buf;
2123 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002124 if (put)
2125 ea.cmdidx = CMD_diffput;
2126 else
2127 ea.cmdidx = CMD_diffget;
2128 ea.addr_count = 0;
2129 ea.line1 = curwin->w_cursor.lnum;
2130 ea.line2 = curwin->w_cursor.lnum;
2131 ex_diffgetput(&ea);
2132}
2133
2134/*
2135 * ":diffget"
2136 * ":diffput"
2137 */
2138 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002139ex_diffgetput(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002140{
2141 linenr_T lnum;
2142 int count;
2143 linenr_T off = 0;
2144 diff_T *dp;
2145 diff_T *dprev;
2146 diff_T *dfree;
2147 int idx_cur;
2148 int idx_other;
2149 int idx_from;
2150 int idx_to;
2151 int i;
2152 int added;
2153 char_u *p;
2154 aco_save_T aco;
2155 buf_T *buf;
2156 int start_skip, end_skip;
2157 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002158 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002159 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002160
2161 /* Find the current buffer in the list of diff buffers. */
2162 idx_cur = diff_buf_idx(curbuf);
2163 if (idx_cur == DB_COUNT)
2164 {
2165 EMSG(_("E99: Current buffer is not in diff mode"));
2166 return;
2167 }
2168
2169 if (*eap->arg == NUL)
2170 {
2171 /* No argument: Find the other buffer in the list of diff buffers. */
2172 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002173 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002174 && curtab->tp_diffbuf[idx_other] != NULL)
2175 {
2176 if (eap->cmdidx != CMD_diffput
2177 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2178 break;
2179 found_not_ma = TRUE;
2180 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002181 if (idx_other == DB_COUNT)
2182 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002183 if (found_not_ma)
2184 EMSG(_("E793: No other buffer in diff mode is modifiable"));
2185 else
2186 EMSG(_("E100: No other buffer in diff mode"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002187 return;
2188 }
2189
2190 /* Check that there isn't a third buffer in the list */
2191 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002192 if (curtab->tp_diffbuf[i] != curbuf
2193 && curtab->tp_diffbuf[i] != NULL
2194 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002195 {
2196 EMSG(_("E101: More than two buffers in diff mode, don't know which one to use"));
2197 return;
2198 }
2199 }
2200 else
2201 {
2202 /* Buffer number or pattern given. Ignore trailing white space. */
2203 p = eap->arg + STRLEN(eap->arg);
2204 while (p > eap->arg && vim_iswhite(p[-1]))
2205 --p;
2206 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2207 ;
2208 if (eap->arg + i == p) /* digits only */
2209 i = atol((char *)eap->arg);
2210 else
2211 {
Bram Moolenaar0c279bb2013-03-19 14:25:54 +01002212 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002213 if (i < 0)
2214 return; /* error message already given */
2215 }
2216 buf = buflist_findnr(i);
2217 if (buf == NULL)
2218 {
2219 EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg);
2220 return;
2221 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002222 if (buf == curbuf)
2223 return; /* nothing to do */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002224 idx_other = diff_buf_idx(buf);
2225 if (idx_other == DB_COUNT)
2226 {
2227 EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg);
2228 return;
2229 }
2230 }
2231
2232 diff_busy = TRUE;
2233
2234 /* When no range given include the line above or below the cursor. */
2235 if (eap->addr_count == 0)
2236 {
2237 /* Make it possible that ":diffget" on the last line gets line below
2238 * the cursor line when there is no difference above the cursor. */
2239 if (eap->cmdidx == CMD_diffget
2240 && eap->line1 == curbuf->b_ml.ml_line_count
2241 && diff_check(curwin, eap->line1) == 0
2242 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2243 ++eap->line2;
2244 else if (eap->line1 > 0)
2245 --eap->line1;
2246 }
2247
2248 if (eap->cmdidx == CMD_diffget)
2249 {
2250 idx_from = idx_other;
2251 idx_to = idx_cur;
2252 }
2253 else
2254 {
2255 idx_from = idx_cur;
2256 idx_to = idx_other;
2257 /* Need to make the other buffer the current buffer to be able to make
2258 * changes in it. */
2259 /* set curwin/curbuf to buf and save a few things */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002260 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002261 }
2262
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002263 /* May give the warning for a changed buffer here, which can trigger the
2264 * FileChangedRO autocommand, which may do nasty things and mess
2265 * everything up. */
2266 if (!curbuf->b_changed)
2267 {
2268 change_warning(0);
2269 if (diff_buf_idx(curbuf) != idx_to)
2270 {
2271 EMSG(_("E787: Buffer changed unexpectedly"));
2272 return;
2273 }
2274 }
2275
Bram Moolenaar071d4272004-06-13 20:20:40 +00002276 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002277 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002278 {
2279 if (dp->df_lnum[idx_cur] > eap->line2 + off)
2280 break; /* past the range that was specified */
2281
2282 dfree = NULL;
2283 lnum = dp->df_lnum[idx_to];
2284 count = dp->df_count[idx_to];
2285 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2286 && u_save(lnum - 1, lnum + count) != FAIL)
2287 {
2288 /* Inside the specified range and saving for undo worked. */
2289 start_skip = 0;
2290 end_skip = 0;
2291 if (eap->addr_count > 0)
2292 {
2293 /* A range was specified: check if lines need to be skipped. */
2294 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2295 if (start_skip > 0)
2296 {
2297 /* range starts below start of current diff block */
2298 if (start_skip > count)
2299 {
2300 lnum += count;
2301 count = 0;
2302 }
2303 else
2304 {
2305 count -= start_skip;
2306 lnum += start_skip;
2307 }
2308 }
2309 else
2310 start_skip = 0;
2311
2312 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2313 - (eap->line2 + off);
2314 if (end_skip > 0)
2315 {
2316 /* range ends above end of current/from diff block */
2317 if (idx_cur == idx_from) /* :diffput */
2318 {
2319 i = dp->df_count[idx_cur] - start_skip - end_skip;
2320 if (count > i)
2321 count = i;
2322 }
2323 else /* :diffget */
2324 {
2325 count -= end_skip;
2326 end_skip = dp->df_count[idx_from] - start_skip - count;
2327 if (end_skip < 0)
2328 end_skip = 0;
2329 }
2330 }
2331 else
2332 end_skip = 0;
2333 }
2334
Bram Moolenaare962c672014-10-15 12:56:49 +02002335 buf_empty = bufempty();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002336 added = 0;
2337 for (i = 0; i < count; ++i)
2338 {
Bram Moolenaar280f1262006-01-30 00:14:18 +00002339 /* remember deleting the last line of the buffer */
2340 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002341 ml_delete(lnum, FALSE);
2342 --added;
2343 }
2344 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2345 {
2346 linenr_T nr;
2347
2348 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002349 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002350 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002351 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2352 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002353 if (p != NULL)
2354 {
2355 ml_append(lnum + i - 1, p, 0, FALSE);
2356 vim_free(p);
2357 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002358 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2359 {
2360 /* Added the first line into an empty buffer, need to
2361 * delete the dummy empty line. */
2362 buf_empty = FALSE;
2363 ml_delete((linenr_T)2, FALSE);
2364 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002365 }
2366 }
2367 new_count = dp->df_count[idx_to] + added;
2368 dp->df_count[idx_to] = new_count;
2369
2370 if (start_skip == 0 && end_skip == 0)
2371 {
2372 /* Check if there are any other buffers and if the diff is
2373 * equal in them. */
2374 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002375 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2376 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002377 && !diff_equal_entry(dp, idx_from, i))
2378 break;
2379 if (i == DB_COUNT)
2380 {
2381 /* delete the diff entry, the buffers are now equal here */
2382 dfree = dp;
2383 dp = dp->df_next;
2384 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002385 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002386 else
2387 dprev->df_next = dp;
2388 }
2389 }
2390
2391 /* Adjust marks. This will change the following entries! */
2392 if (added != 0)
2393 {
2394 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2395 if (curwin->w_cursor.lnum >= lnum)
2396 {
2397 /* Adjust the cursor position if it's in/after the changed
2398 * lines. */
2399 if (curwin->w_cursor.lnum >= lnum + count)
2400 curwin->w_cursor.lnum += added;
2401 else if (added < 0)
2402 curwin->w_cursor.lnum = lnum;
2403 }
2404 }
2405 changed_lines(lnum, 0, lnum + count, (long)added);
2406
2407 if (dfree != NULL)
2408 {
2409 /* Diff is deleted, update folds in other windows. */
2410#ifdef FEAT_FOLDING
2411 diff_fold_update(dfree, idx_to);
2412#endif
2413 vim_free(dfree);
2414 }
2415 else
2416 /* mark_adjust() may have changed the count in a wrong way */
2417 dp->df_count[idx_to] = new_count;
2418
2419 /* When changing the current buffer, keep track of line numbers */
2420 if (idx_cur == idx_to)
2421 off += added;
2422 }
2423
2424 /* If before the range or not deleted, go to next diff. */
2425 if (dfree == NULL)
2426 {
2427 dprev = dp;
2428 dp = dp->df_next;
2429 }
2430 }
2431
2432 /* restore curwin/curbuf and a few other things */
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002433 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 {
2435 /* Syncing undo only works for the current buffer, but we change
2436 * another buffer. Sync undo if the command was typed. This isn't
2437 * 100% right when ":diffput" is used in a function or mapping. */
2438 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002439 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002440 aucmd_restbuf(&aco);
2441 }
2442
2443 diff_busy = FALSE;
2444
2445 /* Check that the cursor is on a valid character and update it's position.
2446 * When there were filler lines the topline has become invalid. */
2447 check_cursor();
2448 changed_line_abv_curs();
2449
2450 /* Also need to redraw the other buffers. */
2451 diff_redraw(FALSE);
2452}
2453
2454#ifdef FEAT_FOLDING
2455/*
2456 * Update folds for all diff buffers for entry "dp".
2457 * Skip buffer with index "skip_idx".
2458 * When there are no diffs, all folds are removed.
2459 */
2460 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002461diff_fold_update(diff_T *dp, int skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002462{
2463 int i;
2464 win_T *wp;
2465
Bram Moolenaar29323592016-07-24 22:04:11 +02002466 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002467 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002468 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469 foldUpdate(wp, dp->df_lnum[i],
2470 dp->df_lnum[i] + dp->df_count[i]);
2471}
2472#endif
2473
2474/*
2475 * Return TRUE if buffer "buf" is in diff-mode.
2476 */
2477 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002478diff_mode_buf(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002480 tabpage_T *tp;
2481
Bram Moolenaar29323592016-07-24 22:04:11 +02002482 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002483 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
2484 return TRUE;
2485 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002486}
2487
2488/*
2489 * Move "count" times in direction "dir" to the next diff block.
2490 * Return FAIL if there isn't such a diff block.
2491 */
2492 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002493diff_move_to(int dir, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002494{
2495 int idx;
2496 linenr_T lnum = curwin->w_cursor.lnum;
2497 diff_T *dp;
2498
2499 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002500 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002501 return FAIL;
2502
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002503 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002504 ex_diffupdate(NULL); /* update after a big change */
2505
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002506 if (curtab->tp_first_diff == NULL) /* no diffs today */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002507 return FAIL;
2508
2509 while (--count >= 0)
2510 {
2511 /* Check if already before first diff. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002512 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002513 break;
2514
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002515 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002516 {
2517 if (dp == NULL)
2518 break;
2519 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
2520 || (dir == BACKWARD
2521 && (dp->df_next == NULL
2522 || lnum <= dp->df_next->df_lnum[idx])))
2523 {
2524 lnum = dp->df_lnum[idx];
2525 break;
2526 }
2527 }
2528 }
2529
2530 /* don't end up past the end of the file */
2531 if (lnum > curbuf->b_ml.ml_line_count)
2532 lnum = curbuf->b_ml.ml_line_count;
2533
2534 /* When the cursor didn't move at all we fail. */
2535 if (lnum == curwin->w_cursor.lnum)
2536 return FAIL;
2537
2538 setpcmark();
2539 curwin->w_cursor.lnum = lnum;
2540 curwin->w_cursor.col = 0;
2541
2542 return OK;
2543}
2544
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002545/*
2546 * Return the line number in the current window that is closest to "lnum1" in
2547 * "buf1" in diff mode.
2548 */
2549 static linenr_T
2550diff_get_corresponding_line_int(
Bram Moolenaar7454a062016-01-30 15:14:10 +01002551 buf_T *buf1,
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002552 linenr_T lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02002553{
2554 int idx1;
2555 int idx2;
2556 diff_T *dp;
2557 int baseline = 0;
Bram Moolenaar860cae12010-06-05 23:22:07 +02002558
2559 idx1 = diff_buf_idx(buf1);
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002560 idx2 = diff_buf_idx(curbuf);
Bram Moolenaar860cae12010-06-05 23:22:07 +02002561 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
2562 return lnum1;
2563
2564 if (curtab->tp_diff_invalid)
2565 ex_diffupdate(NULL); /* update after a big change */
2566
2567 if (curtab->tp_first_diff == NULL) /* no diffs today */
2568 return lnum1;
2569
2570 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
2571 {
2572 if (dp->df_lnum[idx1] > lnum1)
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002573 return lnum1 - baseline;
2574 if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02002575 {
2576 /* Inside the diffblock */
2577 baseline = lnum1 - dp->df_lnum[idx1];
2578 if (baseline > dp->df_count[idx2])
2579 baseline = dp->df_count[idx2];
2580
2581 return dp->df_lnum[idx2] + baseline;
2582 }
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002583 if ( (dp->df_lnum[idx1] == lnum1)
2584 && (dp->df_count[idx1] == 0)
2585 && (dp->df_lnum[idx2] <= curwin->w_cursor.lnum)
2586 && ((dp->df_lnum[idx2] + dp->df_count[idx2])
2587 > curwin->w_cursor.lnum))
Bram Moolenaar860cae12010-06-05 23:22:07 +02002588 /*
2589 * Special case: if the cursor is just after a zero-count
2590 * block (i.e. all filler) and the target cursor is already
2591 * inside the corresponding block, leave the target cursor
2592 * unmoved. This makes repeated CTRL-W W operations work
2593 * as expected.
2594 */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002595 return curwin->w_cursor.lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02002596 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
2597 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
2598 }
2599
2600 /* If we get here then the cursor is after the last diff */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002601 return lnum1 - baseline;
2602}
Bram Moolenaar860cae12010-06-05 23:22:07 +02002603
Bram Moolenaar025e3e02016-10-18 14:50:18 +02002604/*
2605 * Return the line number in the current window that is closest to "lnum1" in
2606 * "buf1" in diff mode. Checks the line number to be valid.
2607 */
2608 linenr_T
2609diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1)
2610{
2611 linenr_T lnum = diff_get_corresponding_line_int(buf1, lnum1);
2612
2613 /* don't end up past the end of the file */
2614 if (lnum > curbuf->b_ml.ml_line_count)
2615 return curbuf->b_ml.ml_line_count;
2616 return lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02002617}
Bram Moolenaar860cae12010-06-05 23:22:07 +02002618
Bram Moolenaar071d4272004-06-13 20:20:40 +00002619/*
2620 * For line "lnum" in the current window find the equivalent lnum in window
2621 * "wp", compensating for inserted/deleted lines.
2622 */
2623 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01002624diff_lnum_win(linenr_T lnum, win_T *wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002625{
2626 diff_T *dp;
2627 int idx;
2628 int i;
2629 linenr_T n;
2630
2631 idx = diff_buf_idx(curbuf);
2632 if (idx == DB_COUNT) /* safety check */
2633 return (linenr_T)0;
2634
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002635 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002636 ex_diffupdate(NULL); /* update after a big change */
2637
2638 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002639 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002640 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2641 break;
2642
2643 /* When after the last change, compute relative to the last line number. */
2644 if (dp == NULL)
2645 return wp->w_buffer->b_ml.ml_line_count
2646 - (curbuf->b_ml.ml_line_count - lnum);
2647
2648 /* Find index for "wp". */
2649 i = diff_buf_idx(wp->w_buffer);
2650 if (i == DB_COUNT) /* safety check */
2651 return (linenr_T)0;
2652
2653 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
2654 if (n > dp->df_lnum[i] + dp->df_count[i])
2655 n = dp->df_lnum[i] + dp->df_count[i];
2656 return n;
2657}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002658
2659#endif /* FEAT_DIFF */