ark_ff_asm/context/
data_structures.rs

1use std::fmt;
2
3#[derive(Clone)]
4pub enum AssemblyVar {
5    Memory(String),
6    Variable(String),
7    Fixed(String),
8}
9
10impl AssemblyVar {
11    pub fn memory_access(&self, offset: usize) -> Option<AssemblyVar> {
12        match self {
13            Self::Variable(a) | Self::Fixed(a) => Some(Self::Memory(format!("{}({})", offset, a))),
14            _ => None,
15        }
16    }
17
18    pub fn memory_accesses(&self, range: usize) -> Vec<AssemblyVar> {
19        (0..range)
20            .map(|i| {
21                let offset = i * 8;
22                self.memory_access(offset).unwrap()
23            })
24            .collect()
25    }
26}
27
28impl fmt::Display for AssemblyVar {
29    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> Result<(), fmt::Error> {
30        match self {
31            Self::Variable(a) | Self::Fixed(a) | Self::Memory(a) => write!(f, "{}", a),
32        }
33    }
34}
35
36impl<'a> From<Declaration<'a>> for AssemblyVar {
37    fn from(other: Declaration<'a>) -> Self {
38        Self::Variable(format!("{{{}}}", other.name))
39    }
40}
41
42impl<'a> From<Register<'a>> for AssemblyVar {
43    fn from(other: Register<'a>) -> Self {
44        Self::Fixed(format!("%{}", other.0))
45    }
46}
47
48#[derive(Copy, Clone, PartialEq, Eq)]
49pub struct Register<'a>(pub &'a str);
50impl fmt::Display for Register<'_> {
51    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> Result<(), fmt::Error> {
52        write!(f, "\"{}\"", self.0)
53    }
54}
55
56#[derive(Copy, Clone)]
57pub struct Declaration<'a> {
58    /// Name of the assembly template variable declared by `self`.
59    pub name: &'a str,
60    /// Rust expression whose value is declared in `self`.
61    pub expr: &'a str,
62}
63
64impl fmt::Display for Declaration<'_> {
65    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> Result<(), fmt::Error> {
66        write!(f, "{} = in(reg) {},", self.name, self.expr)
67    }
68}
ark_ff_asm/context/data_structures.rs

ark_ff_asm/context/
data_structures.rs
