feat: single page index scan with predicate

chips/src/is_equal_vec/columns.rs

@@ -5,12 +5,47 @@ pub struct IsEqualVecIOCols<T> {
     pub prod: T,
 }
 
+impl<T: Clone> IsEqualVecIOCols<T> {
+    pub fn flatten(&self) -> Vec<T> {
+        let mut res: Vec<T> = self.x.iter().chain(self.y.iter()).cloned().collect();
+        res.push(self.prod.clone());
+        res
+    }
+
+    pub fn from_slice(slc: &[T], vec_len: usize) -> Self {
+        let x = slc[0..vec_len].to_vec();
+        let y = slc[vec_len..2 * vec_len].to_vec();
+        let prod = slc[2 * vec_len].clone();
+        Self { x, y, prod }
+    }
+
+    pub fn get_width(vec_len: usize) -> usize {
+        vec_len + vec_len + 1
+    }
+}
+
 #[derive(Default)]
 pub struct IsEqualVecAuxCols<T> {
     pub prods: Vec<T>,
     pub invs: Vec<T>,
 }
 
+impl<T: Clone> IsEqualVecAuxCols<T> {
+    pub fn flatten(&self) -> Vec<T> {
+        self.prods.iter().chain(self.invs.iter()).cloned().collect()
+    }
+
+    pub fn from_slice(slc: &[T], vec_len: usize) -> Self {
+        let prods = slc[0..vec_len].to_vec();
+        let invs = slc[vec_len..2 * vec_len].to_vec();
+        Self { prods, invs }
+    }
+
+    pub fn get_width(vec_len: usize) -> usize {
+        vec_len + vec_len
+    }
+}
+
 #[derive(Default)]
 pub struct IsEqualVecCols<T> {
     pub io: IsEqualVecIOCols<T>,

chips/src/is_equal_vec/mod.rs

@@ -11,6 +11,10 @@ pub struct IsEqualVecAir {
 }
 
 impl IsEqualVecAir {
+    pub fn new(vec_len: usize) -> Self {
+        Self { vec_len }
+    }
+
     pub fn request<F: Clone + PartialEq>(&self, x: &[F], y: &[F]) -> bool {
         x == y
     }

chips/src/lib.rs

@@ -10,6 +10,7 @@ pub mod page_controller;
 pub mod page_read;
 pub mod range;
 pub mod range_gate;
+pub mod single_page_index_scan;
 pub mod sub_chip;
 pub mod sum;
 mod utils;

chips/src/range_gate/mod.rs

@@ -36,6 +36,18 @@ impl RangeCheckerGateChip {
         }
     }
 
+    pub fn bus_index(&self) -> usize {
+        self.air.bus_index
+    }
+
+    pub fn range_max(&self) -> u32 {
+        self.air.range_max
+    }
+
+    pub fn air_width(&self) -> usize {
+        2
+    }
+
     pub fn add_count(&self, val: u32) {
         let val_atomic = &self.count[val as usize];
         val_atomic.fetch_add(1, std::sync::atomic::Ordering::Relaxed);

chips/src/range_gate/trace.rs

@@ -1,10 +1,10 @@
-use p3_field::PrimeField64;
+use p3_field::PrimeField;
 use p3_matrix::dense::RowMajorMatrix;
 
 use super::{columns::NUM_RANGE_GATE_COLS, RangeCheckerGateChip};
 
 impl RangeCheckerGateChip {
-    pub fn generate_trace<F: PrimeField64>(&self) -> RowMajorMatrix<F> {
+    pub fn generate_trace<F: PrimeField>(&self) -> RowMajorMatrix<F> {
         let rows = self
             .count
             .iter()

chips/src/single_page_index_scan/mod.rs

@@ -0,0 +1,6 @@
+pub mod page_controller;
+pub mod page_index_scan_input;
+pub mod page_index_scan_output;
+
+#[cfg(test)]
+pub mod tests;