use super::read::{FontRead, ReadError};
use crate::font_data::FontData;
use types::{Nullable, Offset16, Offset24, Offset32};
pub trait Offset: Copy {
fn to_usize(self) -> usize;
fn non_null(self) -> Option<usize> {
match self.to_usize() {
0 => None,
other => Some(other),
}
}
}
macro_rules! impl_offset {
($name:ident, $width:literal) => {
impl Offset for $name {
#[inline]
fn to_usize(self) -> usize {
self.to_u32() as _
}
}
};
}
impl_offset!(Offset16, 2);
impl_offset!(Offset24, 3);
impl_offset!(Offset32, 4);
pub trait ResolveOffset {
fn resolve_with_args<'a, T: FontRead<'a>>(
&self,
data: FontData<'a>,
args: T::Args,
) -> Result<T, ReadError>;
fn resolve<'a, T: FontRead<'a, Args = ()>>(&self, data: FontData<'a>) -> Result<T, ReadError> {
self.resolve_with_args(data, ())
}
}
pub trait ResolveNullableOffset {
fn resolve_with_args<'a, T: FontRead<'a>>(
&self,
data: FontData<'a>,
args: T::Args,
) -> Option<Result<T, ReadError>>;
fn resolve<'a, T: FontRead<'a, Args = ()>>(
&self,
data: FontData<'a>,
) -> Option<Result<T, ReadError>> {
self.resolve_with_args(data, ())
}
}
impl<O: Offset> ResolveNullableOffset for Nullable<O> {
fn resolve_with_args<'a, T: FontRead<'a>>(
&self,
data: FontData<'a>,
args: T::Args,
) -> Option<Result<T, ReadError>> {
match self.offset().resolve_with_args(data, args) {
Ok(thing) => Some(Ok(thing)),
Err(ReadError::NullOffset) => None,
Err(e) => Some(Err(e)),
}
}
}
impl<O: Offset> ResolveOffset for O {
fn resolve_with_args<'a, T: FontRead<'a>>(
&self,
data: FontData<'a>,
args: T::Args,
) -> Result<T, ReadError> {
self.non_null()
.ok_or(ReadError::NullOffset)
.and_then(|off| data.split_off(off).ok_or(ReadError::OutOfBounds))
.and_then(|data| T::read_with_args(data, args))
}
}
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
#[repr(transparent)]
pub(crate) struct CheckedOffset(Option<usize>);
impl CheckedOffset {
pub(crate) fn new(offset: usize) -> Self {
Self(Some(offset))
}
#[must_use]
pub(crate) fn add(self, x: usize) -> Self {
Self(self.0.and_then(|o| o.checked_add(x)))
}
#[must_use]
pub(crate) fn mul(self, x: usize) -> Self {
Self(self.0.and_then(|o| o.checked_mul(x)))
}
pub(crate) fn get(&self) -> Option<usize> {
self.0
}
pub(crate) fn ok_or<E>(&self, err: E) -> Result<usize, E> {
self.get().ok_or(err)
}
pub(crate) fn ok_or_oob(&self) -> Result<usize, ReadError> {
self.ok_or(ReadError::OutOfBounds)
}
}