added get_mem

diff --git a/src/soc/simple/test/teststate.py b/src/soc/simple/test/teststate.py
index a446d41503b8710502752acb5f6a213edb8f520b..3b7eeb208351547eadf235eafee18ca10feeae33 100644 (file)
--- a/src/soc/simple/test/teststate.py
+++ b/src/soc/simple/test/teststate.py
@@ -49,6 +49,19 @@ class HDLState(State):
         self.pcl.append(self.pc)
         log("class hdl pc", hex(self.pc))
 
+    def get_mem(self):
+        if hasattr(self.core.l0.pimem, 'lsui'):
+            hdlmem = self.core.l0.pimem.lsui.mem
+        else:
+            hdlmem = self.core.l0.pimem.mem
+            if not isinstance(hdlmem, Memory):
+                hdlmem = hdlmem.mem
+        self.mem = []
+        for i in range(hdlmem.depth):
+            value = yield hdlmem._array[i]
+            if value != 0:  # for comparison purposes
+                self.mem.append(((i*8), value))
+
 
 # add to State Factory
 state_add('hdl', HDLState)