88 lines
2.8 KiB
Ruby
88 lines
2.8 KiB
Ruby
|
# -*- coding: utf-8 -*- #
|
||
|
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
# Based on Chroma's MiniZinc lexer:
|
||
|
|
# https://github.com/alecthomas/chroma/blob/5152194c717b394686d3d7a7e1946a360ec0728f/lexers/m/minizinc.go
|
||
|
|
|
||
|
|
module Rouge
|
||
|
|
module Lexers
|
||
|
|
class MiniZinc < RegexLexer
|
||
|
|
title "MiniZinc"
|
||
|
|
desc "MiniZinc is a free and open-source constraint modeling language (minizinc.org)"
|
||
|
|
tag 'minizinc'
|
||
|
|
filenames '*.mzn', '*.fzn', '*.dzn'
|
||
|
|
mimetypes 'text/minizinc'
|
||
|
|
|
||
|
|
def self.builtins
|
||
|
|
@builtins = Set.new %w[
|
||
|
|
abort abs acosh array_intersect array_union array1d array2d array3d
|
||
|
|
array4d array5d array6d asin assert atan bool2int card ceil concat
|
||
|
|
cos cosh dom dom_array dom_size fix exp floor index_set
|
||
|
|
index_set_1of2 index_set_2of2 index_set_1of3 index_set_2of3
|
||
|
|
index_set_3of3 int2float is_fixed join lb lb_array length ln log log2
|
||
|
|
log10 min max pow product round set2array show show_int show_float
|
||
|
|
sin sinh sqrt sum tan tanh trace ub ub_array
|
||
|
|
]
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.keywords
|
||
|
|
@keywords = Set.new %w[
|
||
|
|
ann annotation any constraint else endif function for forall if
|
||
|
|
include list of op output minimize maximize par predicate record
|
||
|
|
satisfy solve test then type var where
|
||
|
|
]
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.keywords_type
|
||
|
|
@keywords_type ||= Set.new %w(
|
||
|
|
array set bool enum float int string tuple
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.operators
|
||
|
|
@operators ||= Set.new %w(
|
||
|
|
in subset superset union diff symdiff intersect
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
id = /[$a-zA-Z_]\w*/
|
||
|
|
|
||
|
|
state :root do
|
||
|
|
rule %r(\s+)m, Text::Whitespace
|
||
|
|
rule %r(\\\n)m, Text::Whitespace
|
||
|
|
rule %r(%.*), Comment::Single
|
||
|
|
rule %r(/(\\\n)?[*](.|\n)*?[*](\\\n)?/)m, Comment::Multiline
|
||
|
|
rule %r/"(\\\\|\\"|[^"])*"/, Literal::String
|
||
|
|
|
||
|
|
rule %r(not|<->|->|<-|\\/|xor|/\\), Operator
|
||
|
|
rule %r(<|>|<=|>=|==|=|!=), Operator
|
||
|
|
rule %r(\+|-|\*|/|div|mod), Operator
|
||
|
|
rule %r(\\|\.\.|\+\+), Operator
|
||
|
|
rule %r([|()\[\]{},:;]), Punctuation
|
||
|
|
rule %r((true|false)\b), Keyword::Constant
|
||
|
|
rule %r(([+-]?)\d+(\.(?!\.)\d*)?([eE][-+]?\d+)?), Literal::Number
|
||
|
|
|
||
|
|
rule id do |m|
|
||
|
|
if self.class.keywords.include? m[0]
|
||
|
|
token Keyword
|
||
|
|
elsif self.class.keywords_type.include? m[0]
|
||
|
|
token Keyword::Type
|
||
|
|
elsif self.class.builtins.include? m[0]
|
||
|
|
token Name::Builtin
|
||
|
|
elsif self.class.operators.include? m[0]
|
||
|
|
token Operator
|
||
|
|
else
|
||
|
|
token Name::Other
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
rule %r(::\s*([^\W\d]\w*)(\s*\([^\)]*\))?), Name::Decorator
|
||
|
|
rule %r(\b([^\W\d]\w*)\b(\()) do
|
||
|
|
groups Name::Function, Punctuation
|
||
|
|
end
|
||
|
|
rule %r([^\W\d]\w*), Name::Other
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|