U |e+@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)ZllvmircontextZstrbuferrmsgmodrT/var/www/website-v5/atlas_env/lib/python3.8/site-packages/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)self 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 MemoryErrorLLVMPY_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. globalr;)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_GetModuleNamer8rrrrAuszModuleRef.namecCstj|t|dSr")r rLLVMPY_SetModuleNamer )r&r1rrrrA|scCsttj|S)z8 The module's original source file name )r r rLLVMPY_GetModuleSourceFileNamer8rrr source_fileszModuleRef.source_filec Cs:tjdd$}tj||t|W5QRSQRXdS)zG This module's data layout specification, as a string. FownedN)r rrLLVMPY_GetDataLayoutr+rJrrr data_layoutszModuleRef.data_layoutcCstj|t|ddSNutf8)r rLLVMPY_SetDataLayoutrencoder&ZstrreprrrrSs c Cs:tjdd$}tj||t|W5QRSQRXdS)zK This module's target "triple" specification, as a string. FrPN)r rrLLVMPY_GetTargetr+rJrrrtripleszModuleRef.triplecCstj|t|ddSrT)r rLLVMPY_SetTargetrrWrXrrrrZs 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) r;)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. r;)r rLLVMPY_ModuleFunctionsIter_FunctionsIteratorr?rbrrr functionss zModuleRef.functionscCstj|}t|t|dS)z| Return an iterator over the struct types defined in the module. The iterator will yield a TypeRef. r;)r rLLVMPY_ModuleTypesIter_TypesIteratorr?rbrrr struct_typess zModuleRef.struct_typescCsttj||jSr")rr rLLVMPY_CloneModuler%r8rrrr\szModuleRef.clone)F)__name__ __module__ __qualname____doc__r$r,r5r9rCrFrHrKpropertyrAsetterrOrSrZr_rdrgrjr\ __classcell__rrr(rr/s>               rc@s,eZdZdZddZddZeZddZdS) _IteratorNcCs&tj||||_|jdk s"tdSr")r ObjectRefr$_parentskindr2)r&r3parentsrrrr$sz_Iterator.__init__cCs$|}|rt||j|jStdSr")_nextr rvru StopIterationr&Zvprrr__next__sz_Iterator.__next__cCs|Sr"rr8rrr__iter__sz_Iterator.__iter__)rlrmrnrvr$r{nextr|rrrrrss rsc@s eZdZdZddZddZdS)rarDcCs|j|dSr")r6LLVMPY_DisposeGlobalsIterr8rrrr9sz_GlobalsIterator._disposecCs tj|Sr")r rLLVMPY_GlobalsIterNextr8rrrrxsz_GlobalsIterator._nextNrlrmrnrvr9rxrrrrrasrac@s eZdZdZddZddZdS)rfr:cCs|j|dSr")r6LLVMPY_DisposeFunctionsIterr8rrrr9sz_FunctionsIterator._disposecCs tj|Sr")r rLLVMPY_FunctionsIterNextr8rrrrxsz_FunctionsIterator._nextNrrrrrrfsrfc@s,eZdZdZddZddZddZeZdS) ritypecCs|j|dSr")r6LLVMPY_DisposeTypesIterr8rrrr9sz_TypesIterator._disposecCs|}|rt|StdSr")rxrryrzrrrr{sz_TypesIterator.__next__cCs tj|Sr")r rLLVMPY_TypesIterNextr8rrrrx sz_TypesIterator._nextN)rlrmrnrvr9r{rxr}rrrrris ri)N)N)@ctypesrrrrrrrllvmlite.bindingr Zllvmlite.binding.linkerr llvmlite.binding.commonr r Zllvmlite.binding.valuer rZllvmlite.binding.contextrrr!rtrrsrarfriLLVMContextRefrrargtypes LLVMModuleRefrestyperr7r*r.r= LLVMValueRefrIrRrVrYr[rErG LLVMTypeRefr`LLVMGlobalsIteratorr~rreLLVMFunctionsIteratorrhLLVMTypesIteratorrrrrrkrLrMrNrrrrs$     %