set_migrations

Author: molpopgen

src/table_collection.rs

@@ -1008,7 +1008,7 @@ impl TableCollection {
         handle_tsk_return_value!(rv)
     }
 
-    /// Set the individual table from an [`OwnedMutationTable`](`crate::OwnedSiteTable`)
+    /// Set the individual table from an [`OwnedIndividualTable`](`crate::OwnedSiteTable`)
     ///
     /// # Errors
     ///
@@ -1024,7 +1024,7 @@ impl TableCollection {
     /// tables.set_individuals(&individuals).unwrap();
     /// assert_eq!(tables.individuals().num_rows(), 1);
     /// let expected = vec![tskit::Location::from(0.1), tskit::Location::from(10.0)];
-    /// assert_eq!(tables.individuals().location(0).unwrap(), Some(expected));
+    /// assert_eq!(tables.individuals().location(0).unwrap(), Some(expected.as_slice()));
     /// # individuals.clear().unwrap();
     /// # assert_eq!(individuals.num_rows(), 0);
     /// ```
@@ -1047,7 +1047,43 @@ impl TableCollection {
         handle_tsk_return_value!(rv)
     }
 
-    pub fn set_migrations(&mut self, migrations: &crate::OwnedMigrationTable) -> TskReturnValue {}
+
+    /// Set the migration table from an [`OwnedMigrationTable`](`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 migrations = tskit::OwnedMigrationTable::default();
+    /// migrations.add_row((0.25, 0.37), 1, (0, 1), 111.0).unwrap();
+    /// tables.set_migrations(&migrations).unwrap();
+    /// assert_eq!(tables.migrations().num_rows(), 1);
+    /// assert_eq!(tables.migrations().time(0).unwrap(), 111.0);
+    /// # migrations.clear().unwrap();
+    /// # assert_eq!(migrations.num_rows(), 0);
+    /// ```
+    pub fn set_migrations(&mut self, migrations: &crate::OwnedMigrationTable) -> TskReturnValue {
+        let rv = unsafe {
+            ll_bindings::tsk_migration_table_set_columns(
+                &mut (*self.inner).migrations,
+                (*migrations.as_ptr()).num_rows,
+                (*migrations.as_ptr()).left,
+                (*migrations.as_ptr()).right,
+                (*migrations.as_ptr()).node,
+                (*migrations.as_ptr()).source,
+                (*migrations.as_ptr()).dest,
+                (*migrations.as_ptr()).time,
+                (*migrations.as_ptr()).metadata,
+                (*migrations.as_ptr()).metadata_offset,
+            )
+        };
+        handle_tsk_return_value!(rv)
+    }
 }
 
 impl TableAccess for TableCollection {