| /* vi:set ts=8 sts=4 sw=4 noet: |
| * |
| * VIM - Vi IMproved by Bram Moolenaar |
| * |
| * Do ":help uganda" in Vim to read copying and usage conditions. |
| * Do ":help credits" in Vim to see a list of people who contributed. |
| * See README.txt for an overview of the Vim source code. |
| */ |
| |
| /* |
| * Text properties implementation. See ":help text-properties". |
| * |
| * TODO: |
| * - Adjust text property column and length when text is inserted/deleted. |
| * -> a :substitute with a multi-line match |
| * -> search for changed_bytes() from misc1.c |
| * -> search for mark_col_adjust() |
| * - Perhaps we only need TP_FLAG_CONT_NEXT and can drop TP_FLAG_CONT_PREV? |
| * - Add an arrray for global_proptypes, to quickly lookup a prop type by ID |
| * - Add an arrray for b_proptypes, to quickly lookup a prop type by ID |
| * - Checking the text length to detect text properties is slow. Use a flag in |
| * the index, like DB_MARKED? |
| * - Also test line2byte() with many lines, so that ml_updatechunk() is taken |
| * into account. |
| * - Perhaps have a window-local option to disable highlighting from text |
| * properties? |
| */ |
| |
| #include "vim.h" |
| |
| #if defined(FEAT_TEXT_PROP) || defined(PROTO) |
| |
| /* |
| * In a hashtable item "hi_key" points to "pt_name" in a proptype_T. |
| * This avoids adding a pointer to the hashtable item. |
| * PT2HIKEY() converts a proptype pointer to a hashitem key pointer. |
| * HIKEY2PT() converts a hashitem key pointer to a proptype pointer. |
| * HI2PT() converts a hashitem pointer to a proptype pointer. |
| */ |
| #define PT2HIKEY(p) ((p)->pt_name) |
| #define HIKEY2PT(p) ((proptype_T *)((p) - offsetof(proptype_T, pt_name))) |
| #define HI2PT(hi) HIKEY2PT((hi)->hi_key) |
| |
| // The global text property types. |
| static hashtab_T *global_proptypes = NULL; |
| |
| // The last used text property type ID. |
| static int proptype_id = 0; |
| |
| static char_u e_type_not_exist[] = N_("E971: Property type %s does not exist"); |
| static char_u e_invalid_col[] = N_("E964: Invalid column number: %ld"); |
| static char_u e_invalid_lnum[] = N_("E966: Invalid line number: %ld"); |
| |
| /* |
| * Find a property type by name, return the hashitem. |
| * Returns NULL if the item can't be found. |
| */ |
| static hashitem_T * |
| find_prop_hi(char_u *name, buf_T *buf) |
| { |
| hashtab_T *ht; |
| hashitem_T *hi; |
| |
| if (*name == NUL) |
| return NULL; |
| if (buf == NULL) |
| ht = global_proptypes; |
| else |
| ht = buf->b_proptypes; |
| |
| if (ht == NULL) |
| return NULL; |
| hi = hash_find(ht, name); |
| if (HASHITEM_EMPTY(hi)) |
| return NULL; |
| return hi; |
| } |
| |
| /* |
| * Like find_prop_hi() but return the property type. |
| */ |
| static proptype_T * |
| find_prop(char_u *name, buf_T *buf) |
| { |
| hashitem_T *hi = find_prop_hi(name, buf); |
| |
| if (hi == NULL) |
| return NULL; |
| return HI2PT(hi); |
| } |
| |
| /* |
| * Lookup a property type by name. First in "buf" and when not found in the |
| * global types. |
| * When not found gives an error message and returns NULL. |
| */ |
| static proptype_T * |
| lookup_prop_type(char_u *name, buf_T *buf) |
| { |
| proptype_T *type = find_prop(name, buf); |
| |
| if (type == NULL) |
| type = find_prop(name, NULL); |
| if (type == NULL) |
| semsg(_(e_type_not_exist), name); |
| return type; |
| } |
| |
| /* |
| * Get an optional "bufnr" item from the dict in "arg". |
| * When the argument is not used or "bufnr" is not present then "buf" is |
| * unchanged. |
| * If "bufnr" is valid or not present return OK. |
| * When "arg" is not a dict or "bufnr" is invalide return FAIL. |
| */ |
| static int |
| get_bufnr_from_arg(typval_T *arg, buf_T **buf) |
| { |
| dictitem_T *di; |
| |
| if (arg->v_type != VAR_DICT) |
| { |
| emsg(_(e_dictreq)); |
| return FAIL; |
| } |
| if (arg->vval.v_dict == NULL) |
| return OK; // NULL dict is like an empty dict |
| di = dict_find(arg->vval.v_dict, (char_u *)"bufnr", -1); |
| if (di != NULL) |
| { |
| *buf = tv_get_buf(&di->di_tv, FALSE); |
| if (*buf == NULL) |
| return FAIL; |
| } |
| return OK; |
| } |
| |
| /* |
| * prop_add({lnum}, {col}, {props}) |
| */ |
| void |
| f_prop_add(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| linenr_T lnum; |
| linenr_T start_lnum; |
| linenr_T end_lnum; |
| colnr_T start_col; |
| colnr_T end_col; |
| dict_T *dict; |
| char_u *type_name; |
| proptype_T *type; |
| buf_T *buf = curbuf; |
| int id = 0; |
| char_u *newtext; |
| int proplen; |
| size_t textlen; |
| char_u *props = NULL; |
| char_u *newprops; |
| textprop_T tmp_prop; |
| int i; |
| |
| start_lnum = tv_get_number(&argvars[0]); |
| start_col = tv_get_number(&argvars[1]); |
| if (start_col < 1) |
| { |
| semsg(_(e_invalid_col), (long)start_col); |
| return; |
| } |
| if (argvars[2].v_type != VAR_DICT) |
| { |
| emsg(_(e_dictreq)); |
| return; |
| } |
| dict = argvars[2].vval.v_dict; |
| |
| if (dict == NULL || dict_find(dict, (char_u *)"type", -1) == NULL) |
| { |
| emsg(_("E965: missing property type name")); |
| return; |
| } |
| type_name = dict_get_string(dict, (char_u *)"type", FALSE); |
| |
| if (dict_find(dict, (char_u *)"end_lnum", -1) != NULL) |
| { |
| end_lnum = dict_get_number(dict, (char_u *)"end_lnum"); |
| if (end_lnum < start_lnum) |
| { |
| semsg(_(e_invargval), "end_lnum"); |
| return; |
| } |
| } |
| else |
| end_lnum = start_lnum; |
| |
| if (dict_find(dict, (char_u *)"length", -1) != NULL) |
| { |
| long length = dict_get_number(dict, (char_u *)"length"); |
| |
| if (length < 0 || end_lnum > start_lnum) |
| { |
| semsg(_(e_invargval), "length"); |
| return; |
| } |
| end_col = start_col + length; |
| } |
| else if (dict_find(dict, (char_u *)"end_col", -1) != NULL) |
| { |
| end_col = dict_get_number(dict, (char_u *)"end_col"); |
| if (end_col <= 0) |
| { |
| semsg(_(e_invargval), "end_col"); |
| return; |
| } |
| } |
| else if (start_lnum == end_lnum) |
| end_col = start_col; |
| else |
| end_col = 1; |
| |
| if (dict_find(dict, (char_u *)"id", -1) != NULL) |
| id = dict_get_number(dict, (char_u *)"id"); |
| |
| if (get_bufnr_from_arg(&argvars[2], &buf) == FAIL) |
| return; |
| |
| type = lookup_prop_type(type_name, buf); |
| if (type == NULL) |
| return; |
| |
| if (start_lnum < 1 || start_lnum > buf->b_ml.ml_line_count) |
| { |
| semsg(_(e_invalid_lnum), (long)start_lnum); |
| return; |
| } |
| if (end_lnum < start_lnum || end_lnum > buf->b_ml.ml_line_count) |
| { |
| semsg(_(e_invalid_lnum), (long)end_lnum); |
| return; |
| } |
| |
| for (lnum = start_lnum; lnum <= end_lnum; ++lnum) |
| { |
| colnr_T col; // start column |
| long length; // in bytes |
| |
| // Fetch the line to get the ml_line_len field updated. |
| proplen = get_text_props(buf, lnum, &props, TRUE); |
| textlen = buf->b_ml.ml_line_len - proplen * sizeof(textprop_T); |
| |
| if (lnum == start_lnum) |
| col = start_col; |
| else |
| col = 1; |
| if (col - 1 > (colnr_T)textlen) |
| { |
| semsg(_(e_invalid_col), (long)start_col); |
| return; |
| } |
| |
| if (lnum == end_lnum) |
| length = end_col - col; |
| else |
| length = (int)textlen - col + 1; |
| if (length > (long)textlen) |
| length = (int)textlen; // can include the end-of-line |
| if (length < 0) |
| length = 0; // zero-width property |
| |
| // Allocate the new line with space for the new proprety. |
| newtext = alloc(buf->b_ml.ml_line_len + sizeof(textprop_T)); |
| if (newtext == NULL) |
| return; |
| // Copy the text, including terminating NUL. |
| mch_memmove(newtext, buf->b_ml.ml_line_ptr, textlen); |
| |
| // Find the index where to insert the new property. |
| // Since the text properties are not aligned properly when stored with the |
| // text, we need to copy them as bytes before using it as a struct. |
| for (i = 0; i < proplen; ++i) |
| { |
| mch_memmove(&tmp_prop, props + i * sizeof(textprop_T), |
| sizeof(textprop_T)); |
| if (tmp_prop.tp_col >= col) |
| break; |
| } |
| newprops = newtext + textlen; |
| if (i > 0) |
| mch_memmove(newprops, props, sizeof(textprop_T) * i); |
| |
| tmp_prop.tp_col = col; |
| tmp_prop.tp_len = length; |
| tmp_prop.tp_id = id; |
| tmp_prop.tp_type = type->pt_id; |
| tmp_prop.tp_flags = (lnum > start_lnum ? TP_FLAG_CONT_PREV : 0) |
| | (lnum < end_lnum ? TP_FLAG_CONT_NEXT : 0); |
| mch_memmove(newprops + i * sizeof(textprop_T), &tmp_prop, |
| sizeof(textprop_T)); |
| |
| if (i < proplen) |
| mch_memmove(newprops + (i + 1) * sizeof(textprop_T), |
| props + i * sizeof(textprop_T), |
| sizeof(textprop_T) * (proplen - i)); |
| |
| if (buf->b_ml.ml_flags & ML_LINE_DIRTY) |
| vim_free(buf->b_ml.ml_line_ptr); |
| buf->b_ml.ml_line_ptr = newtext; |
| buf->b_ml.ml_line_len += sizeof(textprop_T); |
| buf->b_ml.ml_flags |= ML_LINE_DIRTY; |
| } |
| |
| buf->b_has_textprop = TRUE; // this is never reset |
| redraw_buf_later(buf, NOT_VALID); |
| } |
| |
| /* |
| * Fetch the text properties for line "lnum" in buffer "buf". |
| * Returns the number of text properties and, when non-zero, a pointer to the |
| * first one in "props" (note that it is not aligned, therefore the char_u |
| * pointer). |
| */ |
| int |
| get_text_props(buf_T *buf, linenr_T lnum, char_u **props, int will_change) |
| { |
| char_u *text; |
| size_t textlen; |
| size_t proplen; |
| |
| // Be quick when no text property types have been defined or the buffer, |
| // unless we are adding one. |
| if (!buf->b_has_textprop && !will_change) |
| return 0; |
| |
| // Fetch the line to get the ml_line_len field updated. |
| text = ml_get_buf(buf, lnum, will_change); |
| textlen = STRLEN(text) + 1; |
| proplen = buf->b_ml.ml_line_len - textlen; |
| if (proplen % sizeof(textprop_T) != 0) |
| { |
| iemsg(_("E967: text property info corrupted")); |
| return 0; |
| } |
| if (proplen > 0) |
| *props = text + textlen; |
| return (int)(proplen / sizeof(textprop_T)); |
| } |
| |
| /* |
| * Set the text properties for line "lnum" to "props" with length "len". |
| * If "len" is zero text properties are removed, "props" is not used. |
| * Any existing text properties are dropped. |
| * Only works for the current buffer. |
| */ |
| static void |
| set_text_props(linenr_T lnum, char_u *props, int len) |
| { |
| char_u *text; |
| char_u *newtext; |
| int textlen; |
| |
| text = ml_get(lnum); |
| textlen = (int)STRLEN(text) + 1; |
| newtext = alloc(textlen + len); |
| if (newtext == NULL) |
| return; |
| mch_memmove(newtext, text, textlen); |
| if (len > 0) |
| mch_memmove(newtext + textlen, props, len); |
| if (curbuf->b_ml.ml_flags & ML_LINE_DIRTY) |
| vim_free(curbuf->b_ml.ml_line_ptr); |
| curbuf->b_ml.ml_line_ptr = newtext; |
| curbuf->b_ml.ml_line_len = textlen + len; |
| curbuf->b_ml.ml_flags |= ML_LINE_DIRTY; |
| } |
| |
| static proptype_T * |
| find_type_by_id(hashtab_T *ht, int id) |
| { |
| long todo; |
| hashitem_T *hi; |
| |
| if (ht == NULL) |
| return NULL; |
| |
| // TODO: Make this faster by keeping a list of types sorted on ID and use |
| // a binary search. |
| |
| todo = (long)ht->ht_used; |
| for (hi = ht->ht_array; todo > 0; ++hi) |
| { |
| if (!HASHITEM_EMPTY(hi)) |
| { |
| proptype_T *prop = HI2PT(hi); |
| |
| if (prop->pt_id == id) |
| return prop; |
| --todo; |
| } |
| } |
| return NULL; |
| } |
| |
| /* |
| * Find a property type by ID in "buf" or globally. |
| * Returns NULL if not found. |
| */ |
| proptype_T * |
| text_prop_type_by_id(buf_T *buf, int id) |
| { |
| proptype_T *type; |
| |
| type = find_type_by_id(buf->b_proptypes, id); |
| if (type == NULL) |
| type = find_type_by_id(global_proptypes, id); |
| return type; |
| } |
| |
| /* |
| * prop_clear({lnum} [, {lnum_end} [, {bufnr}]]) |
| */ |
| void |
| f_prop_clear(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| linenr_T start = tv_get_number(&argvars[0]); |
| linenr_T end = start; |
| linenr_T lnum; |
| buf_T *buf = curbuf; |
| |
| if (argvars[1].v_type != VAR_UNKNOWN) |
| { |
| end = tv_get_number(&argvars[1]); |
| if (argvars[2].v_type != VAR_UNKNOWN) |
| { |
| if (get_bufnr_from_arg(&argvars[2], &buf) == FAIL) |
| return; |
| } |
| } |
| if (start < 1 || end < 1) |
| { |
| emsg(_(e_invrange)); |
| return; |
| } |
| |
| for (lnum = start; lnum <= end; ++lnum) |
| { |
| char_u *text; |
| size_t len; |
| |
| if (lnum > buf->b_ml.ml_line_count) |
| break; |
| text = ml_get_buf(buf, lnum, FALSE); |
| len = STRLEN(text) + 1; |
| if ((size_t)buf->b_ml.ml_line_len > len) |
| { |
| if (!(buf->b_ml.ml_flags & ML_LINE_DIRTY)) |
| { |
| char_u *newtext = vim_strsave(text); |
| |
| // need to allocate the line now |
| if (newtext == NULL) |
| return; |
| buf->b_ml.ml_line_ptr = newtext; |
| buf->b_ml.ml_flags |= ML_LINE_DIRTY; |
| } |
| buf->b_ml.ml_line_len = (int)len; |
| } |
| } |
| redraw_buf_later(buf, NOT_VALID); |
| } |
| |
| /* |
| * prop_list({lnum} [, {bufnr}]) |
| */ |
| void |
| f_prop_list(typval_T *argvars, typval_T *rettv) |
| { |
| linenr_T lnum = tv_get_number(&argvars[0]); |
| buf_T *buf = curbuf; |
| |
| if (argvars[1].v_type != VAR_UNKNOWN) |
| { |
| if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) |
| return; |
| } |
| if (lnum < 1 || lnum > buf->b_ml.ml_line_count) |
| { |
| emsg(_(e_invrange)); |
| return; |
| } |
| |
| if (rettv_list_alloc(rettv) == OK) |
| { |
| char_u *text = ml_get_buf(buf, lnum, FALSE); |
| size_t textlen = STRLEN(text) + 1; |
| int count = (int)((buf->b_ml.ml_line_len - textlen) |
| / sizeof(textprop_T)); |
| int i; |
| textprop_T prop; |
| proptype_T *pt; |
| |
| for (i = 0; i < count; ++i) |
| { |
| dict_T *d = dict_alloc(); |
| |
| if (d == NULL) |
| break; |
| mch_memmove(&prop, text + textlen + i * sizeof(textprop_T), |
| sizeof(textprop_T)); |
| dict_add_number(d, "col", prop.tp_col); |
| dict_add_number(d, "length", prop.tp_len); |
| dict_add_number(d, "id", prop.tp_id); |
| dict_add_number(d, "start", !(prop.tp_flags & TP_FLAG_CONT_PREV)); |
| dict_add_number(d, "end", !(prop.tp_flags & TP_FLAG_CONT_NEXT)); |
| pt = text_prop_type_by_id(buf, prop.tp_type); |
| if (pt != NULL) |
| dict_add_string(d, "type", pt->pt_name); |
| |
| list_append_dict(rettv->vval.v_list, d); |
| } |
| } |
| } |
| |
| /* |
| * prop_remove({props} [, {lnum} [, {lnum_end}]]) |
| */ |
| void |
| f_prop_remove(typval_T *argvars, typval_T *rettv) |
| { |
| linenr_T start = 1; |
| linenr_T end = 0; |
| linenr_T lnum; |
| dict_T *dict; |
| buf_T *buf = curbuf; |
| dictitem_T *di; |
| int do_all = FALSE; |
| int id = -1; |
| int type_id = -1; |
| |
| rettv->vval.v_number = 0; |
| if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL) |
| { |
| emsg(_(e_invarg)); |
| return; |
| } |
| |
| if (argvars[1].v_type != VAR_UNKNOWN) |
| { |
| start = tv_get_number(&argvars[1]); |
| end = start; |
| if (argvars[2].v_type != VAR_UNKNOWN) |
| end = tv_get_number(&argvars[2]); |
| if (start < 1 || end < 1) |
| { |
| emsg(_(e_invrange)); |
| return; |
| } |
| } |
| |
| dict = argvars[0].vval.v_dict; |
| di = dict_find(dict, (char_u *)"bufnr", -1); |
| if (di != NULL) |
| { |
| buf = tv_get_buf(&di->di_tv, FALSE); |
| if (buf == NULL) |
| return; |
| } |
| |
| di = dict_find(dict, (char_u*)"all", -1); |
| if (di != NULL) |
| do_all = dict_get_number(dict, (char_u *)"all"); |
| |
| if (dict_find(dict, (char_u *)"id", -1) != NULL) |
| id = dict_get_number(dict, (char_u *)"id"); |
| if (dict_find(dict, (char_u *)"type", -1)) |
| { |
| char_u *name = dict_get_string(dict, (char_u *)"type", FALSE); |
| proptype_T *type = lookup_prop_type(name, buf); |
| |
| if (type == NULL) |
| return; |
| type_id = type->pt_id; |
| } |
| if (id == -1 && type_id == -1) |
| { |
| emsg(_("E968: Need at least one of 'id' or 'type'")); |
| return; |
| } |
| |
| if (end == 0) |
| end = buf->b_ml.ml_line_count; |
| for (lnum = start; lnum <= end; ++lnum) |
| { |
| char_u *text; |
| size_t len; |
| |
| if (lnum > buf->b_ml.ml_line_count) |
| break; |
| text = ml_get_buf(buf, lnum, FALSE); |
| len = STRLEN(text) + 1; |
| if ((size_t)buf->b_ml.ml_line_len > len) |
| { |
| static textprop_T textprop; // static because of alignment |
| unsigned idx; |
| |
| for (idx = 0; idx < (buf->b_ml.ml_line_len - len) |
| / sizeof(textprop_T); ++idx) |
| { |
| char_u *cur_prop = buf->b_ml.ml_line_ptr + len |
| + idx * sizeof(textprop_T); |
| size_t taillen; |
| |
| mch_memmove(&textprop, cur_prop, sizeof(textprop_T)); |
| if (textprop.tp_id == id || textprop.tp_type == type_id) |
| { |
| if (!(buf->b_ml.ml_flags & ML_LINE_DIRTY)) |
| { |
| char_u *newptr = alloc(buf->b_ml.ml_line_len); |
| |
| // need to allocate the line to be able to change it |
| if (newptr == NULL) |
| return; |
| mch_memmove(newptr, buf->b_ml.ml_line_ptr, |
| buf->b_ml.ml_line_len); |
| buf->b_ml.ml_line_ptr = newptr; |
| buf->b_ml.ml_flags |= ML_LINE_DIRTY; |
| |
| cur_prop = buf->b_ml.ml_line_ptr + len |
| + idx * sizeof(textprop_T); |
| } |
| |
| taillen = buf->b_ml.ml_line_len - len |
| - (idx + 1) * sizeof(textprop_T); |
| if (taillen > 0) |
| mch_memmove(cur_prop, cur_prop + sizeof(textprop_T), |
| taillen); |
| buf->b_ml.ml_line_len -= sizeof(textprop_T); |
| --idx; |
| |
| ++rettv->vval.v_number; |
| if (!do_all) |
| break; |
| } |
| } |
| } |
| } |
| redraw_buf_later(buf, NOT_VALID); |
| } |
| |
| /* |
| * Common for f_prop_type_add() and f_prop_type_change(). |
| */ |
| void |
| prop_type_set(typval_T *argvars, int add) |
| { |
| char_u *name; |
| buf_T *buf = NULL; |
| dict_T *dict; |
| dictitem_T *di; |
| proptype_T *prop; |
| |
| name = tv_get_string(&argvars[0]); |
| if (*name == NUL) |
| { |
| emsg(_(e_invarg)); |
| return; |
| } |
| |
| if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) |
| return; |
| dict = argvars[1].vval.v_dict; |
| |
| prop = find_prop(name, buf); |
| if (add) |
| { |
| hashtab_T **htp; |
| |
| if (prop != NULL) |
| { |
| semsg(_("E969: Property type %s already defined"), name); |
| return; |
| } |
| prop = (proptype_T *)alloc_clear((int)(sizeof(proptype_T) + STRLEN(name))); |
| if (prop == NULL) |
| return; |
| STRCPY(prop->pt_name, name); |
| prop->pt_id = ++proptype_id; |
| htp = buf == NULL ? &global_proptypes : &buf->b_proptypes; |
| if (*htp == NULL) |
| { |
| *htp = (hashtab_T *)alloc(sizeof(hashtab_T)); |
| if (*htp == NULL) |
| { |
| vim_free(prop); |
| return; |
| } |
| hash_init(*htp); |
| } |
| hash_add(*htp, PT2HIKEY(prop)); |
| } |
| else |
| { |
| if (prop == NULL) |
| { |
| semsg(_(e_type_not_exist), name); |
| return; |
| } |
| } |
| |
| if (dict != NULL) |
| { |
| di = dict_find(dict, (char_u *)"highlight", -1); |
| if (di != NULL) |
| { |
| char_u *highlight; |
| int hl_id = 0; |
| |
| highlight = dict_get_string(dict, (char_u *)"highlight", FALSE); |
| if (highlight != NULL && *highlight != NUL) |
| hl_id = syn_name2id(highlight); |
| if (hl_id <= 0) |
| { |
| semsg(_("E970: Unknown highlight group name: '%s'"), |
| highlight == NULL ? (char_u *)"" : highlight); |
| return; |
| } |
| prop->pt_hl_id = hl_id; |
| } |
| |
| di = dict_find(dict, (char_u *)"combine", -1); |
| if (di != NULL) |
| { |
| if (tv_get_number(&di->di_tv)) |
| prop->pt_flags |= PT_FLAG_COMBINE; |
| else |
| prop->pt_flags &= ~PT_FLAG_COMBINE; |
| } |
| |
| di = dict_find(dict, (char_u *)"priority", -1); |
| if (di != NULL) |
| prop->pt_priority = tv_get_number(&di->di_tv); |
| |
| di = dict_find(dict, (char_u *)"start_incl", -1); |
| if (di != NULL) |
| { |
| if (tv_get_number(&di->di_tv)) |
| prop->pt_flags |= PT_FLAG_INS_START_INCL; |
| else |
| prop->pt_flags &= ~PT_FLAG_INS_START_INCL; |
| } |
| |
| di = dict_find(dict, (char_u *)"end_incl", -1); |
| if (di != NULL) |
| { |
| if (tv_get_number(&di->di_tv)) |
| prop->pt_flags |= PT_FLAG_INS_END_INCL; |
| else |
| prop->pt_flags &= ~PT_FLAG_INS_END_INCL; |
| } |
| } |
| } |
| |
| /* |
| * prop_type_add({name}, {props}) |
| */ |
| void |
| f_prop_type_add(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| prop_type_set(argvars, TRUE); |
| } |
| |
| /* |
| * prop_type_change({name}, {props}) |
| */ |
| void |
| f_prop_type_change(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| prop_type_set(argvars, FALSE); |
| } |
| |
| /* |
| * prop_type_delete({name} [, {bufnr}]) |
| */ |
| void |
| f_prop_type_delete(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| char_u *name; |
| buf_T *buf = NULL; |
| hashitem_T *hi; |
| |
| name = tv_get_string(&argvars[0]); |
| if (*name == NUL) |
| { |
| emsg(_(e_invarg)); |
| return; |
| } |
| |
| if (argvars[1].v_type != VAR_UNKNOWN) |
| { |
| if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) |
| return; |
| } |
| |
| hi = find_prop_hi(name, buf); |
| if (hi != NULL) |
| { |
| hashtab_T *ht; |
| proptype_T *prop = HI2PT(hi); |
| |
| if (buf == NULL) |
| ht = global_proptypes; |
| else |
| ht = buf->b_proptypes; |
| hash_remove(ht, hi); |
| vim_free(prop); |
| } |
| } |
| |
| /* |
| * prop_type_get({name} [, {bufnr}]) |
| */ |
| void |
| f_prop_type_get(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| char_u *name = tv_get_string(&argvars[0]); |
| |
| if (*name == NUL) |
| { |
| emsg(_(e_invarg)); |
| return; |
| } |
| if (rettv_dict_alloc(rettv) == OK) |
| { |
| proptype_T *prop = NULL; |
| buf_T *buf = NULL; |
| |
| if (argvars[1].v_type != VAR_UNKNOWN) |
| { |
| if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) |
| return; |
| } |
| |
| prop = find_prop(name, buf); |
| if (prop != NULL) |
| { |
| dict_T *d = rettv->vval.v_dict; |
| |
| if (prop->pt_hl_id > 0) |
| dict_add_string(d, "highlight", syn_id2name(prop->pt_hl_id)); |
| dict_add_number(d, "priority", prop->pt_priority); |
| dict_add_number(d, "combine", |
| (prop->pt_flags & PT_FLAG_COMBINE) ? 1 : 0); |
| dict_add_number(d, "start_incl", |
| (prop->pt_flags & PT_FLAG_INS_START_INCL) ? 1 : 0); |
| dict_add_number(d, "end_incl", |
| (prop->pt_flags & PT_FLAG_INS_END_INCL) ? 1 : 0); |
| if (buf != NULL) |
| dict_add_number(d, "bufnr", buf->b_fnum); |
| } |
| } |
| } |
| |
| static void |
| list_types(hashtab_T *ht, list_T *l) |
| { |
| long todo; |
| hashitem_T *hi; |
| |
| todo = (long)ht->ht_used; |
| for (hi = ht->ht_array; todo > 0; ++hi) |
| { |
| if (!HASHITEM_EMPTY(hi)) |
| { |
| proptype_T *prop = HI2PT(hi); |
| |
| list_append_string(l, prop->pt_name, -1); |
| --todo; |
| } |
| } |
| } |
| |
| /* |
| * prop_type_list([{bufnr}]) |
| */ |
| void |
| f_prop_type_list(typval_T *argvars, typval_T *rettv UNUSED) |
| { |
| buf_T *buf = NULL; |
| |
| if (rettv_list_alloc(rettv) == OK) |
| { |
| if (argvars[0].v_type != VAR_UNKNOWN) |
| { |
| if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL) |
| return; |
| } |
| if (buf == NULL) |
| { |
| if (global_proptypes != NULL) |
| list_types(global_proptypes, rettv->vval.v_list); |
| } |
| else if (buf->b_proptypes != NULL) |
| list_types(buf->b_proptypes, rettv->vval.v_list); |
| } |
| } |
| |
| /* |
| * Free all property types in "ht". |
| */ |
| static void |
| clear_ht_prop_types(hashtab_T *ht) |
| { |
| long todo; |
| hashitem_T *hi; |
| |
| if (ht == NULL) |
| return; |
| |
| todo = (long)ht->ht_used; |
| for (hi = ht->ht_array; todo > 0; ++hi) |
| { |
| if (!HASHITEM_EMPTY(hi)) |
| { |
| proptype_T *prop = HI2PT(hi); |
| |
| vim_free(prop); |
| --todo; |
| } |
| } |
| |
| hash_clear(ht); |
| vim_free(ht); |
| } |
| |
| #if defined(EXITFREE) || defined(PROTO) |
| /* |
| * Free all global property types. |
| */ |
| void |
| clear_global_prop_types(void) |
| { |
| clear_ht_prop_types(global_proptypes); |
| global_proptypes = NULL; |
| } |
| #endif |
| |
| /* |
| * Free all property types for "buf". |
| */ |
| void |
| clear_buf_prop_types(buf_T *buf) |
| { |
| clear_ht_prop_types(buf->b_proptypes); |
| buf->b_proptypes = NULL; |
| } |
| |
| /* |
| * Adjust the columns of text properties in line "lnum" after position "col" to |
| * shift by "bytes_added" (can be negative). |
| * Note that "col" is zero-based, while tp_col is one-based. |
| * Only for the current buffer. |
| * When "save_for_undo" is TRUE then call u_savesub() before making changes to |
| * the line. |
| * Caller is expected to check b_has_textprop and "bytes_added" being non-zero. |
| * Returns TRUE when props were changed. |
| */ |
| int |
| adjust_prop_columns( |
| linenr_T lnum, |
| colnr_T col, |
| int bytes_added, |
| int save_for_undo) |
| { |
| int proplen; |
| char_u *props; |
| textprop_T tmp_prop; |
| proptype_T *pt; |
| int dirty = FALSE; |
| int ri, wi; |
| size_t textlen; |
| |
| if (text_prop_frozen > 0) |
| return FALSE; |
| |
| proplen = get_text_props(curbuf, lnum, &props, TRUE); |
| if (proplen == 0) |
| return FALSE; |
| textlen = curbuf->b_ml.ml_line_len - proplen * sizeof(textprop_T); |
| |
| wi = 0; // write index |
| for (ri = 0; ri < proplen; ++ri) |
| { |
| mch_memmove(&tmp_prop, props + ri * sizeof(textprop_T), |
| sizeof(textprop_T)); |
| pt = text_prop_type_by_id(curbuf, tmp_prop.tp_type); |
| |
| if (bytes_added > 0 |
| ? (tmp_prop.tp_col >= col |
| + (pt != NULL && (pt->pt_flags & PT_FLAG_INS_START_INCL) |
| ? 2 : 1)) |
| : (tmp_prop.tp_col > col + 1)) |
| { |
| if (tmp_prop.tp_col + bytes_added < col + 1) |
| { |
| tmp_prop.tp_len += (tmp_prop.tp_col - 1 - col) + bytes_added; |
| tmp_prop.tp_col = col + 1; |
| } |
| else |
| tmp_prop.tp_col += bytes_added; |
| // Save for undo if requested and not done yet. |
| if (save_for_undo && !dirty) |
| u_savesub(lnum); |
| dirty = TRUE; |
| if (tmp_prop.tp_len <= 0) |
| continue; // drop this text property |
| } |
| else if (tmp_prop.tp_len > 0 |
| && tmp_prop.tp_col + tmp_prop.tp_len > col |
| + ((pt != NULL && (pt->pt_flags & PT_FLAG_INS_END_INCL)) |
| ? 0 : 1)) |
| { |
| int after = col - bytes_added |
| - (tmp_prop.tp_col - 1 + tmp_prop.tp_len); |
| if (after > 0) |
| tmp_prop.tp_len += bytes_added + after; |
| else |
| tmp_prop.tp_len += bytes_added; |
| // Save for undo if requested and not done yet. |
| if (save_for_undo && !dirty) |
| u_savesub(lnum); |
| dirty = TRUE; |
| if (tmp_prop.tp_len <= 0) |
| continue; // drop this text property |
| } |
| mch_memmove(props + wi * sizeof(textprop_T), &tmp_prop, |
| sizeof(textprop_T)); |
| ++wi; |
| } |
| if (dirty) |
| { |
| colnr_T newlen = (int)textlen + wi * (colnr_T)sizeof(textprop_T); |
| |
| if ((curbuf->b_ml.ml_flags & ML_LINE_DIRTY) == 0) |
| curbuf->b_ml.ml_line_ptr = |
| vim_memsave(curbuf->b_ml.ml_line_ptr, newlen); |
| curbuf->b_ml.ml_flags |= ML_LINE_DIRTY; |
| curbuf->b_ml.ml_line_len = newlen; |
| } |
| return dirty; |
| } |
| |
| /* |
| * Adjust text properties for a line that was split in two. |
| * "lnum_props" is the line that has the properties from before the split. |
| * "lnum_top" is the top line. |
| * "kept" is the number of bytes kept in the first line, while |
| * "deleted" is the number of bytes deleted. |
| */ |
| void |
| adjust_props_for_split( |
| linenr_T lnum_props, |
| linenr_T lnum_top, |
| int kept, |
| int deleted) |
| { |
| char_u *props; |
| int count; |
| garray_T prevprop; |
| garray_T nextprop; |
| int i; |
| int skipped = kept + deleted; |
| |
| if (!curbuf->b_has_textprop) |
| return; |
| |
| // Get the text properties from "lnum_props". |
| count = get_text_props(curbuf, lnum_props, &props, FALSE); |
| ga_init2(&prevprop, sizeof(textprop_T), 10); |
| ga_init2(&nextprop, sizeof(textprop_T), 10); |
| |
| // Keep the relevant ones in the first line, reducing the length if needed. |
| // Copy the ones that include the split to the second line. |
| // Move the ones after the split to the second line. |
| for (i = 0; i < count; ++i) |
| { |
| textprop_T prop; |
| textprop_T *p; |
| |
| // copy the prop to an aligned structure |
| mch_memmove(&prop, props + i * sizeof(textprop_T), sizeof(textprop_T)); |
| |
| if (prop.tp_col < kept && ga_grow(&prevprop, 1) == OK) |
| { |
| p = ((textprop_T *)prevprop.ga_data) + prevprop.ga_len; |
| *p = prop; |
| if (p->tp_col + p->tp_len >= kept) |
| p->tp_len = kept - p->tp_col; |
| ++prevprop.ga_len; |
| } |
| |
| // Only add the property to the next line if the length is bigger than |
| // zero. |
| if (prop.tp_col + prop.tp_len > skipped && ga_grow(&nextprop, 1) == OK) |
| { |
| p = ((textprop_T *)nextprop.ga_data) + nextprop.ga_len; |
| *p = prop; |
| if (p->tp_col > skipped) |
| p->tp_col -= skipped - 1; |
| else |
| { |
| p->tp_len -= skipped - p->tp_col; |
| p->tp_col = 1; |
| } |
| ++nextprop.ga_len; |
| } |
| } |
| |
| set_text_props(lnum_top, prevprop.ga_data, |
| prevprop.ga_len * sizeof(textprop_T)); |
| ga_clear(&prevprop); |
| set_text_props(lnum_top + 1, nextprop.ga_data, |
| nextprop.ga_len * sizeof(textprop_T)); |
| ga_clear(&nextprop); |
| } |
| |
| /* |
| * Line "lnum" has been joined and will end up at column "col" in the new line. |
| * "removed" bytes have been removed from the start of the line, properties |
| * there are to be discarded. |
| * Move the adjusted text properties to an allocated string, store it in |
| * "prop_line" and adjust the columns. |
| */ |
| void |
| adjust_props_for_join( |
| linenr_T lnum, |
| textprop_T **prop_line, |
| int *prop_length, |
| long col, |
| int removed) |
| { |
| int proplen; |
| char_u *props; |
| int ri; |
| int wi = 0; |
| |
| proplen = get_text_props(curbuf, lnum, &props, FALSE); |
| if (proplen > 0) |
| { |
| *prop_line = (textprop_T *)alloc(proplen * (int)sizeof(textprop_T)); |
| if (*prop_line != NULL) |
| { |
| for (ri = 0; ri < proplen; ++ri) |
| { |
| textprop_T *cp = *prop_line + wi; |
| |
| mch_memmove(cp, props + ri * sizeof(textprop_T), |
| sizeof(textprop_T)); |
| if (cp->tp_col + cp->tp_len > removed) |
| { |
| if (cp->tp_col > removed) |
| cp->tp_col += col; |
| else |
| { |
| // property was partly deleted, make it shorter |
| cp->tp_len -= removed - cp->tp_col; |
| cp->tp_col = col; |
| } |
| ++wi; |
| } |
| } |
| } |
| *prop_length = wi; |
| } |
| } |
| |
| /* |
| * After joining lines: concatenate the text and the properties of all joined |
| * lines into one line and replace the line. |
| */ |
| void |
| join_prop_lines( |
| linenr_T lnum, |
| char_u *newp, |
| textprop_T **prop_lines, |
| int *prop_lengths, |
| int count) |
| { |
| size_t proplen = 0; |
| size_t oldproplen; |
| char_u *props; |
| int i; |
| int len; |
| char_u *line; |
| size_t l; |
| |
| for (i = 0; i < count - 1; ++i) |
| proplen += prop_lengths[i]; |
| if (proplen == 0) |
| { |
| ml_replace(lnum, newp, FALSE); |
| return; |
| } |
| |
| // get existing properties of the joined line |
| oldproplen = get_text_props(curbuf, lnum, &props, FALSE); |
| |
| len = (int)STRLEN(newp) + 1; |
| line = alloc(len + (oldproplen + proplen) * (int)sizeof(textprop_T)); |
| if (line == NULL) |
| return; |
| mch_memmove(line, newp, len); |
| l = oldproplen * sizeof(textprop_T); |
| mch_memmove(line + len, props, l); |
| len += l; |
| |
| for (i = 0; i < count - 1; ++i) |
| if (prop_lines[i] != NULL) |
| { |
| l = prop_lengths[i] * sizeof(textprop_T); |
| mch_memmove(line + len, prop_lines[i], l); |
| len += l; |
| vim_free(prop_lines[i]); |
| } |
| |
| ml_replace_len(lnum, line, len, TRUE, FALSE); |
| vim_free(newp); |
| vim_free(prop_lines); |
| vim_free(prop_lengths); |
| } |
| |
| #endif // FEAT_TEXT_PROP |