From 22b72d753f92d323e33f4887743094a3dcabcaa0 Mon Sep 17 00:00:00 2001
From: Franklin Delehelle <franklin.delehelle@odena.eu>
Date: Sun, 17 Dec 2023 10:35:10 +0100
Subject: [PATCH] perf: faster line count for the hub

---
 .../java/net/consensys/linea/zktracer/module/hub/Hub.java | 8 ++++----
 .../consensys/linea/zktracer/module/romLex/RomLex.java    | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/arithmetization/src/main/java/net/consensys/linea/zktracer/module/hub/Hub.java b/arithmetization/src/main/java/net/consensys/linea/zktracer/module/hub/Hub.java
index 1fe325bc01..129418af6d 100644
--- a/arithmetization/src/main/java/net/consensys/linea/zktracer/module/hub/Hub.java
+++ b/arithmetization/src/main/java/net/consensys/linea/zktracer/module/hub/Hub.java
@@ -237,7 +237,7 @@ public Hub() {
                     this.rom,
                     this.shf,
                     this.trm,
-//                    this.txnData,
+                    //                    this.txnData,
                     this.stp,
                     this.wcp),
                 this.precompileLimitModules.stream())
@@ -271,7 +271,7 @@ public List<Module> getModulesToTrace() {
         this.romLex,
         this.shf,
         this.stp,
-//        this.txnData,
+        //        this.txnData,
         this.wcp);
   }
 
@@ -296,7 +296,7 @@ public List<Module> getModulesToCount() {
                 this.rom,
                 this.shf,
                 this.trm,
-//                this.txnData,
+                //                this.txnData,
                 this.wcp),
             this.precompileLimitModules.stream())
         .toList();
@@ -508,7 +508,7 @@ void triggerModules(MessageFrame frame) {
       // TODO: this.oob.tracePreOpcode(frame);
     }
     if (this.pch.signals().stp()) {
-//      this.stp.tracePreOpcode(frame);
+      //      this.stp.tracePreOpcode(frame);
     }
     if (this.pch.signals().exp()) {
       this.modexp.tracePreOpcode(frame);
diff --git a/arithmetization/src/main/java/net/consensys/linea/zktracer/module/romLex/RomLex.java b/arithmetization/src/main/java/net/consensys/linea/zktracer/module/romLex/RomLex.java
index a1c4b17402..39239ee02f 100644
--- a/arithmetization/src/main/java/net/consensys/linea/zktracer/module/romLex/RomLex.java
+++ b/arithmetization/src/main/java/net/consensys/linea/zktracer/module/romLex/RomLex.java
@@ -244,7 +244,7 @@ public void tracePreOpcode(MessageFrame frame) {
                     codeIdentifierBeforeLexOrder += 1;
                     this.chunks.add(
                         new RomChunk(
-                            extAddress,
+                            calledAddress,
                             depNumber,
                             isDeploying,
                             true,