openvm-circuit 2.0.1

OpenVM circuits
Documentation
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> {
    /// The receive frequency. To send, set to negative.
    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);
    }
}