blob: d9aaec828726d7899006b862b4f4e3bf898b3d52 [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 */
32#if defined(MSWIN) || defined(MSDOS)
33static 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
38static int diff_buf_idx __ARGS((buf_T *buf));
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000039static int diff_buf_idx_tp __ARGS((buf_T *buf, tabpage_T *tp));
40static void diff_mark_adjust_tp __ARGS((tabpage_T *tp, int idx, linenr_T line1, linenr_T line2, long amount, long amount_after));
41static void diff_check_unchanged __ARGS((tabpage_T *tp, diff_T *dp));
42static int diff_check_sanity __ARGS((tabpage_T *tp, diff_T *dp));
Bram Moolenaar071d4272004-06-13 20:20:40 +000043static void diff_redraw __ARGS((int dofold));
44static int diff_write __ARGS((buf_T *buf, char_u *fname));
45static void diff_file __ARGS((char_u *tmp_orig, char_u *tmp_new, char_u *tmp_diff));
Bram Moolenaar071d4272004-06-13 20:20:40 +000046static int diff_equal_entry __ARGS((diff_T *dp, int idx1, int idx2));
47static int diff_cmp __ARGS((char_u *s1, char_u *s2));
48#ifdef FEAT_FOLDING
49static void diff_fold_update __ARGS((diff_T *dp, int skip_idx));
50#endif
51static void diff_read __ARGS((int idx_orig, int idx_new, char_u *fname));
52static void diff_copy_entry __ARGS((diff_T *dprev, diff_T *dp, int idx_orig, int idx_new));
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000053static diff_T *diff_alloc_new __ARGS((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
63diff_buf_delete(buf)
64 buf_T *buf;
65{
66 int i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000067 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +000068
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000069 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +000070 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000071 i = diff_buf_idx_tp(buf, tp);
72 if (i != DB_COUNT)
73 {
74 tp->tp_diffbuf[i] = NULL;
75 tp->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +000076 if (tp == curtab)
77 diff_redraw(TRUE);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000078 }
Bram Moolenaar071d4272004-06-13 20:20:40 +000079 }
80}
81
82/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +000083 * Check if the current buffer should be added to or removed from the list of
84 * diff buffers.
85 */
86 void
87diff_buf_adjust(win)
88 win_T *win;
89{
90 win_T *wp;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +000091 int i;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +000092
93 if (!win->w_p_diff)
94 {
95 /* When there is no window showing a diff for this buffer, remove
96 * it from the diffs. */
97 for (wp = firstwin; wp != NULL; wp = wp->w_next)
98 if (wp->w_buffer == win->w_buffer && wp->w_p_diff)
99 break;
100 if (wp == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000101 {
102 i = diff_buf_idx(win->w_buffer);
103 if (i != DB_COUNT)
104 {
105 curtab->tp_diffbuf[i] = NULL;
106 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000107 diff_redraw(TRUE);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000108 }
109 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000110 }
111 else
112 diff_buf_add(win->w_buffer);
113}
114
115/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000116 * Add a buffer to make diffs for.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000117 * Call this when a new buffer is being edited in the current window where
118 * 'diff' is set.
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000119 * Marks the current buffer as being part of the diff and requiring updating.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000120 * This must be done before any autocmd, because a command may use info
121 * about the screen contents.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000122 */
123 void
124diff_buf_add(buf)
125 buf_T *buf;
126{
127 int i;
128
129 if (diff_buf_idx(buf) != DB_COUNT)
130 return; /* It's already there. */
131
132 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000133 if (curtab->tp_diffbuf[i] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000134 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000135 curtab->tp_diffbuf[i] = buf;
136 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000137 diff_redraw(TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000138 return;
139 }
140
141 EMSGN(_("E96: Can not diff more than %ld buffers"), DB_COUNT);
142}
143
144/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000145 * Find buffer "buf" in the list of diff buffers for the current tab page.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000146 * Return its index or DB_COUNT if not found.
147 */
148 static int
149diff_buf_idx(buf)
150 buf_T *buf;
151{
152 int idx;
153
154 for (idx = 0; idx < DB_COUNT; ++idx)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000155 if (curtab->tp_diffbuf[idx] == buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000156 break;
157 return idx;
158}
159
160/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000161 * Find buffer "buf" in the list of diff buffers for tab page "tp".
162 * Return its index or DB_COUNT if not found.
163 */
164 static int
165diff_buf_idx_tp(buf, tp)
166 buf_T *buf;
167 tabpage_T *tp;
168{
169 int idx;
170
171 for (idx = 0; idx < DB_COUNT; ++idx)
172 if (tp->tp_diffbuf[idx] == buf)
173 break;
174 return idx;
175}
176
177/*
178 * Mark the diff info involving buffer "buf" as invalid, it will be updated
179 * when info is requested.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000180 */
181 void
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000182diff_invalidate(buf)
183 buf_T *buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000184{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000185 tabpage_T *tp;
186 int i;
187
188 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000189 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000190 i = diff_buf_idx_tp(buf, tp);
191 if (i != DB_COUNT)
192 {
193 tp->tp_diff_invalid = TRUE;
194 if (tp == curtab)
195 diff_redraw(TRUE);
196 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000197 }
198}
199
200/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000201 * Called by mark_adjust(): update line numbers in "curbuf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000202 */
203 void
204diff_mark_adjust(line1, line2, amount, amount_after)
205 linenr_T line1;
206 linenr_T line2;
207 long amount;
208 long amount_after;
209{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000210 int idx;
211 tabpage_T *tp;
212
213 /* Handle all tab pages that use the current buffer in a diff. */
214 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
215 {
216 idx = diff_buf_idx_tp(curbuf, tp);
217 if (idx != DB_COUNT)
218 diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after);
219 }
220}
221
222/*
223 * Update line numbers in tab page "tp" for "curbuf" with index "idx".
224 * This attempts to update the changes as much as possible:
225 * When inserting/deleting lines outside of existing change blocks, create a
226 * new change block and update the line numbers in following blocks.
227 * When inserting/deleting lines in existing change blocks, update them.
228 */
229 static void
230diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after)
231 tabpage_T *tp;
232 int idx;
233 linenr_T line1;
234 linenr_T line2;
235 long amount;
236 long amount_after;
237{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000238 diff_T *dp;
239 diff_T *dprev;
240 diff_T *dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000241 int i;
242 int inserted, deleted;
243 int n, off;
244 linenr_T last;
245 linenr_T lnum_deleted = line1; /* lnum of remaining deletion */
246 int check_unchanged;
247
Bram Moolenaar071d4272004-06-13 20:20:40 +0000248 if (line2 == MAXLNUM)
249 {
250 /* mark_adjust(99, MAXLNUM, 9, 0): insert lines */
251 inserted = amount;
252 deleted = 0;
253 }
254 else if (amount_after > 0)
255 {
256 /* mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines*/
257 inserted = amount_after;
258 deleted = 0;
259 }
260 else
261 {
262 /* mark_adjust(98, 99, MAXLNUM, -2): delete lines */
263 inserted = 0;
264 deleted = -amount_after;
265 }
266
267 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000268 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000269 for (;;)
270 {
271 /* If the change is after the previous diff block and before the next
272 * diff block, thus not touching an existing change, create a new diff
273 * block. Don't do this when ex_diffgetput() is busy. */
274 if ((dp == NULL || dp->df_lnum[idx] - 1 > line2
275 || (line2 == MAXLNUM && dp->df_lnum[idx] > line1))
276 && (dprev == NULL
277 || dprev->df_lnum[idx] + dprev->df_count[idx] < line1)
278 && !diff_busy)
279 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000280 dnext = diff_alloc_new(tp, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000281 if (dnext == NULL)
282 return;
283
284 dnext->df_lnum[idx] = line1;
285 dnext->df_count[idx] = inserted;
286 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000287 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000288 {
289 if (dprev == NULL)
290 dnext->df_lnum[i] = line1;
291 else
292 dnext->df_lnum[i] = line1
293 + (dprev->df_lnum[i] + dprev->df_count[i])
294 - (dprev->df_lnum[idx] + dprev->df_count[idx]);
295 dnext->df_count[i] = deleted;
296 }
297 }
298
299 /* if at end of the list, quit */
300 if (dp == NULL)
301 break;
302
303 /*
304 * Check for these situations:
305 * 1 2 3
306 * 1 2 3
307 * line1 2 3 4 5
308 * 2 3 4 5
309 * 2 3 4 5
310 * line2 2 3 4 5
311 * 3 5 6
312 * 3 5 6
313 */
314 /* compute last line of this change */
315 last = dp->df_lnum[idx] + dp->df_count[idx] - 1;
316
317 /* 1. change completely above line1: nothing to do */
318 if (last >= line1 - 1)
319 {
320 /* 6. change below line2: only adjust for amount_after; also when
321 * "deleted" became zero when deleted all lines between two diffs */
322 if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2)
323 {
324 if (amount_after == 0)
325 break; /* nothing left to change */
326 dp->df_lnum[idx] += amount_after;
327 }
328 else
329 {
330 check_unchanged = FALSE;
331
332 /* 2. 3. 4. 5.: inserted/deleted lines touching this diff. */
333 if (deleted > 0)
334 {
335 if (dp->df_lnum[idx] >= line1)
336 {
337 off = dp->df_lnum[idx] - lnum_deleted;
338 if (last <= line2)
339 {
340 /* 4. delete all lines of diff */
341 if (dp->df_next != NULL
342 && dp->df_next->df_lnum[idx] - 1 <= line2)
343 {
344 /* delete continues in next diff, only do
345 * lines until that one */
346 n = dp->df_next->df_lnum[idx] - lnum_deleted;
347 deleted -= n;
348 n -= dp->df_count[idx];
349 lnum_deleted = dp->df_next->df_lnum[idx];
350 }
351 else
352 n = deleted - dp->df_count[idx];
353 dp->df_count[idx] = 0;
354 }
355 else
356 {
357 /* 5. delete lines at or just before top of diff */
358 n = off;
359 dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1;
360 check_unchanged = TRUE;
361 }
362 dp->df_lnum[idx] = line1;
363 }
364 else
365 {
366 off = 0;
367 if (last < line2)
368 {
369 /* 2. delete at end of of diff */
370 dp->df_count[idx] -= last - lnum_deleted + 1;
371 if (dp->df_next != NULL
372 && dp->df_next->df_lnum[idx] - 1 <= line2)
373 {
374 /* delete continues in next diff, only do
375 * lines until that one */
376 n = dp->df_next->df_lnum[idx] - 1 - last;
377 deleted -= dp->df_next->df_lnum[idx]
378 - lnum_deleted;
379 lnum_deleted = dp->df_next->df_lnum[idx];
380 }
381 else
382 n = line2 - last;
383 check_unchanged = TRUE;
384 }
385 else
386 {
387 /* 3. delete lines inside the diff */
388 n = 0;
389 dp->df_count[idx] -= deleted;
390 }
391 }
392
393 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000394 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000395 {
396 dp->df_lnum[i] -= off;
397 dp->df_count[i] += n;
398 }
399 }
400 else
401 {
402 if (dp->df_lnum[idx] <= line1)
403 {
404 /* inserted lines somewhere in this diff */
405 dp->df_count[idx] += inserted;
406 check_unchanged = TRUE;
407 }
408 else
409 /* inserted lines somewhere above this diff */
410 dp->df_lnum[idx] += inserted;
411 }
412
413 if (check_unchanged)
414 /* Check if inserted lines are equal, may reduce the
415 * size of the diff. TODO: also check for equal lines
416 * in the middle and perhaps split the block. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000417 diff_check_unchanged(tp, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418 }
419 }
420
421 /* check if this block touches the previous one, may merge them. */
422 if (dprev != NULL && dprev->df_lnum[idx] + dprev->df_count[idx]
423 == dp->df_lnum[idx])
424 {
425 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000426 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000427 dprev->df_count[i] += dp->df_count[i];
428 dprev->df_next = dp->df_next;
429 vim_free(dp);
430 dp = dprev->df_next;
431 }
432 else
433 {
434 /* Advance to next entry. */
435 dprev = dp;
436 dp = dp->df_next;
437 }
438 }
439
440 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000441 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000442 while (dp != NULL)
443 {
444 /* All counts are zero, remove this entry. */
445 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000446 if (tp->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000447 break;
448 if (i == DB_COUNT)
449 {
450 dnext = dp->df_next;
451 vim_free(dp);
452 dp = dnext;
453 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000454 tp->tp_first_diff = dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000455 else
456 dprev->df_next = dnext;
457 }
458 else
459 {
460 /* Advance to next entry. */
461 dprev = dp;
462 dp = dp->df_next;
463 }
464
465 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000466
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000467 if (tp == curtab)
468 {
469 diff_redraw(TRUE);
470
471 /* Need to recompute the scroll binding, may remove or add filler
472 * lines (e.g., when adding lines above w_topline). But it's slow when
473 * making many changes, postpone until redrawing. */
474 diff_need_scrollbind = TRUE;
475 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000476}
477
478/*
479 * Allocate a new diff block and link it between "dprev" and "dp".
480 */
481 static diff_T *
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000482diff_alloc_new(tp, dprev, dp)
483 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000484 diff_T *dprev;
485 diff_T *dp;
486{
487 diff_T *dnew;
488
489 dnew = (diff_T *)alloc((unsigned)sizeof(diff_T));
490 if (dnew != NULL)
491 {
492 dnew->df_next = dp;
493 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000494 tp->tp_first_diff = dnew;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000495 else
496 dprev->df_next = dnew;
497 }
498 return dnew;
499}
500
501/*
502 * Check if the diff block "dp" can be made smaller for lines at the start and
503 * end that are equal. Called after inserting lines.
504 * This may result in a change where all buffers have zero lines, the caller
505 * must take care of removing it.
506 */
507 static void
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000508diff_check_unchanged(tp, dp)
509 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000510 diff_T *dp;
511{
512 int i_org;
513 int i_new;
514 int off_org, off_new;
515 char_u *line_org;
516 int dir = FORWARD;
517
518 /* Find the first buffers, use it as the original, compare the other
519 * buffer lines against this one. */
520 for (i_org = 0; i_org < DB_COUNT; ++i_org)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000521 if (tp->tp_diffbuf[i_org] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000522 break;
523 if (i_org == DB_COUNT) /* safety check */
524 return;
525
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000526 if (diff_check_sanity(tp, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000527 return;
528
529 /* First check lines at the top, then at the bottom. */
530 off_org = 0;
531 off_new = 0;
532 for (;;)
533 {
534 /* Repeat until a line is found which is different or the number of
535 * lines has become zero. */
536 while (dp->df_count[i_org] > 0)
537 {
538 /* Copy the line, the next ml_get() will invalidate it. */
539 if (dir == BACKWARD)
540 off_org = dp->df_count[i_org] - 1;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000541 line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000542 dp->df_lnum[i_org] + off_org, FALSE));
543 if (line_org == NULL)
544 return;
545 for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new)
546 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000547 if (tp->tp_diffbuf[i_new] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000548 continue;
549 if (dir == BACKWARD)
550 off_new = dp->df_count[i_new] - 1;
551 /* if other buffer doesn't have this line, it was inserted */
552 if (off_new < 0 || off_new >= dp->df_count[i_new])
553 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000554 if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 dp->df_lnum[i_new] + off_new, FALSE)) != 0)
556 break;
557 }
558 vim_free(line_org);
559
560 /* Stop when a line isn't equal in all diff buffers. */
561 if (i_new != DB_COUNT)
562 break;
563
564 /* Line matched in all buffers, remove it from the diff. */
565 for (i_new = i_org; i_new < DB_COUNT; ++i_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000566 if (tp->tp_diffbuf[i_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000567 {
568 if (dir == FORWARD)
569 ++dp->df_lnum[i_new];
570 --dp->df_count[i_new];
571 }
572 }
573 if (dir == BACKWARD)
574 break;
575 dir = BACKWARD;
576 }
577}
578
579/*
580 * Check if a diff block doesn't contain invalid line numbers.
581 * This can happen when the diff program returns invalid results.
582 */
583 static int
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000584diff_check_sanity(tp, dp)
585 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000586 diff_T *dp;
587{
588 int i;
589
590 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000591 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000592 if (dp->df_lnum[i] + dp->df_count[i] - 1
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000593 > tp->tp_diffbuf[i]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000594 return FAIL;
595 return OK;
596}
597
598/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000599 * Mark all diff buffers in the current tab page for redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000600 */
601 static void
602diff_redraw(dofold)
603 int dofold; /* also recompute the folds */
604{
605 win_T *wp;
606 int n;
607
608 for (wp = firstwin; wp != NULL; wp = wp->w_next)
609 if (wp->w_p_diff)
610 {
Bram Moolenaar60f83772006-03-12 21:52:47 +0000611 redraw_win_later(wp, SOME_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000612#ifdef FEAT_FOLDING
613 if (dofold && foldmethodIsDiff(wp))
614 foldUpdateAll(wp);
615#endif
616 /* A change may have made filler lines invalid, need to take care
617 * of that for other windows. */
618 if (wp != curwin && wp->w_topfill > 0)
619 {
620 n = diff_check(wp, wp->w_topline);
621 if (wp->w_topfill > n)
622 wp->w_topfill = (n < 0 ? 0 : n);
623 }
624 }
625}
626
627/*
628 * Write buffer "buf" to file "name".
629 * Always use 'fileformat' set to "unix".
630 * Return FAIL for failure
631 */
632 static int
633diff_write(buf, fname)
634 buf_T *buf;
635 char_u *fname;
636{
637 int r;
638 char_u *save_ff;
639
640 save_ff = buf->b_p_ff;
641 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX);
642 r = buf_write(buf, fname, NULL, (linenr_T)1, buf->b_ml.ml_line_count,
643 NULL, FALSE, FALSE, FALSE, TRUE);
644 free_string_option(buf->b_p_ff);
645 buf->b_p_ff = save_ff;
646 return r;
647}
648
649/*
650 * Completely update the diffs for the buffers involved.
651 * This uses the ordinary "diff" command.
652 * The buffers are written to a file, also for unmodified buffers (the file
653 * could have been produced by autocommands, e.g. the netrw plugin).
654 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000655 void
656ex_diffupdate(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +0000657 exarg_T *eap UNUSED; /* can be NULL */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000658{
659 buf_T *buf;
660 int idx_orig;
661 int idx_new;
662 char_u *tmp_orig;
663 char_u *tmp_new;
664 char_u *tmp_diff;
665 FILE *fd;
666 int ok;
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000667 int io_error = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668
669 /* Delete all diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000670 diff_clear(curtab);
671 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672
673 /* Use the first buffer as the original text. */
674 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000675 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000676 break;
677 if (idx_orig == DB_COUNT)
678 return;
679
680 /* Only need to do something when there is another buffer. */
681 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000682 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000683 break;
684 if (idx_new == DB_COUNT)
685 return;
686
687 /* We need three temp file names. */
688 tmp_orig = vim_tempname('o');
689 tmp_new = vim_tempname('n');
690 tmp_diff = vim_tempname('d');
691 if (tmp_orig == NULL || tmp_new == NULL || tmp_diff == NULL)
692 goto theend;
693
694 /*
695 * Do a quick test if "diff" really works. Otherwise it looks like there
696 * are no differences. Can't use the return value, it's non-zero when
697 * there are differences.
698 * May try twice, first with "-a" and then without.
699 */
700 for (;;)
701 {
702 ok = FALSE;
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000703 fd = mch_fopen((char *)tmp_orig, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000704 if (fd == NULL)
705 io_error = TRUE;
706 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000707 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000708 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
709 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 fclose(fd);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000711 fd = mch_fopen((char *)tmp_new, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000712 if (fd == NULL)
713 io_error = TRUE;
714 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000715 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000716 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
717 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 fclose(fd);
719 diff_file(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000720 fd = mch_fopen((char *)tmp_diff, "r");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000721 if (fd == NULL)
722 io_error = TRUE;
723 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000724 {
725 char_u linebuf[LBUFLEN];
726
727 for (;;)
728 {
729 /* There must be a line that contains "1c1". */
730 if (tag_fgets(linebuf, LBUFLEN, fd))
731 break;
732 if (STRNCMP(linebuf, "1c1", 3) == 0)
733 ok = TRUE;
734 }
735 fclose(fd);
736 }
737 mch_remove(tmp_diff);
738 mch_remove(tmp_new);
739 }
740 mch_remove(tmp_orig);
741 }
742
743#ifdef FEAT_EVAL
744 /* When using 'diffexpr' break here. */
745 if (*p_dex != NUL)
746 break;
747#endif
748
749#if defined(MSWIN) || defined(MSDOS)
750 /* If the "-a" argument works, also check if "--binary" works. */
751 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
752 {
753 diff_a_works = TRUE;
754 diff_bin_works = TRUE;
755 continue;
756 }
757 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
758 {
759 /* Tried --binary, but it failed. "-a" works though. */
760 diff_bin_works = FALSE;
761 ok = TRUE;
762 }
763#endif
764
765 /* If we checked if "-a" works already, break here. */
766 if (diff_a_works != MAYBE)
767 break;
768 diff_a_works = ok;
769
770 /* If "-a" works break here, otherwise retry without "-a". */
771 if (ok)
772 break;
773 }
774 if (!ok)
775 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000776 if (io_error)
777 EMSG(_("E810: Cannot read or write temp files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000778 EMSG(_("E97: Cannot create diffs"));
779 diff_a_works = MAYBE;
780#if defined(MSWIN) || defined(MSDOS)
781 diff_bin_works = MAYBE;
782#endif
783 goto theend;
784 }
785
Bram Moolenaarbd1d5602012-05-18 18:47:17 +0200786 /* :diffupdate! */
787 if (eap != NULL && eap->forceit)
788 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new)
789 {
790 buf = curtab->tp_diffbuf[idx_new];
791 if (buf_valid(buf))
792 buf_check_timestamp(buf, FALSE);
793 }
794
Bram Moolenaar071d4272004-06-13 20:20:40 +0000795 /* Write the first buffer to a tempfile. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000796 buf = curtab->tp_diffbuf[idx_orig];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000797 if (diff_write(buf, tmp_orig) == FAIL)
798 goto theend;
799
800 /* Make a difference between the first buffer and every other. */
801 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
802 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000803 buf = curtab->tp_diffbuf[idx_new];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000804 if (buf == NULL)
805 continue;
806 if (diff_write(buf, tmp_new) == FAIL)
807 continue;
808 diff_file(tmp_orig, tmp_new, tmp_diff);
809
810 /* Read the diff output and add each entry to the diff list. */
811 diff_read(idx_orig, idx_new, tmp_diff);
812 mch_remove(tmp_diff);
813 mch_remove(tmp_new);
814 }
815 mch_remove(tmp_orig);
816
Bram Moolenaar60a44dc2007-10-19 16:58:12 +0000817 /* force updating cursor position on screen */
818 curwin->w_valid_cursor.lnum = 0;
819
Bram Moolenaar071d4272004-06-13 20:20:40 +0000820 diff_redraw(TRUE);
821
822theend:
823 vim_free(tmp_orig);
824 vim_free(tmp_new);
825 vim_free(tmp_diff);
826}
827
828/*
829 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
830 */
831 static void
832diff_file(tmp_orig, tmp_new, tmp_diff)
833 char_u *tmp_orig;
834 char_u *tmp_new;
835 char_u *tmp_diff;
836{
837 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000838 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000839
840#ifdef FEAT_EVAL
841 if (*p_dex != NUL)
842 /* Use 'diffexpr' to generate the diff file. */
843 eval_diff(tmp_orig, tmp_new, tmp_diff);
844 else
845#endif
846 {
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000847 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
848 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
849 cmd = alloc((unsigned)len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000850 if (cmd != NULL)
851 {
Bram Moolenaar95fb60a2005-03-08 22:29:20 +0000852 /* We don't want $DIFF_OPTIONS to get in the way. */
853 if (getenv("DIFF_OPTIONS"))
854 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
855
Bram Moolenaar071d4272004-06-13 20:20:40 +0000856 /* Build the diff command and execute it. Always use -a, binary
857 * differences are of no use. Ignore errors, diff returns
858 * non-zero when differences have been found. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000859 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +0000860 diff_a_works == FALSE ? "" : "-a ",
861#if defined(MSWIN) || defined(MSDOS)
862 diff_bin_works == TRUE ? "--binary " : "",
863#else
864 "",
865#endif
866 (diff_flags & DIFF_IWHITE) ? "-b " : "",
867 (diff_flags & DIFF_ICASE) ? "-i " : "",
868 tmp_orig, tmp_new);
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000869 append_redir(cmd, (int)len, p_srr, tmp_diff);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000870#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000871 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000872#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000873 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000874#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000875 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000876#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000877 vim_free(cmd);
878 }
879 }
880}
881
882/*
883 * Create a new version of a file from the current buffer and a diff file.
884 * The buffer is written to a file, also for unmodified buffers (the file
885 * could have been produced by autocommands, e.g. the netrw plugin).
886 */
887 void
888ex_diffpatch(eap)
889 exarg_T *eap;
890{
891 char_u *tmp_orig; /* name of original temp file */
892 char_u *tmp_new; /* name of patched temp file */
893 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000894 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 win_T *old_curwin = curwin;
896 char_u *newname = NULL; /* name of patched file buffer */
897#ifdef UNIX
898 char_u dirbuf[MAXPATHL];
899 char_u *fullname = NULL;
900#endif
901#ifdef FEAT_BROWSE
902 char_u *browseFile = NULL;
903 int browse_flag = cmdmod.browse;
904#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +0000905 struct stat st;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000906
907#ifdef FEAT_BROWSE
908 if (cmdmod.browse)
909 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +0000910 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaar071d4272004-06-13 20:20:40 +0000911 eap->arg, NULL, NULL, BROWSE_FILTER_ALL_FILES, NULL);
912 if (browseFile == NULL)
913 return; /* operation cancelled */
914 eap->arg = browseFile;
915 cmdmod.browse = FALSE; /* don't let do_ecmd() browse again */
916 }
917#endif
918
919 /* We need two temp file names. */
920 tmp_orig = vim_tempname('o');
921 tmp_new = vim_tempname('n');
922 if (tmp_orig == NULL || tmp_new == NULL)
923 goto theend;
924
925 /* Write the current buffer to "tmp_orig". */
926 if (buf_write(curbuf, tmp_orig, NULL,
927 (linenr_T)1, curbuf->b_ml.ml_line_count,
928 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
929 goto theend;
930
931#ifdef UNIX
932 /* Get the absolute path of the patchfile, changing directory below. */
933 fullname = FullName_save(eap->arg, FALSE);
934#endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000935 buflen = STRLEN(tmp_orig) + (
Bram Moolenaar071d4272004-06-13 20:20:40 +0000936# ifdef UNIX
937 fullname != NULL ? STRLEN(fullname) :
938# endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000939 STRLEN(eap->arg)) + STRLEN(tmp_new) + 16;
940 buf = alloc((unsigned)buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000941 if (buf == NULL)
942 goto theend;
943
944#ifdef UNIX
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000945 /* Temporarily chdir to /tmp, to avoid patching files in the current
Bram Moolenaar071d4272004-06-13 20:20:40 +0000946 * directory when the patch file contains more than one patch. When we
947 * have our own temp dir use that instead, it will be cleaned up when we
948 * exit (any .rej files created). Don't change directory if we can't
949 * return to the current. */
950 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
951 dirbuf[0] = NUL;
952 else
953 {
954# ifdef TEMPDIRNAMES
955 if (vim_tempdir != NULL)
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000956 ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000957 else
958# endif
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000959 ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +0000960 shorten_fnames(TRUE);
961 }
962#endif
963
964#ifdef FEAT_EVAL
965 if (*p_pex != NUL)
966 /* Use 'patchexpr' to generate the new file. */
967 eval_patch(tmp_orig,
968# ifdef UNIX
969 fullname != NULL ? fullname :
970# endif
971 eap->arg, tmp_new);
972 else
973#endif
974 {
975 /* Build the patch command and execute it. Ignore errors. Switch to
976 * cooked mode to allow the user to respond to prompts. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000977 vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"",
978 tmp_new, tmp_orig,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000979# ifdef UNIX
980 fullname != NULL ? fullname :
981# endif
982 eap->arg);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000983#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000984 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000985#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000986 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000987#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000988 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000989#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000990 }
991
992#ifdef UNIX
993 if (dirbuf[0] != NUL)
994 {
995 if (mch_chdir((char *)dirbuf) != 0)
996 EMSG(_(e_prev_dir));
997 shorten_fnames(TRUE);
998 }
999#endif
1000
1001 /* patch probably has written over the screen */
1002 redraw_later(CLEAR);
1003
1004 /* Delete any .orig or .rej file created. */
1005 STRCPY(buf, tmp_new);
1006 STRCAT(buf, ".orig");
1007 mch_remove(buf);
1008 STRCPY(buf, tmp_new);
1009 STRCAT(buf, ".rej");
1010 mch_remove(buf);
1011
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001012 /* Only continue if the output file was created. */
1013 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
1014 EMSG(_("E816: Cannot read patch output"));
1015 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001016 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001017 if (curbuf->b_fname != NULL)
1018 {
1019 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001020 (int)(STRLEN(curbuf->b_fname) + 4));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001021 if (newname != NULL)
1022 STRCAT(newname, ".new");
1023 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001024
1025#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001026 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001027#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001028 /* don't use a new tab page, each tab page has its own diffs */
1029 cmdmod.tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001030
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001031 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001032 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001033 /* Pretend it was a ":split fname" command */
1034 eap->cmdidx = CMD_split;
1035 eap->arg = tmp_new;
1036 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001037
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001038 /* check that split worked and editing tmp_new */
1039 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001040 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001041 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1042 diff_win_options(curwin, TRUE);
1043 diff_win_options(old_curwin, TRUE);
1044
1045 if (newname != NULL)
1046 {
1047 /* do a ":file filename.new" on the patched buffer */
1048 eap->arg = newname;
1049 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001050
1051#ifdef FEAT_AUTOCMD
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001052 /* Do filetype detection with the new name. */
1053 if (au_has_group((char_u *)"filetypedetect"))
1054 do_cmdline_cmd((char_u *)":doau filetypedetect BufRead");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001055#endif
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001056 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057 }
1058 }
1059 }
1060
1061theend:
1062 if (tmp_orig != NULL)
1063 mch_remove(tmp_orig);
1064 vim_free(tmp_orig);
1065 if (tmp_new != NULL)
1066 mch_remove(tmp_new);
1067 vim_free(tmp_new);
1068 vim_free(newname);
1069 vim_free(buf);
1070#ifdef UNIX
1071 vim_free(fullname);
1072#endif
1073#ifdef FEAT_BROWSE
1074 vim_free(browseFile);
1075 cmdmod.browse = browse_flag;
1076#endif
1077}
1078
1079/*
1080 * Split the window and edit another file, setting options to show the diffs.
1081 */
1082 void
1083ex_diffsplit(eap)
1084 exarg_T *eap;
1085{
1086 win_T *old_curwin = curwin;
1087
1088#ifdef FEAT_GUI
1089 need_mouse_correct = TRUE;
1090#endif
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001091 /* don't use a new tab page, each tab page has its own diffs */
1092 cmdmod.tab = 0;
1093
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001094 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001095 {
1096 /* Pretend it was a ":split fname" command */
1097 eap->cmdidx = CMD_split;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001098 curwin->w_p_diff = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001099 do_exedit(eap, old_curwin);
1100
1101 if (curwin != old_curwin) /* split must have worked */
1102 {
1103 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1104 diff_win_options(curwin, TRUE);
1105 diff_win_options(old_curwin, TRUE);
1106 }
1107 }
1108}
1109
1110/*
1111 * Set options to show difs for the current window.
1112 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001113 void
1114ex_diffthis(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001115 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001116{
1117 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1118 diff_win_options(curwin, TRUE);
1119}
1120
1121/*
1122 * Set options in window "wp" for diff mode.
1123 */
1124 void
1125diff_win_options(wp, addbuf)
1126 win_T *wp;
1127 int addbuf; /* Add buffer to diff. */
1128{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001129# ifdef FEAT_FOLDING
1130 win_T *old_curwin = curwin;
1131
1132 /* close the manually opened folds */
1133 curwin = wp;
1134 newFoldLevel();
1135 curwin = old_curwin;
1136# endif
1137
Bram Moolenaar071d4272004-06-13 20:20:40 +00001138 wp->w_p_diff = TRUE;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001139 /* Use 'scrollbind' and 'cursorbind' when available */
1140#ifdef FEAT_SCROLLBIND
1141 wp->w_p_scb = TRUE;
1142#endif
Bram Moolenaar860cae12010-06-05 23:22:07 +02001143#ifdef FEAT_CURSORBIND
Bram Moolenaar860cae12010-06-05 23:22:07 +02001144 wp->w_p_crb = TRUE;
1145#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001146 wp->w_p_wrap = FALSE;
1147# ifdef FEAT_FOLDING
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001148 curwin = wp;
1149 curbuf = curwin->w_buffer;
1150 set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001151 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001152 curwin = old_curwin;
1153 curbuf = curwin->w_buffer;
1154 wp->w_p_fdc = diff_foldcolumn;
1155 wp->w_p_fen = TRUE;
1156 wp->w_p_fdl = 0;
1157 foldUpdateAll(wp);
1158 /* make sure topline is not halfway a fold */
1159 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001160# endif
1161#ifdef FEAT_SCROLLBIND
1162 if (vim_strchr(p_sbo, 'h') == NULL)
1163 do_cmdline_cmd((char_u *)"set sbo+=hor");
1164#endif
1165
1166 if (addbuf)
1167 diff_buf_add(wp->w_buffer);
1168 redraw_win_later(wp, NOT_VALID);
1169}
1170
1171/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001172 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001173 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001174 */
1175 void
1176ex_diffoff(eap)
1177 exarg_T *eap;
1178{
1179 win_T *wp;
1180 win_T *old_curwin = curwin;
1181#ifdef FEAT_SCROLLBIND
1182 int diffwin = FALSE;
1183#endif
1184
1185 for (wp = firstwin; wp != NULL; wp = wp->w_next)
1186 {
Bram Moolenaar21fa1ed2009-03-11 11:47:18 +00001187 if (wp == curwin || (eap->forceit && wp->w_p_diff))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001188 {
1189 /* Set 'diff', 'scrollbind' off and 'wrap' on. */
1190 wp->w_p_diff = FALSE;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001191 RESET_BINDING(wp);
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001192 wp->w_p_wrap = TRUE;
1193#ifdef FEAT_FOLDING
1194 curwin = wp;
1195 curbuf = curwin->w_buffer;
1196 set_string_option_direct((char_u *)"fdm", -1,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001197 (char_u *)"manual", OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001198 curwin = old_curwin;
1199 curbuf = curwin->w_buffer;
1200 wp->w_p_fdc = 0;
1201 wp->w_p_fen = FALSE;
1202 wp->w_p_fdl = 0;
1203 foldUpdateAll(wp);
1204 /* make sure topline is not halfway a fold */
1205 changed_window_setting_win(wp);
1206#endif
1207 diff_buf_adjust(wp);
1208 }
1209#ifdef FEAT_SCROLLBIND
1210 diffwin |= wp->w_p_diff;
1211#endif
1212 }
1213
1214#ifdef FEAT_SCROLLBIND
1215 /* Remove "hor" from from 'scrollopt' if there are no diff windows left. */
1216 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1217 do_cmdline_cmd((char_u *)"set sbo-=hor");
1218#endif
1219}
1220
1221/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001222 * Read the diff output and add each entry to the diff list.
1223 */
1224 static void
1225diff_read(idx_orig, idx_new, fname)
1226 int idx_orig; /* idx of original file */
1227 int idx_new; /* idx of new file */
1228 char_u *fname; /* name of diff output file */
1229{
1230 FILE *fd;
1231 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001232 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001233 diff_T *dn, *dpl;
1234 long f1, l1, f2, l2;
1235 char_u linebuf[LBUFLEN]; /* only need to hold the diff line */
1236 int difftype;
1237 char_u *p;
1238 long off;
1239 int i;
1240 linenr_T lnum_orig, lnum_new;
1241 long count_orig, count_new;
1242 int notset = TRUE; /* block "*dp" not set yet */
1243
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001244 fd = mch_fopen((char *)fname, "r");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001245 if (fd == NULL)
1246 {
1247 EMSG(_("E98: Cannot read diff output"));
1248 return;
1249 }
1250
1251 for (;;)
1252 {
1253 if (tag_fgets(linebuf, LBUFLEN, fd))
1254 break; /* end of file */
1255 if (!isdigit(*linebuf))
1256 continue; /* not the start of a diff block */
1257
1258 /* This line must be one of three formats:
1259 * {first}[,{last}]c{first}[,{last}]
1260 * {first}a{first}[,{last}]
1261 * {first}[,{last}]d{first}
1262 */
1263 p = linebuf;
1264 f1 = getdigits(&p);
1265 if (*p == ',')
1266 {
1267 ++p;
1268 l1 = getdigits(&p);
1269 }
1270 else
1271 l1 = f1;
1272 if (*p != 'a' && *p != 'c' && *p != 'd')
1273 continue; /* invalid diff format */
1274 difftype = *p++;
1275 f2 = getdigits(&p);
1276 if (*p == ',')
1277 {
1278 ++p;
1279 l2 = getdigits(&p);
1280 }
1281 else
1282 l2 = f2;
1283 if (l1 < f1 || l2 < f2)
1284 continue; /* invalid line range */
1285
1286 if (difftype == 'a')
1287 {
1288 lnum_orig = f1 + 1;
1289 count_orig = 0;
1290 }
1291 else
1292 {
1293 lnum_orig = f1;
1294 count_orig = l1 - f1 + 1;
1295 }
1296 if (difftype == 'd')
1297 {
1298 lnum_new = f2 + 1;
1299 count_new = 0;
1300 }
1301 else
1302 {
1303 lnum_new = f2;
1304 count_new = l2 - f2 + 1;
1305 }
1306
1307 /* Go over blocks before the change, for which orig and new are equal.
1308 * Copy blocks from orig to new. */
1309 while (dp != NULL
1310 && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig])
1311 {
1312 if (notset)
1313 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1314 dprev = dp;
1315 dp = dp->df_next;
1316 notset = TRUE;
1317 }
1318
1319 if (dp != NULL
1320 && lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig]
1321 && lnum_orig + count_orig >= dp->df_lnum[idx_orig])
1322 {
1323 /* New block overlaps with existing block(s).
1324 * First find last block that overlaps. */
1325 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
1326 if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig])
1327 break;
1328
1329 /* If the newly found block starts before the old one, set the
1330 * start back a number of lines. */
1331 off = dp->df_lnum[idx_orig] - lnum_orig;
1332 if (off > 0)
1333 {
1334 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001335 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001336 dp->df_lnum[i] -= off;
1337 dp->df_lnum[idx_new] = lnum_new;
1338 dp->df_count[idx_new] = count_new;
1339 }
1340 else if (notset)
1341 {
1342 /* new block inside existing one, adjust new block */
1343 dp->df_lnum[idx_new] = lnum_new + off;
1344 dp->df_count[idx_new] = count_new - off;
1345 }
1346 else
1347 /* second overlap of new block with existing block */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001348 dp->df_count[idx_new] += count_new - count_orig
1349 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1350 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001351
1352 /* Adjust the size of the block to include all the lines to the
1353 * end of the existing block or the new diff, whatever ends last. */
1354 off = (lnum_orig + count_orig)
1355 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1356 if (off < 0)
1357 {
1358 /* new change ends in existing block, adjust the end if not
1359 * done already */
1360 if (notset)
1361 dp->df_count[idx_new] += -off;
1362 off = 0;
1363 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001364 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001365 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001366 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1367 - dp->df_lnum[i] + off;
1368
1369 /* Delete the diff blocks that have been merged into one. */
1370 dn = dp->df_next;
1371 dp->df_next = dpl->df_next;
1372 while (dn != dp->df_next)
1373 {
1374 dpl = dn->df_next;
1375 vim_free(dn);
1376 dn = dpl;
1377 }
1378 }
1379 else
1380 {
1381 /* Allocate a new diffblock. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001382 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001383 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001384 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001385
1386 dp->df_lnum[idx_orig] = lnum_orig;
1387 dp->df_count[idx_orig] = count_orig;
1388 dp->df_lnum[idx_new] = lnum_new;
1389 dp->df_count[idx_new] = count_new;
1390
1391 /* Set values for other buffers, these must be equal to the
1392 * original buffer, otherwise there would have been a change
1393 * already. */
1394 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001395 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001396 diff_copy_entry(dprev, dp, idx_orig, i);
1397 }
1398 notset = FALSE; /* "*dp" has been set */
1399 }
1400
1401 /* for remaining diff blocks orig and new are equal */
1402 while (dp != NULL)
1403 {
1404 if (notset)
1405 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1406 dprev = dp;
1407 dp = dp->df_next;
1408 notset = TRUE;
1409 }
1410
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001411done:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001412 fclose(fd);
1413}
1414
1415/*
1416 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1417 */
1418 static void
1419diff_copy_entry(dprev, dp, idx_orig, idx_new)
1420 diff_T *dprev;
1421 diff_T *dp;
1422 int idx_orig;
1423 int idx_new;
1424{
1425 long off;
1426
1427 if (dprev == NULL)
1428 off = 0;
1429 else
1430 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1431 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1432 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1433 dp->df_count[idx_new] = dp->df_count[idx_orig];
1434}
1435
1436/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001437 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001438 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001439 void
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001440diff_clear(tp)
1441 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001442{
1443 diff_T *p, *next_p;
1444
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001445 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001446 {
1447 next_p = p->df_next;
1448 vim_free(p);
1449 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001450 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001451}
1452
1453/*
1454 * Check diff status for line "lnum" in buffer "buf":
1455 * Returns 0 for nothing special
1456 * Returns -1 for a line that should be highlighted as changed.
1457 * Returns -2 for a line that should be highlighted as added/deleted.
1458 * Returns > 0 for inserting that many filler lines above it (never happens
1459 * when 'diffopt' doesn't contain "filler").
1460 * This should only be used for windows where 'diff' is set.
1461 */
1462 int
1463diff_check(wp, lnum)
1464 win_T *wp;
1465 linenr_T lnum;
1466{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001467 int idx; /* index in tp_diffbuf[] for this buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001468 diff_T *dp;
1469 int maxcount;
1470 int i;
1471 buf_T *buf = wp->w_buffer;
1472 int cmp;
1473
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001474 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475 ex_diffupdate(NULL); /* update after a big change */
1476
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001477 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) /* no diffs at all */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001478 return 0;
1479
1480 /* safety check: "lnum" must be a buffer line */
1481 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1482 return 0;
1483
1484 idx = diff_buf_idx(buf);
1485 if (idx == DB_COUNT)
1486 return 0; /* no diffs for buffer "buf" */
1487
1488#ifdef FEAT_FOLDING
1489 /* A closed fold never has filler lines. */
1490 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1491 return 0;
1492#endif
1493
1494 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001495 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001496 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1497 break;
1498 if (dp == NULL || lnum < dp->df_lnum[idx])
1499 return 0;
1500
1501 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1502 {
1503 int zero = FALSE;
1504
1505 /* Changed or inserted line. If the other buffers have a count of
1506 * zero, the lines were inserted. If the other buffers have the same
1507 * count, check if the lines are identical. */
1508 cmp = FALSE;
1509 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001510 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001511 {
1512 if (dp->df_count[i] == 0)
1513 zero = TRUE;
1514 else
1515 {
1516 if (dp->df_count[i] != dp->df_count[idx])
1517 return -1; /* nr of lines changed. */
1518 cmp = TRUE;
1519 }
1520 }
1521 if (cmp)
1522 {
1523 /* Compare all lines. If they are equal the lines were inserted
1524 * in some buffers, deleted in others, but not changed. */
1525 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001526 if (i != idx && curtab->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001527 if (!diff_equal_entry(dp, idx, i))
1528 return -1;
1529 }
1530 /* If there is no buffer with zero lines then there is no difference
1531 * any longer. Happens when making a change (or undo) that removes
1532 * the difference. Can't remove the entry here, we might be halfway
1533 * updating the window. Just report the text as unchanged. Other
1534 * windows might still show the change though. */
1535 if (zero == FALSE)
1536 return 0;
1537 return -2;
1538 }
1539
1540 /* If 'diffopt' doesn't contain "filler", return 0. */
1541 if (!(diff_flags & DIFF_FILLER))
1542 return 0;
1543
1544 /* Insert filler lines above the line just below the change. Will return
1545 * 0 when this buf had the max count. */
1546 maxcount = 0;
1547 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001548 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001549 maxcount = dp->df_count[i];
1550 return maxcount - dp->df_count[idx];
1551}
1552
1553/*
1554 * Compare two entries in diff "*dp" and return TRUE if they are equal.
1555 */
1556 static int
1557diff_equal_entry(dp, idx1, idx2)
1558 diff_T *dp;
1559 int idx1;
1560 int idx2;
1561{
1562 int i;
1563 char_u *line;
1564 int cmp;
1565
1566 if (dp->df_count[idx1] != dp->df_count[idx2])
1567 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001568 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001569 return FALSE;
1570 for (i = 0; i < dp->df_count[idx1]; ++i)
1571 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001572 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001573 dp->df_lnum[idx1] + i, FALSE));
1574 if (line == NULL)
1575 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001576 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001577 dp->df_lnum[idx2] + i, FALSE));
1578 vim_free(line);
1579 if (cmp != 0)
1580 return FALSE;
1581 }
1582 return TRUE;
1583}
1584
1585/*
1586 * Compare strings "s1" and "s2" according to 'diffopt'.
1587 * Return non-zero when they are different.
1588 */
1589 static int
1590diff_cmp(s1, s2)
1591 char_u *s1;
1592 char_u *s2;
1593{
1594 char_u *p1, *p2;
1595#ifdef FEAT_MBYTE
1596 int l;
1597#endif
1598
1599 if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0)
1600 return STRCMP(s1, s2);
1601 if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE))
1602 return MB_STRICMP(s1, s2);
1603
1604 /* Ignore white space changes and possibly ignore case. */
1605 p1 = s1;
1606 p2 = s2;
1607 while (*p1 != NUL && *p2 != NUL)
1608 {
1609 if (vim_iswhite(*p1) && vim_iswhite(*p2))
1610 {
1611 p1 = skipwhite(p1);
1612 p2 = skipwhite(p2);
1613 }
1614 else
1615 {
1616#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001617 l = (*mb_ptr2len)(p1);
1618 if (l != (*mb_ptr2len)(p2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001619 break;
1620 if (l > 1)
1621 {
1622 if (STRNCMP(p1, p2, l) != 0
1623 && (!enc_utf8
1624 || !(diff_flags & DIFF_ICASE)
1625 || utf_fold(utf_ptr2char(p1))
1626 != utf_fold(utf_ptr2char(p2))))
1627 break;
1628 p1 += l;
1629 p2 += l;
1630 }
1631 else
1632#endif
1633 {
1634 if (*p1 != *p2 && (!(diff_flags & DIFF_ICASE)
1635 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
1636 break;
1637 ++p1;
1638 ++p2;
1639 }
1640 }
1641 }
1642
1643 /* Ignore trailing white space. */
1644 p1 = skipwhite(p1);
1645 p2 = skipwhite(p2);
1646 if (*p1 != NUL || *p2 != NUL)
1647 return 1;
1648 return 0;
1649}
1650
1651/*
1652 * Return the number of filler lines above "lnum".
1653 */
1654 int
1655diff_check_fill(wp, lnum)
1656 win_T *wp;
1657 linenr_T lnum;
1658{
1659 int n;
1660
1661 /* be quick when there are no filler lines */
1662 if (!(diff_flags & DIFF_FILLER))
1663 return 0;
1664 n = diff_check(wp, lnum);
1665 if (n <= 0)
1666 return 0;
1667 return n;
1668}
1669
1670/*
1671 * Set the topline of "towin" to match the position in "fromwin", so that they
1672 * show the same diff'ed lines.
1673 */
1674 void
1675diff_set_topline(fromwin, towin)
1676 win_T *fromwin;
1677 win_T *towin;
1678{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001679 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001680 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001681 int fromidx;
1682 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001683 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001684 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001685 int i;
1686
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001687 fromidx = diff_buf_idx(frombuf);
1688 if (fromidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001689 return; /* safety check */
1690
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001691 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001692 ex_diffupdate(NULL); /* update after a big change */
1693
1694 towin->w_topfill = 0;
1695
1696 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001697 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001698 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001699 break;
1700 if (dp == NULL)
1701 {
1702 /* After last change, compute topline relative to end of file; no
1703 * filler lines. */
1704 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001705 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001706 }
1707 else
1708 {
1709 /* Find index for "towin". */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001710 toidx = diff_buf_idx(towin->w_buffer);
1711 if (toidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001712 return; /* safety check */
1713
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001714 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
1715 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001716 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001717 /* Inside a change: compute filler lines. With three or more
1718 * buffers we need to know the largest count. */
1719 max_count = 0;
1720 for (i = 0; i < DB_COUNT; ++i)
1721 if (curtab->tp_diffbuf[i] != NULL
1722 && max_count < dp->df_count[i])
1723 max_count = dp->df_count[i];
1724
1725 if (dp->df_count[toidx] == dp->df_count[fromidx])
1726 {
1727 /* same number of lines: use same filler count */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001728 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001729 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001730 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001731 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001732 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001733 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001734 /* more lines in towin and fromwin doesn't show diff
1735 * lines, only filler lines */
1736 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
1737 {
1738 /* towin also only shows filler lines */
1739 towin->w_topline = dp->df_lnum[toidx]
1740 + dp->df_count[toidx];
1741 towin->w_topfill = fromwin->w_topfill;
1742 }
1743 else
1744 /* towin still has some diff lines to show */
1745 towin->w_topline = dp->df_lnum[toidx]
1746 + max_count - fromwin->w_topfill;
1747 }
1748 }
1749 else if (towin->w_topline >= dp->df_lnum[toidx]
1750 + dp->df_count[toidx])
1751 {
1752 /* less lines in towin and no diff lines to show: compute
1753 * filler lines */
1754 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
1755 if (diff_flags & DIFF_FILLER)
1756 {
1757 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
1758 /* fromwin is also out of diff lines */
1759 towin->w_topfill = fromwin->w_topfill;
1760 else
1761 /* fromwin has some diff lines */
1762 towin->w_topfill = dp->df_lnum[fromidx]
1763 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001764 }
1765 }
1766 }
1767 }
1768
1769 /* safety check (if diff info gets outdated strange things may happen) */
1770 towin->w_botfill = FALSE;
1771 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
1772 {
1773 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
1774 towin->w_botfill = TRUE;
1775 }
1776 if (towin->w_topline < 1)
1777 {
1778 towin->w_topline = 1;
1779 towin->w_topfill = 0;
1780 }
1781
1782 /* When w_topline changes need to recompute w_botline and cursor position */
1783 invalidate_botline_win(towin);
1784 changed_line_abv_curs_win(towin);
1785
1786 check_topfill(towin, FALSE);
1787#ifdef FEAT_FOLDING
1788 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
1789 NULL, TRUE, NULL);
1790#endif
1791}
1792
1793/*
1794 * This is called when 'diffopt' is changed.
1795 */
1796 int
1797diffopt_changed()
1798{
1799 char_u *p;
1800 int diff_context_new = 6;
1801 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001802 int diff_foldcolumn_new = 2;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001803 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001804
1805 p = p_dip;
1806 while (*p != NUL)
1807 {
1808 if (STRNCMP(p, "filler", 6) == 0)
1809 {
1810 p += 6;
1811 diff_flags_new |= DIFF_FILLER;
1812 }
1813 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
1814 {
1815 p += 8;
1816 diff_context_new = getdigits(&p);
1817 }
1818 else if (STRNCMP(p, "icase", 5) == 0)
1819 {
1820 p += 5;
1821 diff_flags_new |= DIFF_ICASE;
1822 }
1823 else if (STRNCMP(p, "iwhite", 6) == 0)
1824 {
1825 p += 6;
1826 diff_flags_new |= DIFF_IWHITE;
1827 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001828 else if (STRNCMP(p, "horizontal", 10) == 0)
1829 {
1830 p += 10;
1831 diff_flags_new |= DIFF_HORIZONTAL;
1832 }
1833 else if (STRNCMP(p, "vertical", 8) == 0)
1834 {
1835 p += 8;
1836 diff_flags_new |= DIFF_VERTICAL;
1837 }
1838 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
1839 {
1840 p += 11;
1841 diff_foldcolumn_new = getdigits(&p);
1842 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001843 if (*p != ',' && *p != NUL)
1844 return FAIL;
1845 if (*p == ',')
1846 ++p;
1847 }
1848
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001849 /* Can't have both "horizontal" and "vertical". */
1850 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
1851 return FAIL;
1852
Bram Moolenaar071d4272004-06-13 20:20:40 +00001853 /* If "icase" or "iwhite" was added or removed, need to update the diff. */
1854 if (diff_flags != diff_flags_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001855 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
1856 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001857
1858 diff_flags = diff_flags_new;
1859 diff_context = diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001860 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001861
1862 diff_redraw(TRUE);
1863
1864 /* recompute the scroll binding with the new option value, may
1865 * remove or add filler lines */
1866 check_scrollbind((linenr_T)0, 0L);
1867
1868 return OK;
1869}
1870
1871/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001872 * Return TRUE if 'diffopt' contains "horizontal".
1873 */
1874 int
1875diffopt_horizontal()
1876{
1877 return (diff_flags & DIFF_HORIZONTAL) != 0;
1878}
1879
1880/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001881 * Find the difference within a changed line.
1882 * Returns TRUE if the line was added, no other buffer has it.
1883 */
1884 int
1885diff_find_change(wp, lnum, startp, endp)
1886 win_T *wp;
1887 linenr_T lnum;
1888 int *startp; /* first char of the change */
1889 int *endp; /* last char of the change */
1890{
1891 char_u *line_org;
1892 char_u *line_new;
1893 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001894 int si_org, si_new;
1895 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001896 diff_T *dp;
1897 int idx;
1898 int off;
1899 int added = TRUE;
1900
1901 /* Make a copy of the line, the next ml_get() will invalidate it. */
1902 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
1903 if (line_org == NULL)
1904 return FALSE;
1905
1906 idx = diff_buf_idx(wp->w_buffer);
1907 if (idx == DB_COUNT) /* cannot happen */
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001908 {
1909 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001910 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001911 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001912
1913 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001914 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001915 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1916 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001917 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001918 {
1919 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001921 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001922
1923 off = lnum - dp->df_lnum[idx];
1924
1925 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001926 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001927 {
1928 /* Skip lines that are not in the other change (filler lines). */
1929 if (off >= dp->df_count[i])
1930 continue;
1931 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001932 line_new = ml_get_buf(curtab->tp_diffbuf[i],
1933 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001934
1935 /* Search for start of difference */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001936 si_org = si_new = 0;
1937 while (line_org[si_org] != NUL)
1938 {
1939 if ((diff_flags & DIFF_IWHITE)
1940 && vim_iswhite(line_org[si_org])
1941 && vim_iswhite(line_new[si_new]))
1942 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001943 si_org = (int)(skipwhite(line_org + si_org) - line_org);
1944 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001945 }
1946 else
1947 {
1948 if (line_org[si_org] != line_new[si_new])
1949 break;
1950 ++si_org;
1951 ++si_new;
1952 }
1953 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001954#ifdef FEAT_MBYTE
1955 if (has_mbyte)
1956 {
1957 /* Move back to first byte of character in both lines (may
1958 * have "nn^" in line_org and "n^ in line_new). */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001959 si_org -= (*mb_head_off)(line_org, line_org + si_org);
1960 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001961 }
1962#endif
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001963 if (*startp > si_org)
1964 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001965
1966 /* Search for end of difference, if any. */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001967 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001968 {
1969 ei_org = (int)STRLEN(line_org);
1970 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001971 while (ei_org >= *startp && ei_new >= si_new
1972 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001973 {
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001974 if ((diff_flags & DIFF_IWHITE)
1975 && vim_iswhite(line_org[ei_org])
1976 && vim_iswhite(line_new[ei_new]))
1977 {
1978 while (ei_org >= *startp
1979 && vim_iswhite(line_org[ei_org]))
1980 --ei_org;
1981 while (ei_new >= si_new
1982 && vim_iswhite(line_new[ei_new]))
1983 --ei_new;
1984 }
1985 else
1986 {
1987 if (line_org[ei_org] != line_new[ei_new])
1988 break;
1989 --ei_org;
1990 --ei_new;
1991 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 }
1993 if (*endp < ei_org)
1994 *endp = ei_org;
1995 }
1996 }
1997
1998 vim_free(line_org);
1999 return added;
2000}
2001
2002#if defined(FEAT_FOLDING) || defined(PROTO)
2003/*
2004 * Return TRUE if line "lnum" is not close to a diff block, this line should
2005 * be in a fold.
2006 * Return FALSE if there are no diff blocks at all in this window.
2007 */
2008 int
2009diff_infold(wp, lnum)
2010 win_T *wp;
2011 linenr_T lnum;
2012{
2013 int i;
2014 int idx = -1;
2015 int other = FALSE;
2016 diff_T *dp;
2017
2018 /* Return if 'diff' isn't set. */
2019 if (!wp->w_p_diff)
2020 return FALSE;
2021
2022 for (i = 0; i < DB_COUNT; ++i)
2023 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002024 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002025 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002026 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002027 other = TRUE;
2028 }
2029
2030 /* return here if there are no diffs in the window */
2031 if (idx == -1 || !other)
2032 return FALSE;
2033
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002034 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002035 ex_diffupdate(NULL); /* update after a big change */
2036
2037 /* Return if there are no diff blocks. All lines will be folded. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002038 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039 return TRUE;
2040
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002041 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002042 {
2043 /* If this change is below the line there can't be any further match. */
2044 if (dp->df_lnum[idx] - diff_context > lnum)
2045 break;
2046 /* If this change ends before the line we have a match. */
2047 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2048 return FALSE;
2049 }
2050 return TRUE;
2051}
2052#endif
2053
2054/*
2055 * "dp" and "do" commands.
2056 */
2057 void
2058nv_diffgetput(put)
2059 int put;
2060{
2061 exarg_T ea;
2062
2063 ea.arg = (char_u *)"";
2064 if (put)
2065 ea.cmdidx = CMD_diffput;
2066 else
2067 ea.cmdidx = CMD_diffget;
2068 ea.addr_count = 0;
2069 ea.line1 = curwin->w_cursor.lnum;
2070 ea.line2 = curwin->w_cursor.lnum;
2071 ex_diffgetput(&ea);
2072}
2073
2074/*
2075 * ":diffget"
2076 * ":diffput"
2077 */
2078 void
2079ex_diffgetput(eap)
2080 exarg_T *eap;
2081{
2082 linenr_T lnum;
2083 int count;
2084 linenr_T off = 0;
2085 diff_T *dp;
2086 diff_T *dprev;
2087 diff_T *dfree;
2088 int idx_cur;
2089 int idx_other;
2090 int idx_from;
2091 int idx_to;
2092 int i;
2093 int added;
2094 char_u *p;
2095 aco_save_T aco;
2096 buf_T *buf;
2097 int start_skip, end_skip;
2098 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002099 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002100 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002101
2102 /* Find the current buffer in the list of diff buffers. */
2103 idx_cur = diff_buf_idx(curbuf);
2104 if (idx_cur == DB_COUNT)
2105 {
2106 EMSG(_("E99: Current buffer is not in diff mode"));
2107 return;
2108 }
2109
2110 if (*eap->arg == NUL)
2111 {
2112 /* No argument: Find the other buffer in the list of diff buffers. */
2113 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002114 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002115 && curtab->tp_diffbuf[idx_other] != NULL)
2116 {
2117 if (eap->cmdidx != CMD_diffput
2118 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2119 break;
2120 found_not_ma = TRUE;
2121 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002122 if (idx_other == DB_COUNT)
2123 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002124 if (found_not_ma)
2125 EMSG(_("E793: No other buffer in diff mode is modifiable"));
2126 else
2127 EMSG(_("E100: No other buffer in diff mode"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002128 return;
2129 }
2130
2131 /* Check that there isn't a third buffer in the list */
2132 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002133 if (curtab->tp_diffbuf[i] != curbuf
2134 && curtab->tp_diffbuf[i] != NULL
2135 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002136 {
2137 EMSG(_("E101: More than two buffers in diff mode, don't know which one to use"));
2138 return;
2139 }
2140 }
2141 else
2142 {
2143 /* Buffer number or pattern given. Ignore trailing white space. */
2144 p = eap->arg + STRLEN(eap->arg);
2145 while (p > eap->arg && vim_iswhite(p[-1]))
2146 --p;
2147 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2148 ;
2149 if (eap->arg + i == p) /* digits only */
2150 i = atol((char *)eap->arg);
2151 else
2152 {
2153 i = buflist_findpat(eap->arg, p, FALSE, TRUE);
2154 if (i < 0)
2155 return; /* error message already given */
2156 }
2157 buf = buflist_findnr(i);
2158 if (buf == NULL)
2159 {
2160 EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg);
2161 return;
2162 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002163 if (buf == curbuf)
2164 return; /* nothing to do */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002165 idx_other = diff_buf_idx(buf);
2166 if (idx_other == DB_COUNT)
2167 {
2168 EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg);
2169 return;
2170 }
2171 }
2172
2173 diff_busy = TRUE;
2174
2175 /* When no range given include the line above or below the cursor. */
2176 if (eap->addr_count == 0)
2177 {
2178 /* Make it possible that ":diffget" on the last line gets line below
2179 * the cursor line when there is no difference above the cursor. */
2180 if (eap->cmdidx == CMD_diffget
2181 && eap->line1 == curbuf->b_ml.ml_line_count
2182 && diff_check(curwin, eap->line1) == 0
2183 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2184 ++eap->line2;
2185 else if (eap->line1 > 0)
2186 --eap->line1;
2187 }
2188
2189 if (eap->cmdidx == CMD_diffget)
2190 {
2191 idx_from = idx_other;
2192 idx_to = idx_cur;
2193 }
2194 else
2195 {
2196 idx_from = idx_cur;
2197 idx_to = idx_other;
2198 /* Need to make the other buffer the current buffer to be able to make
2199 * changes in it. */
2200 /* set curwin/curbuf to buf and save a few things */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002201 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002202 }
2203
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002204 /* May give the warning for a changed buffer here, which can trigger the
2205 * FileChangedRO autocommand, which may do nasty things and mess
2206 * everything up. */
2207 if (!curbuf->b_changed)
2208 {
2209 change_warning(0);
2210 if (diff_buf_idx(curbuf) != idx_to)
2211 {
2212 EMSG(_("E787: Buffer changed unexpectedly"));
2213 return;
2214 }
2215 }
2216
Bram Moolenaar071d4272004-06-13 20:20:40 +00002217 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002218 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002219 {
2220 if (dp->df_lnum[idx_cur] > eap->line2 + off)
2221 break; /* past the range that was specified */
2222
2223 dfree = NULL;
2224 lnum = dp->df_lnum[idx_to];
2225 count = dp->df_count[idx_to];
2226 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2227 && u_save(lnum - 1, lnum + count) != FAIL)
2228 {
2229 /* Inside the specified range and saving for undo worked. */
2230 start_skip = 0;
2231 end_skip = 0;
2232 if (eap->addr_count > 0)
2233 {
2234 /* A range was specified: check if lines need to be skipped. */
2235 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2236 if (start_skip > 0)
2237 {
2238 /* range starts below start of current diff block */
2239 if (start_skip > count)
2240 {
2241 lnum += count;
2242 count = 0;
2243 }
2244 else
2245 {
2246 count -= start_skip;
2247 lnum += start_skip;
2248 }
2249 }
2250 else
2251 start_skip = 0;
2252
2253 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2254 - (eap->line2 + off);
2255 if (end_skip > 0)
2256 {
2257 /* range ends above end of current/from diff block */
2258 if (idx_cur == idx_from) /* :diffput */
2259 {
2260 i = dp->df_count[idx_cur] - start_skip - end_skip;
2261 if (count > i)
2262 count = i;
2263 }
2264 else /* :diffget */
2265 {
2266 count -= end_skip;
2267 end_skip = dp->df_count[idx_from] - start_skip - count;
2268 if (end_skip < 0)
2269 end_skip = 0;
2270 }
2271 }
2272 else
2273 end_skip = 0;
2274 }
2275
Bram Moolenaar280f1262006-01-30 00:14:18 +00002276 buf_empty = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002277 added = 0;
2278 for (i = 0; i < count; ++i)
2279 {
Bram Moolenaar280f1262006-01-30 00:14:18 +00002280 /* remember deleting the last line of the buffer */
2281 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002282 ml_delete(lnum, FALSE);
2283 --added;
2284 }
2285 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2286 {
2287 linenr_T nr;
2288
2289 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002290 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002291 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002292 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2293 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002294 if (p != NULL)
2295 {
2296 ml_append(lnum + i - 1, p, 0, FALSE);
2297 vim_free(p);
2298 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002299 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2300 {
2301 /* Added the first line into an empty buffer, need to
2302 * delete the dummy empty line. */
2303 buf_empty = FALSE;
2304 ml_delete((linenr_T)2, FALSE);
2305 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002306 }
2307 }
2308 new_count = dp->df_count[idx_to] + added;
2309 dp->df_count[idx_to] = new_count;
2310
2311 if (start_skip == 0 && end_skip == 0)
2312 {
2313 /* Check if there are any other buffers and if the diff is
2314 * equal in them. */
2315 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002316 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2317 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002318 && !diff_equal_entry(dp, idx_from, i))
2319 break;
2320 if (i == DB_COUNT)
2321 {
2322 /* delete the diff entry, the buffers are now equal here */
2323 dfree = dp;
2324 dp = dp->df_next;
2325 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002326 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002327 else
2328 dprev->df_next = dp;
2329 }
2330 }
2331
2332 /* Adjust marks. This will change the following entries! */
2333 if (added != 0)
2334 {
2335 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2336 if (curwin->w_cursor.lnum >= lnum)
2337 {
2338 /* Adjust the cursor position if it's in/after the changed
2339 * lines. */
2340 if (curwin->w_cursor.lnum >= lnum + count)
2341 curwin->w_cursor.lnum += added;
2342 else if (added < 0)
2343 curwin->w_cursor.lnum = lnum;
2344 }
2345 }
2346 changed_lines(lnum, 0, lnum + count, (long)added);
2347
2348 if (dfree != NULL)
2349 {
2350 /* Diff is deleted, update folds in other windows. */
2351#ifdef FEAT_FOLDING
2352 diff_fold_update(dfree, idx_to);
2353#endif
2354 vim_free(dfree);
2355 }
2356 else
2357 /* mark_adjust() may have changed the count in a wrong way */
2358 dp->df_count[idx_to] = new_count;
2359
2360 /* When changing the current buffer, keep track of line numbers */
2361 if (idx_cur == idx_to)
2362 off += added;
2363 }
2364
2365 /* If before the range or not deleted, go to next diff. */
2366 if (dfree == NULL)
2367 {
2368 dprev = dp;
2369 dp = dp->df_next;
2370 }
2371 }
2372
2373 /* restore curwin/curbuf and a few other things */
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002374 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002375 {
2376 /* Syncing undo only works for the current buffer, but we change
2377 * another buffer. Sync undo if the command was typed. This isn't
2378 * 100% right when ":diffput" is used in a function or mapping. */
2379 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002380 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002381 aucmd_restbuf(&aco);
2382 }
2383
2384 diff_busy = FALSE;
2385
2386 /* Check that the cursor is on a valid character and update it's position.
2387 * When there were filler lines the topline has become invalid. */
2388 check_cursor();
2389 changed_line_abv_curs();
2390
2391 /* Also need to redraw the other buffers. */
2392 diff_redraw(FALSE);
2393}
2394
2395#ifdef FEAT_FOLDING
2396/*
2397 * Update folds for all diff buffers for entry "dp".
2398 * Skip buffer with index "skip_idx".
2399 * When there are no diffs, all folds are removed.
2400 */
2401 static void
2402diff_fold_update(dp, skip_idx)
2403 diff_T *dp;
2404 int skip_idx;
2405{
2406 int i;
2407 win_T *wp;
2408
2409 for (wp = firstwin; wp != NULL; wp = wp->w_next)
2410 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002411 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002412 foldUpdate(wp, dp->df_lnum[i],
2413 dp->df_lnum[i] + dp->df_count[i]);
2414}
2415#endif
2416
2417/*
2418 * Return TRUE if buffer "buf" is in diff-mode.
2419 */
2420 int
2421diff_mode_buf(buf)
2422 buf_T *buf;
2423{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002424 tabpage_T *tp;
2425
2426 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
2427 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
2428 return TRUE;
2429 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002430}
2431
2432/*
2433 * Move "count" times in direction "dir" to the next diff block.
2434 * Return FAIL if there isn't such a diff block.
2435 */
2436 int
2437diff_move_to(dir, count)
2438 int dir;
2439 long count;
2440{
2441 int idx;
2442 linenr_T lnum = curwin->w_cursor.lnum;
2443 diff_T *dp;
2444
2445 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002446 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002447 return FAIL;
2448
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002449 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002450 ex_diffupdate(NULL); /* update after a big change */
2451
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002452 if (curtab->tp_first_diff == NULL) /* no diffs today */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002453 return FAIL;
2454
2455 while (--count >= 0)
2456 {
2457 /* Check if already before first diff. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002458 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002459 break;
2460
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002461 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002462 {
2463 if (dp == NULL)
2464 break;
2465 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
2466 || (dir == BACKWARD
2467 && (dp->df_next == NULL
2468 || lnum <= dp->df_next->df_lnum[idx])))
2469 {
2470 lnum = dp->df_lnum[idx];
2471 break;
2472 }
2473 }
2474 }
2475
2476 /* don't end up past the end of the file */
2477 if (lnum > curbuf->b_ml.ml_line_count)
2478 lnum = curbuf->b_ml.ml_line_count;
2479
2480 /* When the cursor didn't move at all we fail. */
2481 if (lnum == curwin->w_cursor.lnum)
2482 return FAIL;
2483
2484 setpcmark();
2485 curwin->w_cursor.lnum = lnum;
2486 curwin->w_cursor.col = 0;
2487
2488 return OK;
2489}
2490
Bram Moolenaar860cae12010-06-05 23:22:07 +02002491#if defined(FEAT_CURSORBIND) || defined(PROTO)
2492 linenr_T
2493diff_get_corresponding_line(buf1, lnum1, buf2, lnum3)
2494 buf_T *buf1;
2495 linenr_T lnum1;
2496 buf_T *buf2;
2497 linenr_T lnum3;
2498{
2499 int idx1;
2500 int idx2;
2501 diff_T *dp;
2502 int baseline = 0;
2503 linenr_T lnum2;
2504
2505 idx1 = diff_buf_idx(buf1);
2506 idx2 = diff_buf_idx(buf2);
2507 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
2508 return lnum1;
2509
2510 if (curtab->tp_diff_invalid)
2511 ex_diffupdate(NULL); /* update after a big change */
2512
2513 if (curtab->tp_first_diff == NULL) /* no diffs today */
2514 return lnum1;
2515
2516 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
2517 {
2518 if (dp->df_lnum[idx1] > lnum1)
2519 {
2520 lnum2 = lnum1 - baseline;
2521 /* don't end up past the end of the file */
2522 if (lnum2 > buf2->b_ml.ml_line_count)
2523 lnum2 = buf2->b_ml.ml_line_count;
2524
2525 return lnum2;
2526 }
2527 else if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
2528 {
2529 /* Inside the diffblock */
2530 baseline = lnum1 - dp->df_lnum[idx1];
2531 if (baseline > dp->df_count[idx2])
2532 baseline = dp->df_count[idx2];
2533
2534 return dp->df_lnum[idx2] + baseline;
2535 }
2536 else if ( (dp->df_lnum[idx1] == lnum1)
2537 && (dp->df_count[idx1] == 0)
2538 && (dp->df_lnum[idx2] <= lnum3)
2539 && ((dp->df_lnum[idx2] + dp->df_count[idx2]) > lnum3))
2540 /*
2541 * Special case: if the cursor is just after a zero-count
2542 * block (i.e. all filler) and the target cursor is already
2543 * inside the corresponding block, leave the target cursor
2544 * unmoved. This makes repeated CTRL-W W operations work
2545 * as expected.
2546 */
2547 return lnum3;
2548 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
2549 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
2550 }
2551
2552 /* If we get here then the cursor is after the last diff */
2553 lnum2 = lnum1 - baseline;
2554 /* don't end up past the end of the file */
2555 if (lnum2 > buf2->b_ml.ml_line_count)
2556 lnum2 = buf2->b_ml.ml_line_count;
2557
2558 return lnum2;
2559}
2560#endif
2561
Bram Moolenaar071d4272004-06-13 20:20:40 +00002562#if defined(FEAT_FOLDING) || defined(PROTO)
2563/*
2564 * For line "lnum" in the current window find the equivalent lnum in window
2565 * "wp", compensating for inserted/deleted lines.
2566 */
2567 linenr_T
2568diff_lnum_win(lnum, wp)
2569 linenr_T lnum;
2570 win_T *wp;
2571{
2572 diff_T *dp;
2573 int idx;
2574 int i;
2575 linenr_T n;
2576
2577 idx = diff_buf_idx(curbuf);
2578 if (idx == DB_COUNT) /* safety check */
2579 return (linenr_T)0;
2580
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002581 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002582 ex_diffupdate(NULL); /* update after a big change */
2583
2584 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002585 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002586 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2587 break;
2588
2589 /* When after the last change, compute relative to the last line number. */
2590 if (dp == NULL)
2591 return wp->w_buffer->b_ml.ml_line_count
2592 - (curbuf->b_ml.ml_line_count - lnum);
2593
2594 /* Find index for "wp". */
2595 i = diff_buf_idx(wp->w_buffer);
2596 if (i == DB_COUNT) /* safety check */
2597 return (linenr_T)0;
2598
2599 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
2600 if (n > dp->df_lnum[i] + dp->df_count[i])
2601 n = dp->df_lnum[i] + dp->df_count[i];
2602 return n;
2603}
2604#endif
2605
2606#endif /* FEAT_DIFF */