use plonky2::field::extension::Extendable;
use plonky2::field::packed::PackedField;
use plonky2::hash::hash_types::RichField;
use plonky2::iop::ext_target::ExtensionTarget;

use crate::constraint_consumer::{ConstraintConsumer, RecursiveConstraintConsumer};
use crate::cpu::columns::CpuColumnsView;

/// Check the correct updating of `clock`.
pub fn eval_packed<P: PackedField>(
    lv: &CpuColumnsView<P>,
    nv: &CpuColumnsView<P>,
    yield_constr: &mut ConstraintConsumer<P>,
) {
    // The clock is 0 at the beginning.
    yield_constr.constraint_first_row(lv.clock);
    // The clock is incremented by 1 at each row.
    yield_constr.constraint_transition(nv.clock - lv.clock - P::ONES);
}

/// Circuit version of `eval_packed`.
/// Check the correct updating of `clock`.
pub fn eval_ext_circuit<F: RichField + Extendable<D>, const D: usize>(
    builder: &mut plonky2::plonk::circuit_builder::CircuitBuilder<F, D>,
    lv: &CpuColumnsView<ExtensionTarget<D>>,
    nv: &CpuColumnsView<ExtensionTarget<D>>,
    yield_constr: &mut RecursiveConstraintConsumer<F, D>,
) {
    // The clock is 0 at the beginning.
    yield_constr.constraint_first_row(builder, lv.clock);
    // The clock is incremented by 1 at each row.
    {
        let new_clock = builder.add_const_extension(lv.clock, F::ONE);
        let constr = builder.sub_extension(nv.clock, new_clock);
        yield_constr.constraint_transition(builder, constr);
    }
}