blob: b7f0a1015d25322e1b7a7534033e2e769879337a [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 */
655/*ARGSUSED*/
656 void
657ex_diffupdate(eap)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000658 exarg_T *eap; /* can be NULL, it's not used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659{
660 buf_T *buf;
661 int idx_orig;
662 int idx_new;
663 char_u *tmp_orig;
664 char_u *tmp_new;
665 char_u *tmp_diff;
666 FILE *fd;
667 int ok;
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000668 int io_error = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000669
670 /* Delete all diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000671 diff_clear(curtab);
672 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000673
674 /* Use the first buffer as the original text. */
675 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000676 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000677 break;
678 if (idx_orig == DB_COUNT)
679 return;
680
681 /* Only need to do something when there is another buffer. */
682 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000683 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000684 break;
685 if (idx_new == DB_COUNT)
686 return;
687
688 /* We need three temp file names. */
689 tmp_orig = vim_tempname('o');
690 tmp_new = vim_tempname('n');
691 tmp_diff = vim_tempname('d');
692 if (tmp_orig == NULL || tmp_new == NULL || tmp_diff == NULL)
693 goto theend;
694
695 /*
696 * Do a quick test if "diff" really works. Otherwise it looks like there
697 * are no differences. Can't use the return value, it's non-zero when
698 * there are differences.
699 * May try twice, first with "-a" and then without.
700 */
701 for (;;)
702 {
703 ok = FALSE;
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000704 fd = mch_fopen((char *)tmp_orig, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000705 if (fd == NULL)
706 io_error = TRUE;
707 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000708 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000709 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
710 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000711 fclose(fd);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000712 fd = mch_fopen((char *)tmp_new, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000713 if (fd == NULL)
714 io_error = TRUE;
715 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000716 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000717 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
718 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000719 fclose(fd);
720 diff_file(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000721 fd = mch_fopen((char *)tmp_diff, "r");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000722 if (fd == NULL)
723 io_error = TRUE;
724 else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000725 {
726 char_u linebuf[LBUFLEN];
727
728 for (;;)
729 {
730 /* There must be a line that contains "1c1". */
731 if (tag_fgets(linebuf, LBUFLEN, fd))
732 break;
733 if (STRNCMP(linebuf, "1c1", 3) == 0)
734 ok = TRUE;
735 }
736 fclose(fd);
737 }
738 mch_remove(tmp_diff);
739 mch_remove(tmp_new);
740 }
741 mch_remove(tmp_orig);
742 }
743
744#ifdef FEAT_EVAL
745 /* When using 'diffexpr' break here. */
746 if (*p_dex != NUL)
747 break;
748#endif
749
750#if defined(MSWIN) || defined(MSDOS)
751 /* If the "-a" argument works, also check if "--binary" works. */
752 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
753 {
754 diff_a_works = TRUE;
755 diff_bin_works = TRUE;
756 continue;
757 }
758 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
759 {
760 /* Tried --binary, but it failed. "-a" works though. */
761 diff_bin_works = FALSE;
762 ok = TRUE;
763 }
764#endif
765
766 /* If we checked if "-a" works already, break here. */
767 if (diff_a_works != MAYBE)
768 break;
769 diff_a_works = ok;
770
771 /* If "-a" works break here, otherwise retry without "-a". */
772 if (ok)
773 break;
774 }
775 if (!ok)
776 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000777 if (io_error)
778 EMSG(_("E810: Cannot read or write temp files"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000779 EMSG(_("E97: Cannot create diffs"));
780 diff_a_works = MAYBE;
781#if defined(MSWIN) || defined(MSDOS)
782 diff_bin_works = MAYBE;
783#endif
784 goto theend;
785 }
786
787 /* Write the first buffer to a tempfile. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000788 buf = curtab->tp_diffbuf[idx_orig];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000789 if (diff_write(buf, tmp_orig) == FAIL)
790 goto theend;
791
792 /* Make a difference between the first buffer and every other. */
793 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
794 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000795 buf = curtab->tp_diffbuf[idx_new];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796 if (buf == NULL)
797 continue;
798 if (diff_write(buf, tmp_new) == FAIL)
799 continue;
800 diff_file(tmp_orig, tmp_new, tmp_diff);
801
802 /* Read the diff output and add each entry to the diff list. */
803 diff_read(idx_orig, idx_new, tmp_diff);
804 mch_remove(tmp_diff);
805 mch_remove(tmp_new);
806 }
807 mch_remove(tmp_orig);
808
Bram Moolenaar60a44dc2007-10-19 16:58:12 +0000809 /* force updating cursor position on screen */
810 curwin->w_valid_cursor.lnum = 0;
811
Bram Moolenaar071d4272004-06-13 20:20:40 +0000812 diff_redraw(TRUE);
813
814theend:
815 vim_free(tmp_orig);
816 vim_free(tmp_new);
817 vim_free(tmp_diff);
818}
819
820/*
821 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
822 */
823 static void
824diff_file(tmp_orig, tmp_new, tmp_diff)
825 char_u *tmp_orig;
826 char_u *tmp_new;
827 char_u *tmp_diff;
828{
829 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000830 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831
832#ifdef FEAT_EVAL
833 if (*p_dex != NUL)
834 /* Use 'diffexpr' to generate the diff file. */
835 eval_diff(tmp_orig, tmp_new, tmp_diff);
836 else
837#endif
838 {
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000839 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
840 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
841 cmd = alloc((unsigned)len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000842 if (cmd != NULL)
843 {
Bram Moolenaar95fb60a2005-03-08 22:29:20 +0000844 /* We don't want $DIFF_OPTIONS to get in the way. */
845 if (getenv("DIFF_OPTIONS"))
846 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
847
Bram Moolenaar071d4272004-06-13 20:20:40 +0000848 /* Build the diff command and execute it. Always use -a, binary
849 * differences are of no use. Ignore errors, diff returns
850 * non-zero when differences have been found. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000851 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +0000852 diff_a_works == FALSE ? "" : "-a ",
853#if defined(MSWIN) || defined(MSDOS)
854 diff_bin_works == TRUE ? "--binary " : "",
855#else
856 "",
857#endif
858 (diff_flags & DIFF_IWHITE) ? "-b " : "",
859 (diff_flags & DIFF_ICASE) ? "-i " : "",
860 tmp_orig, tmp_new);
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000861 append_redir(cmd, (int)len, p_srr, tmp_diff);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000862#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000863 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000864#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000866#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000867 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000868#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000869 vim_free(cmd);
870 }
871 }
872}
873
874/*
875 * Create a new version of a file from the current buffer and a diff file.
876 * The buffer is written to a file, also for unmodified buffers (the file
877 * could have been produced by autocommands, e.g. the netrw plugin).
878 */
879 void
880ex_diffpatch(eap)
881 exarg_T *eap;
882{
883 char_u *tmp_orig; /* name of original temp file */
884 char_u *tmp_new; /* name of patched temp file */
885 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000886 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000887 win_T *old_curwin = curwin;
888 char_u *newname = NULL; /* name of patched file buffer */
889#ifdef UNIX
890 char_u dirbuf[MAXPATHL];
891 char_u *fullname = NULL;
892#endif
893#ifdef FEAT_BROWSE
894 char_u *browseFile = NULL;
895 int browse_flag = cmdmod.browse;
896#endif
897
898#ifdef FEAT_BROWSE
899 if (cmdmod.browse)
900 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +0000901 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaar071d4272004-06-13 20:20:40 +0000902 eap->arg, NULL, NULL, BROWSE_FILTER_ALL_FILES, NULL);
903 if (browseFile == NULL)
904 return; /* operation cancelled */
905 eap->arg = browseFile;
906 cmdmod.browse = FALSE; /* don't let do_ecmd() browse again */
907 }
908#endif
909
910 /* We need two temp file names. */
911 tmp_orig = vim_tempname('o');
912 tmp_new = vim_tempname('n');
913 if (tmp_orig == NULL || tmp_new == NULL)
914 goto theend;
915
916 /* Write the current buffer to "tmp_orig". */
917 if (buf_write(curbuf, tmp_orig, NULL,
918 (linenr_T)1, curbuf->b_ml.ml_line_count,
919 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
920 goto theend;
921
922#ifdef UNIX
923 /* Get the absolute path of the patchfile, changing directory below. */
924 fullname = FullName_save(eap->arg, FALSE);
925#endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000926 buflen = STRLEN(tmp_orig) + (
Bram Moolenaar071d4272004-06-13 20:20:40 +0000927# ifdef UNIX
928 fullname != NULL ? STRLEN(fullname) :
929# endif
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000930 STRLEN(eap->arg)) + STRLEN(tmp_new) + 16;
931 buf = alloc((unsigned)buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000932 if (buf == NULL)
933 goto theend;
934
935#ifdef UNIX
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000936 /* Temporarily chdir to /tmp, to avoid patching files in the current
Bram Moolenaar071d4272004-06-13 20:20:40 +0000937 * directory when the patch file contains more than one patch. When we
938 * have our own temp dir use that instead, it will be cleaned up when we
939 * exit (any .rej files created). Don't change directory if we can't
940 * return to the current. */
941 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
942 dirbuf[0] = NUL;
943 else
944 {
945# ifdef TEMPDIRNAMES
946 if (vim_tempdir != NULL)
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000947 ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000948 else
949# endif
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +0000950 ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +0000951 shorten_fnames(TRUE);
952 }
953#endif
954
955#ifdef FEAT_EVAL
956 if (*p_pex != NUL)
957 /* Use 'patchexpr' to generate the new file. */
958 eval_patch(tmp_orig,
959# ifdef UNIX
960 fullname != NULL ? fullname :
961# endif
962 eap->arg, tmp_new);
963 else
964#endif
965 {
966 /* Build the patch command and execute it. Ignore errors. Switch to
967 * cooked mode to allow the user to respond to prompts. */
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +0000968 vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"",
969 tmp_new, tmp_orig,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000970# ifdef UNIX
971 fullname != NULL ? fullname :
972# endif
973 eap->arg);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000974#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000975 block_autocmds(); /* Avoid ShellCmdPost stuff */
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000976#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000977 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000978#ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000979 unblock_autocmds();
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000980#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000981 }
982
983#ifdef UNIX
984 if (dirbuf[0] != NUL)
985 {
986 if (mch_chdir((char *)dirbuf) != 0)
987 EMSG(_(e_prev_dir));
988 shorten_fnames(TRUE);
989 }
990#endif
991
992 /* patch probably has written over the screen */
993 redraw_later(CLEAR);
994
995 /* Delete any .orig or .rej file created. */
996 STRCPY(buf, tmp_new);
997 STRCAT(buf, ".orig");
998 mch_remove(buf);
999 STRCPY(buf, tmp_new);
1000 STRCAT(buf, ".rej");
1001 mch_remove(buf);
1002
1003 if (curbuf->b_fname != NULL)
1004 {
1005 newname = vim_strnsave(curbuf->b_fname,
1006 (int)(STRLEN(curbuf->b_fname) + 4));
1007 if (newname != NULL)
1008 STRCAT(newname, ".new");
1009 }
1010
1011#ifdef FEAT_GUI
1012 need_mouse_correct = TRUE;
1013#endif
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001014 /* don't use a new tab page, each tab page has its own diffs */
1015 cmdmod.tab = 0;
1016
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001017 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001018 {
1019 /* Pretend it was a ":split fname" command */
1020 eap->cmdidx = CMD_split;
1021 eap->arg = tmp_new;
1022 do_exedit(eap, old_curwin);
1023
1024 if (curwin != old_curwin) /* split must have worked */
1025 {
1026 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1027 diff_win_options(curwin, TRUE);
1028 diff_win_options(old_curwin, TRUE);
1029
1030 if (newname != NULL)
1031 {
1032 /* do a ":file filename.new" on the patched buffer */
1033 eap->arg = newname;
1034 ex_file(eap);
1035
1036#ifdef FEAT_AUTOCMD
1037 /* Do filetype detection with the new name. */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001038 if (au_has_group((char_u *)"filetypedetect"))
1039 do_cmdline_cmd((char_u *)":doau filetypedetect BufRead");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001040#endif
1041 }
1042 }
1043 }
1044
1045theend:
1046 if (tmp_orig != NULL)
1047 mch_remove(tmp_orig);
1048 vim_free(tmp_orig);
1049 if (tmp_new != NULL)
1050 mch_remove(tmp_new);
1051 vim_free(tmp_new);
1052 vim_free(newname);
1053 vim_free(buf);
1054#ifdef UNIX
1055 vim_free(fullname);
1056#endif
1057#ifdef FEAT_BROWSE
1058 vim_free(browseFile);
1059 cmdmod.browse = browse_flag;
1060#endif
1061}
1062
1063/*
1064 * Split the window and edit another file, setting options to show the diffs.
1065 */
1066 void
1067ex_diffsplit(eap)
1068 exarg_T *eap;
1069{
1070 win_T *old_curwin = curwin;
1071
1072#ifdef FEAT_GUI
1073 need_mouse_correct = TRUE;
1074#endif
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001075 /* don't use a new tab page, each tab page has its own diffs */
1076 cmdmod.tab = 0;
1077
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001078 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001079 {
1080 /* Pretend it was a ":split fname" command */
1081 eap->cmdidx = CMD_split;
Bram Moolenaar4be06f92005-07-29 22:36:03 +00001082 curwin->w_p_diff = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001083 do_exedit(eap, old_curwin);
1084
1085 if (curwin != old_curwin) /* split must have worked */
1086 {
1087 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1088 diff_win_options(curwin, TRUE);
1089 diff_win_options(old_curwin, TRUE);
1090 }
1091 }
1092}
1093
1094/*
1095 * Set options to show difs for the current window.
1096 */
1097/*ARGSUSED*/
1098 void
1099ex_diffthis(eap)
1100 exarg_T *eap;
1101{
1102 /* Set 'diff', 'scrollbind' on and 'wrap' off. */
1103 diff_win_options(curwin, TRUE);
1104}
1105
1106/*
1107 * Set options in window "wp" for diff mode.
1108 */
1109 void
1110diff_win_options(wp, addbuf)
1111 win_T *wp;
1112 int addbuf; /* Add buffer to diff. */
1113{
1114 wp->w_p_diff = TRUE;
1115 wp->w_p_scb = TRUE;
1116 wp->w_p_wrap = FALSE;
1117# ifdef FEAT_FOLDING
1118 {
1119 win_T *old_curwin = curwin;
1120
1121 curwin = wp;
1122 curbuf = curwin->w_buffer;
1123 set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001124 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001125 curwin = old_curwin;
1126 curbuf = curwin->w_buffer;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001127 wp->w_p_fdc = diff_foldcolumn;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001128 wp->w_p_fen = TRUE;
1129 wp->w_p_fdl = 0;
1130 foldUpdateAll(wp);
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001131 /* make sure topline is not halfway a fold */
1132 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001133 }
1134# endif
1135#ifdef FEAT_SCROLLBIND
1136 if (vim_strchr(p_sbo, 'h') == NULL)
1137 do_cmdline_cmd((char_u *)"set sbo+=hor");
1138#endif
1139
1140 if (addbuf)
1141 diff_buf_add(wp->w_buffer);
1142 redraw_win_later(wp, NOT_VALID);
1143}
1144
1145/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001146 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001147 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001148 */
1149 void
1150ex_diffoff(eap)
1151 exarg_T *eap;
1152{
1153 win_T *wp;
1154 win_T *old_curwin = curwin;
1155#ifdef FEAT_SCROLLBIND
1156 int diffwin = FALSE;
1157#endif
1158
1159 for (wp = firstwin; wp != NULL; wp = wp->w_next)
1160 {
Bram Moolenaar21fa1ed2009-03-11 11:47:18 +00001161 if (wp == curwin || (eap->forceit && wp->w_p_diff))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001162 {
1163 /* Set 'diff', 'scrollbind' off and 'wrap' on. */
1164 wp->w_p_diff = FALSE;
1165 wp->w_p_scb = FALSE;
1166 wp->w_p_wrap = TRUE;
1167#ifdef FEAT_FOLDING
1168 curwin = wp;
1169 curbuf = curwin->w_buffer;
1170 set_string_option_direct((char_u *)"fdm", -1,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001171 (char_u *)"manual", OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001172 curwin = old_curwin;
1173 curbuf = curwin->w_buffer;
1174 wp->w_p_fdc = 0;
1175 wp->w_p_fen = FALSE;
1176 wp->w_p_fdl = 0;
1177 foldUpdateAll(wp);
1178 /* make sure topline is not halfway a fold */
1179 changed_window_setting_win(wp);
1180#endif
1181 diff_buf_adjust(wp);
1182 }
1183#ifdef FEAT_SCROLLBIND
1184 diffwin |= wp->w_p_diff;
1185#endif
1186 }
1187
1188#ifdef FEAT_SCROLLBIND
1189 /* Remove "hor" from from 'scrollopt' if there are no diff windows left. */
1190 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1191 do_cmdline_cmd((char_u *)"set sbo-=hor");
1192#endif
1193}
1194
1195/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001196 * Read the diff output and add each entry to the diff list.
1197 */
1198 static void
1199diff_read(idx_orig, idx_new, fname)
1200 int idx_orig; /* idx of original file */
1201 int idx_new; /* idx of new file */
1202 char_u *fname; /* name of diff output file */
1203{
1204 FILE *fd;
1205 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001206 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001207 diff_T *dn, *dpl;
1208 long f1, l1, f2, l2;
1209 char_u linebuf[LBUFLEN]; /* only need to hold the diff line */
1210 int difftype;
1211 char_u *p;
1212 long off;
1213 int i;
1214 linenr_T lnum_orig, lnum_new;
1215 long count_orig, count_new;
1216 int notset = TRUE; /* block "*dp" not set yet */
1217
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001218 fd = mch_fopen((char *)fname, "r");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001219 if (fd == NULL)
1220 {
1221 EMSG(_("E98: Cannot read diff output"));
1222 return;
1223 }
1224
1225 for (;;)
1226 {
1227 if (tag_fgets(linebuf, LBUFLEN, fd))
1228 break; /* end of file */
1229 if (!isdigit(*linebuf))
1230 continue; /* not the start of a diff block */
1231
1232 /* This line must be one of three formats:
1233 * {first}[,{last}]c{first}[,{last}]
1234 * {first}a{first}[,{last}]
1235 * {first}[,{last}]d{first}
1236 */
1237 p = linebuf;
1238 f1 = getdigits(&p);
1239 if (*p == ',')
1240 {
1241 ++p;
1242 l1 = getdigits(&p);
1243 }
1244 else
1245 l1 = f1;
1246 if (*p != 'a' && *p != 'c' && *p != 'd')
1247 continue; /* invalid diff format */
1248 difftype = *p++;
1249 f2 = getdigits(&p);
1250 if (*p == ',')
1251 {
1252 ++p;
1253 l2 = getdigits(&p);
1254 }
1255 else
1256 l2 = f2;
1257 if (l1 < f1 || l2 < f2)
1258 continue; /* invalid line range */
1259
1260 if (difftype == 'a')
1261 {
1262 lnum_orig = f1 + 1;
1263 count_orig = 0;
1264 }
1265 else
1266 {
1267 lnum_orig = f1;
1268 count_orig = l1 - f1 + 1;
1269 }
1270 if (difftype == 'd')
1271 {
1272 lnum_new = f2 + 1;
1273 count_new = 0;
1274 }
1275 else
1276 {
1277 lnum_new = f2;
1278 count_new = l2 - f2 + 1;
1279 }
1280
1281 /* Go over blocks before the change, for which orig and new are equal.
1282 * Copy blocks from orig to new. */
1283 while (dp != NULL
1284 && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig])
1285 {
1286 if (notset)
1287 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1288 dprev = dp;
1289 dp = dp->df_next;
1290 notset = TRUE;
1291 }
1292
1293 if (dp != NULL
1294 && lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig]
1295 && lnum_orig + count_orig >= dp->df_lnum[idx_orig])
1296 {
1297 /* New block overlaps with existing block(s).
1298 * First find last block that overlaps. */
1299 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
1300 if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig])
1301 break;
1302
1303 /* If the newly found block starts before the old one, set the
1304 * start back a number of lines. */
1305 off = dp->df_lnum[idx_orig] - lnum_orig;
1306 if (off > 0)
1307 {
1308 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001309 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001310 dp->df_lnum[i] -= off;
1311 dp->df_lnum[idx_new] = lnum_new;
1312 dp->df_count[idx_new] = count_new;
1313 }
1314 else if (notset)
1315 {
1316 /* new block inside existing one, adjust new block */
1317 dp->df_lnum[idx_new] = lnum_new + off;
1318 dp->df_count[idx_new] = count_new - off;
1319 }
1320 else
1321 /* second overlap of new block with existing block */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001322 dp->df_count[idx_new] += count_new - count_orig
1323 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1324 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001325
1326 /* Adjust the size of the block to include all the lines to the
1327 * end of the existing block or the new diff, whatever ends last. */
1328 off = (lnum_orig + count_orig)
1329 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1330 if (off < 0)
1331 {
1332 /* new change ends in existing block, adjust the end if not
1333 * done already */
1334 if (notset)
1335 dp->df_count[idx_new] += -off;
1336 off = 0;
1337 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001338 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001339 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001340 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1341 - dp->df_lnum[i] + off;
1342
1343 /* Delete the diff blocks that have been merged into one. */
1344 dn = dp->df_next;
1345 dp->df_next = dpl->df_next;
1346 while (dn != dp->df_next)
1347 {
1348 dpl = dn->df_next;
1349 vim_free(dn);
1350 dn = dpl;
1351 }
1352 }
1353 else
1354 {
1355 /* Allocate a new diffblock. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001356 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001357 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001358 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001359
1360 dp->df_lnum[idx_orig] = lnum_orig;
1361 dp->df_count[idx_orig] = count_orig;
1362 dp->df_lnum[idx_new] = lnum_new;
1363 dp->df_count[idx_new] = count_new;
1364
1365 /* Set values for other buffers, these must be equal to the
1366 * original buffer, otherwise there would have been a change
1367 * already. */
1368 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001369 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001370 diff_copy_entry(dprev, dp, idx_orig, i);
1371 }
1372 notset = FALSE; /* "*dp" has been set */
1373 }
1374
1375 /* for remaining diff blocks orig and new are equal */
1376 while (dp != NULL)
1377 {
1378 if (notset)
1379 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1380 dprev = dp;
1381 dp = dp->df_next;
1382 notset = TRUE;
1383 }
1384
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001385done:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001386 fclose(fd);
1387}
1388
1389/*
1390 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1391 */
1392 static void
1393diff_copy_entry(dprev, dp, idx_orig, idx_new)
1394 diff_T *dprev;
1395 diff_T *dp;
1396 int idx_orig;
1397 int idx_new;
1398{
1399 long off;
1400
1401 if (dprev == NULL)
1402 off = 0;
1403 else
1404 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1405 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1406 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1407 dp->df_count[idx_new] = dp->df_count[idx_orig];
1408}
1409
1410/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001411 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001412 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001413 void
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001414diff_clear(tp)
1415 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001416{
1417 diff_T *p, *next_p;
1418
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001419 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001420 {
1421 next_p = p->df_next;
1422 vim_free(p);
1423 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001424 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001425}
1426
1427/*
1428 * Check diff status for line "lnum" in buffer "buf":
1429 * Returns 0 for nothing special
1430 * Returns -1 for a line that should be highlighted as changed.
1431 * Returns -2 for a line that should be highlighted as added/deleted.
1432 * Returns > 0 for inserting that many filler lines above it (never happens
1433 * when 'diffopt' doesn't contain "filler").
1434 * This should only be used for windows where 'diff' is set.
1435 */
1436 int
1437diff_check(wp, lnum)
1438 win_T *wp;
1439 linenr_T lnum;
1440{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001441 int idx; /* index in tp_diffbuf[] for this buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001442 diff_T *dp;
1443 int maxcount;
1444 int i;
1445 buf_T *buf = wp->w_buffer;
1446 int cmp;
1447
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001448 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001449 ex_diffupdate(NULL); /* update after a big change */
1450
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001451 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) /* no diffs at all */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001452 return 0;
1453
1454 /* safety check: "lnum" must be a buffer line */
1455 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1456 return 0;
1457
1458 idx = diff_buf_idx(buf);
1459 if (idx == DB_COUNT)
1460 return 0; /* no diffs for buffer "buf" */
1461
1462#ifdef FEAT_FOLDING
1463 /* A closed fold never has filler lines. */
1464 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1465 return 0;
1466#endif
1467
1468 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001469 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001470 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1471 break;
1472 if (dp == NULL || lnum < dp->df_lnum[idx])
1473 return 0;
1474
1475 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1476 {
1477 int zero = FALSE;
1478
1479 /* Changed or inserted line. If the other buffers have a count of
1480 * zero, the lines were inserted. If the other buffers have the same
1481 * count, check if the lines are identical. */
1482 cmp = FALSE;
1483 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001484 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001485 {
1486 if (dp->df_count[i] == 0)
1487 zero = TRUE;
1488 else
1489 {
1490 if (dp->df_count[i] != dp->df_count[idx])
1491 return -1; /* nr of lines changed. */
1492 cmp = TRUE;
1493 }
1494 }
1495 if (cmp)
1496 {
1497 /* Compare all lines. If they are equal the lines were inserted
1498 * in some buffers, deleted in others, but not changed. */
1499 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001500 if (i != idx && curtab->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001501 if (!diff_equal_entry(dp, idx, i))
1502 return -1;
1503 }
1504 /* If there is no buffer with zero lines then there is no difference
1505 * any longer. Happens when making a change (or undo) that removes
1506 * the difference. Can't remove the entry here, we might be halfway
1507 * updating the window. Just report the text as unchanged. Other
1508 * windows might still show the change though. */
1509 if (zero == FALSE)
1510 return 0;
1511 return -2;
1512 }
1513
1514 /* If 'diffopt' doesn't contain "filler", return 0. */
1515 if (!(diff_flags & DIFF_FILLER))
1516 return 0;
1517
1518 /* Insert filler lines above the line just below the change. Will return
1519 * 0 when this buf had the max count. */
1520 maxcount = 0;
1521 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001522 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001523 maxcount = dp->df_count[i];
1524 return maxcount - dp->df_count[idx];
1525}
1526
1527/*
1528 * Compare two entries in diff "*dp" and return TRUE if they are equal.
1529 */
1530 static int
1531diff_equal_entry(dp, idx1, idx2)
1532 diff_T *dp;
1533 int idx1;
1534 int idx2;
1535{
1536 int i;
1537 char_u *line;
1538 int cmp;
1539
1540 if (dp->df_count[idx1] != dp->df_count[idx2])
1541 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001542 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001543 return FALSE;
1544 for (i = 0; i < dp->df_count[idx1]; ++i)
1545 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001546 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001547 dp->df_lnum[idx1] + i, FALSE));
1548 if (line == NULL)
1549 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001550 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00001551 dp->df_lnum[idx2] + i, FALSE));
1552 vim_free(line);
1553 if (cmp != 0)
1554 return FALSE;
1555 }
1556 return TRUE;
1557}
1558
1559/*
1560 * Compare strings "s1" and "s2" according to 'diffopt'.
1561 * Return non-zero when they are different.
1562 */
1563 static int
1564diff_cmp(s1, s2)
1565 char_u *s1;
1566 char_u *s2;
1567{
1568 char_u *p1, *p2;
1569#ifdef FEAT_MBYTE
1570 int l;
1571#endif
1572
1573 if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0)
1574 return STRCMP(s1, s2);
1575 if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE))
1576 return MB_STRICMP(s1, s2);
1577
1578 /* Ignore white space changes and possibly ignore case. */
1579 p1 = s1;
1580 p2 = s2;
1581 while (*p1 != NUL && *p2 != NUL)
1582 {
1583 if (vim_iswhite(*p1) && vim_iswhite(*p2))
1584 {
1585 p1 = skipwhite(p1);
1586 p2 = skipwhite(p2);
1587 }
1588 else
1589 {
1590#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001591 l = (*mb_ptr2len)(p1);
1592 if (l != (*mb_ptr2len)(p2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001593 break;
1594 if (l > 1)
1595 {
1596 if (STRNCMP(p1, p2, l) != 0
1597 && (!enc_utf8
1598 || !(diff_flags & DIFF_ICASE)
1599 || utf_fold(utf_ptr2char(p1))
1600 != utf_fold(utf_ptr2char(p2))))
1601 break;
1602 p1 += l;
1603 p2 += l;
1604 }
1605 else
1606#endif
1607 {
1608 if (*p1 != *p2 && (!(diff_flags & DIFF_ICASE)
1609 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
1610 break;
1611 ++p1;
1612 ++p2;
1613 }
1614 }
1615 }
1616
1617 /* Ignore trailing white space. */
1618 p1 = skipwhite(p1);
1619 p2 = skipwhite(p2);
1620 if (*p1 != NUL || *p2 != NUL)
1621 return 1;
1622 return 0;
1623}
1624
1625/*
1626 * Return the number of filler lines above "lnum".
1627 */
1628 int
1629diff_check_fill(wp, lnum)
1630 win_T *wp;
1631 linenr_T lnum;
1632{
1633 int n;
1634
1635 /* be quick when there are no filler lines */
1636 if (!(diff_flags & DIFF_FILLER))
1637 return 0;
1638 n = diff_check(wp, lnum);
1639 if (n <= 0)
1640 return 0;
1641 return n;
1642}
1643
1644/*
1645 * Set the topline of "towin" to match the position in "fromwin", so that they
1646 * show the same diff'ed lines.
1647 */
1648 void
1649diff_set_topline(fromwin, towin)
1650 win_T *fromwin;
1651 win_T *towin;
1652{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001653 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001654 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001655 int fromidx;
1656 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001657 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001658 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001659 int i;
1660
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001661 fromidx = diff_buf_idx(frombuf);
1662 if (fromidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001663 return; /* safety check */
1664
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001665 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001666 ex_diffupdate(NULL); /* update after a big change */
1667
1668 towin->w_topfill = 0;
1669
1670 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001671 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001672 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001673 break;
1674 if (dp == NULL)
1675 {
1676 /* After last change, compute topline relative to end of file; no
1677 * filler lines. */
1678 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001679 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001680 }
1681 else
1682 {
1683 /* Find index for "towin". */
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001684 toidx = diff_buf_idx(towin->w_buffer);
1685 if (toidx == DB_COUNT)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001686 return; /* safety check */
1687
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001688 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
1689 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001690 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001691 /* Inside a change: compute filler lines. With three or more
1692 * buffers we need to know the largest count. */
1693 max_count = 0;
1694 for (i = 0; i < DB_COUNT; ++i)
1695 if (curtab->tp_diffbuf[i] != NULL
1696 && max_count < dp->df_count[i])
1697 max_count = dp->df_count[i];
1698
1699 if (dp->df_count[toidx] == dp->df_count[fromidx])
1700 {
1701 /* same number of lines: use same filler count */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001702 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001703 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001704 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001705 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001706 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001707 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001708 /* more lines in towin and fromwin doesn't show diff
1709 * lines, only filler lines */
1710 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
1711 {
1712 /* towin also only shows filler lines */
1713 towin->w_topline = dp->df_lnum[toidx]
1714 + dp->df_count[toidx];
1715 towin->w_topfill = fromwin->w_topfill;
1716 }
1717 else
1718 /* towin still has some diff lines to show */
1719 towin->w_topline = dp->df_lnum[toidx]
1720 + max_count - fromwin->w_topfill;
1721 }
1722 }
1723 else if (towin->w_topline >= dp->df_lnum[toidx]
1724 + dp->df_count[toidx])
1725 {
1726 /* less lines in towin and no diff lines to show: compute
1727 * filler lines */
1728 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
1729 if (diff_flags & DIFF_FILLER)
1730 {
1731 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
1732 /* fromwin is also out of diff lines */
1733 towin->w_topfill = fromwin->w_topfill;
1734 else
1735 /* fromwin has some diff lines */
1736 towin->w_topfill = dp->df_lnum[fromidx]
1737 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001738 }
1739 }
1740 }
1741 }
1742
1743 /* safety check (if diff info gets outdated strange things may happen) */
1744 towin->w_botfill = FALSE;
1745 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
1746 {
1747 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
1748 towin->w_botfill = TRUE;
1749 }
1750 if (towin->w_topline < 1)
1751 {
1752 towin->w_topline = 1;
1753 towin->w_topfill = 0;
1754 }
1755
1756 /* When w_topline changes need to recompute w_botline and cursor position */
1757 invalidate_botline_win(towin);
1758 changed_line_abv_curs_win(towin);
1759
1760 check_topfill(towin, FALSE);
1761#ifdef FEAT_FOLDING
1762 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
1763 NULL, TRUE, NULL);
1764#endif
1765}
1766
1767/*
1768 * This is called when 'diffopt' is changed.
1769 */
1770 int
1771diffopt_changed()
1772{
1773 char_u *p;
1774 int diff_context_new = 6;
1775 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001776 int diff_foldcolumn_new = 2;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001777 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001778
1779 p = p_dip;
1780 while (*p != NUL)
1781 {
1782 if (STRNCMP(p, "filler", 6) == 0)
1783 {
1784 p += 6;
1785 diff_flags_new |= DIFF_FILLER;
1786 }
1787 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
1788 {
1789 p += 8;
1790 diff_context_new = getdigits(&p);
1791 }
1792 else if (STRNCMP(p, "icase", 5) == 0)
1793 {
1794 p += 5;
1795 diff_flags_new |= DIFF_ICASE;
1796 }
1797 else if (STRNCMP(p, "iwhite", 6) == 0)
1798 {
1799 p += 6;
1800 diff_flags_new |= DIFF_IWHITE;
1801 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001802 else if (STRNCMP(p, "horizontal", 10) == 0)
1803 {
1804 p += 10;
1805 diff_flags_new |= DIFF_HORIZONTAL;
1806 }
1807 else if (STRNCMP(p, "vertical", 8) == 0)
1808 {
1809 p += 8;
1810 diff_flags_new |= DIFF_VERTICAL;
1811 }
1812 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
1813 {
1814 p += 11;
1815 diff_foldcolumn_new = getdigits(&p);
1816 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001817 if (*p != ',' && *p != NUL)
1818 return FAIL;
1819 if (*p == ',')
1820 ++p;
1821 }
1822
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001823 /* Can't have both "horizontal" and "vertical". */
1824 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
1825 return FAIL;
1826
Bram Moolenaar071d4272004-06-13 20:20:40 +00001827 /* If "icase" or "iwhite" was added or removed, need to update the diff. */
1828 if (diff_flags != diff_flags_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001829 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
1830 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001831
1832 diff_flags = diff_flags_new;
1833 diff_context = diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001834 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001835
1836 diff_redraw(TRUE);
1837
1838 /* recompute the scroll binding with the new option value, may
1839 * remove or add filler lines */
1840 check_scrollbind((linenr_T)0, 0L);
1841
1842 return OK;
1843}
1844
1845/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00001846 * Return TRUE if 'diffopt' contains "horizontal".
1847 */
1848 int
1849diffopt_horizontal()
1850{
1851 return (diff_flags & DIFF_HORIZONTAL) != 0;
1852}
1853
1854/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001855 * Find the difference within a changed line.
1856 * Returns TRUE if the line was added, no other buffer has it.
1857 */
1858 int
1859diff_find_change(wp, lnum, startp, endp)
1860 win_T *wp;
1861 linenr_T lnum;
1862 int *startp; /* first char of the change */
1863 int *endp; /* last char of the change */
1864{
1865 char_u *line_org;
1866 char_u *line_new;
1867 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001868 int si_org, si_new;
1869 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001870 diff_T *dp;
1871 int idx;
1872 int off;
1873 int added = TRUE;
1874
1875 /* Make a copy of the line, the next ml_get() will invalidate it. */
1876 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
1877 if (line_org == NULL)
1878 return FALSE;
1879
1880 idx = diff_buf_idx(wp->w_buffer);
1881 if (idx == DB_COUNT) /* cannot happen */
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001882 {
1883 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001884 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001885 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001886
1887 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001888 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001889 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1890 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001891 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001892 {
1893 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001894 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00001895 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001896
1897 off = lnum - dp->df_lnum[idx];
1898
1899 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001900 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901 {
1902 /* Skip lines that are not in the other change (filler lines). */
1903 if (off >= dp->df_count[i])
1904 continue;
1905 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001906 line_new = ml_get_buf(curtab->tp_diffbuf[i],
1907 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908
1909 /* Search for start of difference */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001910 si_org = si_new = 0;
1911 while (line_org[si_org] != NUL)
1912 {
1913 if ((diff_flags & DIFF_IWHITE)
1914 && vim_iswhite(line_org[si_org])
1915 && vim_iswhite(line_new[si_new]))
1916 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001917 si_org = (int)(skipwhite(line_org + si_org) - line_org);
1918 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001919 }
1920 else
1921 {
1922 if (line_org[si_org] != line_new[si_new])
1923 break;
1924 ++si_org;
1925 ++si_new;
1926 }
1927 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001928#ifdef FEAT_MBYTE
1929 if (has_mbyte)
1930 {
1931 /* Move back to first byte of character in both lines (may
1932 * have "nn^" in line_org and "n^ in line_new). */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001933 si_org -= (*mb_head_off)(line_org, line_org + si_org);
1934 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001935 }
1936#endif
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001937 if (*startp > si_org)
1938 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001939
1940 /* Search for end of difference, if any. */
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001941 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001942 {
1943 ei_org = (int)STRLEN(line_org);
1944 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001945 while (ei_org >= *startp && ei_new >= si_new
1946 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947 {
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00001948 if ((diff_flags & DIFF_IWHITE)
1949 && vim_iswhite(line_org[ei_org])
1950 && vim_iswhite(line_new[ei_new]))
1951 {
1952 while (ei_org >= *startp
1953 && vim_iswhite(line_org[ei_org]))
1954 --ei_org;
1955 while (ei_new >= si_new
1956 && vim_iswhite(line_new[ei_new]))
1957 --ei_new;
1958 }
1959 else
1960 {
1961 if (line_org[ei_org] != line_new[ei_new])
1962 break;
1963 --ei_org;
1964 --ei_new;
1965 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001966 }
1967 if (*endp < ei_org)
1968 *endp = ei_org;
1969 }
1970 }
1971
1972 vim_free(line_org);
1973 return added;
1974}
1975
1976#if defined(FEAT_FOLDING) || defined(PROTO)
1977/*
1978 * Return TRUE if line "lnum" is not close to a diff block, this line should
1979 * be in a fold.
1980 * Return FALSE if there are no diff blocks at all in this window.
1981 */
1982 int
1983diff_infold(wp, lnum)
1984 win_T *wp;
1985 linenr_T lnum;
1986{
1987 int i;
1988 int idx = -1;
1989 int other = FALSE;
1990 diff_T *dp;
1991
1992 /* Return if 'diff' isn't set. */
1993 if (!wp->w_p_diff)
1994 return FALSE;
1995
1996 for (i = 0; i < DB_COUNT; ++i)
1997 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001998 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001999 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002000 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002001 other = TRUE;
2002 }
2003
2004 /* return here if there are no diffs in the window */
2005 if (idx == -1 || !other)
2006 return FALSE;
2007
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002008 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002009 ex_diffupdate(NULL); /* update after a big change */
2010
2011 /* Return if there are no diff blocks. All lines will be folded. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002012 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013 return TRUE;
2014
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002015 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002016 {
2017 /* If this change is below the line there can't be any further match. */
2018 if (dp->df_lnum[idx] - diff_context > lnum)
2019 break;
2020 /* If this change ends before the line we have a match. */
2021 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2022 return FALSE;
2023 }
2024 return TRUE;
2025}
2026#endif
2027
2028/*
2029 * "dp" and "do" commands.
2030 */
2031 void
2032nv_diffgetput(put)
2033 int put;
2034{
2035 exarg_T ea;
2036
2037 ea.arg = (char_u *)"";
2038 if (put)
2039 ea.cmdidx = CMD_diffput;
2040 else
2041 ea.cmdidx = CMD_diffget;
2042 ea.addr_count = 0;
2043 ea.line1 = curwin->w_cursor.lnum;
2044 ea.line2 = curwin->w_cursor.lnum;
2045 ex_diffgetput(&ea);
2046}
2047
2048/*
2049 * ":diffget"
2050 * ":diffput"
2051 */
2052 void
2053ex_diffgetput(eap)
2054 exarg_T *eap;
2055{
2056 linenr_T lnum;
2057 int count;
2058 linenr_T off = 0;
2059 diff_T *dp;
2060 diff_T *dprev;
2061 diff_T *dfree;
2062 int idx_cur;
2063 int idx_other;
2064 int idx_from;
2065 int idx_to;
2066 int i;
2067 int added;
2068 char_u *p;
2069 aco_save_T aco;
2070 buf_T *buf;
2071 int start_skip, end_skip;
2072 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002073 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002074 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002075
2076 /* Find the current buffer in the list of diff buffers. */
2077 idx_cur = diff_buf_idx(curbuf);
2078 if (idx_cur == DB_COUNT)
2079 {
2080 EMSG(_("E99: Current buffer is not in diff mode"));
2081 return;
2082 }
2083
2084 if (*eap->arg == NUL)
2085 {
2086 /* No argument: Find the other buffer in the list of diff buffers. */
2087 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002088 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002089 && curtab->tp_diffbuf[idx_other] != NULL)
2090 {
2091 if (eap->cmdidx != CMD_diffput
2092 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2093 break;
2094 found_not_ma = TRUE;
2095 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096 if (idx_other == DB_COUNT)
2097 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002098 if (found_not_ma)
2099 EMSG(_("E793: No other buffer in diff mode is modifiable"));
2100 else
2101 EMSG(_("E100: No other buffer in diff mode"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102 return;
2103 }
2104
2105 /* Check that there isn't a third buffer in the list */
2106 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002107 if (curtab->tp_diffbuf[i] != curbuf
2108 && curtab->tp_diffbuf[i] != NULL
2109 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002110 {
2111 EMSG(_("E101: More than two buffers in diff mode, don't know which one to use"));
2112 return;
2113 }
2114 }
2115 else
2116 {
2117 /* Buffer number or pattern given. Ignore trailing white space. */
2118 p = eap->arg + STRLEN(eap->arg);
2119 while (p > eap->arg && vim_iswhite(p[-1]))
2120 --p;
2121 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2122 ;
2123 if (eap->arg + i == p) /* digits only */
2124 i = atol((char *)eap->arg);
2125 else
2126 {
2127 i = buflist_findpat(eap->arg, p, FALSE, TRUE);
2128 if (i < 0)
2129 return; /* error message already given */
2130 }
2131 buf = buflist_findnr(i);
2132 if (buf == NULL)
2133 {
2134 EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg);
2135 return;
2136 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002137 if (buf == curbuf)
2138 return; /* nothing to do */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002139 idx_other = diff_buf_idx(buf);
2140 if (idx_other == DB_COUNT)
2141 {
2142 EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg);
2143 return;
2144 }
2145 }
2146
2147 diff_busy = TRUE;
2148
2149 /* When no range given include the line above or below the cursor. */
2150 if (eap->addr_count == 0)
2151 {
2152 /* Make it possible that ":diffget" on the last line gets line below
2153 * the cursor line when there is no difference above the cursor. */
2154 if (eap->cmdidx == CMD_diffget
2155 && eap->line1 == curbuf->b_ml.ml_line_count
2156 && diff_check(curwin, eap->line1) == 0
2157 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2158 ++eap->line2;
2159 else if (eap->line1 > 0)
2160 --eap->line1;
2161 }
2162
2163 if (eap->cmdidx == CMD_diffget)
2164 {
2165 idx_from = idx_other;
2166 idx_to = idx_cur;
2167 }
2168 else
2169 {
2170 idx_from = idx_cur;
2171 idx_to = idx_other;
2172 /* Need to make the other buffer the current buffer to be able to make
2173 * changes in it. */
2174 /* set curwin/curbuf to buf and save a few things */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002175 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002176 }
2177
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002178 /* May give the warning for a changed buffer here, which can trigger the
2179 * FileChangedRO autocommand, which may do nasty things and mess
2180 * everything up. */
2181 if (!curbuf->b_changed)
2182 {
2183 change_warning(0);
2184 if (diff_buf_idx(curbuf) != idx_to)
2185 {
2186 EMSG(_("E787: Buffer changed unexpectedly"));
2187 return;
2188 }
2189 }
2190
Bram Moolenaar071d4272004-06-13 20:20:40 +00002191 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002192 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002193 {
2194 if (dp->df_lnum[idx_cur] > eap->line2 + off)
2195 break; /* past the range that was specified */
2196
2197 dfree = NULL;
2198 lnum = dp->df_lnum[idx_to];
2199 count = dp->df_count[idx_to];
2200 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2201 && u_save(lnum - 1, lnum + count) != FAIL)
2202 {
2203 /* Inside the specified range and saving for undo worked. */
2204 start_skip = 0;
2205 end_skip = 0;
2206 if (eap->addr_count > 0)
2207 {
2208 /* A range was specified: check if lines need to be skipped. */
2209 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2210 if (start_skip > 0)
2211 {
2212 /* range starts below start of current diff block */
2213 if (start_skip > count)
2214 {
2215 lnum += count;
2216 count = 0;
2217 }
2218 else
2219 {
2220 count -= start_skip;
2221 lnum += start_skip;
2222 }
2223 }
2224 else
2225 start_skip = 0;
2226
2227 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2228 - (eap->line2 + off);
2229 if (end_skip > 0)
2230 {
2231 /* range ends above end of current/from diff block */
2232 if (idx_cur == idx_from) /* :diffput */
2233 {
2234 i = dp->df_count[idx_cur] - start_skip - end_skip;
2235 if (count > i)
2236 count = i;
2237 }
2238 else /* :diffget */
2239 {
2240 count -= end_skip;
2241 end_skip = dp->df_count[idx_from] - start_skip - count;
2242 if (end_skip < 0)
2243 end_skip = 0;
2244 }
2245 }
2246 else
2247 end_skip = 0;
2248 }
2249
Bram Moolenaar280f1262006-01-30 00:14:18 +00002250 buf_empty = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002251 added = 0;
2252 for (i = 0; i < count; ++i)
2253 {
Bram Moolenaar280f1262006-01-30 00:14:18 +00002254 /* remember deleting the last line of the buffer */
2255 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002256 ml_delete(lnum, FALSE);
2257 --added;
2258 }
2259 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2260 {
2261 linenr_T nr;
2262
2263 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002264 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002265 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002266 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2267 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002268 if (p != NULL)
2269 {
2270 ml_append(lnum + i - 1, p, 0, FALSE);
2271 vim_free(p);
2272 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002273 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2274 {
2275 /* Added the first line into an empty buffer, need to
2276 * delete the dummy empty line. */
2277 buf_empty = FALSE;
2278 ml_delete((linenr_T)2, FALSE);
2279 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280 }
2281 }
2282 new_count = dp->df_count[idx_to] + added;
2283 dp->df_count[idx_to] = new_count;
2284
2285 if (start_skip == 0 && end_skip == 0)
2286 {
2287 /* Check if there are any other buffers and if the diff is
2288 * equal in them. */
2289 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002290 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2291 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002292 && !diff_equal_entry(dp, idx_from, i))
2293 break;
2294 if (i == DB_COUNT)
2295 {
2296 /* delete the diff entry, the buffers are now equal here */
2297 dfree = dp;
2298 dp = dp->df_next;
2299 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002300 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002301 else
2302 dprev->df_next = dp;
2303 }
2304 }
2305
2306 /* Adjust marks. This will change the following entries! */
2307 if (added != 0)
2308 {
2309 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2310 if (curwin->w_cursor.lnum >= lnum)
2311 {
2312 /* Adjust the cursor position if it's in/after the changed
2313 * lines. */
2314 if (curwin->w_cursor.lnum >= lnum + count)
2315 curwin->w_cursor.lnum += added;
2316 else if (added < 0)
2317 curwin->w_cursor.lnum = lnum;
2318 }
2319 }
2320 changed_lines(lnum, 0, lnum + count, (long)added);
2321
2322 if (dfree != NULL)
2323 {
2324 /* Diff is deleted, update folds in other windows. */
2325#ifdef FEAT_FOLDING
2326 diff_fold_update(dfree, idx_to);
2327#endif
2328 vim_free(dfree);
2329 }
2330 else
2331 /* mark_adjust() may have changed the count in a wrong way */
2332 dp->df_count[idx_to] = new_count;
2333
2334 /* When changing the current buffer, keep track of line numbers */
2335 if (idx_cur == idx_to)
2336 off += added;
2337 }
2338
2339 /* If before the range or not deleted, go to next diff. */
2340 if (dfree == NULL)
2341 {
2342 dprev = dp;
2343 dp = dp->df_next;
2344 }
2345 }
2346
2347 /* restore curwin/curbuf and a few other things */
2348 if (idx_other == idx_to)
2349 {
2350 /* Syncing undo only works for the current buffer, but we change
2351 * another buffer. Sync undo if the command was typed. This isn't
2352 * 100% right when ":diffput" is used in a function or mapping. */
2353 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002354 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002355 aucmd_restbuf(&aco);
2356 }
2357
2358 diff_busy = FALSE;
2359
2360 /* Check that the cursor is on a valid character and update it's position.
2361 * When there were filler lines the topline has become invalid. */
2362 check_cursor();
2363 changed_line_abv_curs();
2364
2365 /* Also need to redraw the other buffers. */
2366 diff_redraw(FALSE);
2367}
2368
2369#ifdef FEAT_FOLDING
2370/*
2371 * Update folds for all diff buffers for entry "dp".
2372 * Skip buffer with index "skip_idx".
2373 * When there are no diffs, all folds are removed.
2374 */
2375 static void
2376diff_fold_update(dp, skip_idx)
2377 diff_T *dp;
2378 int skip_idx;
2379{
2380 int i;
2381 win_T *wp;
2382
2383 for (wp = firstwin; wp != NULL; wp = wp->w_next)
2384 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002385 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002386 foldUpdate(wp, dp->df_lnum[i],
2387 dp->df_lnum[i] + dp->df_count[i]);
2388}
2389#endif
2390
2391/*
2392 * Return TRUE if buffer "buf" is in diff-mode.
2393 */
2394 int
2395diff_mode_buf(buf)
2396 buf_T *buf;
2397{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002398 tabpage_T *tp;
2399
2400 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
2401 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
2402 return TRUE;
2403 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404}
2405
2406/*
2407 * Move "count" times in direction "dir" to the next diff block.
2408 * Return FAIL if there isn't such a diff block.
2409 */
2410 int
2411diff_move_to(dir, count)
2412 int dir;
2413 long count;
2414{
2415 int idx;
2416 linenr_T lnum = curwin->w_cursor.lnum;
2417 diff_T *dp;
2418
2419 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002420 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002421 return FAIL;
2422
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002423 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002424 ex_diffupdate(NULL); /* update after a big change */
2425
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002426 if (curtab->tp_first_diff == NULL) /* no diffs today */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002427 return FAIL;
2428
2429 while (--count >= 0)
2430 {
2431 /* Check if already before first diff. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002432 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002433 break;
2434
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002435 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002436 {
2437 if (dp == NULL)
2438 break;
2439 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
2440 || (dir == BACKWARD
2441 && (dp->df_next == NULL
2442 || lnum <= dp->df_next->df_lnum[idx])))
2443 {
2444 lnum = dp->df_lnum[idx];
2445 break;
2446 }
2447 }
2448 }
2449
2450 /* don't end up past the end of the file */
2451 if (lnum > curbuf->b_ml.ml_line_count)
2452 lnum = curbuf->b_ml.ml_line_count;
2453
2454 /* When the cursor didn't move at all we fail. */
2455 if (lnum == curwin->w_cursor.lnum)
2456 return FAIL;
2457
2458 setpcmark();
2459 curwin->w_cursor.lnum = lnum;
2460 curwin->w_cursor.col = 0;
2461
2462 return OK;
2463}
2464
2465#if defined(FEAT_FOLDING) || defined(PROTO)
2466/*
2467 * For line "lnum" in the current window find the equivalent lnum in window
2468 * "wp", compensating for inserted/deleted lines.
2469 */
2470 linenr_T
2471diff_lnum_win(lnum, wp)
2472 linenr_T lnum;
2473 win_T *wp;
2474{
2475 diff_T *dp;
2476 int idx;
2477 int i;
2478 linenr_T n;
2479
2480 idx = diff_buf_idx(curbuf);
2481 if (idx == DB_COUNT) /* safety check */
2482 return (linenr_T)0;
2483
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002484 if (curtab->tp_diff_invalid)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002485 ex_diffupdate(NULL); /* update after a big change */
2486
2487 /* search for a change that includes "lnum" in the list of diffblocks. */
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002488 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2490 break;
2491
2492 /* When after the last change, compute relative to the last line number. */
2493 if (dp == NULL)
2494 return wp->w_buffer->b_ml.ml_line_count
2495 - (curbuf->b_ml.ml_line_count - lnum);
2496
2497 /* Find index for "wp". */
2498 i = diff_buf_idx(wp->w_buffer);
2499 if (i == DB_COUNT) /* safety check */
2500 return (linenr_T)0;
2501
2502 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
2503 if (n > dp->df_lnum[i] + dp->df_count[i])
2504 n = dp->df_lnum[i] + dp->df_count[i];
2505 return n;
2506}
2507#endif
2508
2509#endif /* FEAT_DIFF */