Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
Stan: Added missing keywords and HOFs (#3238)
- Loading branch information
1 parent
642d93e
commit afd77ed
Showing
5 changed files
with
179 additions
and
47 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,49 +1,65 @@ | ||
// https://mc-stan.org/docs/2_24/reference-manual/bnf-grammars.html | ||
(function (Prism) { | ||
|
||
Prism.languages.stan = { | ||
'comment': /\/\/.*|\/\*[\s\S]*?\*\/|#(?!include).*/, | ||
'string': { | ||
// String literals can contain spaces and any printable ASCII characters except for " and \ | ||
// https://mc-stan.org/docs/2_24/reference-manual/print-statements-section.html#string-literals | ||
pattern: /"[\x20\x21\x23-\x5B\x5D-\x7E]*"/, | ||
greedy: true | ||
}, | ||
'directive': { | ||
pattern: /^([ \t]*)#include\b.*/m, | ||
lookbehind: true, | ||
alias: 'property' | ||
}, | ||
// https://mc-stan.org/docs/2_28/reference-manual/bnf-grammars.html | ||
|
||
'function-arg': { | ||
pattern: /(\b(?:algebra_solver|integrate_1d|integrate_ode|integrate_ode_bdf|integrate_ode_rk45|map_rect)\s*\(\s*)[a-zA-Z]\w*/, | ||
lookbehind: true, | ||
alias: 'function' | ||
}, | ||
'constraint': { | ||
pattern: /(\b(?:int|matrix|real|row_vector|vector)\s*)<[^<>]*>/, | ||
lookbehind: true, | ||
inside: { | ||
'expression': { | ||
pattern: /(=\s*)\S(?:\S|\s+(?!\s))*?(?=\s*(?:>$|,\s*\w+\s*=))/, | ||
lookbehind: true, | ||
inside: null // see below | ||
var higherOrderFunctions = /\b(?:algebra_solver|algebra_solver_newton|integrate_1d|integrate_ode|integrate_ode_bdf|integrate_ode_rk45|map_rect|ode_(?:adams|bdf|ckrk|rk45)(?:_tol)?|ode_adjoint_tol_ctl|reduce_sum|reduce_sum_static)\b/; | ||
|
||
Prism.languages.stan = { | ||
'comment': /\/\/.*|\/\*[\s\S]*?\*\/|#(?!include).*/, | ||
'string': { | ||
// String literals can contain spaces and any printable ASCII characters except for " and \ | ||
// https://mc-stan.org/docs/2_24/reference-manual/print-statements-section.html#string-literals | ||
pattern: /"[\x20\x21\x23-\x5B\x5D-\x7E]*"/, | ||
greedy: true | ||
}, | ||
'directive': { | ||
pattern: /^([ \t]*)#include\b.*/m, | ||
lookbehind: true, | ||
alias: 'property' | ||
}, | ||
|
||
'function-arg': { | ||
pattern: RegExp( | ||
'(' + | ||
higherOrderFunctions.source + | ||
/\s*\(\s*/.source + | ||
')' + | ||
/[a-zA-Z]\w*/.source | ||
), | ||
lookbehind: true, | ||
alias: 'function' | ||
}, | ||
'constraint': { | ||
pattern: /(\b(?:int|matrix|real|row_vector|vector)\s*)<[^<>]*>/, | ||
lookbehind: true, | ||
inside: { | ||
'expression': { | ||
pattern: /(=\s*)\S(?:\S|\s+(?!\s))*?(?=\s*(?:>$|,\s*\w+\s*=))/, | ||
lookbehind: true, | ||
inside: null // see below | ||
}, | ||
'property': /\b[a-z]\w*(?=\s*=)/i, | ||
'operator': /=/, | ||
'punctuation': /^<|>$|,/ | ||
} | ||
}, | ||
'keyword': [ | ||
{ | ||
pattern: /\bdata(?=\s*\{)|\b(?:functions|generated|model|parameters|quantities|transformed)\b/, | ||
alias: 'program-block' | ||
}, | ||
'property': /\b[a-z]\w*(?=\s*=)/i, | ||
'operator': /=/, | ||
'punctuation': /^<|>$|,/ | ||
} | ||
}, | ||
'keyword': [ | ||
/\b(?:break|cholesky_factor_corr|cholesky_factor_cov|continue|corr_matrix|cov_matrix|data|else|for|functions|generated|if|in|increment_log_prob|int|matrix|model|ordered|parameters|positive_ordered|print|quantities|real|reject|return|row_vector|simplex|target|transformed|unit_vector|vector|void|while)\b/, | ||
// these are functions that are known to take another function as their first argument. | ||
/\b(?:algebra_solver|integrate_1d|integrate_ode|integrate_ode_bdf|integrate_ode_rk45|map_rect)\b/ | ||
], | ||
'function': /\b[a-z]\w*(?=\s*\()/i, | ||
'number': /(?:\b\d+(?:\.\d*)?|\B\.\d+)(?:E[+-]?\d+)?\b/i, | ||
'boolean': /\b(?:false|true)\b/, | ||
/\b(?:array|break|cholesky_factor_corr|cholesky_factor_cov|complex|continue|corr_matrix|cov_matrix|data|else|for|if|in|increment_log_prob|int|matrix|ordered|positive_ordered|print|real|reject|return|row_vector|simplex|target|unit_vector|vector|void|while)\b/, | ||
// these are functions that are known to take another function as their first argument. | ||
higherOrderFunctions | ||
], | ||
'function': /\b[a-z]\w*(?=\s*\()/i, | ||
'number': /(?:\b\d+(?:_\d+)*(?:\.(?:\d+(?:_\d+)*)?)?|\B\.\d+(?:_\d+)*)(?:E[+-]?\d+(?:_\d+)*)?i?(?!\w)/i, | ||
'boolean': /\b(?:false|true)\b/, | ||
|
||
'operator': /<-|\.[*/]=?|\|\|?|&&|[!=<>+\-*/]=?|['^%~?:]/, | ||
'punctuation': /[()\[\]{},;]/ | ||
}; | ||
|
||
'operator': /<-|\.[*/]=?|\|\|?|&&|[!=<>+\-*/]=?|['^%~?:]/, | ||
'punctuation': /[()\[\]{},;]/ | ||
}; | ||
Prism.languages.stan.constraint.inside.expression.inside = Prism.languages.stan; | ||
|
||
Prism.languages.stan.constraint.inside.expression.inside = Prism.languages.stan; | ||
}(Prism)); |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,76 @@ | ||
functions { | ||
// ... function declarations and definitions ... | ||
} | ||
data { | ||
// ... declarations ... | ||
} | ||
transformed data { | ||
// ... declarations ... statements ... | ||
} | ||
parameters { | ||
// ... declarations ... | ||
} | ||
transformed parameters { | ||
// ... declarations ... statements ... | ||
} | ||
model { | ||
// ... declarations ... statements ... | ||
} | ||
generated quantities { | ||
// ... declarations ... statements ... | ||
} | ||
|
||
// data-only quantifiers | ||
real foo(data real x) { | ||
return x^2; | ||
} | ||
|
||
---------------------------------------------------- | ||
|
||
<span class="token keyword program-block">functions</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... function declarations and definitions ...</span> | ||
<span class="token punctuation">}</span> | ||
<span class="token keyword program-block">data</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... declarations ...</span> | ||
<span class="token punctuation">}</span> | ||
<span class="token keyword program-block">transformed</span> | ||
<span class="token keyword program-block">data</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... declarations ... statements ...</span> | ||
<span class="token punctuation">}</span> | ||
<span class="token keyword program-block">parameters</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... declarations ...</span> | ||
<span class="token punctuation">}</span> | ||
<span class="token keyword program-block">transformed</span> | ||
<span class="token keyword program-block">parameters</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... declarations ... statements ...</span> | ||
<span class="token punctuation">}</span> | ||
<span class="token keyword program-block">model</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... declarations ... statements ...</span> | ||
<span class="token punctuation">}</span> | ||
<span class="token keyword program-block">generated</span> | ||
<span class="token keyword program-block">quantities</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token comment">// ... declarations ... statements ...</span> | ||
<span class="token punctuation">}</span> | ||
|
||
<span class="token comment">// data-only quantifiers</span> | ||
<span class="token keyword">real</span> | ||
<span class="token function">foo</span> | ||
<span class="token punctuation">(</span> | ||
<span class="token keyword">data</span> | ||
<span class="token keyword">real</span> | ||
x | ||
<span class="token punctuation">)</span> | ||
<span class="token punctuation">{</span> | ||
<span class="token keyword">return</span> | ||
x | ||
<span class="token operator">^</span> | ||
<span class="token number">2</span> | ||
<span class="token punctuation">;</span> | ||
<span class="token punctuation">}</span> |