|Le dZddlZddlZddlZddlZddlZGddejZGddejZ Gdde Z d Z d Z d Z dd ZdZdZdZdS)zJ Utilities for manipulating the Abstract Syntax Tree of Python constructs Nc4eZdZdZfdZdZdZdZxZS) NameVisitorzQ NodeVisitor that builds a set of all of the named identifiers in an AST cntt|j|i|t|_dSN)superr__init__setnames)selfargskwargs __class__s 1lib/python3.11/site-packages/datashader/macros.pyrzNameVisitor.__init__s4)k4  )4:6:::UU cD|j|jdSr)r addidr nodes r visit_NamezNameVisitor.visit_Names twrct|dr!|j|jdSt|dr!|j|jdSdS)Nargr)hasattrr rrrrs r visit_argzNameVisitor.visit_argsh 4   $ JNN48 $ $ $ $ $ T4  $ JNN47 # # # # # $ $rctjdfd|jD}|rtd|Ddz}nd}dt |||zDS)aD Returns a list of new names that are not already present in the AST. New names will have the form _N, for N a non-negative integer. If the AST has no existing identifiers of this form, then the returned names will start at 0 ('_0', '_1', '_2'). If the AST already has identifiers of this form, then the names returned will not include the existing identifiers. Parameters ---------- num_names: int The number of new names to return Returns ------- list of str z^_(\d+)$c>g|]}||S)match).0nprop_res r z-NameVisitor.get_new_names..0s*DDD7==3C3CD!DDDrc<g|]}t|ddS)N)intrr s rr"z-NameVisitor.get_new_names..2s&CCCqAabbE CCCrr$rc2g|]}dt|zS)_)strr&s rr"z-NameVisitor.get_new_names..6s.???c!ff ???r)recompiler maxrange)r num_namesmatching_names start_numberr!s @r get_new_nameszNameVisitor.get_new_namess&*[))DDDDTZDDD  CCNCCCDDqHLLL??lL9$<==??? ?r) __name__ __module__ __qualname____doc__rrrr1 __classcell__rs@rrr so   $$$ ???????rrc"eZdZdZfdZxZS)ExpandVarargTransformerz Node transformer that replaces the starred use of a variable in an AST with a collection of unstarred named variables. cdtt|j|i|||_||_dS)a Parameters ---------- starred_name: str The name of the starred variable to replace expand_names: list of stf List of the new names that should be used to replace the starred variable N)rr9r starred_name expand_names)r r;r<r r rs rrz ExpandVarargTransformer.__init__?s> 6%t,,5tFvFFF((r)r2r3r4r5rr6r7s@rr9r9:sB ) ) ) ) ) ) ) ) )rr9ceZdZdZdS)ExpandVarargTransformerStarredcXjj|jkrfd|jDSS)NcFg|]}tj|jS))rctx)astNamerA)rnamers rr"z@ExpandVarargTransformerStarred.visit_Starred..Ss6'''CH$(333'''r)valuerr;r<rs `r visit_Starredz,ExpandVarargTransformerStarred.visit_StarredQsJ :=D- - -''''%''' 'KrN)r2r3r4rFrrrr>r>Os#rr>cztjtj|}t j|}|S)z2 Get the AST representation of a function )textwrapdedentinspect getsourcerBparse)fn fn_sourcefn_asts rfunction_to_astrPYs6  1" 5 566IYy ! !F Mrc4ddl}||S)z9Convert AST to source code string using the astor packagerN)astor to_source)rBrRs r ast_to_sourcerTjsLLL ??3  rct|tjsJ|jd}t|tjsJt |d|jzdS)zO Compile function AST into a code object suitable for use in eval/exec rz<%s>exec)mode) isinstancerBModulebody FunctionDefr+rD)rO fndef_asts rcompile_function_astr]ps[ fcj ) )))) AI i 1 1111 66IN2 @ @ @@rr$ct|tjsJ|jd}t|tjsJt |}t j}|}t|D] }|j } |j }|j }~tj |} | |t|| | |jS)Nr)rXrBrYrZr[r]rJ currentframer-f_backf_locals f_globalscopyupdateevalrD) rO stacklevelr\code current_frame eval_framer( eval_locals eval_globalsscopes rfunction_ast_to_functionrmzs fcj ) )))) AI i 1 1111  ' 'D(**MJ :  ''& %K'L Il # #E LL u   rc,tj|S)N)r)rBrrDs r _build_argrps 7t   rcPt|tjsJtj|}|jd}t|tjsJ|j}|j}|std|sJt|tr|}n|j }t}| |||}t}||| |} | jd} | jjd|Dd| j_t} | | || jvr#td|g|_tj| | S)a Given a function AST that use a variable length positional argument (e.g. *args), return a function that replaces the use of this argument with one or more fixed arguments. To be supported, a function must have a starred argument in the function signature, and it may only use this argument in starred form as the input to other functions. For example, suppose expand_number is 3 and fn_ast is an AST representing this function... def my_fn1(a, b, *args): print(a, b) other_fn(a, b, *args) Then this function will return the AST of a function equivalent to... def my_fn1(a, b, _0, _1, _2): print(a, b) other_fn(a, b, _0, _1, _2) If the input function uses `args` for anything other than passing it to other functions in starred form, an error will be raised. Parameters ---------- fn_ast: ast.FunctionDef expand_number: int Returns ------- ast.FunctionDef rzmInput function AST does not have a variable length positional argument (e.g. *args) in the function signaturec.g|]}t|S)ro)rp)rrDs rr"z/expand_function_ast_varargs..s#8884   888rNzNThe variable length positional argument {n} is used in an unsupported context )r )rXrBrYrcdeepcopyrZr[r vararg ValueErrorr)rrvisitr1r>extendr formatdecorator_listfix_missing_locations) rO expand_numberr\fn_args fn_vararg vararg_namebefore_name_visitorr<expand_transformer new_fn_ast new_fndef_astafter_name_visitors rexpand_function_ast_varargsrsF fcj ) ))))]6 " "F AI i 1 1111nGI +*++ + )S!!$ m &--f%%%&44]CCL8##\ eFmmOA&M""88<888!%M%Z((((... F[F  "Ij))) rcfttrdkrtdfd}|S)aX Decorator to expand the variable length (starred) argument in a function signature with a fixed number of arguments. Parameters ---------- expand_number: int The number of fixed arguments that should replace the variable length argument Returns ------- function Decorator Function rz,expand_number must be a non-negative integercdt|}t|}t|dS)N)rf)rPrrm)rMrOfn_expanded_astr{s r_expand_varargsz'expand_varargs.._expand_varargss2 $$5fmLL'AFFFFr)rXr%ru)r{rs` rexpand_varargsrsW mS ) )I]Q->->GHHHGGGGG r)r$)r5r*rcrJrBrH NodeVisitorrNodeTransformerr9r>rPrTr]rmrprrrrrrsB  ,?,?,?,?,?#/,?,?,?^)))))c1)))*%<"   AAA!!!!:cccLr