From 6fcbf3e7a56ff1a29a84ec9ebf93af1ced19e4fa Mon Sep 17 00:00:00 2001
From: Pieter Donker <donker@astron.nl>
Date: Thu, 16 Mar 2017 13:01:41 +0000
Subject: [PATCH] Task #893 backup

---
 tools/oneclick/prestudy/YAML/mm_menu.py              | 6 +++++-
 tools/oneclick/prestudy/YAML/py_mm_lib/system_rom.py | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/tools/oneclick/prestudy/YAML/mm_menu.py b/tools/oneclick/prestudy/YAML/mm_menu.py
index 58082a4425..a565da47b8 100755
--- a/tools/oneclick/prestudy/YAML/mm_menu.py
+++ b/tools/oneclick/prestudy/YAML/mm_menu.py
@@ -35,7 +35,7 @@ from py_mm_lib import *
 
 
 def main():
-    """ main function """
+    """ main """
     while True:
         menu_choice = show_menu().lower()
         if menu_choice == 'q':
@@ -68,10 +68,12 @@ def show_menu():
 
 
 def get_filename():
+    """ ask user for filename to load """
     return raw_input("wich file: ")
 
 
 def generate_report(filename):
+    """ generate report (pdf) for loaded file. """
     try:
         config = yaml.load(open(filename, "r"))
         name = config.keys()[0]
@@ -92,6 +94,7 @@ def generate_report(filename):
 
 
 def show_overview(filename):
+    """ show overview of loaded file """
     try:
         config = yaml.load(open(filename, "r"))
         name = config.keys()[0]
@@ -110,6 +113,7 @@ def generate_system_rom(filename):
     """ generate system_rom reg and txt file """
     system_rom = SystemRom()
     system_rom.read_system(filename)
+    system_rom.calc_offsets()
     system_rom.generate_reg()
     system_rom.generate_txt()
 
diff --git a/tools/oneclick/prestudy/YAML/py_mm_lib/system_rom.py b/tools/oneclick/prestudy/YAML/py_mm_lib/system_rom.py
index 9dc5656001..1dab7a45aa 100644
--- a/tools/oneclick/prestudy/YAML/py_mm_lib/system_rom.py
+++ b/tools/oneclick/prestudy/YAML/py_mm_lib/system_rom.py
@@ -70,7 +70,7 @@ class SystemRom(object):
         for name in slave_order:
             peripheral = peripherals[name]
             for key, val in peripheral.rams.items():
-                width = int(val['with'])
+                width = int(val['width'])
                 depth = int(val['depth'])
                 n_words = int(ceil(float(width) / self.word_size))
                 n_addresses = n_words * depth
-- 
GitLab