o
    -wi!2                     @   s   d Z ddlZddlmZmZmZmZmZ ddlm	Z	m
Z
mZmZmZmZmZmZ ddlmZ ddlmZ g dZG dd	 d	eZG d
d deZG dd deZG dd deZdS )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 Zd ZdZdZdgZdgZdgZej	ej
B ZdZdefdejfd	ejfged
dejdfdefdefdejfdejfdejfdejfdejfdejdfdejfdejdfeefged
dejfdejdfeejdfe dged
eej!dfe dgdejdfdej"fde#e$fdejfgdZ%d S )!r   z\
    For `Modelica <http://www.modelica.org/>`_ source code.

    .. versionadded:: 1.1
    Modelicamodelicaz*.moztext/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)\bzwithin\bpackage-prefixz(function|record)\bz(if|for|when|while)\bz#popz\\[\'"?\\abfnrtv]z6(?i)<\s*html\s*>([^\\"]|\\.)+?(<\s*/\s*html\s*>|(?="))z<|\\?[^"\\<]+)r   rootr   r   r   N)&__name__
__module____qualname____doc__namealiases	filenames	mimetypesreDOTALL	MULTILINEflags_namer   r   Single	Multiliner   r   Doubler   r	   r   FloatIntegerr   Builtinr
   ReservedWordConstantClassr   	NamespaceEscaper   r   tokens r7   r7   b/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/vendor/pygments/lexers/modeling.pyr      sd    


!




r   c                   @   s   e Zd ZdZdZg dZdgZdZ	 dZ	 de	fgde
jfged	ed
deeje	efdejfddee  ejfdefdefdefdefdefdefgdZdd ZdS )r   zv
    Pygments Lexer for `OpenBugs <http://www.openbugs.net/>`_ and WinBugs
    models.

    .. versionadded:: 1.6
    BUGS)bugswinbugsopenbugs*.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phizpost.p.valuepowzprior.p.valueprobitzreplicate.postzreplicate.priorroundsinsinhsolutionsqrtsteptantanhtruncinprod
interp.lininverselogdetmeanz
eigen.valsodeprodzp.valueMrankrankedz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dwish\s+#.*$commentsr   z(model)(\s+)(\{)(for|in)(?![\w.])(%s)(?=\s*\()|z[A-Za-z][\w.]*&[-+]?[0-9]*\.?[0-9]+([eE][-+]?[0-9]+)?\[|\]|\(|\)|:|,|;<-|~z	\+|-|\*|/[{}]r   r   r   c                 C      t d| t jr
dS dS )Nz^\s*model\s*{gffffff?        r%   searchMtextr7   r7   r8   analyse_text      zBugsLexer.analyse_textN)r   r   r   r    r!   r"   r#   
_FUNCTIONS_DISTRIBUTIONSr   r   r*   r   r   r
   r4   r   r0   joinr   r/   r   r	   r6   r   r7   r7   r7   r8   r   c   sB    %r   c                   @   s   e Zd ZdZdZdgZddgZdZedd d	D Z	d
Z
defgdefgdejfdejfgededdeejeefdejfdejfddee	 e
  ejfeddefdefdefdefdefgdZdd ZdS ) r   z=
    Pygments Lexer for JAGS.

    .. versionadded:: 1.6
    JAGSjagsz*.jagr=   )2r>   r?   r@   rA   rB   rC   rD   rF   rG   rE   rL   exprP   ifelserO   rR   rS   rT   rU   rX   rY   rZ   r[   r\   r]   r_   r`   ra   rb   rc   rd   re   rg   rV   rh   rW   rj   ro   rm   rf   rk   rn   tacosacoshasinasinhatanrs   rq   c                 c   s    | ]}d | V  qdS )z[dpq]%sNr7   ).0xr7   r7   r8   	<genexpr>   s    zJagsLexer.<genexpr>)bernbetadchiqsqrr}   r~   r   gammaz	gen.gammalogislnormnegbinnchisqrnormparpoisweib)r   r   dbetabinru   rv   rw   rz   r   r   r   r   r   dbinomdchisqdnbinomdweibullddirichr   z[a-zA-Z][\w.]*\b(?s)/\*.*?\*/r   r   r   z(model|data)(\s+)(\{)zvar(?![\w.])r   r   r   namesr   r   r   z&\+|-|\*|\/|\|\|[&]{2}|[<>=]=?|\^|%.*?%r   )r   r   r   r   c                 C   s@   t d| t jrt d| t jrdS t d| t jrdS dS dS )Nz^\s*model\s*\{z^\s*data\s*\{g?z^\s*varg333333?r   r   r   r7   r7   r8   r     s   zJagsLexer.analyse_textN)r   r   r   r    r!   r"   r#   r   tupler   _OTHER_DISTRIBUTIONSr   r   r   r+   r*   r   r   r
   r4   r   Declarationr0   r   r/   r   r	   r6   r   r7   r7   r7   r8   r      sP    +r   c                   @   s  e Zd ZdZdZdgZdgZdefgdej	fdej
fgdefed	ed
ddd eejeefddej efdefddej ejfdefddejej  ejfdejjfddej ejfdejfdefdejfdejfdej fde!fde!fdefgd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
    Stanstanz*.stanr   r   z	(//|#).*$z"[^"]*"r   r   z(%s)(\s*)(\{)r   )	functionsdataztransformed\s+?data
parametersztransformed\s+parametersmodelzgenerated\s+quantitiesz(%s)\bz
T(?=\s*\[)z[;:,\[\]()]r   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]+r   z/\+|-|\.?\*|\.?/|\\|'|\^|==?|!=?|<=?|>=?|\|\||&&r   r   c                 C   r   )Nz^\s*parameters\s*\{g      ?r   r   r   r7   r7   r8   r   b  r   zStanLexer.analyse_textN)$r   r   r   r    r!   r"   r#   r   r   r+   r*   r   r   r   r   r
   r4   r   r   KEYWORDSTYPESType	FUNCTIONSDISTRIBUTIONSr   r/   PseudoRESERVEDr0   Functionr   r-   r.   r	   r6   r   r7   r7   r7   r8   r     sT    	
9r   )r    r%   pygments.lexerr   r   r   r   r   pygments.tokenr   r   r	   r
   r   r   r   r   pygments.lexers.htmlr   pygments.lexersr   __all__r   r   r   r   r7   r7   r7   r8   <module>   s   
(K`X