i ddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZmZdd Zdd ZddZddZdS)) annotations)ClassIR) Dependency)FuncIR)AssignCallC PrimitiveOp)RStructRTupleRTypeRUnionRVecfnrreturnset[Dependency] | Nonecd}t||}|jD]}|jD]}t|tr Jdt|t r6|j/|jD]'}|t}||(t|j |}t|trt|j j |}|S)aFind implicit dependencies that need to be imported. Using primitives or types defined in librt submodules such as "librt.base64" requires dependency imports (e.g., capsule imports). Note that a module can depend on a librt module even if it doesn't explicitly import it, for example via re-exported names or via return types of functions defined in other modules. NzLowered IR is expected) find_type_dependenciesblocksops isinstancer r dependenciessetaddcollect_type_depstyperdest)rdepsblockopdeps Z/root/projects/qq-shell/venv/lib64/python3.11/site-packages/mypyc/analysis/capsule_deps.pyfind_implicit_op_dependenciesr" s$(D !"d + +D = =) = =B!"k22 L L4L L L L"e$$ ")D?""C|"uuHHSMMMM$RWd33D"f%% =(t<< = Krc|jjjD]}t|j|}t|jjj|}|S)zFind dependencies from RTypes in function signatures. Some RTypes (e.g., those for librt types) have associated dependencies that need to be imported when the type is used. )declsigargsrrret_type)rrargs r!rr&sHw{11 400 RW[14 8 8D Kr#clrcvd}|jD].}|jD]}t||}/|S)z-Find dependencies from class attribute types.N)mro attributesvaluesr)r*rbase attr_types r!find_class_dependenciesr14sP#'D66//11 6 6I$Y55DD 6 Kr#typr c|j/|jD]'}|t}||(t|tr|jD]}t ||}nt|tr|jD]}t ||}nZt|tr|jD]}t ||}n*t|trt |j |}|S)zHCollect dependencies from an RType, recursively checking compound types.) rrrrr itemsrr typesr r item_type)r2rr items r!rr=s ##  C|uu HHSMMMM#v 6I 1 1D$T400DD 1 C 6I 1 1D$T400DD 1 C ! !6I 1 1D$T400DD 1 C  6 55 Kr#N)rrrr)rrrrrr)r*rrr)r2r rrrr) __future__rmypyc.ir.class_irr mypyc.ir.depsrmypyc.ir.func_irr mypyc.ir.opsrrr mypyc.ir.rtypesr r r r rr"rr1rr#r!r?s""""""%%%%%%$$$$$$######3333333333@@@@@@@@@@@@@@8    r#