o @aE@sdZddlmZmZddlmZddlmZmZm Z m Z ddlm Z m Z m Z mZddlmZddlmZmZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddl m!Z!m"Z"ddl m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6m7Z7ddl8Z9ddl:Z:ddl;ZZ>ddl?Z?ddl@mAZAddlBZBe:CdZDddZEddZFdd ZGd!d"ZH  d4d#d$ZId5d%d&ZJ  d4d'd(ZKd6d)d*ZLd7d,d-ZM  + d8d.d/ZN  +d9d0d1ZOd2d3ZPdS):z This module contains all the stuff to make your way from python code to a dynamic library, see __init__.py for exported interfaces. )CxxPython)cfg) PythonModuleIncludeLine Statement) FunctionBodyFunctionDeclarationValueBlock)ReturnStatement)PythranExtensionPythranBuildExt)refinemark_unexported_functions) PassManager) pythran_ward)tog)pytype_to_depspytype_to_ctype) load_specfileSpec)spec_to_string) check_specs check_exportsPythranSyntaxError) __version__)cxxidN)datetime) CompileError) sysconfig)setup)mkdtempNamedTemporaryFile)reducepythrancCspt}|jD]}|D]}|D] }|t|qq q|jD]}|D] }|t|q%q!t|dddS)zD Extract types dependencies from specs for each exported signature. cSsd|vS)NZinclude)xr(r(3/usr/lib/python3/dist-packages/pythran/toolchain.py;sz-_extract_specs_dependencies..)key)set functionsvaluesupdatercapsulessorted)specsZdeps signatures signaturetr(r(r*_extract_specs_dependencies+sr7cCs>|d}t|dkrddg|}tt|ddt|dS)zqTurns an optimization of the form my_optim my_package.my_optim into the associated symbol.r' optimizationsNr)splitlenr&getattr __import__) optimizationZsplittedr(r(r*_parse_optimization>s   r@cCsBtd|dd}|||jWdS1swYdS)zwrite `content` to a temporary XXX`suffix` file and return the filename. It is user's responsibility to delete when done.wF)modesuffixdeleteN)r%writename)contentrCoutr(r(r* _write_tempIs $rIcCs<|jD]}t|tjr|j|krdd|jjDSqgS)z(Checks if a given function has argumentscSsg|]}t|jqSr()rid).0argr(r(r* Usz has_argument..)body isinstanceastZ FunctionDefrFargs)modulefnamenr(r(r* has_argumentQs rUcCsht||}t||\}}|durt||}|dur"tdd}dd|D}t||||||fS)z*Front-end and middle-end compilation stepsNr'r:cSg|]}t|qSr()r@)rKZoptr(r(r*rMhz$front_middle_end..)rfrontendparserrgetr;r) module_namecoder: module_dir entry_pointspmir docstringsr(r(r*front_middle_endYs    rbcCs t||||\}}}|t|S)zYpython + pythran spec -> py code Prints and returns the optimized python code. )rbdumpr)r[r\r:r]r_r`_r(r(r* generate_pyqs recs2r t}nd}t|||||d\}}|t}dur7Gdddt} | |} fdd} | | fSttrAtifdd}  |t |t|t r[|} n| dd } t | ttd } t||| } | td td td tdtd| jddtD| j|j| tddd}jD]~\}}t|}|s| |d|||t|D]c\}}d||}dd|D}t|}ddt ||D}t!d}d"|}||||rd|nd}d|}| #t$t%t&||ddt ||Dt't(d|||d"|g|||qqj)D]g\}}t|}d d|D}t|}d!dt ||D}t!d}d"|}||||r`d|nd}d|}t*||}| +t$t%t&||d"dt ||Dt't,d#|||d"|g||q-| | fS)$zpython + pythran spec -> c++ code returns a PythonModule object and an error checker the error checker can be used to print more detailed info on the origin of a compile error (e.g. due to bad typing) N)r^c@s eZdZddZddZeZdS)zgenerate_cxx..GenerablecSs ||_dSN)rG)selfrGr(r(r*__init__ z(generate_cxx..Generable.__init__cSs t|jSrf)strrG)rgr(r(r*__str__riz'generate_cxx..Generable.__str__N)__name__ __module__ __qualname__rhrkZgenerater(r(r(r* GenerablesrocstdSrf)r typecheckr()r`r(r* error_checkersz#generate_cxx..error_checkercst}t|dSrf)rrpr)typesr`r3r(r*rqs asciiignore)hashversiondatezpythonic/core.hppzpythonic/python/core.hppzpythonic/types/bool.hppzpythonic/types/int.hppz&#ifdef _OPENMP #include #endifcSrVr()r)rKZincr(r(r*rMrWz generate_cxx..z%pythonic/python/exception_handler.hppcSstd||S)Nz{0}::{1})rformat)r[Z internal_namer(r(r*wardedszgenerate_cxx..wardedz{}()()z{0}{1}cSrVr(rrKr6r(r(r*rMrWcSg|]\}}|qSr(r(rKrTrdr(r(r*rMrWz{0}::{1}::type{2}z, z<{0}>ztypename %s::result_typecSsg|] \}}t|d|qS)z&&r rKr6ar(r(r*rMsa PyThreadState *_save = PyEval_SaveThread(); try {{ auto res = {0}()({1}); PyEval_RestoreThread(_save); return res; }} catch(...) {{ PyEval_RestoreThread(_save); throw; }} cSrVr(rr{r(r(r*rMrWcSr|r(r(r}r(r(r*rMrWcSsg|] \}}t||qSr(rrr(r(r*rMsz {0}()({1}))-r-keysrbrcrobjectrOdictrZ to_docstringsrbytesencodehashlibZsha256Z hexdigestrr ZnowrZadd_to_includesrrr7rNr.itemsrZadd_global_varry enumeraterUziprjoinZadd_pyfunctionr r r r rr1rZ add_capsuler )r[r\r3r:r]r^r_rarGromodrqZ code_bytesZmetainforzZ function_namer4Zinternal_func_nameZsigidr5Znumbered_function_nameZarguments_typesZarguments_namesZ argumentsZname_fmtZ args_listZspecialized_fnameZ result_typeZ docstringr(rsr* generate_cxx|s   z             *    rc Ksbt}t}t||gfi|}zt||gdtidttjr!dnddd|d|gdWnty>}zt t |d}~wwd d }t d } t tj||d D]B} | | ru|rd|d | }n tjt|| }|| |qT|r|d d}tj|} nt} || tj| tj| qTt|t|td|td||S)zyc++ file -> native module Return the filename of the produced shared library Raises CompileError on failure Z build_extzsetup.pyz --verbosez--quietz --build-libz --build-temp)rFZ ext_modulesZcmdclass script_nameZ script_argsNc Ssxt|d-}t|d}||Wdn1swYWddSWddS1s5wYdS)Nrbwb)openrEread)Zsrc_fileZ dest_filesrcdestr(r(r*copy:s  "zcompile_cxxfile..copyZ EXT_SUFFIX*%{ext}r~zGenerated module: zOutput: )r$rr#rloggerZ isEnabledForloggingINFO SystemExitr!rjr"Zget_config_varglobospathrendswithreplacegetcwddirnamebasenameshutilrmtreeinfo) r[Zcxxfile output_binarykwargsZbuilddirZbuildtmp extensionerextfZoutput_directoryr(r(r*compile_cxxfilesT         rFcKsBt|d}t|||fi|}|st||Std||S)zZc++ code (string) -> temporary file -> native module. Returns the generated .so. .cppz!Keeping temporary generated file:)rIrrremoverwarning)r[ZcxxcoderZ keep_temprZfdpathr(r(r*compile_cxxcodeYs  rcKsX|r*t||||} |durt| dSt| d} |d}t| |td|ddlm } |dur8| |}t |||||\} } d| dgvra| j dtd| j dtd tjj|rtt| d } |rs|d d }n|d }t| |td ||Szt|t| fd |i|}W|Stytd| tdw)a Pythran code (string) -> c++ code -> native module if `cpponly` is set to true, return the generated C++ filename if `pyonly` is set to true, prints the generated Python filename, unless `output_file` is set otherwise, return the generated native library filename Nz.pyzGenerated Python source file: r) spec_parserZENABLE_PYTHON_MODULEZ undef_macrosz#undef ENABLE_PYTHON_MODULEz#define PY_MAJOR_VERSION {}rrzGenerated C++ source file: rz4Compilation error, trying hard to find its origin...z,Nop, I'm going to flood you with C++ errors!)reprintrIryrmoverr pythran.specrrrZZpreambleinsertrsys version_infomajorrjrrr!r)r[Z pythrancoder3Zoptscpponlypyonly output_filer]rrGZtmp_filerrRrqr(r(r*compile_pythrancodemsV           rc Ks|stj|\}}|ptj|d}ntj|dd\}}|p*|ddd}tj|}tj|dd} tj| rKt| } |d| z)t |} t || f||||d|}Wd W|S1smwYW|St y} z | j d ur|| _ d } ~ ww) a Pythran file -> c++ file -> native module. Returns the generated .so (or .cpp if `cpponly` is set to true). Usage without an existing spec file >>> with open('pythran_test.py', 'w') as fd: ... _ = fd.write('def foo(i): return i ** 2') >>> cpp_path = compile_pythranfile('pythran_test.py', cpponly=True) Usage with an existing spec file: >>> with open('pythran_test.pythran', 'w') as fd: ... _ = fd.write('export foo(int)') >>> so_path = compile_pythranfile('pythran_test.py') Specify the output file: >>> import sysconfig >>> ext = sysconfig.get_config_vars()["SO"] >>> so_path = compile_pythranfile('pythran_test.py', output_file='foo'+ext) rrr~r8r9z.pythranr3)rrrr]N)rrr;splitextrrisfiler setdefaultrrrrfilename) Z file_pathrr[rrrrdrr]Z spec_filer3fdrr(r(r*compile_pythranfiles:         rcCs&d}td|}|ot|dSdS)zMSimple passthrough compile test. May raises CompileError Exception. z* #include testN)rrr)r\rr(r(r* test_compiles r)NNN)NNrf)NF)NNFFNN)NNFF)Q__doc__Zpythran.backendrrZpythran.configrZpythran.cxxgenrrrrr r r r r Z pythran.distrrZpythran.middlendrrZpythran.passmanagerrZpythran.tablesrZ pythran.typesrZpythran.types.type_dependenciesrZpythran.types.conversionrrrrrZpythran.syntaxrrrZpythran.versionrZ pythran.utilsrZpythran.frontendrXr Zdistutils.errorsr!Z distutilsr"Znumpy.distutils.corer#Ztempfiler$r%ZgastrPros.pathrrrr functoolsr&rZ getLoggerrr7r@rIrUrbrerrrrrrr(r(r(r*sn                      ! = ? :