pooled_writer/
bgzf.rs

1///! An implementation of [`Compressor`] for the `BGZF` format.
2use crate::Compressor;
3
4/// A BGZF compressor.
5pub struct BgzfCompressor {
6    inner: bgzf::Compressor,
7}
8
9impl Compressor for BgzfCompressor {
10    type Error = bgzf::BgzfError;
11    type CompressionLevel = bgzf::CompressionLevel;
12
13    const BLOCK_SIZE: usize = bgzf::BGZF_BLOCK_SIZE;
14
15    fn new(compression_level: Self::CompressionLevel) -> Self {
16        Self { inner: bgzf::Compressor::new(compression_level) }
17    }
18
19    fn default_compression_level() -> Self::CompressionLevel {
20        bgzf::CompressionLevel::new(5).unwrap()
21    }
22
23    fn new_compression_level(compression_level: u8) -> Result<Self::CompressionLevel, Self::Error> {
24        bgzf::CompressionLevel::new(compression_level)
25    }
26
27    fn compress(
28        &mut self,
29        input: &[u8],
30        output: &mut Vec<u8>,
31        is_last: bool,
32    ) -> Result<(), Self::Error> {
33        self.inner.compress(input, output)?;
34        if is_last {
35            bgzf::Compressor::append_eof(output);
36        }
37        Ok(())
38    }
39}