a `"@sddlZddlZddlZddlZddlZGdddejZGdddejZGdddeZ Gdd d eZ d d Z d d Z ddZ dddZddZddZddZdS)Ncs8eZdZdZfddZddZddZdd ZZS) NameVisitorzQ NodeVisitor that builds a set of all of the named identifiers in an AST cs"tt|j|i|t|_dSN)superr__init__setnames)selfargskwargs __class__0lib/python3.9/site-packages/datashader/macros.pyrszNameVisitor.__init__cCs|j|jdSr)raddidrnoder r r visit_NameszNameVisitor.visit_NamecCs6t|dr|j|jnt|dr2|j|jdS)Nargr)hasattrrrrrrr r r visit_args  zNameVisitor.visit_argcsVtdfdd|jD}|r:tdd|Dd}nd}ddt|||DS)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+)$csg|]}|r|qSr )match.0nZprop_rer r 0z-NameVisitor.get_new_names..cSsg|]}t|ddqS)N)intrr r rr2rrrcSsg|]}dt|qS)_)strrr r rr6r)recompilermaxrange)rZ num_namesZmatching_namesZ start_numberr rr get_new_namess  zNameVisitor.get_new_names) __name__ __module__ __qualname____doc__rrrr& __classcell__r r r rr s  rcs eZdZdZfddZZS)ExpandVarargTransformerz Node transformer that replaces the starred use of a variable in an AST with a collection of unstarred named variables. cs&tt|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)rr,r starred_name expand_names)rr-r.r r r r rr?s z ExpandVarargTransformer.__init__)r'r(r)r*rr+r r r rr,:sr,c@seZdZddZdS)ExpandVarargTransformerStarredcs*jj|jkr"fdd|jDSSdS)Ncsg|]}tj|jdqSrctx)astNamer2rnamerr rrSrz@ExpandVarargTransformerStarred.visit_Starred..)valuerr-r.rr r7r visit_StarredQs  z,ExpandVarargTransformerStarred.visit_StarredN)r'r(r)r9r r r rr/Osr/c@seZdZddZdS)ExpandVarargTransformerCallArgcCsDt|ddr<|jj|jkr.)getattrr;rr-r extendr.rr r r visit_Call[s z)ExpandVarargTransformerCallArg.visit_CallN)r'r(r)r>r r r rr:Ysr:cCstt|}t|}|S)z2 Get the AST representation of a function )textwrapdedentinspectZ getsourcer3parse)fnZ fn_sourcefn_astr r rfunction_to_astes rEcCsddl}||S)z9Convert AST to source code string using the astor packagerN)astorZ to_source)r3rFr r r ast_to_sourcevsrGcCs>t|tjsJ|jd}t|tjs*Jt|d|jddS)zO Compile function AST into a code object suitable for use in eval/exec rz<%s>exec)mode) isinstancer3Modulebody FunctionDefr#r6)rD fndef_astr r rcompile_function_ast|s rOrc Cst|tjsJ|jd}t|tjs*Jt|}t}|}t|D] }|j }qF|j }|j }~t |} | |t|| | |jS)Nr)rJr3rKrLrMrOrAZ currentframer%f_backf_locals f_globalscopyupdateevalr6) rD stacklevelrNcodeZ current_frameZ eval_framer Z eval_localsZ eval_globalsZscoper r rfunction_ast_to_functions     rXcCs8ztj|dWSty2tj|tdYS0dS)N)rr1)r3rAttributeErrorr4ZParamr6r r r _build_args r[c Cst|tjsJt|}|jd}t|tjs4J|j}|j}|sLt d|sTJt|t rd|}n|j }t }| |||}ttdrt}nt}||| |} | jd} | jjdd|Dd| j_t } | | || jvrt dj|dg|_t| | 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 signatureZStarredcSsg|]}t|dqS)rZ)r[r5r r rrrz/expand_function_ast_varargs..NzNThe variable length positional argument {n} is used in an unsupported context )r)rJr3rKrSdeepcopyrLrMr vararg ValueErrorr!rrZvisitr&rr/r:r=rformatdecorator_listZfix_missing_locations) rD expand_numberrNZfn_argsZ fn_varargZ vararg_nameZbefore_name_visitorr.Zexpand_transformerZ new_fn_astZ new_fndef_astZafter_name_visitorr r rexpand_function_ast_varargssJ#           rbcs*ttrdkrtdfdd}|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 integercst|}t|}t|ddS)N)rV)rErbrX)rCrDZfn_expanded_astrar r_expand_varargs*s z'expand_varargs.._expand_varargs)rJrr^)rarer rdrexpand_varargss rf)r)r"rSrAr3r?Z NodeVisitorrZNodeTransformerr,r/r:rErGrOrXr[rbrfr r r rs/    k