ADD: from sites into sequences

src/lib.rs

@@ -27,9 +27,11 @@ pub mod ffi {
         // model.h
         generate!("argweaver::ArgModel")
         // sequences.h
+        generate!("argweaver::Sequences")
         generate!("argweaver::Sites")
         generate!("argweaver::read_sites")
         generate!("argweaver::read_sites1")
+        generate!("argweaver::make_sequences_from_sites")
     }
     pub use ffi::{argweaver::*, spidir::*, *};
 }

src/sites.rs

@@ -207,6 +207,23 @@ fn parse_locs(input: &str) -> IResult<&str, (Vec<u32>, Vec<Vec<u32>>)> {
     Ok((input, (pos, seqs)))
 }
 
+impl TryInto<UniquePtr<ffi::Sequences>> for Sites {
+    type Error = Box<dyn std::error::Error>;
+
+    fn try_into(self) -> Result<UniquePtr<ffi::Sequences>> {
+        let mut sites: UniquePtr<ffi::Sites> = self.try_into()?;
+        let mut sequences = ffi::Sequences::new(c_int(0)).within_unique_ptr();
+        unsafe {
+            ffi::make_sequences_from_sites(
+                std::pin::Pin::<&mut ffi::Sites>::into_inner_unchecked(sites.pin_mut()),
+                std::pin::Pin::<&mut ffi::Sequences>::into_inner_unchecked(sequences.pin_mut()),
+                'A' as i8,
+            )
+        };
+        Ok(sequences)
+    }
+}
+
 impl Sites {
     pub fn from_path(path: &std::path::PathBuf) -> Result<Self> {
         let content = read_to_string(path)?;
@@ -351,6 +368,10 @@ impl Sites {
             .collect();
         self.data.hstack_mut(&columns).unwrap();
     }
+
+    pub fn set_samples(&mut self, samples: &[&str]) {
+        self.data = self.data.select(&samples.to_vec()).unwrap();
+    }
 }
 
 #[cfg(feature = "extension-module")]