o ÉÙ¾iÒ#ã@srdZddlmZddlZddlZddlmZmZmZm Z m Z m Z e dkr&e ne Zz7e dkr0e ƒ‚ejdkrJdd lmZmZdd lmZdd lmZndd lmZmZdd lmZdd lmZWney|dd lmZmZdd lmZdd lmZYnwejZGd d „d eƒZGdd„deƒZd!dd„Zd"dd„Zdd„Zdd„Z dd„Z!d#dd„Z"dd„Z#dd „Z$eeƒdS)$zFunction namespace.é)Úabsolute_importNé)Ú_LIBÚ check_callÚpy_strÚc_strÚ string_typesÚ _FFI_MODEÚcythonÚctypes)ér)Ú_set_class_functionÚ_set_class_module)Ú FunctionBase)Úconvert_to_decord_funcc@seZdZdZdS)ÚFunctionaRThe PackedFunc object. Function plays an key role to bridge front and backend in DECORD. Function provide a type-erased interface, you can call function with positional arguments. The compiled module returns Function. DECORD backend also registers and exposes its API as Functions. For example, the developer function exposed in decord.ir_pass are actually C++ functions that are registered as PackedFunc The following are list of common usage scenario of decord.Function. - Automatic exposure of C++ API into python - To call PackedFunc from python side - To call python callbacks to inspect results in generated code - Bring python hook into C++ backend See Also -------- decord.register_func: How to register global function. decord.get_global_func: How to get global function. N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©rrúH/home/ubuntu/.local/lib/python3.10/site-packages/decord/_ffi/function.pyrsrc@sVeZdZdZgd¢Zdd„Zdd„Zedd„ƒZdd d „Z d d „Z dd„Z dd„Z dS)Ú ModuleBasezBase class for module©ÚhandleÚ_entryÚ entry_namecCs||_d|_d|_dS)NÚ__decord_main__r)ÚselfrrrrÚ__init__=s zModuleBase.__init__cCstt |j¡ƒdS©N)rrÚ DECORDModFreer©rrrrÚ__del__BszModuleBase.__del__cCs |jr|jS| |j¡|_|jS)z}Get the entry function Returns ------- f : Function The entry function if exist )rÚ get_functionrr"rrrÚ entry_funcEs zModuleBase.entry_funcFc CsHtƒ}tt |jt|ƒt |¡t |¡¡ƒ|j st d|ƒ‚t |dƒS)a:Get function from the module. Parameters ---------- name : str The name of the function query_imports : bool Whether also query modules imported by this module. Returns ------- f : Function The result function. zModule has no function '%s'F) ÚFunctionHandlerrÚDECORDModGetFunctionrrr Úc_intÚbyrefÚvalueÚAttributeErrorr)rÚnameÚ query_importsÚ ret_handlerrrr$Ss ýÿ zModuleBase.get_functioncCstt |j|j¡ƒdS)z“Add module to the import list of current one. Parameters ---------- module : Module The other module. N)rrÚDECORDModImportr)rÚmodulerrrÚ import_modulemszModuleBase.import_modulecCst|tƒs tdƒ‚| |¡S)Nz%Can only take string as function name)Ú isinstancerÚ ValueErrorr$)rr,rrrÚ __getitem__ws  zModuleBase.__getitem__cGs|jr|j|ŽS|j}||ŽSr )rr%)rÚargsÚfrrrÚ__call__|s zModuleBase.__call__N©F) rrrrÚ __slots__rr#Úpropertyr%r$r1r4r7rrrrr9s   rFcsLtˆƒr ˆ}|j‰tˆtƒstdƒ‚t |¡‰‡‡fdd„}|r$||ƒS|S)aúRegister global function Parameters ---------- func_name : str or function The function name f : function, optional The function to be registered. override: boolean optional Whether override existing entry. Returns ------- fregister : function Register function if f is not specified. Examples -------- The following code registers my_packed_func as global function. Note that we simply get it back from global function table to invoke it from python side. However, we can also invoke the same function from C++ backend, or in the compiled DECORD code. .. code-block:: python targs = (10, 10.0, "hello") @decord.register_func def my_packed_func(*args): assert(tuple(args) == targs) return 10 # Get it out from global function table f = decord.get_global_func("my_packed_func") assert isinstance(f, decord.nd.Function) y = f(*targs) assert y == 10 zexpect string function namecs.t|tƒs t|ƒ}tt tˆƒ|jˆ¡ƒ|S)zinternal register function)r2rrrrÚDECORDFuncRegisterGlobalrr)Úmyf©Ú func_nameÚ ioverriderrÚregister²s  ÿzregister_func..register)Úcallablerr2Ústrr3r r()r>r6Úoverrider@rr=rÚ register_funcƒs'  rDcCsDtƒ}tt t|ƒt |¡¡ƒ|jrt|dƒS|rdSt d|ƒ‚)aBGet a global function by name Parameters ---------- name : str The name of the global function allow_missing : bool Whether allow missing function or raise an error. Returns ------- func : decord.Function The function to be returned, None if function is missing. FNzCannot find global function %s) r&rrÚDECORDFuncGetGlobalrr r)r*rr3)r,Ú allow_missingrrrrÚget_global_func¾s  rGcCs\t tj¡ƒ}t ¡}tt t |¡t |¡¡ƒg}t|j ƒD] }|  t ||ƒ¡q |S)z~Get list of global functions registered. Returns ------- names : list List of global functions names. ) r ÚPOINTERÚc_char_pÚc_uintrrÚDECORDFuncListGlobalNamesr)Úranger*Úappendr)ÚplistÚsizeÚfnamesÚirrrÚlist_global_func_namesÚsÿrRcs>i‰‡fdd„}t|ƒ}||jƒ}|}|dkrtd|ƒ‚ˆS)a Extract the extension PackedFuncs from a C module. Parameters ---------- finit : ctypes function a ctypes that takes signature of DECORDExtensionDeclarer Returns ------- fdict : dict of str to Function The extracted functions cs |ˆ|<dSr r)r,Úfunc©ÚfdictrrÚ_listüs z extract_ext_funcs.._listrzcannot initialize with %s)rrÚ RuntimeError)ÚfinitrVr<ÚretÚ_rrTrÚextract_ext_funcsís   r[cCs|}d|_|S)NT)Ú is_global)r6ÚflocalrrrÚ_get_apisr^cCs:|r|n|}| d¡rt||dd…ƒdSt||ƒdS)zÄInitialize api for a given module name namespace : str The namespace of the source registry target_module_name : str The target module name if different from namespace zdecord.éN)Ú startswithÚ_init_api_prefix)Ú namespaceÚtarget_module_namerrrÚ _init_api s ÿ rdcCs”tj|}tƒD]?}| d¡rq| |¡sq|t|ƒdd…}|}| d¡dkr0td|ƒqt|ƒ}t|ƒ}||_ d||_ t ||j |ƒqdS)NrZrÚ.éÿÿÿÿúWarning: invalid API name "%s".úDECORD PackedFunc %s. ) ÚsysÚmodulesrRr`ÚlenÚfindÚprintrGr^rrÚsetattr)Ú module_nameÚprefixr0r,ÚfnameÚ target_moduler6Úffrrrras"      ñracCsttƒD]4}| d¡s qtjd}|}| d¡dkr td|ƒqt|ƒ}t|ƒ}||_d||_ t ||j|ƒqdS)NrZzdecord._api_internalrerfrgrh) rRr`rirjrlrmrGr^rrrn)r,rrrqr6rsrrrÚ_init_internal_api/s     ôrt)NFr8r )%rÚ __future__rrir Úbaserrrrrr rWÚ ImportErrorÚ IMPORT_EXCEPTÚ version_infoÚ _cy3.corer rrÚ _FunctionBaserÚ _cy2.coreÚ_ctypes.functionÚc_void_pr&rÚobjectrrDrGrRr[r^rdrartrrrrÚsD      €  ü J ;