1 files changed, 73 insertions, 0 deletions
diff --git a/docs/util/YoscryptLexer.py b/docs/util/YoscryptLexer.py
new file mode 100644
index 000000000..8cb31c81a
--- /dev/null
+++ b/docs/util/YoscryptLexer.py
@@ -0,0 +1,73 @@
+from pygments.lexer import RegexLexer, bygroups, include
+from pygments.token import (Comment, Error, Keyword, Name, Number, Operator,
+                            String, Whitespace)
+
+__all__ = ['YoscryptLexer']
+
+class YoscryptLexer(RegexLexer):
+    name = 'Yosys Script'
+    aliases = ['yoscrypt']
+    filenames = ['*.ys']
+    
+    
+    
+    tokens = {
+        'common': [
+            (r'\s+', Whitespace),
+            (r'#.*', Comment.Single),
+            (r'"', String, 'string'),
+            (r'(\d+)(\')([bdho]? ?\w+)', bygroups(Number, Operator, Number)),
+            (r'(\d+\.\d+)', Number.Float),
+            (r'(\d+)', Number),
+            (r'(\$[A-Za-z_0-9]*)', Name.Builtin),
+            (r'([A-Za-z_][A-Za-z_0-9\.\\/:-]*)', Name),
+            (r'(\[)(-\S*)(\])', # optional command
+             	bygroups(Operator, Name.Attribute, Operator)),
+            (r'([\[<]\w*[\]>])', Name), # arguments
+            (r'[\{\}\|=\[\],]', Operator),
+            (r'.', Comment),
+        ],
+        'root': [
+            (r'([A-Za-z_][A-Za-z_0-9]*)', Keyword, 'command'),
+            (r'(-[A-Za-z_][A-Za-z_0-9]*)', Name.Attribute, 'command'), # shortcut for options
+            include('common'),
+        ],
+        'command': [
+            (r'(-[A-Za-z_][A-Za-z_0-9]*)', Name.Attribute),
+            (r'\+/[^\s]+', Name.Class),
+            (r'$', Whitespace, '#pop'),
+            (r';(?=\s)', Operator, '#pop'),
+            (r';{2,3}(?=\s)', Name.Class, '#pop'),
+            (r';{1,3}', Error, '#pop'),
+            (r'([ANwismctparn]:)', Keyword.Type, 'pattern'),
+            (r'@', Keyword.Type),
+            (r'%(x|ci|co)e?', Keyword.Type, 'expansion'),
+            (r'%[%nuidDcasmMCR]?', Keyword.Type),
+            (r'/', Operator),
+            include('common'),
+        ],
+        'pattern': [
+            (r'<<', Name), # Not an operator
+            (r'(=|<|<=|>|>=)', Operator),
+            (r':', Keyword.Type),
+            (r'$', Whitespace, '#pop:2'),
+            (r'\s+', Whitespace, '#pop'),
+            include('common'),
+        ],
+        'expansion': [
+            (r'$', Name.Class, '#pop:2'),
+            (r';(?=\s)', Operator, '#pop:2'),
+            (r';{2,3}(?=\s)', Name.Class, '#pop:2'),
+            (r'\s', Whitespace, '#pop'),
+            (r'[0-9\*]{1,3}', Number),
+            (r'[:+-,\[\]]', Operator),
+            include('common'),
+        ],
+        'string': [
+            (r'"', String, '#pop'),
+            (r'\\([\\abfnrtv"\']|x[a-fA-F0-9]{2,4}|[0-7]{1,3})', String.Escape),
+            (r'[^\\"\n]+', String),  # all other characters
+            (r'(\\)(\n)', bygroups(String.Escape, Whitespace)),  # line continuation
+            (r'\\', String),  # stray backslash
+        ]
+    }