B Ꮚ\)@s(ddlmZmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZmZddlmZmZddlmZdd d Zdd d ZGddde jZGddde jZGdddeZGdddeZGdddeZe jeeege j j!_"e j#e j j!_$e jee eege j j%_"e j#e j j%_$e j#ge j j&_"e j#eege j j'_"e j#eeee ge j j(_"e j#ege j j)_"e j*e j j)_$e j#eege j j+_"ee j j+_$e j#eege j j,_"e j#ege j j-_"e j#eege j j._"e j#ege j j/_"e j#ege j j0_"e j*e j j0_$e j#ege j j1_"e j2e j j1_$e j#ge j j3_"e j4e j j3_$e j4ge j j5_"e j4ge j j6_"e j*e j j6_$e j#ge j j7_"e j8e j j7_$e j#ge j j9_"e j:e j j9_$e j8ge j j;_"e j:ge j j<_"e j8ge j j=_"e j*e j j=_$e j:ge j j>_"e j2e j j>_$e j#ge j j?_"e j#e j j?_$e j#ge j j@_"ee j j@_$e j#ege j jA_"d S))print_functionabsolute_import)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 |WdQRX|S)z- Create Module from a LLVM IR string NzLLVM IR parsing error {0}) rrrr OutputString ModuleReflibLLVMPY_ParseAssemblyclose RuntimeErrorformat)ZllvmircontextZstrbuferrmsgmodr6lib/python3.7/site-packages/llvmlite/binding/module.pyparse_assembly s rc Csh|dkrt}t|}t|}t8}ttj|||||}|rZ|t d |WdQRX|S)z? Create Module from a LLVM *bitcode* (a bytes object). NzLLVM bitcode parsing error {0}) rrlenr rrrLLVMPY_ParseBitcoderrr)ZbitcoderZbufbufsizerrrrr parse_bitcodes r#cseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ e ddZ e jddZ e ddZejddZe ddZejddZd&ddZe ddZe d d!Ze d"d#Zd$d%ZZS)'rz' A reference to a LLVM module. cstt||||_dS)N)superr__init___context)selfZ module_ptrr) __class__rrr%5szModuleRef.__init__c Cs*t}tj||t|SQRXdS)N)r rrLLVMPY_PrintModuleToStringstr)r'Zoutstrrrr__str__9s zModuleRef.__str__c Cs^td}td}tj|t|t||s0tz|jdks@tt ||jStj |XdS)zF Return the module's LLVM bitcode, as a bytes object. Nr) rr r rLLVMPY_WriteBitcodeToStringr MemoryErrorvalueAssertionErrorr ZLLVMPY_DisposeString)r'ptrsizerrr as_bitcode>s zModuleRef.as_bitcodecCs|j|dS)N)_capiLLVMPY_DisposeModule)r'rrr_disposeMszModuleRef._disposecCs*tj|t|}|st|t||dS)z~ Get a ValueRef pointing to the function named *name*. NameError is raised if the symbol isn't found. )module)r rLLVMPY_GetNamedFunctionr NameErrorr)r'nameprrr get_functionPszModuleRef.get_functioncCs*tj|t|}|st|t||dS)z Get a ValueRef pointing to the global variable named *name*. NameError is raised if the symbol isn't found. )r7)r rLLVMPY_GetNamedGlobalVariablerr9r)r'r:r;rrrget_global_variableZszModuleRef.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_GetNamedStructTyperr9r)r'r:r;rrrget_struct_typedszModuleRef.get_struct_typec Cs2t }tj||r$tt|WdQRXdS)zW Verify the module IR's correctness. RuntimeError is raised on error. N)r rrLLVMPY_VerifyModulerr*)r'outmsgrrrverifyns zModuleRef.verifycCsttj|S)z* The module's identifier. )rr rLLVMPY_GetModuleName)r'rrrr:vszModuleRef.namecCstj|t|dS)N)r rLLVMPY_SetModuleNamer)r'r/rrrr:}sc Cs.tjdd}tj||t|SQRXdS)zG This module's data layout specification, as a string. F)ownedN)r rrLLVMPY_GetDataLayoutr*)r'rBrrr data_layoutszModuleRef.data_layoutcCstj|t|ddS)Nutf8)r rLLVMPY_SetDataLayoutrencode)r'strreprrrrHsc Cs.tjdd}tj||t|SQRXdS)zK This module's target "triple" specification, as a string. F)rFN)r rrLLVMPY_GetTargetr*)r'rBrrrtripleszModuleRef.triplecCstj|t|ddS)NrI)r rLLVMPY_SetTargetrrK)r'rLrrrrNsFcCs|r |}t||dS)z Link the *other* module into this one. The *other* module will be destroyed unless *preserve* is true. N)cloner )r'otherZpreserverrrlink_inszModuleRef.link_incCstj|}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) )r7)r rLLVMPY_ModuleGlobalsIter_GlobalsIterator)r'itrrrglobal_variabless zModuleRef.global_variablescCstj|}t||dS)z Return an iterator over this module's functions. The iterator will yield a ValueRef for each function. )r7)r rLLVMPY_ModuleFunctionsIter_FunctionsIterator)r'rUrrr functionss zModuleRef.functionscCstj|}t||dS)z| Return an iterator over the struct types defined in the module. The iterator will yield a TypeRef. )r7)r rLLVMPY_ModuleTypesIter_TypesIterator)r'rUrrr struct_typess zModuleRef.struct_typescCsttj||jS)N)rr rLLVMPY_CloneModuler&)r'rrrrPszModuleRef.clone)F)__name__ __module__ __qualname____doc__r%r+r3r6r<r>r@rCpropertyr:setterrHrNrRrVrYr\rP __classcell__rr)r(rr0s(        rc@s(eZdZddZddZeZddZdS) _IteratorcCstj||||_dS)N)r ObjectRefr%_module)r'r1r7rrrr%sz_Iterator.__init__cCs |}|rt||jStdS)N)_nextrrg StopIteration)r'vprrr__next__s z_Iterator.__next__cCs|S)Nr)r'rrr__iter__sz_Iterator.__iter__N)r^r_r`r%rknextrlrrrrresrec@seZdZddZddZdS)rTcCs|j|dS)N)r4LLVMPY_DisposeGlobalsIter)r'rrrr6sz_GlobalsIterator._disposecCs tj|S)N)r rLLVMPY_GlobalsIterNext)r'rrrrhsz_GlobalsIterator._nextN)r^r_r`r6rhrrrrrTsrTc@seZdZddZddZdS)rXcCs|j|dS)N)r4LLVMPY_DisposeFunctionsIter)r'rrrr6sz_FunctionsIterator._disposecCs tj|S)N)r rLLVMPY_FunctionsIterNext)r'rrrrhsz_FunctionsIterator._nextN)r^r_r`r6rhrrrrrXsrXc@s(eZdZddZddZddZeZdS)r[cCs|j|dS)N)r4LLVMPY_DisposeTypesIter)r'rrrr6sz_TypesIterator._disposecCs|}|rt|StdS)N)rhrri)r'rjrrrrksz_TypesIterator.__next__cCs tj|S)N)r rLLVMPY_TypesIterNext)r'rrrrhsz_TypesIterator._nextN)r^r_r`r6rkrhrmrrrrr[sr[)N)N)BZ __future__rrZctypesrrrrrr r r Zlinkerr commonrrr/rrrrrr#rfrrerTrXr[ZLLVMContextRefrrZargtypesZ LLVMModuleRefZrestyper!r5r)r-r8Z LLVMValueRefrArGrJrMrOr=r?Z LLVMTypeRefrSZLLVMGlobalsIteratorrnrorWZLLVMFunctionsIteratorrZZLLVMTypesIteratorrprrrqrsr]rDrErrrrs|$