From 3df844558b54d577709848a12cdd3eba37566331 Mon Sep 17 00:00:00 2001
From: malatrax <simon.malatrait@grenoble-inp.org>
Date: Tue, 3 Dec 2024 10:28:19 +0100
Subject: [PATCH] feat: add log size of Memory interaction trace

---
 crates/brainfuck_prover/src/components/memory/component.rs | 2 +-
 crates/brainfuck_prover/src/components/mod.rs              | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/crates/brainfuck_prover/src/components/memory/component.rs b/crates/brainfuck_prover/src/components/memory/component.rs
index 4eb73c1..38dc7d4 100644
--- a/crates/brainfuck_prover/src/components/memory/component.rs
+++ b/crates/brainfuck_prover/src/components/memory/component.rs
@@ -187,7 +187,7 @@ mod tests {
 
         // Generate interaction trace
         let (interaction_trace, interaction_claim) =
-            interaction_trace_evaluation(&main_trace, &memory_lookup_elements);
+            interaction_trace_evaluation(&main_trace, &memory_lookup_elements).unwrap();
 
         // Create the trace evaluation TreeVec
         let trace = TreeVec::new(vec![preprocessed_trace, main_trace, interaction_trace]);
diff --git a/crates/brainfuck_prover/src/components/mod.rs b/crates/brainfuck_prover/src/components/mod.rs
index 37b5311..5b01e05 100644
--- a/crates/brainfuck_prover/src/components/mod.rs
+++ b/crates/brainfuck_prover/src/components/mod.rs
@@ -5,7 +5,7 @@ use processor::table::ProcessorColumn;
 use stwo_prover::core::{
     backend::simd::SimdBackend,
     channel::Channel,
-    fields::m31::BaseField,
+    fields::{m31::BaseField, secure_column::SECURE_EXTENSION_DEGREE},
     pcs::TreeVec,
     poly::{circle::CircleEvaluation, BitReversedOrder},
     ColumnVec,
@@ -75,7 +75,7 @@ impl<T: TraceColumn> Claim<T> {
         // TODO: Add the preprocessed and interaction trace correct sizes
         let preprocessed_trace_log_sizes: Vec<u32> = vec![self.log_size];
         let trace_log_sizes = vec![self.log_size; T::count()];
-        let interaction_trace_log_sizes: Vec<u32> = vec![];
+        let interaction_trace_log_sizes: Vec<u32> = vec![self.log_size; SECURE_EXTENSION_DEGREE];
         TreeVec::new(vec![
             preprocessed_trace_log_sizes,
             trace_log_sizes,