Added functionality for loading selected strip

This commit is contained in:
Alan Race committed 2022-05-02 15:59:48 +02:00
1 parent 8f34748d01
commit dd8c9e4f2b
4 files changed
+33 -30

No files matched your search

+29 -28
View File
@@ -259,6 +259,7 @@ pub enum ChunkType {
Tile,
}
#[derive(Debug)]
#[non_exhaustive]
pub struct ChunkInfo {
/// Width of the chunk as specified (includes potential padding)
@@ -1165,7 +1166,7 @@ impl<R: Read + Seek> Decoder<R> {
Some(ChunkInfo {
chunk_width: sized_width,
chunk_height: sized_height,
chunk_height: rows_per_strip,
data_width: sized_width,
data_height: strip_height,
@@ -1332,18 +1333,16 @@ impl<R: Read + Seek> Decoder<R> {
.image()
.chunk_offsets
.get(index)
.ok_or(TiffError::FormatError(
TiffFormatError::InconsistentSizesEncountered,
))?;
.ok_or(TiffFormatError::InvalidChunkIndex(index))?;
let byte_count = *self
.image()
.chunk_bytes
.get(index)
.ok_or(TiffError::FormatError(
TiffFormatError::InconsistentSizesEncountered,
))?;
.ok_or(TiffFormatError::InvalidChunkIndex(index))?;
let chunk_info = self.chunk_info(index).ok_or(TiffError::IntSizeError)?;
let chunk_info = self
.chunk_info(index)
.ok_or(TiffFormatError::InvalidChunkIndex(index))?;
let buffer_size = chunk_info
.data_width
@@ -1378,30 +1377,25 @@ impl<R: Read + Seek> Decoder<R> {
fn read_tile_to_buffer(
&mut self,
result: &mut DecodingBuffer,
tile: usize,
tile_index: usize,
output_width: usize,
) -> TiffResult<()> {
let file_offset = *self
.image()
.chunk_offsets
.get(tile)
.ok_or(TiffError::FormatError(
TiffFormatError::InconsistentSizesEncountered,
))?;
.get(tile_index)
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
let compressed_bytes =
*self
.image()
.chunk_bytes
.get(tile)
.ok_or(TiffError::FormatError(
TiffFormatError::InconsistentSizesEncountered,
))?;
let compressed_bytes = *self
.image()
.chunk_bytes
.get(tile_index)
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
// Return the same error as above if the chunk doesn't exist
let chunk_info = self.chunk_info(tile).ok_or(TiffError::FormatError(
TiffFormatError::InconsistentSizesEncountered,
))?;
let chunk_info = self
.chunk_info(tile_index)
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
self.expand_tile(
result.copy(),
@@ -1485,10 +1479,15 @@ impl<R: Read + Seek> Decoder<R> {
/// Read a single strip from the image and return it as a Vector
pub fn read_strip(&mut self) -> TiffResult<(DecodingResult, ChunkInfo)> {
self.check_chunk_type(ChunkType::Strip)?;
let index = self.current_chunk;
self.read_strip_at(self.current_chunk)
}
let chunk_info = self.chunk_info(index).unwrap();
pub fn read_strip_at(&mut self, strip_index: usize) -> TiffResult<(DecodingResult, ChunkInfo)> {
self.check_chunk_type(ChunkType::Strip)?;
let chunk_info = self
.chunk_info(strip_index)
.ok_or(TiffFormatError::InvalidChunkIndex(strip_index))?;
let mut result = self.result_buffer(chunk_info.data_width, chunk_info.data_height)?;
self.read_strip_to_buffer(result.as_buffer(0))?;
@@ -1508,7 +1507,9 @@ impl<R: Read + Seek> Decoder<R> {
pub fn read_tile_at(&mut self, tile_index: usize) -> TiffResult<(DecodingResult, ChunkInfo)> {
self.check_chunk_type(ChunkType::Tile)?;
let chunk_info = self.chunk_info(tile_index).unwrap();
let chunk_info = self
.chunk_info(tile_index)
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
let mut result = self.result_buffer(chunk_info.data_width, chunk_info.data_height)?;