U +d+@s0ddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z mZddlmZmZddlmZddd Zdd d ZGd d d e jZGddde jZGdddeZGdddeZGdddeZe jeeege jj_e j e jj_!e jeeeege jj"_e j e jj"_!e j ge jj#_e j eege jj$_e j eeeege jj%_e j ege jj&_e j'e jj&_!e j eege jj(_ee jj(_!e j eege jj)_e j ege jj*_e j eege jj+_e j ege jj,_e j ege jj-_e j'e jj-_!e j ege jj._e j/e jj._!e j ge jj0_e j1e jj0_!e j1ge jj2_e j1ge jj3_e j'e jj3_!e j ge jj4_e j5e jj4_!e j ge jj6_e j7e jj6_!e j5ge jj8_e j7ge jj9_e j5ge jj:_e j'e jj:_!e j7ge jj;_e j/e jj;_!e j ge jj<_e j e jj<_!e j ge jj=_ee jj=_!e j ege jj>_e j ge jj?_ee jj?_!dS))c_char_pbyrefPOINTERc_boolcreate_string_bufferc_size_t string_at)ffi) link_modules)_decode_string_encode_string)ValueRefTypeRef)get_global_contextNc Csf|dkrt}t|}t|}t6}ttj||||}|rX|t d |W5QRX|S)z- Create Module from a LLVM IR string NzLLVM IR parsing error {0}) rr rr OutputString ModuleReflibLLVMPY_ParseAssemblyclose RuntimeErrorformat)ZllvmircontextZstrbuferrmsgmodr;/tmp/pip-unpacked-wheel-stw2luzp/llvmlite/binding/module.pyparse_assembly s rc Csh|dkrt}t|}t|}t8}ttj|||||}|rZ|t d |W5QRX|S)z? Create Module from a LLVM *bitcode* (a bytes object). NzLLVM bitcode parsing error {0}) rrlenr rrrLLVMPY_ParseBitcoderrr)Zbitcoderbufbufsizerrrrr parse_bitcodes& r!cseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ e ddZ e jddZ e ddZe ddZejddZe ddZejddZd(ddZe d d!Ze d"d#Ze d$d%Zd&d'ZZS))rz' A reference to a LLVM module. cstt||||_dSN)superr__init___context)selfZ module_ptrr __class__rrr$4szModuleRef.__init__c Cs6t$}tj||t|W5QRSQRXdSr")r rrLLVMPY_PrintModuleToStringstr)r&Zoutstrrrr__str__8s zModuleRef.__str__c Csbtd}td}tj|t|t||s0tz|jdks@t t ||jWStj|XdS)zF Return the module's LLVM bitcode, as a bytes object. Nr) rrr rLLVMPY_WriteBitcodeToStringr MemoryErrorZLLVMPY_DisposeStringvalueAssertionErrorr)r&ptrsizerrr as_bitcode=szModuleRef.as_bitcodecCs|j|dSr")_capiLLVMPY_DisposeModuler&rrr_disposeLszModuleRef._disposecCs0tj|t|}|st|t|dt|dS)z~ Get a ValueRef pointing to the function named *name*. NameError is raised if the symbol isn't found. functionmodule)r rLLVMPY_GetNamedFunctionr NameErrorr dictr&nameprrr get_functionOszModuleRef.get_functioncCs0tj|t|}|st|t|dt|dS)z Get a ValueRef pointing to the global variable named *name*. NameError is raised if the symbol isn't found. globalr9)r rLLVMPY_GetNamedGlobalVariabler r<r r=r>rrrget_global_variableYszModuleRef.get_global_variablecCs&tj|t|}|st|t|S)z Get a TypeRef pointing to a structure type named *name*. NameError is raised if the struct type isn't found. )r rLLVMPY_GetNamedStructTyper r<rr>rrrget_struct_typecszModuleRef.get_struct_typec Cs2t }tj||r$tt|W5QRXdS)zW Verify the module IR's correctness. RuntimeError is raised on error. N)r rrLLVMPY_VerifyModulerr*r&Zoutmsgrrrverifyms zModuleRef.verifycCsttj|S)z* The module's identifier. )r r rLLVMPY_GetModuleNamer6rrrr?uszModuleRef.namecCstj|t|dSr")r rLLVMPY_SetModuleNamer )r&r/rrrr?|scCsttj|S)z8 The module's original source file name )r r rLLVMPY_GetModuleSourceFileNamer6rrr source_fileszModuleRef.source_filec Cs:tjdd$}tj||t|W5QRSQRXdS)zG This module's data layout specification, as a string. FZownedN)r rrLLVMPY_GetDataLayoutr*rHrrr data_layoutszModuleRef.data_layoutcCstj|t|ddSNutf8)r rLLVMPY_SetDataLayoutrencoder&ZstrreprrrrPs c Cs:tjdd$}tj||t|W5QRSQRXdS)zK This module's target "triple" specification, as a string. FrNN)r rrLLVMPY_GetTargetr*rHrrrtripleszModuleRef.triplecCstj|t|ddSrQ)r rLLVMPY_SetTargetrrTrUrrrrWs FcCs|r |}t||dS)z Link the *other* module into this one. The *other* module will be destroyed unless *preserve* is true. N)cloner )r&otherpreserverrrlink_inszModuleRef.link_incCstj|}t|t|dS)a) Return an iterator over this module's global variables. The iterator will yield a ValueRef for each global variable. Note that global variables don't include functions (a function is a "global value" but not a "global variable" in LLVM parlance) r9)r rLLVMPY_ModuleGlobalsIter_GlobalsIteratorr=r&itrrrglobal_variabless zModuleRef.global_variablescCstj|}t|t|dS)z Return an iterator over this module's functions. The iterator will yield a ValueRef for each function. r9)r rLLVMPY_ModuleFunctionsIter_FunctionsIteratorr=r_rrr functionss zModuleRef.functionscCstj|}t|t|dS)z| Return an iterator over the struct types defined in the module. The iterator will yield a TypeRef. r9)r rLLVMPY_ModuleTypesIter_TypesIteratorr=r_rrr struct_typess zModuleRef.struct_typescCsttj||jSr")rr rLLVMPY_CloneModuler%r6rrrrYszModuleRef.clone)F)__name__ __module__ __qualname____doc__r$r+r3r7rArDrFrIpropertyr?setterrMrPrWr\rardrgrY __classcell__rrr'rr/s>               rc@s,eZdZdZddZddZeZddZdS) _IteratorNcCs&tj||||_|jdk s"tdSr")r ObjectRefr$_parentskindr0)r&r1parentsrrrr$sz_Iterator.__init__cCs$|}|rt||j|jStdSr")_nextr rsrr StopIterationr&Zvprrr__next__sz_Iterator.__next__cCs|Sr"rr6rrr__iter__sz_Iterator.__iter__)rirjrkrsr$rxnextryrrrrrps rpc@s eZdZdZddZddZdS)r^rBcCs|j|dSr")r4LLVMPY_DisposeGlobalsIterr6rrrr7sz_GlobalsIterator._disposecCs tj|Sr")r rLLVMPY_GlobalsIterNextr6rrrrusz_GlobalsIterator._nextNrirjrkrsr7rurrrrr^sr^c@s eZdZdZddZddZdS)rcr8cCs|j|dSr")r4LLVMPY_DisposeFunctionsIterr6rrrr7sz_FunctionsIterator._disposecCs tj|Sr")r rLLVMPY_FunctionsIterNextr6rrrrusz_FunctionsIterator._nextNr}rrrrrcsrcc@s,eZdZdZddZddZddZeZdS) rftypecCs|j|dSr")r4LLVMPY_DisposeTypesIterr6rrrr7sz_TypesIterator._disposecCs|}|rt|StdSr")rurrvrwrrrrxsz_TypesIterator.__next__cCs tj|Sr")r rLLVMPY_TypesIterNextr6rrrru sz_TypesIterator._nextN)rirjrkrsr7rxrurzrrrrrfs rf)N)N)@ctypesrrrrrrrZllvmlite.bindingr Zllvmlite.binding.linkerr Zllvmlite.binding.commonr r Zllvmlite.binding.valuer rZllvmlite.binding.contextrrr!rqrrpr^rcrfZLLVMContextRefrrargtypesZ LLVMModuleRefrestyperr5r)r-r;Z LLVMValueRefrGrOrSrVrXrCrEZ LLVMTypeRefr]ZLLVMGlobalsIteratorr{r|rbZLLVMFunctionsIteratorreZLLVMTypesIteratorr~rrrrhrJrKrLrrrrs$     %