diff options
Diffstat (limited to 'src/nvim/diff.c')
-rw-r--r-- | src/nvim/diff.c | 2611 |
1 files changed, 2611 insertions, 0 deletions
diff --git a/src/nvim/diff.c b/src/nvim/diff.c new file mode 100644 index 0000000000..e360edc665 --- /dev/null +++ b/src/nvim/diff.c @@ -0,0 +1,2611 @@ +/// @file diff.c +/// +/// Code for diff'ing two, three or four buffers. + +#include "vim.h" +#include "diff.h" +#include "buffer.h" +#include "charset.h" +#include "eval.h" +#include "ex_cmds.h" +#include "ex_docmd.h" +#include "fileio.h" +#include "fold.h" +#include "mark.h" +#include "mbyte.h" +#include "memline.h" +#include "message.h" +#include "misc1.h" +#include "misc2.h" +#include "memory.h" +#include "move.h" +#include "normal.h" +#include "option.h" +#include "path.h" +#include "screen.h" +#include "undo.h" +#include "window.h" +#include "os/os.h" +#include "os/shell.h" + +static int diff_busy = FALSE; // ex_diffgetput() is busy + +// Flags obtained from the 'diffopt' option +#define DIFF_FILLER 1 // display filler lines +#define DIFF_ICASE 2 // ignore case +#define DIFF_IWHITE 4 // ignore change in white space +#define DIFF_HORIZONTAL 8 // horizontal splits +#define DIFF_VERTICAL 16 // vertical splits +static int diff_flags = DIFF_FILLER; + +#define LBUFLEN 50 // length of line in diff file + +// TRUE when "diff -a" works, FALSE when it doesn't work, MAYBE when not +// checked yet +static int diff_a_works = MAYBE; + +static int diff_buf_idx(buf_T *buf); +static int diff_buf_idx_tp(buf_T *buf, tabpage_T *tp); +static void diff_mark_adjust_tp(tabpage_T *tp, int idx, linenr_T line1, + linenr_T line2, long amount, + long amount_after); +static void diff_check_unchanged(tabpage_T *tp, diff_T *dp); +static int diff_check_sanity(tabpage_T *tp, diff_T *dp); +static void diff_redraw(int dofold); +static int diff_write(buf_T *buf, char_u *fname); +static void diff_file(char_u *tmp_orig, char_u *tmp_new, char_u *tmp_diff); +static int diff_equal_entry(diff_T *dp, int idx1, int idx2); +static int diff_cmp(char_u *s1, char_u *s2); +static void diff_fold_update(diff_T *dp, int skip_idx); +static void diff_read(int idx_orig, int idx_new, char_u *fname); +static void diff_copy_entry(diff_T *dprev, diff_T *dp, int idx_orig, + int idx_new); +static diff_T* diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp); + +#ifndef USE_CR +# define tag_fgets vim_fgets +#endif // ifndef USE_CR + +/// Called when deleting or unloading a buffer: No longer make a diff with it. +/// +/// @param buf +void diff_buf_delete(buf_T *buf) +{ + tabpage_T *tp; + for (tp = first_tabpage; tp != NULL; tp = tp->tp_next) { + int i = diff_buf_idx_tp(buf, tp); + + if (i != DB_COUNT) { + tp->tp_diffbuf[i] = NULL; + tp->tp_diff_invalid = TRUE; + + if (tp == curtab) { + diff_redraw(TRUE); + } + } + } +} + +/// Check if the current buffer should be added to or removed from the list of +/// diff buffers. +/// +/// @param win +void diff_buf_adjust(win_T *win) +{ + + if (!win->w_p_diff) { + // When there is no window showing a diff for this buffer, remove + // it from the diffs. + win_T *wp; + for (wp = firstwin; wp != NULL; wp = wp->w_next) { + if ((wp->w_buffer == win->w_buffer) && wp->w_p_diff) { + break; + } + } + + if (wp == NULL) { + int i = diff_buf_idx(win->w_buffer); + if (i != DB_COUNT) { + curtab->tp_diffbuf[i] = NULL; + curtab->tp_diff_invalid = TRUE; + diff_redraw(TRUE); + } + } + } else { + diff_buf_add(win->w_buffer); + } +} + +/// Add a buffer to make diffs for. +/// +/// Call this when a new buffer is being edited in the current window where +/// 'diff' is set. +/// Marks the current buffer as being part of the diff and requiring updating. +/// This must be done before any autocmd, because a command may use info +/// about the screen contents. +/// +/// @param buf The buffer to add. +void diff_buf_add(buf_T *buf) +{ + if (diff_buf_idx(buf) != DB_COUNT) { + // It's already there. + return; + } + + int i; + for (i = 0; i < DB_COUNT; ++i) { + if (curtab->tp_diffbuf[i] == NULL) { + curtab->tp_diffbuf[i] = buf; + curtab->tp_diff_invalid = TRUE; + diff_redraw(TRUE); + return; + } + } + + EMSGN(_("E96: Can not diff more than %" PRId64 " buffers"), DB_COUNT); +} + +/// Find buffer "buf" in the list of diff buffers for the current tab page. +/// +/// @param buf The buffer to find. +/// +/// @return Its index or DB_COUNT if not found. +static int diff_buf_idx(buf_T *buf) +{ + int idx; + for (idx = 0; idx < DB_COUNT; ++idx) { + if (curtab->tp_diffbuf[idx] == buf) { + break; + } + } + return idx; +} + +/// Find buffer "buf" in the list of diff buffers for tab page "tp". +/// +/// @param buf +/// @param tp +/// +/// @return its index or DB_COUNT if not found. +static int diff_buf_idx_tp(buf_T *buf, tabpage_T *tp) +{ + int idx; + for (idx = 0; idx < DB_COUNT; ++idx) { + if (tp->tp_diffbuf[idx] == buf) { + break; + } + } + return idx; +} + +/// Mark the diff info involving buffer "buf" as invalid, it will be updated +/// when info is requested. +/// +/// @param buf +void diff_invalidate(buf_T *buf) +{ + tabpage_T *tp; + for (tp = first_tabpage; tp != NULL; tp = tp->tp_next) { + int i = diff_buf_idx_tp(buf, tp); + if (i != DB_COUNT) { + tp->tp_diff_invalid = TRUE; + if (tp == curtab) { + diff_redraw(TRUE); + } + } + } +} + +/// Called by mark_adjust(): update line numbers in "curbuf". +/// +/// @param line1 +/// @param line2 +/// @param amount +/// @param amount_after +void diff_mark_adjust(linenr_T line1, linenr_T line2, long amount, + long amount_after) +{ + // Handle all tab pages that use the current buffer in a diff. + tabpage_T *tp; + for (tp = first_tabpage; tp != NULL; tp = tp->tp_next) { + int idx = diff_buf_idx_tp(curbuf, tp); + if (idx != DB_COUNT) { + diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after); + } + } +} + +/// Update line numbers in tab page "tp" for "curbuf" with index "idx". +/// +/// This attempts to update the changes as much as possible: +/// When inserting/deleting lines outside of existing change blocks, create a +/// new change block and update the line numbers in following blocks. +/// When inserting/deleting lines in existing change blocks, update them. +/// +/// @param tp +/// @param idx +/// @param line1 +/// @param line2 +/// @param amount +/// @amount_after +static void diff_mark_adjust_tp(tabpage_T *tp, int idx, linenr_T line1, + linenr_T line2, long amount, long amount_after) +{ + int inserted; + int deleted; + if (line2 == MAXLNUM) { + // mark_adjust(99, MAXLNUM, 9, 0): insert lines + inserted = amount; + deleted = 0; + } else if (amount_after > 0) { + // mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines + inserted = amount_after; + deleted = 0; + } else { + // mark_adjust(98, 99, MAXLNUM, -2): delete lines + inserted = 0; + deleted = -amount_after; + } + + diff_T *dprev = NULL; + diff_T *dp = tp->tp_first_diff; + + linenr_T last; + linenr_T lnum_deleted = line1; // lnum of remaining deletion + int n; + int off; + for (;;) { + // If the change is after the previous diff block and before the next + // diff block, thus not touching an existing change, create a new diff + // block. Don't do this when ex_diffgetput() is busy. + if (((dp == NULL) + || (dp->df_lnum[idx] - 1 > line2) + || ((line2 == MAXLNUM) && (dp->df_lnum[idx] > line1))) + && ((dprev == NULL) + || (dprev->df_lnum[idx] + dprev->df_count[idx] < line1)) + && !diff_busy) { + diff_T *dnext = diff_alloc_new(tp, dprev, dp); + + dnext->df_lnum[idx] = line1; + dnext->df_count[idx] = inserted; + int i; + for (i = 0; i < DB_COUNT; ++i) { + if ((tp->tp_diffbuf[i] != NULL) && (i != idx)) { + if (dprev == NULL) { + dnext->df_lnum[i] = line1; + } else { + dnext->df_lnum[i] = line1 + + (dprev->df_lnum[i] + dprev->df_count[i]) + - (dprev->df_lnum[idx] + dprev->df_count[idx]); + } + dnext->df_count[i] = deleted; + } + } + } + + // if at end of the list, quit + if (dp == NULL) { + break; + } + + // + // Check for these situations: + // 1 2 3 + // 1 2 3 + // line1 2 3 4 5 + // 2 3 4 5 + // 2 3 4 5 + // line2 2 3 4 5 + // 3 5 6 + // 3 5 6 + + // compute last line of this change + last = dp->df_lnum[idx] + dp->df_count[idx] - 1; + + // 1. change completely above line1: nothing to do + if (last >= line1 - 1) { + // 6. change below line2: only adjust for amount_after; also when + // "deleted" became zero when deleted all lines between two diffs. + if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2) { + if (amount_after == 0) { + // nothing left to change + break; + } + dp->df_lnum[idx] += amount_after; + } else { + int check_unchanged = FALSE; + + // 2. 3. 4. 5.: inserted/deleted lines touching this diff. + if (deleted > 0) { + if (dp->df_lnum[idx] >= line1) { + off = dp->df_lnum[idx] - lnum_deleted; + + if (last <= line2) { + // 4. delete all lines of diff + if ((dp->df_next != NULL) + && (dp->df_next->df_lnum[idx] - 1 <= line2)) { + // delete continues in next diff, only do + // lines until that one + n = dp->df_next->df_lnum[idx] - lnum_deleted; + deleted -= n; + n -= dp->df_count[idx]; + lnum_deleted = dp->df_next->df_lnum[idx]; + } else { + n = deleted - dp->df_count[idx]; + } + dp->df_count[idx] = 0; + } else { + // 5. delete lines at or just before top of diff + n = off; + dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1; + check_unchanged = TRUE; + } + dp->df_lnum[idx] = line1; + } else { + off = 0; + + if (last < line2) { + // 2. delete at end of of diff + dp->df_count[idx] -= last - lnum_deleted + 1; + + if ((dp->df_next != NULL) + && (dp->df_next->df_lnum[idx] - 1 <= line2)) { + // delete continues in next diff, only do + // lines until that one + n = dp->df_next->df_lnum[idx] - 1 - last; + deleted -= dp->df_next->df_lnum[idx] - lnum_deleted; + lnum_deleted = dp->df_next->df_lnum[idx]; + } else { + n = line2 - last; + } + check_unchanged = TRUE; + } else { + // 3. delete lines inside the diff + n = 0; + dp->df_count[idx] -= deleted; + } + } + + int i; + for (i = 0; i < DB_COUNT; ++i) { + if ((tp->tp_diffbuf[i] != NULL) && (i != idx)) { + dp->df_lnum[i] -= off; + dp->df_count[i] += n; + } + } + } else { + if (dp->df_lnum[idx] <= line1) { + // inserted lines somewhere in this diff + dp->df_count[idx] += inserted; + check_unchanged = TRUE; + } else { + // inserted lines somewhere above this diff + dp->df_lnum[idx] += inserted; + } + } + + if (check_unchanged) { + // Check if inserted lines are equal, may reduce the size of the + // diff. + // + // TODO: also check for equal lines in the middle and perhaps split + // the block. + diff_check_unchanged(tp, dp); + } + } + } + + // check if this block touches the previous one, may merge them. + if ((dprev != NULL) + && (dprev->df_lnum[idx] + dprev->df_count[idx] == dp->df_lnum[idx])) { + int i; + for (i = 0; i < DB_COUNT; ++i) { + if (tp->tp_diffbuf[i] != NULL) { + dprev->df_count[i] += dp->df_count[i]; + } + } + dprev->df_next = dp->df_next; + free(dp); + dp = dprev->df_next; + } else { + // Advance to next entry. + dprev = dp; + dp = dp->df_next; + } + } + + dprev = NULL; + dp = tp->tp_first_diff; + + while (dp != NULL) { + // All counts are zero, remove this entry. + int i; + for (i = 0; i < DB_COUNT; ++i) { + if ((tp->tp_diffbuf[i] != NULL) && (dp->df_count[i] != 0)) { + break; + } + } + + if (i == DB_COUNT) { + diff_T *dnext = dp->df_next; + free(dp); + dp = dnext; + + if (dprev == NULL) { + tp->tp_first_diff = dnext; + } else { + dprev->df_next = dnext; + } + } else { + // Advance to next entry. + dprev = dp; + dp = dp->df_next; + } + } + + if (tp == curtab) { + diff_redraw(TRUE); + + // Need to recompute the scroll binding, may remove or add filler + // lines (e.g., when adding lines above w_topline). But it's slow when + // making many changes, postpone until redrawing. + diff_need_scrollbind = TRUE; + } +} + +/// Allocate a new diff block and link it between "dprev" and "dp". +/// +/// @param tp +/// @param dprev +/// @param dp +/// +/// @return The new diff block. +static diff_T* diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp) +{ + diff_T *dnew = xmalloc(sizeof(*dnew)); + + dnew->df_next = dp; + if (dprev == NULL) { + tp->tp_first_diff = dnew; + } else { + dprev->df_next = dnew; + } + + return dnew; +} + +/// Check if the diff block "dp" can be made smaller for lines at the start and +/// end that are equal. Called after inserting lines. +/// +/// This may result in a change where all buffers have zero lines, the caller +/// must take care of removing it. +/// +/// @param tp +/// @param dp +static void diff_check_unchanged(tabpage_T *tp, diff_T *dp) +{ + // Find the first buffers, use it as the original, compare the other + // buffer lines against this one. + int i_org; + for (i_org = 0; i_org < DB_COUNT; ++i_org) { + if (tp->tp_diffbuf[i_org] != NULL) { + break; + } + } + + // safety check + if (i_org == DB_COUNT) { + return; + } + + if (diff_check_sanity(tp, dp) == FAIL) { + return; + } + + // First check lines at the top, then at the bottom. + int off_org = 0; + int off_new = 0; + int dir = FORWARD; + for (;;) { + // Repeat until a line is found which is different or the number of + // lines has become zero. + while (dp->df_count[i_org] > 0) { + // Copy the line, the next ml_get() will invalidate it. + if (dir == BACKWARD) { + off_org = dp->df_count[i_org] - 1; + } + char_u *line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org], + dp->df_lnum[i_org] + off_org, + FALSE)); + + if (line_org == NULL) { + return; + } + + int i_new; + for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new) { + if (tp->tp_diffbuf[i_new] == NULL) { + continue; + } + + if (dir == BACKWARD) { + off_new = dp->df_count[i_new] - 1; + } + + // if other buffer doesn't have this line, it was inserted + if ((off_new < 0) || (off_new >= dp->df_count[i_new])) { + break; + } + + if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new], + dp->df_lnum[i_new] + off_new, + FALSE)) != 0) { + break; + } + } + free(line_org); + + // Stop when a line isn't equal in all diff buffers. + if (i_new != DB_COUNT) { + break; + } + + // Line matched in all buffers, remove it from the diff. + for (i_new = i_org; i_new < DB_COUNT; ++i_new) { + if (tp->tp_diffbuf[i_new] != NULL) { + if (dir == FORWARD) { + dp->df_lnum[i_new]++; + } + dp->df_count[i_new]--; + } + } + } + + if (dir == BACKWARD) { + break; + } + dir = BACKWARD; + } +} + +/// Check if a diff block doesn't contain invalid line numbers. +/// This can happen when the diff program returns invalid results. +/// +/// @param tp +/// @param dp +/// +/// @return OK if the diff block doesn't contain invalid line numbers. +static int diff_check_sanity(tabpage_T *tp, diff_T *dp) +{ + int i; + for (i = 0; i < DB_COUNT; ++i) { + if (tp->tp_diffbuf[i] != NULL) { + if (dp->df_lnum[i] + dp->df_count[i] - 1 + > tp->tp_diffbuf[i]->b_ml.ml_line_count) { + return FAIL; + } + } + } + return OK; +} + +/// Mark all diff buffers in the current tab page for redraw. +/// +/// @param dofold Also recompute the folds +static void diff_redraw(int dofold) +{ + win_T *wp; + for (wp = firstwin; wp != NULL; wp = wp->w_next) { + if (wp->w_p_diff) { + redraw_win_later(wp, SOME_VALID); + if (dofold && foldmethodIsDiff(wp)) { + foldUpdateAll(wp); + } + + /* A change may have made filler lines invalid, need to take care + * of that for other windows. */ + int n = diff_check(wp, wp->w_topline); + + if (((wp != curwin) && (wp->w_topfill > 0)) || (n > 0)) { + if (wp->w_topfill > n) { + wp->w_topfill = (n < 0 ? 0 : n); + } else if ((n > 0) && (n > wp->w_topfill)) { + wp->w_topfill = n; + } + } + } + } +} + +/// Write buffer "buf" to file "name". +/// +/// Always use 'fileformat' set to "unix". +/// +/// @param buf +/// @param fname +/// +/// @return FAIL for failure +static int diff_write(buf_T *buf, char_u *fname) +{ + char_u *save_ff = buf->b_p_ff; + buf->b_p_ff = vim_strsave((char_u *)FF_UNIX); + int r = buf_write(buf, fname, NULL, (linenr_T)1, buf->b_ml.ml_line_count, + NULL, FALSE, FALSE, FALSE, TRUE); + free_string_option(buf->b_p_ff); + buf->b_p_ff = save_ff; + return r; +} + +/// Completely update the diffs for the buffers involved. +/// +/// This uses the ordinary "diff" command. +/// The buffers are written to a file, also for unmodified buffers (the file +/// could have been produced by autocommands, e.g. the netrw plugin). +/// +/// @param eap can be NULL +void ex_diffupdate(exarg_T *eap) +{ + // Delete all diffblocks. + diff_clear(curtab); + curtab->tp_diff_invalid = FALSE; + + // Use the first buffer as the original text. + int idx_orig; + for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig) { + if (curtab->tp_diffbuf[idx_orig] != NULL) { + break; + } + } + + if (idx_orig == DB_COUNT) { + return; + } + + // Only need to do something when there is another buffer. + int idx_new; + for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new) { + if (curtab->tp_diffbuf[idx_new] != NULL) { + break; + } + } + + if (idx_new == DB_COUNT) { + return; + } + + // We need three temp file names. + char_u *tmp_orig = vim_tempname('o'); + char_u *tmp_new = vim_tempname('n'); + char_u *tmp_diff = vim_tempname('d'); + + if ((tmp_orig == NULL) || (tmp_new == NULL) || (tmp_diff == NULL)) { + goto theend; + } + + // Do a quick test if "diff" really works. Otherwise it looks like there + // are no differences. Can't use the return value, it's non-zero when + // there are differences. + // May try twice, first with "-a" and then without. + int io_error = FALSE; + int ok = FALSE; + for (;;) { + ok = FALSE; + FILE *fd = mch_fopen((char *)tmp_orig, "w"); + + if (fd == NULL) { + io_error = TRUE; + } else { + if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1) { + io_error = TRUE; + } + fclose(fd); + fd = mch_fopen((char *)tmp_new, "w"); + + if (fd == NULL) { + io_error = TRUE; + } else { + if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1) { + io_error = TRUE; + } + fclose(fd); + diff_file(tmp_orig, tmp_new, tmp_diff); + fd = mch_fopen((char *)tmp_diff, "r"); + + if (fd == NULL) { + io_error = TRUE; + } else { + char_u linebuf[LBUFLEN]; + + for (;;) { + // There must be a line that contains "1c1". + if (tag_fgets(linebuf, LBUFLEN, fd)) { + break; + } + + if (STRNCMP(linebuf, "1c1", 3) == 0) { + ok = TRUE; + } + } + fclose(fd); + } + os_remove((char *)tmp_diff); + os_remove((char *)tmp_new); + } + os_remove((char *)tmp_orig); + } + + // When using 'diffexpr' break here. + if (*p_dex != NUL) { + break; + } + + // If we checked if "-a" works already, break here. + if (diff_a_works != MAYBE) { + break; + } + diff_a_works = ok; + + // If "-a" works break here, otherwise retry without "-a". + if (ok) { + break; + } + } + + if (!ok) { + if (io_error) { + EMSG(_("E810: Cannot read or write temp files")); + } + EMSG(_("E97: Cannot create diffs")); + diff_a_works = MAYBE; + goto theend; + } + + // :diffupdate! + if ((eap != NULL) && eap->forceit) { + for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new) { + buf_T *buf = curtab->tp_diffbuf[idx_new]; + if (buf_valid(buf)) { + buf_check_timestamp(buf, FALSE); + } + } + } + + // Write the first buffer to a tempfile. + buf_T *buf = curtab->tp_diffbuf[idx_orig]; + if (diff_write(buf, tmp_orig) == FAIL) { + goto theend; + } + + // Make a difference between the first buffer and every other. + for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new) { + buf_T *buf = curtab->tp_diffbuf[idx_new]; + if (buf == NULL) { + continue; + } + + if (diff_write(buf, tmp_new) == FAIL) { + continue; + } + diff_file(tmp_orig, tmp_new, tmp_diff); + + // Read the diff output and add each entry to the diff list. + diff_read(idx_orig, idx_new, tmp_diff); + os_remove((char *)tmp_diff); + os_remove((char *)tmp_new); + } + os_remove((char *)tmp_orig); + + // force updating cursor position on screen + curwin->w_valid_cursor.lnum = 0; + + diff_redraw(TRUE); + +theend: + free(tmp_orig); + free(tmp_new); + free(tmp_diff); +} + +/// Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff". +/// +/// @param tmp_orig +/// @param tmp_new +/// @param tmp_diff +static void diff_file(char_u *tmp_orig, char_u *tmp_new, char_u *tmp_diff) +{ + if (*p_dex != NUL) { + // Use 'diffexpr' to generate the diff file. + eval_diff(tmp_orig, tmp_new, tmp_diff); + } else { + size_t len = STRLEN(tmp_orig) + STRLEN(tmp_new) + STRLEN(tmp_diff) + + STRLEN(p_srr) + 27; + char_u *cmd = xmalloc(len); + + /* We don't want $DIFF_OPTIONS to get in the way. */ + if (os_getenv("DIFF_OPTIONS")) { + vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)""); + } + + /* Build the diff command and execute it. Always use -a, binary + * differences are of no use. Ignore errors, diff returns + * non-zero when differences have been found. */ + vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s %s", + diff_a_works == FALSE ? "" : "-a ", + "", + (diff_flags & DIFF_IWHITE) ? "-b " : "", + (diff_flags & DIFF_ICASE) ? "-i " : "", + tmp_orig, tmp_new); + append_redir(cmd, (int)len, p_srr, tmp_diff); + block_autocmds(); /* Avoid ShellCmdPost stuff */ + (void)call_shell( + cmd, + kShellOptFilter | kShellOptSilent | kShellOptDoOut, + NULL + ); + unblock_autocmds(); + free(cmd); + } +} + +/// Create a new version of a file from the current buffer and a diff file. +/// +/// The buffer is written to a file, also for unmodified buffers (the file +/// could have been produced by autocommands, e.g. the netrw plugin). +/// +/// @param eap +void ex_diffpatch(exarg_T *eap) +{ + char_u *buf = NULL; + win_T *old_curwin = curwin; + char_u *newname = NULL; // name of patched file buffer + +#ifdef UNIX + char_u dirbuf[MAXPATHL]; + char_u *fullname = NULL; +#endif // ifdef UNIX + // We need two temp file names. + // Name of original temp file. + char_u *tmp_orig = vim_tempname('o'); + // Name of patched temp file. + char_u *tmp_new = vim_tempname('n'); + + if ((tmp_orig == NULL) || (tmp_new == NULL)) { + goto theend; + } + + // Write the current buffer to "tmp_orig". + if (buf_write(curbuf, tmp_orig, NULL, + (linenr_T)1, curbuf->b_ml.ml_line_count, + NULL, FALSE, FALSE, FALSE, TRUE) == FAIL) { + goto theend; + } + +#ifdef UNIX + // Get the absolute path of the patchfile, changing directory below. + fullname = FullName_save(eap->arg, FALSE); +#endif // ifdef UNIX + +#ifdef UNIX + size_t buflen = STRLEN(tmp_orig) + + (fullname != NULL ? STRLEN(fullname) : STRLEN(eap->arg)) + + STRLEN(tmp_new) + 16; +#else + size_t buflen = STRLEN(tmp_orig) + (STRLEN(eap->arg)) + STRLEN(tmp_new) + 16; +#endif // ifdef UNIX + + buf = xmalloc(buflen); + +#ifdef UNIX + + // Temporarily chdir to /tmp, to avoid patching files in the current + // directory when the patch file contains more than one patch. When we + // have our own temp dir use that instead, it will be cleaned up when we + // exit (any .rej files created). Don't change directory if we can't + // return to the current. + if ((os_dirname(dirbuf, MAXPATHL) != OK) + || (os_chdir((char *)dirbuf) != 0)) { + dirbuf[0] = NUL; + } else { +# ifdef TEMPDIRNAMES + if (vim_tempdir != NULL) { + ignored = os_chdir((char *)vim_tempdir); + } else { + ignored = os_chdir("/tmp"); + } +# else + ignored = os_chdir("/tmp"); +# endif // ifdef TEMPDIRNAMES + shorten_fnames(TRUE); + } +#endif // ifdef UNIX + + if (*p_pex != NUL) { + // Use 'patchexpr' to generate the new file. +#ifdef UNIX + eval_patch(tmp_orig, fullname != NULL ? fullname : eap->arg, tmp_new); +#else + eval_patch(tmp_orig, eap->arg, tmp_new); +#endif // ifdef UNIX + } else { + // Build the patch command and execute it. Ignore errors. Switch to + // cooked mode to allow the user to respond to prompts. +#ifdef UNIX + vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"", + tmp_new, tmp_orig, fullname != NULL ? fullname : eap->arg); +#else + vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"", + tmp_new, tmp_orig, eap->arg); +#endif // ifdef UNIX + // Avoid ShellCmdPost stuff + block_autocmds(); + (void)call_shell(buf, kShellOptFilter | kShellOptCooked, NULL); + unblock_autocmds(); + } + +#ifdef UNIX + if (dirbuf[0] != NUL) { + if (os_chdir((char *)dirbuf) != 0) { + EMSG(_(e_prev_dir)); + } + shorten_fnames(TRUE); + } +#endif // ifdef UNIX + + // patch probably has written over the screen + redraw_later(CLEAR); + + // Delete any .orig or .rej file created. + STRCPY(buf, tmp_new); + STRCAT(buf, ".orig"); + os_remove((char *)buf); + STRCPY(buf, tmp_new); + STRCAT(buf, ".rej"); + os_remove((char *)buf); + + // Only continue if the output file was created. + off_t file_size; + bool file_size_success = os_get_file_size((char *)tmp_new, &file_size); + if (!file_size_success || file_size == 0) { + EMSG(_("E816: Cannot read patch output")); + } else { + if (curbuf->b_fname != NULL) { + newname = vim_strnsave(curbuf->b_fname, + (int)(STRLEN(curbuf->b_fname) + 4)); + + if (newname != NULL) { + STRCAT(newname, ".new"); + } + } + + // don't use a new tab page, each tab page has its own diffs + cmdmod.tab = 0; + + if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL) { + // Pretend it was a ":split fname" command + eap->cmdidx = CMD_split; + eap->arg = tmp_new; + do_exedit(eap, old_curwin); + + // check that split worked and editing tmp_new + if ((curwin != old_curwin) && win_valid(old_curwin)) { + // Set 'diff', 'scrollbind' on and 'wrap' off. + diff_win_options(curwin, TRUE); + diff_win_options(old_curwin, TRUE); + + if (newname != NULL) { + // do a ":file filename.new" on the patched buffer + eap->arg = newname; + ex_file(eap); + + // Do filetype detection with the new name. + if (au_has_group((char_u *)"filetypedetect")) { + do_cmdline_cmd((char_u *)":doau filetypedetect BufRead"); + } + } + } + } + } + +theend: + if (tmp_orig != NULL) { + os_remove((char *)tmp_orig); + } + free(tmp_orig); + + if (tmp_new != NULL) { + os_remove((char *)tmp_new); + } + free(tmp_new); + free(newname); + free(buf); +#ifdef UNIX + free(fullname); +#endif // ifdef UNIX +} + +/// Split the window and edit another file, setting options to show the diffs. +/// +/// @param eap +void ex_diffsplit(exarg_T *eap) +{ + win_T *old_curwin = curwin; + + // don't use a new tab page, each tab page has its own diffs + cmdmod.tab = 0; + + if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL) { + // Pretend it was a ":split fname" command + eap->cmdidx = CMD_split; + curwin->w_p_diff = TRUE; + do_exedit(eap, old_curwin); + + // split must have worked + if (curwin != old_curwin) { + // Set 'diff', 'scrollbind' on and 'wrap' off. + diff_win_options(curwin, TRUE); + diff_win_options(old_curwin, TRUE); + } + } +} + +// Set options to show diffs for the current window. +void ex_diffthis(exarg_T *eap) +{ + // Set 'diff', 'scrollbind' on and 'wrap' off. + diff_win_options(curwin, TRUE); +} + +/// Set options in window "wp" for diff mode. +/// +/// @param addbuf Add buffer to diff. +void diff_win_options(win_T *wp, int addbuf) +{ + win_T *old_curwin = curwin; + + // close the manually opened folds + curwin = wp; + newFoldLevel(); + curwin = old_curwin; + + wp->w_p_diff = TRUE; + + // Use 'scrollbind' and 'cursorbind' when available + if (!wp->w_p_diff_saved) { + wp->w_p_scb_save = wp->w_p_scb; + } + wp->w_p_scb = TRUE; + + if (!wp->w_p_diff_saved) { + wp->w_p_crb_save = wp->w_p_crb; + } + wp->w_p_crb = TRUE; + + if (!wp->w_p_diff_saved) { + wp->w_p_wrap_save = wp->w_p_wrap; + } + wp->w_p_wrap = FALSE; + curwin = wp; + curbuf = curwin->w_buffer; + + if (!wp->w_p_diff_saved) { + wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm); + } + set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff", + OPT_LOCAL | OPT_FREE, 0); + curwin = old_curwin; + curbuf = curwin->w_buffer; + + if (!wp->w_p_diff_saved) { + wp->w_p_fdc_save = wp->w_p_fdc; + wp->w_p_fen_save = wp->w_p_fen; + wp->w_p_fdl_save = wp->w_p_fdl; + } + wp->w_p_fdc = diff_foldcolumn; + wp->w_p_fen = TRUE; + wp->w_p_fdl = 0; + foldUpdateAll(wp); + + // make sure topline is not halfway a fold + changed_window_setting_win(wp); + if (vim_strchr(p_sbo, 'h') == NULL) { + do_cmdline_cmd((char_u *)"set sbo+=hor"); + } + + // Saved the current values, to be restored in ex_diffoff(). + wp->w_p_diff_saved = TRUE; + + if (addbuf) { + diff_buf_add(wp->w_buffer); + } + redraw_win_later(wp, NOT_VALID); +} + +/// Set options not to show diffs. For the current window or all windows. +/// Only in the current tab page. +/// +/// @param eap +void ex_diffoff(exarg_T *eap) +{ + win_T *old_curwin = curwin; + int diffwin = FALSE; + + win_T *wp; + for (wp = firstwin; wp != NULL; wp = wp->w_next) { + if (eap->forceit ? wp->w_p_diff : (wp == curwin)) { + // Set 'diff', 'scrollbind' off and 'wrap' on. If option values + // were saved in diff_win_options() restore them. + wp->w_p_diff = FALSE; + + if (wp->w_p_scb) { + wp->w_p_scb = wp->w_p_diff_saved ? wp->w_p_scb_save : FALSE; + } + + if (wp->w_p_crb) { + wp->w_p_crb = wp->w_p_diff_saved ? wp->w_p_crb_save : FALSE; + } + + if (!wp->w_p_wrap) { + wp->w_p_wrap = wp->w_p_diff_saved ? wp->w_p_wrap_save : TRUE; + } + curwin = wp; + curbuf = curwin->w_buffer; + + if (wp->w_p_diff_saved) { + free_string_option(wp->w_p_fdm); + wp->w_p_fdm = wp->w_p_fdm_save; + wp->w_p_fdm_save = empty_option; + } else { + set_string_option_direct((char_u *)"fdm", -1, + (char_u *)"manual", OPT_LOCAL | OPT_FREE, 0); + } + curwin = old_curwin; + curbuf = curwin->w_buffer; + + if (wp->w_p_fdc == diff_foldcolumn) { + wp->w_p_fdc = wp->w_p_diff_saved ? wp->w_p_fdc_save : 0; + } + + if ((wp->w_p_fdl == 0) + && wp->w_p_diff_saved) { + wp->w_p_fdl = wp->w_p_fdl_save; + } + + if (wp->w_p_fen) { + // Only restore 'foldenable' when 'foldmethod' is not + // "manual", otherwise we continue to show the diff folds. + if (foldmethodIsManual(wp) || !wp->w_p_diff_saved) { + wp->w_p_fen = FALSE; + } else { + wp->w_p_fen = wp->w_p_fen_save; + } + } + + foldUpdateAll(wp); + + // make sure topline is not halfway a fold + changed_window_setting_win(wp); + + // Note: 'sbo' is not restored, it's a global option. + diff_buf_adjust(wp); + + wp->w_p_diff_saved = FALSE; + } + diffwin |= wp->w_p_diff; + } + + // Remove "hor" from from 'scrollopt' if there are no diff windows left. + if (!diffwin && (vim_strchr(p_sbo, 'h') != NULL)) { + do_cmdline_cmd((char_u *)"set sbo-=hor"); + } +} + +/// Read the diff output and add each entry to the diff list. +/// +/// @param idx_orig idx of original file +/// @param idx_new idx of new file +/// @param fname name of diff output file +static void diff_read(int idx_orig, int idx_new, char_u *fname) +{ + FILE *fd; + diff_T *dprev = NULL; + diff_T *dp = curtab->tp_first_diff; + diff_T *dn, *dpl; + long f1, l1, f2, l2; + char_u linebuf[LBUFLEN]; // only need to hold the diff line + int difftype; + char_u *p; + long off; + int i; + linenr_T lnum_orig, lnum_new; + long count_orig, count_new; + int notset = TRUE; // block "*dp" not set yet + + fd = mch_fopen((char *)fname, "r"); + + if (fd == NULL) { + EMSG(_("E98: Cannot read diff output")); + return; + } + + for (;;) { + if (tag_fgets(linebuf, LBUFLEN, fd)) { + // end of file + break; + } + + if (!isdigit(*linebuf)) { + // not the start of a diff block + continue; + } + + // This line must be one of three formats: + // {first}[,{last}]c{first}[,{last}] + // {first}a{first}[,{last}] + // {first}[,{last}]d{first} + p = linebuf; + f1 = getdigits(&p); + + if (*p == ',') { + ++p; + l1 = getdigits(&p); + } else { + l1 = f1; + } + + if ((*p != 'a') && (*p != 'c') && (*p != 'd')) { + // invalid diff format + continue; + } + difftype = *p++; + f2 = getdigits(&p); + + if (*p == ',') { + ++p; + l2 = getdigits(&p); + } else { + l2 = f2; + } + + if ((l1 < f1) || (l2 < f2)) { + // invalid line range + continue; + } + + if (difftype == 'a') { + lnum_orig = f1 + 1; + count_orig = 0; + } else { + lnum_orig = f1; + count_orig = l1 - f1 + 1; + } + + if (difftype == 'd') { + lnum_new = f2 + 1; + count_new = 0; + } else { + lnum_new = f2; + count_new = l2 - f2 + 1; + } + + // Go over blocks before the change, for which orig and new are equal. + // Copy blocks from orig to new. + while (dp != NULL + && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig]) { + if (notset) { + diff_copy_entry(dprev, dp, idx_orig, idx_new); + } + dprev = dp; + dp = dp->df_next; + notset = TRUE; + } + + if ((dp != NULL) + && (lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig]) + && (lnum_orig + count_orig >= dp->df_lnum[idx_orig])) { + // New block overlaps with existing block(s). + // First find last block that overlaps. + for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next) { + if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig]) { + break; + } + } + + // If the newly found block starts before the old one, set the + // start back a number of lines. + off = dp->df_lnum[idx_orig] - lnum_orig; + + if (off > 0) { + for (i = idx_orig; i < idx_new; ++i) { + if (curtab->tp_diffbuf[i] != NULL) { + dp->df_lnum[i] -= off; + } + } + dp->df_lnum[idx_new] = lnum_new; + dp->df_count[idx_new] = count_new; + } else if (notset) { + // new block inside existing one, adjust new block + dp->df_lnum[idx_new] = lnum_new + off; + dp->df_count[idx_new] = count_new - off; + } else { + // second overlap of new block with existing block + dp->df_count[idx_new] += count_new - count_orig + + dpl->df_lnum[idx_orig] + + dpl->df_count[idx_orig] + - (dp->df_lnum[idx_orig] + + dp->df_count[idx_orig]); + } + + // Adjust the size of the block to include all the lines to the + // end of the existing block or the new diff, whatever ends last. + off = (lnum_orig + count_orig) + - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]); + + if (off < 0) { + // new change ends in existing block, adjust the end if not + // done already + if (notset) { + dp->df_count[idx_new] += -off; + } + off = 0; + } + + for (i = idx_orig; i < idx_new; ++i) { + if (curtab->tp_diffbuf[i] != NULL) { + dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i] + - dp->df_lnum[i] + off; + } + } + + // Delete the diff blocks that have been merged into one. + dn = dp->df_next; + dp->df_next = dpl->df_next; + + while (dn != dp->df_next) { + dpl = dn->df_next; + free(dn); + dn = dpl; + } + } else { + // Allocate a new diffblock. + dp = diff_alloc_new(curtab, dprev, dp); + + dp->df_lnum[idx_orig] = lnum_orig; + dp->df_count[idx_orig] = count_orig; + dp->df_lnum[idx_new] = lnum_new; + dp->df_count[idx_new] = count_new; + + // Set values for other buffers, these must be equal to the + // original buffer, otherwise there would have been a change + // already. + for (i = idx_orig + 1; i < idx_new; ++i) { + if (curtab->tp_diffbuf[i] != NULL) { + diff_copy_entry(dprev, dp, idx_orig, i); + } + } + } + notset = FALSE; // "*dp" has been set + } + + // for remaining diff blocks orig and new are equal + while (dp != NULL) { + if (notset) { + diff_copy_entry(dprev, dp, idx_orig, idx_new); + } + dprev = dp; + dp = dp->df_next; + notset = TRUE; + } + + fclose(fd); +} + +/// Copy an entry at "dp" from "idx_orig" to "idx_new". +/// +/// @param dprev +/// @param dp +/// @param idx_orig +/// @param idx_new +static void diff_copy_entry(diff_T *dprev, diff_T *dp, int idx_orig, + int idx_new) +{ + long off; + + if (dprev == NULL) { + off = 0; + } else { + off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig]) + - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]); + } + dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off; + dp->df_count[idx_new] = dp->df_count[idx_orig]; +} + +/// Clear the list of diffblocks for tab page "tp". +/// +/// @param tp +void diff_clear(tabpage_T *tp) +{ + diff_T *p; + diff_T *next_p; + for (p = tp->tp_first_diff; p != NULL; p = next_p) { + next_p = p->df_next; + free(p); + } + tp->tp_first_diff = NULL; +} + +/// Check diff status for line "lnum" in buffer "buf": +/// +/// Returns 0 for nothing special +/// Returns -1 for a line that should be highlighted as changed. +/// Returns -2 for a line that should be highlighted as added/deleted. +/// Returns > 0 for inserting that many filler lines above it (never happens +/// when 'diffopt' doesn't contain "filler"). +/// This should only be used for windows where 'diff' is set. +/// +/// @param wp +/// @param lnum +/// +/// @return diff status. +int diff_check(win_T *wp, linenr_T lnum) +{ + int idx; // index in tp_diffbuf[] for this buffer + diff_T *dp; + int maxcount; + int i; + buf_T *buf = wp->w_buffer; + int cmp; + + if (curtab->tp_diff_invalid) { + // update after a big change + ex_diffupdate(NULL); + } + + // no diffs at all + if ((curtab->tp_first_diff == NULL) || !wp->w_p_diff) { + return 0; + } + + // safety check: "lnum" must be a buffer line + if ((lnum < 1) || (lnum > buf->b_ml.ml_line_count + 1)) { + return 0; + } + + idx = diff_buf_idx(buf); + + if (idx == DB_COUNT) { + // no diffs for buffer "buf" + return 0; + } + + // A closed fold never has filler lines. + if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL)) { + return 0; + } + + // search for a change that includes "lnum" in the list of diffblocks. + for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) { + if (lnum <= dp->df_lnum[idx] + dp->df_count[idx]) { + break; + } + } + + if ((dp == NULL) || (lnum < dp->df_lnum[idx])) { + return 0; + } + + if (lnum < dp->df_lnum[idx] + dp->df_count[idx]) { + int zero = FALSE; + + // Changed or inserted line. If the other buffers have a count of + // zero, the lines were inserted. If the other buffers have the same + // count, check if the lines are identical. + cmp = FALSE; + + for (i = 0; i < DB_COUNT; ++i) { + if ((i != idx) && (curtab->tp_diffbuf[i] != NULL)) { + if (dp->df_count[i] == 0) { + zero = TRUE; + } else { + if (dp->df_count[i] != dp->df_count[idx]) { + // nr of lines changed. + return -1; + } + cmp = TRUE; + } + } + } + + if (cmp) { + // Compare all lines. If they are equal the lines were inserted + // in some buffers, deleted in others, but not changed. + for (i = 0; i < DB_COUNT; ++i) { + if ((i != idx) + && (curtab->tp_diffbuf[i] != NULL) + && (dp->df_count[i] != 0)) { + if (!diff_equal_entry(dp, idx, i)) { + return -1; + } + } + } + } + + // If there is no buffer with zero lines then there is no difference + // any longer. Happens when making a change (or undo) that removes + // the difference. Can't remove the entry here, we might be halfway + // updating the window. Just report the text as unchanged. Other + // windows might still show the change though. + if (zero == FALSE) { + return 0; + } + return -2; + } + + // If 'diffopt' doesn't contain "filler", return 0. + if (!(diff_flags & DIFF_FILLER)) { + return 0; + } + + // Insert filler lines above the line just below the change. Will return + // 0 when this buf had the max count. + maxcount = 0; + for (i = 0; i < DB_COUNT; ++i) { + if ((curtab->tp_diffbuf[i] != NULL) && (dp->df_count[i] > maxcount)) { + maxcount = dp->df_count[i]; + } + } + return maxcount - dp->df_count[idx]; +} + +/// Compare two entries in diff "*dp" and return TRUE if they are equal. +/// +/// @param dp +/// @param idx1 First entry in diff "*dp" +/// @param idx2 Second entry in diff "*dp" +/// +/// @return return TRUE if two entires are equal. +static int diff_equal_entry(diff_T *dp, int idx1, int idx2) +{ + if (dp->df_count[idx1] != dp->df_count[idx2]) { + return FALSE; + } + + if (diff_check_sanity(curtab, dp) == FAIL) { + return FALSE; + } + + int i; + for (i = 0; i < dp->df_count[idx1]; ++i) { + char_u *line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1], + dp->df_lnum[idx1] + i, FALSE)); + + if (line == NULL) { + return FALSE; + } + int cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2], + dp->df_lnum[idx2] + i, FALSE)); + free(line); + + if (cmp != 0) { + return FALSE; + } + } + return TRUE; +} + +/// Compare strings "s1" and "s2" according to 'diffopt'. +/// Return non-zero when they are different. +/// +/// @param s1 The first string +/// @param s2 The second string +/// +/// @return on-zero if the two strings are different. +static int diff_cmp(char_u *s1, char_u *s2) +{ + if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0) { + return STRCMP(s1, s2); + } + + if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE)) { + return MB_STRICMP(s1, s2); + } + + // Ignore white space changes and possibly ignore case. + char_u *p1 = s1; + char_u *p2 = s2; + + while (*p1 != NUL && *p2 != NUL) { + if (vim_iswhite(*p1) && vim_iswhite(*p2)) { + p1 = skipwhite(p1); + p2 = skipwhite(p2); + } else { + int l = (*mb_ptr2len)(p1); + if (l != (*mb_ptr2len)(p2)) { + break; + } + + if (l > 1) { + if ((STRNCMP(p1, p2, l) != 0) + && (!enc_utf8 + || !(diff_flags & DIFF_ICASE) + || (utf_fold(utf_ptr2char(p1)) + != utf_fold(utf_ptr2char(p2))))) { + break; + } + p1 += l; + p2 += l; + } else { + if ((*p1 != *p2) + && (!(diff_flags & DIFF_ICASE) + || (TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))) { + break; + } + ++p1; + ++p2; + } + } + } + + // Ignore trailing white space. + p1 = skipwhite(p1); + p2 = skipwhite(p2); + + if ((*p1 != NUL) || (*p2 != NUL)) { + return 1; + } + return 0; +} + +/// Return the number of filler lines above "lnum". +/// +/// @param wp +/// @param lnum +/// +/// @return Number of filler lines above lnum +int diff_check_fill(win_T *wp, linenr_T lnum) +{ + // be quick when there are no filler lines + if (!(diff_flags & DIFF_FILLER)) { + return 0; + } + int n = diff_check(wp, lnum); + + if (n <= 0) { + return 0; + } + return n; +} + +/// Set the topline of "towin" to match the position in "fromwin", so that they +/// show the same diff'ed lines. +/// +/// @param fromwin +/// @param towin +void diff_set_topline(win_T *fromwin, win_T *towin) +{ + buf_T *frombuf = fromwin->w_buffer; + linenr_T lnum = fromwin->w_topline; + diff_T *dp; + int max_count; + int i; + + int fromidx = diff_buf_idx(frombuf); + if (fromidx == DB_COUNT) { + // safety check + return; + } + + if (curtab->tp_diff_invalid) { + // update after a big change + ex_diffupdate(NULL); + } + towin->w_topfill = 0; + + // search for a change that includes "lnum" in the list of diffblocks. + for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) { + if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx]) { + break; + } + } + + if (dp == NULL) { + // After last change, compute topline relative to end of file; no + // filler lines. + towin->w_topline = towin->w_buffer->b_ml.ml_line_count + - (frombuf->b_ml.ml_line_count - lnum); + } else { + // Find index for "towin". + int toidx = diff_buf_idx(towin->w_buffer); + + if (toidx == DB_COUNT) { + // safety check + return; + } + towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]); + + if (lnum >= dp->df_lnum[fromidx]) { + // Inside a change: compute filler lines. With three or more + // buffers we need to know the largest count. + max_count = 0; + + for (i = 0; i < DB_COUNT; ++i) { + if ((curtab->tp_diffbuf[i] != NULL) && (max_count < dp->df_count[i])) { + max_count = dp->df_count[i]; + } + } + + if (dp->df_count[toidx] == dp->df_count[fromidx]) { + // same number of lines: use same filler count + towin->w_topfill = fromwin->w_topfill; + } else if (dp->df_count[toidx] > dp->df_count[fromidx]) { + if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx]) { + // more lines in towin and fromwin doesn't show diff + // lines, only filler lines + if (max_count - fromwin->w_topfill >= dp->df_count[toidx]) { + // towin also only shows filler lines + towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx]; + towin->w_topfill = fromwin->w_topfill; + } else { + // towin still has some diff lines to show + towin->w_topline = dp->df_lnum[toidx] + + max_count - fromwin->w_topfill; + } + } + } else if (towin->w_topline >= dp->df_lnum[toidx] + + dp->df_count[toidx]) { + // less lines in towin and no diff lines to show: compute + // filler lines + towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx]; + + if (diff_flags & DIFF_FILLER) { + if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx]) { + // fromwin is also out of diff lines + towin->w_topfill = fromwin->w_topfill; + } else { + // fromwin has some diff lines + towin->w_topfill = dp->df_lnum[fromidx] + max_count - lnum; + } + } + } + } + } + + // safety check (if diff info gets outdated strange things may happen) + towin->w_botfill = FALSE; + + if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count) { + towin->w_topline = towin->w_buffer->b_ml.ml_line_count; + towin->w_botfill = TRUE; + } + + if (towin->w_topline < 1) { + towin->w_topline = 1; + towin->w_topfill = 0; + } + + // When w_topline changes need to recompute w_botline and cursor position + invalidate_botline_win(towin); + changed_line_abv_curs_win(towin); + + check_topfill(towin, FALSE); + (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline, + NULL, TRUE, NULL); +} + +/// This is called when 'diffopt' is changed. +/// +/// @return +int diffopt_changed(void) +{ + int diff_context_new = 6; + int diff_flags_new = 0; + int diff_foldcolumn_new = 2; + + char_u *p = p_dip; + while (*p != NUL) { + if (STRNCMP(p, "filler", 6) == 0) { + p += 6; + diff_flags_new |= DIFF_FILLER; + } else if ((STRNCMP(p, "context:", 8) == 0) && VIM_ISDIGIT(p[8])) { + p += 8; + diff_context_new = getdigits(&p); + } else if (STRNCMP(p, "icase", 5) == 0) { + p += 5; + diff_flags_new |= DIFF_ICASE; + } else if (STRNCMP(p, "iwhite", 6) == 0) { + p += 6; + diff_flags_new |= DIFF_IWHITE; + } else if (STRNCMP(p, "horizontal", 10) == 0) { + p += 10; + diff_flags_new |= DIFF_HORIZONTAL; + } else if (STRNCMP(p, "vertical", 8) == 0) { + p += 8; + diff_flags_new |= DIFF_VERTICAL; + } else if ((STRNCMP(p, "foldcolumn:", 11) == 0) && VIM_ISDIGIT(p[11])) { + p += 11; + diff_foldcolumn_new = getdigits(&p); + } + + if ((*p != ',') && (*p != NUL)) { + return FAIL; + } + + if (*p == ',') { + ++p; + } + } + + // Can't have both "horizontal" and "vertical". + if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL)) { + return FAIL; + } + + // If "icase" or "iwhite" was added or removed, need to update the diff. + if (diff_flags != diff_flags_new) { + tabpage_T *tp; + for (tp = first_tabpage; tp != NULL; tp = tp->tp_next) { + tp->tp_diff_invalid = TRUE; + } + } + + diff_flags = diff_flags_new; + diff_context = diff_context_new; + diff_foldcolumn = diff_foldcolumn_new; + + diff_redraw(TRUE); + + // recompute the scroll binding with the new option value, may + // remove or add filler lines + check_scrollbind((linenr_T)0, 0L); + return OK; +} + +/// Return TRUE if 'diffopt' contains "horizontal". +/// +/// @return TRUE if 'diffopt' contains "horizontal" +int diffopt_horizontal(void) +{ + return (diff_flags & DIFF_HORIZONTAL) != 0; +} + +/// Find the difference within a changed line. +/// +/// @param startp first char of the change +/// @param endp last char of the change +/// +/// @returns TRUE if the line was added, no other buffer has it. +int diff_find_change(win_T *wp, linenr_T lnum, int *startp, int *endp) +{ + char_u *line_new; + int si_org; + int si_new; + int ei_org; + int ei_new; + int added = TRUE; + + // Make a copy of the line, the next ml_get() will invalidate it. + char_u *line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE)); + if (line_org == NULL) { + return FALSE; + } + + int idx = diff_buf_idx(wp->w_buffer); + if (idx == DB_COUNT) { + // cannot happen + free(line_org); + return FALSE; + } + + // search for a change that includes "lnum" in the list of diffblocks. + diff_T *dp; + for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) { + if (lnum <= dp->df_lnum[idx] + dp->df_count[idx]) { + break; + } + } + + if ((dp == NULL) || (diff_check_sanity(curtab, dp) == FAIL)) { + free(line_org); + return FALSE; + } + + int off = lnum - dp->df_lnum[idx]; + int i; + for (i = 0; i < DB_COUNT; ++i) { + if ((curtab->tp_diffbuf[i] != NULL) && (i != idx)) { + // Skip lines that are not in the other change (filler lines). + if (off >= dp->df_count[i]) { + continue; + } + added = FALSE; + line_new = ml_get_buf(curtab->tp_diffbuf[i], + dp->df_lnum[i] + off, FALSE); + + // Search for start of difference + si_org = si_new = 0; + + while (line_org[si_org] != NUL) { + if ((diff_flags & DIFF_IWHITE) + && vim_iswhite(line_org[si_org]) + && vim_iswhite(line_new[si_new])) { + si_org = (int)(skipwhite(line_org + si_org) - line_org); + si_new = (int)(skipwhite(line_new + si_new) - line_new); + } else { + if (line_org[si_org] != line_new[si_new]) { + break; + } + ++si_org; + ++si_new; + } + } + + if (has_mbyte) { + // Move back to first byte of character in both lines (may + // have "nn^" in line_org and "n^ in line_new). + si_org -= (*mb_head_off)(line_org, line_org + si_org); + si_new -= (*mb_head_off)(line_new, line_new + si_new); + } + + if (*startp > si_org) { + *startp = si_org; + } + + // Search for end of difference, if any. + if ((line_org[si_org] != NUL) || (line_new[si_new] != NUL)) { + ei_org = (int)STRLEN(line_org); + ei_new = (int)STRLEN(line_new); + + while (ei_org >= *startp + && ei_new >= si_new + && ei_org >= 0 + && ei_new >= 0) { + if ((diff_flags & DIFF_IWHITE) + && vim_iswhite(line_org[ei_org]) + && vim_iswhite(line_new[ei_new])) { + while (ei_org >= *startp && vim_iswhite(line_org[ei_org])) { + ei_org--; + } + + while (ei_new >= si_new && vim_iswhite(line_new[ei_new])) { + ei_new--; + } + } else { + if (line_org[ei_org] != line_new[ei_new]) { + break; + } + ei_org--; + ei_new--; + } + } + + if (*endp < ei_org) { + *endp = ei_org; + } + } + } + } + + free(line_org); + return added; +} + +/// Return TRUE if line "lnum" is not close to a diff block, this line should +/// be in a fold. +/// +/// @param wp +/// @param lnum +/// +/// @return FALSE if there are no diff blocks at all in this window. +int diff_infold(win_T *wp, linenr_T lnum) +{ + int other = FALSE; + diff_T *dp; + + // Return if 'diff' isn't set. + if (!wp->w_p_diff) { + return FALSE; + } + + int idx = -1; + int i; + for (i = 0; i < DB_COUNT; ++i) { + if (curtab->tp_diffbuf[i] == wp->w_buffer) { + idx = i; + } else if (curtab->tp_diffbuf[i] != NULL) { + other = TRUE; + } + } + + // return here if there are no diffs in the window + if ((idx == -1) || !other) { + return FALSE; + } + + if (curtab->tp_diff_invalid) { + // update after a big change + ex_diffupdate(NULL); + } + + // Return if there are no diff blocks. All lines will be folded. + if (curtab->tp_first_diff == NULL) { + return TRUE; + } + + for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) { + // If this change is below the line there can't be any further match. + if (dp->df_lnum[idx] - diff_context > lnum) { + break; + } + + // If this change ends before the line we have a match. + if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum) { + return FALSE; + } + } + return TRUE; +} + +/// "dp" and "do" commands. +/// +/// @param put +void nv_diffgetput(int put) +{ + exarg_T ea; + ea.arg = (char_u *)""; + if (put) { + ea.cmdidx = CMD_diffput; + } else { + ea.cmdidx = CMD_diffget; + } + ea.addr_count = 0; + ea.line1 = curwin->w_cursor.lnum; + ea.line2 = curwin->w_cursor.lnum; + ex_diffgetput(&ea); +} + +/// ":diffget" and ":diffput" +/// +/// @param eap +void ex_diffgetput(exarg_T *eap) +{ + linenr_T lnum; + int count; + linenr_T off = 0; + diff_T *dp; + diff_T *dprev; + diff_T *dfree; + int i; + int added; + char_u *p; + aco_save_T aco; + buf_T *buf; + int start_skip, end_skip; + int new_count; + int buf_empty; + int found_not_ma = FALSE; + int idx_other; + int idx_from; + int idx_to; + + // Find the current buffer in the list of diff buffers. + int idx_cur = diff_buf_idx(curbuf); + if (idx_cur == DB_COUNT) { + EMSG(_("E99: Current buffer is not in diff mode")); + return; + } + + if (*eap->arg == NUL) { + // No argument: Find the other buffer in the list of diff buffers. + for (idx_other = 0; idx_other < DB_COUNT; ++idx_other) { + if ((curtab->tp_diffbuf[idx_other] != curbuf) + && (curtab->tp_diffbuf[idx_other] != NULL)) { + if ((eap->cmdidx != CMD_diffput) + || curtab->tp_diffbuf[idx_other]->b_p_ma) { + break; + } + found_not_ma = TRUE; + } + } + + if (idx_other == DB_COUNT) { + if (found_not_ma) { + EMSG(_("E793: No other buffer in diff mode is modifiable")); + } else { + EMSG(_("E100: No other buffer in diff mode")); + } + return; + } + + // Check that there isn't a third buffer in the list + for (i = idx_other + 1; i < DB_COUNT; ++i) { + if ((curtab->tp_diffbuf[i] != curbuf) + && (curtab->tp_diffbuf[i] != NULL) + && ((eap->cmdidx != CMD_diffput) || curtab->tp_diffbuf[i]->b_p_ma)) { + EMSG(_("E101: More than two buffers in diff mode, don't know " + "which one to use")); + return; + } + } + } else { + // Buffer number or pattern given. Ignore trailing white space. + p = eap->arg + STRLEN(eap->arg); + while (p > eap->arg && vim_iswhite(p[-1])) { + p--; + } + + for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i) { + } + + if (eap->arg + i == p) { + // digits only + i = atol((char *)eap->arg); + } else { + i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE); + + if (i < 0) { + // error message already given + return; + } + } + buf = buflist_findnr(i); + + if (buf == NULL) { + EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg); + return; + } + + if (buf == curbuf) { + // nothing to do + return; + } + idx_other = diff_buf_idx(buf); + + if (idx_other == DB_COUNT) { + EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg); + return; + } + } + + diff_busy = TRUE; + + // When no range given include the line above or below the cursor. + if (eap->addr_count == 0) { + // Make it possible that ":diffget" on the last line gets line below + // the cursor line when there is no difference above the cursor. + if ((eap->cmdidx == CMD_diffget) + && (eap->line1 == curbuf->b_ml.ml_line_count) + && (diff_check(curwin, eap->line1) == 0) + && ((eap->line1 == 1) || (diff_check(curwin, eap->line1 - 1) == 0))) { + ++eap->line2; + } else if (eap->line1 > 0) { + --eap->line1; + } + } + + if (eap->cmdidx == CMD_diffget) { + idx_from = idx_other; + idx_to = idx_cur; + } else { + idx_from = idx_cur; + idx_to = idx_other; + + // Need to make the other buffer the current buffer to be able to make + // changes in it. + + // set curwin/curbuf to buf and save a few things + aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]); + } + + // May give the warning for a changed buffer here, which can trigger the + // FileChangedRO autocommand, which may do nasty things and mess + // everything up. + if (!curbuf->b_changed) { + change_warning(0); + if (diff_buf_idx(curbuf) != idx_to) { + EMSG(_("E787: Buffer changed unexpectedly")); + return; + } + } + + dprev = NULL; + + for (dp = curtab->tp_first_diff; dp != NULL;) { + if (dp->df_lnum[idx_cur] > eap->line2 + off) { + // past the range that was specified + break; + } + dfree = NULL; + lnum = dp->df_lnum[idx_to]; + count = dp->df_count[idx_to]; + + if ((dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off) + && (u_save(lnum - 1, lnum + count) != FAIL)) { + // Inside the specified range and saving for undo worked. + start_skip = 0; + end_skip = 0; + + if (eap->addr_count > 0) { + // A range was specified: check if lines need to be skipped. + start_skip = eap->line1 + off - dp->df_lnum[idx_cur]; + if (start_skip > 0) { + // range starts below start of current diff block + if (start_skip > count) { + lnum += count; + count = 0; + } else { + count -= start_skip; + lnum += start_skip; + } + } else { + start_skip = 0; + } + + end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1 + - (eap->line2 + off); + + if (end_skip > 0) { + // range ends above end of current/from diff block + if (idx_cur == idx_from) { + // :diffput + i = dp->df_count[idx_cur] - start_skip - end_skip; + + if (count > i) { + count = i; + } + } else { + // :diffget + count -= end_skip; + end_skip = dp->df_count[idx_from] - start_skip - count; + + if (end_skip < 0) { + end_skip = 0; + } + } + } else { + end_skip = 0; + } + } + + buf_empty = FALSE; + added = 0; + + for (i = 0; i < count; ++i) { + // remember deleting the last line of the buffer + buf_empty = curbuf->b_ml.ml_line_count == 1; + ml_delete(lnum, FALSE); + added--; + } + + for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i) { + linenr_T nr = dp->df_lnum[idx_from] + start_skip + i; + if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count) { + break; + } + p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from], nr, FALSE)); + if (p != NULL) { + ml_append(lnum + i - 1, p, 0, FALSE); + free(p); + added++; + if (buf_empty && (curbuf->b_ml.ml_line_count == 2)) { + // Added the first line into an empty buffer, need to + // delete the dummy empty line. + buf_empty = FALSE; + ml_delete((linenr_T)2, FALSE); + } + } + } + new_count = dp->df_count[idx_to] + added; + dp->df_count[idx_to] = new_count; + + if ((start_skip == 0) && (end_skip == 0)) { + // Check if there are any other buffers and if the diff is + // equal in them. + for (i = 0; i < DB_COUNT; ++i) { + if ((curtab->tp_diffbuf[i] != NULL) + && (i != idx_from) + && (i != idx_to) + && !diff_equal_entry(dp, idx_from, i)) { + break; + } + } + + if (i == DB_COUNT) { + // delete the diff entry, the buffers are now equal here + dfree = dp; + dp = dp->df_next; + + if (dprev == NULL) { + curtab->tp_first_diff = dp; + } else { + dprev->df_next = dp; + } + } + } + + // Adjust marks. This will change the following entries! + if (added != 0) { + mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added); + if (curwin->w_cursor.lnum >= lnum) { + // Adjust the cursor position if it's in/after the changed + // lines. + if (curwin->w_cursor.lnum >= lnum + count) { + curwin->w_cursor.lnum += added; + } else if (added < 0) { + curwin->w_cursor.lnum = lnum; + } + } + } + changed_lines(lnum, 0, lnum + count, (long)added); + + if (dfree != NULL) { + // Diff is deleted, update folds in other windows. + diff_fold_update(dfree, idx_to); + free(dfree); + } else { + // mark_adjust() may have changed the count in a wrong way + dp->df_count[idx_to] = new_count; + } + + // When changing the current buffer, keep track of line numbers + if (idx_cur == idx_to) { + off += added; + } + } + + // If before the range or not deleted, go to next diff. + if (dfree == NULL) { + dprev = dp; + dp = dp->df_next; + } + } + + // restore curwin/curbuf and a few other things + if (eap->cmdidx != CMD_diffget) { + // Syncing undo only works for the current buffer, but we change + // another buffer. Sync undo if the command was typed. This isn't + // 100% right when ":diffput" is used in a function or mapping. + if (KeyTyped) { + u_sync(FALSE); + } + aucmd_restbuf(&aco); + } + + diff_busy = FALSE; + + // Check that the cursor is on a valid character and update it's position. + // When there were filler lines the topline has become invalid. + check_cursor(); + changed_line_abv_curs(); + + // Also need to redraw the other buffers. + diff_redraw(FALSE); +} + +/// Update folds for all diff buffers for entry "dp". +/// +/// Skip buffer with index "skip_idx". +/// When there are no diffs, all folds are removed. +/// +/// @param dp +/// @param skip_idx +static void diff_fold_update(diff_T *dp, int skip_idx) +{ + win_T *wp; + for (wp = firstwin; wp != NULL; wp = wp->w_next) { + int i; + for (i = 0; i < DB_COUNT; ++i) { + if ((curtab->tp_diffbuf[i] == wp->w_buffer) && (i != skip_idx)) { + foldUpdate(wp, dp->df_lnum[i], dp->df_lnum[i] + dp->df_count[i]); + } + } + } +} + +/// Checks if the buffer is in diff-mode. +/// +/// @param buf The buffer to check. +/// +/// @return TRUE if buffer "buf" is in diff-mode. +int diff_mode_buf(buf_T *buf) +{ + tabpage_T *tp; + for (tp = first_tabpage; tp != NULL; tp = tp->tp_next) { + if (diff_buf_idx_tp(buf, tp) != DB_COUNT) { + return TRUE; + } + } + return FALSE; +} + +/// Move "count" times in direction "dir" to the next diff block. +/// +/// @param dir +/// @param count +/// +/// @return FAIL if there isn't such a diff block. +int diff_move_to(int dir, long count) +{ + linenr_T lnum = curwin->w_cursor.lnum; + int idx = diff_buf_idx(curbuf); + if ((idx == DB_COUNT) || (curtab->tp_first_diff == NULL)) { + return FAIL; + } + + if (curtab->tp_diff_invalid) { + // update after a big change + ex_diffupdate(NULL); + } + + if (curtab->tp_first_diff == NULL) { + // no diffs today + return FAIL; + } + + while (--count >= 0) { + // Check if already before first diff. + if ((dir == BACKWARD) && (lnum <= curtab->tp_first_diff->df_lnum[idx])) { + break; + } + + diff_T *dp; + for (dp = curtab->tp_first_diff;; dp = dp->df_next) { + if (dp == NULL) { + break; + } + + if (((dir == FORWARD) && (lnum < dp->df_lnum[idx])) + || ((dir == BACKWARD) + && ((dp->df_next == NULL) + || (lnum <= dp->df_next->df_lnum[idx])))) { + lnum = dp->df_lnum[idx]; + break; + } + } + } + + // don't end up past the end of the file + if (lnum > curbuf->b_ml.ml_line_count) { + lnum = curbuf->b_ml.ml_line_count; + } + + // When the cursor didn't move at all we fail. + if (lnum == curwin->w_cursor.lnum) { + return FAIL; + } + + setpcmark(); + curwin->w_cursor.lnum = lnum; + curwin->w_cursor.col = 0; + + return OK; +} + +/// Finds the corresponding line in a diff. +/// +/// @param buf1 +/// @param lnum1 +/// @param buf2 +/// @param lnum3 +/// +/// @return The corresponding line. +linenr_T diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1, buf_T *buf2, + linenr_T lnum3) +{ + int idx1; + int idx2; + diff_T *dp; + int baseline = 0; + linenr_T lnum2; + + idx1 = diff_buf_idx(buf1); + idx2 = diff_buf_idx(buf2); + + if ((idx1 == DB_COUNT) + || (idx2 == DB_COUNT) + || (curtab->tp_first_diff == NULL)) { + return lnum1; + } + + if (curtab->tp_diff_invalid) { + // update after a big change + ex_diffupdate(NULL); + } + + if (curtab->tp_first_diff == NULL) { + // no diffs today + return lnum1; + } + + for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) { + if (dp->df_lnum[idx1] > lnum1) { + lnum2 = lnum1 - baseline; + + // don't end up past the end of the file + if (lnum2 > buf2->b_ml.ml_line_count) { + lnum2 = buf2->b_ml.ml_line_count; + } + + return lnum2; + } else if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1) { + // Inside the diffblock + baseline = lnum1 - dp->df_lnum[idx1]; + + if (baseline > dp->df_count[idx2]) { + baseline = dp->df_count[idx2]; + } + + return dp->df_lnum[idx2] + baseline; + } else if ((dp->df_lnum[idx1] == lnum1) + && (dp->df_count[idx1] == 0) + && (dp->df_lnum[idx2] <= lnum3) + && ((dp->df_lnum[idx2] + dp->df_count[idx2]) > lnum3)) { + // Special case: if the cursor is just after a zero-count + // block (i.e. all filler) and the target cursor is already + // inside the corresponding block, leave the target cursor + // unmoved. This makes repeated CTRL-W W operations work + // as expected. + return lnum3; + } + baseline = (dp->df_lnum[idx1] + dp->df_count[idx1]) + - (dp->df_lnum[idx2] + dp->df_count[idx2]); + } + + // If we get here then the cursor is after the last diff + lnum2 = lnum1 - baseline; + + // don't end up past the end of the file + if (lnum2 > buf2->b_ml.ml_line_count) { + lnum2 = buf2->b_ml.ml_line_count; + } + + return lnum2; +} + +/// For line "lnum" in the current window find the equivalent lnum in window +/// "wp", compensating for inserted/deleted lines. +linenr_T diff_lnum_win(linenr_T lnum, win_T *wp) +{ + diff_T *dp; + int idx; + int i; + linenr_T n; + + idx = diff_buf_idx(curbuf); + + if (idx == DB_COUNT) { + // safety check + return (linenr_T)0; + } + + if (curtab->tp_diff_invalid) { + // update after a big change + ex_diffupdate(NULL); + } + + // search for a change that includes "lnum" in the list of diffblocks. + for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) { + if (lnum <= dp->df_lnum[idx] + dp->df_count[idx]) { + break; + } + } + + // When after the last change, compute relative to the last line number. + if (dp == NULL) { + return wp->w_buffer->b_ml.ml_line_count + - (curbuf->b_ml.ml_line_count - lnum); + } + + // Find index for "wp". + i = diff_buf_idx(wp->w_buffer); + + if (i == DB_COUNT) { + // safety check + return (linenr_T)0; + } + + n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]); + if (n > dp->df_lnum[i] + dp->df_count[i]) { + n = dp->df_lnum[i] + dp->df_count[i]; + } + return n; +} |