3
� �X!2 � @ s� d Z ddlZddlmZmZmZmZmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZ ddlmZ ddlmZ ddd d
gZG d d� de�ZG d d� de�ZG d
d � d e�ZG dd
� d
e�ZdS )z�
pygments.lexers.modeling
~~~~~~~~~~~~~~~~~~~~~~~~
Lexers for modeling languages.
:copyright: Copyright 2006-2017 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
� N)�
RegexLexer�include�bygroups�using�default)�Text�Comment�Operator�Keyword�Name�String�Number� Punctuation)� HtmlLexer)�_stan_builtins�
ModelicaLexer� BugsLexer� JagsLexer� StanLexerc @ s$ e Zd ZdZdZdgZdgZdgZej ej
B Z dZ de
fdejfd ejfged
�d ejd fd
efdefdejfdejfdejfdej fdej fdej dfdej fdej dfe efg
ed
�dej fdej dfe ej dfe d�ged
�e ej!dfe d�gd ejdfd ej"fd e#e$�fd ejfgd �Z%d S )!r z\
For `Modelica <http://www.modelica.org/>`_ source code.
.. versionadded:: 1.1
ZModelicaZmodelicaz*.moztext/x-modelicaz"(?:'(?:[^\\']|\\.)+'|[a-zA-Z_]\w*)u [\s]+z //[^\n]*\n?z /\*.*?\*/�
whitespace�"�stringz
[()\[\]{},;]+z\.?[*^/+-]|\.|<>|[<>:=]=?z \d+(\.?\d*[eE][-+]?\d+|\.\d*)z\d+al (abs|acos|actualStream|array|asin|assert|AssertionLevel|atan|atan2|backSample|Boolean|cardinality|cat|ceil|change|Clock|Connections|cos|cosh|cross|delay|diagonal|div|edge|exp|ExternalObject|fill|floor|getInstanceName|hold|homotopy|identity|inStream|integer|Integer|interval|inverse|isPresent|linspace|log|log10|matrix|max|min|mod|ndims|noClock|noEvent|ones|outerProduct|pre|previous|product|Real|reinit|rem|rooted|sample|scalar|semiLinear|shiftSample|sign|sin|sinh|size|skew|smooth|spatialDistribution|sqrt|StateSelect|String|subSample|sum|superSample|symmetric|tan|tanh|terminal|terminate|time|transpose|vector|zeros)\baX (algorithm|annotation|break|connect|constant|constrainedby|der|discrete|each|else|elseif|elsewhen|encapsulated|enumeration|equation|exit|expandable|extends|external|final|flow|for|if|import|impure|in|initial|inner|input|loop|nondiscrete|outer|output|parameter|partial|protected|public|pure|redeclare|replaceable|return|stream|then|when|while)\bz(and|not|or)\bzI(block|class|connector|end|function|model|operator|package|record|type)\b�classz(false|true)\bzwithin\bzpackage-prefixz(function|record)\bz(if|for|when|while)\bz#popz\\[\'"?\\abfnrtv]z6(?i)<\s*html\s*>([^\\"]|\\.)+?(<\s*/\s*html\s*>|(?="))z
<|\\?[^"\\<]+)r �rootr zpackage-prefixr N)&�__name__�
__module__� __qualname__�__doc__�name�aliases� filenamesZ mimetypes�re�DOTALL� MULTILINE�flags�_namer r �Single� Multiliner r �Doubler r r
�Float�Integerr �Builtinr
�ReservedZWordZConstantZClassr � NamespaceZEscaper r �tokens� r/ r/ � /usr/lib/python3.6/modeling.pyr sN
c > @ s� e Zd ZdZdZdddgZdgZdtZduZdbe fgdce
j fge dd�e de�dfe
eje e�fdgejfdhdijee � ejfdjefdkefdlefdmefdnefdoefg dp�Zdqdr� ZdsS )vr zv
Pygments Lexer for `OpenBugs <http://www.openbugs.net/>`_ and WinBugs
models.
.. versionadded:: 1.6
ZBUGSZbugsZwinbugsZopenbugsz*.bug�abs�arccos�arccosh�arcsin�arcsinh�arctan�arctanh�cloglog�cos�cosh�
cumulative�cut�density�deviance�equals�expr�gammap�ilogit�icloglog�integral�log�logfact�loggam�logit�max�min�phi� post.p.value�pow�
prior.p.value�probit�replicate.post�replicate.prior�round�sin�sinh�solution�sqrt�step�tan�tanh�trunc�inprod�
interp.lin�inverse�logdet�mean�
eigen.vals�ode�prod�p.valueM�rank�ranked�replicate.postM�sd�sort�sum�D�I�F�T�C�dbern�dbin�dcat�dnegbin�dpois�dhyper�dbeta�dchisqr�ddexp�dexp�dflat�dgamma�dgev�df�dggamma�dgpar�dloglik�dlnorm�dlogis�dnorm�dpar�dt�dunif�dweib�dmulti�ddirch�dmnorm�dmt�dwishz\s+z#.*$�commentsr z(model)(\s+)(\{)z(for|in)(?![\w.])z
(%s)(?=\s*\()�|z[A-Za-z][\w.]*z&[-+]?[0-9]*\.?[0-9]+([eE][-+]?[0-9]+)?z\[|\]|\(|\)|:|,|;z<-|~z \+|-|\*|/z[{}])r r� r c C s t jd| t j�rdS dS d S )Nz
^\s*model\s*{gffffff�?g )r! �search�M)�textr/ r/ r0 � analyse_text� s zBugsLexer.analyse_textN)>r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ rK rL rM rN rO rP rQ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ) ro rp rq rr rs rt ru rv rw rx ry rz r{ r| r} r~ r r� r� r� r� r� r� r� r� r� r� r� r� )r r r r r r r �
_FUNCTIONS�_DISTRIBUTIONSr r r& r r r
r- r r, �joinr r+ r
r r. r� r/ r/ r/ r0 r c sL
c 2 @ s� e Zd ZdZdZdgZddgZdpZed8d9� dqD ��Z drZ
d[e fgd\e fgd]e
jfd^e
jfged_�ed`�daeeje e�fdbejfdcejfdddejee e
� e jfedf�dgefdhefdiefdjefdkefg dl�Zdmdn� Z doS )sr z=
Pygments Lexer for JAGS.
.. versionadded:: 1.6
ZJAGSZjagsz*.jagz*.bugr1 r2 r3 r4 r5 r6 r7 r9 r: r8 r? �exprC �ifelserB rE rF rG rH rK rM rO rR rS rT rV rW rX rY rZ r[ �
interp.linr^ rI r_ rJ rb ri rg r] rd rh �t�acos�acosh�asin�asinh�atanrm rk c c s | ]}d | V qdS )z[dpq]%sNr/ )�.0�xr/ r/ r0 � <genexpr>� s zJagsLexer.<genexpr>�bern�beta�dchiqsqrrw rx r| �gamma� gen.gamma�logis�lnorm�negbin�nchisqr�norm�par�pois�weibr� r� �dbetabinro rp rq rt r� r� r� r� r� �dbinom�dchisq�dnbinom�dweibull�ddirichz\s+z[a-zA-Z][\w.]*\bz
(?s)/\*.*?\*/z#.*$r� r z(model|data)(\s+)(\{)z var(?![\w.])z(for|in)(?![\w.])z
(%s)(?=\s*\()r� �namesz&[-+]?[0-9]*\.?[0-9]+([eE][-+]?[0-9]+)?z\[|\]|\(|\)|:|,|;z<-|~z&\+|-|\*|\/|\|\|[&]{2}|[<>=]=?|\^|%.*?%z[{}])r r� r� r c C sF t jd| t j�r>t jd| t j�r$dS t jd| t j�r8dS dS ndS d S )Nz^\s*model\s*\{z
^\s*data\s*\{g�������?z^\s*varg333333�?r )r! r� r� )r� r/ r/ r0 r� s zJagsLexer.analyse_textN)2r1 r2 r3 r4 r5 r6 r7 r9 r: r8 r? r� rC r� rB rE rF rG rH rK rM rO rR rS rT rV rW rX rY rZ r[ r� r^ rI r_ rJ rb ri rg r] rd rh r� r� r� r� r� r� rm rk )r� r� r� rw rx r| r� r� r� r� r� r� r� r� r� r� )r� r� r� ro rp rq rt r� r� r� r� r� r� r� r� r� r� ) r r r r r r r r� �tupler� Z_OTHER_DISTRIBUTIONSr r r r' r& r r r
r- r Z Declarationr, r� r+ r
r r. r� r/ r/ r/ r0 r � sN
c @ s e Zd ZdZdZdgZdgZdefgdej fdej
fgde fe d �e d
�d d j
d%� eejee�fdd j
ej� efdefdd j
ej� ejfdefdd j
ejej � ejfdejjfdd j
ej� ej fdej fdefde j fd e j fd e j fd e!fd e!fd efgd!�Z"d"d#� Z#d$S )&r a Pygments Lexer for Stan models.
The Stan modeling language is specified in the *Stan Modeling Language
User's Guide and Reference Manual, v2.8.0*,
`pdf <https://github.com/stan-dev/stan/releases/download/v2.8.8/stan-reference-2.8.0.pdf>`__.
.. versionadded:: 1.6
ZStanZstanz*.stanz\s+z
(?s)/\*.*?\*/z (//|#).*$z"[^"]*"r� r z
(%s)(\s*)(\{)r� � functions�data�transformed\s+?data�
parameters�transformed\s+parameters�model�generated\s+quantitiesz(%s)\bz
T(?=\s*\[)z [;:,\[\]()]z
(%s)(?=\s*\()z[A-Za-z]\w*__\bz[A-Za-z]\w*(?=\s*\()]z
[A-Za-z]\w*\bz -?[0-9]+(\.[0-9]+)?[eE]-?[0-9]+z-?[0-9]*\.[0-9]*z-?[0-9]+z<-|~z/\+|-|\.?\*|\.?/|\\|'|\^|==?|!=?|<=?|>=?|\|\||&&z[{}])r r� r c C s t jd| t j�rdS dS d S )Nz^\s*parameters\s*\{g �?g )r! r� r� )r� r/ r/ r0 r� b s zStanLexer.analyse_textN)r� r� r� r� r� r� r� )$r r r r r r r r r r' r& r r r� r r
r- r r ZKEYWORDSZTYPESZTypeZ FUNCTIONSZ
DISTRIBUTIONSr r+ ZPseudoZRESERVEDr, ZFunctionr
r) r* r r. r� r/ r/ r/ r0 r sD
)r r! Zpygments.lexerr r r r r Zpygments.tokenr r r r
r r r
r Zpygments.lexers.htmlr Zpygments.lexersr �__all__r r r r r/ r/ r/ r0 �<module>
s ( K`X