%gJ2dZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZd d lmZej,d k(rd nd ZdefdZdedefdZdeddfdZdede efdZdeddfdZdeddfdZdeddfdZdefdZ de e e!dffdZ" d.de!de!d e e eefde#fd!Z$ d/de ede ede edffd"Z% d0d#d$de ed%e ed&e#d'e#d(e#d)e#d*e!d+e#de#fd,Z&ejNde ee!e!e!gdffd-Z(y)1z. Utilities for finding and installing CmdStan N) OrderedDict)CallableDictOptionalTupleUnion)tqdm) _DOT_CMDSTAN)progress) get_loggerWindowsz.exereturnctj}d}|dk(rd}|S|dk(r?tjgddtjd}d|j vrd }|Sd }|S|d k(rd }|S|d k(s|dk(rd }|S|dk(rd}|S)Nraarch64arm64armv7l)readelfz-Az/proc/self/exeT)checkstdouttextTag_ABI_VFP_argsarmelarmhfmips64mips64elppc64elppc64les390x)platformmachine subprocessrunPIPEr)r#archrs T/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/cmdstanpy/utils/cmdstan.pydetermine_linux_archr)s G D)( K' H .. /??    /D KD K H  K I I!5 K G  Kversionctjjdd}|s!tjdk(r t }|r |j dk7r d|d|d|d}n d|d|d}d |S) N CMDSTAN_ARCHrLinuxfalsevz /cmdstan-z-linux-z.tar.gzz6https://github.com/stan-dev/cmdstan/releases/download/)osenvirongetr"systemr)lower)r+r'url_ends r(get_download_urlr70sz ::>>." -D HOO%0#%  'gYiyvWEgYiy8 CG9 MMr* install_dirctjj|s tj|ytjj|stdj| tdd5 dddtjdy#tt t f$r }tdj||d}~wwxYw#1swYVxYw#t $r }tdj||d}~wwxYw)z9Check that specified install directory exists, can write.zCannot create directory: {}Nz&File exists, should be a directory: {} tmp_test_wwz"Cannot write files to directory {}) r1pathexistsmakedirsIOErrorOSErrorPermissionError ValueErrorformatisdiropenremove)r8es r( validate_dirrH=s 77>>+ &  KK $ ww}}[)8?? L  lC(   IIl #/2 -44[A     4;;KH  sGB1 C =C?C C1C  CCC D )DD  cmdstan_dirc tj|Dcgc]U}tjjtjj ||r|j dr|ddW}}t |dk(ryt |dk(rd|dzS|Dcgc]-}|djs|jddk(s,|/}}tt |D]k}d||vs ||jd}|djd}|d}tt|d z }dj |d|d|g||<m|jd |t |dz } | jd}t|ddkr3tt|dd z}d |z|d<dj |} d| zScc}wcc}w) a Given a valid directory path, find all installed CmdStan versions and return highest (i.e., latest) version number. Assumes directory consists of CmdStan releases, created by function `install_cmdstan`, and therefore dirnames have format "cmdstan-.." or "cmdstan-..-rc", which is CmdStan release practice as of v 2.24. cmdstan-Nrr .r z-rcdcRttt|jdS)NrM)listmapintsplit)ss r(z$get_latest_cmdstan..tsSaggcl%; <r*)keyz0-rc)r1listdirr<rDjoin startswithlenisdigitcountrangerSstrrRsort) rInameversionsr0icompsmmprc_numpatchlatests r(get_latest_cmdstanrhUsJJ{+  77==k48 9 OOJ ' QRH  8} 8}HQK''#Laqt||~!''#,!:KLHL 3x= !< HQK QK%%e,E(..%C1XFF c)*E((CFCFE#:;HQK < MM>"'',,tUGi4GH I8?9 9   Jr*c@t||tjd<y)z4 Validate, then set CmdStan directory path. CMDSTANN)rnr1r2rms r(set_cmdstan_pathrqs$ BJJyr*makec*|tjd<y)z* set MAKE environmental variable. MAKEN)r1r2)rrs r( set_make_envrusBJJvr*cPd}dtjvr3ttjddkDrtjd}ntjj tjj dt }tjj|s tdt|}| tdtjj ||}|tjd<t|tjj|S)z7 Validate, then return CmdStan directory path. rrpr~zcNo CmdStan installation found, run command "install_cmdstan"or (re)activate your conda environment!) r1r2rZr< expanduserrXr r=rBrhrnnormpath)cmdstanrIlatest_cmdstans r( cmdstan_pathr|sGBJJ3rzz)'<#=#A**Y'gg((c<)HI ww~~k*: ,K8  !: '',,{N; ' 9'" 77  G $$r*.c tjjtd}tjj|s#t j dtyt|d5}|j}dddjd}|dkrt j d |y|tdz }|jd |}|||}|jd }t|d k7rt j d |ytd|ddDS#t$r=}t j dt jd|Yd}~yd}~wwxYw#1swYxYw)ai Parses version string out of CmdStan makefile variable CMDSTAN_VERSION, returns Tuple(Major, minor). If CmdStan installation is not found or cannot parse version from makefile logs warning and returns None. Lenient behavoir required for CI tests, per comment: https://github.com/stan-dev/cmdstanpy/pull/321#issuecomment-733817554 makefilezNo CmdStan installation found.z%sNz=CmdStan installation %s missing makefile, cannot get version.rzCMDSTAN_VERSION := rz'Cannot parse version from makefile: %s. rMzFCannot parse version, expected "..", found: "%s".c32K|]}t|ywN)rR).0xs r( z"cmdstan_version..s-AQ-sr )r1r<rXr|rBrinfodebugr=rEreadfindrZrStuple)r~rGfdcontents start_idxend_idxr+splitss r(cmdstan_versionrs^77<<  ; 77>>( #  K N  h 779 34I1}  5   *++ImmD),Gy)G ]]3 F 6{a     -!- --G  :; 4#s#(D68E?6 E<?3E77E<?Fmajorminorrcd}|d|vr t}nt|dt|df}|tjd||y|d|ks|d|k(r |d|kryy) z Check that CmdStan version is less than Major.minor version. :param major: Major version number :param minor: Minor version number :return: True if version at or above major.minor, else False. Nstan_version_majorstan_version_minorz1Cannot determine whether version is before %d.%d.Frr T)rrRrr)rrr cur_versions r(cmdstan_version_beforersK |+47%'  )* + )* +   ? 1~A%KNU$: r*c  tjdk7r td|t|ts t dt }dtjvrtjd}tjjtjj|drtjj|tjdkDrdndd}tjj|r[tjj|d d}tjj|s1d }d }|jd |d }nd }|jd |d }ntjjtjj|d rtjj|tjdkDrd nd d}tjj|rZtjj|d}tjj|s,d }d }|jd|d }nd }|jd|d }ntjjd}tjj!tjjdt"}||gng|~tjj|dtjj|dtjj|dtjj|dgngztjj|dtjjtjj%ddtjj|dtjjtjj%ddtjj|dtjjtjj%ddtjjtjj%ddgzD]}d }d }tjj|r|dvrtjj|tjdkDrdndd}tjj|r[tjj|d d}tjj|sd }d }|jd|d }nd }|jd|d }tjj|tjdkDrd nd d}tjj|r[tjj|d}tjj|sd }d }|jd|d }nd }|jd|d }d }s t'd|j)d|djt+t-j.gtj0dd j3dztjd<||fS)z? Validate, then activate C++ toolchain directory path. rz4Functionality is currently only supported on Windowsz!Format version number as a stringCMDSTAN_TOOLCHAINmingw64lmingw32rjusrrz+Found invalid installion for RTools40 on %smingw_64mingw_32z+Found invalid installion for RTools35 on %s RTOOLS40_HOMErwRTools40RTools35RTools30RTools/ RBuildTools)35z3.53z-Found invalid installation for RTools40 on %sz-Found invalid installation for RTools35 on %szkno RTools toolchain installation found, run command line script "python -m cmdstanpy.install_cxx_toolchain"zAdd C++ toolchain to $PATH: %s;PATH)r"r4 RuntimeError isinstancer^ TypeErrorrr1r2r<r=rXsysmaxsizewarningr3rxr abspathrBrrPrfromkeysgetenvrS)r+r8loggertoolchain_root compiler_path tool_path rtools40_homerIs r(cxx_toolchain_pathr sE I% B  :gs#;;<< \Fbjj($78 77>>"'',,~yA BGGLL!kkE1 M ww~~m,GGLLF ww~~i0 "I$&MNNE&&(N " A""$ WW^^BGGLLD EGGLL"{{U2 M ww~~m,GGLL? ww~~i0 "I$&MNNE&&(N " A""$ 7 gg((c<)HI - 9m_r* GGLLj9GGLLj9GGLLj9GGLLh7   [*5 RWW__S1:> [*5 RWW__S1:> [(3 RWW__S18< RWW__S1=A M $N,MIww~~n-"44$&GGLL&&)kkE&9 %M ww~~m4$&GGLL$N !ww~~i8(*I,.M"NN O ..0N!(* K**,$&GGLL&'*{{U': %M ww~~m4$&GGLL$G !ww~~i8(*I,.M"NN O ..0N!(* K**,!#[M $^  :   KK0.A   *RYYvr-B-H-H-MM  BJJv ) ##r*F) interactivedir overwritecompilerr verbosecoresrc t} ddlm} m} m} |r.t |||||||dk7gr|j d| } n| |||||||} | | d | jvr0d | jjd d jd d}nd | j}ttjj| j|y#t$r%} |j dt| Yd} ~ yd} ~ wwxYw)au Download and install a CmdStan release from GitHub. Downloads the release tar.gz file to temporary storage. Retries GitHub requests in order to allow for transient network outages. Builds CmdStan executables and tests the compiler by building example model ``bernoulli.stan``. :param version: CmdStan version string, e.g. "2.29.2". Defaults to latest CmdStan release. If ``git`` is installed, a git tag or branch of stan-dev/cmdstan can be specified, e.g. "git:develop". :param dir: Path to install directory. Defaults to hidden directory ``$HOME/.cmdstan``. If no directory is specified and the above directory does not exist, directory ``$HOME/.cmdstan`` will be created and populated. :param overwrite: Boolean value; when ``True``, will overwrite and rebuild an existing CmdStan installation. Default is ``False``. :param compiler: Boolean value; when ``True`` on WINDOWS ONLY, use the C++ compiler from the ``install_cxx_toolchain`` command or install one if none is found. :param progress: Boolean value; when ``True``, show a progress bar for downloading and unpacking CmdStan. Default is ``False``. :param verbose: Boolean value; when ``True``, show console output from all intallation steps, i.e., download, build, and test CmdStan release. Default is ``False``. :param cores: Integer, number of cores to use in the ``make`` command. Default is 1 core. :param interactive: Boolean value; if true, ignore all other arguments to this function and run in an interactive mode, prompting the user to provide the other information manually through the standard input. This flag should only be used in interactive environments, e.g. on the command line. :return: Boolean value; ``True`` for success. r )InstallationSettingsInteractiveSettings run_installr z`Interactive installation requested but other arguments were used. These values will be ignored!)r+rrrr rrzCmdStan installation failed. %sNFzgit:rK:-r_T)rinstall_cmdstanrrranyr Exceptionr^r+replacerqr1r<rXr)r+rrrr rrrrrrrargsrGfolders r(rrsh\F(   QJ C'(D'#!!D D DLL00c:BB3LMNDLL>*RWW\\$((F34  93q6BsAC D 'DD c\tddddddtdtd td d ffd }|S) z(Sets up tqdm callback for url downloads.BTiblueF)unit unit_scale unit_divisorcolourleaver\ block_size total_sizerNcj|_j||z}j|jz j|k\rj yyr)totalresetupdatenclose)r\rrdownloaded_sizepbars r(download_progress_hookz6wrap_url_progress_hook..download_progress_hooksX :: #DJ JJL*, Odff,- 66Z  JJL r*)r rR)rrs @r(wrap_url_progress_hookrsK  D   # 14    "!r*r)NN)NNFFFFr ))__doc__r1r"r$r collectionsrtypingrrrrr tqdm.autor cmdstanpyr rr progbarloggingrr4rlr^r)r7rHrhrnrqrur|rRrboolrrr wrap_callbackrr*r(rs(  #99""%hoo'94F" c6 Nc Nc Ncd0)C)HSM)X      !3!4!st%c%4/.%S/2/.f>B "*4S>": @AE^$ c]^$08 ^$ 38_^$D"ee c]e #ee e  e  e ee eP "3S/42G)H I""r*