[chore] add feed_input for keccak circuit (#229)

chore: add `feed_input` for keccak circuit
axiom-crypto · Dec 10, 2023 · 6e35ba3 · 6e35ba3
1 parent b6625fa
commit 6e35ba3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/hashes/zkevm/src/keccak/component/circuit/shard.rs b/hashes/zkevm/src/keccak/component/circuit/shard.rs
@@ -243,6 +243,13 @@ impl<F: Field> KeccakComponentShardCircuit<F> {
         }
     }
 
+    /// Set inputs of [KeccakComponentShardCircuit], **overwriting** any old inputs.
+    pub fn feed_input(&mut self, inputs: Vec<Vec<u8>>) {
+        let input_size = inputs.iter().map(|input| get_num_keccak_f(input.len())).sum::<usize>();
+        assert!(input_size < self.params.capacity, "Input size exceeds capacity");
+        self.inputs = inputs;
+    }
+
     /// Get break points of BaseCircuitBuilder.
     pub fn base_circuit_break_points(&self) -> MultiPhaseThreadBreakPoints {
         self.base_circuit_builder.borrow().break_points()