qoi/src/error.rs
2022-01-04 02:59:32 +03:00

83 lines
3 KiB
Rust

use core::convert::Infallible;
use core::fmt::{self, Display};
use crate::consts::QOI_MAGIC;
/// Errors that can occur during encoding or decoding.
#[derive(Debug)]
pub enum Error {
/// Leading 4 magic bytes don't match when decoding
InvalidMagic { magic: u32 },
/// Invalid number of channels: expected 3 or 4
InvalidChannels { channels: u8 },
/// Invalid color space: expected 0 or 1
InvalidColorSpace { colorspace: u8 },
/// Invalid image dimensions: can't be empty or larger than 400Mp
InvalidImageDimensions { width: u32, height: u32 },
/// Image dimensions are inconsistent with image buffer length
InvalidImageLength { size: usize, width: u32, height: u32 },
/// Output buffer is too small to fit encoded/decoded image
OutputBufferTooSmall { size: usize, required: usize },
/// Input buffer ended unexpectedly before decoding was finished
UnexpectedBufferEnd,
/// Invalid stream end marker encountered when decoding
InvalidPadding,
#[cfg(feature = "std")]
/// Generic I/O error from the wrapped reader/writer
IoError(std::io::Error),
}
/// Alias for [`Result`](std::result::Result) with the error type of [`Error`].
pub type Result<T> = core::result::Result<T, Error>;
impl Display for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match *self {
Self::InvalidMagic { magic } => {
write!(f, "invalid magic: expected {:?}, got {:?}", QOI_MAGIC, magic.to_be_bytes())
}
Self::InvalidChannels { channels } => {
write!(f, "invalid number of channels: {}", channels)
}
Self::InvalidColorSpace { colorspace } => {
write!(f, "invalid color space: {} (expected 0 or 1)", colorspace)
}
Self::InvalidImageDimensions { width, height } => {
write!(f, "invalid image dimensions: {}x{}", width, height)
}
Self::InvalidImageLength { size, width, height } => {
write!(f, "invalid image length: {} bytes for {}x{}", size, width, height)
}
Self::OutputBufferTooSmall { size, required } => {
write!(f, "output buffer size too small: {} (required: {})", size, required)
}
Self::UnexpectedBufferEnd => {
write!(f, "unexpected input buffer end while decoding")
}
Self::InvalidPadding => {
write!(f, "invalid padding (stream end marker mismatch)")
}
#[cfg(feature = "std")]
Self::IoError(ref err) => {
write!(f, "i/o error: {}", err)
}
}
}
}
#[cfg(feature = "std")]
impl std::error::Error for Error {}
impl From<Infallible> for Error {
fn from(_: Infallible) -> Self {
unreachable!()
}
}
#[cfg(feature = "std")]
impl From<std::io::Error> for Error {
fn from(err: std::io::Error) -> Self {
Self::IoError(err)
}
}