@aztec/prover-client 0.0.1-commit.7b97ef96e → 0.0.1-commit.7cbc774

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (82) hide show
  1. package/dest/config.d.ts +1 -1
  2. package/dest/config.d.ts.map +1 -1
  3. package/dest/config.js +16 -2
  4. package/dest/light/lightweight_checkpoint_builder.d.ts +6 -2
  5. package/dest/light/lightweight_checkpoint_builder.d.ts.map +1 -1
  6. package/dest/light/lightweight_checkpoint_builder.js +40 -11
  7. package/dest/mocks/test_context.d.ts +3 -1
  8. package/dest/mocks/test_context.d.ts.map +1 -1
  9. package/dest/mocks/test_context.js +14 -8
  10. package/dest/orchestrator/block-building-helpers.d.ts +1 -1
  11. package/dest/orchestrator/checkpoint-proving-state.d.ts +8 -1
  12. package/dest/orchestrator/checkpoint-proving-state.d.ts.map +1 -1
  13. package/dest/orchestrator/checkpoint-proving-state.js +10 -1
  14. package/dest/orchestrator/checkpoint-sub-tree-orchestrator.d.ts +107 -0
  15. package/dest/orchestrator/checkpoint-sub-tree-orchestrator.d.ts.map +1 -0
  16. package/dest/orchestrator/checkpoint-sub-tree-orchestrator.js +151 -0
  17. package/dest/orchestrator/epoch-proving-context.d.ts +51 -0
  18. package/dest/orchestrator/epoch-proving-context.d.ts.map +1 -0
  19. package/dest/orchestrator/epoch-proving-context.js +81 -0
  20. package/dest/orchestrator/epoch-proving-state.d.ts +1 -1
  21. package/dest/orchestrator/index.d.ts +4 -1
  22. package/dest/orchestrator/index.d.ts.map +1 -1
  23. package/dest/orchestrator/index.js +3 -0
  24. package/dest/orchestrator/orchestrator.d.ts +15 -25
  25. package/dest/orchestrator/orchestrator.d.ts.map +1 -1
  26. package/dest/orchestrator/orchestrator.js +33 -154
  27. package/dest/orchestrator/proving-scheduler.d.ts +72 -0
  28. package/dest/orchestrator/proving-scheduler.d.ts.map +1 -0
  29. package/dest/orchestrator/proving-scheduler.js +117 -0
  30. package/dest/orchestrator/top-tree-orchestrator.d.ts +83 -0
  31. package/dest/orchestrator/top-tree-orchestrator.d.ts.map +1 -0
  32. package/dest/orchestrator/top-tree-orchestrator.js +182 -0
  33. package/dest/orchestrator/top-tree-proving-scheduler.d.ts +62 -0
  34. package/dest/orchestrator/top-tree-proving-scheduler.d.ts.map +1 -0
  35. package/dest/orchestrator/top-tree-proving-scheduler.js +73 -0
  36. package/dest/orchestrator/top-tree-proving-state.d.ts +61 -0
  37. package/dest/orchestrator/top-tree-proving-state.d.ts.map +1 -0
  38. package/dest/orchestrator/top-tree-proving-state.js +185 -0
  39. package/dest/prover-client/prover-client.d.ts +62 -3
  40. package/dest/prover-client/prover-client.d.ts.map +1 -1
  41. package/dest/prover-client/prover-client.js +50 -2
  42. package/dest/proving_broker/broker_prover_facade.d.ts +1 -1
  43. package/dest/proving_broker/broker_prover_facade.d.ts.map +1 -1
  44. package/dest/proving_broker/broker_prover_facade.js +13 -19
  45. package/dest/proving_broker/config.d.ts +9 -73
  46. package/dest/proving_broker/config.d.ts.map +1 -1
  47. package/dest/proving_broker/config.js +3 -3
  48. package/dest/proving_broker/index.d.ts +2 -1
  49. package/dest/proving_broker/index.d.ts.map +1 -1
  50. package/dest/proving_broker/index.js +1 -0
  51. package/dest/proving_broker/proving_broker.d.ts +2 -2
  52. package/dest/proving_broker/proving_broker.d.ts.map +1 -1
  53. package/dest/proving_broker/proving_broker.js +39 -10
  54. package/dest/proving_broker/proving_broker_database/persisted.js +2 -2
  55. package/dest/proving_broker/proving_broker_instrumentation.d.ts +3 -1
  56. package/dest/proving_broker/proving_broker_instrumentation.d.ts.map +1 -1
  57. package/dest/proving_broker/proving_broker_instrumentation.js +7 -0
  58. package/dest/proving_broker/rpc.d.ts +3 -1
  59. package/dest/proving_broker/rpc.d.ts.map +1 -1
  60. package/dest/proving_broker/rpc.js +80 -24
  61. package/dest/test/mock_prover.d.ts +3 -3
  62. package/package.json +18 -19
  63. package/src/config.ts +18 -2
  64. package/src/light/lightweight_checkpoint_builder.ts +46 -20
  65. package/src/mocks/test_context.ts +12 -9
  66. package/src/orchestrator/checkpoint-proving-state.ts +14 -1
  67. package/src/orchestrator/checkpoint-sub-tree-orchestrator.ts +271 -0
  68. package/src/orchestrator/epoch-proving-context.ts +101 -0
  69. package/src/orchestrator/index.ts +8 -0
  70. package/src/orchestrator/orchestrator.ts +46 -210
  71. package/src/orchestrator/proving-scheduler.ts +156 -0
  72. package/src/orchestrator/top-tree-orchestrator.ts +314 -0
  73. package/src/orchestrator/top-tree-proving-scheduler.ts +154 -0
  74. package/src/orchestrator/top-tree-proving-state.ts +220 -0
  75. package/src/prover-client/prover-client.ts +127 -2
  76. package/src/proving_broker/broker_prover_facade.ts +17 -20
  77. package/src/proving_broker/config.ts +3 -2
  78. package/src/proving_broker/index.ts +1 -0
  79. package/src/proving_broker/proving_broker.ts +34 -7
  80. package/src/proving_broker/proving_broker_database/persisted.ts +2 -2
  81. package/src/proving_broker/proving_broker_instrumentation.ts +9 -0
  82. package/src/proving_broker/rpc.ts +36 -24
@@ -0,0 +1,182 @@
1
+ import { BatchedBlobAccumulator } from '@aztec/blob-lib';
2
+ import { BLOBS_PER_CHECKPOINT, FIELDS_PER_BLOB, OUT_HASH_TREE_HEIGHT } from '@aztec/constants';
3
+ import { padArrayEnd } from '@aztec/foundation/collection';
4
+ import { BLS12Point } from '@aztec/foundation/curves/bls12';
5
+ import { Fr } from '@aztec/foundation/curves/bn254';
6
+ import { promiseWithResolvers } from '@aztec/foundation/promise';
7
+ import { MerkleTreeCalculator, shaMerkleHash } from '@aztec/foundation/trees';
8
+ import { computeCheckpointOutHash } from '@aztec/stdlib/messaging';
9
+ import { CheckpointRootRollupHints, CheckpointRootRollupPrivateInputs, CheckpointRootSingleBlockRollupPrivateInputs } from '@aztec/stdlib/rollup';
10
+ import { AppendOnlyTreeSnapshot } from '@aztec/stdlib/trees';
11
+ import { getTelemetryClient } from '@aztec/telemetry-client';
12
+ import { buildBlobHints, toProofData } from './block-building-helpers.js';
13
+ import { TopTreeProvingScheduler } from './top-tree-proving-scheduler.js';
14
+ import { TopTreeProvingState } from './top-tree-proving-state.js';
15
+ /**
16
+ * Sentinel thrown by `cancel` so callers can distinguish reorg-driven cancellation from
17
+ * a genuine proving failure (and choose to rebuild + retry instead of failing the epoch).
18
+ */ export class TopTreeCancelledError extends Error {
19
+ constructor(reason = 'Top-tree proving cancelled'){
20
+ super(reason);
21
+ this.name = 'TopTreeCancelledError';
22
+ }
23
+ }
24
+ /**
25
+ * Drives proving from checkpoint root rollups through the epoch root rollup. Owns no
26
+ * world-state forks or tx processing — every input is supplied by the caller.
27
+ *
28
+ * Pipelined start: `prove()` does not wait for block-level proving. It pre-computes the
29
+ * out-hash and blob-accumulator hint chains immediately from archiver-derivable data,
30
+ * and each checkpoint's root rollup fires the moment its sub-tree's `blockProofs`
31
+ * promise resolves. Later checkpoints can still be block-level proving in parallel.
32
+ */ export class TopTreeOrchestrator extends TopTreeProvingScheduler {
33
+ proverId;
34
+ state;
35
+ cancelled;
36
+ constructor(prover, proverId, enqueueConcurrency, _telemetryClient = getTelemetryClient(), bindings){
37
+ super(prover, enqueueConcurrency, 'prover-client:top-tree-orchestrator', bindings), this.proverId = proverId, this.cancelled = false;
38
+ }
39
+ getProverId() {
40
+ return this.proverId;
41
+ }
42
+ /**
43
+ * Proves the top tree from per-checkpoint data and pending block-proofs promises.
44
+ * Resolves with the final epoch proof, or rejects with `TopTreeCancelledError` if
45
+ * `cancel()` is invoked, or any other error if a circuit fails.
46
+ */ async prove(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges, checkpointData) {
47
+ if (checkpointData.length !== totalNumCheckpoints) {
48
+ throw new Error(`checkpointData length (${checkpointData.length}) does not match totalNumCheckpoints (${totalNumCheckpoints}).`);
49
+ }
50
+ if (this.state) {
51
+ throw new Error('TopTreeOrchestrator.prove called twice; construct a new orchestrator per epoch.');
52
+ }
53
+ // If cancel() was already called before prove() ran (e.g. a removeCheckpoint that
54
+ // landed while the caller was still preparing inputs), short-circuit the whole
55
+ // proving path. Without this, prove() would build its state, the per-checkpoint
56
+ // .then handlers would all bail on `this.cancelled`, and the completion promise
57
+ // would never resolve — prove() would hang forever.
58
+ if (this.cancelled) {
59
+ throw new TopTreeCancelledError();
60
+ }
61
+ const { promise: completionPromise, resolve, reject } = promiseWithResolvers();
62
+ // The completion promise is awaited inside the try/catch below. Attach a no-op catch
63
+ // here as well so any spurious unhandled-rejection detection during cancellation
64
+ // (where reject() can fire synchronously before the await microtask installs a handler)
65
+ // is silenced.
66
+ completionPromise.catch(()=>{});
67
+ const startBlobAccumulator = BatchedBlobAccumulator.newWithChallenges(finalBlobBatchingChallenges);
68
+ this.state = new TopTreeProvingState(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges, startBlobAccumulator, resolve, (reason)=>reject(this.cancelled ? new TopTreeCancelledError(reason) : new Error(reason)));
69
+ // Compute the full out-hash hint chain and per-checkpoint start blob accumulators
70
+ // synchronously from archiver data. No proving required.
71
+ const outHashHints = await this.computeOutHashHints(checkpointData);
72
+ const checkpointStartBlobs = [];
73
+ let runningBlobAccumulator = startBlobAccumulator;
74
+ for (const cd of checkpointData){
75
+ checkpointStartBlobs.push(runningBlobAccumulator);
76
+ runningBlobAccumulator = await runningBlobAccumulator.accumulateFields(cd.blobFields);
77
+ }
78
+ this.state.setEndBlobAccumulator(runningBlobAccumulator);
79
+ // For each checkpoint, await its block proofs promise then enqueue the checkpoint root.
80
+ // Each await runs independently — checkpoints whose sub-trees finish first start their
81
+ // root proofs first, in parallel with later checkpoints' block-level proving.
82
+ for(let i = 0; i < checkpointData.length; i++){
83
+ const cd = checkpointData[i];
84
+ const checkpointIndex = i;
85
+ void cd.blockProofs.then((blockProofs)=>{
86
+ if (this.cancelled || !this.state?.verifyState()) {
87
+ return;
88
+ }
89
+ this.enqueueCheckpointRoot(this.state, checkpointIndex, blockProofs, cd, outHashHints[i], checkpointStartBlobs[i]);
90
+ }, (err)=>{
91
+ if (this.cancelled) {
92
+ return;
93
+ }
94
+ this.state?.reject(`Sub-tree for checkpoint ${i} failed: ${err}`);
95
+ });
96
+ }
97
+ try {
98
+ await completionPromise;
99
+ await this.state.finalizeBatchedBlob();
100
+ return this.state.getEpochProofResult();
101
+ } catch (err) {
102
+ if (this.cancelled) {
103
+ throw new TopTreeCancelledError();
104
+ }
105
+ throw err;
106
+ }
107
+ }
108
+ /**
109
+ * Cancels in-flight proving. If `abortJobs` is true, each pending broker job is aborted
110
+ * (used on reorg, when the surviving checkpoint set differs and the in-flight jobs'
111
+ * inputs are no longer valid). On shutdown the caller passes `false` so jobs remain in
112
+ * the broker queue for reuse on restart.
113
+ */ cancel({ abortJobs }) {
114
+ this.cancelled = true;
115
+ this.resetSchedulerState(abortJobs);
116
+ this.state?.cancel();
117
+ }
118
+ /** Standard shutdown — preserve the broker queue (`abortJobs: false`). */ cancelInternal() {
119
+ this.cancel({
120
+ abortJobs: false
121
+ });
122
+ }
123
+ // --- internal: per-checkpoint enqueue path ---
124
+ onRootRollupComplete(state) {
125
+ state.resolve();
126
+ }
127
+ enqueueCheckpointRoot(state, checkpointIndex, blockProofs, cd, outHashHint, startBlobAccumulator) {
128
+ void this.buildCheckpointRootInputs(blockProofs, cd, outHashHint, startBlobAccumulator).then((inputs)=>{
129
+ this.deferredProving(state, (signal)=>{
130
+ if (inputs instanceof CheckpointRootSingleBlockRollupPrivateInputs) {
131
+ return this.prover.getCheckpointRootSingleBlockRollupProof(inputs, signal, state.epochNumber);
132
+ }
133
+ return this.prover.getCheckpointRootRollupProof(inputs, signal, state.epochNumber);
134
+ }, (result)=>{
135
+ this.logger.debug(`Completed checkpoint root proof for checkpoint ${checkpointIndex}`);
136
+ const leafLocation = state.setCheckpointRootRollupProof(checkpointIndex, result);
137
+ if (state.totalNumCheckpoints === 1) {
138
+ this.enqueueEpochPadding(state);
139
+ } else {
140
+ this.checkAndEnqueueNextCheckpointMergeRollup(state, leafLocation);
141
+ }
142
+ });
143
+ });
144
+ }
145
+ async buildCheckpointRootInputs(blockProofs, cd, outHashHint, startBlobAccumulator) {
146
+ const { blobCommitments, blobsHash } = await buildBlobHints(cd.blobFields);
147
+ const hints = CheckpointRootRollupHints.from({
148
+ previousBlockHeader: cd.previousBlockHeader,
149
+ previousArchiveSiblingPath: cd.previousArchiveSiblingPath,
150
+ previousOutHash: outHashHint.treeSnapshot,
151
+ newOutHashSiblingPath: outHashHint.siblingPath,
152
+ startBlobAccumulator: startBlobAccumulator.toBlobAccumulator(),
153
+ finalBlobChallenges: this.state.finalBlobBatchingChallenges,
154
+ blobFields: padArrayEnd(cd.blobFields, Fr.ZERO, FIELDS_PER_BLOB * BLOBS_PER_CHECKPOINT),
155
+ blobCommitments: padArrayEnd(blobCommitments, BLS12Point.ZERO, BLOBS_PER_CHECKPOINT),
156
+ blobsHash
157
+ });
158
+ const proofDatas = blockProofs.map((p)=>toProofData(p));
159
+ return proofDatas.length === 1 ? new CheckpointRootSingleBlockRollupPrivateInputs(proofDatas[0], hints) : new CheckpointRootRollupPrivateInputs([
160
+ proofDatas[0],
161
+ proofDatas[1]
162
+ ], hints);
163
+ }
164
+ async computeOutHashHints(checkpointData) {
165
+ const treeCalculator = await MerkleTreeCalculator.create(OUT_HASH_TREE_HEIGHT, undefined, (left, right)=>Promise.resolve(shaMerkleHash(left, right)));
166
+ const computeHint = async (leaves)=>{
167
+ const tree = await treeCalculator.computeTree(leaves.map((l)=>l.toBuffer()));
168
+ const nextAvailableLeafIndex = leaves.length;
169
+ return {
170
+ treeSnapshot: new AppendOnlyTreeSnapshot(Fr.fromBuffer(tree.root), nextAvailableLeafIndex),
171
+ siblingPath: tree.getSiblingPath(nextAvailableLeafIndex).map(Fr.fromBuffer)
172
+ };
173
+ };
174
+ const hints = [];
175
+ const outHashes = [];
176
+ for (const cd of checkpointData){
177
+ hints.push(await computeHint(outHashes));
178
+ outHashes.push(computeCheckpointOutHash(cd.l2ToL1MsgsPerBlock));
179
+ }
180
+ return hints;
181
+ }
182
+ }
@@ -0,0 +1,62 @@
1
+ import type { NESTED_RECURSIVE_PROOF_LENGTH, NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH } from '@aztec/constants';
2
+ import type { EpochNumber } from '@aztec/foundation/branded-types';
3
+ import type { LoggerBindings } from '@aztec/foundation/log';
4
+ import type { TreeNodeLocation } from '@aztec/foundation/trees';
5
+ import type { PublicInputsAndRecursiveProof, ServerCircuitProver } from '@aztec/stdlib/interfaces/server';
6
+ import type { CheckpointMergeRollupPrivateInputs, CheckpointPaddingRollupPrivateInputs, CheckpointRollupPublicInputs, RootRollupPrivateInputs, RootRollupPublicInputs } from '@aztec/stdlib/rollup';
7
+ import { ProvingScheduler, type ProvingStateLike } from './proving-scheduler.js';
8
+ type CheckpointRollupProof = PublicInputsAndRecursiveProof<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>;
9
+ type RootRollupProof = PublicInputsAndRecursiveProof<RootRollupPublicInputs, typeof NESTED_RECURSIVE_PROOF_LENGTH>;
10
+ /**
11
+ * State interface required by the top-tree proving drivers (checkpoint-merge → padding →
12
+ * root-rollup). Both `EpochProvingState` and `TopTreeProvingState` satisfy it structurally;
13
+ * the per-checkpoint state in `EpochProvingState` (block/tx proving, world-state forks)
14
+ * is owned outside this surface.
15
+ */
16
+ export interface TopTreeStateLike extends ProvingStateLike {
17
+ readonly epochNumber: EpochNumber;
18
+ readonly totalNumCheckpoints: number;
19
+ tryStartProvingCheckpointMerge(location: TreeNodeLocation): boolean;
20
+ setCheckpointMergeRollupProof(location: TreeNodeLocation, provingOutput: CheckpointRollupProof): void;
21
+ isReadyForCheckpointMerge(location: TreeNodeLocation): boolean;
22
+ getParentLocation(location: TreeNodeLocation): TreeNodeLocation;
23
+ getCheckpointMergeRollupInputs(location: TreeNodeLocation): CheckpointMergeRollupPrivateInputs;
24
+ tryStartProvingPaddingCheckpoint(): boolean;
25
+ setCheckpointPaddingProof(provingOutput: CheckpointRollupProof): void;
26
+ getPaddingCheckpointInputs(): CheckpointPaddingRollupPrivateInputs;
27
+ tryStartProvingRootRollup(): boolean;
28
+ setRootRollupProof(provingOutput: RootRollupProof): void;
29
+ isReadyForRootRollup(): boolean;
30
+ getRootRollupInputs(): RootRollupPrivateInputs;
31
+ }
32
+ /**
33
+ * Shared scheduling for the top-tree section of epoch proving — checkpoint-merge,
34
+ * padding (single-checkpoint case), and root rollup. Both `ProvingOrchestrator` and
35
+ * `TopTreeOrchestrator` extend this; their per-checkpoint-root drivers diverge (one
36
+ * drains state-derived inputs once block-merge is done, the other builds inputs from
37
+ * caller-supplied checkpoint data), but the rest of the tree is identical.
38
+ *
39
+ * Subclasses provide a `wrapCircuitCall` hook for telemetry (the orchestrator wraps
40
+ * each call in a span; the top-tree leaves it as identity), and an
41
+ * `onRootRollupComplete` hook to invoke the right shape of `state.resolve()` —
42
+ * `EpochProvingState.resolve` takes a `ProvingResult`, `TopTreeProvingState.resolve`
43
+ * is no-arg.
44
+ */
45
+ export declare abstract class TopTreeProvingScheduler extends ProvingScheduler {
46
+ protected readonly prover: ServerCircuitProver;
47
+ constructor(prover: ServerCircuitProver, enqueueConcurrency: number, loggerName?: string, bindings?: LoggerBindings);
48
+ /**
49
+ * Wraps a circuit call for telemetry. Default is identity; the orchestrator overrides
50
+ * to wrap with `wrapCallbackInSpan`.
51
+ */
52
+ protected wrapCircuitCall<T>(_circuitName: string, fn: (signal: AbortSignal) => Promise<T>): (signal: AbortSignal) => Promise<T>;
53
+ /** Called once the root rollup proof has been set; subclasses call `state.resolve(...)` with the right shape. */
54
+ protected abstract onRootRollupComplete(state: TopTreeStateLike): void;
55
+ protected enqueueCheckpointMergeRollup(state: TopTreeStateLike, location: TreeNodeLocation): void;
56
+ protected enqueueEpochPadding(state: TopTreeStateLike): void;
57
+ protected enqueueRootRollup(state: TopTreeStateLike): void;
58
+ protected checkAndEnqueueNextCheckpointMergeRollup(state: TopTreeStateLike, currentLocation: TreeNodeLocation): void;
59
+ protected checkAndEnqueueRootRollup(state: TopTreeStateLike): void;
60
+ }
61
+ export {};
62
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidG9wLXRyZWUtcHJvdmluZy1zY2hlZHVsZXIuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9vcmNoZXN0cmF0b3IvdG9wLXRyZWUtcHJvdmluZy1zY2hlZHVsZXIudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUEsT0FBTyxLQUFLLEVBQUUsNkJBQTZCLEVBQUUseUNBQXlDLEVBQUUsTUFBTSxrQkFBa0IsQ0FBQztBQUNqSCxPQUFPLEtBQUssRUFBRSxXQUFXLEVBQUUsTUFBTSxpQ0FBaUMsQ0FBQztBQUNuRSxPQUFPLEtBQUssRUFBRSxjQUFjLEVBQUUsTUFBTSx1QkFBdUIsQ0FBQztBQUM1RCxPQUFPLEtBQUssRUFBRSxnQkFBZ0IsRUFBRSxNQUFNLHlCQUF5QixDQUFDO0FBQ2hFLE9BQU8sS0FBSyxFQUFFLDZCQUE2QixFQUFFLG1CQUFtQixFQUFFLE1BQU0saUNBQWlDLENBQUM7QUFDMUcsT0FBTyxLQUFLLEVBQ1Ysa0NBQWtDLEVBQ2xDLG9DQUFvQyxFQUNwQyw0QkFBNEIsRUFDNUIsdUJBQXVCLEVBQ3ZCLHNCQUFzQixFQUN2QixNQUFNLHNCQUFzQixDQUFDO0FBRTlCLE9BQU8sRUFBRSxnQkFBZ0IsRUFBRSxLQUFLLGdCQUFnQixFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFFakYsS0FBSyxxQkFBcUIsR0FBRyw2QkFBNkIsQ0FDeEQsNEJBQTRCLEVBQzVCLE9BQU8seUNBQXlDLENBQ2pELENBQUM7QUFFRixLQUFLLGVBQWUsR0FBRyw2QkFBNkIsQ0FBQyxzQkFBc0IsRUFBRSxPQUFPLDZCQUE2QixDQUFDLENBQUM7QUFFbkg7Ozs7O0dBS0c7QUFDSCxNQUFNLFdBQVcsZ0JBQWlCLFNBQVEsZ0JBQWdCO0lBQ3hELFFBQVEsQ0FBQyxXQUFXLEVBQUUsV0FBVyxDQUFDO0lBQ2xDLFFBQVEsQ0FBQyxtQkFBbUIsRUFBRSxNQUFNLENBQUM7SUFFckMsOEJBQThCLENBQUMsUUFBUSxFQUFFLGdCQUFnQixHQUFHLE9BQU8sQ0FBQztJQUNwRSw2QkFBNkIsQ0FBQyxRQUFRLEVBQUUsZ0JBQWdCLEVBQUUsYUFBYSxFQUFFLHFCQUFxQixHQUFHLElBQUksQ0FBQztJQUN0Ryx5QkFBeUIsQ0FBQyxRQUFRLEVBQUUsZ0JBQWdCLEdBQUcsT0FBTyxDQUFDO0lBQy9ELGlCQUFpQixDQUFDLFFBQVEsRUFBRSxnQkFBZ0IsR0FBRyxnQkFBZ0IsQ0FBQztJQUNoRSw4QkFBOEIsQ0FBQyxRQUFRLEVBQUUsZ0JBQWdCLEdBQUcsa0NBQWtDLENBQUM7SUFFL0YsZ0NBQWdDLElBQUksT0FBTyxDQUFDO0lBQzVDLHlCQUF5QixDQUFDLGFBQWEsRUFBRSxxQkFBcUIsR0FBRyxJQUFJLENBQUM7SUFDdEUsMEJBQTBCLElBQUksb0NBQW9DLENBQUM7SUFFbkUseUJBQXlCLElBQUksT0FBTyxDQUFDO0lBQ3JDLGtCQUFrQixDQUFDLGFBQWEsRUFBRSxlQUFlLEdBQUcsSUFBSSxDQUFDO0lBQ3pELG9CQUFvQixJQUFJLE9BQU8sQ0FBQztJQUNoQyxtQkFBbUIsSUFBSSx1QkFBdUIsQ0FBQztDQUNoRDtBQUVEOzs7Ozs7Ozs7Ozs7R0FZRztBQUNILDhCQUFzQix1QkFBd0IsU0FBUSxnQkFBZ0I7SUFFbEUsU0FBUyxDQUFDLFFBQVEsQ0FBQyxNQUFNLEVBQUUsbUJBQW1CO0lBRGhELFlBQ3FCLE1BQU0sRUFBRSxtQkFBbUIsRUFDOUMsa0JBQWtCLEVBQUUsTUFBTSxFQUMxQixVQUFVLENBQUMsRUFBRSxNQUFNLEVBQ25CLFFBQVEsQ0FBQyxFQUFFLGNBQWMsRUFHMUI7SUFFRDs7O09BR0c7SUFDSCxTQUFTLENBQUMsZUFBZSxDQUFDLENBQUMsRUFDekIsWUFBWSxFQUFFLE1BQU0sRUFDcEIsRUFBRSxFQUFFLENBQUMsTUFBTSxFQUFFLFdBQVcsS0FBSyxPQUFPLENBQUMsQ0FBQyxDQUFDLEdBQ3RDLENBQUMsTUFBTSxFQUFFLFdBQVcsS0FBSyxPQUFPLENBQUMsQ0FBQyxDQUFDLENBRXJDO0lBRUQsaUhBQWlIO0lBQ2pILFNBQVMsQ0FBQyxRQUFRLENBQUMsb0JBQW9CLENBQUMsS0FBSyxFQUFFLGdCQUFnQixHQUFHLElBQUksQ0FBQztJQUV2RSxTQUFTLENBQUMsNEJBQTRCLENBQUMsS0FBSyxFQUFFLGdCQUFnQixFQUFFLFFBQVEsRUFBRSxnQkFBZ0IsUUFlekY7SUFFRCxTQUFTLENBQUMsbUJBQW1CLENBQUMsS0FBSyxFQUFFLGdCQUFnQixRQWVwRDtJQUVELFNBQVMsQ0FBQyxpQkFBaUIsQ0FBQyxLQUFLLEVBQUUsZ0JBQWdCLFFBY2xEO0lBRUQsU0FBUyxDQUFDLHdDQUF3QyxDQUFDLEtBQUssRUFBRSxnQkFBZ0IsRUFBRSxlQUFlLEVBQUUsZ0JBQWdCLFFBVTVHO0lBRUQsU0FBUyxDQUFDLHlCQUF5QixDQUFDLEtBQUssRUFBRSxnQkFBZ0IsUUFLMUQ7Q0FDRiJ9
@@ -0,0 +1 @@
1
+ {"version":3,"file":"top-tree-proving-scheduler.d.ts","sourceRoot":"","sources":["../../src/orchestrator/top-tree-proving-scheduler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,6BAA6B,EAAE,yCAAyC,EAAE,MAAM,kBAAkB,CAAC;AACjH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AACnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,KAAK,EAAE,6BAA6B,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC1G,OAAO,KAAK,EACV,kCAAkC,EAClC,oCAAoC,EACpC,4BAA4B,EAC5B,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,gBAAgB,EAAE,KAAK,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAEjF,KAAK,qBAAqB,GAAG,6BAA6B,CACxD,4BAA4B,EAC5B,OAAO,yCAAyC,CACjD,CAAC;AAEF,KAAK,eAAe,GAAG,6BAA6B,CAAC,sBAAsB,EAAE,OAAO,6BAA6B,CAAC,CAAC;AAEnH;;;;;GAKG;AACH,MAAM,WAAW,gBAAiB,SAAQ,gBAAgB;IACxD,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC;IAErC,8BAA8B,CAAC,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC;IACpE,6BAA6B,CAAC,QAAQ,EAAE,gBAAgB,EAAE,aAAa,EAAE,qBAAqB,GAAG,IAAI,CAAC;IACtG,yBAAyB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC;IAC/D,iBAAiB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,gBAAgB,CAAC;IAChE,8BAA8B,CAAC,QAAQ,EAAE,gBAAgB,GAAG,kCAAkC,CAAC;IAE/F,gCAAgC,IAAI,OAAO,CAAC;IAC5C,yBAAyB,CAAC,aAAa,EAAE,qBAAqB,GAAG,IAAI,CAAC;IACtE,0BAA0B,IAAI,oCAAoC,CAAC;IAEnE,yBAAyB,IAAI,OAAO,CAAC;IACrC,kBAAkB,CAAC,aAAa,EAAE,eAAe,GAAG,IAAI,CAAC;IACzD,oBAAoB,IAAI,OAAO,CAAC;IAChC,mBAAmB,IAAI,uBAAuB,CAAC;CAChD;AAED;;;;;;;;;;;;GAYG;AACH,8BAAsB,uBAAwB,SAAQ,gBAAgB;IAElE,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,mBAAmB;IADhD,YACqB,MAAM,EAAE,mBAAmB,EAC9C,kBAAkB,EAAE,MAAM,EAC1B,UAAU,CAAC,EAAE,MAAM,EACnB,QAAQ,CAAC,EAAE,cAAc,EAG1B;IAED;;;OAGG;IACH,SAAS,CAAC,eAAe,CAAC,CAAC,EACzB,YAAY,EAAE,MAAM,EACpB,EAAE,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,CAAC,CAAC,GACtC,CAAC,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,CAAC,CAAC,CAErC;IAED,iHAAiH;IACjH,SAAS,CAAC,QAAQ,CAAC,oBAAoB,CAAC,KAAK,EAAE,gBAAgB,GAAG,IAAI,CAAC;IAEvE,SAAS,CAAC,4BAA4B,CAAC,KAAK,EAAE,gBAAgB,EAAE,QAAQ,EAAE,gBAAgB,QAezF;IAED,SAAS,CAAC,mBAAmB,CAAC,KAAK,EAAE,gBAAgB,QAepD;IAED,SAAS,CAAC,iBAAiB,CAAC,KAAK,EAAE,gBAAgB,QAclD;IAED,SAAS,CAAC,wCAAwC,CAAC,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,gBAAgB,QAU5G;IAED,SAAS,CAAC,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,QAK1D;CACF"}
@@ -0,0 +1,73 @@
1
+ import { ProvingScheduler } from './proving-scheduler.js';
2
+ /**
3
+ * Shared scheduling for the top-tree section of epoch proving — checkpoint-merge,
4
+ * padding (single-checkpoint case), and root rollup. Both `ProvingOrchestrator` and
5
+ * `TopTreeOrchestrator` extend this; their per-checkpoint-root drivers diverge (one
6
+ * drains state-derived inputs once block-merge is done, the other builds inputs from
7
+ * caller-supplied checkpoint data), but the rest of the tree is identical.
8
+ *
9
+ * Subclasses provide a `wrapCircuitCall` hook for telemetry (the orchestrator wraps
10
+ * each call in a span; the top-tree leaves it as identity), and an
11
+ * `onRootRollupComplete` hook to invoke the right shape of `state.resolve()` —
12
+ * `EpochProvingState.resolve` takes a `ProvingResult`, `TopTreeProvingState.resolve`
13
+ * is no-arg.
14
+ */ export class TopTreeProvingScheduler extends ProvingScheduler {
15
+ prover;
16
+ constructor(prover, enqueueConcurrency, loggerName, bindings){
17
+ super(enqueueConcurrency, loggerName, bindings), this.prover = prover;
18
+ }
19
+ /**
20
+ * Wraps a circuit call for telemetry. Default is identity; the orchestrator overrides
21
+ * to wrap with `wrapCallbackInSpan`.
22
+ */ wrapCircuitCall(_circuitName, fn) {
23
+ return fn;
24
+ }
25
+ enqueueCheckpointMergeRollup(state, location) {
26
+ if (!state.verifyState() || !state.tryStartProvingCheckpointMerge(location)) {
27
+ return;
28
+ }
29
+ const inputs = state.getCheckpointMergeRollupInputs(location);
30
+ this.deferredProving(state, this.wrapCircuitCall('rollup-checkpoint-merge', (signal)=>this.prover.getCheckpointMergeRollupProof(inputs, signal, state.epochNumber)), (result)=>{
31
+ state.setCheckpointMergeRollupProof(location, result);
32
+ this.checkAndEnqueueNextCheckpointMergeRollup(state, location);
33
+ });
34
+ }
35
+ enqueueEpochPadding(state) {
36
+ if (!state.verifyState() || !state.tryStartProvingPaddingCheckpoint()) {
37
+ return;
38
+ }
39
+ const inputs = state.getPaddingCheckpointInputs();
40
+ this.deferredProving(state, this.wrapCircuitCall('rollup-checkpoint-padding', (signal)=>this.prover.getCheckpointPaddingRollupProof(inputs, signal, state.epochNumber)), (result)=>{
41
+ state.setCheckpointPaddingProof(result);
42
+ this.checkAndEnqueueRootRollup(state);
43
+ });
44
+ }
45
+ enqueueRootRollup(state) {
46
+ if (!state.verifyState() || !state.tryStartProvingRootRollup()) {
47
+ return;
48
+ }
49
+ const inputs = state.getRootRollupInputs();
50
+ this.deferredProving(state, this.wrapCircuitCall('rollup-root', (signal)=>this.prover.getRootRollupProof(inputs, signal, state.epochNumber)), (result)=>{
51
+ this.logger.verbose(`Completed root rollup for epoch ${state.epochNumber}`);
52
+ state.setRootRollupProof(result);
53
+ this.onRootRollupComplete(state);
54
+ });
55
+ }
56
+ checkAndEnqueueNextCheckpointMergeRollup(state, currentLocation) {
57
+ if (!state.isReadyForCheckpointMerge(currentLocation)) {
58
+ return;
59
+ }
60
+ const parentLocation = state.getParentLocation(currentLocation);
61
+ if (parentLocation.level === 0) {
62
+ this.checkAndEnqueueRootRollup(state);
63
+ } else {
64
+ this.enqueueCheckpointMergeRollup(state, parentLocation);
65
+ }
66
+ }
67
+ checkAndEnqueueRootRollup(state) {
68
+ if (!state.isReadyForRootRollup()) {
69
+ return;
70
+ }
71
+ this.enqueueRootRollup(state);
72
+ }
73
+ }
@@ -0,0 +1,61 @@
1
+ import type { BatchedBlob, BatchedBlobAccumulator, FinalBlobBatchingChallenges } from '@aztec/blob-lib';
2
+ import type { NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH } from '@aztec/constants';
3
+ import { EpochNumber } from '@aztec/foundation/branded-types';
4
+ import { type TreeNodeLocation } from '@aztec/foundation/trees';
5
+ import type { PublicInputsAndRecursiveProof } from '@aztec/stdlib/interfaces/server';
6
+ import type { Proof } from '@aztec/stdlib/proofs';
7
+ import { CheckpointMergeRollupPrivateInputs, CheckpointPaddingRollupPrivateInputs, CheckpointRollupPublicInputs, RootRollupPrivateInputs, type RootRollupPublicInputs } from '@aztec/stdlib/rollup';
8
+ /**
9
+ * Lean top-tree-only state. Owns the merge tree of checkpoint root proofs, the
10
+ * single-checkpoint padding proof slot, the final root rollup proof, and the blob
11
+ * accumulator endpoints needed to finalise the epoch's batched blob proof.
12
+ *
13
+ * Constructed with `totalNumCheckpoints` and `finalBlobBatchingChallenges` upfront —
14
+ * by the time the top tree starts, all checkpoints are known and the challenges are
15
+ * derivable from their blob fields.
16
+ */
17
+ export declare class TopTreeProvingState {
18
+ #private;
19
+ readonly epochNumber: EpochNumber;
20
+ readonly totalNumCheckpoints: number;
21
+ readonly finalBlobBatchingChallenges: FinalBlobBatchingChallenges;
22
+ readonly startBlobAccumulator: BatchedBlobAccumulator;
23
+ private readonly completionCallback;
24
+ private readonly rejectionCallback;
25
+ private checkpointProofs;
26
+ private checkpointPaddingProof;
27
+ private rootRollupProof;
28
+ private endBlobAccumulator;
29
+ private finalBatchedBlob;
30
+ private lifecycle;
31
+ constructor(epochNumber: EpochNumber, totalNumCheckpoints: number, finalBlobBatchingChallenges: FinalBlobBatchingChallenges, startBlobAccumulator: BatchedBlobAccumulator, completionCallback: () => void, rejectionCallback: (reason: string) => void);
32
+ setCheckpointRootRollupProof(checkpointIndex: number, provingOutput: PublicInputsAndRecursiveProof<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>): TreeNodeLocation;
33
+ tryStartProvingCheckpointMerge(location: TreeNodeLocation): boolean;
34
+ setCheckpointMergeRollupProof(location: TreeNodeLocation, provingOutput: PublicInputsAndRecursiveProof<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>): void;
35
+ isReadyForCheckpointMerge(location: TreeNodeLocation): boolean;
36
+ getParentLocation(location: TreeNodeLocation): TreeNodeLocation;
37
+ getCheckpointMergeRollupInputs(mergeLocation: TreeNodeLocation): CheckpointMergeRollupPrivateInputs;
38
+ tryStartProvingPaddingCheckpoint(): boolean;
39
+ setCheckpointPaddingProof(provingOutput: PublicInputsAndRecursiveProof<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>): void;
40
+ getPaddingCheckpointInputs(): CheckpointPaddingRollupPrivateInputs;
41
+ tryStartProvingRootRollup(): boolean;
42
+ setRootRollupProof(provingOutput: PublicInputsAndRecursiveProof<RootRollupPublicInputs>): void;
43
+ isReadyForRootRollup(): boolean;
44
+ getRootRollupInputs(): RootRollupPrivateInputs;
45
+ /**
46
+ * Sets the end-of-epoch blob accumulator, computed by the top-tree orchestrator
47
+ * from the surviving checkpoints' blob fields. Required before `finalizeBatchedBlob`.
48
+ */
49
+ setEndBlobAccumulator(accumulator: BatchedBlobAccumulator): void;
50
+ finalizeBatchedBlob(): Promise<void>;
51
+ getEpochProofResult(): {
52
+ proof: Proof;
53
+ publicInputs: RootRollupPublicInputs;
54
+ batchedBlobInputs: BatchedBlob;
55
+ };
56
+ verifyState(): boolean;
57
+ resolve(): void;
58
+ reject(reason: string): void;
59
+ cancel(): void;
60
+ }
61
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidG9wLXRyZWUtcHJvdmluZy1zdGF0ZS5kLnRzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL29yY2hlc3RyYXRvci90b3AtdHJlZS1wcm92aW5nLXN0YXRlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sS0FBSyxFQUFFLFdBQVcsRUFBRSxzQkFBc0IsRUFBRSwyQkFBMkIsRUFBRSxNQUFNLGlCQUFpQixDQUFDO0FBQ3hHLE9BQU8sS0FBSyxFQUFpQyx5Q0FBeUMsRUFBRSxNQUFNLGtCQUFrQixDQUFDO0FBQ2pILE9BQU8sRUFBRSxXQUFXLEVBQUUsTUFBTSxpQ0FBaUMsQ0FBQztBQUM5RCxPQUFPLEVBQUUsS0FBSyxnQkFBZ0IsRUFBdUIsTUFBTSx5QkFBeUIsQ0FBQztBQUNyRixPQUFPLEtBQUssRUFBRSw2QkFBNkIsRUFBRSxNQUFNLGlDQUFpQyxDQUFDO0FBQ3JGLE9BQU8sS0FBSyxFQUFFLEtBQUssRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ2xELE9BQU8sRUFDTCxrQ0FBa0MsRUFDbEMsb0NBQW9DLEVBQ3BDLDRCQUE0QixFQUM1Qix1QkFBdUIsRUFDdkIsS0FBSyxzQkFBc0IsRUFDNUIsTUFBTSxzQkFBc0IsQ0FBQztBQVc5Qjs7Ozs7Ozs7R0FRRztBQUNILHFCQUFhLG1CQUFtQjs7YUFhWixXQUFXLEVBQUUsV0FBVzthQUN4QixtQkFBbUIsRUFBRSxNQUFNO2FBQzNCLDJCQUEyQixFQUFFLDJCQUEyQjthQUN4RCxvQkFBb0IsRUFBRSxzQkFBc0I7SUFDNUQsT0FBTyxDQUFDLFFBQVEsQ0FBQyxrQkFBa0I7SUFDbkMsT0FBTyxDQUFDLFFBQVEsQ0FBQyxpQkFBaUI7SUFqQnBDLE9BQU8sQ0FBQyxnQkFBZ0IsQ0FFdEI7SUFDRixPQUFPLENBQUMsc0JBQXNCLENBRWhCO0lBQ2QsT0FBTyxDQUFDLGVBQWUsQ0FBdUY7SUFDOUcsT0FBTyxDQUFDLGtCQUFrQixDQUFxQztJQUMvRCxPQUFPLENBQUMsZ0JBQWdCLENBQTBCO0lBQ2xELE9BQU8sQ0FBQyxTQUFTLENBQThCO0lBRS9DLFlBQ2tCLFdBQVcsRUFBRSxXQUFXLEVBQ3hCLG1CQUFtQixFQUFFLE1BQU0sRUFDM0IsMkJBQTJCLEVBQUUsMkJBQTJCLEVBQ3hELG9CQUFvQixFQUFFLHNCQUFzQixFQUMzQyxrQkFBa0IsRUFBRSxNQUFNLElBQUksRUFDOUIsaUJBQWlCLEVBQUUsQ0FBQyxNQUFNLEVBQUUsTUFBTSxLQUFLLElBQUksRUFNN0Q7SUFJTSw0QkFBNEIsQ0FDakMsZUFBZSxFQUFFLE1BQU0sRUFDdkIsYUFBYSxFQUFFLDZCQUE2QixDQUMxQyw0QkFBNEIsRUFDNUIsT0FBTyx5Q0FBeUMsQ0FDakQsR0FDQSxnQkFBZ0IsQ0FFbEI7SUFJTSw4QkFBOEIsQ0FBQyxRQUFRLEVBQUUsZ0JBQWdCLFdBTS9EO0lBRU0sNkJBQTZCLENBQ2xDLFFBQVEsRUFBRSxnQkFBZ0IsRUFDMUIsYUFBYSxFQUFFLDZCQUE2QixDQUMxQyw0QkFBNEIsRUFDNUIsT0FBTyx5Q0FBeUMsQ0FDakQsUUFHRjtJQUVNLHlCQUF5QixDQUFDLFFBQVEsRUFBRSxnQkFBZ0IsV0FFMUQ7SUFFTSxpQkFBaUIsQ0FBQyxRQUFRLEVBQUUsZ0JBQWdCLG9CQUVsRDtJQUVNLDhCQUE4QixDQUFDLGFBQWEsRUFBRSxnQkFBZ0Isc0NBTXBFO0lBSU0sZ0NBQWdDLFlBTXRDO0lBRU0seUJBQXlCLENBQzlCLGFBQWEsRUFBRSw2QkFBNkIsQ0FDMUMsNEJBQTRCLEVBQzVCLE9BQU8seUNBQXlDLENBQ2pELFFBR0Y7SUFFTSwwQkFBMEIseUNBRWhDO0lBSU0seUJBQXlCLFlBTS9CO0lBRU0sa0JBQWtCLENBQUMsYUFBYSxFQUFFLDZCQUE2QixDQUFDLHNCQUFzQixDQUFDLFFBRTdGO0lBRU0sb0JBQW9CLFlBRzFCO0lBRU0sbUJBQW1CLDRCQVF6QjtJQUlEOzs7T0FHRztJQUNJLHFCQUFxQixDQUFDLFdBQVcsRUFBRSxzQkFBc0IsUUFFL0Q7SUFFWSxtQkFBbUIsa0JBSy9CO0lBRU0sbUJBQW1CLElBQUk7UUFBRSxLQUFLLEVBQUUsS0FBSyxDQUFDO1FBQUMsWUFBWSxFQUFFLHNCQUFzQixDQUFDO1FBQUMsaUJBQWlCLEVBQUUsV0FBVyxDQUFBO0tBQUUsQ0FVbkg7SUFJTSxXQUFXLFlBRWpCO0lBRU0sT0FBTyxTQU1iO0lBRU0sTUFBTSxDQUFDLE1BQU0sRUFBRSxNQUFNLFFBTTNCO0lBRU0sTUFBTSxTQUVaO0NBUUYifQ==
@@ -0,0 +1 @@
1
+ {"version":3,"file":"top-tree-proving-state.d.ts","sourceRoot":"","sources":["../../src/orchestrator/top-tree-proving-state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,sBAAsB,EAAE,2BAA2B,EAAE,MAAM,iBAAiB,CAAC;AACxG,OAAO,KAAK,EAAiC,yCAAyC,EAAE,MAAM,kBAAkB,CAAC;AACjH,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,KAAK,gBAAgB,EAAuB,MAAM,yBAAyB,CAAC;AACrF,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,iCAAiC,CAAC;AACrF,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EACL,kCAAkC,EAClC,oCAAoC,EACpC,4BAA4B,EAC5B,uBAAuB,EACvB,KAAK,sBAAsB,EAC5B,MAAM,sBAAsB,CAAC;AAW9B;;;;;;;;GAQG;AACH,qBAAa,mBAAmB;;aAaZ,WAAW,EAAE,WAAW;aACxB,mBAAmB,EAAE,MAAM;aAC3B,2BAA2B,EAAE,2BAA2B;aACxD,oBAAoB,EAAE,sBAAsB;IAC5D,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAjBpC,OAAO,CAAC,gBAAgB,CAEtB;IACF,OAAO,CAAC,sBAAsB,CAEhB;IACd,OAAO,CAAC,eAAe,CAAuF;IAC9G,OAAO,CAAC,kBAAkB,CAAqC;IAC/D,OAAO,CAAC,gBAAgB,CAA0B;IAClD,OAAO,CAAC,SAAS,CAA8B;IAE/C,YACkB,WAAW,EAAE,WAAW,EACxB,mBAAmB,EAAE,MAAM,EAC3B,2BAA2B,EAAE,2BAA2B,EACxD,oBAAoB,EAAE,sBAAsB,EAC3C,kBAAkB,EAAE,MAAM,IAAI,EAC9B,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,EAM7D;IAIM,4BAA4B,CACjC,eAAe,EAAE,MAAM,EACvB,aAAa,EAAE,6BAA6B,CAC1C,4BAA4B,EAC5B,OAAO,yCAAyC,CACjD,GACA,gBAAgB,CAElB;IAIM,8BAA8B,CAAC,QAAQ,EAAE,gBAAgB,WAM/D;IAEM,6BAA6B,CAClC,QAAQ,EAAE,gBAAgB,EAC1B,aAAa,EAAE,6BAA6B,CAC1C,4BAA4B,EAC5B,OAAO,yCAAyC,CACjD,QAGF;IAEM,yBAAyB,CAAC,QAAQ,EAAE,gBAAgB,WAE1D;IAEM,iBAAiB,CAAC,QAAQ,EAAE,gBAAgB,oBAElD;IAEM,8BAA8B,CAAC,aAAa,EAAE,gBAAgB,sCAMpE;IAIM,gCAAgC,YAMtC;IAEM,yBAAyB,CAC9B,aAAa,EAAE,6BAA6B,CAC1C,4BAA4B,EAC5B,OAAO,yCAAyC,CACjD,QAGF;IAEM,0BAA0B,yCAEhC;IAIM,yBAAyB,YAM/B;IAEM,kBAAkB,CAAC,aAAa,EAAE,6BAA6B,CAAC,sBAAsB,CAAC,QAE7F;IAEM,oBAAoB,YAG1B;IAEM,mBAAmB,4BAQzB;IAID;;;OAGG;IACI,qBAAqB,CAAC,WAAW,EAAE,sBAAsB,QAE/D;IAEY,mBAAmB,kBAK/B;IAEM,mBAAmB,IAAI;QAAE,KAAK,EAAE,KAAK,CAAC;QAAC,YAAY,EAAE,sBAAsB,CAAC;QAAC,iBAAiB,EAAE,WAAW,CAAA;KAAE,CAUnH;IAIM,WAAW,YAEjB;IAEM,OAAO,SAMb;IAEM,MAAM,CAAC,MAAM,EAAE,MAAM,QAM3B;IAEM,MAAM,SAEZ;CAQF"}
@@ -0,0 +1,185 @@
1
+ import { UnbalancedTreeStore } from '@aztec/foundation/trees';
2
+ import { CheckpointMergeRollupPrivateInputs, CheckpointPaddingRollupPrivateInputs, RootRollupPrivateInputs } from '@aztec/stdlib/rollup';
3
+ import { toProofData } from './block-building-helpers.js';
4
+ var TOP_TREE_LIFECYCLE = /*#__PURE__*/ function(TOP_TREE_LIFECYCLE) {
5
+ TOP_TREE_LIFECYCLE[TOP_TREE_LIFECYCLE["CREATED"] = 0] = "CREATED";
6
+ TOP_TREE_LIFECYCLE[TOP_TREE_LIFECYCLE["RESOLVED"] = 1] = "RESOLVED";
7
+ TOP_TREE_LIFECYCLE[TOP_TREE_LIFECYCLE["REJECTED"] = 2] = "REJECTED";
8
+ return TOP_TREE_LIFECYCLE;
9
+ }(TOP_TREE_LIFECYCLE || {});
10
+ /**
11
+ * Lean top-tree-only state. Owns the merge tree of checkpoint root proofs, the
12
+ * single-checkpoint padding proof slot, the final root rollup proof, and the blob
13
+ * accumulator endpoints needed to finalise the epoch's batched blob proof.
14
+ *
15
+ * Constructed with `totalNumCheckpoints` and `finalBlobBatchingChallenges` upfront —
16
+ * by the time the top tree starts, all checkpoints are known and the challenges are
17
+ * derivable from their blob fields.
18
+ */ export class TopTreeProvingState {
19
+ epochNumber;
20
+ totalNumCheckpoints;
21
+ finalBlobBatchingChallenges;
22
+ startBlobAccumulator;
23
+ completionCallback;
24
+ rejectionCallback;
25
+ checkpointProofs;
26
+ checkpointPaddingProof;
27
+ rootRollupProof;
28
+ endBlobAccumulator;
29
+ finalBatchedBlob;
30
+ lifecycle;
31
+ constructor(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges, startBlobAccumulator, completionCallback, rejectionCallback){
32
+ this.epochNumber = epochNumber;
33
+ this.totalNumCheckpoints = totalNumCheckpoints;
34
+ this.finalBlobBatchingChallenges = finalBlobBatchingChallenges;
35
+ this.startBlobAccumulator = startBlobAccumulator;
36
+ this.completionCallback = completionCallback;
37
+ this.rejectionCallback = rejectionCallback;
38
+ this.lifecycle = 0;
39
+ if (totalNumCheckpoints < 1) {
40
+ throw new Error(`TopTreeProvingState requires at least one checkpoint; got ${totalNumCheckpoints}.`);
41
+ }
42
+ this.checkpointProofs = new UnbalancedTreeStore(totalNumCheckpoints);
43
+ }
44
+ // --- checkpoint root rollup ---
45
+ setCheckpointRootRollupProof(checkpointIndex, provingOutput) {
46
+ return this.checkpointProofs.setLeaf(checkpointIndex, {
47
+ provingOutput
48
+ });
49
+ }
50
+ // --- checkpoint merge rollup ---
51
+ tryStartProvingCheckpointMerge(location) {
52
+ if (this.checkpointProofs.getNode(location)?.isProving) {
53
+ return false;
54
+ }
55
+ this.checkpointProofs.setNode(location, {
56
+ isProving: true
57
+ });
58
+ return true;
59
+ }
60
+ setCheckpointMergeRollupProof(location, provingOutput) {
61
+ this.checkpointProofs.setNode(location, {
62
+ provingOutput
63
+ });
64
+ }
65
+ isReadyForCheckpointMerge(location) {
66
+ return !!this.checkpointProofs.getSibling(location)?.provingOutput;
67
+ }
68
+ getParentLocation(location) {
69
+ return this.checkpointProofs.getParentLocation(location);
70
+ }
71
+ getCheckpointMergeRollupInputs(mergeLocation) {
72
+ const [left, right] = this.checkpointProofs.getChildren(mergeLocation).map((c)=>c?.provingOutput);
73
+ if (!left || !right) {
74
+ throw new Error('At least one child is not ready for the checkpoint merge rollup.');
75
+ }
76
+ return new CheckpointMergeRollupPrivateInputs([
77
+ toProofData(left),
78
+ toProofData(right)
79
+ ]);
80
+ }
81
+ // --- padding (single-checkpoint case) ---
82
+ tryStartProvingPaddingCheckpoint() {
83
+ if (this.checkpointPaddingProof?.isProving) {
84
+ return false;
85
+ }
86
+ this.checkpointPaddingProof = {
87
+ isProving: true
88
+ };
89
+ return true;
90
+ }
91
+ setCheckpointPaddingProof(provingOutput) {
92
+ this.checkpointPaddingProof = {
93
+ provingOutput
94
+ };
95
+ }
96
+ getPaddingCheckpointInputs() {
97
+ return new CheckpointPaddingRollupPrivateInputs();
98
+ }
99
+ // --- root rollup ---
100
+ tryStartProvingRootRollup() {
101
+ if (this.rootRollupProof?.isProving) {
102
+ return false;
103
+ }
104
+ this.rootRollupProof = {
105
+ isProving: true
106
+ };
107
+ return true;
108
+ }
109
+ setRootRollupProof(provingOutput) {
110
+ this.rootRollupProof = {
111
+ provingOutput
112
+ };
113
+ }
114
+ isReadyForRootRollup() {
115
+ const childProofs = this.#getChildProofsForRoot();
116
+ return childProofs.every((p)=>!!p);
117
+ }
118
+ getRootRollupInputs() {
119
+ const [left, right] = this.#getChildProofsForRoot();
120
+ if (!left || !right) {
121
+ throw new Error('At least one child is not ready for the root rollup.');
122
+ }
123
+ return RootRollupPrivateInputs.from({
124
+ previousRollups: [
125
+ toProofData(left),
126
+ toProofData(right)
127
+ ]
128
+ });
129
+ }
130
+ // --- blob accumulator finalisation ---
131
+ /**
132
+ * Sets the end-of-epoch blob accumulator, computed by the top-tree orchestrator
133
+ * from the surviving checkpoints' blob fields. Required before `finalizeBatchedBlob`.
134
+ */ setEndBlobAccumulator(accumulator) {
135
+ this.endBlobAccumulator = accumulator;
136
+ }
137
+ async finalizeBatchedBlob() {
138
+ if (!this.endBlobAccumulator) {
139
+ throw new Error('End blob accumulator not set; call setEndBlobAccumulator before finalize.');
140
+ }
141
+ this.finalBatchedBlob = await this.endBlobAccumulator.finalize(true);
142
+ }
143
+ getEpochProofResult() {
144
+ const provingOutput = this.rootRollupProof?.provingOutput;
145
+ if (!provingOutput || !this.finalBatchedBlob) {
146
+ throw new Error('Top-tree proof not ready; root rollup or batched blob missing.');
147
+ }
148
+ return {
149
+ proof: provingOutput.proof.binaryProof,
150
+ publicInputs: provingOutput.inputs,
151
+ batchedBlobInputs: this.finalBatchedBlob
152
+ };
153
+ }
154
+ // --- lifecycle ---
155
+ verifyState() {
156
+ return this.lifecycle === 0;
157
+ }
158
+ resolve() {
159
+ if (!this.verifyState()) {
160
+ return;
161
+ }
162
+ this.lifecycle = 1;
163
+ this.completionCallback();
164
+ }
165
+ reject(reason) {
166
+ if (!this.verifyState()) {
167
+ return;
168
+ }
169
+ this.lifecycle = 2;
170
+ this.rejectionCallback(reason);
171
+ }
172
+ cancel() {
173
+ this.reject('Top-tree proving cancelled');
174
+ }
175
+ #getChildProofsForRoot() {
176
+ const rootLocation = {
177
+ level: 0,
178
+ index: 0
179
+ };
180
+ return this.totalNumCheckpoints === 1 ? [
181
+ this.checkpointProofs.getNode(rootLocation)?.provingOutput,
182
+ this.checkpointPaddingProof?.provingOutput
183
+ ] : this.checkpointProofs.getChildren(rootLocation).map((c)=>c?.provingOutput);
184
+ }
185
+ }