tsg 0.1.0

A tool to analyze and manipulate transcript segment graph (TSG)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use std::path::Path;

use crate::graph::TSGraph;
use anyhow::Result;
use std::io::Write;

pub fn to_fa<P: AsRef<Path>, Q: AsRef<Path>>(
    tsg_graph: &mut TSGraph,
    reference_genome_path: P,
    output: Q,
) -> Result<()> {
    tsg_graph.annotate_node_with_sequence(reference_genome_path)?;
    let paths = tsg_graph.traverse()?;

    let output_file = std::fs::File::create(output)?;
    let mut writer = std::io::BufWriter::new(output_file);

    for path in paths {
        let seq = path.to_fa()?;
        writeln!(writer, ">{}", path.get_id().unwrap())?;
        writeln!(writer, "{}", seq)?;
    }
    Ok(())
}