blob: 05ca7ac250a010221a7babb7b304a395ded509f0 [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vim:set ts=8 sts=4 sw=4:
2 *
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 Moolenaar49d7bf12006-02-17 21:45:41 +0000142 * Find buffer "buf" in the list of diff buffers for the current tab page.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143 * Return its index or DB_COUNT if not found.
144 */
145 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100146diff_buf_idx(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000147{
148 int idx;
149
150 for (idx = 0; idx < DB_COUNT; ++idx)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000151 if (curtab->tp_diffbuf[idx] == buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 break;
153 return idx;
154}
155
156/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000157 * Find buffer "buf" in the list of diff buffers for tab page "tp".
158 * Return its index or DB_COUNT if not found.
159 */
160 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100161diff_buf_idx_tp(buf_T *buf, tabpage_T *tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000162{
163 int idx;
164
165 for (idx = 0; idx < DB_COUNT; ++idx)
166 if (tp->tp_diffbuf[idx] == buf)
167 break;
168 return idx;
169}
170
171/*
172 * Mark the diff info involving buffer "buf" as invalid, it will be updated
173 * when info is requested.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000174 */
175 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100176diff_invalidate(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000177{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000178 tabpage_T *tp;
179 int i;
180
Bram Moolenaar29323592016-07-24 22:04:11 +0200181 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000182 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000183 i = diff_buf_idx_tp(buf, tp);
184 if (i != DB_COUNT)
185 {
186 tp->tp_diff_invalid = TRUE;
187 if (tp == curtab)
188 diff_redraw(TRUE);
189 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000190 }
191}
192
193/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000194 * Called by mark_adjust(): update line numbers in "curbuf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000195 */
196 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100197diff_mark_adjust(
198 linenr_T line1,
199 linenr_T line2,
200 long amount,
201 long amount_after)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000202{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000203 int idx;
204 tabpage_T *tp;
205
206 /* Handle all tab pages that use the current buffer in a diff. */
Bram Moolenaar29323592016-07-24 22:04:11 +0200207 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000208 {
209 idx = diff_buf_idx_tp(curbuf, tp);
210 if (idx != DB_COUNT)
211 diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after);
212 }
213}
214
215/*
216 * Update line numbers in tab page "tp" for "curbuf" with index "idx".
217 * This attempts to update the changes as much as possible:
218 * When inserting/deleting lines outside of existing change blocks, create a
219 * new change block and update the line numbers in following blocks.
220 * When inserting/deleting lines in existing change blocks, update them.
221 */
222 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100223diff_mark_adjust_tp(
224 tabpage_T *tp,
225 int idx,
226 linenr_T line1,
227 linenr_T line2,
228 long amount,
229 long amount_after)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000230{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000231 diff_T *dp;
232 diff_T *dprev;
233 diff_T *dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000234 int i;
235 int inserted, deleted;
236 int n, off;
237 linenr_T last;
238 linenr_T lnum_deleted = line1; /* lnum of remaining deletion */
239 int check_unchanged;
240
Bram Moolenaar071d4272004-06-13 20:20:40 +0000241 if (line2 == MAXLNUM)
242 {
243 /* mark_adjust(99, MAXLNUM, 9, 0): insert lines */
244 inserted = amount;
245 deleted = 0;
246 }
247 else if (amount_after > 0)
248 {
249 /* mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines*/
250 inserted = amount_after;
251 deleted = 0;
252 }
253 else
254 {
255 /* mark_adjust(98, 99, MAXLNUM, -2): delete lines */
256 inserted = 0;
257 deleted = -amount_after;
258 }
259
260 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000261 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000262 for (;;)
263 {
264 /* If the change is after the previous diff block and before the next
265 * diff block, thus not touching an existing change, create a new diff
266 * block. Don't do this when ex_diffgetput() is busy. */
267 if ((dp == NULL || dp->df_lnum[idx] - 1 > line2
268 || (line2 == MAXLNUM && dp->df_lnum[idx] > line1))
269 && (dprev == NULL
270 || dprev->df_lnum[idx] + dprev->df_count[idx] < line1)
271 && !diff_busy)
272 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000273 dnext = diff_alloc_new(tp, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000274 if (dnext == NULL)
275 return;
276
277 dnext->df_lnum[idx] = line1;
278 dnext->df_count[idx] = inserted;
279 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000280 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000281 {
282 if (dprev == NULL)
283 dnext->df_lnum[i] = line1;
284 else
285 dnext->df_lnum[i] = line1
286 + (dprev->df_lnum[i] + dprev->df_count[i])
287 - (dprev->df_lnum[idx] + dprev->df_count[idx]);
288 dnext->df_count[i] = deleted;
289 }
290 }
291
292 /* if at end of the list, quit */
293 if (dp == NULL)
294 break;
295
296 /*
297 * Check for these situations:
298 * 1 2 3
299 * 1 2 3
300 * line1 2 3 4 5
301 * 2 3 4 5
302 * 2 3 4 5
303 * line2 2 3 4 5
304 * 3 5 6
305 * 3 5 6
306 */
307 /* compute last line of this change */
308 last = dp->df_lnum[idx] + dp->df_count[idx] - 1;
309
310 /* 1. change completely above line1: nothing to do */
311 if (last >= line1 - 1)
312 {
313 /* 6. change below line2: only adjust for amount_after; also when
314 * "deleted" became zero when deleted all lines between two diffs */
315 if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2)
316 {
317 if (amount_after == 0)
318 break; /* nothing left to change */
319 dp->df_lnum[idx] += amount_after;
320 }
321 else
322 {
323 check_unchanged = FALSE;
324
325 /* 2. 3. 4. 5.: inserted/deleted lines touching this diff. */
326 if (deleted > 0)
327 {
328 if (dp->df_lnum[idx] >= line1)
329 {
330 off = dp->df_lnum[idx] - lnum_deleted;
331 if (last <= line2)
332 {
333 /* 4. delete all lines of diff */
334 if (dp->df_next != NULL
335 && dp->df_next->df_lnum[idx] - 1 <= line2)
336 {
337 /* delete continues in next diff, only do
338 * lines until that one */
339 n = dp->df_next->df_lnum[idx] - lnum_deleted;
340 deleted -= n;
341 n -= dp->df_count[idx];
342 lnum_deleted = dp->df_next->df_lnum[idx];
343 }
344 else
345 n = deleted - dp->df_count[idx];
346 dp->df_count[idx] = 0;
347 }
348 else
349 {
350 /* 5. delete lines at or just before top of diff */
351 n = off;
352 dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1;
353 check_unchanged = TRUE;
354 }
355 dp->df_lnum[idx] = line1;
356 }
357 else
358 {
359 off = 0;
360 if (last < line2)
361 {
362 /* 2. delete at end of of diff */
363 dp->df_count[idx] -= last - lnum_deleted + 1;
364 if (dp->df_next != NULL
365 && dp->df_next->df_lnum[idx] - 1 <= line2)
366 {
367 /* delete continues in next diff, only do
368 * lines until that one */
369 n = dp->df_next->df_lnum[idx] - 1 - last;
370 deleted -= dp->df_next->df_lnum[idx]
371 - lnum_deleted;
372 lnum_deleted = dp->df_next->df_lnum[idx];
373 }
374 else
375 n = line2 - last;
376 check_unchanged = TRUE;
377 }
378 else
379 {
380 /* 3. delete lines inside the diff */
381 n = 0;
382 dp->df_count[idx] -= deleted;
383 }
384 }
385
386 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000387 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000388 {
389 dp->df_lnum[i] -= off;
390 dp->df_count[i] += n;
391 }
392 }
393 else
394 {
395 if (dp->df_lnum[idx] <= line1)
396 {
397 /* inserted lines somewhere in this diff */
398 dp->df_count[idx] += inserted;
399 check_unchanged = TRUE;
400 }
401 else
402 /* inserted lines somewhere above this diff */
403 dp->df_lnum[idx] += inserted;
404 }
405
406 if (check_unchanged)
407 /* Check if inserted lines are equal, may reduce the
408 * size of the diff. TODO: also check for equal lines
409 * in the middle and perhaps split the block. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000410 diff_check_unchanged(tp, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000411 }
412 }
413
414 /* check if this block touches the previous one, may merge them. */
415 if (dprev != NULL && dprev->df_lnum[idx] + dprev->df_count[idx]
416 == dp->df_lnum[idx])
417 {
418 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000419 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420 dprev->df_count[i] += dp->df_count[i];
421 dprev->df_next = dp->df_next;
422 vim_free(dp);
423 dp = dprev->df_next;
424 }
425 else
426 {
427 /* Advance to next entry. */
428 dprev = dp;
429 dp = dp->df_next;
430 }
431 }
432
433 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000434 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000435 while (dp != NULL)
436 {
437 /* All counts are zero, remove this entry. */
438 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000439 if (tp->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000440 break;
441 if (i == DB_COUNT)
442 {
443 dnext = dp->df_next;
444 vim_free(dp);
445 dp = dnext;
446 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000447 tp->tp_first_diff = dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000448 else
449 dprev->df_next = dnext;
450 }
451 else
452 {
453 /* Advance to next entry. */
454 dprev = dp;
455 dp = dp->df_next;
456 }
457
458 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000459
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000460 if (tp == curtab)
461 {
462 diff_redraw(TRUE);
463
464 /* Need to recompute the scroll binding, may remove or add filler
465 * lines (e.g., when adding lines above w_topline). But it's slow when
466 * making many changes, postpone until redrawing. */
467 diff_need_scrollbind = TRUE;
468 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000469}
470
471/*
472 * Allocate a new diff block and link it between "dprev" and "dp".
473 */
474 static diff_T *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100475diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000476{
477 diff_T *dnew;
478
479 dnew = (diff_T *)alloc((unsigned)sizeof(diff_T));
480 if (dnew != NULL)
481 {
482 dnew->df_next = dp;
483 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000484 tp->tp_first_diff = dnew;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000485 else
486 dprev->df_next = dnew;
487 }
488 return dnew;
489}
490
491/*
492 * Check if the diff block "dp" can be made smaller for lines at the start and
493 * end that are equal. Called after inserting lines.
494 * This may result in a change where all buffers have zero lines, the caller
495 * must take care of removing it.
496 */
497 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100498diff_check_unchanged(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000499{
500 int i_org;
501 int i_new;
502 int off_org, off_new;
503 char_u *line_org;
504 int dir = FORWARD;
505
506 /* Find the first buffers, use it as the original, compare the other
507 * buffer lines against this one. */
508 for (i_org = 0; i_org < DB_COUNT; ++i_org)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000509 if (tp->tp_diffbuf[i_org] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000510 break;
511 if (i_org == DB_COUNT) /* safety check */
512 return;
513
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000514 if (diff_check_sanity(tp, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000515 return;
516
517 /* First check lines at the top, then at the bottom. */
518 off_org = 0;
519 off_new = 0;
520 for (;;)
521 {
522 /* Repeat until a line is found which is different or the number of
523 * lines has become zero. */
524 while (dp->df_count[i_org] > 0)
525 {
526 /* Copy the line, the next ml_get() will invalidate it. */
527 if (dir == BACKWARD)
528 off_org = dp->df_count[i_org] - 1;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000529 line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000530 dp->df_lnum[i_org] + off_org, FALSE));
531 if (line_org == NULL)
532 return;
533 for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new)
534 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000535 if (tp->tp_diffbuf[i_new] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000536 continue;
537 if (dir == BACKWARD)
538 off_new = dp->df_count[i_new] - 1;
539 /* if other buffer doesn't have this line, it was inserted */
540 if (off_new < 0 || off_new >= dp->df_count[i_new])
541 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000542 if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000543 dp->df_lnum[i_new] + off_new, FALSE)) != 0)
544 break;
545 }
546 vim_free(line_org);
547
548 /* Stop when a line isn't equal in all diff buffers. */
549 if (i_new != DB_COUNT)
550 break;
551
552 /* Line matched in all buffers, remove it from the diff. */
553 for (i_new = i_org; i_new < DB_COUNT; ++i_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000554 if (tp->tp_diffbuf[i_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 {
556 if (dir == FORWARD)
557 ++dp->df_lnum[i_new];
558 --dp->df_count[i_new];
559 }
560 }
561 if (dir == BACKWARD)
562 break;
563 dir = BACKWARD;
564 }
565}
566
567/*
568 * Check if a diff block doesn't contain invalid line numbers.
569 * This can happen when the diff program returns invalid results.
570 */
571 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100572diff_check_sanity(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000573{
574 int i;
575
576 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000577 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000578 if (dp->df_lnum[i] + dp->df_count[i] - 1
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000579 > tp->tp_diffbuf[i]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000580 return FAIL;
581 return OK;
582}
583
584/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000585 * Mark all diff buffers in the current tab page for redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000586 */
587 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100588diff_redraw(
589 int dofold) /* also recompute the folds */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590{
591 win_T *wp;
592 int n;
593
Bram Moolenaar29323592016-07-24 22:04:11 +0200594 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000595 if (wp->w_p_diff)
596 {
Bram Moolenaar60f83772006-03-12 21:52:47 +0000597 redraw_win_later(wp, SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000598#ifdef FEAT_FOLDING
599 if (dofold && foldmethodIsDiff(wp))
600 foldUpdateAll(wp);
601#endif
602 /* A change may have made filler lines invalid, need to take care
603 * of that for other windows. */
Bram Moolenaara80888d2012-10-21 22:18:21 +0200604 n = diff_check(wp, wp->w_topline);
605 if ((wp != curwin && wp->w_topfill > 0) || n > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000606 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000607 if (wp->w_topfill > n)
608 wp->w_topfill = (n < 0 ? 0 : n);
Bram Moolenaara80888d2012-10-21 22:18:21 +0200609 else if (n > 0 && n > wp->w_topfill)
610 wp->w_topfill = n;
Bram Moolenaar846a2ff2014-05-28 11:35:37 +0200611 check_topfill(wp, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000612 }
613 }
614}
615
616/*
617 * Write buffer "buf" to file "name".
618 * Always use 'fileformat' set to "unix".
619 * Return FAIL for failure
620 */
621 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100622diff_write(buf_T *buf, char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000623{
624 int r;
625 char_u *save_ff;
626
627 save_ff = buf->b_p_ff;
628 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX);
629 r = buf_write(buf, fname, NULL, (linenr_T)1, buf->b_ml.ml_line_count,
630 NULL, FALSE, FALSE, FALSE, TRUE);
631 free_string_option(buf->b_p_ff);
632 buf->b_p_ff = save_ff;
633 return r;
634}
635
636/*
637 * Completely update the diffs for the buffers involved.
638 * This uses the ordinary "diff" command.
639 * The buffers are written to a file, also for unmodified buffers (the file
640 * could have been produced by autocommands, e.g. the netrw plugin).
641 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000642 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100643ex_diffupdate(
Bram Moolenaarf1d25012016-03-03 12:22:53 +0100644 exarg_T *eap) /* can be NULL */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645{
646 buf_T *buf;
647 int idx_orig;
648 int idx_new;
649 char_u *tmp_orig;
650 char_u *tmp_new;
651 char_u *tmp_diff;
652 FILE *fd;
653 int ok;
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000654 int io_error = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000655
656 /* Delete all diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000657 diff_clear(curtab);
658 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659
660 /* Use the first buffer as the original text. */
661 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000662 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000663 break;
664 if (idx_orig == DB_COUNT)
665 return;
666
667 /* Only need to do something when there is another buffer. */
668 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000669 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 break;
671 if (idx_new == DB_COUNT)
672 return;
673
674 /* We need three temp file names. */
Bram Moolenaare5c421c2015-03-31 13:33:08 +0200675 tmp_orig = vim_tempname('o', TRUE);
676 tmp_new = vim_tempname('n', TRUE);
677 tmp_diff = vim_tempname('d', TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000678 if (tmp_orig == NULL || tmp_new == NULL || tmp_diff == NULL)
679 goto theend;
680
681 /*
682 * Do a quick test if "diff" really works. Otherwise it looks like there
683 * are no differences. Can't use the return value, it's non-zero when
684 * there are differences.
685 * May try twice, first with "-a" and then without.
686 */
687 for (;;)
688 {
689 ok = FALSE;
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000690 fd = mch_fopen((char *)tmp_orig, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000691 if (fd == NULL)
692 io_error = TRUE;
693 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000694 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000695 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
696 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000697 fclose(fd);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000698 fd = mch_fopen((char *)tmp_new, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000699 if (fd == NULL)
700 io_error = TRUE;
701 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000702 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000703 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
704 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000705 fclose(fd);
706 diff_file(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000707 fd = mch_fopen((char *)tmp_diff, "r");
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 {
712 char_u linebuf[LBUFLEN];
713
714 for (;;)
715 {
716 /* There must be a line that contains "1c1". */
717 if (tag_fgets(linebuf, LBUFLEN, fd))
718 break;
719 if (STRNCMP(linebuf, "1c1", 3) == 0)
720 ok = TRUE;
721 }
722 fclose(fd);
723 }
724 mch_remove(tmp_diff);
725 mch_remove(tmp_new);
726 }
727 mch_remove(tmp_orig);
728 }
729
730#ifdef FEAT_EVAL
731 /* When using 'diffexpr' break here. */
732 if (*p_dex != NUL)
733 break;
734#endif
735
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100736#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000737 /* If the "-a" argument works, also check if "--binary" works. */
738 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
739 {
740 diff_a_works = TRUE;
741 diff_bin_works = TRUE;
742 continue;
743 }
744 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
745 {
746 /* Tried --binary, but it failed. "-a" works though. */
747 diff_bin_works = FALSE;
748 ok = TRUE;
749 }
750#endif
751
752 /* If we checked if "-a" works already, break here. */
753 if (diff_a_works != MAYBE)
754 break;
755 diff_a_works = ok;
756
757 /* If "-a" works break here, otherwise retry without "-a". */
758 if (ok)
759 break;
760 }
761 if (!ok)
762 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000763 if (io_error)
764 EMSG(_("E810: Cannot read or write temp files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000765 EMSG(_("E97: Cannot create diffs"));
766 diff_a_works = MAYBE;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100767#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000768 diff_bin_works = MAYBE;
769#endif
770 goto theend;
771 }
772
Bram Moolenaarbd1d5602012-05-18 18:47:17 +0200773 /* :diffupdate! */
774 if (eap != NULL && eap->forceit)
775 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new)
776 {
777 buf = curtab->tp_diffbuf[idx_new];
778 if (buf_valid(buf))
779 buf_check_timestamp(buf, FALSE);
780 }
781
Bram Moolenaar071d4272004-06-13 20:20:40 +0000782 /* Write the first buffer to a tempfile. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000783 buf = curtab->tp_diffbuf[idx_orig];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000784 if (diff_write(buf, tmp_orig) == FAIL)
785 goto theend;
786
787 /* Make a difference between the first buffer and every other. */
788 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
789 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000790 buf = curtab->tp_diffbuf[idx_new];
Bram Moolenaar9dd33af2015-08-04 21:51:25 +0200791 if (buf == NULL || buf->b_ml.ml_mfp == NULL)
792 continue; /* skip buffer that isn't loaded */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000793 if (diff_write(buf, tmp_new) == FAIL)
794 continue;
795 diff_file(tmp_orig, tmp_new, tmp_diff);
796
797 /* Read the diff output and add each entry to the diff list. */
798 diff_read(idx_orig, idx_new, tmp_diff);
799 mch_remove(tmp_diff);
800 mch_remove(tmp_new);
801 }
802 mch_remove(tmp_orig);
803
Bram Moolenaar60a44dc2007-10-19 16:58:12 +0000804 /* force updating cursor position on screen */
805 curwin->w_valid_cursor.lnum = 0;
806
Bram Moolenaar071d4272004-06-13 20:20:40 +0000807 diff_redraw(TRUE);
808
809theend:
810 vim_free(tmp_orig);
811 vim_free(tmp_new);
812 vim_free(tmp_diff);
813}
814
815/*
816 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
817 */
818 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100819diff_file(
820 char_u *tmp_orig,
821 char_u *tmp_new,
822 char_u *tmp_diff)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000823{
824 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000825 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000826
827#ifdef FEAT_EVAL
828 if (*p_dex != NUL)
829 /* Use 'diffexpr' to generate the diff file. */
830 eval_diff(tmp_orig, tmp_new, tmp_diff);
831 else
832#endif
833 {
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000834 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
835 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
836 cmd = alloc((unsigned)len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000837 if (cmd != NULL)
838 {
Bram Moolenaar95fb60a2005-03-08 22:29:20 +0000839 /* We don't want $DIFF_OPTIONS to get in the way. */
840 if (getenv("DIFF_OPTIONS"))
841 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
842
Bram Moolenaar071d4272004-06-13 20:20:40 +0000843 /* Build the diff command and execute it. Always use -a, binary
844 * differences are of no use. Ignore errors, diff returns
845 * non-zero when differences have been found. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000846 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +0000847 diff_a_works == FALSE ? "" : "-a ",
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100848#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000849 diff_bin_works == TRUE ? "--binary " : "",
850#else
851 "",
852#endif
853 (diff_flags & DIFF_IWHITE) ? "-b " : "",
854 (diff_flags & DIFF_ICASE) ? "-i " : "",
855 tmp_orig, tmp_new);
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000856 append_redir(cmd, (int)len, p_srr, tmp_diff);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000857#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000858 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000859#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000860 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000861#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000862 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000863#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000864 vim_free(cmd);
865 }
866 }
867}
868
869/*
870 * Create a new version of a file from the current buffer and a diff file.
871 * The buffer is written to a file, also for unmodified buffers (the file
872 * could have been produced by autocommands, e.g. the netrw plugin).
873 */
874 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100875ex_diffpatch(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000876{
877 char_u *tmp_orig; /* name of original temp file */
878 char_u *tmp_new; /* name of patched temp file */
879 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000880 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000881 win_T *old_curwin = curwin;
882 char_u *newname = NULL; /* name of patched file buffer */
883#ifdef UNIX
884 char_u dirbuf[MAXPATHL];
885 char_u *fullname = NULL;
886#endif
887#ifdef FEAT_BROWSE
888 char_u *browseFile = NULL;
889 int browse_flag = cmdmod.browse;
890#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +0200891 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000892
893#ifdef FEAT_BROWSE
894 if (cmdmod.browse)
895 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +0000896 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaar071d4272004-06-13 20:20:40 +0000897 eap->arg, NULL, NULL, BROWSE_FILTER_ALL_FILES, NULL);
898 if (browseFile == NULL)
899 return; /* operation cancelled */
900 eap->arg = browseFile;
901 cmdmod.browse = FALSE; /* don't let do_ecmd() browse again */
902 }
903#endif
904
905 /* We need two temp file names. */
Bram Moolenaare5c421c2015-03-31 13:33:08 +0200906 tmp_orig = vim_tempname('o', FALSE);
907 tmp_new = vim_tempname('n', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000908 if (tmp_orig == NULL || tmp_new == NULL)
909 goto theend;
910
911 /* Write the current buffer to "tmp_orig". */
912 if (buf_write(curbuf, tmp_orig, NULL,
913 (linenr_T)1, curbuf->b_ml.ml_line_count,
914 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
915 goto theend;
916
917#ifdef UNIX
918 /* Get the absolute path of the patchfile, changing directory below. */
919 fullname = FullName_save(eap->arg, FALSE);
920#endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000921 buflen = STRLEN(tmp_orig) + (
Bram Moolenaar071d4272004-06-13 20:20:40 +0000922# ifdef UNIX
923 fullname != NULL ? STRLEN(fullname) :
924# endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000925 STRLEN(eap->arg)) + STRLEN(tmp_new) + 16;
926 buf = alloc((unsigned)buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000927 if (buf == NULL)
928 goto theend;
929
930#ifdef UNIX
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000931 /* Temporarily chdir to /tmp, to avoid patching files in the current
Bram Moolenaar071d4272004-06-13 20:20:40 +0000932 * directory when the patch file contains more than one patch. When we
933 * have our own temp dir use that instead, it will be cleaned up when we
934 * exit (any .rej files created). Don't change directory if we can't
935 * return to the current. */
936 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
937 dirbuf[0] = NUL;
938 else
939 {
940# ifdef TEMPDIRNAMES
941 if (vim_tempdir != NULL)
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000942 ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000943 else
944# endif
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000945 ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +0000946 shorten_fnames(TRUE);
947 }
948#endif
949
950#ifdef FEAT_EVAL
951 if (*p_pex != NUL)
952 /* Use 'patchexpr' to generate the new file. */
953 eval_patch(tmp_orig,
954# ifdef UNIX
955 fullname != NULL ? fullname :
956# endif
957 eap->arg, tmp_new);
958 else
959#endif
960 {
961 /* Build the patch command and execute it. Ignore errors. Switch to
962 * cooked mode to allow the user to respond to prompts. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000963 vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"",
964 tmp_new, tmp_orig,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000965# ifdef UNIX
966 fullname != NULL ? fullname :
967# endif
968 eap->arg);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000969#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000970 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000971#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000972 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000973#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000974 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000975#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000976 }
977
978#ifdef UNIX
979 if (dirbuf[0] != NUL)
980 {
981 if (mch_chdir((char *)dirbuf) != 0)
982 EMSG(_(e_prev_dir));
983 shorten_fnames(TRUE);
984 }
985#endif
986
987 /* patch probably has written over the screen */
988 redraw_later(CLEAR);
989
990 /* Delete any .orig or .rej file created. */
991 STRCPY(buf, tmp_new);
992 STRCAT(buf, ".orig");
993 mch_remove(buf);
994 STRCPY(buf, tmp_new);
995 STRCAT(buf, ".rej");
996 mch_remove(buf);
997
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +0000998 /* Only continue if the output file was created. */
999 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
1000 EMSG(_("E816: Cannot read patch output"));
1001 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001002 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001003 if (curbuf->b_fname != NULL)
1004 {
1005 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001006 (int)(STRLEN(curbuf->b_fname) + 4));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001007 if (newname != NULL)
1008 STRCAT(newname, ".new");
1009 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001010
1011#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001012 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001013#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001014 /* don't use a new tab page, each tab page has its own diffs */
1015 cmdmod.tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001016
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001017 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001018 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001019 /* Pretend it was a ":split fname" command */
1020 eap->cmdidx = CMD_split;
1021 eap->arg = tmp_new;
1022 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001023
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001024 /* check that split worked and editing tmp_new */
1025 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001026 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001027 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1028 diff_win_options(curwin, TRUE);
1029 diff_win_options(old_curwin, TRUE);
1030
1031 if (newname != NULL)
1032 {
1033 /* do a ":file filename.new" on the patched buffer */
1034 eap->arg = newname;
1035 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001036
1037#ifdef FEAT_AUTOCMD
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001038 /* Do filetype detection with the new name. */
1039 if (au_has_group((char_u *)"filetypedetect"))
1040 do_cmdline_cmd((char_u *)":doau filetypedetect BufRead");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001041#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001042 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001043 }
1044 }
1045 }
1046
1047theend:
1048 if (tmp_orig != NULL)
1049 mch_remove(tmp_orig);
1050 vim_free(tmp_orig);
1051 if (tmp_new != NULL)
1052 mch_remove(tmp_new);
1053 vim_free(tmp_new);
1054 vim_free(newname);
1055 vim_free(buf);
1056#ifdef UNIX
1057 vim_free(fullname);
1058#endif
1059#ifdef FEAT_BROWSE
1060 vim_free(browseFile);
1061 cmdmod.browse = browse_flag;
1062#endif
1063}
1064
1065/*
1066 * Split the window and edit another file, setting options to show the diffs.
1067 */
1068 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001069ex_diffsplit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001070{
1071 win_T *old_curwin = curwin;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001072 bufref_T old_curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001073
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001074 set_bufref(&old_curbuf, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001075#ifdef FEAT_GUI
1076 need_mouse_correct = TRUE;
1077#endif
Bram Moolenaar46328f92016-08-28 15:39:57 +02001078 /* Need to compute w_fraction when no redraw happened yet. */
1079 validate_cursor();
1080 set_fraction(curwin);
1081
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001082 /* don't use a new tab page, each tab page has its own diffs */
1083 cmdmod.tab = 0;
1084
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001085 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001086 {
1087 /* Pretend it was a ":split fname" command */
1088 eap->cmdidx = CMD_split;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001089 curwin->w_p_diff = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001090 do_exedit(eap, old_curwin);
1091
1092 if (curwin != old_curwin) /* split must have worked */
1093 {
1094 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1095 diff_win_options(curwin, TRUE);
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001096 if (win_valid(old_curwin))
1097 {
1098 diff_win_options(old_curwin, TRUE);
1099
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001100 if (bufref_valid(&old_curbuf))
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001101 /* Move the cursor position to that of the old window. */
1102 curwin->w_cursor.lnum = diff_get_corresponding_line(
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001103 old_curbuf.br_buf,
Bram Moolenaarf29a82d2015-12-17 15:03:55 +01001104 old_curwin->w_cursor.lnum,
1105 curbuf,
1106 curwin->w_cursor.lnum);
1107 }
Bram Moolenaar46328f92016-08-28 15:39:57 +02001108 /* Now that lines are folded scroll to show the cursor at the same
1109 * relative position. */
1110 scroll_to_fraction(curwin, curwin->w_height);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001111 }
1112 }
1113}
1114
1115/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001116 * Set options to show diffs for the current window.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001117 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001118 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001119ex_diffthis(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001120{
1121 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1122 diff_win_options(curwin, TRUE);
1123}
1124
1125/*
1126 * Set options in window "wp" for diff mode.
1127 */
1128 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001129diff_win_options(
1130 win_T *wp,
1131 int addbuf) /* Add buffer to diff. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001133# ifdef FEAT_FOLDING
1134 win_T *old_curwin = curwin;
1135
1136 /* close the manually opened folds */
1137 curwin = wp;
1138 newFoldLevel();
1139 curwin = old_curwin;
1140# endif
1141
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001142 /* Use 'scrollbind' and 'cursorbind' when available */
1143#ifdef FEAT_SCROLLBIND
Bram Moolenaar43929962015-07-03 15:06:56 +02001144 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001145 wp->w_p_scb_save = wp->w_p_scb;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001146 wp->w_p_scb = TRUE;
1147#endif
Bram Moolenaar860cae12010-06-05 23:22:07 +02001148#ifdef FEAT_CURSORBIND
Bram Moolenaar43929962015-07-03 15:06:56 +02001149 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001150 wp->w_p_crb_save = wp->w_p_crb;
Bram Moolenaar860cae12010-06-05 23:22:07 +02001151 wp->w_p_crb = TRUE;
1152#endif
Bram Moolenaar43929962015-07-03 15:06:56 +02001153 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001154 wp->w_p_wrap_save = wp->w_p_wrap;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001155 wp->w_p_wrap = FALSE;
1156# ifdef FEAT_FOLDING
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001157 curwin = wp;
1158 curbuf = curwin->w_buffer;
Bram Moolenaar43929962015-07-03 15:06:56 +02001159 if (!wp->w_p_diff)
1160 {
1161 if (wp->w_p_diff_saved)
1162 free_string_option(wp->w_p_fdm_save);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001163 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm);
Bram Moolenaar43929962015-07-03 15:06:56 +02001164 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001165 set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001166 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001167 curwin = old_curwin;
1168 curbuf = curwin->w_buffer;
Bram Moolenaar43929962015-07-03 15:06:56 +02001169 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001170 {
1171 wp->w_p_fdc_save = wp->w_p_fdc;
1172 wp->w_p_fen_save = wp->w_p_fen;
1173 wp->w_p_fdl_save = wp->w_p_fdl;
1174 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001175 wp->w_p_fdc = diff_foldcolumn;
1176 wp->w_p_fen = TRUE;
1177 wp->w_p_fdl = 0;
1178 foldUpdateAll(wp);
1179 /* make sure topline is not halfway a fold */
1180 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001181# endif
1182#ifdef FEAT_SCROLLBIND
1183 if (vim_strchr(p_sbo, 'h') == NULL)
1184 do_cmdline_cmd((char_u *)"set sbo+=hor");
1185#endif
Bram Moolenaara87aa802013-07-03 15:47:03 +02001186 /* Saved the current values, to be restored in ex_diffoff(). */
1187 wp->w_p_diff_saved = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001188
Bram Moolenaar43929962015-07-03 15:06:56 +02001189 wp->w_p_diff = TRUE;
1190
Bram Moolenaar071d4272004-06-13 20:20:40 +00001191 if (addbuf)
1192 diff_buf_add(wp->w_buffer);
1193 redraw_win_later(wp, NOT_VALID);
1194}
1195
1196/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001197 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001198 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001199 */
1200 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001201ex_diffoff(exarg_T *eap)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001202{
1203 win_T *wp;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001204#ifdef FEAT_SCROLLBIND
1205 int diffwin = FALSE;
1206#endif
1207
Bram Moolenaar29323592016-07-24 22:04:11 +02001208 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001209 {
Bram Moolenaar00462ff2013-09-20 20:13:53 +02001210 if (eap->forceit ? wp->w_p_diff : wp == curwin)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001211 {
Bram Moolenaar43929962015-07-03 15:06:56 +02001212 /* Set 'diff' off. If option values were saved in
1213 * diff_win_options(), restore the ones whose settings seem to have
1214 * been left over from diff mode. */
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001215 wp->w_p_diff = FALSE;
Bram Moolenaara87aa802013-07-03 15:47:03 +02001216
Bram Moolenaara87aa802013-07-03 15:47:03 +02001217 if (wp->w_p_diff_saved)
1218 {
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001219
Bram Moolenaar43929962015-07-03 15:06:56 +02001220#ifdef FEAT_SCROLLBIND
1221 if (wp->w_p_scb)
1222 wp->w_p_scb = wp->w_p_scb_save;
1223#endif
1224#ifdef FEAT_CURSORBIND
1225 if (wp->w_p_crb)
1226 wp->w_p_crb = wp->w_p_crb_save;
1227#endif
1228 if (!wp->w_p_wrap)
1229 wp->w_p_wrap = wp->w_p_wrap_save;
1230#ifdef FEAT_FOLDING
1231 free_string_option(wp->w_p_fdm);
1232 wp->w_p_fdm = vim_strsave(wp->w_p_fdm_save);
1233
1234 if (wp->w_p_fdc == diff_foldcolumn)
1235 wp->w_p_fdc = wp->w_p_fdc_save;
1236 if (wp->w_p_fdl == 0)
1237 wp->w_p_fdl = wp->w_p_fdl_save;
1238
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001239 /* Only restore 'foldenable' when 'foldmethod' is not
1240 * "manual", otherwise we continue to show the diff folds. */
Bram Moolenaar43929962015-07-03 15:06:56 +02001241 if (wp->w_p_fen)
1242 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE
1243 : wp->w_p_fen_save;
1244
1245 foldUpdateAll(wp);
Bram Moolenaar43929962015-07-03 15:06:56 +02001246#endif
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001247 }
Bram Moolenaare67d5462016-08-27 22:40:42 +02001248 /* remove filler lines */
1249 wp->w_topfill = 0;
1250
1251 /* make sure topline is not halfway a fold and cursor is
1252 * invalidated */
1253 changed_window_setting_win(wp);
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001254
Bram Moolenaara87aa802013-07-03 15:47:03 +02001255 /* Note: 'sbo' is not restored, it's a global option. */
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001256 diff_buf_adjust(wp);
1257 }
1258#ifdef FEAT_SCROLLBIND
1259 diffwin |= wp->w_p_diff;
1260#endif
1261 }
1262
1263#ifdef FEAT_SCROLLBIND
1264 /* Remove "hor" from from 'scrollopt' if there are no diff windows left. */
1265 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1266 do_cmdline_cmd((char_u *)"set sbo-=hor");
1267#endif
1268}
1269
1270/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001271 * Read the diff output and add each entry to the diff list.
1272 */
1273 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001274diff_read(
1275 int idx_orig, /* idx of original file */
1276 int idx_new, /* idx of new file */
1277 char_u *fname) /* name of diff output file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001278{
1279 FILE *fd;
1280 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001281 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001282 diff_T *dn, *dpl;
1283 long f1, l1, f2, l2;
1284 char_u linebuf[LBUFLEN]; /* only need to hold the diff line */
1285 int difftype;
1286 char_u *p;
1287 long off;
1288 int i;
1289 linenr_T lnum_orig, lnum_new;
1290 long count_orig, count_new;
1291 int notset = TRUE; /* block "*dp" not set yet */
1292
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001293 fd = mch_fopen((char *)fname, "r");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001294 if (fd == NULL)
1295 {
1296 EMSG(_("E98: Cannot read diff output"));
1297 return;
1298 }
1299
1300 for (;;)
1301 {
1302 if (tag_fgets(linebuf, LBUFLEN, fd))
1303 break; /* end of file */
1304 if (!isdigit(*linebuf))
1305 continue; /* not the start of a diff block */
1306
1307 /* This line must be one of three formats:
1308 * {first}[,{last}]c{first}[,{last}]
1309 * {first}a{first}[,{last}]
1310 * {first}[,{last}]d{first}
1311 */
1312 p = linebuf;
1313 f1 = getdigits(&p);
1314 if (*p == ',')
1315 {
1316 ++p;
1317 l1 = getdigits(&p);
1318 }
1319 else
1320 l1 = f1;
1321 if (*p != 'a' && *p != 'c' && *p != 'd')
1322 continue; /* invalid diff format */
1323 difftype = *p++;
1324 f2 = getdigits(&p);
1325 if (*p == ',')
1326 {
1327 ++p;
1328 l2 = getdigits(&p);
1329 }
1330 else
1331 l2 = f2;
1332 if (l1 < f1 || l2 < f2)
1333 continue; /* invalid line range */
1334
1335 if (difftype == 'a')
1336 {
1337 lnum_orig = f1 + 1;
1338 count_orig = 0;
1339 }
1340 else
1341 {
1342 lnum_orig = f1;
1343 count_orig = l1 - f1 + 1;
1344 }
1345 if (difftype == 'd')
1346 {
1347 lnum_new = f2 + 1;
1348 count_new = 0;
1349 }
1350 else
1351 {
1352 lnum_new = f2;
1353 count_new = l2 - f2 + 1;
1354 }
1355
1356 /* Go over blocks before the change, for which orig and new are equal.
1357 * Copy blocks from orig to new. */
1358 while (dp != NULL
1359 && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig])
1360 {
1361 if (notset)
1362 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1363 dprev = dp;
1364 dp = dp->df_next;
1365 notset = TRUE;
1366 }
1367
1368 if (dp != NULL
1369 && lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig]
1370 && lnum_orig + count_orig >= dp->df_lnum[idx_orig])
1371 {
1372 /* New block overlaps with existing block(s).
1373 * First find last block that overlaps. */
1374 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
1375 if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig])
1376 break;
1377
1378 /* If the newly found block starts before the old one, set the
1379 * start back a number of lines. */
1380 off = dp->df_lnum[idx_orig] - lnum_orig;
1381 if (off > 0)
1382 {
1383 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001384 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001385 dp->df_lnum[i] -= off;
1386 dp->df_lnum[idx_new] = lnum_new;
1387 dp->df_count[idx_new] = count_new;
1388 }
1389 else if (notset)
1390 {
1391 /* new block inside existing one, adjust new block */
1392 dp->df_lnum[idx_new] = lnum_new + off;
1393 dp->df_count[idx_new] = count_new - off;
1394 }
1395 else
1396 /* second overlap of new block with existing block */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001397 dp->df_count[idx_new] += count_new - count_orig
1398 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1399 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001400
1401 /* Adjust the size of the block to include all the lines to the
1402 * end of the existing block or the new diff, whatever ends last. */
1403 off = (lnum_orig + count_orig)
1404 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1405 if (off < 0)
1406 {
1407 /* new change ends in existing block, adjust the end if not
1408 * done already */
1409 if (notset)
1410 dp->df_count[idx_new] += -off;
1411 off = 0;
1412 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001413 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001414 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001415 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1416 - dp->df_lnum[i] + off;
1417
1418 /* Delete the diff blocks that have been merged into one. */
1419 dn = dp->df_next;
1420 dp->df_next = dpl->df_next;
1421 while (dn != dp->df_next)
1422 {
1423 dpl = dn->df_next;
1424 vim_free(dn);
1425 dn = dpl;
1426 }
1427 }
1428 else
1429 {
1430 /* Allocate a new diffblock. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001431 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001432 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001433 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001434
1435 dp->df_lnum[idx_orig] = lnum_orig;
1436 dp->df_count[idx_orig] = count_orig;
1437 dp->df_lnum[idx_new] = lnum_new;
1438 dp->df_count[idx_new] = count_new;
1439
1440 /* Set values for other buffers, these must be equal to the
1441 * original buffer, otherwise there would have been a change
1442 * already. */
1443 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001444 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001445 diff_copy_entry(dprev, dp, idx_orig, i);
1446 }
1447 notset = FALSE; /* "*dp" has been set */
1448 }
1449
1450 /* for remaining diff blocks orig and new are equal */
1451 while (dp != NULL)
1452 {
1453 if (notset)
1454 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1455 dprev = dp;
1456 dp = dp->df_next;
1457 notset = TRUE;
1458 }
1459
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001460done:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001461 fclose(fd);
1462}
1463
1464/*
1465 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1466 */
1467 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001468diff_copy_entry(
1469 diff_T *dprev,
1470 diff_T *dp,
1471 int idx_orig,
1472 int idx_new)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001473{
1474 long off;
1475
1476 if (dprev == NULL)
1477 off = 0;
1478 else
1479 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1480 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1481 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1482 dp->df_count[idx_new] = dp->df_count[idx_orig];
1483}
1484
1485/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001486 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001487 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001488 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001489diff_clear(tabpage_T *tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001490{
1491 diff_T *p, *next_p;
1492
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001493 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001494 {
1495 next_p = p->df_next;
1496 vim_free(p);
1497 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001498 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001499}
1500
1501/*
1502 * Check diff status for line "lnum" in buffer "buf":
1503 * Returns 0 for nothing special
1504 * Returns -1 for a line that should be highlighted as changed.
1505 * Returns -2 for a line that should be highlighted as added/deleted.
1506 * Returns > 0 for inserting that many filler lines above it (never happens
1507 * when 'diffopt' doesn't contain "filler").
1508 * This should only be used for windows where 'diff' is set.
1509 */
1510 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001511diff_check(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001512{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001513 int idx; /* index in tp_diffbuf[] for this buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001514 diff_T *dp;
1515 int maxcount;
1516 int i;
1517 buf_T *buf = wp->w_buffer;
1518 int cmp;
1519
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001520 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001521 ex_diffupdate(NULL); /* update after a big change */
1522
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001523 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) /* no diffs at all */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001524 return 0;
1525
1526 /* safety check: "lnum" must be a buffer line */
1527 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1528 return 0;
1529
1530 idx = diff_buf_idx(buf);
1531 if (idx == DB_COUNT)
1532 return 0; /* no diffs for buffer "buf" */
1533
1534#ifdef FEAT_FOLDING
1535 /* A closed fold never has filler lines. */
1536 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1537 return 0;
1538#endif
1539
1540 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001541 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001542 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1543 break;
1544 if (dp == NULL || lnum < dp->df_lnum[idx])
1545 return 0;
1546
1547 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1548 {
1549 int zero = FALSE;
1550
1551 /* Changed or inserted line. If the other buffers have a count of
1552 * zero, the lines were inserted. If the other buffers have the same
1553 * count, check if the lines are identical. */
1554 cmp = FALSE;
1555 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001556 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001557 {
1558 if (dp->df_count[i] == 0)
1559 zero = TRUE;
1560 else
1561 {
1562 if (dp->df_count[i] != dp->df_count[idx])
1563 return -1; /* nr of lines changed. */
1564 cmp = TRUE;
1565 }
1566 }
1567 if (cmp)
1568 {
1569 /* Compare all lines. If they are equal the lines were inserted
1570 * in some buffers, deleted in others, but not changed. */
1571 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001572 if (i != idx && curtab->tp_diffbuf[i] != NULL
1573 && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574 if (!diff_equal_entry(dp, idx, i))
1575 return -1;
1576 }
1577 /* If there is no buffer with zero lines then there is no difference
1578 * any longer. Happens when making a change (or undo) that removes
1579 * the difference. Can't remove the entry here, we might be halfway
1580 * updating the window. Just report the text as unchanged. Other
1581 * windows might still show the change though. */
1582 if (zero == FALSE)
1583 return 0;
1584 return -2;
1585 }
1586
1587 /* If 'diffopt' doesn't contain "filler", return 0. */
1588 if (!(diff_flags & DIFF_FILLER))
1589 return 0;
1590
1591 /* Insert filler lines above the line just below the change. Will return
1592 * 0 when this buf had the max count. */
1593 maxcount = 0;
1594 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001595 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001596 maxcount = dp->df_count[i];
1597 return maxcount - dp->df_count[idx];
1598}
1599
1600/*
1601 * Compare two entries in diff "*dp" and return TRUE if they are equal.
1602 */
1603 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001604diff_equal_entry(diff_T *dp, int idx1, int idx2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001605{
1606 int i;
1607 char_u *line;
1608 int cmp;
1609
1610 if (dp->df_count[idx1] != dp->df_count[idx2])
1611 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001612 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001613 return FALSE;
1614 for (i = 0; i < dp->df_count[idx1]; ++i)
1615 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001616 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001617 dp->df_lnum[idx1] + i, FALSE));
1618 if (line == NULL)
1619 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001620 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001621 dp->df_lnum[idx2] + i, FALSE));
1622 vim_free(line);
1623 if (cmp != 0)
1624 return FALSE;
1625 }
1626 return TRUE;
1627}
1628
1629/*
1630 * Compare strings "s1" and "s2" according to 'diffopt'.
1631 * Return non-zero when they are different.
1632 */
1633 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001634diff_cmp(char_u *s1, char_u *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001635{
1636 char_u *p1, *p2;
1637#ifdef FEAT_MBYTE
1638 int l;
1639#endif
1640
1641 if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0)
1642 return STRCMP(s1, s2);
1643 if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE))
1644 return MB_STRICMP(s1, s2);
1645
1646 /* Ignore white space changes and possibly ignore case. */
1647 p1 = s1;
1648 p2 = s2;
1649 while (*p1 != NUL && *p2 != NUL)
1650 {
1651 if (vim_iswhite(*p1) && vim_iswhite(*p2))
1652 {
1653 p1 = skipwhite(p1);
1654 p2 = skipwhite(p2);
1655 }
1656 else
1657 {
1658#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001659 l = (*mb_ptr2len)(p1);
1660 if (l != (*mb_ptr2len)(p2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001661 break;
1662 if (l > 1)
1663 {
1664 if (STRNCMP(p1, p2, l) != 0
1665 && (!enc_utf8
1666 || !(diff_flags & DIFF_ICASE)
1667 || utf_fold(utf_ptr2char(p1))
1668 != utf_fold(utf_ptr2char(p2))))
1669 break;
1670 p1 += l;
1671 p2 += l;
1672 }
1673 else
1674#endif
1675 {
1676 if (*p1 != *p2 && (!(diff_flags & DIFF_ICASE)
1677 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
1678 break;
1679 ++p1;
1680 ++p2;
1681 }
1682 }
1683 }
1684
1685 /* Ignore trailing white space. */
1686 p1 = skipwhite(p1);
1687 p2 = skipwhite(p2);
1688 if (*p1 != NUL || *p2 != NUL)
1689 return 1;
1690 return 0;
1691}
1692
1693/*
1694 * Return the number of filler lines above "lnum".
1695 */
1696 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001697diff_check_fill(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001698{
1699 int n;
1700
1701 /* be quick when there are no filler lines */
1702 if (!(diff_flags & DIFF_FILLER))
1703 return 0;
1704 n = diff_check(wp, lnum);
1705 if (n <= 0)
1706 return 0;
1707 return n;
1708}
1709
1710/*
1711 * Set the topline of "towin" to match the position in "fromwin", so that they
1712 * show the same diff'ed lines.
1713 */
1714 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001715diff_set_topline(win_T *fromwin, win_T *towin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001716{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001717 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001718 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001719 int fromidx;
1720 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001721 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001722 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001723 int i;
1724
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001725 fromidx = diff_buf_idx(frombuf);
1726 if (fromidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001727 return; /* safety check */
1728
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001729 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001730 ex_diffupdate(NULL); /* update after a big change */
1731
1732 towin->w_topfill = 0;
1733
1734 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001735 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001736 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001737 break;
1738 if (dp == NULL)
1739 {
1740 /* After last change, compute topline relative to end of file; no
1741 * filler lines. */
1742 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001743 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001744 }
1745 else
1746 {
1747 /* Find index for "towin". */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001748 toidx = diff_buf_idx(towin->w_buffer);
1749 if (toidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001750 return; /* safety check */
1751
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001752 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
1753 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001754 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001755 /* Inside a change: compute filler lines. With three or more
1756 * buffers we need to know the largest count. */
1757 max_count = 0;
1758 for (i = 0; i < DB_COUNT; ++i)
1759 if (curtab->tp_diffbuf[i] != NULL
1760 && max_count < dp->df_count[i])
1761 max_count = dp->df_count[i];
1762
1763 if (dp->df_count[toidx] == dp->df_count[fromidx])
1764 {
1765 /* same number of lines: use same filler count */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001766 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001767 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001768 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001769 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001770 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001771 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001772 /* more lines in towin and fromwin doesn't show diff
1773 * lines, only filler lines */
1774 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
1775 {
1776 /* towin also only shows filler lines */
1777 towin->w_topline = dp->df_lnum[toidx]
1778 + dp->df_count[toidx];
1779 towin->w_topfill = fromwin->w_topfill;
1780 }
1781 else
1782 /* towin still has some diff lines to show */
1783 towin->w_topline = dp->df_lnum[toidx]
1784 + max_count - fromwin->w_topfill;
1785 }
1786 }
1787 else if (towin->w_topline >= dp->df_lnum[toidx]
1788 + dp->df_count[toidx])
1789 {
1790 /* less lines in towin and no diff lines to show: compute
1791 * filler lines */
1792 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
1793 if (diff_flags & DIFF_FILLER)
1794 {
1795 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
1796 /* fromwin is also out of diff lines */
1797 towin->w_topfill = fromwin->w_topfill;
1798 else
1799 /* fromwin has some diff lines */
1800 towin->w_topfill = dp->df_lnum[fromidx]
1801 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001802 }
1803 }
1804 }
1805 }
1806
1807 /* safety check (if diff info gets outdated strange things may happen) */
1808 towin->w_botfill = FALSE;
1809 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
1810 {
1811 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
1812 towin->w_botfill = TRUE;
1813 }
1814 if (towin->w_topline < 1)
1815 {
1816 towin->w_topline = 1;
1817 towin->w_topfill = 0;
1818 }
1819
1820 /* When w_topline changes need to recompute w_botline and cursor position */
1821 invalidate_botline_win(towin);
1822 changed_line_abv_curs_win(towin);
1823
1824 check_topfill(towin, FALSE);
1825#ifdef FEAT_FOLDING
1826 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
1827 NULL, TRUE, NULL);
1828#endif
1829}
1830
1831/*
1832 * This is called when 'diffopt' is changed.
1833 */
1834 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001835diffopt_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001836{
1837 char_u *p;
1838 int diff_context_new = 6;
1839 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001840 int diff_foldcolumn_new = 2;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001841 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842
1843 p = p_dip;
1844 while (*p != NUL)
1845 {
1846 if (STRNCMP(p, "filler", 6) == 0)
1847 {
1848 p += 6;
1849 diff_flags_new |= DIFF_FILLER;
1850 }
1851 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
1852 {
1853 p += 8;
1854 diff_context_new = getdigits(&p);
1855 }
1856 else if (STRNCMP(p, "icase", 5) == 0)
1857 {
1858 p += 5;
1859 diff_flags_new |= DIFF_ICASE;
1860 }
1861 else if (STRNCMP(p, "iwhite", 6) == 0)
1862 {
1863 p += 6;
1864 diff_flags_new |= DIFF_IWHITE;
1865 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001866 else if (STRNCMP(p, "horizontal", 10) == 0)
1867 {
1868 p += 10;
1869 diff_flags_new |= DIFF_HORIZONTAL;
1870 }
1871 else if (STRNCMP(p, "vertical", 8) == 0)
1872 {
1873 p += 8;
1874 diff_flags_new |= DIFF_VERTICAL;
1875 }
1876 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
1877 {
1878 p += 11;
1879 diff_foldcolumn_new = getdigits(&p);
1880 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001881 if (*p != ',' && *p != NUL)
1882 return FAIL;
1883 if (*p == ',')
1884 ++p;
1885 }
1886
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001887 /* Can't have both "horizontal" and "vertical". */
1888 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
1889 return FAIL;
1890
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891 /* If "icase" or "iwhite" was added or removed, need to update the diff. */
1892 if (diff_flags != diff_flags_new)
Bram Moolenaar29323592016-07-24 22:04:11 +02001893 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001894 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001895
1896 diff_flags = diff_flags_new;
1897 diff_context = diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001898 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899
1900 diff_redraw(TRUE);
1901
1902 /* recompute the scroll binding with the new option value, may
1903 * remove or add filler lines */
1904 check_scrollbind((linenr_T)0, 0L);
1905
1906 return OK;
1907}
1908
1909/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001910 * Return TRUE if 'diffopt' contains "horizontal".
1911 */
1912 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001913diffopt_horizontal(void)
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001914{
1915 return (diff_flags & DIFF_HORIZONTAL) != 0;
1916}
1917
1918/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001919 * Find the difference within a changed line.
1920 * Returns TRUE if the line was added, no other buffer has it.
1921 */
1922 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001923diff_find_change(
1924 win_T *wp,
1925 linenr_T lnum,
1926 int *startp, /* first char of the change */
1927 int *endp) /* last char of the change */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001928{
1929 char_u *line_org;
1930 char_u *line_new;
1931 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001932 int si_org, si_new;
1933 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001934 diff_T *dp;
1935 int idx;
1936 int off;
1937 int added = TRUE;
1938
1939 /* Make a copy of the line, the next ml_get() will invalidate it. */
1940 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
1941 if (line_org == NULL)
1942 return FALSE;
1943
1944 idx = diff_buf_idx(wp->w_buffer);
1945 if (idx == DB_COUNT) /* cannot happen */
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001946 {
1947 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001949 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001950
1951 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001952 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001953 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1954 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001955 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001956 {
1957 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001959 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960
1961 off = lnum - dp->df_lnum[idx];
1962
1963 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001964 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001965 {
1966 /* Skip lines that are not in the other change (filler lines). */
1967 if (off >= dp->df_count[i])
1968 continue;
1969 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001970 line_new = ml_get_buf(curtab->tp_diffbuf[i],
1971 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001972
1973 /* Search for start of difference */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001974 si_org = si_new = 0;
1975 while (line_org[si_org] != NUL)
1976 {
1977 if ((diff_flags & DIFF_IWHITE)
1978 && vim_iswhite(line_org[si_org])
1979 && vim_iswhite(line_new[si_new]))
1980 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001981 si_org = (int)(skipwhite(line_org + si_org) - line_org);
1982 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001983 }
1984 else
1985 {
1986 if (line_org[si_org] != line_new[si_new])
1987 break;
1988 ++si_org;
1989 ++si_new;
1990 }
1991 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992#ifdef FEAT_MBYTE
1993 if (has_mbyte)
1994 {
1995 /* Move back to first byte of character in both lines (may
1996 * have "nn^" in line_org and "n^ in line_new). */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001997 si_org -= (*mb_head_off)(line_org, line_org + si_org);
1998 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001999 }
2000#endif
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002001 if (*startp > si_org)
2002 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003
2004 /* Search for end of difference, if any. */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002005 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002006 {
2007 ei_org = (int)STRLEN(line_org);
2008 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002009 while (ei_org >= *startp && ei_new >= si_new
2010 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011 {
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002012 if ((diff_flags & DIFF_IWHITE)
2013 && vim_iswhite(line_org[ei_org])
2014 && vim_iswhite(line_new[ei_new]))
2015 {
2016 while (ei_org >= *startp
2017 && vim_iswhite(line_org[ei_org]))
2018 --ei_org;
2019 while (ei_new >= si_new
2020 && vim_iswhite(line_new[ei_new]))
2021 --ei_new;
2022 }
2023 else
2024 {
2025 if (line_org[ei_org] != line_new[ei_new])
2026 break;
2027 --ei_org;
2028 --ei_new;
2029 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002030 }
2031 if (*endp < ei_org)
2032 *endp = ei_org;
2033 }
2034 }
2035
2036 vim_free(line_org);
2037 return added;
2038}
2039
2040#if defined(FEAT_FOLDING) || defined(PROTO)
2041/*
2042 * Return TRUE if line "lnum" is not close to a diff block, this line should
2043 * be in a fold.
2044 * Return FALSE if there are no diff blocks at all in this window.
2045 */
2046 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002047diff_infold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002048{
2049 int i;
2050 int idx = -1;
2051 int other = FALSE;
2052 diff_T *dp;
2053
2054 /* Return if 'diff' isn't set. */
2055 if (!wp->w_p_diff)
2056 return FALSE;
2057
2058 for (i = 0; i < DB_COUNT; ++i)
2059 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002060 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002061 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002062 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002063 other = TRUE;
2064 }
2065
2066 /* return here if there are no diffs in the window */
2067 if (idx == -1 || !other)
2068 return FALSE;
2069
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002070 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071 ex_diffupdate(NULL); /* update after a big change */
2072
2073 /* Return if there are no diff blocks. All lines will be folded. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002074 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002075 return TRUE;
2076
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002077 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002078 {
2079 /* If this change is below the line there can't be any further match. */
2080 if (dp->df_lnum[idx] - diff_context > lnum)
2081 break;
2082 /* If this change ends before the line we have a match. */
2083 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2084 return FALSE;
2085 }
2086 return TRUE;
2087}
2088#endif
2089
2090/*
2091 * "dp" and "do" commands.
2092 */
2093 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002094nv_diffgetput(int put, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002095{
2096 exarg_T ea;
Bram Moolenaar6a643652014-10-31 13:54:25 +01002097 char_u buf[30];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002098
Bram Moolenaar6a643652014-10-31 13:54:25 +01002099 if (count == 0)
2100 ea.arg = (char_u *)"";
2101 else
2102 {
2103 vim_snprintf((char *)buf, 30, "%ld", count);
2104 ea.arg = buf;
2105 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002106 if (put)
2107 ea.cmdidx = CMD_diffput;
2108 else
2109 ea.cmdidx = CMD_diffget;
2110 ea.addr_count = 0;
2111 ea.line1 = curwin->w_cursor.lnum;
2112 ea.line2 = curwin->w_cursor.lnum;
2113 ex_diffgetput(&ea);
2114}
2115
2116/*
2117 * ":diffget"
2118 * ":diffput"
2119 */
2120 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002121ex_diffgetput(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002122{
2123 linenr_T lnum;
2124 int count;
2125 linenr_T off = 0;
2126 diff_T *dp;
2127 diff_T *dprev;
2128 diff_T *dfree;
2129 int idx_cur;
2130 int idx_other;
2131 int idx_from;
2132 int idx_to;
2133 int i;
2134 int added;
2135 char_u *p;
2136 aco_save_T aco;
2137 buf_T *buf;
2138 int start_skip, end_skip;
2139 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002140 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002141 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002142
2143 /* Find the current buffer in the list of diff buffers. */
2144 idx_cur = diff_buf_idx(curbuf);
2145 if (idx_cur == DB_COUNT)
2146 {
2147 EMSG(_("E99: Current buffer is not in diff mode"));
2148 return;
2149 }
2150
2151 if (*eap->arg == NUL)
2152 {
2153 /* No argument: Find the other buffer in the list of diff buffers. */
2154 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002155 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002156 && curtab->tp_diffbuf[idx_other] != NULL)
2157 {
2158 if (eap->cmdidx != CMD_diffput
2159 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2160 break;
2161 found_not_ma = TRUE;
2162 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002163 if (idx_other == DB_COUNT)
2164 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002165 if (found_not_ma)
2166 EMSG(_("E793: No other buffer in diff mode is modifiable"));
2167 else
2168 EMSG(_("E100: No other buffer in diff mode"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002169 return;
2170 }
2171
2172 /* Check that there isn't a third buffer in the list */
2173 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002174 if (curtab->tp_diffbuf[i] != curbuf
2175 && curtab->tp_diffbuf[i] != NULL
2176 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002177 {
2178 EMSG(_("E101: More than two buffers in diff mode, don't know which one to use"));
2179 return;
2180 }
2181 }
2182 else
2183 {
2184 /* Buffer number or pattern given. Ignore trailing white space. */
2185 p = eap->arg + STRLEN(eap->arg);
2186 while (p > eap->arg && vim_iswhite(p[-1]))
2187 --p;
2188 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2189 ;
2190 if (eap->arg + i == p) /* digits only */
2191 i = atol((char *)eap->arg);
2192 else
2193 {
Bram Moolenaar0c279bb2013-03-19 14:25:54 +01002194 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002195 if (i < 0)
2196 return; /* error message already given */
2197 }
2198 buf = buflist_findnr(i);
2199 if (buf == NULL)
2200 {
2201 EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg);
2202 return;
2203 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002204 if (buf == curbuf)
2205 return; /* nothing to do */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002206 idx_other = diff_buf_idx(buf);
2207 if (idx_other == DB_COUNT)
2208 {
2209 EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg);
2210 return;
2211 }
2212 }
2213
2214 diff_busy = TRUE;
2215
2216 /* When no range given include the line above or below the cursor. */
2217 if (eap->addr_count == 0)
2218 {
2219 /* Make it possible that ":diffget" on the last line gets line below
2220 * the cursor line when there is no difference above the cursor. */
2221 if (eap->cmdidx == CMD_diffget
2222 && eap->line1 == curbuf->b_ml.ml_line_count
2223 && diff_check(curwin, eap->line1) == 0
2224 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2225 ++eap->line2;
2226 else if (eap->line1 > 0)
2227 --eap->line1;
2228 }
2229
2230 if (eap->cmdidx == CMD_diffget)
2231 {
2232 idx_from = idx_other;
2233 idx_to = idx_cur;
2234 }
2235 else
2236 {
2237 idx_from = idx_cur;
2238 idx_to = idx_other;
2239 /* Need to make the other buffer the current buffer to be able to make
2240 * changes in it. */
2241 /* set curwin/curbuf to buf and save a few things */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002242 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002243 }
2244
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002245 /* May give the warning for a changed buffer here, which can trigger the
2246 * FileChangedRO autocommand, which may do nasty things and mess
2247 * everything up. */
2248 if (!curbuf->b_changed)
2249 {
2250 change_warning(0);
2251 if (diff_buf_idx(curbuf) != idx_to)
2252 {
2253 EMSG(_("E787: Buffer changed unexpectedly"));
2254 return;
2255 }
2256 }
2257
Bram Moolenaar071d4272004-06-13 20:20:40 +00002258 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002259 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002260 {
2261 if (dp->df_lnum[idx_cur] > eap->line2 + off)
2262 break; /* past the range that was specified */
2263
2264 dfree = NULL;
2265 lnum = dp->df_lnum[idx_to];
2266 count = dp->df_count[idx_to];
2267 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2268 && u_save(lnum - 1, lnum + count) != FAIL)
2269 {
2270 /* Inside the specified range and saving for undo worked. */
2271 start_skip = 0;
2272 end_skip = 0;
2273 if (eap->addr_count > 0)
2274 {
2275 /* A range was specified: check if lines need to be skipped. */
2276 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2277 if (start_skip > 0)
2278 {
2279 /* range starts below start of current diff block */
2280 if (start_skip > count)
2281 {
2282 lnum += count;
2283 count = 0;
2284 }
2285 else
2286 {
2287 count -= start_skip;
2288 lnum += start_skip;
2289 }
2290 }
2291 else
2292 start_skip = 0;
2293
2294 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2295 - (eap->line2 + off);
2296 if (end_skip > 0)
2297 {
2298 /* range ends above end of current/from diff block */
2299 if (idx_cur == idx_from) /* :diffput */
2300 {
2301 i = dp->df_count[idx_cur] - start_skip - end_skip;
2302 if (count > i)
2303 count = i;
2304 }
2305 else /* :diffget */
2306 {
2307 count -= end_skip;
2308 end_skip = dp->df_count[idx_from] - start_skip - count;
2309 if (end_skip < 0)
2310 end_skip = 0;
2311 }
2312 }
2313 else
2314 end_skip = 0;
2315 }
2316
Bram Moolenaare962c672014-10-15 12:56:49 +02002317 buf_empty = bufempty();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002318 added = 0;
2319 for (i = 0; i < count; ++i)
2320 {
Bram Moolenaar280f1262006-01-30 00:14:18 +00002321 /* remember deleting the last line of the buffer */
2322 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002323 ml_delete(lnum, FALSE);
2324 --added;
2325 }
2326 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2327 {
2328 linenr_T nr;
2329
2330 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002331 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002332 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002333 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2334 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002335 if (p != NULL)
2336 {
2337 ml_append(lnum + i - 1, p, 0, FALSE);
2338 vim_free(p);
2339 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002340 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2341 {
2342 /* Added the first line into an empty buffer, need to
2343 * delete the dummy empty line. */
2344 buf_empty = FALSE;
2345 ml_delete((linenr_T)2, FALSE);
2346 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002347 }
2348 }
2349 new_count = dp->df_count[idx_to] + added;
2350 dp->df_count[idx_to] = new_count;
2351
2352 if (start_skip == 0 && end_skip == 0)
2353 {
2354 /* Check if there are any other buffers and if the diff is
2355 * equal in them. */
2356 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002357 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2358 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002359 && !diff_equal_entry(dp, idx_from, i))
2360 break;
2361 if (i == DB_COUNT)
2362 {
2363 /* delete the diff entry, the buffers are now equal here */
2364 dfree = dp;
2365 dp = dp->df_next;
2366 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002367 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368 else
2369 dprev->df_next = dp;
2370 }
2371 }
2372
2373 /* Adjust marks. This will change the following entries! */
2374 if (added != 0)
2375 {
2376 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2377 if (curwin->w_cursor.lnum >= lnum)
2378 {
2379 /* Adjust the cursor position if it's in/after the changed
2380 * lines. */
2381 if (curwin->w_cursor.lnum >= lnum + count)
2382 curwin->w_cursor.lnum += added;
2383 else if (added < 0)
2384 curwin->w_cursor.lnum = lnum;
2385 }
2386 }
2387 changed_lines(lnum, 0, lnum + count, (long)added);
2388
2389 if (dfree != NULL)
2390 {
2391 /* Diff is deleted, update folds in other windows. */
2392#ifdef FEAT_FOLDING
2393 diff_fold_update(dfree, idx_to);
2394#endif
2395 vim_free(dfree);
2396 }
2397 else
2398 /* mark_adjust() may have changed the count in a wrong way */
2399 dp->df_count[idx_to] = new_count;
2400
2401 /* When changing the current buffer, keep track of line numbers */
2402 if (idx_cur == idx_to)
2403 off += added;
2404 }
2405
2406 /* If before the range or not deleted, go to next diff. */
2407 if (dfree == NULL)
2408 {
2409 dprev = dp;
2410 dp = dp->df_next;
2411 }
2412 }
2413
2414 /* restore curwin/curbuf and a few other things */
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002415 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002416 {
2417 /* Syncing undo only works for the current buffer, but we change
2418 * another buffer. Sync undo if the command was typed. This isn't
2419 * 100% right when ":diffput" is used in a function or mapping. */
2420 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002421 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002422 aucmd_restbuf(&aco);
2423 }
2424
2425 diff_busy = FALSE;
2426
2427 /* Check that the cursor is on a valid character and update it's position.
2428 * When there were filler lines the topline has become invalid. */
2429 check_cursor();
2430 changed_line_abv_curs();
2431
2432 /* Also need to redraw the other buffers. */
2433 diff_redraw(FALSE);
2434}
2435
2436#ifdef FEAT_FOLDING
2437/*
2438 * Update folds for all diff buffers for entry "dp".
2439 * Skip buffer with index "skip_idx".
2440 * When there are no diffs, all folds are removed.
2441 */
2442 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002443diff_fold_update(diff_T *dp, int skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002444{
2445 int i;
2446 win_T *wp;
2447
Bram Moolenaar29323592016-07-24 22:04:11 +02002448 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002449 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002450 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002451 foldUpdate(wp, dp->df_lnum[i],
2452 dp->df_lnum[i] + dp->df_count[i]);
2453}
2454#endif
2455
2456/*
2457 * Return TRUE if buffer "buf" is in diff-mode.
2458 */
2459 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002460diff_mode_buf(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002461{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002462 tabpage_T *tp;
2463
Bram Moolenaar29323592016-07-24 22:04:11 +02002464 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002465 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
2466 return TRUE;
2467 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002468}
2469
2470/*
2471 * Move "count" times in direction "dir" to the next diff block.
2472 * Return FAIL if there isn't such a diff block.
2473 */
2474 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002475diff_move_to(int dir, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002476{
2477 int idx;
2478 linenr_T lnum = curwin->w_cursor.lnum;
2479 diff_T *dp;
2480
2481 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002482 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002483 return FAIL;
2484
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002485 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002486 ex_diffupdate(NULL); /* update after a big change */
2487
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002488 if (curtab->tp_first_diff == NULL) /* no diffs today */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489 return FAIL;
2490
2491 while (--count >= 0)
2492 {
2493 /* Check if already before first diff. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002494 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002495 break;
2496
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002497 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002498 {
2499 if (dp == NULL)
2500 break;
2501 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
2502 || (dir == BACKWARD
2503 && (dp->df_next == NULL
2504 || lnum <= dp->df_next->df_lnum[idx])))
2505 {
2506 lnum = dp->df_lnum[idx];
2507 break;
2508 }
2509 }
2510 }
2511
2512 /* don't end up past the end of the file */
2513 if (lnum > curbuf->b_ml.ml_line_count)
2514 lnum = curbuf->b_ml.ml_line_count;
2515
2516 /* When the cursor didn't move at all we fail. */
2517 if (lnum == curwin->w_cursor.lnum)
2518 return FAIL;
2519
2520 setpcmark();
2521 curwin->w_cursor.lnum = lnum;
2522 curwin->w_cursor.col = 0;
2523
2524 return OK;
2525}
2526
Bram Moolenaar860cae12010-06-05 23:22:07 +02002527 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01002528diff_get_corresponding_line(
2529 buf_T *buf1,
2530 linenr_T lnum1,
2531 buf_T *buf2,
2532 linenr_T lnum3)
Bram Moolenaar860cae12010-06-05 23:22:07 +02002533{
2534 int idx1;
2535 int idx2;
2536 diff_T *dp;
2537 int baseline = 0;
2538 linenr_T lnum2;
2539
2540 idx1 = diff_buf_idx(buf1);
2541 idx2 = diff_buf_idx(buf2);
2542 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
2543 return lnum1;
2544
2545 if (curtab->tp_diff_invalid)
2546 ex_diffupdate(NULL); /* update after a big change */
2547
2548 if (curtab->tp_first_diff == NULL) /* no diffs today */
2549 return lnum1;
2550
2551 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
2552 {
2553 if (dp->df_lnum[idx1] > lnum1)
2554 {
2555 lnum2 = lnum1 - baseline;
2556 /* don't end up past the end of the file */
2557 if (lnum2 > buf2->b_ml.ml_line_count)
2558 lnum2 = buf2->b_ml.ml_line_count;
2559
2560 return lnum2;
2561 }
2562 else if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
2563 {
2564 /* Inside the diffblock */
2565 baseline = lnum1 - dp->df_lnum[idx1];
2566 if (baseline > dp->df_count[idx2])
2567 baseline = dp->df_count[idx2];
2568
2569 return dp->df_lnum[idx2] + baseline;
2570 }
2571 else if ( (dp->df_lnum[idx1] == lnum1)
2572 && (dp->df_count[idx1] == 0)
2573 && (dp->df_lnum[idx2] <= lnum3)
2574 && ((dp->df_lnum[idx2] + dp->df_count[idx2]) > lnum3))
2575 /*
2576 * Special case: if the cursor is just after a zero-count
2577 * block (i.e. all filler) and the target cursor is already
2578 * inside the corresponding block, leave the target cursor
2579 * unmoved. This makes repeated CTRL-W W operations work
2580 * as expected.
2581 */
2582 return lnum3;
2583 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
2584 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
2585 }
2586
2587 /* If we get here then the cursor is after the last diff */
2588 lnum2 = lnum1 - baseline;
2589 /* don't end up past the end of the file */
2590 if (lnum2 > buf2->b_ml.ml_line_count)
2591 lnum2 = buf2->b_ml.ml_line_count;
2592
2593 return lnum2;
2594}
Bram Moolenaar860cae12010-06-05 23:22:07 +02002595
Bram Moolenaar071d4272004-06-13 20:20:40 +00002596#if defined(FEAT_FOLDING) || defined(PROTO)
2597/*
2598 * For line "lnum" in the current window find the equivalent lnum in window
2599 * "wp", compensating for inserted/deleted lines.
2600 */
2601 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01002602diff_lnum_win(linenr_T lnum, win_T *wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002603{
2604 diff_T *dp;
2605 int idx;
2606 int i;
2607 linenr_T n;
2608
2609 idx = diff_buf_idx(curbuf);
2610 if (idx == DB_COUNT) /* safety check */
2611 return (linenr_T)0;
2612
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002613 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 ex_diffupdate(NULL); /* update after a big change */
2615
2616 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002617 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002618 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2619 break;
2620
2621 /* When after the last change, compute relative to the last line number. */
2622 if (dp == NULL)
2623 return wp->w_buffer->b_ml.ml_line_count
2624 - (curbuf->b_ml.ml_line_count - lnum);
2625
2626 /* Find index for "wp". */
2627 i = diff_buf_idx(wp->w_buffer);
2628 if (i == DB_COUNT) /* safety check */
2629 return (linenr_T)0;
2630
2631 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
2632 if (n > dp->df_lnum[i] + dp->df_count[i])
2633 n = dp->df_lnum[i] + dp->df_count[i];
2634 return n;
2635}
2636#endif
2637
2638#endif /* FEAT_DIFF */