noodles-util 0.18.0

noodles support utilities
Documentation 
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use std::{
    fs::File,
    io::{self, BufReader},
    path::Path,
};

use noodles_bcf as bcf;
use noodles_vcf as vcf;

use super::IndexedReader;
use crate::variant::{Compression, Format};

/// An indexed variant reader builder.
#[derive(Default)]
pub struct Builder {
    compression: Option<Option<Compression>>,
    format: Option<Format>,
}

impl Builder {
    /// Builds an indexed variant reader from a path.
    pub fn build_from_path<P>(self, src: P) -> io::Result<IndexedReader<File>>
    where
        P: AsRef<Path>,
    {
        use crate::variant::reader::builder::{detect_compression, detect_format};

        let mut reader = File::open(src.as_ref()).map(BufReader::new)?;

        let compression = match self.compression {
            Some(compression) => compression,
            None => detect_compression(&mut reader)?,
        };

        let format = match self.format {
            Some(format) => format,
            None => detect_format(&mut reader, compression)?,
        };

        match (format, compression) {
            (Format::Vcf, Some(Compression::Bgzf)) => vcf::indexed_reader::Builder::default()
                .build_from_path(src)
                .map(IndexedReader::Vcf),
            (Format::Bcf, Some(Compression::Bgzf)) => bcf::indexed_reader::Builder::default()
                .build_from_path(src)
                .map(IndexedReader::Bcf),
            (_, None) => Err(io::Error::new(
                io::ErrorKind::InvalidData,
                "source not bgzip-compressed",
            )),
        }
    }
}