`i7 dZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m ZddlmZmZdd lmZmZmZmZdd lmZmZmZmZd"d Zd#dZd$dZd%dZd&d Z e!d!krej"edSdS)'z-TVM-FFI Stub Generator (``tvm-ffi-stubgen``).) annotationsNPath)codegen)consts)FileInfo collect_files)collect_global_funcscollect_type_keysobject_info_from_type_keytoposort_objects)FuncInfo ImportItem InitConfigOptionsreturnintc &t}|jpgD]}tj|d|jD}t d|jD}t}d}|jrGt|jd }| r|j }tj }|D]f} t||#t $rFt#tjd|jdt)jtjYcwxYw|jr%| Jdt1|||j|||D]}|jr/t#tjd |jtj t7||||L#t $rFt#tjd|jdt)jtjYwxYw~dS) zCommand line entry point for ``tvm-ffi-stubgen``. This generates in-place type stubs inside special ``tvm-ffi-stubgen`` blocks in the given files or directories. See the module docstring for an overview and examples of the block syntax. c6g|]}tj|S)ctypesCDLL).0libs d/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/tvm_ffi/stub/cli.py z__main__..2s" 1 1 1FK   1 1 1c,g|]}t|Srr)rfs rrz__main__..3s*F*F*Fq477*F*F*FrNrz[Failed] File "z": z!init-path could not be determined)init_cfg init_path global_funcsz[File] ) _parse_argsimports importlib import_moduledllsr filesr rresolveis_fileparentCTY_MAP_DEFAULTScopy_stage_1 ExceptionprintTERM_REDpath traceback format_exc TERM_RESETinit_stage_2verbose TERM_CYAN_stage_3)optimpr(r)r#r"ty_mapfiles r__main__rA(sz --C{ b%%$$$$ 1 1 1 1 1D)*F*FCI*F*F*FGGE.B.D.DL!I y)1&&..00      )!(I .3355F  T6 " " " "    :bbdibbIzInvalid ty_map format at line z. Example: `A.B -> C.D`) code_blockskind isinstanceparamstrsplit ValueError lineno_startstrip)r@r?codelhsrhses rr0r0cs . . 9  !$*c22222:++D11SS    _T5F___ #&))++F399;;  . .s9A A8A33A8r)list[FileInfo]r!rr"rr#dict[str, list[FuncInfo]]c dfd }dD}dDtjz}|j}|r|ds|dz }|d} t } |D]} | | g| D]\} } | | ks| |s"t| |vrgn| | gd } tt| |z }t|}| s|s|| dd z }|d d |dz }||}|dd5}|t%j|j|| ||| | kdddn #1swxYwY||dz }||}|dd5}|t%j|j| ddddn #1swxYwY|dS)Nr4rrr c|st|dg}nHD] }||jr|cS!tj|d}| Jd|||S)Nr)r4linesrET)r@ include_emptyzFailed to read file: )existsr samefiler4 from_fileappend)r4retr@r)s r_find_or_insert_filez&_stage_2.._find_or_insert_filezs{{}} CBB???CC  ==++ KKK $$dCCCC??$BD$B$B??? S rcRh|]$}|jD]}|jdk |jd%S)globalrrErFrHrr@rNs r z_stage_2..sG'''D4D'',0 U]H]H] 1 H]H]H]H]rcFh|]}|jD]}|jdk |jS)objectr`ras rrbz_stage_2..sC1A)-TYRZEZEZ EZEZEZEZr.c|jjS)N)schemaname)r s rz_stage_2..s !(-r)key/T)parentsexist_okz _ffi_api.pyazutf-8)encoding)is_rootz __init__.py_ffi_api) submodule)r4rrr )r-BUILTIN_TYPE_KEYSprefixrMendswithrstripr setdefaultitems startswithsortedgetsetrreplacemkdiropenwriteGgenerate_ffi_apirEreload generate_init)r)r?r!r"r#r]defined_func_prefixes defined_objs prefix_filter root_prefixprefixesrt obj_namesfuncsobjs object_infos directory target_path target_filer s` rr9r9ss      ''"''''$ L O))++M]33C88 &&s++K%6%8%8H((FB''''%^^--"" +%%):):=)I)I% 111BB|7G7GPR7S7S''   c)nn|344'--  \  sC 8 88 t444-/ **;77   cG  4 4  GG"+ "k1                    -/ **;77   cG  4 4 \ GGAOK$;VzZZZ [ [ [ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \E""s$ 7GG G 0II I r=rc  t}tg}d}|jD]}|jdkr|j\}} } |t |t | o,t| to| dk| r| nd| r| dks| drd}|jD]j}|jdkr]| |jd g} | D]!} | | j j"tj|| |||k|jD]}|jd kr|j} t| tsJt#| }| | | } t | j} |tj|||||fd |D}|jD]%}|jd krtj|||n&|jD]2}|jd kr%|z}|r|dhz}tj|||n3|jD]!}|jdkrtj|"||j|jdS)NFz import-objecttrue)type_checking_onlyalias _FFI_LOAD_LIBzlibinfo.load_lib_moduleTr_rrdc&g|] }|jv |Sr) full_name)ri defined_typess rrz_stage_3..s%FFFQQ[ %E%Eq%E%E%Erzimport-section__all__LIBexport)r:dry_run)r|rErFrHr[rboolrGrIlowerrur{addrgrhrgenerate_global_funcsr rgenerate_objectgenerate_import_section generate_allgenerate_exportupdater:r)r@r=r?r# defined_funcsr%ffi_load_lib_importedrNrhrrrfunctype_keyobj_infor export_namesrs @rr<r<s "eeM!eeM "G! -- 9 ' '.2j +D$e NN/00A&'93??A.4466&@#(2%%d    -%?22t}}E^7_7_2(,% GG 9  $$TZ]B77E 4 4!!$+"23333 #D%# F F F DD 9 zHh,, , ,,0::Hzz(H55H"8,,6I   i ( ( ( dFGS( C C CFFFF'FFFG  9( ( ( %dGS 9 9 9 E )  9 ! !(=8L$ 6+ug5 N4s 3 3 3 E " $$ 9 d # # #KK S[K99999rc Gddtjtj}d.d}tjdd tjtjtjd | }|d td dd|dtd dd|dtd d|dtd d|dtd d|dtddtj d|dd d!d"#|d$d%d&'|d(d%d)'| }|j |j|jg}d}t!|rEt#|s|d*t'|j |j|j+}|js(|t-jd,t1||j||j||j|j|j|j-S)/NceZdZdS)"_parse_args..HelpFormatterN)__name__ __module__ __qualname__rrr HelpFormatterrs rrarg str | Noner list[str]cH|sgSd|dDS)Nc^g|]*}||+Sr)rM)ritems rrz8_parse_args.._split_list_arg..s-HHH4::<<H HHHr;)rJ)rs r_split_list_argz$_parse_args.._split_list_args- IHH3HHHHrztvm-ffi-stubgenaGenerate type stubs for TVM FFI extensions. It supports two modes - In `--init-*` mode, it generates missing `_ffi_api.py` and `__init__.py` files, based on the registered global functions and object types in the loaded libraries. - In normal mode, it processes the given files/directories in-place, generating type stubs inside special `tvm-ffi-stubgen` directive blocks. Documentation: re)prog descriptionformatter_classz --importsIMPORTSz\Additional imports to load before generation, separated by ';' (e.g. 'pkgA;pkgB.submodule').)typedefaultmetavarhelpz--dllsLIBSzShared libraries to preload before generation (e.g. TVM runtime or your extension), separated by ';'. This ensures global function and object metadata is available. Platform-specific suffixes like .so/.dylib/.dll are supported.z --init-pypkgzuPython package name to generate stubs for (e.g. apache-tvm-ffi). Required together with --init-lib and --init-prefix.)rrrz --init-libzCMake target that produces the shared library to load for stub generation (e.g. tvm_ffi_shared). Required together with --init-pypkg and --init-prefix.z --init-prefixzGlobal function/object prefix to include when generating stubs (e.g. tvm_ffi.). Required together with --init-pypkg and --init-lib.z--indentzbExtra spaces added inside each generated block, relative to the indentation of the corresponding 'z' line.r)*PATHzFiles or directories to process. Directories are scanned recursively; only .py and .pyi files are modified. Use tvm-ffi-stubgen directives to select where stubs are generated.)nargsrrz --verbose store_truezvPrint a unified diff of changes to each file. This is useful for debugging or previewing changes before applying them.)actionrz --dry-runz`Don't write changes to files. This is useful for previewing changes without modifying any files.zE--init-pypkg, --init-lib, and --init-prefix must be provided together)pkg shared_targetrtr)r%r(r8indentr)r:r)rrrr)argparseArgumentDefaultsHelpFormatterRawTextHelpFormatterArgumentParserr-r;DOC_URLr7 add_argumentrIr STUB_BEGIN parse_args init_pypkginit_lib init_prefixanyallerrorrr) print_helpsysexitrr%r(rr:r)rrparserargs init_flagsr!s rr$r$sL     >@]   IIII  $  F  k  F ,-9  F 78l  F F F&   F   ,       -       C           S    G12 G G G   0      D   +      D/4=$2BCJ"&H : : b LL` a a a-#    :   -- _TY ' ' {j     rrA)rr)r@r r?rBrrC) r)rRr?rBr!rr"rr#rSrrC) r@r r=rr?rBr#rSrrC)rr)#__doc__ __future__rrrr&rr5pathlibrrrrrr- file_utilsr r lib_stater r r rutilsrrrrrAr0r9r<r$rrrrrrs"43""""""  //////// =<<<<<<<<<<<8 8 8 8 v . . . . FFFFR?:?:?:?:DCCCCL z CHXXZZr