Bladeren bron

charts un pipe

Your Name 1 jaar geleden
bovenliggende
commit
4852387c1a
1 gewijzigde bestanden met toevoegingen van 5 en 3 verwijderingen
  1. 5 3
      src/variants.rs

+ 5 - 3
src/variants.rs

@@ -1075,7 +1075,7 @@ impl Variants {
         Ok(())
     }
 
-    pub fn stats_json(&self, path: &str) -> Result<()> {
+    pub fn stats_json(&self, path: &str) -> Result<AllStats> {
         let variants_stats = self.stats()?;
         let all_stats = AllStats {
             variants_stats,
@@ -1085,7 +1085,7 @@ impl Variants {
         };
         let s = serde_json::to_string(&all_stats)?;
         fs::write(path, s)?;
-        Ok(())
+        Ok(all_stats)
     }
 
     pub fn save_bytes(&self, path: &str) -> Result<()> {
@@ -1973,6 +1973,7 @@ pub fn run_pipe(name: &str, multi: &MultiProgress) -> Result<()> {
     let af_init_path = format!("{}/{name}_variants_af_init.csv", report_data_dir);
 
     let af_final_path = format!("{}/{name}_variants_af_final.csv", report_data_dir);
+    let graphs_prefix = format!("{}/{name}_barcharts", report_data_dir);
 
     let sources = vec![
         (
@@ -2041,9 +2042,10 @@ pub fn run_pipe(name: &str, multi: &MultiProgress) -> Result<()> {
     variants.filter_snp()?;
 
     variants.save_bytes(&bytes_path)?;
-    variants
+    let all_stats = variants
         .stats_json(&stats_path)
         .context("Can't write stats")?;
+    all_stats.generate_graph(&graphs_prefix)?;
 
     write_af_data(&variants, &af_final_path).context("Can't write final AF data")?;