Repository: alecthomas/chroma
Pull request #239 (2019-03-07 10:06:35)

add matlab lexer
add support for matlab

alecthomas: This is great, thank you!

Patch:
diff --git a/lexers/m/matlab.go b/lexers/m/matlab.go
new file mode 100644
index 0000000..5b0baa5
--- /dev/null
+++ b/lexers/m/matlab.go
@@ -0,0 +1,51 @@
+package m
+
+import (
+	. "github.com/alecthomas/chroma" // nolint
+	"github.com/alecthomas/chroma/lexers/internal"
+)
+
+// Matlab lexer.
+var Matlab = internal.Register(MustNewLexer(
+	&Config{
+		Name:      "Matlab",
+		Aliases:   []string{"matlab"},
+		Filenames: []string{"*.m"},
+		MimeTypes: []string{"text/matlab"},
+	},
+	Rules{
+		"root": {
+			{`\n`, Text, nil},
+			{`^!.*`, LiteralStringOther, nil},
+			{`%\{\s*\n`, CommentMultiline, Push("blockcomment")},
+			{`%.*$`, Comment, nil},
+			{`^\s*function`, Keyword, Push("deffunc")},
+			{Words(``, `\b`, `break`, `case`, `catch`, `classdef`, `continue`, `else`, `elseif`, `end`, `enumerated`, `events`, `for`, `function`, `global`, `if`, `methods`, `otherwise`, `parfor`, `persistent`, `properties`, `return`, `spmd`, `switch`, `try`, `while`), Keyword, nil},
+			{`(sin|sind|sinh|asin|asind|asinh|cos|cosd|cosh|acos|acosd|acosh|tan|tand|tanh|atan|atand|atan2|atanh|sec|secd|sech|asec|asecd|asech|csc|cscd|csch|acsc|acscd|acsch|cot|cotd|coth|acot|acotd|acoth|hypot|exp|expm1|log|log1p|log10|log2|pow2|realpow|reallog|realsqrt|sqrt|nthroot|nextpow2|abs|angle|complex|conj|imag|real|unwrap|isreal|cplxpair|fix|floor|ceil|round|mod|rem|sign|airy|besselj|bessely|besselh|besseli|besselk|beta|betainc|betaln|ellipj|ellipke|erf|erfc|erfcx|erfinv|expint|gamma|gammainc|gammaln|psi|legendre|cross|dot|factor|isprime|primes|gcd|lcm|rat|rats|perms|nchoosek|factorial|cart2sph|cart2pol|pol2cart|sph2cart|hsv2rgb|rgb2hsv|zeros|ones|eye|repmat|rand|randn|linspace|logspace|freqspace|meshgrid|accumarray|size|length|ndims|numel|disp|isempty|isequal|isequalwithequalnans|cat|reshape|diag|blkdiag|tril|triu|fliplr|flipud|flipdim|rot90|find|end|sub2ind|ind2sub|bsxfun|ndgrid|permute|ipermute|shiftdim|circshift|squeeze|isscalar|isvector|ans|eps|realmax|realmin|pi|i|inf|nan|isnan|isinf|isfinite|j|why|compan|gallery|hadamard|hankel|hilb|invhilb|magic|pascal|rosser|toeplitz|vander|wilkinson)\b`, NameBuiltin, nil},
+			{`\.\.\..*$`, Comment, nil},
+			{`-|==|~=|<|>|<=|>=|&&|&|~|\|\|?`, Operator, nil},
+			{`\.\*|\*|\+|\.\^|\.\\|\.\/|\/|\\`, Operator, nil},
+			{`\[|\]|\(|\)|\{|\}|:|@|\.|,`, Punctuation, nil},
+			{`=|:|;`, Punctuation, nil},
+			{`(?<=[\w)\].])\'+`, Operator, nil},
+			{`(\d+\.\d*|\d*\.\d+)([eEf][+-]?[0-9]+)?`, LiteralNumberFloat, nil},
+			{`\d+[eEf][+-]?[0-9]+`, LiteralNumberFloat, nil},
+			{`\d+`, LiteralNumberInteger, nil},
+			{`(?<![\w)\].])\'`, LiteralString, Push("string")},
+			{`[a-zA-Z_]\w*`, Name, nil},
+			{`.`, Text, nil},
+		},
+		"string": {
+			{`[^\']*\'`, LiteralString, Pop(1)},
+		},
+		"blockcomment": {
+			{`^\s*%\}`, CommentMultiline, Pop(1)},
+			{`^.*\n`, CommentMultiline, nil},
+			{`.`, CommentMultiline, nil},
+		},
+		"deffunc": {
+			{`(\s*)(?:(.+)(\s*)(=)(\s*))?(.+)(\()(.*)(\))(\s*)`, ByGroups(TextWhitespace, Text, TextWhitespace, Punctuation, TextWhitespace, NameFunction, Punctuation, Text, Punctuation, TextWhitespace), Pop(1)},
+			{`(\s*)([a-zA-Z_]\w*)`, ByGroups(Text, NameFunction), Pop(1)},
+		},
+	},
+))


Test patch:
diff --git a/lexers/testdata/matlab.actual b/lexers/testdata/matlab.actual
new file mode 100644
index 0000000..1e63fac
--- /dev/null
+++ b/lexers/testdata/matlab.actual
@@ -0,0 +1,10 @@
+% a sum function
+function p = vector_sum(x)
+p = 0;
+for k1 = 1:length(x)
+    p = p + x(k1);
+end
+end
+
+z = 1:10;
+sum_from_1_to_10 = vector_sum(z)
diff --git a/lexers/testdata/matlab.expected b/lexers/testdata/matlab.expected
new file mode 100644
index 0000000..392acbc
--- /dev/null
+++ b/lexers/testdata/matlab.expected
@@ -0,0 +1,69 @@
+[
+  {"type":"Comment","value":"% a sum function"},
+  {"type":"Text","value":"\n"},
+  {"type":"Keyword","value":"function"},
+  {"type":"TextWhitespace","value":" "},
+  {"type":"Text","value":"p "},
+  {"type":"Punctuation","value":"="},
+  {"type":"TextWhitespace","value":" "},
+  {"type":"NameFunction","value":"vector_sum"},
+  {"type":"Punctuation","value":"("},
+  {"type":"Text","value":"x"},
+  {"type":"Punctuation","value":")"},
+  {"type":"TextWhitespace","value":"\n"},
+  {"type":"Name","value":"p"},
+  {"type":"Text","value":" "},
+  {"type":"Punctuation","value":"="},
+  {"type":"Text","value":" "},
+  {"type":"LiteralNumberInteger","value":"0"},
+  {"type":"Punctuation","value":";"},
+  {"type":"Text","value":"\n"},
+  {"type":"Keyword","value":"for"},
+  {"type":"Text","value":" "},
+  {"type":"Name","value":"k1"},
+  {"type":"Text","value":" "},
+  {"type":"Punctuation","value":"="},
+  {"type":"Text","value":" "},
+  {"type":"LiteralNumberInteger","value":"1"},
+  {"type":"Punctuation","value":":"},
+  {"type":"NameBuiltin","value":"length"},
+  {"type":"Punctuation","value":"("},
+  {"type":"Name","value":"x"},
+  {"type":"Punctuation","value":")"},
+  {"type":"Text","value":"\n    "},
+  {"type":"Name","value":"p"},
+  {"type":"Text","value":" "},
+  {"type":"Punctuation","value":"="},
+  {"type":"Text","value":" "},
+  {"type":"Name","value":"p"},
+  {"type":"Text","value":" "},
+  {"type":"Operator","value":"+"},
+  {"type":"Text","value":" "},
+  {"type":"Name","value":"x"},
+  {"type":"Punctuation","value":"("},
+  {"type":"Name","value":"k1"},
+  {"type":"Punctuation","value":");"},
+  {"type":"Text","value":"\n"},
+  {"type":"Keyword","value":"end"},
+  {"type":"Text","value":"\n"},
+  {"type":"Keyword","value":"end"},
+  {"type":"Text","value":"\n\n"},
+  {"type":"Name","value":"z"},
+  {"type":"Text","value":" "},
+  {"type":"Punctuation","value":"="},
+  {"type":"Text","value":" "},
+  {"type":"LiteralNumberInteger","value":"1"},
+  {"type":"Punctuation","value":":"},
+  {"type":"LiteralNumberInteger","value":"10"},
+  {"type":"Punctuation","value":";"},
+  {"type":"Text","value":"\n"},
+  {"type":"Name","value":"sum_from_1_to_10"},
+  {"type":"Text","value":" "},
+  {"type":"Punctuation","value":"="},
+  {"type":"Text","value":" "},
+  {"type":"Name","value":"vector_sum"},
+  {"type":"Punctuation","value":"("},
+  {"type":"Name","value":"z"},
+  {"type":"Punctuation","value":")"},
+  {"type":"Text","value":"\n"}
+]