category iter

src/lib.rs

@@ -169,11 +169,11 @@ mod tests {
 
         let mut variants = load_variants_name(name, &multi)?;
         let v = variants.get_cat(&variants::VariantCategory::Constit);
-        let v = v.iter().filter(|v| 0.25 < v.vaf.unwrap() && v.vaf.unwrap() < 0.75 ).map(|v| {
-
-        }).collect::<Vec<Variant>>();
-
-        lev 
+        // let v = v.iter().filter(|v| 0.25 < v.vaf.unwrap() && v.vaf.unwrap() < 0.75 )
+        //     .map(|v| {
+        //     
+        // }).collect::<Vec<Variant>>();
+        //
 
         variants.write_vcf_cat("test.vcf.gz", &variants::VariantCategory::Somatic)?;
         println!("{} variants loaded from db.", variants.len());

src/variants.rs

@@ -426,7 +426,7 @@ impl Variants {
         pg.finish();
         info!("{}. Executed in {}s", self.stats_vcf, elapsed.as_secs());
     }
-    pub fn get_cat(&mut self, cat: VariantCategory) -> Vec<Variant> {
+    pub fn get_cat(&mut self, cat: &VariantCategory) -> Vec<Variant> {
         let pg = self.mp.add(new_pg_speed(self.data.len() as u64));
         pg.set_message(format!("Get cat {:?}", cat));
         self.data
@@ -436,7 +436,7 @@ impl Variants {
                 if e.annotations
                     .iter()
                     .filter(|e| match e {
-                        AnnotationType::VariantCategory(vc) => *vc == cat,
+                        AnnotationType::VariantCategory(vc) => vc == cat,
                         _ => false,
                     })
                     .count()