Updated create_reader to only require &mut SmartReader and not &mut Self
This commit is contained in:
1 parent
2cf33d634d
commit
9e6f9771d3
2 files changed
+32
-21
No files matched your search
+31
-18
@@ -1031,7 +1031,17 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
self.goto_offset_u64(offset)?;
|
self.goto_offset_u64(offset)?;
|
||||||
let byte_order = self.reader.byte_order;
|
let byte_order = self.reader.byte_order;
|
||||||
|
|
||||||
let reader = Self::create_reader(self, length)?;
|
let photometric_interpretation = self.image().photometric_interpretation;
|
||||||
|
let compression_method = self.image().compression_method;
|
||||||
|
let jpeg_tables = self.image().jpeg_tables.clone();
|
||||||
|
|
||||||
|
let reader = Self::create_reader(
|
||||||
|
&mut self.reader,
|
||||||
|
photometric_interpretation,
|
||||||
|
compression_method,
|
||||||
|
length,
|
||||||
|
jpeg_tables,
|
||||||
|
)?;
|
||||||
|
|
||||||
// Read into output buffer.
|
// Read into output buffer.
|
||||||
{
|
{
|
||||||
@@ -1089,8 +1099,16 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
|
|
||||||
let line_samples = output_width * self.image().bits_per_sample.len();
|
let line_samples = output_width * self.image().bits_per_sample.len();
|
||||||
let photometric_interpretation = self.image().photometric_interpretation;
|
let photometric_interpretation = self.image().photometric_interpretation;
|
||||||
|
let compression_method = self.image().compression_method;
|
||||||
|
let jpeg_tables = self.image().jpeg_tables.clone();
|
||||||
|
|
||||||
let mut reader = Self::create_reader(self, compressed_length)?;
|
let mut reader = Self::create_reader(
|
||||||
|
&mut self.reader,
|
||||||
|
photometric_interpretation,
|
||||||
|
compression_method,
|
||||||
|
compressed_length,
|
||||||
|
jpeg_tables,
|
||||||
|
)?;
|
||||||
|
|
||||||
for row in 0..(tile_length - padding_down) {
|
for row in 0..(tile_length - padding_down) {
|
||||||
let buf = match &mut buffer {
|
let buf = match &mut buffer {
|
||||||
@@ -1129,35 +1147,30 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn create_reader<'r>(
|
fn create_reader<'r>(
|
||||||
decoder: &'r mut Self,
|
reader: &'r mut SmartReader<R>,
|
||||||
|
photometric_interpretation: PhotometricInterpretation,
|
||||||
|
compression_method: CompressionMethod,
|
||||||
compressed_length: u64,
|
compressed_length: u64,
|
||||||
|
jpeg_tables: Option<Vec<u8>>,
|
||||||
) -> TiffResult<Box<dyn Read + 'r>> {
|
) -> TiffResult<Box<dyn Read + 'r>> {
|
||||||
let compression_method = decoder.image().compression_method;
|
|
||||||
|
|
||||||
Ok(match compression_method {
|
Ok(match compression_method {
|
||||||
CompressionMethod::None => Box::new(&mut decoder.reader),
|
CompressionMethod::None => Box::new(reader),
|
||||||
CompressionMethod::LZW => Box::new(LZWReader::new(
|
CompressionMethod::LZW => {
|
||||||
&mut decoder.reader,
|
Box::new(LZWReader::new(reader, usize::try_from(compressed_length)?))
|
||||||
usize::try_from(compressed_length)?,
|
|
||||||
)),
|
|
||||||
CompressionMethod::PackBits => {
|
|
||||||
Box::new(PackBitsReader::new(&mut decoder.reader, compressed_length))
|
|
||||||
}
|
}
|
||||||
|
CompressionMethod::PackBits => Box::new(PackBitsReader::new(reader, compressed_length)),
|
||||||
CompressionMethod::Deflate | CompressionMethod::OldDeflate => {
|
CompressionMethod::Deflate | CompressionMethod::OldDeflate => {
|
||||||
Box::new(DeflateReader::new(&mut decoder.reader))
|
Box::new(DeflateReader::new(reader))
|
||||||
}
|
}
|
||||||
CompressionMethod::ModernJPEG => {
|
CompressionMethod::ModernJPEG => {
|
||||||
if decoder.image().jpeg_tables.is_some() && compressed_length < 2 {
|
if jpeg_tables.is_some() && compressed_length < 2 {
|
||||||
return Err(TiffError::FormatError(
|
return Err(TiffError::FormatError(
|
||||||
TiffFormatError::InvalidTagValueType(Tag::JPEGTables),
|
TiffFormatError::InvalidTagValueType(Tag::JPEGTables),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let jpeg_tables = decoder.image().jpeg_tables.clone();
|
|
||||||
let photometric_interpretation = decoder.image().photometric_interpretation;
|
|
||||||
|
|
||||||
let jpeg_reader = JpegReader::new(
|
let jpeg_reader = JpegReader::new(
|
||||||
&mut decoder.reader,
|
reader,
|
||||||
compressed_length,
|
compressed_length,
|
||||||
jpeg_tables,
|
jpeg_tables,
|
||||||
&photometric_interpretation,
|
&photometric_interpretation,
|
||||||
|
|||||||
@@ -390,10 +390,8 @@ fn oom() {
|
|||||||
let err = decoder.read_image().unwrap_err();
|
let err = decoder.read_image().unwrap_err();
|
||||||
|
|
||||||
match err {
|
match err {
|
||||||
tiff::TiffError::LimitsExceeded => {}
|
|
||||||
tiff::TiffError::UnsupportedError(tiff::TiffUnsupportedError::InterpretationWithBits(
|
tiff::TiffError::UnsupportedError(tiff::TiffUnsupportedError::InterpretationWithBits(
|
||||||
_,
|
..,
|
||||||
_,
|
|
||||||
)) => {}
|
)) => {}
|
||||||
unexpected => panic!("Unexpected error {}", unexpected),
|
unexpected => panic!("Unexpected error {}", unexpected),
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in new issue
Block a user