openvm-circuit 2.0.1

OpenVM circuits
Documentation
use openvm_circuit_primitives::ColumnsAir;
use openvm_stark_backend::{
    interaction::InteractionBuilder,
    p3_air::{Air, BaseAir},
    p3_field::Field,
    p3_matrix::Matrix,
    BaseAirWithPublicValues, PartitionedBaseAir,
};

use super::ProgramTester;
use crate::system::program::ProgramBus;

#[derive(Clone, Copy, Debug, derive_new::new)]
pub struct ProgramDummyAir {
    pub bus: ProgramBus,
}

// No columns provided: test dummy whose row is `ProgramExecutionCols` plus a trailing `count`
// column, with no matching `Cols` struct.
impl ColumnsAir for ProgramDummyAir {}
impl<F: Field> BaseAirWithPublicValues<F> for ProgramDummyAir {}
impl<F: Field> PartitionedBaseAir<F> for ProgramDummyAir {}
impl<F: Field> BaseAir<F> for ProgramDummyAir {
    fn width(&self) -> usize {
        ProgramTester::<F>::width()
    }
}

impl<AB: InteractionBuilder> Air<AB> for ProgramDummyAir {
    fn eval(&self, builder: &mut AB) {
        let main = builder.main();
        let local = main.row_slice(0).expect("row 0 present");
        let local = local
            .iter()
            .cloned()
            .map(Into::into)
            .collect::<Vec<AB::Expr>>();
        self.bus.inner.add_key_with_lookups(
            builder,
            local[..local.len() - 1].iter().cloned(),
            local[local.len() - 1].clone(),
        );
    }
}