From cd939ee5538392059d68647e6d423c417b2b0b9f Mon Sep 17 00:00:00 2001
From: Bob McWhirter <bmcwhirt@redhat.com>
Date: Mon, 2 Aug 2021 11:05:50 -0400
Subject: [PATCH] Download cubeprogdb and point the parser to that.

---
 d                    | 5 +++++
 util/parse_memory.py | 4 ++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/d b/d
index e2ac187..9884a6c 100755
--- a/d
+++ b/d
@@ -18,6 +18,7 @@ case "$CMD" in
         ./d download-svd
         ./d download-headers
         ./d download-cubedb
+        ./d download-cubeprogdb
     ;;
     download-mcufinder)
         mkdir -p sources/mcufinder
@@ -53,6 +54,10 @@ case "$CMD" in
         rm -rf sources/cubedb
         git clone --depth 1 https://github.com/embassy-rs/stm32cube-database.git sources/cubedb
     ;;
+    download-cubeprogdb)
+        rm -rf sources/cubeprogdb
+        git clone --depth 1 https://github.com/embassy-rs/stm32cubeprog-database.git sources/cubeprogdb
+    ;;
     install-chiptool)
         cargo install --git https://github.com/embassy-rs/chiptool
     ;;
diff --git a/util/parse_memory.py b/util/parse_memory.py
index 661e961..3146414 100644
--- a/util/parse_memory.py
+++ b/util/parse_memory.py
@@ -113,11 +113,11 @@ def parse_files(dir):
 
         memories.append( chunk )
 
-dir = sys.argv[1]
+dir = "sources/cubeprogdb/db"
 
 parse_files(dir)
 
 with open('data/memories.yaml', 'w') as f:
     f.write(yaml.dump(memories, width=500))
 
-#print(yaml.dump(memories, width=500))
\ No newline at end of file
+#print(yaml.dump(memories, width=500))