Added functionality for loading selected strip
This commit is contained in:
1 parent
ca4da33dd7
commit
e57ca1a926
4 files changed
+30
-22
No files matched your search
+26
-20
@@ -241,6 +241,7 @@ pub enum ChunkType {
|
|||||||
Tile,
|
Tile,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub struct ChunkInfo {
|
pub struct ChunkInfo {
|
||||||
/// Width of the chunk as specified (includes potential padding)
|
/// Width of the chunk as specified (includes potential padding)
|
||||||
@@ -1085,6 +1086,9 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
pub fn read_strip_to_buffer(&mut self, mut buffer: DecodingBuffer) -> TiffResult<()> {
|
pub fn read_strip_to_buffer(&mut self, mut buffer: DecodingBuffer) -> TiffResult<()> {
|
||||||
self.check_chunk_type(ChunkType::Strip)?;
|
self.check_chunk_type(ChunkType::Strip)?;
|
||||||
|
|
||||||
|
let chunk_info = self
|
||||||
|
.chunk_info(index)
|
||||||
|
.ok_or(TiffFormatError::InvalidChunkIndex(index))?;
|
||||||
let offset = self.image.chunk_file_range(self.current_chunk)?.0;
|
let offset = self.image.chunk_file_range(self.current_chunk)?.0;
|
||||||
self.goto_offset_u64(offset)?;
|
self.goto_offset_u64(offset)?;
|
||||||
let chunk_info = self.chunk_info(index).ok_or(TiffError::IntSizeError)?;
|
let chunk_info = self.chunk_info(index).ok_or(TiffError::IntSizeError)?;
|
||||||
@@ -1107,30 +1111,25 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
fn read_tile_to_buffer(
|
fn read_tile_to_buffer(
|
||||||
&mut self,
|
&mut self,
|
||||||
result: &mut DecodingBuffer,
|
result: &mut DecodingBuffer,
|
||||||
tile: usize,
|
tile_index: usize,
|
||||||
output_width: usize,
|
output_width: usize,
|
||||||
) -> TiffResult<()> {
|
) -> TiffResult<()> {
|
||||||
let file_offset = *self
|
let file_offset = *self
|
||||||
.image()
|
.image()
|
||||||
.chunk_offsets
|
.chunk_offsets
|
||||||
.get(tile)
|
.get(tile_index)
|
||||||
.ok_or(TiffError::FormatError(
|
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
|
||||||
TiffFormatError::InconsistentSizesEncountered,
|
|
||||||
))?;
|
|
||||||
|
|
||||||
let compressed_bytes =
|
let compressed_bytes = *self
|
||||||
*self
|
.image()
|
||||||
.image()
|
.chunk_bytes
|
||||||
.chunk_bytes
|
.get(tile_index)
|
||||||
.get(tile)
|
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
|
||||||
.ok_or(TiffError::FormatError(
|
|
||||||
TiffFormatError::InconsistentSizesEncountered,
|
|
||||||
))?;
|
|
||||||
|
|
||||||
// Return the same error as above if the chunk doesn't exist
|
// Return the same error as above if the chunk doesn't exist
|
||||||
let chunk_info = self.chunk_info(tile).ok_or(TiffError::FormatError(
|
let chunk_info = self
|
||||||
TiffFormatError::InconsistentSizesEncountered,
|
.chunk_info(tile_index)
|
||||||
))?;
|
.ok_or(TiffFormatError::InvalidChunkIndex(tile_index))?;
|
||||||
|
|
||||||
self.expand_tile(
|
self.expand_tile(
|
||||||
result.copy(),
|
result.copy(),
|
||||||
@@ -1214,10 +1213,15 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
|
|
||||||
/// Read a single strip from the image and return it as a Vector
|
/// Read a single strip from the image and return it as a Vector
|
||||||
pub fn read_strip(&mut self) -> TiffResult<(DecodingResult, ChunkInfo)> {
|
pub fn read_strip(&mut self) -> TiffResult<(DecodingResult, ChunkInfo)> {
|
||||||
self.check_chunk_type(ChunkType::Strip)?;
|
self.read_strip_at(self.current_chunk)
|
||||||
let index = 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)?;
|
let mut result = self.result_buffer(chunk_info.data_width, chunk_info.data_height)?;
|
||||||
self.read_strip_to_buffer(result.as_buffer(0))?;
|
self.read_strip_to_buffer(result.as_buffer(0))?;
|
||||||
@@ -1237,7 +1241,9 @@ impl<R: Read + Seek> Decoder<R> {
|
|||||||
pub fn read_tile_at(&mut self, tile_index: usize) -> TiffResult<(DecodingResult, ChunkInfo)> {
|
pub fn read_tile_at(&mut self, tile_index: usize) -> TiffResult<(DecodingResult, ChunkInfo)> {
|
||||||
self.check_chunk_type(ChunkType::Tile)?;
|
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)?;
|
let mut result = self.result_buffer(chunk_info.data_width, chunk_info.data_height)?;
|
||||||
|
|
||||||
|
|||||||
@@ -56,6 +56,7 @@ pub enum TiffFormatError {
|
|||||||
actual_samples: usize,
|
actual_samples: usize,
|
||||||
required_samples: usize,
|
required_samples: usize,
|
||||||
},
|
},
|
||||||
|
InvalidChunkIndex(usize),
|
||||||
InvalidTag,
|
InvalidTag,
|
||||||
InvalidTagValueType(Tag),
|
InvalidTagValueType(Tag),
|
||||||
RequiredTagNotFound(Tag),
|
RequiredTagNotFound(Tag),
|
||||||
@@ -100,6 +101,7 @@ impl fmt::Display for TiffFormatError {
|
|||||||
actual_samples, required_samples
|
actual_samples, required_samples
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
InvalidChunkIndex(index) => write!(fmt, "Image chunk index ({}) requested.", index),
|
||||||
InvalidTag => write!(fmt, "Image contains invalid tag."),
|
InvalidTag => write!(fmt, "Image contains invalid tag."),
|
||||||
InvalidTagValueType(ref tag) => {
|
InvalidTagValueType(ref tag) => {
|
||||||
write!(fmt, "Tag `{:?}` did not have the expected value type.", tag)
|
write!(fmt, "Tag `{:?}` did not have the expected value type.", tag)
|
||||||
|
|||||||
@@ -276,7 +276,7 @@ fn test_tiled_incremental() {
|
|||||||
|
|
||||||
for tile in 0..tiles {
|
for tile in 0..tiles {
|
||||||
match decoder.read_tile().unwrap() {
|
match decoder.read_tile().unwrap() {
|
||||||
(DecodingResult::U8(res), chunk_info) => {
|
(DecodingResult::U8(res), _chunk_info) => {
|
||||||
let sum: u64 = res.into_iter().map(<u64>::from).sum();
|
let sum: u64 = res.into_iter().map(<u64>::from).sum();
|
||||||
assert_eq!(sum, sums[tile as usize]);
|
assert_eq!(sum, sums[tile as usize]);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -513,7 +513,7 @@ fn test_rows_per_strip() {
|
|||||||
for i in 0..50 {
|
for i in 0..50 {
|
||||||
let img2 = [i; 2 * 100];
|
let img2 = [i; 2 * 100];
|
||||||
match decoder.read_strip().unwrap() {
|
match decoder.read_strip().unwrap() {
|
||||||
DecodingResult::U8(data) => assert_eq!(&img2[..], &data[..]),
|
(DecodingResult::U8(data), _chunk_info) => assert_eq!(&img2[..], &data[..]),
|
||||||
other => panic!("Incorrect strip type {:?}", other),
|
other => panic!("Incorrect strip type {:?}", other),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in new issue
Block a user