stav: extract information for the prover from the runner

CHANGELOG.md

@@ -2,6 +2,8 @@
 
 #### Upcoming Changes
 
+* feat: Add `ProverInfo` and extract the relevant information for it from the runner [#1982](https://github.com/lambdaclass/cairo-vm/pull/1982)
+
 #### [2.0.0] - 2025-02-26
 
 * fix: Check overflow in cairo pie address calculation [#1945](https://github.com/lambdaclass/cairo-vm/pull/1945)

vm/src/vm/runners/cairo_runner.rs

@@ -1477,6 +1477,28 @@ impl CairoRunner {
             })
             .collect()
     }
+
+    pub fn get_info_for_prover_input(&self) -> Result<ProverInfo, RunnerError> {
+        Ok(ProverInfo {
+            relocatble_trace: self
+            .vm
+            .trace.clone()
+            .ok_or(RunnerError::Trace(TraceError::TraceNotEnabled))?,
+            relocatble_memory: self.vm.segments.memory.data.clone(),
+            public_memory_addresses: self.vm.segments.public_memory_offsets.clone(),
+            builtins_segments: self.get_builtin_segment_info_for_pie()?
+        })
+    }
+}
+
+//* ----------------------
+//*   ProverInfo
+//* ---------------------- 
+pub struct ProverInfo {
+    pub relocatble_trace: Vec<TraceEntry>,
+    pub relocatble_memory: Vec<Vec<MemoryCell>>,
+    pub public_memory_addresses: HashMap<usize, Vec<(usize, usize)>>,
+    pub builtins_segments: HashMap<BuiltinName, cairo_pie::SegmentInfo>,
 }
 
 #[derive(Clone, Debug, Eq, PartialEq)]