use super::ValueEncoder;
use crate::{
Codec,
CodecEncodeError,
CodecValueExt,
codec::assert_unit_bounds,
};
#[derive(Clone, Copy, Debug, Default, Eq, Hash, PartialEq)]
pub struct CodecValueEncoder<C> {
codec: C,
}
impl<C> CodecValueEncoder<C>
where
C: Codec,
{
#[must_use]
#[inline]
pub fn new(codec: C) -> Self {
assert_unit_bounds::<C>(&codec);
Self { codec }
}
}
impl<C> ValueEncoder<C::Value> for CodecValueEncoder<C>
where
C: Codec,
{
type Output = Vec<C::Unit>;
type Error = CodecEncodeError<C::EncodeError>;
fn encode(
&mut self,
input: &C::Value,
) -> Result<Self::Output, Self::Error> {
if !self.codec.can_encode_value(input) {
return Err(CodecEncodeError::unencodable_value(0));
}
let units = self
.codec
.max_encode_reset_units()
.checked_add(self.codec.encode_len(input).get())
.ok_or_else(CodecEncodeError::output_length_overflow)?;
let mut output = vec![C::Unit::default(); units];
let written =
self.codec.encode_value_with_reset(input, &mut output, 0)?;
output.truncate(written);
Ok(output)
}
}