use alloc::{string::ToString, vec::Vec};
use core::fmt;
use miden_crypto::hash::blake::Blake3_256;
use num_traits::ToBytes;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
mod assembly_op;
pub use assembly_op::AssemblyOp;
mod debug;
pub use debug::DebugOptions;
use crate::mast::{DecoratedOpLink, DecoratorFingerprint};
#[derive(Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(all(feature = "arbitrary", test), miden_test_serde_macros::serde_test)]
pub enum Decorator {
AsmOp(AssemblyOp),
Debug(DebugOptions),
Trace(u32),
}
impl Decorator {
pub fn fingerprint(&self) -> DecoratorFingerprint {
match self {
Self::AsmOp(asm_op) => {
let mut bytes_to_hash = Vec::new();
if let Some(location) = asm_op.location() {
bytes_to_hash.extend(location.uri.as_str().as_bytes());
bytes_to_hash.extend(location.start.to_u32().to_le_bytes());
bytes_to_hash.extend(location.end.to_u32().to_le_bytes());
}
bytes_to_hash.extend(asm_op.context_name().as_bytes());
bytes_to_hash.extend(asm_op.op().as_bytes());
bytes_to_hash.push(asm_op.num_cycles());
bytes_to_hash.push(asm_op.should_break() as u8);
Blake3_256::hash(&bytes_to_hash)
},
Self::Debug(debug) => Blake3_256::hash(debug.to_string().as_bytes()),
Self::Trace(trace) => Blake3_256::hash(&trace.to_le_bytes()),
}
}
}
impl crate::prettier::PrettyPrint for Decorator {
fn render(&self) -> crate::prettier::Document {
crate::prettier::display(self)
}
}
impl fmt::Display for Decorator {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
Self::AsmOp(assembly_op) => {
write!(f, "asmOp({}, {})", assembly_op.op(), assembly_op.num_cycles())
},
Self::Debug(options) => write!(f, "debug({options})"),
Self::Trace(trace_id) => write!(f, "trace({trace_id})"),
}
}
}
pub type DecoratorList = Vec<DecoratedOpLink>;