a bR@sdZddlmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z eZejddkZzeWneyddlmZYn0z ddlZWneyddlZYn0ddlmZmZddlmZddlmZdd lmZdd lm Z m!Z!m"Z"zdd l#m$Z$Wn ey:dd l%m$Z$Yn0dd l&m'Z'd dl(m)Z*d dl+m,Z,ddl-m.Z.ddl/m0Z0ddggdddgdgddZ1e1de1d<erddZ2nddZ2e!Gdd d e Z3ej4e'e3j5j6d!d"e'e3j.je'e3j7jd#ZdS)$uA ===================== Cython related magics ===================== Magic command interface for interactive work with Cython .. note:: The ``Cython`` package needs to be installed separately. It can be obtained using ``easy_install`` or ``pip``. Usage ===== To enable the magics below, execute ``%load_ext cython``. ``%%cython`` {CYTHON_DOC} ``%%cython_inline`` {CYTHON_INLINE_DOC} ``%%cython_pyximport`` {CYTHON_PYXIMPORT_DOC} Author: * Brian Granger Code moved from IPython and adapted by: * Martín Gaitán Parts of this code were taken from Cython.inline. )absolute_importprint_functionN)reload) Distribution Extension) build_ext)display)magic_arguments)Magics magics_class cell_magic)get_ipython_cache_dir)dedent) __version__) CompileError) cython_inline) cythonizez-fprofile-generate-fprofile-dir={TEMPDIR})z -fprofile-usez-fprofile-correctionr)genusez -prof-genz -prof-use)gcciccrZmingw32cCst|tr|S|tSN) isinstancebytesencode IO_ENCODINGnamer"i/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/Cython/Build/IpythonMagic.py encode_fsjsr$cCs|Srr"r r"r"r#r$mscseZdZfddZddZeddZeddZeej d d d d d dej ddd d ddej dddddddej dddddddej ddd d ddej dd d!gd"dej d#d!gd$dej d%d&d!gd'dej d(d)d*d+ej d,d-d.d!gd/d0ej d1d2d!gd3dej d4d5d!gd6dej d7d8d d d9d:ej d;dd?d:ed@dAZ dBdCZ dRdDdEZ dSdFdGZ dHdIZedJdKZdLdMZddddefdNdOZedPdQZZS)T CythonMagicscs&tt||i|_i|_d|_dS)NF)superr%__init___reloads _code_cache_pyximport_installed)selfshell __class__r"r#r'tszCythonMagics.__init__c Csp|j}d|vr|d}ndd|D}|D]@}z|j|||iWq*tyhd|}t|Yq*0q*dS)N__all__cSsg|]}|ds|qS)_) startswith).0kr"r"r# z,CythonMagics._import_all..z%'module' object has no attribute '%s')__dict__r,pushKeyErrorAttributeError)r+moduleZmdictkeysr3msgr"r"r# _import_allzs  zCythonMagics._import_allcCs|jj}|jj}t|||dS)aCompile and run a Cython code cell using Cython.inline. This magic simply passes the body of the cell to Cython.inline and returns the result. If the variables `a` and `b` are defined in the user's namespace, here is a simple example that returns their sum:: %%cython_inline return a+b For most purposes, we recommend the usage of the `%%cython` magic. )localsglobals)r,Zuser_global_nsZuser_nsr)r+linecellZlocsZglobsr"r"r#rszCythonMagics.cython_inlinecCs|}|std|d}tj|ddd}||Wdn1sL0Ydtjvsf|js|ddl}| d |_||j vr|j |}nt |tj|}||j |<| |dS) a)Compile and import a Cython code cell using pyximport. The contents of the cell are written to a `.pyx` file in the current working directory, which is then imported using `pyximport`. This magic requires a module name to be passed:: %%cython_pyximport modulename def f(x): return 2.0*x The compiled module is then imported and all of its symbols are injected into the user's namespace. For most purposes, we recommend the usage of the `%%cython` magic. zmodule name must be given.pyxwutf-8encodingN pyximportrT) strip ValueErrorioopenwritesysmodulesr*rGinstallr( __import__r=)r+r@rA module_namefnamefrGr:r"r"r#cython_pyximports (    zCythonMagics.cython_pyximportz-az --annotate store_trueFz/Produce a colorized HTML version of the source.)actiondefaulthelpz-+z--cplusz Output a C++ rather than C file.z-3language_level store_constrNzSelect Python 3 syntax.)destrVconstrWrXz-2rzSelect Python 2 syntax.z-fz--forcezWForce the compilation of a new module, even if the source has been previously compiled.z-cz--compile-argsappendzoExtra flags to pass to compiler via the `extra_compile_args` Extension flag (can be specified multiple times).z --link-argszjExtra flags to pass to linker via the `extra_link_args` Extension flag (can be specified multiple times).z-lz--libzNAdd a library to link the extension against (can be specified multiple times).z-nz--namez%Specify a name for the Cython module.)rXz-L library_dirsdirzPAdd a path to the list of library directories (can be specified multiple times).)r[metavarrVrWrXz-Iz --includezPAdd a path to the list of include directories (can be specified multiple times).z-Sz--srczFAdd a path to the list of src files (can be specified multiple times).z--pgopgozEnable profile guided optimisation in the C compiler. Compiles the cell twice and executes it in between to generate a runtime profile.)r[rVrWrXz --verbosequiet store_falseTz_Print debug information like generated .c/.cpp file location and exact gcc/g++ command invoked.c Cs\t|j|}|dr|n|d}tjtd}||tj tj t f}tj |s\t ||jrj|d7}|jr~|tf7}|jrt|j}ndtt|d}tj||d}tj|||j} tj| } |jp| } |jrtj|sd} d} | rj|j|||||jd } | dur2dSt| d ksDJ| d } ||j|<|jrj|| |z$|j| ||jrd nd|jd Wnt j!j"yYdS0t#$|| }|%||jrXz.z6No PGO %s configuration known for C compiler type '%s'rg) compiler compiler_type compiler_so PGO_CONFIGgetrr]rrrrMr) r+rZ step_namerrZ compiler_cmdconfigZ orig_flagsflagsrr"rr#_add_pgo_flagss2     zCythonMagics._add_pgo_flagscCs6z|jWSty0|d|_|jYS0dS)z*The extension suffix for compiled modules.N)Z_so_extr9rZget_ext_filenamer+r"r"r#rxs  zCythonMagics.so_extcCs0zddlm}Wnty"Yn 0|dS)zzclear distutils mkpath cache prevents distutils from skipping re-creation of dirs that have been removed r) _path_createdN)distutils.dir_utilr ImportErrorclear)r+rr"r"r#_clear_distutils_mkpath_caches  z*CythonMagics._clear_distutils_mkpath_cachec s|t}|}z|dWnty6Yn0||sJ||jrr|Gfddd|}||}|rt|_ |rt|}||_ |dur|g|_ |S)Nz setup.cfgcseZdZfddZdS)z5CythonMagics._get_build_extension.._build_extcs||dSr)build_extensionsrZ add_pgo_flagsZbase_build_extrfrr"r#rs zFCythonMagics._get_build_extension.._build_ext.build_extensionsN)__name__ __module__ __qualname__rr"rr"r# _build_extsr) rrfind_config_filesremoverIparse_config_filesrfinalize_optionsr$ build_temp build_libr) r+rrrrfrdist config_filesrr"rr#rs2  z!CythonMagics._get_build_extensioncs*tddfdd|D}|S)zClean up the annotated HTML source. Strips the link to the generated C or C++ file, which we do not present to the user. z&

Raw output: (.*)rdc3s|]}|s|VqdSr)match)r2lrr"r# +r5z4CythonMagics.clean_annotated_html..)recompilerl splitlines)htmlr"rr#r#s z!CythonMagics.clean_annotated_html)T)NNT)rrrr'r=r rrTr Zargumentrer}r{r~rpropertyrxrrr staticmethodr __classcell__r"r"r-r#r%qs   &   (m/ $    "r%z -+, --cplusz --cplus )Z CYTHON_DOCZCYTHON_INLINE_DOCZCYTHON_PYXIMPORT_DOC)8__doc__ __future__rrrrJrjrrMrsrZ distutils.logrrgetfilesystemencodingrrmZIS_PY2r NameErrorrurrvdistutils.corerrZdistutils.command.build_extrZ IPython.corer r ZIPython.core.magicr r r Z IPython.pathsrZIPython.utils.pathZIPython.utils.textrZShadowrroZCompiler.ErrorsrZInlinerZ Dependenciesrrr$r%rrereplacerTr"r"r"r#sr-             ?