Add old-style DEFLATE support

This commit is contained in:
Masterchef365 committed 2019-10-02 18:51:07 -07:00
1 parent b79128d766
commit ce73d39d57
4 files changed
+44 -1

No files matched your search

+1
View File
@@ -21,6 +21,7 @@ byteorder = "1.2"
lzw = "0.10" lzw = "0.10"
num-derive = "0.3" num-derive = "0.3"
num-traits = "0.2" num-traits = "0.2"
libflate = "0.1" # TODO: Put behind a feature?
[dev-dependencies] [dev-dependencies]
tempfile = "3.0" tempfile = "3.0"
+7 -1
View File
@@ -7,7 +7,7 @@ use {ColorType, TiffError, TiffFormatError, TiffResult, TiffUnsupportedError};
use self::ifd::Directory; use self::ifd::Directory;
use self::stream::{ByteOrder, EndianReader, LZWReader, PackBitsReader, SmartReader}; use self::stream::{ByteOrder, EndianReader, LZWReader, DeflateReader, PackBitsReader, SmartReader};
pub mod ifd; pub mod ifd;
mod stream; mod stream;
@@ -579,6 +579,12 @@ impl<R: Read + Seek> Decoder<R> {
)?; )?;
(bytes, Box::new(reader)) (bytes, Box::new(reader))
} }
CompressionMethod::OldDeflate => {
let (bytes, reader) = DeflateReader::new(
&mut self.reader,
max_uncompressed_length)?;
(bytes, Box::new(reader))
}
method => { method => {
return Err(TiffError::UnsupportedError( return Err(TiffError::UnsupportedError(
TiffUnsupportedError::UnsupportedCompressionMethod(method), TiffUnsupportedError::UnsupportedCompressionMethod(method),
+35
View File
@@ -3,6 +3,7 @@
use byteorder::{BigEndian, LittleEndian, ReadBytesExt}; use byteorder::{BigEndian, LittleEndian, ReadBytesExt};
use lzw; use lzw;
use std::io::{self, Read, Seek}; use std::io::{self, Read, Seek};
use libflate::zlib;
/// Byte order of the TIFF file. /// Byte order of the TIFF file.
#[derive(Clone, Copy, Debug)] #[derive(Clone, Copy, Debug)]
@@ -67,6 +68,40 @@ pub trait EndianReader: Read {
} }
} }
/// Reader that decompresses DEFLATE streams
pub struct DeflateReader {
buffer: io::Cursor<Vec<u8>>,
byte_order: ByteOrder,
}
impl DeflateReader {
pub fn new<R: Read + Seek>(
reader: &mut SmartReader<R>,
max_uncompressed_length: usize,
) -> io::Result<(usize, Self)> {
let byte_order = reader.byte_order;
let mut decoder = zlib::Decoder::new(reader)?;
let mut uncompressed = Vec::with_capacity(max_uncompressed_length);
let bytes_read = decoder.read_to_end(&mut uncompressed)?;
Ok((bytes_read, Self {
byte_order,
buffer: io::Cursor::new(uncompressed),
}))
}
}
impl Read for DeflateReader {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
self.buffer.read(buf)
}
}
impl EndianReader for DeflateReader {
fn byte_order(&self) -> ByteOrder {
self.byte_order
}
}
/// Reader that decompresses LZW streams /// Reader that decompresses LZW streams
pub struct LZWReader { pub struct LZWReader {
buffer: io::Cursor<Vec<u8>>, buffer: io::Cursor<Vec<u8>>,
+1
View File
@@ -8,6 +8,7 @@
extern crate byteorder; extern crate byteorder;
extern crate lzw; extern crate lzw;
extern crate libflate;
#[macro_use] #[macro_use]
extern crate num_derive; extern crate num_derive;
extern crate num_traits; extern crate num_traits;