\ K]c@`sddlmZmZmZddlZddlZddlZddlmZddl m Z ddl j j ZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZeje Z!d gZ"ej#j$ej%Z&ej#j'ej(dZ)ej#j'ej(dZ*ej#j'ej(dZ+dZ,de-fdYZ.de-fdYZ/de/fdYZ0de/fdYZ1ere1ne0Z2dS(i(tprint_functiontdivisiontabsolute_importN(tir(tLinkage(tcgutils(tIS_PY3i(t llvm_types(t compile_extratFlags(tglobal_compiler_lock(t cpu_target(t nrtdynmodtCompilericC`s2ddl}ddl}|jdt|dS(Nis #include #ifndef HAVE_LONGDOUBLE #define HAVE_LONGDOUBLE %d #endif typedef struct { float real; float imag; } complex64; typedef struct { double real; double imag; } complex128; #if HAVE_LONGDOUBLE typedef struct { long double real; long double imag; } complex256; #endif typedef float float32; typedef double float64; #if HAVE_LONGDOUBLE typedef long double float128; #endif t complex256(tnumpyttextwraptdedentthasattr(RR((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt get_header s  t ExportEntrycB`s eZdZdZdZRS(s0 A simple record for exporting symbols. cC`s||_||_||_dS(N(tsymbolt signaturetfunction(tselfRRR((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt__init__Is  cC`sd|j|jfS(NsExportEntry(%r, %r)(RR(R((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt__repr__Ns(t__name__t __module__t__doc__RR(((s2lib/python2.7/site-packages/numba/pycc/compiler.pyRDs t_ModuleCompilercB`seZdZejjejejej ejfZ ejj e Z ejjejej fZ ejj e ZedZdZdZedZedZedZdZdZd Zd Zd Zd ZRS( sA base class to compile Python modules to a single shared library or extension module. :param export_entries: a list of ExportEntry instances. :param module_name: the name of the exported module. cK`sa||_t|_g|_||_d|_||_tj |_ tj j |j||_ dS(N( t module_nametFalsetexport_python_wrapt dll_exportstexport_entriestNonetexternal_init_functiontuse_nrtR ttyping_contextttarget_contexttwith_aot_codegentcontext(RR#RR&t aot_options((s2lib/python2.7/site-packages/numba/pycc/compiler.pyRls        cC`s d|fS(Ns._pycc_method_%s((Rt func_name((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt_mangle_method_symbolzscC`s tdS(sAEmit generated Python wrapper and extension module code. N(tNotImplementedError(Rt llvm_module((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt_emit_python_wrapper}sc C`si|_i|_i|_|jj}|j|j}t}|jd|j sk|jdn|j r|jdt j |j\}}|j |nx5|jD]*}t|j|j|j|jj|jj|did|}|jj}|jj|} |j rtj| _|jj} |jj| } |j|j| _tj | _|j!j"j#|jj$|jj%} | |j|<|j&|j|<|jj'|j|ssextern %s %s(%s); ( tostpathtsplitextReRfRR#RRyRRBtjoinRA( RRhtfnametextRkt export_entryRJRNRA((Rs2lib/python2.7/site-packages/numba/pycc/compiler.pyt emit_headers  cC`s2g}x|jD]}|j}|j|}|j|}|j|d|}|jj||}tjj |tjj |t j t tf} |j| qWtjj ttttg} |j| tjj|j|} |j| jd} | | _tj| _tjj| ttg} | S(s Collect exported methods and emit a PyMethodDef array. :returns: a pointer to the PyMethodDef array. RJs.module_methods(R#RR-R9t add_functionR*tinsert_const_stringRFtConstanttstructtbitcasttltt _void_startMETH_VARARGS_AND_KEYWORDStNULLRRtZEROtarrayt method_def_tytadd_global_variablettypet initializerRGRHtgep(RR/t method_defsR\RJRDR`tlfunct method_nametmethod_def_consttsentineltmethod_array_initt method_arraytmethod_array_ptr((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt_emit_method_arrays*         c C`sg}x=|jD]2}|j|}|j|j}|j|qWtjj|j|}|j j |d|} | j t t gS(s Emit an array of env_def_t structures (see modulemixin.c) storing the pickled environment constants for each of the exported functions. s.module_environments( R#R:tserialize_uncachedtconstsRRRFRRt env_def_tyR*tinsert_unique_constRR( RR/tbuildertpyapitenv_defsR\tenvtenv_deft env_defs_inittgv((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt_emit_environment_arrays   c C`sg}xR|jD]G}|j|}|jj||}|jtj}|j|qWtj j tj|} |jj |d| }|j t t gS(sj Emit an array of Environment pointers that needs to be filled at initialization. s.module_envgvs(R#R;R*tdeclare_env_globalRRRRRRFRRRRR( RR/RRt env_settersR\t envgv_nameRtenvgvtenv_setters_init((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt_emit_envgvs_arrays   c C`so|jrgtjtj|j|j|j|jg}|j||j}|j |||||gSdSdS(s@ Emit call to "external" init function, if any. N( R%Rt FunctionTypeRt_int32Rtmethod_def_ptrt env_def_ptrRtcallR$( RR/RtmodobjRt env_arrayt envgv_arrayR`Rb((s2lib/python2.7/site-packages/numba/pycc/compiler.pyt_emit_module_init_code#s    (RRRRFtTypeRRt _int8_starRRRtpointerRRRR RR-R0R RcRlRnRyRRRRR(((s2lib/python2.7/site-packages/numba/pycc/compiler.pyRRs(    @      tModuleCompilerPy2cB`s/eZedZedZdZRS(cC`sxtjjtjtj|jtjtjtjf}d}tjrR|d7}ntj dkrn|d7}n||fS(sPReturn the signature and name of the function to initialize the module. tPy_InitModule4t TraceRefsi@t_64( RFRRRt_pyobject_head_pRRRt _trace_refs_t _plat_bits(RRRJ((s2lib/python2.7/site-packages/numba/pycc/compiler.pytmodule_create_definition5s   cC`s)tjjtjjdd|jfS(sSReturn the signature and name of the function to initialize the extension. tinit((RFRRtvoidR(R((s2lib/python2.7/site-packages/numba/pycc/compiler.pytmodule_init_definitionIsc C`s3|j|j}|jd}tj|}|jj|}|j|j}tj|_ |jj ||j }|j |}|j |||ttjjtjtjjtjtjf} |j|||} |j|||} |j||| || | |j|jj|jdS(NtEntry(RRtappend_basic_blockRFtBuilderR*tget_python_apiRRKRHRRRRRRtnullRRRoRtsyst api_versionRRRtret_voidR"RRRJ( RR/t mod_init_fnR\RRtcreate_module_fntmod_name_constRtmodRR((s2lib/python2.7/site-packages/numba/pycc/compiler.pyR0Os*    !  (RRtpropertyRRR0(((s2lib/python2.7/site-packages/numba/pycc/compiler.pyR3stModuleCompilerPy3c B`seZdZeejejZeejejZeejejeejZ eejejZ eejZ ej Z ejjeje ejejfZejjee e ejejee ee f ZedZedZdZRS(cG`stjjtjj||S(N(RFRRR(tretRA((s2lib/python2.7/site-packages/numba/pycc/compiler.pytvtcC`sVtjjtjtjj|jtjf}d}tjrL|d7}n||fS(sn Return the signature and name of the Python C API function to initialize the module. tPyModule_Create2R( RFRRRRRt module_def_tyRR(RRRJ((s2lib/python2.7/site-packages/numba/pycc/compiler.pyRs  cC`s)tjjtjd}|d|jfS(sX Return the name and signature of the module's initialization function. tPyInit_((RFRRRRR(RR((s2lib/python2.7/site-packages/numba/pycc/compiler.pyRsc C`s|j|j}tj|_|jj||j}tjj t j tjj |j tjj t jtjj t jf}|j|jd}||_tj|_|j|}tjj ||tjj |jtjjt jd|tjj |jtjj |jtjj |jtjj |jf }|j|jd}||_tj|_|j|j} | jd} tj| } |jj| } | j||tjjt jt j!f} | j"t#j$| | $| j%t&j'| jj(j)WdQX|j*|| | }|j+|| | }|j,|| | |||}|dk r| j"t#j.| |!| j%tjj | jWdQXn| j%| |j/j0| j1dS(Ns.module_def_baseis .module_defR(2RRRFRKRHR*RRRRRt_pyobject_head_initRt m_init_tyt_llvm_py_ssize_tRRRRRGRt _char_starRot inquiry_tyttraverseproc_tyt freefunc_tyRRRRRRRRtif_thenRtis_nullRRRtpointeeRBRRRR$t is_not_nullR"RRRJ(RR/RRtmod_def_base_initt mod_def_baseRt mod_def_inittmod_defRR\RRRRRR((s2lib/python2.7/site-packages/numba/pycc/compiler.pyR0s\              !% % (RRt_ptr_funRt_int8Rt visitproc_tyRRRRRRRRFRRt_pyobject_headRtmodule_def_base_tyRRRRRRR0(((s2lib/python2.7/site-packages/numba/pycc/compiler.pyRts:             i(3t __future__RRRtloggingR}RtllvmliteRtllvmlite.bindingRtllvmlite.llvmpy.coretllvmpytcoreRFtnumbaRt numba.utilsRRRRtnumba.compilerRR tnumba.compiler_lockR tnumba.targets.registryR t numba.runtimeR t getLoggerRtloggert__all__RRRRRoRRtONERRtobjectRRRRtModuleCompiler(((s2lib/python2.7/site-packages/numba/pycc/compiler.pyts2     $A