Compare commits
4
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a8abdd2484 | ||
|
|
934fde185a | ||
|
|
3302936b08 | ||
|
|
a5b9a57e09 |
No files matched your search
+1
-1
@@ -24,7 +24,7 @@ quick-error = "2.0.1"
|
|||||||
fax34 = { package = "fax", version = "0.2.6", optional = true }
|
fax34 = { package = "fax", version = "0.2.6", optional = true }
|
||||||
flate2 = { version = "1.0.20", optional = true }
|
flate2 = { version = "1.0.20", optional = true }
|
||||||
weezl = { version = "0.2.1", optional = true }
|
weezl = { version = "0.2.1", optional = true }
|
||||||
zstd = { version = "0.13", optional = true }
|
zstd = { version = "0.14", optional = true }
|
||||||
zrip-decode = { version = "0.8.2", features = ["paranoid"], optional = true }
|
zrip-decode = { version = "0.8.2", features = ["paranoid"], optional = true }
|
||||||
zune-jpeg = { version = "0.5.1", optional = true }
|
zune-jpeg = { version = "0.5.1", optional = true }
|
||||||
image-webp = {version = "0.2.4", optional = true }
|
image-webp = {version = "0.2.4", optional = true }
|
||||||
|
|||||||
@@ -5,6 +5,8 @@
|
|||||||
|
|
||||||
TIFF decoding and encoding library in pure Rust
|
TIFF decoding and encoding library in pure Rust
|
||||||
|
|
||||||
|
Forked from https://github.com/image-rs/image-tiff. Changes: read images containing RGBPalette frames, ignoring colormaps.
|
||||||
|
|
||||||
## Supported
|
## Supported
|
||||||
|
|
||||||
### Features
|
### Features
|
||||||
|
|||||||
+8
-15
@@ -1,9 +1,8 @@
|
|||||||
use super::ifd::Value;
|
use super::ifd::Value;
|
||||||
use super::stream::{PackBitsReader, ReadSamples};
|
use super::stream::PackBitsReader;
|
||||||
use super::tag_reader::TagReader;
|
use super::tag_reader::TagReader;
|
||||||
use super::ChunkType;
|
use super::ChunkType;
|
||||||
use super::{predict_f16, predict_f32, predict_f64, ValueReader};
|
use super::{predict_f16, predict_f32, predict_f64, ValueReader};
|
||||||
use crate::decoder::stream::CompressionEngines;
|
|
||||||
use crate::tags::{
|
use crate::tags::{
|
||||||
ByteOrder, CompressionMethod, ExtraSamples, PhotometricInterpretation, PlanarConfiguration,
|
ByteOrder, CompressionMethod, ExtraSamples, PhotometricInterpretation, PlanarConfiguration,
|
||||||
Predictor, SampleFormat, Tag,
|
Predictor, SampleFormat, Tag,
|
||||||
@@ -12,7 +11,7 @@ use crate::{
|
|||||||
ColorType, Directory, TiffError, TiffFormatError, TiffResult, TiffUnsupportedError, UsageError,
|
ColorType, Directory, TiffError, TiffFormatError, TiffResult, TiffUnsupportedError, UsageError,
|
||||||
};
|
};
|
||||||
|
|
||||||
use std::io::{Cursor, Read, Seek};
|
use std::io::{self, Cursor, Read, Seek};
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
@@ -177,7 +176,7 @@ impl Image {
|
|||||||
.map(Value::into_u16)
|
.map(Value::into_u16)
|
||||||
.transpose()?
|
.transpose()?
|
||||||
.and_then(PhotometricInterpretation::from_u16)
|
.and_then(PhotometricInterpretation::from_u16)
|
||||||
.ok_or(TiffUnsupportedError::UnknownInterpretation)?;
|
.unwrap_or(PhotometricInterpretation::BlackIsZero);
|
||||||
|
|
||||||
// Try to parse both the compression method and the number, format, and bits of the included samples.
|
// Try to parse both the compression method and the number, format, and bits of the included samples.
|
||||||
// If they are not explicitly specified, those tags are reset to their default values and not carried from previous images.
|
// If they are not explicitly specified, those tags are reset to their default values and not carried from previous images.
|
||||||
@@ -699,8 +698,7 @@ impl Image {
|
|||||||
dimensions: (u32, u32),
|
dimensions: (u32, u32),
|
||||||
samples: u16,
|
samples: u16,
|
||||||
#[cfg_attr(not(feature = "fax"), allow(unused_variables))] fill_order: u16,
|
#[cfg_attr(not(feature = "fax"), allow(unused_variables))] fill_order: u16,
|
||||||
engines: &mut CompressionEngines,
|
) -> TiffResult<Box<dyn Read + 'r>> {
|
||||||
) -> TiffResult<Box<dyn ReadSamples + 'r>> {
|
|
||||||
Ok(match compression_method {
|
Ok(match compression_method {
|
||||||
CompressionMethod::None => Box::new(reader),
|
CompressionMethod::None => Box::new(reader),
|
||||||
CompressionMethod::SgiLog => match samples {
|
CompressionMethod::SgiLog => match samples {
|
||||||
@@ -731,7 +729,6 @@ impl Image {
|
|||||||
CompressionMethod::LZW => Box::new(super::stream::LZWReader::new(
|
CompressionMethod::LZW => Box::new(super::stream::LZWReader::new(
|
||||||
reader,
|
reader,
|
||||||
usize::try_from(compressed_length)?,
|
usize::try_from(compressed_length)?,
|
||||||
engines,
|
|
||||||
)),
|
)),
|
||||||
#[cfg(feature = "zstd")]
|
#[cfg(feature = "zstd")]
|
||||||
CompressionMethod::ZSTD => Box::new(zstd::Decoder::new(reader)?),
|
CompressionMethod::ZSTD => Box::new(zstd::Decoder::new(reader)?),
|
||||||
@@ -1036,7 +1033,6 @@ impl Image {
|
|||||||
layout: &ReadoutLayout,
|
layout: &ReadoutLayout,
|
||||||
chunk_index: u32,
|
chunk_index: u32,
|
||||||
scratch: &mut Vec<u8>,
|
scratch: &mut Vec<u8>,
|
||||||
engines: &mut CompressionEngines,
|
|
||||||
) -> TiffResult<()> {
|
) -> TiffResult<()> {
|
||||||
let ValueReader {
|
let ValueReader {
|
||||||
reader,
|
reader,
|
||||||
@@ -1183,7 +1179,6 @@ impl Image {
|
|||||||
chunk_dims,
|
chunk_dims,
|
||||||
self.samples,
|
self.samples,
|
||||||
self.fill_order,
|
self.fill_order,
|
||||||
engines,
|
|
||||||
)?;
|
)?;
|
||||||
|
|
||||||
// Extended bit depths (9-15, 17-31): packed N-bit samples must be unpacked to
|
// Extended bit depths (9-15, 17-31): packed N-bit samples must be unpacked to
|
||||||
@@ -1313,12 +1308,11 @@ impl Image {
|
|||||||
// Two ways how we get here: we have more bytes in our chunk data than in the image
|
// Two ways how we get here: we have more bytes in our chunk data than in the image
|
||||||
// we are to read. Then we need to skip the rest of the data. Or we have a bigger
|
// we are to read. Then we need to skip the rest of the data. Or we have a bigger
|
||||||
// row stride than the chunk contains data, then we need to fill only the front.
|
// row stride than the chunk contains data, then we need to fill only the front.
|
||||||
|
reader.read_exact(&mut row[..used])?;
|
||||||
|
// Skip horizontal padding
|
||||||
if chunk_row_bytes > data_row_bytes {
|
if chunk_row_bytes > data_row_bytes {
|
||||||
// Skip horizontal padding
|
let len = u64::try_from(chunk_row_bytes - data_row_bytes)?;
|
||||||
let skip = u64::try_from(chunk_row_bytes - data_row_bytes)?;
|
io::copy(&mut reader.by_ref().take(len), &mut io::sink())?;
|
||||||
reader.read_padded_samples(&mut row[..used], skip)?;
|
|
||||||
} else {
|
|
||||||
reader.read_exact(&mut row[..used])?;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
super::fix_endianness_and_predict(
|
super::fix_endianness_and_predict(
|
||||||
@@ -1408,7 +1402,6 @@ impl Image {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
reader.cache_or_destroy(engines);
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -83,14 +83,6 @@ impl<R: Read> Read for LogLuv24<R> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<R: Read> super::stream::ReadSamples for LogLuv24<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> std::io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn packbits_decode<R: Read, const N: usize>(
|
fn packbits_decode<R: Read, const N: usize>(
|
||||||
reader: &mut R,
|
reader: &mut R,
|
||||||
buf: &mut [[u8; N]],
|
buf: &mut [[u8; N]],
|
||||||
@@ -220,14 +212,6 @@ impl<R: Read> Read for LogLuv32<R> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<R: Read> super::stream::ReadSamples for LogLuv32<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> std::io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Pack-Bits like encoding for a single 16-bit log-L sample.
|
/// Pack-Bits like encoding for a single 16-bit log-L sample.
|
||||||
pub struct LogLuv16<R> {
|
pub struct LogLuv16<R> {
|
||||||
reader: Take<R>,
|
reader: Take<R>,
|
||||||
@@ -294,14 +278,6 @@ impl<R: Read> Read for LogLuv16<R> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<R: Read> super::stream::ReadSamples for LogLuv16<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> std::io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Table information from: `libtiff`, 1997 by Greg Ward Larson, SGI
|
// Table information from: `libtiff`, 1997 by Greg Ward Larson, SGI
|
||||||
#[doc(alias = "UV_SQSIZ")]
|
#[doc(alias = "UV_SQSIZ")]
|
||||||
const UV_QUANTIZATION_WIDTH: f32 = 0.003500;
|
const UV_QUANTIZATION_WIDTH: f32 = 0.003500;
|
||||||
|
|||||||
+5
-17
@@ -259,7 +259,6 @@ pub struct Decoder<R> {
|
|||||||
scratch: Vec<u8>,
|
scratch: Vec<u8>,
|
||||||
/// Leniency flag that allows libtiff-like TIFF color interpretation behavior
|
/// Leniency flag that allows libtiff-like TIFF color interpretation behavior
|
||||||
lenient: bool,
|
lenient: bool,
|
||||||
compression: stream::CompressionEngines,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// All the information needed to read and interpret byte slices from the underlying file, i.e. to
|
/// All the information needed to read and interpret byte slices from the underlying file, i.e. to
|
||||||
@@ -854,7 +853,6 @@ impl TiffHeader {
|
|||||||
image: None,
|
image: None,
|
||||||
scratch: Vec::new(),
|
scratch: Vec::new(),
|
||||||
lenient: false,
|
lenient: false,
|
||||||
compression: stream::CompressionEngines::default(),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -951,18 +949,12 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
#[expect(clippy::unnecessary_unwrap)] // Lifetimes would overlap due to return.
|
#[expect(clippy::unnecessary_unwrap)] // Lifetimes would overlap due to return.
|
||||||
fn ensure_image_and_reader(
|
fn ensure_image_and_reader(
|
||||||
&mut self,
|
&mut self,
|
||||||
) -> TiffResult<(
|
) -> TiffResult<(&mut Image, &mut ValueReader<R>, &mut Vec<u8>)> {
|
||||||
&mut Image,
|
|
||||||
&mut ValueReader<R>,
|
|
||||||
&mut Vec<u8>,
|
|
||||||
&mut stream::CompressionEngines,
|
|
||||||
)> {
|
|
||||||
if self.image.is_some() {
|
if self.image.is_some() {
|
||||||
return Ok((
|
return Ok((
|
||||||
self.image.as_mut().unwrap(),
|
self.image.as_mut().unwrap(),
|
||||||
&mut self.value_reader,
|
&mut self.value_reader,
|
||||||
&mut self.scratch,
|
&mut self.scratch,
|
||||||
&mut self.compression,
|
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -971,7 +963,6 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
self.image.insert(image),
|
self.image.insert(image),
|
||||||
&mut self.value_reader,
|
&mut self.value_reader,
|
||||||
&mut self.scratch,
|
&mut self.scratch,
|
||||||
&mut self.compression,
|
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1104,7 +1095,6 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
image: core::mem::take(&mut self.image),
|
image: core::mem::take(&mut self.image),
|
||||||
scratch: core::mem::take(&mut self.scratch),
|
scratch: core::mem::take(&mut self.scratch),
|
||||||
lenient: self.lenient,
|
lenient: self.lenient,
|
||||||
compression: core::mem::take(&mut self.compression),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1588,10 +1578,10 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
chunk_index: u32,
|
chunk_index: u32,
|
||||||
layout: &image::ReadoutLayout,
|
layout: &image::ReadoutLayout,
|
||||||
) -> TiffResult<()> {
|
) -> TiffResult<()> {
|
||||||
let (image, value_reader, scratch, engines) = self.ensure_image_and_reader()?;
|
let (image, value_reader, scratch) = self.ensure_image_and_reader()?;
|
||||||
let offset = image.chunk_file_range(chunk_index)?.0;
|
let offset = image.chunk_file_range(chunk_index)?.0;
|
||||||
value_reader.reader.goto_offset(offset)?;
|
value_reader.reader.goto_offset(offset)?;
|
||||||
image.expand_chunk(value_reader, buffer, layout, chunk_index, scratch, engines)?;
|
image.expand_chunk(value_reader, buffer, layout, chunk_index, scratch)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1743,7 +1733,7 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
slice: TiffCodingUnit,
|
slice: TiffCodingUnit,
|
||||||
buffer: &mut [u8],
|
buffer: &mut [u8],
|
||||||
) -> TiffResult<()> {
|
) -> TiffResult<()> {
|
||||||
let (image, value_reader, scratch, engines) = self.ensure_image_and_reader()?;
|
let (image, value_reader, scratch) = self.ensure_image_and_reader()?;
|
||||||
|
|
||||||
let (width, height) = image.chunk_data_dimensions(slice.0)?;
|
let (width, height) = image.chunk_data_dimensions(slice.0)?;
|
||||||
let readout = image.readout_for_size(width, height)?;
|
let readout = image.readout_for_size(width, height)?;
|
||||||
@@ -1776,7 +1766,6 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
readout,
|
readout,
|
||||||
chunk,
|
chunk,
|
||||||
scratch,
|
scratch,
|
||||||
engines,
|
|
||||||
)?;
|
)?;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1948,7 +1937,7 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
/// Returns an error if the buffer fits less than one plane. In particular, for non-planar
|
/// Returns an error if the buffer fits less than one plane. In particular, for non-planar
|
||||||
/// images returns an error if the buffer does not fit the required size.
|
/// images returns an error if the buffer does not fit the required size.
|
||||||
pub fn read_image_bytes(&mut self, buffer: &mut [u8]) -> TiffResult<()> {
|
pub fn read_image_bytes(&mut self, buffer: &mut [u8]) -> TiffResult<()> {
|
||||||
let (image, value_reader, scratch, engines) = self.ensure_image_and_reader()?;
|
let (image, value_reader, scratch) = self.ensure_image_and_reader()?;
|
||||||
let readout = image.readout_for_image()?;
|
let readout = image.readout_for_image()?;
|
||||||
|
|
||||||
let ref layout @ image::PlaneLayout {
|
let ref layout @ image::PlaneLayout {
|
||||||
@@ -1982,7 +1971,6 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
readout,
|
readout,
|
||||||
chunk,
|
chunk,
|
||||||
scratch,
|
scratch,
|
||||||
engines,
|
|
||||||
)?;
|
)?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+8
-190
@@ -122,86 +122,9 @@ impl<R: Read> EndianReader<R> {
|
|||||||
// # READERS
|
// # READERS
|
||||||
//
|
//
|
||||||
|
|
||||||
#[derive(Default)]
|
|
||||||
pub(crate) struct CompressionEngines {
|
|
||||||
#[cfg(feature = "lzw")]
|
|
||||||
lzw: Option<weezl::decode::Decoder>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl core::fmt::Debug for CompressionEngines {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
let mut dbg = f.debug_struct("CompressionEngines");
|
|
||||||
#[cfg(feature = "lzw")]
|
|
||||||
dbg.field("lzw", &self.lzw.is_some().then_some("<weezl>"));
|
|
||||||
dbg.finish()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// A variant of `io::Read` that takes one block to the end.
|
|
||||||
///
|
|
||||||
/// This is derived from `Read` so that any special optimization done by `std` (or any nightly crate
|
|
||||||
/// for that matter) surely remains usable.
|
|
||||||
///
|
|
||||||
/// We want our decompression engines to be much more specialized than the standard library trait.
|
|
||||||
/// Firstly there is more information available to communicate (the tiff Type, known decompressed
|
|
||||||
/// size, …) and secondly they can potentially *reset*. This could be solved with an extension trait
|
|
||||||
/// but since we want to avoid `read` in favor of `read_exact` (or an equivalent if we want to pass
|
|
||||||
/// in a strided block buffer) a separate trait that's auto derived is favored.
|
|
||||||
///
|
|
||||||
/// FIXME: Also consider that we will want to read subsets of samples. Via IO this is done by
|
|
||||||
/// discarding data *after* it is put into the output buffer. Yet, for some decompression that write
|
|
||||||
/// alone is considerable work where it could simply skip bits more efficiently. So the reader would
|
|
||||||
/// likely want to know about a sample mask.
|
|
||||||
pub trait ReadSamples: Read {
|
|
||||||
/// Read a row of samples and throw away some additional padding samples at the end.
|
|
||||||
///
|
|
||||||
/// This is automatically implemented for `Self: Read` as a copy into `io::Sink`.
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()>;
|
|
||||||
|
|
||||||
/// Put this back into our cache of compression engines.
|
|
||||||
fn cache_or_destroy(self: Box<Self>, _: &mut CompressionEngines) {}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Note: the `Seek` bound ensures this does not overlap actual decompression.. This is an odd
|
|
||||||
/// design problem born out of wanting both the underlying `Read` for uncompressed streams due to
|
|
||||||
/// performance as well as the override to `cache_or_destroy` for all compressed streams.
|
|
||||||
impl<R: Read + Seek> ReadSamples for R {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
if discard > 0 {
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Type alias for the deflate Reader
|
/// Type alias for the deflate Reader
|
||||||
#[cfg(feature = "deflate")]
|
#[cfg(feature = "deflate")]
|
||||||
pub struct DeflateReader<R> {
|
pub type DeflateReader<R> = flate2::read::ZlibDecoder<R>;
|
||||||
inner: flate2::read::ZlibDecoder<R>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<R: Read> DeflateReader<R> {
|
|
||||||
pub fn new(inner: R) -> Self {
|
|
||||||
DeflateReader {
|
|
||||||
inner: flate2::read::ZlibDecoder::new(inner),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<R: Read> Read for DeflateReader<R> {
|
|
||||||
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
|
||||||
self.inner.read(buf)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<R: Read> ReadSamples for DeflateReader<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
//
|
||||||
// ## LZW Reader
|
// ## LZW Reader
|
||||||
@@ -212,17 +135,12 @@ impl<R: Read> ReadSamples for DeflateReader<R> {
|
|||||||
pub struct LZWReader<R: Read> {
|
pub struct LZWReader<R: Read> {
|
||||||
reader: BufReader<Take<R>>,
|
reader: BufReader<Take<R>>,
|
||||||
decoder: weezl::decode::Decoder,
|
decoder: weezl::decode::Decoder,
|
||||||
is_old_style: bool,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(feature = "lzw")]
|
#[cfg(feature = "lzw")]
|
||||||
impl<R: Read> LZWReader<R> {
|
impl<R: Read> LZWReader<R> {
|
||||||
/// Wraps a reader
|
/// Wraps a reader
|
||||||
pub fn new(
|
pub fn new(reader: R, compressed_length: usize) -> LZWReader<R> {
|
||||||
reader: R,
|
|
||||||
compressed_length: usize,
|
|
||||||
engine: &mut CompressionEngines,
|
|
||||||
) -> LZWReader<R> {
|
|
||||||
let mut buffered = BufReader::with_capacity(
|
let mut buffered = BufReader::with_capacity(
|
||||||
(32 * 1024).min(compressed_length),
|
(32 * 1024).min(compressed_length),
|
||||||
reader.take(u64::try_from(compressed_length).unwrap()),
|
reader.take(u64::try_from(compressed_length).unwrap()),
|
||||||
@@ -240,75 +158,26 @@ impl<R: Read> LZWReader<R> {
|
|||||||
Ok(head) if head.len() >= 2 && head[0] == 0x00 && head[1] & 0x01 != 0
|
Ok(head) if head.len() >= 2 && head[0] == 0x00 && head[1] & 0x01 != 0
|
||||||
);
|
);
|
||||||
|
|
||||||
let decoder = if old_style {
|
let configuration = if old_style {
|
||||||
weezl::decode::Configuration::new(weezl::BitOrder::Lsb, 8)
|
weezl::decode::Configuration::new(weezl::BitOrder::Lsb, 8)
|
||||||
.with_yield_on_full_buffer(true)
|
|
||||||
.build()
|
|
||||||
} else {
|
} else {
|
||||||
// All standard LZW decoders have the same configuration. We can thus reuse any that we
|
weezl::decode::Configuration::with_tiff_size_switch(weezl::BitOrder::Msb, 8)
|
||||||
// have previously constructed, skipping internal buffer allocation. We only need to
|
}
|
||||||
// reset them to a new stream.
|
.with_yield_on_full_buffer(true);
|
||||||
if let Some(mut lzw) = engine.lzw.take() {
|
|
||||||
lzw.reset();
|
|
||||||
lzw
|
|
||||||
} else {
|
|
||||||
weezl::decode::Configuration::with_tiff_size_switch(weezl::BitOrder::Msb, 8)
|
|
||||||
.with_yield_on_full_buffer(true)
|
|
||||||
.build()
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
Self {
|
Self {
|
||||||
reader: buffered,
|
reader: buffered,
|
||||||
decoder: decoder,
|
decoder: configuration.build(),
|
||||||
is_old_style: old_style,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn discard_n(&mut self, mut n: u64) -> io::Result<()> {
|
|
||||||
let mut buf = [0u8; 2048];
|
|
||||||
|
|
||||||
while n > 0 {
|
|
||||||
let limit = n.min(2048) as usize;
|
|
||||||
let result = self
|
|
||||||
.decoder
|
|
||||||
.decode_bytes(self.reader.fill_buf()?, &mut buf[..limit]);
|
|
||||||
|
|
||||||
self.reader.consume(result.consumed_in);
|
|
||||||
n -= result.consumed_out as u64;
|
|
||||||
|
|
||||||
match result.status {
|
|
||||||
Ok(weezl::LzwStatus::Ok) => {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
Ok(weezl::LzwStatus::NoProgress) => {
|
|
||||||
return Err(io::Error::new(
|
|
||||||
io::ErrorKind::UnexpectedEof,
|
|
||||||
"no lzw end code found",
|
|
||||||
));
|
|
||||||
}
|
|
||||||
Ok(weezl::LzwStatus::Done) => {
|
|
||||||
if n != 0 {
|
|
||||||
// stream is too short.
|
|
||||||
return Err(std::io::ErrorKind::UnexpectedEof)?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(err) => return Err(io::Error::new(io::ErrorKind::InvalidData, err)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(feature = "lzw")]
|
#[cfg(feature = "lzw")]
|
||||||
impl<R: Read> Read for LZWReader<R> {
|
impl<R: Read> Read for LZWReader<R> {
|
||||||
fn read(&mut self, mut buf: &mut [u8]) -> io::Result<usize> {
|
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
||||||
loop {
|
loop {
|
||||||
let result = self.decoder.decode_bytes(self.reader.fill_buf()?, buf);
|
let result = self.decoder.decode_bytes(self.reader.fill_buf()?, buf);
|
||||||
|
|
||||||
self.reader.consume(result.consumed_in);
|
self.reader.consume(result.consumed_in);
|
||||||
let _ = buf.split_off_mut(..result.consumed_out);
|
|
||||||
|
|
||||||
match result.status {
|
match result.status {
|
||||||
Ok(weezl::LzwStatus::Ok) => {
|
Ok(weezl::LzwStatus::Ok) => {
|
||||||
@@ -333,24 +202,6 @@ impl<R: Read> Read for LZWReader<R> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<R: Read> ReadSamples for LZWReader<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
|
|
||||||
if discard > 0 {
|
|
||||||
self.discard_n(discard)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn cache_or_destroy(self: Box<Self>, engines: &mut CompressionEngines) {
|
|
||||||
if !self.is_old_style && engines.lzw.is_none() {
|
|
||||||
engines.lzw = Some(self.decoder);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
//
|
||||||
// ## PackBits Reader
|
// ## PackBits Reader
|
||||||
//
|
//
|
||||||
@@ -423,14 +274,6 @@ impl<R: Read> Read for PackBitsReader<R> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<R: Read> ReadSamples for PackBitsReader<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Iterator adapter that reverses bits in each byte for FillOrder=2 (LSB-to-MSB).
|
/// Iterator adapter that reverses bits in each byte for FillOrder=2 (LSB-to-MSB).
|
||||||
#[cfg(feature = "fax")]
|
#[cfg(feature = "fax")]
|
||||||
type FaxBytes<R> =
|
type FaxBytes<R> =
|
||||||
@@ -547,14 +390,6 @@ impl<R: Read> Read for Group4Reader<R> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<R: Read> ReadSamples for Group4Reader<R> {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Owning byte reader for Group 3 decoding: the compressed data is consumed
|
/// Owning byte reader for Group 3 decoding: the compressed data is consumed
|
||||||
/// into an iterator so the reader doesn't borrow the buffer.
|
/// into an iterator so the reader doesn't borrow the buffer.
|
||||||
#[cfg(feature = "fax")]
|
#[cfg(feature = "fax")]
|
||||||
@@ -693,14 +528,6 @@ impl Read for Group3Reader {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ReadSamples for Group3Reader {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "webp")]
|
#[cfg(feature = "webp")]
|
||||||
pub struct WebPReader {
|
pub struct WebPReader {
|
||||||
inner: Cursor<Vec<u8>>,
|
inner: Cursor<Vec<u8>>,
|
||||||
@@ -756,15 +583,6 @@ impl Read for WebPReader {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(feature = "webp")]
|
|
||||||
impl ReadSamples for WebPReader {
|
|
||||||
fn read_padded_samples(&mut self, buf: &mut [u8], discard: u64) -> io::Result<()> {
|
|
||||||
self.read_exact(buf)?;
|
|
||||||
std::io::copy(&mut self.by_ref().take(discard), &mut std::io::sink())?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|||||||
Reference in new issue
Block a user