provenance

Author: molpopgen

src/_macros.rs

@@ -987,6 +987,26 @@ macro_rules! migration_table_add_row_with_metadata {
         }
     };
 }
+
+macro_rules! provenance_table_add_row {
+    ($(#[$attr:meta])* => $name: ident, $self: ident, $table: expr) => {
+        $(#[$attr])*
+        pub fn $name(&mut $self, record: &str) -> Result<$crate::ProvenanceId, $crate::TskitError> {
+            let timestamp = humantime::format_rfc3339(std::time::SystemTime::now()).to_string();
+            let rv = unsafe {
+                $crate::bindings::tsk_provenance_table_add_row(
+                    &mut $table,
+                    timestamp.as_ptr() as *mut i8,
+                    timestamp.len() as tsk_size_t,
+                    record.as_ptr() as *mut i8,
+                    record.len() as tsk_size_t,
+                )
+            };
+            handle_tsk_return_value!(rv, rv.into())
+        }
+    };
+}
+
 #[cfg(test)]
 mod test {
     use crate::error::TskitError;

src/table_collection.rs

@@ -808,6 +808,7 @@ impl TableCollection {
     }
 
     #[cfg(any(feature = "provenance", doc))]
+    provenance_table_add_row!(
     /// Add provenance record with a time stamp.
     ///
     /// All implementation of this trait provided by `tskit` use
@@ -856,19 +857,7 @@ impl TableCollection {
     /// assert_eq!(treeseq.provenances().record(0).unwrap(), row_0.record);
     /// # }
     /// ```
-    pub fn add_provenance(&mut self, record: &str) -> Result<crate::ProvenanceId, TskitError> {
-        let timestamp = humantime::format_rfc3339(std::time::SystemTime::now()).to_string();
-        let rv = unsafe {
-            ll_bindings::tsk_provenance_table_add_row(
-                &mut (*self.inner).provenances,
-                timestamp.as_ptr() as *mut i8,
-                timestamp.len() as tsk_size_t,
-                record.as_ptr() as *mut i8,
-                record.len() as tsk_size_t,
-            )
-        };
-        handle_tsk_return_value!(rv, crate::ProvenanceId::from(rv))
-    }
+    => add_provenance, self, (*self.inner).provenances);
 }
 
 impl TableAccess for TableCollection {