@aztec/sequencer-client 0.26.2 → 0.26.5

package/src/simulator/rollup.ts

@@ -8,9 +8,20 @@ import {
 } from '@aztec/circuits.js';
 import { createDebugLogger } from '@aztec/foundation/log';
 import { elapsed } from '@aztec/foundation/timer';
-import { executeBaseRollup, executeMergeRollup, executeRootRollup } from '@aztec/noir-protocol-circuits-types';
+import {
+  BaseRollupArtifact,
+  MergeRollupArtifact,
+  RootRollupArtifact,
+  convertBaseRollupInputsToWitnessMap,
+  convertBaseRollupOutputsFromWitnessMap,
+  convertMergeRollupInputsToWitnessMap,
+  convertMergeRollupOutputsFromWitnessMap,
+  convertRootRollupInputsToWitnessMap,
+  convertRootRollupOutputsFromWitnessMap,
+} from '@aztec/noir-protocol-circuits-types';
 
-import { RollupSimulator } from './index.js';
+import { RollupSimulator, WASMSimulator } from './index.js';
+import { SimulationProvider } from './simulation_provider.js';
 
 /**
  * Implements the rollup circuit simulator.
@@ -18,21 +29,22 @@ import { RollupSimulator } from './index.js';
 export class RealRollupCircuitSimulator implements RollupSimulator {
   private log = createDebugLogger('aztec:rollup-simulator');
 
+  // Some circuits are so small it is faster to use WASM
+  private wasmSimulator: WASMSimulator = new WASMSimulator();
+
+  constructor(private simulationProvider: SimulationProvider) {}
+
   /**
    * Simulates the base rollup circuit from its inputs.
    * @param input - Inputs to the circuit.
    * @returns The public inputs as outputs of the simulation.
    */
   public async baseRollupCircuit(input: BaseRollupInputs): Promise<BaseOrMergeRollupPublicInputs> {
-    const [duration, result] = await elapsed(() => executeBaseRollup(input));
+    const witnessMap = convertBaseRollupInputsToWitnessMap(input);
 
-    this.log(`Simulated base rollup circuit`, {
-      eventName: 'circuit-simulation',
-      circuitName: 'base-rollup',
-      duration,
-      inputSize: input.toBuffer().length,
-      outputSize: result.toBuffer().length,
-    } satisfies CircuitSimulationStats);
+    const witness = await this.simulationProvider.simulateCircuit(witnessMap, BaseRollupArtifact);
+
+    const result = convertBaseRollupOutputsFromWitnessMap(witness);
 
     return Promise.resolve(result);
   }
@@ -42,15 +54,11 @@ export class RealRollupCircuitSimulator implements RollupSimulator {
    * @returns The public inputs as outputs of the simulation.
    */
   public async mergeRollupCircuit(input: MergeRollupInputs): Promise<BaseOrMergeRollupPublicInputs> {
-    const [duration, result] = await elapsed(() => executeMergeRollup(input));
+    const witnessMap = convertMergeRollupInputsToWitnessMap(input);
 
-    this.log(`Simulated merge rollup circuit`, {
-      eventName: 'circuit-simulation',
-      circuitName: 'merge-rollup',
-      duration,
-      inputSize: input.toBuffer().length,
-      outputSize: result.toBuffer().length,
-    } satisfies CircuitSimulationStats);
+    const witness = await this.wasmSimulator.simulateCircuit(witnessMap, MergeRollupArtifact);
+
+    const result = convertMergeRollupOutputsFromWitnessMap(witness);
 
     return result;
   }
@@ -61,7 +69,11 @@ export class RealRollupCircuitSimulator implements RollupSimulator {
    * @returns The public inputs as outputs of the simulation.
    */
   public async rootRollupCircuit(input: RootRollupInputs): Promise<RootRollupPublicInputs> {
-    const [duration, result] = await elapsed(() => executeRootRollup(input));
+    const witnessMap = convertRootRollupInputsToWitnessMap(input);
+
+    const [duration, witness] = await elapsed(() => this.wasmSimulator.simulateCircuit(witnessMap, RootRollupArtifact));
+
+    const result = convertRootRollupOutputsFromWitnessMap(witness);
 
     this.log(`Simulated root rollup circuit`, {
       eventName: 'circuit-simulation',

package/src/simulator/simulation_provider.ts

@@ -0,0 +1,10 @@
+import { NoirCompiledCircuit } from '@aztec/types/noir';
+
+import { WitnessMap } from '@noir-lang/types';
+
+/**
+ * Low level simulation interface
+ */
+export interface SimulationProvider {
+  simulateCircuit(input: WitnessMap, compiledCircuit: NoirCompiledCircuit): Promise<WitnessMap>;
+}