B
    ØÓY\†Ÿ  ã               @   s  d Z ddlZddlZddlZddlZddlZddlmZ ddl	m
Z
 ddlmZ e ¡ dkrpddl	mZ ejZnG d	d
„ d
ƒZeƒ ZeejjfZyddlmZ W n ek
r´   Y nX dd„ Zd dd„Zdd„ Zdd„ Zd!dd„ZG dd„ dƒZG dd„ dƒZG dd„ dƒZG dd„ dƒZ dS )"a@  
Improved support for Microsoft Visual C++ compilers.

Known supported compilers:
--------------------------
Microsoft Visual C++ 9.0:
    Microsoft Visual C++ Compiler for Python 2.7 (x86, amd64)
    Microsoft Windows SDK 6.1 (x86, x64, ia64)
    Microsoft Windows SDK 7.0 (x86, x64, ia64)

Microsoft Visual C++ 10.0:
    Microsoft Windows SDK 7.1 (x86, x64, ia64)

Microsoft Visual C++ 14.0:
    Microsoft Visual C++ Build Tools 2015 (x86, x64, arm)
    Microsoft Visual Studio 2017 (x86, x64, arm, arm64)
    Microsoft Visual Studio Build Tools 2017 (x86, x64, arm, arm64)
é    N)ÚLegacyVersion)Úfilterfalseé   )Úget_unpatchedÚWindows)Úwinregc               @   s   e Zd ZdZdZdZdZdS )r   N)Ú__name__Ú
__module__Ú__qualname__Ú
HKEY_USERSÚHKEY_CURRENT_USERÚHKEY_LOCAL_MACHINEÚHKEY_CLASSES_ROOT© r   r   ú.lib/python3.7/site-packages/setuptools/msvc.pyr   (   s   r   )ÚRegc             C   sž   d}|d| f }yt  |d¡}W nJ tk
rj   y|d| f }t  |d¡}W n tk
rd   d}Y nX Y nX |r’tjjj |d¡}tj |¡r’|S ttƒ| ƒS )a+  
    Patched "distutils.msvc9compiler.find_vcvarsall" to use the standalone
    compiler build for Python (VCForPython). Fall back to original behavior
    when the standalone compiler is not available.

    Redirect the path of "vcvarsall.bat".

    Known supported compilers
    -------------------------
    Microsoft Visual C++ 9.0:
        Microsoft Visual C++ Compiler for Python 2.7 (x86, amd64)

    Parameters
    ----------
    version: float
        Required Microsoft Visual C++ version.

    Return
    ------
    vcvarsall.bat path: str
    z-Software\%sMicrosoft\DevDiv\VCForPython\%0.1fÚ Ú
installdirzWow6432Node\Nzvcvarsall.bat)	r   Z	get_valueÚKeyErrorÚosÚpathÚjoinÚisfiler   Úmsvc9_find_vcvarsall)ÚversionZVC_BASEÚkeyÚ
productdirÚ	vcvarsallr   r   r   r   ?   s    r   Úx86c          
   O   sŽ   yt tƒ}|| |f|ž|ŽS  tjjk
r2   Y n tk
rD   Y nX yt|| ƒ ¡ S  tjjk
rˆ } zt|| |ƒ ‚ W dd}~X Y nX dS )aœ  
    Patched "distutils.msvc9compiler.query_vcvarsall" for support extra
    compilers.

    Set environment without use of "vcvarsall.bat".

    Known supported compilers
    -------------------------
    Microsoft Visual C++ 9.0:
        Microsoft Visual C++ Compiler for Python 2.7 (x86, amd64)
        Microsoft Windows SDK 6.1 (x86, x64, ia64)
        Microsoft Windows SDK 7.0 (x86, x64, ia64)

    Microsoft Visual C++ 10.0:
        Microsoft Windows SDK 7.1 (x86, x64, ia64)

    Parameters
    ----------
    ver: float
        Required Microsoft Visual C++ version.
    arch: str
        Target architecture.

    Return
    ------
    environment: dict
    N)	r   Úmsvc9_query_vcvarsallÚ	distutilsÚerrorsÚDistutilsPlatformErrorÚ
ValueErrorÚEnvironmentInfoÚ
return_envÚ_augment_exception)ÚverÚarchÚargsÚkwargsZorigÚexcr   r   r   r   j   s    r   c          
   C   sn   yt tƒ| ƒS  tjjk
r$   Y nX yt| dd ¡ S  tjjk
rh } zt|dƒ ‚ W dd}~X Y nX dS )a'  
    Patched "distutils._msvccompiler._get_vc_env" for support extra
    compilers.

    Set environment without use of "vcvarsall.bat".

    Known supported compilers
    -------------------------
    Microsoft Visual C++ 14.0:
        Microsoft Visual C++ Build Tools 2015 (x86, x64, arm)
        Microsoft Visual Studio 2017 (x86, x64, arm, arm64)
        Microsoft Visual Studio Build Tools 2017 (x86, x64, arm, arm64)

    Parameters
    ----------
    plat_spec: str
        Target architecture.

    Return
    ------
    environment: dict
    g      ,@)Ú
vc_min_verN)r   Úmsvc14_get_vc_envr    r!   r"   r$   r%   r&   )Z	plat_specr+   r   r   r   r-   ™   s    
r-   c              O   sB   dt jkr4ddl}t|jƒtdƒk r4|jjj| |ŽS tt	ƒ| |ŽS )z¦
    Patched "distutils._msvccompiler.gen_lib_options" for fix
    compatibility between "numpy.distutils" and "distutils._msvccompiler"
    (for Numpy < 1.11.2)
    znumpy.distutilsr   Nz1.11.2)
ÚsysÚmodulesZnumpyr   Ú__version__r    Z	ccompilerZgen_lib_optionsr   Úmsvc14_gen_lib_options)r)   r*   Znpr   r   r   r1   ¿   s
    
r1   r   c             C   s¬   | j d }d| ¡ ks"d| ¡ kr d}|jf tƒ Ž}d}|dkrr| ¡  d¡dkrh|d	7 }||d
 7 }q |d7 }n.|dkr|d7 }||d 7 }n|dkr |d7 }|f| _ dS )zl
    Add details to the exception message to help guide the user
    as to what action will resolve it.
    r   r   zvisual cz0Microsoft Visual C++ {version:0.1f} is required.z-www.microsoft.com/download/details.aspx?id=%dg      "@Zia64éÿÿÿÿz* Get it with "Microsoft Windows SDK 7.0": iB  z% Get it from http://aka.ms/vcpython27g      $@z* Get it with "Microsoft Windows SDK 7.1": iW   g      ,@z^ Get it with "Microsoft Visual C++ Build Tools": https://visualstudio.microsoft.com/downloads/N)r)   ÚlowerÚformatÚlocalsÚfind)r+   r   r(   ÚmessageZtmplZ
msdownloadr   r   r   r&   Ì   s     

r&   c               @   sb   e Zd ZdZe dd¡ ¡ Zdd„ Ze	dd„ ƒZ
dd	„ Zd
d„ Zddd„Zddd„Zddd„ZdS )ÚPlatformInfoz‚
    Current and Target Architectures informations.

    Parameters
    ----------
    arch: str
        Target architecture.
    Zprocessor_architecturer   c             C   s   |  ¡  dd¡| _d S )NÚx64Úamd64)r3   Úreplacer(   )Úselfr(   r   r   r   Ú__init__û   s    zPlatformInfo.__init__c             C   s   | j | j  d¡d d … S )NÚ_r   )r(   r6   )r<   r   r   r   Ú
target_cpuþ   s    zPlatformInfo.target_cpuc             C   s
   | j dkS )Nr   )r?   )r<   r   r   r   Útarget_is_x86  s    zPlatformInfo.target_is_x86c             C   s
   | j dkS )Nr   )Úcurrent_cpu)r<   r   r   r   Úcurrent_is_x86  s    zPlatformInfo.current_is_x86Fc             C   s.   | j dkr|rdS | j dkr$|r$dS d| j  S )uk  
        Current platform specific subfolder.

        Parameters
        ----------
        hidex86: bool
            return '' and not 'Â†' if architecture is x86.
        x64: bool
            return 'd' and not 'md64' if architecture is amd64.

        Return
        ------
        subfolder: str
            '	arget', or '' (see hidex86 parameter)
        r   r   r:   z\x64z\%s)rA   )r<   Úhidex86r9   r   r   r   Úcurrent_dir  s    zPlatformInfo.current_dirc             C   s.   | j dkr|rdS | j dkr$|r$dS d| j  S )ar  
        Target platform specific subfolder.

        Parameters
        ----------
        hidex86: bool
            return '' and not '\x86' if architecture is x86.
        x64: bool
            return '\x64' and not '\amd64' if architecture is amd64.

        Return
        ------
        subfolder: str
            '\current', or '' (see hidex86 parameter)
        r   r   r:   z\x64z\%s)r?   )r<   rC   r9   r   r   r   Ú
target_dir  s    zPlatformInfo.target_dirc             C   s0   |rdn| j }| j|krdS |  ¡  dd| ¡S )ao  
        Cross platform specific subfolder.

        Parameters
        ----------
        forcex86: bool
            Use 'x86' as current architecture even if current acritecture is
            not x86.

        Return
        ------
        subfolder: str
            '' if target architecture is current architecture,
            '\current_target' if not.
        r   r   ú\z\%s_)rA   r?   rE   r;   )r<   Úforcex86Zcurrentr   r   r   Ú	cross_dir4  s    zPlatformInfo.cross_dirN)FF)FF)F)r   r	   r
   Ú__doc__Úsafe_envÚgetr3   rA   r=   Úpropertyr?   r@   rB   rD   rE   rH   r   r   r   r   r8   ð   s   

r8   c               @   sª   e Zd ZdZejejejejfZ	dd„ Z
edd„ ƒZedd„ ƒZedd	„ ƒZed
d„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZddd„Zdd„ ZdS )ÚRegistryInfoz 
    Microsoft Visual Studio related registry informations.

    Parameters
    ----------
    platform_info: PlatformInfo
        "PlatformInfo" instance.
    c             C   s
   || _ d S )N)Úpi)r<   Zplatform_infor   r   r   r=   Y  s    zRegistryInfo.__init__c             C   s   dS )z<
        Microsoft Visual Studio root registry key.
        ZVisualStudior   )r<   r   r   r   Úvisualstudio\  s    zRegistryInfo.visualstudioc             C   s   t j | jd¡S )z;
        Microsoft Visual Studio SxS registry key.
        ZSxS)r   r   r   rO   )r<   r   r   r   Úsxsc  s    zRegistryInfo.sxsc             C   s   t j | jd¡S )z8
        Microsoft Visual C++ VC7 registry key.
        ZVC7)r   r   r   rP   )r<   r   r   r   Úvcj  s    zRegistryInfo.vcc             C   s   t j | jd¡S )z;
        Microsoft Visual Studio VS7 registry key.
        ZVS7)r   r   r   rP   )r<   r   r   r   Úvsq  s    zRegistryInfo.vsc             C   s   dS )z?
        Microsoft Visual C++ for Python registry key.
        zDevDiv\VCForPythonr   )r<   r   r   r   Úvc_for_pythonx  s    zRegistryInfo.vc_for_pythonc             C   s   dS )z-
        Microsoft SDK registry key.
        zMicrosoft SDKsr   )r<   r   r   r   Úmicrosoft_sdk  s    zRegistryInfo.microsoft_sdkc             C   s   t j | jd¡S )z>
        Microsoft Windows/Platform SDK registry key.
        r   )r   r   r   rT   )r<   r   r   r   Úwindows_sdk†  s    zRegistryInfo.windows_sdkc             C   s   t j | jd¡S )z<
        Microsoft .NET Framework SDK registry key.
        ZNETFXSDK)r   r   r   rT   )r<   r   r   r   Ú	netfx_sdk  s    zRegistryInfo.netfx_sdkc             C   s   dS )z<
        Microsoft Windows Kits Roots registry key.
        zWindows Kits\Installed Rootsr   )r<   r   r   r   Úwindows_kits_roots”  s    zRegistryInfo.windows_kits_rootsFc             C   s(   | j  ¡ s|rdnd}tj d|d|¡S )a
  
        Return key in Microsoft software registry.

        Parameters
        ----------
        key: str
            Registry key path where look.
        x86: str
            Force x86 software registry.

        Return
        ------
        str: value
        r   ZWow6432NodeZSoftwareZ	Microsoft)rN   rB   r   r   r   )r<   r   r   Znode64r   r   r   Ú	microsoft›  s    zRegistryInfo.microsoftc             C   sÂ   t j}t j}| j}xª| jD ] }y||||ƒd|ƒ}W nZ ttfk
rŽ   | j ¡ sˆy||||dƒd|ƒ}W qŠ ttfk
r„   wY qŠX nwY nX yt  	||¡d S  ttfk
r¸   Y qX qW dS )a  
        Look for values in registry in Microsoft software registry.

        Parameters
        ----------
        key: str
            Registry key path where look.
        name: str
            Value name to find.

        Return
        ------
        str: value
        r   TN)
r   ÚKEY_READÚOpenKeyrX   ÚHKEYSÚOSErrorÚIOErrorrN   rB   ZQueryValueEx)r<   r   ÚnamerY   ZopenkeyÚmsÚhkeyÚbkeyr   r   r   Úlookup­  s"    

zRegistryInfo.lookupN)F)r   r	   r
   rI   r   r   r   r   r   r[   r=   rL   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rb   r   r   r   r   rM   K  s"   
rM   c               @   s$  e Zd ZdZe dd¡Ze dd¡Ze de¡Zd3dd„Z	d	d
„ Z
dd„ Zedd„ ƒZedd„ ƒZdd„ Zdd„ Zedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZedd „ ƒZed!d"„ ƒZed#d$„ ƒZed%d&„ ƒZed'd(„ ƒZed)d*„ ƒZed+d,„ ƒZed-d.„ ƒZd/d0„ Zd4d1d2„ZdS )5Ú
SystemInfozê
    Microsoft Windows and Visual Studio related system inormations.

    Parameters
    ----------
    registry_info: RegistryInfo
        "RegistryInfo" instance.
    vc_ver: float
        Required Microsoft Visual C++ version.
    ÚWinDirr   ÚProgramFileszProgramFiles(x86)Nc             C   s"   || _ | j j| _|p|  ¡ | _d S )N)ÚrirN   Ú_find_latest_available_vc_verÚvc_ver)r<   Zregistry_inforh   r   r   r   r=   â  s    
zSystemInfo.__init__c             C   s6   y|   ¡ d S  tk
r0   d}tj |¡‚Y nX d S )Nr2   z%No Microsoft Visual C++ version found)Úfind_available_vc_versÚ
IndexErrorr    r!   r"   )r<   Úerrr   r   r   rg   ç  s
    z(SystemInfo._find_latest_available_vc_verc             C   s6  | j j}| j j| j j| j jf}g }x| j jD ]ü}xö|D ]î}yt |||ƒdtj¡}W n t	t
fk
rp   w8Y nX t |¡\}}}	xPt|ƒD ]D}
y*tt ||
¡d ƒ}||kr¸| |¡ W qŒ tk
rÎ   Y qŒX qŒW xPt|ƒD ]D}
y(tt ||
¡ƒ}||kr| |¡ W qÞ tk
r    Y qÞX qÞW q8W q.W t|ƒS )zC
        Find all available Microsoft Visual C++ versions.
        r   )rf   rX   rQ   rS   rR   r[   r   rZ   rY   r\   r]   ZQueryInfoKeyÚrangeÚfloatZ	EnumValueÚappendr#   ZEnumKeyÚsorted)r<   r_   ZvckeysZvc_versr`   r   ra   ZsubkeysÚvaluesr>   Úir'   r   r   r   ri   î  s2    


z!SystemInfo.find_available_vc_versc             C   s6   d| j  }tj | j|¡}| j | jjd| j  ¡p4|S )z4
        Microsoft Visual Studio directory.
        zMicrosoft Visual Studio %0.1fz%0.1f)rh   r   r   r   ÚProgramFilesx86rf   rb   rR   )r<   r^   Údefaultr   r   r   ÚVSInstallDir  s    
zSystemInfo.VSInstallDirc             C   sŽ   | j  |  ¡ p|  ¡ }tj | jjd| j ¡}| j 	|d¡}|rNtj |d¡n|}| j 	| jj
d| j ¡pl|}tj |¡sŠd}tj |¡‚|S )z1
        Microsoft Visual C++ directory.
        z%0.1fr   ZVCz(Microsoft Visual C++ directory not found)rt   Ú	_guess_vcÚ_guess_vc_legacyr   r   r   rf   rS   rh   rb   rQ   Úisdirr    r!   r"   )r<   Úguess_vcZreg_pathZ	python_vcZ
default_vcr   Úmsgr   r   r   ÚVCInstallDir  s    zSystemInfo.VCInstallDirc          
   C   s^   | j dkrdS d}tj | j|¡}yt |¡d }tj ||¡S  tttfk
rX   Y nX dS )z*
        Locate Visual C for 2017
        g      ,@NzVC\Tools\MSVCr2   )	rh   r   r   r   rt   Úlistdirr\   r]   rj   )r<   rs   rx   Zvc_exact_verr   r   r   ru   /  s    
zSystemInfo._guess_vcc             C   s   d| j  }tj | j|¡S )z<
        Locate Visual C for versions prior to 2017
        z Microsoft Visual Studio %0.1f\VC)rh   r   r   r   rr   )r<   rs   r   r   r   rv   ?  s    
zSystemInfo._guess_vc_legacyc             C   sJ   | j dkrdS | j dkrdS | j dkr*dS | j dkr8dS | j d	krFd
S dS )zN
        Microsoft Windows SDK versions for specified MSVC++ version.
        g      "@)z7.0z6.1z6.0ag      $@)z7.1z7.0ag      &@)z8.0z8.0ag      (@)z8.1z8.1ag      ,@)z10.0z8.1N)rh   )r<   r   r   r   ÚWindowsSdkVersionF  s    




zSystemInfo.WindowsSdkVersionc             C   s   |   tj | jd¡¡S )z4
        Microsoft Windows SDK last version
        Úlib)Ú_use_last_dir_namer   r   r   ÚWindowsSdkDir)r<   r   r   r   ÚWindowsSdkLastVersionV  s    
z SystemInfo.WindowsSdkLastVersionc             C   sD  d}x8| j D ].}tj | jjd| ¡}| j |d¡}|rP qW |rNtj |¡s†tj | jjd| j	 ¡}| j |d¡}|r†tj |d¡}|r–tj |¡sàxH| j D ]>}|d| 
d¡… }d	| }tj | j|¡}tj |¡rž|}qžW |ròtj |¡s*x6| j D ],}d
| }tj | j|¡}tj |¡rú|}qúW |s@tj | jd¡}|S )z2
        Microsoft Windows SDK directory.
        r   zv%sÚinstallationfolderz%0.1fr   ZWinSDKNÚ.zMicrosoft SDKs\Windows Kits\%szMicrosoft SDKs\Windows\v%sZPlatformSDK)r|   r   r   r   rf   rU   rb   rw   rS   rh   Úrfindre   rz   )r<   Úsdkdirr'   Úlocr   Zinstall_baseZintverÚdr   r   r   r   ^  s6    zSystemInfo.WindowsSdkDirc       	      C   sÚ   | j dkrd}d}n&d}| j dkr&dnd}| jjd|d}d	|| d
d¡f }g }| j dkr†x(| jD ]}|tj | jj	||¡g7 }qdW x,| j
D ]"}|tj | jjd| |¡g7 }qŽW x |D ]}| j |d¡}|rºP qºW |S )z=
        Microsoft Windows SDK executable directory.
        g      &@é#   r   é(   g      (@TF)r9   rC   zWinSDK-NetFx%dTools%srF   ú-g      ,@zv%sAr   )rh   rN   rD   r;   ÚNetFxSdkVersionr   r   r   rf   rV   r|   rU   rb   )	r<   Znetfxverr(   rC   ZfxZregpathsr'   r   Zexecpathr   r   r   ÚWindowsSDKExecutablePath„  s$    

"
z#SystemInfo.WindowsSDKExecutablePathc             C   s.   d| j  }tj | jj|¡}| j |d¡p,dS )z0
        Microsoft Visual F# directory.
        z%0.1f\Setup\F#r   r   )rh   r   r   r   rf   rO   rb   )r<   r   r   r   r   ÚFSharpInstallDir£  s    
zSystemInfo.FSharpInstallDirc             C   sF   | j dkrd}nd}x(|D ] }| j | jjd| ¡}|rP qW |pDdS )z8
        Microsoft Universal CRT SDK directory.
        g      ,@)Z10Z81r   z
kitsroot%sr   )rh   rf   rb   rW   )r<   Zversr'   r„   r   r   r   ÚUniversalCRTSdkDir¬  s    


zSystemInfo.UniversalCRTSdkDirc             C   s   |   tj | jd¡¡S )z@
        Microsoft Universal C Runtime SDK last version
        r}   )r~   r   r   r   r   )r<   r   r   r   ÚUniversalCRTSdkLastVersion¿  s    
z%SystemInfo.UniversalCRTSdkLastVersionc             C   s   | j dkrdS dS dS )z8
        Microsoft .NET Framework SDK versions.
        g      ,@)z4.6.1z4.6r   N)rh   )r<   r   r   r   rŠ   Ç  s    
zSystemInfo.NetFxSdkVersionc             C   s>   x4| j D ]*}tj | jj|¡}| j |d¡}|rP qW |p<dS )z9
        Microsoft .NET Framework SDK directory.
        Zkitsinstallationfolderr   )rŠ   r   r   r   rf   rV   rb   )r<   r'   r…   r„   r   r   r   ÚNetFxSdkDirÒ  s    zSystemInfo.NetFxSdkDirc             C   s&   t j | jd¡}| j | jjd¡p$|S )z;
        Microsoft .NET Framework 32bit directory.
        zMicrosoft.NET\FrameworkZframeworkdir32)r   r   r   rd   rf   rb   rQ   )r<   Úguess_fwr   r   r   ÚFrameworkDir32Þ  s    zSystemInfo.FrameworkDir32c             C   s&   t j | jd¡}| j | jjd¡p$|S )z;
        Microsoft .NET Framework 64bit directory.
        zMicrosoft.NET\Framework64Zframeworkdir64)r   r   r   rd   rf   rb   rQ   )r<   r   r   r   r   ÚFrameworkDir64é  s    zSystemInfo.FrameworkDir64c             C   s
   |   d¡S )z:
        Microsoft .NET Framework 32bit versions.
        é    )Ú_find_dot_net_versions)r<   r   r   r   ÚFrameworkVersion32ô  s    zSystemInfo.FrameworkVersion32c             C   s
   |   d¡S )z:
        Microsoft .NET Framework 64bit versions.
        é@   )r”   )r<   r   r   r   ÚFrameworkVersion64û  s    zSystemInfo.FrameworkVersion64c             C   s˜   | j  | j jd| ¡}t| d| ƒ}|p6|  |d¡p6d}| jdkrL|df}n:| jdkrx| ¡ dd	… d
krndn|df}n| jdkr†d}| jdkr”d}|S )z¡
        Find Microsoft .NET Framework versions.

        Parameters
        ----------
        bits: int
            Platform number of bits: 32 or 64.
        zframeworkver%dzFrameworkDir%dÚvr   g      (@zv4.0g      $@Né   Zv4z
v4.0.30319zv3.5g      "@)zv3.5z
v2.0.50727g       @)zv3.0z
v2.0.50727)rf   rb   rQ   Úgetattrr~   rh   r3   )r<   ÚbitsZreg_verZdot_net_dirr'   Zframeworkverr   r   r   r”     s    





z!SystemInfo._find_dot_net_versionsc                s,   ‡ ‡fdd„t t ˆ ¡ƒD ƒ}t|dƒp*dS )zî
        Return name of the last dir in path or '' if no dir found.

        Parameters
        ----------
        path: str
            Use dirs in this path
        prefix: str
            Use only dirs startings by this prefix
        c             3   s2   | ]*}t j t j ˆ |¡¡r| ˆ¡r|V  qd S )N)r   r   rw   r   Ú
startswith)Ú.0Zdir_name)r   Úprefixr   r   ú	<genexpr>(  s   z0SystemInfo._use_last_dir_name.<locals>.<genexpr>Nr   )Úreversedr   r{   Únext)r<   r   rž   Zmatching_dirsr   )r   rž   r   r~     s    zSystemInfo._use_last_dir_name)N)r   ) r   r	   r
   rI   rJ   rK   rd   re   rr   r=   rg   ri   rL   rt   rz   ru   rv   r|   r€   r   r‹   rŒ   r   rŽ   rŠ   r   r‘   r’   r•   r—   r”   r~   r   r   r   r   rc   Ð  s4   

&	rc   c               @   sR  e Zd ZdZd=dd„Zedd„ ƒZedd	„ ƒZed
d„ ƒZedd„ ƒZ	edd„ ƒZ
edd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZdd„ Zedd„ ƒZedd„ ƒZed d!„ ƒZed"d#„ ƒZed$d%„ ƒZed&d'„ ƒZed(d)„ ƒZed*d+„ ƒZed,d-„ ƒZed.d/„ ƒZed0d1„ ƒZed2d3„ ƒZed4d5„ ƒZd>d7d8„Zd9d:„ Zd?d;d<„Z dS )@r$   aY  
    Return environment variables for specified Microsoft Visual C++ version
    and platform : Lib, Include, Path and libpath.

    This function is compatible with Microsoft Visual C++ 9.0 to 14.0.

    Script created by analysing Microsoft environment configuration files like
    "vcvars[...].bat", "SetEnv.Cmd", "vcbuildtools.bat", ...

    Parameters
    ----------
    arch: str
        Target architecture.
    vc_ver: float
        Required Microsoft Visual C++ version. If not set, autodetect the last
        version.
    vc_min_ver: float
        Minimum Microsoft Visual C++ version.
    Nr   c             C   sB   t |ƒ| _t| jƒ| _t| j|ƒ| _| j|k r>d}tj 	|¡‚d S )Nz.No suitable Microsoft Visual C++ version found)
r8   rN   rM   rf   rc   Úsirh   r    r!   r"   )r<   r(   rh   r,   rk   r   r   r   r=   H  s    

zEnvironmentInfo.__init__c             C   s   | j jS )z/
        Microsoft Visual C++ version.
        )r¢   rh   )r<   r   r   r   rh   Q  s    zEnvironmentInfo.vc_verc                sV   ddg}ˆ j dkrDˆ jjddd}|dg7 }|dg7 }|d| g7 }‡ fd	d
„|D ƒS )z/
        Microsoft Visual Studio Tools
        zCommon7\IDEzCommon7\Toolsg      ,@T)rC   r9   z1Common7\IDE\CommonExtensions\Microsoft\TestWindowzTeam Tools\Performance ToolszTeam Tools\Performance Tools%sc                s   g | ]}t j ˆ jj|¡‘qS r   )r   r   r   r¢   rt   )r   r   )r<   r   r   ú
<listcomp>e  s    z+EnvironmentInfo.VSTools.<locals>.<listcomp>)rh   rN   rD   )r<   ÚpathsÚarch_subdirr   )r<   r   ÚVSToolsX  s    


zEnvironmentInfo.VSToolsc             C   s$   t j | jjd¡t j | jjd¡gS )zL
        Microsoft Visual C++ & Microsoft Foundation Class Includes
        ZIncludezATLMFC\Include)r   r   r   r¢   rz   )r<   r   r   r   Ú
VCIncludesg  s    zEnvironmentInfo.VCIncludesc                sb   ˆ j dkrˆ jjdd}nˆ jjdd}d| d| g}ˆ j dkrP|d| g7 }‡ fd	d
„|D ƒS )zM
        Microsoft Visual C++ & Microsoft Foundation Class Libraries
        g      .@T)r9   )rC   zLib%szATLMFC\Lib%sg      ,@zLib\store%sc                s   g | ]}t j ˆ jj|¡‘qS r   )r   r   r   r¢   rz   )r   r   )r<   r   r   r£   }  s    z/EnvironmentInfo.VCLibraries.<locals>.<listcomp>)rh   rN   rE   )r<   r¥   r¤   r   )r<   r   ÚVCLibrarieso  s    

zEnvironmentInfo.VCLibrariesc             C   s"   | j dk rg S tj | jjd¡gS )zA
        Microsoft Visual C++ store references Libraries
        g      ,@zLib\store\references)rh   r   r   r   r¢   rz   )r<   r   r   r   ÚVCStoreRefs  s    
zEnvironmentInfo.VCStoreRefsc             C   s  | j }tj |jd¡g}| jdkr&dnd}| j |¡}|rT|tj |jd| ¡g7 }| jdkrˆd| jjdd }|tj |j|¡g7 }nŒ| jdkrþ| j 	¡ r d	nd
}|tj |j|| jj
dd ¡g7 }| jj| jjkrü|tj |j|| jjdd ¡g7 }n|tj |jd¡g7 }|S )z,
        Microsoft Visual C++ Tools
        Z
VCPackagesg      $@TFzBin%sg      ,@)rC   g      .@zbin\HostX86%szbin\HostX64%s)r9   ÚBin)r¢   r   r   r   rz   rh   rN   rH   rD   rB   rE   rA   r?   )r<   r¢   ÚtoolsrG   r¥   r   Zhost_dirr   r   r   ÚVCToolsˆ  s&    

zEnvironmentInfo.VCToolsc             C   st   | j dkr2| jjddd}tj | jjd| ¡gS | jjdd}tj | jjd¡}| j}tj |d||f ¡gS dS )	z1
        Microsoft Windows SDK Libraries
        g      $@T)rC   r9   zLib%s)r9   r}   z%sum%sN)	rh   rN   rE   r   r   r   r¢   r   Ú_sdk_subdir)r<   r¥   r}   Zlibverr   r   r   ÚOSLibraries¨  s    
zEnvironmentInfo.OSLibrariesc             C   s|   t j | jjd¡}| jdkr.|t j |d¡gS | jdkr@| j}nd}t j |d| ¡t j |d| ¡t j |d| ¡gS d	S )
z/
        Microsoft Windows SDK Include
        Úincludeg      $@Zglg      ,@r   z%ssharedz%sumz%swinrtN)r   r   r   r¢   r   rh   r­   )r<   r¯   Úsdkverr   r   r   Ú
OSIncludes·  s    

zEnvironmentInfo.OSIncludesc             C   sº   t j | jjd¡}g }| jdkr*|| j7 }| jdkrH|t j |d¡g7 }| jdkr¶||t j | jjd¡t j |dd¡t j |d	d¡t j |d
d¡t j | jjddd| j ddd¡g7 }|S )z7
        Microsoft Windows SDK Libraries Paths
        Z
Referencesg      "@g      &@zCommonConfiguration\Neutralg      ,@ZUnionMetadataz'Windows.Foundation.UniversalApiContractz1.0.0.0z%Windows.Foundation.FoundationContractz,Windows.Networking.Connectivity.WwanContractZExtensionSDKszMicrosoft.VCLibsz%0.1fZCommonConfigurationZneutral)r   r   r   r¢   r   rh   r®   )r<   ÚrefÚlibpathr   r   r   Ú	OSLibpathÊ  s>    




zEnvironmentInfo.OSLibpathc             C   s   t |  ¡ ƒS )z-
        Microsoft Windows SDK Tools
        )ÚlistÚ
_sdk_tools)r<   r   r   r   ÚSdkTools÷  s    zEnvironmentInfo.SdkToolsc             c   s  | j dk r0| j dkrdnd}tj | jj|¡V  | j ¡ sd| jjdd}d| }tj | jj|¡V  | j dksx| j dkr¶| j 	¡ rˆd	}n| jjddd
}d| }tj | jj|¡V  nL| j dkrtj | jjd¡}| jjdd}| jj
}tj |d||f ¡V  | jjr| jjV  dS )z=
        Microsoft Windows SDK Tools paths generator
        g      .@g      &@rª   zBin\x86T)r9   zBin%sg      $@r   )rC   r9   zBin\NETFX 4.0 Tools%sz%s%sN)rh   r   r   r   r¢   r   rN   rB   rD   r@   r€   r‹   )r<   Zbin_dirr¥   r   r°   r   r   r   r¶   þ  s(    



zEnvironmentInfo._sdk_toolsc             C   s   | j j}|rd| S dS )z6
        Microsoft Windows SDK version subdir
        z%s\r   )r¢   r€   )r<   Úucrtverr   r   r   r­     s    zEnvironmentInfo._sdk_subdirc             C   s"   | j dkrg S tj | jjd¡gS )z-
        Microsoft Windows SDK Setup
        g      "@ZSetup)rh   r   r   r   r¢   r   )r<   r   r   r   ÚSdkSetup$  s    
zEnvironmentInfo.SdkSetupc                s”   | j }| j‰ | jdkr0d}| ¡  o,| ¡  }n$| ¡ p>| ¡ }|jdkpR|jdk}g }|rt|‡ fdd„ˆ jD ƒ7 }|r|‡ fdd„ˆ jD ƒ7 }|S )z0
        Microsoft .NET Framework Tools
        g      $@Tr:   c                s   g | ]}t j ˆ j|¡‘qS r   )r   r   r   r‘   )r   r'   )r¢   r   r   r£   ?  s   z+EnvironmentInfo.FxTools.<locals>.<listcomp>c                s   g | ]}t j ˆ j|¡‘qS r   )r   r   r   r’   )r   r'   )r¢   r   r   r£   B  s   )	rN   r¢   rh   r@   rB   rA   r?   r•   r—   )r<   rN   Z	include32Z	include64r«   r   )r¢   r   ÚFxTools.  s    
zEnvironmentInfo.FxToolsc             C   s<   | j dk s| jjsg S | jjdd}tj | jjd| ¡gS )z8
        Microsoft .Net Framework SDK Libraries
        g      ,@T)r9   zlib\um%s)rh   r¢   r   rN   rE   r   r   r   )r<   r¥   r   r   r   ÚNetFxSDKLibrariesF  s    z!EnvironmentInfo.NetFxSDKLibrariesc             C   s*   | j dk s| jjsg S tj | jjd¡gS )z7
        Microsoft .Net Framework SDK Includes
        g      ,@z
include\um)rh   r¢   r   r   r   r   )r<   r   r   r   ÚNetFxSDKIncludesQ  s    z EnvironmentInfo.NetFxSDKIncludesc             C   s   t j | jjd¡gS )z>
        Microsoft Visual Studio Team System Database
        zVSTSDB\Deploy)r   r   r   r¢   rt   )r<   r   r   r   ÚVsTDb[  s    zEnvironmentInfo.VsTDbc             C   s~   | j dk rg S | j dk r0| jj}| jjdd}n| jj}d}d| j |f }tj ||¡g}| j dkrz|tj ||d¡g7 }|S )z(
        Microsoft Build Engine
        g      (@g      .@T)rC   r   zMSBuild\%0.1f\bin%sZRoslyn)	rh   r¢   rr   rN   rD   rt   r   r   r   )r<   Ú	base_pathr¥   r   Zbuildr   r   r   ÚMSBuildb  s    


zEnvironmentInfo.MSBuildc             C   s"   | j dk rg S tj | jjd¡gS )z.
        Microsoft HTML Help Workshop
        g      &@zHTML Help Workshop)rh   r   r   r   r¢   rr   )r<   r   r   r   ÚHTMLHelpWorkshopy  s    
z EnvironmentInfo.HTMLHelpWorkshopc             C   sL   | j dk rg S | jjdd}tj | jjd¡}| j}tj |d||f ¡gS )z=
        Microsoft Universal C Runtime SDK Libraries
        g      ,@T)r9   r}   z%sucrt%s)	rh   rN   rE   r   r   r   r¢   r   Ú_ucrt_subdir)r<   r¥   r}   r¸   r   r   r   ÚUCRTLibrariesƒ  s    
zEnvironmentInfo.UCRTLibrariesc             C   s6   | j dk rg S tj | jjd¡}tj |d| j ¡gS )z;
        Microsoft Universal C Runtime SDK Include
        g      ,@r¯   z%sucrt)rh   r   r   r   r¢   r   rÁ   )r<   r¯   r   r   r   ÚUCRTIncludes  s    
zEnvironmentInfo.UCRTIncludesc             C   s   | j j}|rd| S dS )zB
        Microsoft Universal C Runtime SDK version subdir
        z%s\r   )r¢   rŽ   )r<   r¸   r   r   r   rÁ   ›  s    zEnvironmentInfo._ucrt_subdirc             C   s    | j dk r| j dkrg S | jjS )z%
        Microsoft Visual F#
        g      &@g      (@)rh   r¢   rŒ   )r<   r   r   r   ÚFSharp£  s    zEnvironmentInfo.FSharpc             C   sl   | j jdd}| jdk r&| jj}d}n| jj dd¡}d}| jdkrHdn| j}||| j|f }tj ||¡S )	zA
        Microsoft Visual C++ runtime redistribuable dll
        T)r9   é   z-redist%s\Microsoft.VC%d0.CRT\vcruntime%d0.dllz\Toolsz\Redistz.onecore%s\Microsoft.VC%d0.CRT\vcruntime%d0.dllg      ,@)	rN   rE   rh   r¢   rz   r;   r   r   r   )r<   r¥   Zredist_pathZ	vcruntimeZdll_verr   r   r   ÚVCRuntimeRedist­  s    
zEnvironmentInfo.VCRuntimeRedistTc             C   s¶   t |  d| j| j| j| jg|¡|  d| j| j| j| j	| j
g|¡|  d| j| j| j| jg|¡|  d| j| j| j| j| j| j| j| j| jg	|¡d}| jdkr²tj | j¡r²| j|d< |S )z—
        Return environment dict.

        Parameters
        ----------
        exists: bool
            It True, only return existing paths.
        r¯   r}   r³   r   )r¯   r}   r³   r   é   Zpy_vcruntime_redist)ÚdictÚ_build_pathsr§   r±   rÃ   r¼   r¨   r®   rº   rÂ   r»   r©   r´   r¬   r¦   r½   r·   r¹   r¿   rÀ   rÄ   rh   r   r   r   rÆ   )r<   ÚexistsÚenvr   r   r   r%   À  sD    	

zEnvironmentInfo.return_envc       
      C   sx   t j |¡}t |d¡ tj¡}t  ||¡}|rBtt	tj
j|ƒƒn|}|sbd| ¡  }tj |¡‚|  |¡}	tj |	¡S )a  
        Given an environment variable name and specified paths,
        return a pathsep-separated string of paths containing
        unique, extant, directories from those paths and from
        the environment variable. Raise an error if no paths
        are resolved.
        r   z %s environment variable is empty)Ú	itertoolsÚchainÚfrom_iterablerJ   rK   Úsplitr   Úpathseprµ   Úfilterr   rw   Úupperr    r!   r"   Ú_unique_everseenr   )
r<   r^   Zspec_path_listsrÊ   Z
spec_pathsZ	env_pathsr¤   Zextant_pathsry   Zunique_pathsr   r   r   rÉ   í  s    	
zEnvironmentInfo._build_pathsc             c   sj   t ƒ }|j}|dkr:xPt|j|ƒD ]}||ƒ |V  q"W n,x*|D ]"}||ƒ}||kr@||ƒ |V  q@W dS )z×
        List unique elements, preserving order.
        Remember all elements ever seen.

        _unique_everseen('AAAABBBCCDAABBB') --> A B C D

        _unique_everseen('ABBCcAD', str.lower) --> A B C D
        N)ÚsetÚaddr   Ú__contains__)r<   Úiterabler   ÚseenZseen_addÚelementÚkr   r   r   rÓ     s    	
z EnvironmentInfo._unique_everseen)Nr   )T)N)!r   r	   r
   rI   r=   rL   rh   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Ó   r   r   r   r   r$   0  s:   
		 -




-r$   )r   )r   )!rI   r   r.   ÚplatformrÌ   Zdistutils.errorsr    Z#setuptools.extern.packaging.versionr   Zsetuptools.extern.six.movesr   Zmonkeyr   Úsystemr   ÚenvironrJ   rÈ   ÚImportErrorr!   r"   Z_msvc9_suppress_errorsZdistutils.msvc9compilerr   r   r   r-   r1   r&   r8   rM   rc   r$   r   r   r   r   Ú<module>   s>   
+
/&
$[   b