o e@sddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZejrQdd lmZdd lmZddlmZee je jhdddZeed<GdddeZdS)) annotations)PathN) OptionKey)DependencyMethods)CMakeDependency)packages)PkgConfigDependency)factory_methods) Environment) MachineChoice)DependencyGeneratorenv 'Environment' for_machine'MachineChoice'kwargsT.Dict[str, T.Any]methodsT.List[DependencyMethods]returnT.List['DependencyGenerator']c Csg}tj|vr4|d|jtd}|rdnd}|tt |||dD] }|tt |||q&tj |vrD|tt d|||S)NstaticZ prefer_staticzmkl-static-lp64-iompzmkl-dynamic-lp64-iomp)zscalapack-openmpi scalapackZ Scalapack) r PKGCONFIGgetZcoredataZ get_optionrappend functoolspartialMKLPkgConfigDependencyr CMAKEr)rrrr candidatesZ static_optZmklpkgr$F/usr/lib/python3.10/site-packages/mesonbuild/dependencies/scalapack.pyscalapack_factory"s    r&rcs>eZdZdZ ddfd d Zdfdd ZdddZZS)r zPkgConfigDependency for Intel MKL. MKL's pkg-config is pretty much borked in every way. We need to apply a bunch of fixups to make it work correctly. NnamestrrrrrlanguageT.Optional[str]cstjd}|rt|nd|_tj||||d|jr,|j|j  r/|j j dkr/d|_ |j s4dS|jdks=Jd|jdkrx|jddd }|sj|jrjz|jd d d d d }Wn tyiYnw|rzt|tssJ||_dSdSdS)NZMKLROOT)r)gccFz0This should not happen if we didn't return aboveunknownVersion) pkgconfig default_valueZcompilers_and_libraries_r/r)osenvironrrresolve _MKLPkgConfigDependency__mklrootsuper__init__machinesr is_windows clib_compileridZis_foundversionZ get_variableas_posixsplit IndexError isinstancer()selfr'rrr)_mv __class__r$r%r7Cs6       zMKLPkgConfigDependency.__init__rNonecsHt|jj|jrd}n|jrd}nd}|jd}|jj dkrLt |j D]"\}}t j |\}}d|vrK|dd}d t j |||j |<q)d }t |j D]\}}|d ra|d }qS|d krgnqS|jj|jst|jr|j |t|d||j |d t|d|dS|j |d|j |d ddS)Nz.libz.ar,z lib/intel64r+Zmkl_intel_lp64intelZgfr1r)z-Lz-WlrZmkl_scalapack_lp64Zmkl_blacs_intelmpi_lp64z-lmkl_scalapack_lp64z-lmkl_blacs_intelmpi_lp64)r6 _set_libsrr8rr9rr5r:r; enumerateZ link_argsr2pathr>replacejoin startswithinsertr()rAsuffixlibdiriadirnamebasenamejrDr$r%rIls<      z MKLPkgConfigDependency._set_libscCsBd}|jdkr d}|jj|j|d|jffd}|||_dS)NFZfortranTprefix)Zdefine_variable)r)r/cflagsr'r5r=Z_convert_mingw_pathsZ compile_args)rAZ allow_systemrXr$r$r% _set_cargss   z!MKLPkgConfigDependency._set_cargs)N)r'r(rrrrr)r*)rrF)__name__ __module__ __qualname____doc__r7rIrY __classcell__r$r$rDr%r ;s )%r ) rrrrrrrrrr) __future__rpathlibrrr2typingTZmesonlibrbaserZcmakerdetectr r/r factoryr TYPE_CHECKING environmentr r rrr!r&r r$r$r$r%s&