Switch to TABLE_INCREF

2 files changed, 1 insertions, 6 deletions

diff --git a/builtins/table.c b/builtins/table.c
index e0c94593..f3f97ec3 100644
--- a/builtins/table.c
+++ b/builtins/table.c
@@ -107,12 +107,6 @@ static void maybe_copy_on_write(table_t *t, const TypeInfo *type)
     }
 }
 
-public void Table$mark_copy_on_write(table_t *t)
-{
-    ARRAY_INCREF(t->entries);
-    if (t->bucket_info) t->bucket_info->data_refcount = TABLE_MAX_DATA_REFCOUNT;
-}
-
 // Return address of value or NULL
 public void *Table$get_raw(table_t t, const void *key, const TypeInfo *type)
 {
diff --git a/builtins/table.h b/builtins/table.h
index 82e12286..e136f460 100644
--- a/builtins/table.h
+++ b/builtins/table.h
@@ -41,6 +41,7 @@ void Table$remove(table_t *t, const void *key, const TypeInfo *type);
 void Table$clear(table_t *t);
 table_t Table$sorted(table_t t, const TypeInfo *type);
 void Table$mark_copy_on_write(table_t *t);
+#define TABLE_INCREF(t) ({ ARRAY_INCREF((t).entries); if ((t).bucket_info) (t).bucket_info.data_refcount += ((t).bucket_info.data_refcount < TABLE_MAX_DATA_REFCOUNT); })
 int32_t Table$compare(const table_t *x, const table_t *y, const TypeInfo *type);
 bool Table$equal(const table_t *x, const table_t *y, const TypeInfo *type);
 uint32_t Table$hash(const table_t *t, const TypeInfo *type);