From 3570b8b1e95328227246b9a26130ee4ff9c49471 Mon Sep 17 00:00:00 2001
From: Luke Kenneth Casson Leighton <lkcl@lkcl.net>
Date: Sun, 5 Apr 2020 15:32:49 +0100
Subject: [PATCH] add switch case and default keywords

---
 src/soc/decoder/pseudo/lexer.py    | 21 +++++++++++++++------
 src/soc/decoder/pseudo/pywriter.py |  3 ++-
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/src/soc/decoder/pseudo/lexer.py b/src/soc/decoder/pseudo/lexer.py
index bd84e2f1..7c1d6420 100644
--- a/src/soc/decoder/pseudo/lexer.py
+++ b/src/soc/decoder/pseudo/lexer.py
@@ -31,10 +31,13 @@ NO_INDENT = 0
 MAY_INDENT = 1
 MUST_INDENT = 2
 
-# turn into python-like colon syntax from pseudo-code syntax
+# turn into python-like colon syntax from pseudo-code syntax.
+# identify tokens which tell us whether a "hidden colon" is needed.
+# this in turn means that track_tokens_filter "works" without needing
+# complex grammar rules
 def python_colonify(lexer, tokens):
 
-    forwhile_seen = False
+    fake_colon_needed = False
     for token in tokens:
         #print ("track colon token", token, token.type)
 
@@ -47,15 +50,15 @@ def python_colonify(lexer, tokens):
             token = copy(token)
             token.type = "COLON"
             yield token
-        elif token.type in ['DO', 'WHILE', 'FOR']:
-            forwhile_seen = True
+        elif token.type in ['DO', 'WHILE', 'FOR', 'SWITCH', 'CASE', 'DEFAULT']:
+            fake_colon_needed = True
             yield token
         elif token.type == 'NEWLINE':
-            if forwhile_seen:
+            if fake_colon_needed:
                 ctok = copy(token)
                 ctok.type = "COLON"
                 yield ctok
-                forwhile_seen = False
+                fake_colon_needed = False
             yield token
         else:
             yield token
@@ -260,6 +263,9 @@ class PowerLexer:
         'BITAND',
         'BITXOR',
         'RETURN',
+        'SWITCH',
+        'CASE',
+        'DEFAULT',
         'WS',
         'NEWLINE',
         'COMMA',
@@ -328,6 +334,9 @@ class PowerLexer:
       "while": "WHILE",
       "do": "DO",
       "return": "RETURN",
+      "switch": "SWITCH",
+      "case": "CASE",
+      "default": "DEFAULT",
       }
 
     def t_NAME(self, t):
diff --git a/src/soc/decoder/pseudo/pywriter.py b/src/soc/decoder/pseudo/pywriter.py
index d8e60686..052ee95e 100644
--- a/src/soc/decoder/pseudo/pywriter.py
+++ b/src/soc/decoder/pseudo/pywriter.py
@@ -72,8 +72,9 @@ class PyISAWriter(ISA):
 
 if __name__ == '__main__':
     isa = PyISAWriter()
-    isa.write_pysource('stringldst')
+    isa.write_pysource('system')
     exit(0)
+    isa.write_pysource('stringldst')
     isa.write_pysource('fixedshift')
     isa.write_pysource('condition')
     isa.write_pysource('fixedtrap')
-- 
2.30.2