o e #@sddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZddlmZdd lmZdd lmZejrSdd lmZdd lmZdd lmZee je je jhd#ddZeed<GdddeZGdddeZGdd d eZ Gd!d"d"e Z!dS)$) annotationsN)detect_cpu_family)DependencyMethodsdetect_compilerSystemDependency)ConfigToolDependency)packages)factory_methods)PkgConfigDependency)DependencyGenerator) Environment) MachineChoiceenv 'Environment' for_machine'MachineChoice'kwargsT.Dict[str, T.Any]methodsT.List[DependencyMethods]returnT.List['DependencyGenerator']c Cs|dd}|dvr gSg}td|||}|sgS|dv}tj|vrI|sId}|dkr/d}n |dkr6d}n|d krfortrancpprmpi>intel-clintelzompi-crzompi-cxxrz ompi-fortr-v version_argZreturncode_valueZI_MPI_CCZmpiiccZ I_MPI_CXXZmpiicpcZ I_MPI_F90ZmpiifortZMPICCZmpiccZMPICXXzmpic++ZmpicxxZmpiCCcSsg|]}tj|qS)osenvironget).0er%r%@/usr/lib/python3.10/site-packages/mesonbuild/dependencies/mpi.py Wszmpi_factory..)ZMPIFCZMPIF90ZMPIF77)ZmpifortZmpif90Zmpif77cSsg|]}|r|qSr%r%)r)tr%r%r+r,\stoolsrmsmpi)r(rZget_idr PKGCONFIGappend functoolspartialr CONFIG_TOOLcopymachines is_windowsr&r'IntelMPIConfigToolDependencyextendOpenMPIConfigToolDependencySYSTEMMSMPIDependency) rrrrr candidatescompilerZcompiler_is_intelpkg_nameZnwargsZ tool_namesclsr%r%r+ mpi_factory"sf      rArc@s*eZdZdddZdddZdd d Zd S)_MPIConfigToolDependencyargs T.List[str]rcCsg}d}|jdkr|jjj|jd}||7}d}|D]0}|d|s8|dks8|drD|dkrD|dsD||||vrCd }q|rMd}||q|S) r MPI wrappers return a bunch of garbage args. Drop -O2 and everything that is not needed. )-IrF)z-Dz-f-pthread-W-Wall-WerrorT)rrcoredata compilersrZget_module_incdir_args startswithr1)selfrCresultZ multi_argsZfc include_nextfr%r%r+_filter_compile_argsns(    z-_MPIConfigToolDependency._filter_compile_argscCsJg}d}|D]}||r|||dvrd}q|r"d}||q|S)rEF>-Xlinker-LT) _is_link_argr1)rNrCrOrPrQr%r%r+_filter_link_argss   z*_MPIConfigToolDependency._filter_link_argsrQstrboolcCsX|jjdkr|dkp|dp|dS|dp+|dkp+|do+|dko+|d  S) Nrz/linkz/LIBPATHz.lib)rT-lrSrGrHrIrJ)Z clib_compileridrMendswith)rNrQr%r%r+rUs  z%_MPIConfigToolDependency._is_link_argN)rCrDrrD)rQrWrrX)__name__ __module__ __qualname__rRrVrUr%r%r%r+rBls  rBc4eZdZdZdZ ddfd d ZdddZZS)r8z*Wrapper around Intel's mpiicc and friends.r"NnamerWrrrrrT.Optional[str]csHtj||||d|jsdS|dgd}|||_|||_dS)Nr!z-showzlink and compile argssuper__init__is_foundZget_config_valuerR compile_argsrV link_args)rNr`rrrrC __class__r%r+rds  z%IntelMPIConfigToolDependency.__init__outrcCs,td|}|rd|d|dS|S)Nz(\d{4}) Update (\d)z{}.{}rr)researchformatgrouprNrjvr%r%r+_sanitize_versions z.IntelMPIConfigToolDependency._sanitize_versionNr`rWrrrrrrarjrWrrWr\r]r^__doc__r#rdrq __classcell__r%r%rhr+r8s  r8cr_)r:z)Wrapper around OpenMPI mpicc and friends.z--showme:versionNr`rWrrrrrracsVtj||||d|jsdS|dgd}|||_|dgd}|||_dS)Nr!z--showme:compilerfz --showme:linkrgrb)rNr`rrrZc_argsZl_argsrhr%r+rds z$OpenMPIConfigToolDependency.__init__rjrcCstd|}|r |dS|S)Nz \d+.\d+.\d+r)rkrlrnror%r%r+rqs  z-OpenMPIConfigToolDependency._sanitize_versionrrrsrtrur%r%rhr+r:s  r:cs&eZdZdZ d dfd d ZZS)r<zThe Microsoft MPI.Nr`rWrrrrrrac s tj||||d|dvrd|_dS|jj|jsdStj d}t |jj j j }d}|dkr;tj d}d}n |dkrGtj d}d }|dusO|durTd|_dSd |_d tj|d g|_d |d tj||g|_|jdkr|jd tj|ddSdS)Nr!>NrrFZ MSMPI_INCx86Z MSMPI_LIB32x86_64Z MSMPI_LIB64x64TrYr/rFrZmsmpifec)rcrdrerr6rr7r&r'r(rrKrLhostpathjoinrgrfrr1) rNr`rrrincdirarchlibdirpostrhr%r+rds0    zMSMPIDependency.__init__rrrs)r\r]r^rvrdrwr%r%rhr+r<sr<) rrrrrrrrrr)" __future__rr2typingTr&rk environmentrbaserrrZ configtoolr detectr factoryr Z pkgconfigr TYPE_CHECKINGr rZmesonlibrr0r4r;rArBr8r:r<r%r%r%r+s,          F4