use std::{borrow::Borrow, mem::size_of};
use openvm_circuit_primitives::{ColumnsAir, StructReflection, StructReflectionHelper};
use openvm_circuit_primitives_derive::AlignedBorrow;
use openvm_stark_backend::{
interaction::InteractionBuilder,
p3_air::{Air, BaseAir},
p3_field::Field,
p3_matrix::Matrix,
BaseAirWithPublicValues, PartitionedBaseAir,
};
use crate::arch::{ExecutionBus, ExecutionState};
#[derive(Clone, Copy, Debug, AlignedBorrow, StructReflection, derive_new::new)]
#[repr(C)]
pub struct DummyExecutionInteractionCols<T> {
pub count: T,
pub initial_state: ExecutionState<T>,
pub final_state: ExecutionState<T>,
}
#[derive(Clone, Copy, Debug, derive_new::new, ColumnsAir)]
#[columns_via(DummyExecutionInteractionCols<u8>)]
pub struct ExecutionDummyAir {
pub bus: ExecutionBus,
}
impl<F: Field> BaseAirWithPublicValues<F> for ExecutionDummyAir {}
impl<F: Field> PartitionedBaseAir<F> for ExecutionDummyAir {}
impl<F: Field> BaseAir<F> for ExecutionDummyAir {
fn width(&self) -> usize {
size_of::<DummyExecutionInteractionCols<u8>>()
}
}
impl<AB: InteractionBuilder> Air<AB> for ExecutionDummyAir {
fn eval(&self, builder: &mut AB) {
let main = builder.main();
let local = main.row_slice(0).expect("row 0 present");
let local: &DummyExecutionInteractionCols<AB::Var> = (*local).borrow();
self.bus
.execute(builder, local.count, local.initial_state, local.final_state);
}
}