o í@Ëaåã@sždZddlmZddlmZmZmZddlmZm Z ddl Z ddl Z ddl Z zddl mZWn ey;dd„ZYnweƒZGd d „d eƒZeƒe je<dS) z:OpenMP wrapper using a libgomp dynamically loaded library.é)Ú find_library)Ú check_outputÚCalledProcessErrorÚDEVNULL)Úmsvc_runtime_majorÚget_shared_lib_extensionN)ÚcompilercCstj dd¡S)NZCXXzc++)ÚosÚenvironÚget©r r ú./usr/lib/python3/dist-packages/omp/__init__.pyrsrc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚOpenMPz¦ Internal representation of the OpenMP module. Custom class is used to dynamically add omp runtime function to this library when function is called. cCs(tƒ}|dur | ¡dS| |¡dS)N)rÚ init_not_msvcÚ init_msvc)ÚselfÚverr r r Ú__init__s zOpenMP.__init__cCs.td|ƒ}|s tdƒ‚t |¡|_d|_dS)Nz vcomp%d.dllz(I can't find a shared library for vcomp.é)rÚ ImportErrorÚctypesÚCDLLÚlibompÚversion)rrZ vcomp_pathr r r r%s   zOpenMP.init_msvccCsgd¢S)z&Return list of OpenMP libraries to try)ZompZgompZiomp5r )rr r r Úget_libomp_names.szOpenMP.get_libomp_namesc Cs`g}tjdkr ddg}ndg}g}|D]}tj |d¡}|r'| | tj¡¡q| ¡}t durb|D]/}t d  |t ƒ¡g}zt |t d}tj | ¡ ¡¡} | rU| | ¡Wq2ttfyaYq2w|D]B}t|ƒ} | s‡|D]} tj | d   |t ƒ¡¡} tj | ¡r†| } nqn| r¦zt | ¡|_WntyŸtd   | ¡ƒ‚wd |_dSqdtd   |d ¡ƒ‚)z? Find OpenMP library and try to load if using ctype interface. ÚdarwinZDYLD_LIBRARY_PATHZDYLD_FALLBACK_LIBRARY_PATHZLD_LIBRARY_PATHÚNz-print-file-name=lib{}{})Ústderrzlib{}{}z[found openMP library '{}' but couldn't load it. This may happen if you are cross-compiling.é-zkI can't find a shared library for libomp, you may need to install it or adjust the {} environment variable.r)ÚsysÚplatformr r r ÚextendÚsplitÚpathseprÚcxxÚformatrrrÚpathÚdirnameÚdecodeÚstripÚappendÚOSErrorrrÚjoinÚisfilerrrrr) rZenv_varsÚpathsZenv_varZ env_pathsZ libomp_namesZ libomp_nameÚcmdÚoutputr&Z libomp_pathZcandidate_pathr r r r2st  €þÿÿ €ÿÿþ þ ÿÿù ÿzOpenMP.init_not_msvccCs|dkr|jSt|jd|ƒS)z£ Get correct function name from libgomp ready to be use. __getattr__ is call only `name != libomp` as libomp is a real attribute. ZVERSIONZomp_)rÚgetattrr)rÚnamer r r Ú __getattr__vszOpenMP.__getattr__N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr3r r r r rs  Dr)r7Z ctypes.utilrÚ subprocessrrrZnumpy.distutils.misc_utilrrrr rZpythran.configrrr$ÚobjectrÚmodulesr4r r r r Ús   ÿm