stak_native/
arithmetic.rs

1use stak_vm::{Error, Heap, Memory, PrimitiveSet};
2use winter_maybe_async::maybe_async;
3
4/// An arithmetic primitive.
5pub enum ArithmeticPrimitive {
6    /// A `quotient` procedure.
7    Quotient,
8}
9
10impl ArithmeticPrimitive {
11    const QUOTIENT: usize = Self::Quotient as _;
12}
13
14/// An arithmetic primitive set.
15#[derive(Debug, Default)]
16pub struct ArithmeticPrimitiveSet {}
17
18impl ArithmeticPrimitiveSet {
19    /// Creates a primitive set.
20    pub fn new() -> Self {
21        Self::default()
22    }
23}
24
25impl<H: Heap> PrimitiveSet<H> for ArithmeticPrimitiveSet {
26    type Error = Error;
27
28    #[maybe_async]
29    fn operate(&mut self, memory: &mut Memory<H>, primitive: usize) -> Result<(), Self::Error> {
30        match primitive {
31            ArithmeticPrimitive::QUOTIENT => {
32                let [x, y] = memory.pop_many()?;
33                let x = x.assume_number();
34                let y = y.assume_number();
35
36                memory.push(((x - x % y) / y).into())?;
37            }
38            _ => return Err(Error::IllegalPrimitive),
39        }
40
41        Ok(())
42    }
43}