refactor: frontend of trace_decoder

Cargo.toml

@@ -1,5 +1,6 @@
 [workspace]
-members = ["mpt_trie",
+members = [
+    "mpt_trie",
     "smt_trie",
     "proof_gen",
     "trace_decoder",
@@ -11,8 +12,9 @@ members = ["mpt_trie",
     "zero_bin/ops",
     "zero_bin/verifier",
     "zero_bin/rpc",
-    "zero_bin/prover", 
-    "compat"]
+    "zero_bin/prover",
+    "compat",
+]
 resolver = "2"
 
 [workspace.package]
@@ -24,7 +26,8 @@ keywords = ["cryptography", "STARK", "plonky2", "ethereum", "zk"]
 categories = ["cryptography::cryptocurrencies"]
 
 [workspace.dependencies]
-alloy = { git = "https://github.com/alloy-rs/alloy", tag='v0.1.1', default-features = false, features = [
+__compat_primitive_types = { version = "0.12.2", package = "primitive-types" }
+alloy = { git = "https://github.com/alloy-rs/alloy", tag = 'v0.1.1', default-features = false, features = [
     "consensus",
     "reqwest",
     "json-rpc",
@@ -36,24 +39,26 @@ alloy = { git = "https://github.com/alloy-rs/alloy", tag='v0.1.1', default-featu
     "providers",
     "transports",
     "transport-http",
-    "rpc-types-debug"
+    "rpc-types-debug",
 ] }
 anyhow = "1.0.86"
 async-stream = "0.3.5"
 axum = "0.7.5"
+bitflags = "2.5.0"
+bitvec = "1.0.1"
 bytes = "1.6.0"
 ciborium = "0.2.2"
 ciborium-io = "0.2.2"
 clap = { version = "4.5.7", features = ["derive", "env"] }
 compat = { path = "compat" }
-__compat_primitive_types = { version = "0.12.2", package = "primitive-types" }
 criterion = "0.5.1"
 dotenvy = "0.15.7"
+either = "1.12.0"
 enum-as-inner = "0.6.0"
 enumn = "0.1.13"
 env_logger = "0.11.3"
-ethereum-types = "0.14.1"
 eth_trie = "0.4.0"
+ethereum-types = "0.14.1"
 evm_arithmetization = { path = "evm_arithmetization", version = "0.2.0" }
 futures = "0.3.30"
 hashbrown = "0.14.5"
@@ -70,6 +75,7 @@ mpt_trie = { path = "mpt_trie", version = "0.3.0" }
 num = "0.4.3"
 num-bigint = "0.4.5"
 num-traits = "0.2.19"
+nunny = "0.2.1"
 once_cell = "1.19.0"
 paladin-core = "0.4.2"
 parking_lot = "0.12.3"
@@ -88,8 +94,8 @@ serde = "1.0.203"
 serde_json = "1.0.118"
 serde_path_to_error = "0.1.16"
 serde_with = "3.8.1"
-smt_trie = { path = "smt_trie", version = "0.1.0" }
 sha2 = "0.10.8"
+smt_trie = { path = "smt_trie", version = "0.1.0" }
 static_assertions = "1.1.0"
 thiserror = "1.0.61"
 tiny-keccak = "2.0.2"
@@ -99,8 +105,11 @@ tower = "0.4"
 trace_decoder = { path = "trace_decoder", version = "0.4.0" }
 tracing = "0.1"
 tracing-subscriber = { version = "0.3", features = ["env-filter"] }
+u4 = "0.1.0"
 uint = "0.9.5"
 url = "2.5.2"
+winnow = "0.6.13"
+
 
 # zero-bin related dependencies
 ops = { path = "zero_bin/ops" }

trace_decoder/Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "trace_decoder"
-description = "Processes trace payloads into Intermediate Representation (IR) format."
-authors = ["Polygon Zero <[email protected]>"]
+description = "Ethereum node witness -> Prover input"
+authors = ["Polygon Zero"]
 version = "0.4.0"
 edition.workspace = true
 license.workspace = true
@@ -10,25 +10,33 @@ homepage.workspace = true
 keywords.workspace = true
 
 [dependencies]
+anyhow.workspace = true
+bitflags = { workspace = true }
+bitvec = { workspace = true }
+bytes = { workspace = true }
 ciborium = { workspace = true }
+ciborium-io = { workspace = true }
+either = { workspace = true }
 enum-as-inner = { workspace = true }
-enumn = { workspace = true }
 ethereum-types = { workspace = true }
+evm_arithmetization = { workspace = true }
 hex = { workspace = true }
+hex-literal = { workspace = true }
+itertools.workspace = true
 keccak-hash = { workspace = true }
 log = { workspace = true }
+mpt_trie = { workspace = true }
+nunny = { workspace = true, features = ["serde"] }
+plonky2 = { workspace = true }
 rlp = { workspace = true }
 serde = { workspace = true }
-serde_with = { workspace = true }
+smt_trie = { workspace = true }
 thiserror = { workspace = true }
-
-# Local dependencies
-mpt_trie = { workspace = true }
-evm_arithmetization = { workspace = true }
+u4 = { workspace = true }
+winnow = { workspace = true }
 
 [dev-dependencies]
 criterion = { workspace = true }
-pretty_env_logger = { workspace = true }
 serde_json = { workspace = true }
 
 [[bench]]

trace_decoder/benches/block_processing.rs

@@ -5,37 +5,27 @@
 //! (<https://etherscan.io/block/19240650>) containing 201 transactions and 16 withdrawals.
 
 use criterion::{criterion_group, criterion_main, BatchSize, Criterion};
-use serde::{Deserialize, Serialize};
-use trace_decoder::{
-    processed_block_trace::ProcessingMeta,
-    trace_protocol::BlockTrace,
-    types::{CodeHash, OtherBlockData},
-};
+use trace_decoder::{BlockTrace, OtherBlockData};
 
-#[derive(Clone, Debug, Deserialize, Serialize)]
+#[derive(Clone, Debug, serde::Deserialize)]
 pub struct ProverInput {
     pub block_trace: BlockTrace,
     pub other_data: OtherBlockData,
 }
 
-fn resolve_code_hash_fn(_: &CodeHash) -> Vec<u8> {
-    todo!()
-}
-
 fn criterion_benchmark(c: &mut Criterion) {
-    let bytes = std::fs::read("benches/block_input.json").unwrap();
-    let prover_input: ProverInput = serde_json::from_slice(&bytes).unwrap();
+    let prover_input =
+        serde_json::from_slice::<ProverInput>(include_bytes!("block_input.json").as_slice())
+            .unwrap();
 
     c.bench_function("Block 19240650 processing", |b| {
         b.iter_batched(
             || prover_input.clone(),
-            |pi| {
-                pi.block_trace
-                    .into_txn_proof_gen_ir(
-                        &ProcessingMeta::new(resolve_code_hash_fn),
-                        prover_input.other_data.clone(),
-                    )
-                    .unwrap()
+            |ProverInput {
+                 block_trace,
+                 other_data,
+             }| {
+                trace_decoder::entrypoint(block_trace, other_data, |_| unimplemented!()).unwrap()
             },
             BatchSize::LargeInput,
         )