97 lines
3.0 KiB
Rust
97 lines
3.0 KiB
Rust
extern crate tiff;
|
|
extern crate tempfile;
|
|
|
|
use tiff::ColorType;
|
|
use tiff::decoder::{Decoder, DecodingResult};
|
|
use tiff::encoder::{colortype, TiffEncoder};
|
|
|
|
use std::io::{Seek, SeekFrom};
|
|
use std::fs::File;
|
|
|
|
#[test]
|
|
fn encode_decode() {
|
|
let mut image_data = Vec::new();
|
|
for x in 0..100 {
|
|
for y in 0..100u8 {
|
|
let val = x + y;
|
|
image_data.push(val);
|
|
image_data.push(val);
|
|
image_data.push(val);
|
|
}
|
|
}
|
|
let mut file = tempfile::tempfile().unwrap();
|
|
{
|
|
let mut tiff = TiffEncoder::new(&mut file).unwrap();
|
|
|
|
tiff.write_image::<colortype::RGB8>(100, 100, &image_data).unwrap();
|
|
}
|
|
{
|
|
file.seek(SeekFrom::Start(0)).unwrap();
|
|
let mut decoder = Decoder::new(&mut file).unwrap();
|
|
assert_eq!(decoder.colortype().unwrap(), ColorType::RGB(8));
|
|
assert_eq!(decoder.dimensions().unwrap(), (100, 100));
|
|
if let DecodingResult::U8(img_res) = decoder.read_image().unwrap() {
|
|
assert_eq!(image_data, img_res);
|
|
}
|
|
else {
|
|
panic!("Wrong data type");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn test_gray_u8_roundtrip()
|
|
{
|
|
let img_file = File::open("./tests/images/minisblack-1c-8b.tiff").expect("Cannot find test image!");
|
|
let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
|
|
assert_eq!(decoder.colortype().unwrap(), ColorType::Gray(8));
|
|
let img_res = decoder.read_image();
|
|
assert!(img_res.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn test_rgb_u8()
|
|
{
|
|
let img_file = File::open("./tests/images/rgb-3c-8b.tiff").expect("Cannot find test image!");
|
|
let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
|
|
assert_eq!(decoder.colortype().unwrap(), ColorType::RGB(8));
|
|
let img_res = decoder.read_image();
|
|
|
|
let mut file = tempfile::tempfile().unwrap();
|
|
let mut tiff = TiffEncoder::new(&mut file).unwrap();
|
|
|
|
tiff.write_image::<colortype::RGB8>(100, 100, &image_data).unwrap();
|
|
file.seek(SeekFrom::Start(0)).unwrap();
|
|
|
|
let mut decoder = Decoder::new(&mut file).unwrap();
|
|
assert_eq!(decoder.colortype().unwrap(), ColorType::RGB(8));
|
|
assert_eq!(decoder.dimensions().unwrap(), (100, 100));
|
|
if let DecodingResult::U8(img_res) = decoder.read_image().unwrap() {
|
|
assert_eq!(image_data, img_res);
|
|
}
|
|
else {
|
|
panic!("Wrong data type");
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_gray_u16()
|
|
{
|
|
let img_file = File::open("./tests/images/minisblack-1c-16b.tiff").expect("Cannot find test image!");
|
|
let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
|
|
assert_eq!(decoder.colortype().unwrap(), ColorType::Gray(16));
|
|
let img_res = decoder.read_image();
|
|
assert!(img_res.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn test_rgb_u16()
|
|
{
|
|
let img_file = File::open("./tests/images/rgb-3c-16b.tiff").expect("Cannot find test image!");
|
|
let mut decoder = Decoder::new(img_file).expect("Cannot create decoder");
|
|
assert_eq!(decoder.colortype().unwrap(), ColorType::RGB(16));
|
|
let img_res = decoder.read_image();
|
|
assert!(img_res.is_ok());
|
|
}
|