tsg 0.1.1

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
25
26
27
28
29
30
31

use anyhow::Result;
use std::path::Path;

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

pub fn to_json<P: AsRef<Path>>(tsg_graph: &TSGraph, pretty: bool, output: P) -> Result<()> {
    let output_file = std::fs::File::create(output)?;
    let mut writer = std::io::BufWriter::new(output_file);
    let json = tsg_graph.to_json()?;

    if pretty {
        let json = serde_json::to_string_pretty(&json)?;
        writeln!(writer, "{}", json)?;
        return Ok(());
    }
    writeln!(writer, "{}", json)?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_to_gtf() {
        let tsg_graph = TSGraph::from_file("tests/data/test.tsg").unwrap();
        let output = "tests/data/test.json";
        to_json(&tsg_graph, true, output).unwrap();
    }
}