set_populations

Author: molpopgen

src/table_collection.rs

@@ -1047,7 +1047,6 @@ impl TableCollection {
         handle_tsk_return_value!(rv)
     }
 
-
     /// Set the migration table from an [`OwnedMigrationTable`](`crate::OwnedSiteTable`)
     ///
     /// # Errors
@@ -1084,6 +1083,36 @@ impl TableCollection {
         };
         handle_tsk_return_value!(rv)
     }
+
+    /// Set the population table from an [`OwnedPopulationTable`](`crate::OwnedSiteTable`)
+    ///
+    /// # Errors
+    ///
+    /// Any errors from the C API propagate.
+    ///
+    /// # Example
+    ///
+    /// ```rust
+    /// # use tskit::TableAccess;
+    /// let mut tables = tskit::TableCollection::new(1.0).unwrap();
+    /// let mut populations = tskit::OwnedPopulationTable::default();
+    /// populations.add_row().unwrap();
+    /// tables.set_populations(&populations).unwrap();
+    /// assert_eq!(tables.populations().num_rows(), 1);
+    /// # populations.clear().unwrap();
+    /// # assert_eq!(populations.num_rows(), 0);
+    /// ```
+    pub fn set_populations(&mut self, populations: &crate::OwnedPopulationTable) -> TskReturnValue {
+        let rv = unsafe {
+            ll_bindings::tsk_population_table_set_columns(
+                &mut (*self.inner).populations,
+                (*populations.as_ptr()).num_rows,
+                (*populations.as_ptr()).metadata,
+                (*populations.as_ptr()).metadata_offset,
+            )
+        };
+        handle_tsk_return_value!(rv)
+    }
 }
 
 impl TableAccess for TableCollection {