provenance, too

Author: molpopgen

src/provenance.rs

@@ -262,6 +262,51 @@ impl ProvenanceTable {
         make_provenance_row(self, row.into().0)
     }
 
+    /// Obtain a [`ProvenanceTableRowView`] for row `row`.
+    ///
+    /// # Returns
+    ///
+    /// * `Some(row view)` if `r` is valid
+    /// * `None` otherwise
+    pub fn row_view<P: Into<ProvenanceId> + Copy>(&self, row: P) -> Option<ProvenanceTableRowView> {
+        match u64::try_from(row.into().0).ok() {
+            Some(x) if x < self.num_rows() => {
+                let record_slice = unsafe_tsk_ragged_char_column_access_to_slice_u8!(
+                    row.into().0,
+                    0,
+                    self.num_rows(),
+                    self.as_ref(),
+                    record,
+                    record_offset,
+                    record_length
+                );
+                let timestamp_slice = unsafe_tsk_ragged_char_column_access_to_slice_u8!(
+                    row.into().0,
+                    0,
+                    self.num_rows(),
+                    self.as_ref(),
+                    timestamp,
+                    timestamp_offset,
+                    timestamp_length
+                );
+                let view = ProvenanceTableRowView {
+                    table: self,
+                    id: row.into(),
+                    record: match record_slice {
+                        Some(r) => std::str::from_utf8(r).unwrap(),
+                        None => "",
+                    },
+                    timestamp: match timestamp_slice {
+                        Some(t) => std::str::from_utf8(t).unwrap(),
+                        None => "",
+                    },
+                };
+                Some(view)
+            }
+            _ => None,
+        }
+    }
+
     /// Return an iterator over rows of the table.
     /// The value of the iterator is [`ProvenanceTableRow`].
     pub fn iter(&self) -> impl Iterator<Item = ProvenanceTableRow> + '_ {