Proposal for table set_row in C

Author: jeromekelleher

c/tskit/tables.c

@@ -984,6 +984,53 @@ tsk_individual_table_get_row(
     return ret;
 }
 
+int
+tsk_individual_table_set_row(
+    tsk_individual_table_t *self, tsk_id_t index, const tsk_individual_t *row)
+{
+    int ret = 0;
+    tsk_id_t ret_id;
+    tsk_individual_table_t copy;
+    tsk_id_t j;
+    tsk_size_t num_rows;
+    tsk_id_t *rows = NULL;
+
+    ret = tsk_individual_table_copy(self, &copy, 0);
+    if (ret != 0) {
+        goto out;
+    }
+    rows = malloc(self->num_rows * sizeof(*rows));
+    if (rows == NULL) {
+        ret = TSK_ERR_NO_MEMORY;
+        goto out;
+    }
+
+    ret = tsk_individual_table_truncate(self, (tsk_size_t) index);
+    if (ret != 0) {
+        goto out;
+    }
+    ret_id = tsk_individual_table_add_row(self, row->flags, row->location,
+        row->location_length, row->parents, self->parents_length, self->metadata,
+        self->metadata_length);
+    if (ret_id < 0) {
+        ret = ret_id;
+        goto out;
+    }
+    num_rows = 0;
+    for (j = index + 1; j < (tsk_id_t) self->num_rows; j++) {
+        rows[num_rows] = j;
+        num_rows++;
+    }
+    ret = tsk_individual_table_extend(self, &copy, num_rows, rows, 0);
+    if (ret != 0) {
+        goto out;
+    }
+out:
+    tsk_individual_table_free(&copy);
+    tsk_safe_free(rows);
+    return ret;
+}
+
 int
 tsk_individual_table_set_metadata_schema(tsk_individual_table_t *self,
     const char *metadata_schema, tsk_size_t metadata_schema_length)

c/tskit/tables.h

@@ -957,6 +957,9 @@ next operation that modifies the table (e.g., by adding a new row), but not afte
 int tsk_individual_table_get_row(
     const tsk_individual_table_t *self, tsk_id_t index, tsk_individual_t *row);
 
+int tsk_individual_table_set_row(
+    tsk_individual_table_t *self, tsk_id_t index, const tsk_individual_t *row);
+
 /**
 @brief Set the metadata schema