709 lines
24 KiB
Rust
709 lines
24 KiB
Rust
//
|
|
// Copyright (c) 2016 KAMADA Ken'ichi.
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions
|
|
// are met:
|
|
// 1. Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// 2. Redistributions in binary form must reproduce the above copyright
|
|
// notice, this list of conditions and the following disclaimer in the
|
|
// documentation and/or other materials provided with the distribution.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
// OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
// HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
// OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
// SUCH DAMAGE.
|
|
//
|
|
|
|
use std::fmt;
|
|
use std::mem;
|
|
|
|
use endian::Endian;
|
|
|
|
/// Types and values of TIFF fields (for Exif attributes).
|
|
#[derive(Debug)]
|
|
pub enum Value<'a> {
|
|
/// Vector of 8-bit unsigned integers.
|
|
Byte(Vec<u8>),
|
|
/// Vector of slices of 8-bit bytes containing 7-bit ASCII characters.
|
|
/// The trailing null character is not included. Note that
|
|
/// the 8th bits may present if a non-conforming data is given.
|
|
Ascii(Vec<&'a [u8]>),
|
|
/// Vector of 16-bit unsigned integers.
|
|
Short(Vec<u16>),
|
|
/// Vector of 32-bit unsigned integers.
|
|
Long(Vec<u32>),
|
|
/// Vector of unsigned rationals.
|
|
/// An unsigned rational number is a pair of 32-bit unsigned integers.
|
|
Rational(Vec<Rational>),
|
|
/// Vector of 8-bit signed integers. Unused in the Exif specification.
|
|
SByte(Vec<i8>),
|
|
/// Slice of 8-bit bytes.
|
|
Undefined(&'a [u8]),
|
|
/// Vector of 16-bit signed integers. Unused in the Exif specification.
|
|
SShort(Vec<i16>),
|
|
/// Vector of 32-bit signed integers.
|
|
SLong(Vec<i32>),
|
|
/// Vector of signed rationals.
|
|
/// A signed rational number is a pair of 32-bit signed integers.
|
|
SRational(Vec<SRational>),
|
|
/// Vector of 32-bit (single precision) floating-point numbers.
|
|
/// Unused in the Exif specification.
|
|
Float(Vec<f32>),
|
|
/// Vector of 64-bit (double precision) floating-point numbers.
|
|
/// Unused in the Exif specification.
|
|
Double(Vec<f64>),
|
|
/// The type is unknown to this implementation.
|
|
/// The associated values are the type, the count, and the
|
|
/// offset of the "Value Offset" element.
|
|
Unknown(u16, u32, u32),
|
|
}
|
|
|
|
impl<'a> Value<'a> {
|
|
/// Returns an object that implements `std::fmt::Display` for
|
|
/// printing a value in a tag-specific format.
|
|
/// The tag of the value is specified as the argument.
|
|
///
|
|
/// # Examples
|
|
///
|
|
/// ```
|
|
/// use exif::{Value, tag};
|
|
/// let val = Value::Undefined(b"0231");
|
|
/// assert_eq!(format!("{}", val.display_as(tag::ExifVersion)),
|
|
/// "2.31");
|
|
/// let val = Value::Short(vec![2]);
|
|
/// assert_eq!(format!("{}", val.display_as(tag::ResolutionUnit)),
|
|
/// "pixels per inch");
|
|
/// ```
|
|
#[inline]
|
|
pub fn display_as(&self, tag: ::tag_priv::Tag) -> Display {
|
|
::tag_priv::display_value_as(self, tag)
|
|
}
|
|
|
|
/// Returns the unsigned integer at the given position.
|
|
/// None is returned if the value type is not unsigned integer
|
|
/// (BYTE, SHORT, or LONG) or the position is out of bounds.
|
|
pub fn get_uint(&self, index: usize) -> Option<u32> {
|
|
match *self {
|
|
Value::Byte(ref v) if v.len() > index => Some(v[index] as u32),
|
|
Value::Short(ref v) if v.len() > index => Some(v[index] as u32),
|
|
Value::Long(ref v) if v.len() > index => Some(v[index]),
|
|
_ => None,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Helper struct for printing a value in a tag-specific format.
|
|
pub struct Display<'a> {
|
|
pub fmt: fn(&mut fmt::Write, &Value) -> fmt::Result,
|
|
pub value: &'a Value<'a>,
|
|
}
|
|
|
|
impl<'a> fmt::Display for Display<'a> {
|
|
#[inline]
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
(self.fmt)(f, self.value)
|
|
}
|
|
}
|
|
|
|
// Static default values.
|
|
pub enum DefaultValue {
|
|
None,
|
|
Byte(&'static [u8]),
|
|
Ascii(&'static [&'static [u8]]),
|
|
Short(&'static [u16]),
|
|
Rational(&'static [(u32, u32)]),
|
|
Undefined(&'static [u8]),
|
|
// Depends on other tags, JPEG markers, etc.
|
|
ContextDependent,
|
|
// Unspecified in the Exif standard.
|
|
Unspecified,
|
|
}
|
|
|
|
impl<'a> From<&'a DefaultValue> for Option<Value<'a>> {
|
|
fn from(defval: &DefaultValue) -> Option<Value> {
|
|
match *defval {
|
|
DefaultValue::None => None,
|
|
DefaultValue::Byte(s) => Some(Value::Byte(s.to_vec())),
|
|
DefaultValue::Ascii(s) => Some(Value::Ascii(s.to_vec())),
|
|
DefaultValue::Short(s) => Some(Value::Short(s.to_vec())),
|
|
DefaultValue::Rational(s) => Some(Value::Rational(
|
|
s.iter().map(|&t| tuple2rational(t)).collect())),
|
|
DefaultValue::Undefined(s) => Some(Value::Undefined(s)),
|
|
DefaultValue::ContextDependent => None,
|
|
DefaultValue::Unspecified => None,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// An unsigned rational number, which is a pair of 32-bit unsigned integers.
|
|
#[derive(Copy, Clone)]
|
|
pub struct Rational { pub num: u32, pub denom: u32 }
|
|
|
|
impl Rational {
|
|
/// Converts the value to an f64.
|
|
#[inline]
|
|
pub fn to_f64(&self) -> f64 {
|
|
self.num as f64 / self.denom as f64
|
|
}
|
|
}
|
|
|
|
fn tuple2rational(t: (u32, u32)) -> Rational {
|
|
Rational { num: t.0, denom: t.1 }
|
|
}
|
|
|
|
impl fmt::Debug for Rational {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "Rational({}/{})", self.num, self.denom)
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for Rational {
|
|
/// Formatting parameters other than width are not supported.
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
let buf = fmt_rational_sub(f, self.num, self.denom);
|
|
f.pad_integral(true, "", &buf)
|
|
}
|
|
}
|
|
|
|
impl From<Rational> for f64 {
|
|
#[inline]
|
|
fn from(r: Rational) -> f64 { r.to_f64() }
|
|
}
|
|
|
|
impl From<Rational> for f32 {
|
|
#[inline]
|
|
fn from(r: Rational) -> f32 { r.to_f64() as f32 }
|
|
}
|
|
|
|
/// A signed rational number, which is a pair of 32-bit signed integers.
|
|
#[derive(Copy, Clone)]
|
|
pub struct SRational { pub num: i32, pub denom: i32 }
|
|
|
|
impl SRational {
|
|
/// Converts the value to an f64.
|
|
#[inline]
|
|
pub fn to_f64(&self) -> f64 {
|
|
self.num as f64 / self.denom as f64
|
|
}
|
|
}
|
|
|
|
impl fmt::Debug for SRational {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "SRational({}/{})", self.num, self.denom)
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for SRational {
|
|
/// Formatting parameters other than width are not supported.
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
let buf = fmt_rational_sub(
|
|
f, self.num.wrapping_abs() as u32, self.denom);
|
|
f.pad_integral(self.num >= 0, "", &buf)
|
|
}
|
|
}
|
|
|
|
impl From<SRational> for f64 {
|
|
#[inline]
|
|
fn from(r: SRational) -> f64 { r.to_f64() }
|
|
}
|
|
|
|
impl From<SRational> for f32 {
|
|
#[inline]
|
|
fn from(r: SRational) -> f32 { r.to_f64() as f32 }
|
|
}
|
|
|
|
// Only u32 or i32 are expected for T.
|
|
fn fmt_rational_sub<T>(f: &mut fmt::Formatter, num: u32, denom: T)
|
|
-> String where T: fmt::Display {
|
|
// The API to get the alignment is not yet stable as of Rust 1.16,
|
|
// so it is not fully supported.
|
|
match (f.sign_plus(), f.precision(), f.sign_aware_zero_pad()) {
|
|
(true, Some(prec), true) =>
|
|
format!("{}/{:+0w$}", num, denom, w = prec),
|
|
(true, Some(prec), false) =>
|
|
format!("{}/{:+w$}", num, denom, w = prec),
|
|
(true, None, _) =>
|
|
format!("{}/{:+}", num, denom),
|
|
(false, Some(prec), true) =>
|
|
format!("{}/{:0w$}", num, denom, w = prec),
|
|
(false, Some(prec), false) =>
|
|
format!("{}/{:w$}", num, denom, w = prec),
|
|
(false, None, _) =>
|
|
format!("{}/{}", num, denom),
|
|
}
|
|
}
|
|
|
|
type Parser<'a> = fn(&'a [u8], usize, usize) -> Value<'a>;
|
|
|
|
// Return the length of a single value and the parser of the type.
|
|
pub fn get_type_info<'a, E>(typecode: u16)
|
|
-> (usize, Parser<'a>) where E: Endian {
|
|
match typecode {
|
|
1 => (1, parse_byte),
|
|
2 => (1, parse_ascii),
|
|
3 => (2, parse_short::<E>),
|
|
4 => (4, parse_long::<E>),
|
|
5 => (8, parse_rational::<E>),
|
|
6 => (1, parse_sbyte),
|
|
7 => (1, parse_undefined),
|
|
8 => (2, parse_sshort::<E>),
|
|
9 => (4, parse_slong::<E>),
|
|
10 => (8, parse_srational::<E>),
|
|
11 => (4, parse_float::<E>),
|
|
12 => (8, parse_double::<E>),
|
|
_ => (0, parse_unknown),
|
|
}
|
|
}
|
|
|
|
fn parse_byte<'a>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> {
|
|
Value::Byte(data[offset .. offset + count].to_vec())
|
|
}
|
|
|
|
fn parse_ascii<'a>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> {
|
|
// Any ASCII field can contain multiple strings [TIFF6 Image File
|
|
// Directory].
|
|
let iter = (&data[offset .. offset + count]).split(|&b| b == b'\0');
|
|
let mut v: Vec<&[u8]> = iter.collect();
|
|
if v.last().map_or(false, |&s| s.len() == 0) {
|
|
v.pop();
|
|
}
|
|
Value::Ascii(v)
|
|
}
|
|
|
|
fn parse_short<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(E::loadu16(data, offset + i * 2));
|
|
}
|
|
Value::Short(val)
|
|
}
|
|
|
|
fn parse_long<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(E::loadu32(data, offset + i * 4));
|
|
}
|
|
Value::Long(val)
|
|
}
|
|
|
|
fn parse_rational<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(Rational {
|
|
num: E::loadu32(data, offset + i * 8),
|
|
denom: E::loadu32(data, offset + i * 8 + 4),
|
|
});
|
|
}
|
|
Value::Rational(val)
|
|
}
|
|
|
|
fn parse_sbyte<'a>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> {
|
|
let uslice = &data[offset .. offset + count];
|
|
let islice = unsafe { ::std::slice::from_raw_parts(
|
|
uslice.as_ptr() as *const i8, count) };
|
|
Value::SByte(islice.to_vec())
|
|
}
|
|
|
|
fn parse_undefined<'a>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> {
|
|
Value::Undefined(&data[offset .. offset + count])
|
|
}
|
|
|
|
fn parse_sshort<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(E::loadu16(data, offset + i * 2) as i16);
|
|
}
|
|
Value::SShort(val)
|
|
}
|
|
|
|
fn parse_slong<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(E::loadu32(data, offset + i * 4) as i32);
|
|
}
|
|
Value::SLong(val)
|
|
}
|
|
|
|
fn parse_srational<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(SRational {
|
|
num: E::loadu32(data, offset + i * 8) as i32,
|
|
denom: E::loadu32(data, offset + i * 8 + 4) as i32,
|
|
});
|
|
}
|
|
Value::SRational(val)
|
|
}
|
|
|
|
// TIFF and Rust use IEEE 754 format, so no conversion is required.
|
|
fn parse_float<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(unsafe { mem::transmute(E::loadu32(data, offset + i * 4)) });
|
|
}
|
|
Value::Float(val)
|
|
}
|
|
|
|
// TIFF and Rust use IEEE 754 format, so no conversion is required.
|
|
fn parse_double<'a, E>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> where E: Endian {
|
|
let mut val = Vec::with_capacity(count);
|
|
for i in 0..count {
|
|
val.push(unsafe { mem::transmute(E::loadu64(data, offset + i * 8)) });
|
|
}
|
|
Value::Double(val)
|
|
}
|
|
|
|
// This is a dummy function and will never be called.
|
|
#[allow(unused_variables)]
|
|
fn parse_unknown<'a>(data: &'a [u8], offset: usize, count: usize)
|
|
-> Value<'a> {
|
|
unreachable!()
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use endian::BigEndian;
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn byte() {
|
|
let sets: &[(&[u8], &[u8])] = &[
|
|
(b"x", b""),
|
|
(b"x\xbe\xad", b"\xbe\xad"),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(1);
|
|
for &(data, ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Byte(v) => assert_eq!(v, ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn ascii() {
|
|
let sets: &[(&[u8], Vec<&[u8]>)] = &[
|
|
(b"x", vec![]), // malformed
|
|
(b"x\0", vec![b""]),
|
|
(b"x\0\0", vec![b"", b""]),
|
|
(b"xA", vec![b"A"]), // malformed
|
|
(b"xA\0", vec![b"A"]),
|
|
(b"xA\0B", vec![b"A", b"B"]), // malformed
|
|
(b"xA\0B\0", vec![b"A", b"B"]),
|
|
(b"xA\0\xbe\0", vec![b"A", b"\xbe"]), // not ASCII
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(2);
|
|
for &(data, ref ans) in sets {
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Ascii(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn short() {
|
|
let sets: &[(&[u8], Vec<u16>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\x01\x02\x03\x04", vec![0x0102, 0x0304]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(3);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Short(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn long() {
|
|
let sets: &[(&[u8], Vec<u32>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\x01\x02\x03\x04\x05\x06\x07\x08",
|
|
vec![0x01020304, 0x05060708]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(4);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Long(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn rational() {
|
|
let sets: &[(&[u8], Vec<Rational>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\xa1\x02\x03\x04\x05\x06\x07\x08\
|
|
\x09\x0a\x0b\x0c\xbd\x0e\x0f\x10",
|
|
vec![Rational { num: 0xa1020304, denom: 0x05060708 },
|
|
Rational { num: 0x090a0b0c, denom: 0xbd0e0f10 }]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(5);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Rational(v) => {
|
|
assert_eq!(v.len(), ans.len());
|
|
for (x, y) in v.iter().zip(ans.iter()) {
|
|
assert!(x.num == y.num && x.denom == y.denom);
|
|
}
|
|
},
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn sbyte() {
|
|
let sets: &[(&[u8], &[i8])] = &[
|
|
(b"x", &[]),
|
|
(b"x\xbe\x7d", &[-0x42, 0x7d]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(6);
|
|
for &(data, ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::SByte(v) => assert_eq!(v, ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn undefined() {
|
|
let sets: &[(&[u8], &[u8])] = &[
|
|
(b"x", b""),
|
|
(b"x\xbe\xad", b"\xbe\xad"),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(7);
|
|
for &(data, ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Undefined(v) => assert_eq!(v, ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn sshort() {
|
|
let sets: &[(&[u8], Vec<i16>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\x01\x02\xf3\x04", vec![0x0102, -0x0cfc]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(8);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::SShort(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn slong() {
|
|
let sets: &[(&[u8], Vec<i32>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\x01\x02\x03\x04\x85\x06\x07\x08",
|
|
vec![0x01020304, -0x7af9f8f8]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(9);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::SLong(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn srational() {
|
|
let sets: &[(&[u8], Vec<SRational>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\xa1\x02\x03\x04\x05\x06\x07\x08\
|
|
\x09\x0a\x0b\x0c\xbd\x0e\x0f\x10",
|
|
vec![SRational { num: -0x5efdfcfc, denom: 0x05060708 },
|
|
SRational { num: 0x090a0b0c, denom: -0x42f1f0f0 }]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(10);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::SRational(v) => {
|
|
assert_eq!(v.len(), ans.len());
|
|
for (x, y) in v.iter().zip(ans.iter()) {
|
|
assert!(x.num == y.num && x.denom == y.denom);
|
|
}
|
|
},
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn float() {
|
|
let sets: &[(&[u8], Vec<f32>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\x7f\x7f\xff\xff\x80\x80\x00\x00\x40\x00\x00\x00",
|
|
vec![::std::f32::MAX, -::std::f32::MIN_POSITIVE, 2.0]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(11);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Float(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn double() {
|
|
let sets: &[(&[u8], Vec<f64>)] = &[
|
|
(b"x", vec![]),
|
|
(b"x\x7f\xef\xff\xff\xff\xff\xff\xff\
|
|
\x80\x10\x00\x00\x00\x00\x00\x00\
|
|
\x40\x00\x00\x00\x00\x00\x00\x00",
|
|
vec![::std::f64::MAX, -::std::f64::MIN_POSITIVE, 2.0]),
|
|
];
|
|
let (unitlen, parser) = get_type_info::<BigEndian>(12);
|
|
for &(data, ref ans) in sets {
|
|
assert!((data.len() - 1) % unitlen == 0);
|
|
match parser(data, 1, (data.len() - 1) / unitlen) {
|
|
Value::Double(v) => assert_eq!(v, *ans),
|
|
v => panic!("wrong variant {:?}", v),
|
|
}
|
|
}
|
|
}
|
|
|
|
// These functions are never called in a way that an out-of-range access
|
|
// could happen, so this test is hypothetical but just for safety.
|
|
#[test]
|
|
#[should_panic(expected = "index 5 out of range for slice of length 4")]
|
|
fn short_oor() {
|
|
parse_short::<BigEndian>(b"\x01\x02\x03\x04", 1, 2);
|
|
}
|
|
|
|
#[test]
|
|
fn unknown() {
|
|
let (unitlen, _parser) = get_type_info::<BigEndian>(0xffff);
|
|
assert_eq!(unitlen, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn get_uint() {
|
|
let v = Value::Byte(vec![1, 2]);
|
|
assert_eq!(v.get_uint(0), Some(1));
|
|
assert_eq!(v.get_uint(1), Some(2));
|
|
assert_eq!(v.get_uint(2), None);
|
|
let v = Value::Short(vec![1, 2]);
|
|
assert_eq!(v.get_uint(0), Some(1));
|
|
assert_eq!(v.get_uint(1), Some(2));
|
|
assert_eq!(v.get_uint(2), None);
|
|
let v = Value::Long(vec![1, 2]);
|
|
assert_eq!(v.get_uint(0), Some(1));
|
|
assert_eq!(v.get_uint(1), Some(2));
|
|
assert_eq!(v.get_uint(2), None);
|
|
let v = Value::SLong(vec![1, 2]);
|
|
assert_eq!(v.get_uint(0), None);
|
|
assert_eq!(v.get_uint(1), None);
|
|
assert_eq!(v.get_uint(2), None);
|
|
}
|
|
|
|
#[test]
|
|
fn rational_fmt_display() {
|
|
let r = Rational { num: u32::max_value(), denom: u32::max_value() };
|
|
assert_eq!(format!("{}", r), "4294967295/4294967295");
|
|
|
|
let r = Rational { num: 10, denom: 20 };
|
|
assert_eq!(format!("{}", r), "10/20");
|
|
assert_eq!(format!("{:11}", r), " 10/20");
|
|
assert_eq!(format!("{:3}", r), "10/20");
|
|
}
|
|
|
|
#[test]
|
|
fn srational_fmt_display() {
|
|
let r = SRational { num: i32::min_value(), denom: i32::min_value() };
|
|
assert_eq!(format!("{}", r), "-2147483648/-2147483648");
|
|
let r = SRational { num: i32::max_value(), denom: i32::max_value() };
|
|
assert_eq!(format!("{}", r), "2147483647/2147483647");
|
|
|
|
let r = SRational { num: -10, denom: 20 };
|
|
assert_eq!(format!("{}", r), "-10/20");
|
|
assert_eq!(format!("{:11}", r), " -10/20");
|
|
assert_eq!(format!("{:3}", r), "-10/20");
|
|
|
|
let r = SRational { num: 10, denom: -20 };
|
|
assert_eq!(format!("{}", r), "10/-20");
|
|
assert_eq!(format!("{:11}", r), " 10/-20");
|
|
assert_eq!(format!("{:3}", r), "10/-20");
|
|
|
|
let r = SRational { num: -10, denom: -20 };
|
|
assert_eq!(format!("{}", r), "-10/-20");
|
|
assert_eq!(format!("{:11}", r), " -10/-20");
|
|
assert_eq!(format!("{:3}", r), "-10/-20");
|
|
}
|
|
|
|
#[test]
|
|
fn ratioanl_f64() {
|
|
use std::{f64, u32};
|
|
assert_eq!(f64::from(Rational { num: 1, denom: 2 }), 0.5);
|
|
assert_eq!(f64::from(Rational { num: 1, denom: u32::MAX }),
|
|
2.3283064370807974e-10);
|
|
assert_eq!(f64::from(Rational { num: u32::MAX, denom: 1 }),
|
|
u32::MAX as f64);
|
|
assert_eq!(f64::from(Rational { num: u32::MAX - 1, denom: u32::MAX }),
|
|
0.9999999997671694);
|
|
assert_eq!(f64::from(Rational { num: u32::MAX, denom: u32::MAX - 1 }),
|
|
1.0000000002328306);
|
|
assert_eq!(f64::from(Rational { num: 1, denom: 0 }), f64::INFINITY);
|
|
assert!(f64::from(Rational { num: 0, denom: 0 }).is_nan());
|
|
|
|
assert_eq!(f64::from(SRational { num: 1, denom: 2 }), 0.5);
|
|
assert_eq!(f64::from(SRational { num: -1, denom: 2 }), -0.5);
|
|
assert_eq!(f64::from(SRational { num: 1, denom: -2 }), -0.5);
|
|
assert_eq!(f64::from(SRational { num: -1, denom: -2 }), 0.5);
|
|
assert_eq!(f64::from(SRational { num: 1, denom: 0 }), f64::INFINITY);
|
|
assert_eq!(f64::from(SRational { num: -1, denom: 0 }),
|
|
f64::NEG_INFINITY);
|
|
}
|
|
|
|
#[test]
|
|
fn rational_f32() {
|
|
// If num and demon are converted to f32 before the division,
|
|
// the precision is lost in this example.
|
|
assert_eq!(f32::from(Rational { num: 1, denom: 16777217 }),
|
|
5.960464e-8);
|
|
}
|
|
}
|