o j,@sddlmZddlZddlmZGdddZGdddZdZd Ze d kr_ej ej e Zej ed Zej ed ZeeZeed ZeeWddS1sXwYdSdS))TemplateN)IOc@s4eZdZd ddZdeeddfddZd d ZdS) ASTCodeGenerator _c_ast.cfgcCs ||_dd||D|_dS)zIInitialize the code generator from a configuration file. cSsg|] \}}t||qS)NodeCfg).0namecontentsrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/pycparser/_ast_gen.py sz-ASTCodeGenerator.__init__..N) cfg_filename parse_cfgfilenode_cfg)selfr rrr __init__s zASTCodeGenerator.__init__filereturnNcCsDttj|jd}|t7}|jD] }||d7}q||dS)z2Generates the code into file, an open file buffer.)r z N)r_PROLOGUE_COMMENT substituter _PROLOGUE_CODErgenerate_sourcewrite)rrsrcrrrr generate s  zASTCodeGenerator.generatec cst|da}|D]U}|}|r|drq |d}|d}|d}|dks2||ks2||kr=td|d|d |d |}||d|}|rWd d |d Dng} || fVq Wd d S1sjwYd S)z`Parse the configuration file and yield pairs of (name, contents) for each node. r#:[]zInvalid line in :  NcSsg|]}|qSr)strip)rvrrr r ;sz2ASTCodeGenerator.parse_cfgfile..,)openr# startswithfind RuntimeErrorsplit) rfilenameflinecolon_i lbracket_i rbracket_ir valvallistrrr r*s"      "zASTCodeGenerator.parse_cfgfile)r)__name__ __module__ __qualname__rrstrrrrrrr rs  rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)rzNode configuration. name: node name contents: a list of contents - attributes and child nodes See comment at the top of the configuration file for details. cCs~||_g|_g|_g|_g|_|D]+}|d}|j||dr*|j|q|dr6|j|q|j|qdS)N*z**)r all_entriesattrchild seq_childrstripappendendswith)rr r entry clean_entryrrr rGs    zNodeCfg.__init__cCs<|}|d|7}|d|7}|d|7}|S)Nr") _gen_init _gen_children _gen_iter_gen_attr_namesrrrrr rYs zNodeCfg.generate_sourcecCsd|jd}|jr&d|j}ddd|jD}|d7}d|d}nd }d }|d |d 7}|d |d7}|jdgD] }|d|d|d7}q@|S)Nzclass z(Node): , css|] }d|dVqdS)'Nr)rerrr esz$NodeCfg._gen_init..z, 'coord', '__weakref__'z(self, z , coord=None)z'coord', '__weakref__'z(self, coord=None)z __slots__ = (z) z def __init__r!coordz self.z = r")r r8join)rrargsslotsarglistr rrr rA`s zNodeCfg._gen_initcCsd}|jr>|d7}|jD]}|d|d7}|d|d|d7}q |jD]}|d|d 7}|d |d 7}q%|d 7}|S|d 7}|S)Nz def children(self): z nodelist = []  if self. is not None: z nodelist.append(("z", self.z)) z' for i, child in enumerate(self. or []): z nodelist.append((f"z[{i}]", child)) z return tuple(nodelist) z return () r8r:r;rrr:r;rrr rBts  zNodeCfg._gen_childrencCs~d}|jr9|jD]}|d|d7}|d|d7}q|jD]}|d|d7}|d7}q|js7|js7|d 7}|S|d 7}|S) Nz def __iter__(self): rOrPz yield self.r"z for child in (self.rQz yield child z return yield rRrSrrr rCs    zNodeCfg._gen_itercCs"dddd|jDd}|S)Nz attr_names = (css|]}|dVqdS)rFNr)rnmrrr rIsz*NodeCfg._gen_attr_names..))rKr9rErrr rDszNodeCfg._gen_attr_namesN) r3r4r5__doc__rrrArBrCrDrrrr r?s ra#----------------------------------------------------------------- # ** ATTENTION ** # This code was automatically generated from _c_ast.cfg # # Do not modify it directly. Modify the configuration file and # run the generator again. # ** ** *** ** ** # # pycparser: c_ast.py # # AST Node classes. # # Eli Bendersky [https://eli.thegreenplace.net/] # License: BSD #----------------------------------------------------------------- ae import sys from typing import Any, ClassVar, IO, Optional def _repr(obj): """ Get the representation of an object, with dedicated pprint-like format for lists. """ if isinstance(obj, list): return '[' + (',\n '.join((_repr(e).replace('\n', '\n ') for e in obj))) + '\n]' else: return repr(obj) class Node: __slots__ = () """ Abstract base class for AST nodes. """ attr_names: ClassVar[tuple[str, ...]] = () coord: Optional[Any] def __repr__(self): """ Generates a python representation of the current node """ result = self.__class__.__name__ + '(' indent = '' separator = '' for name in self.__slots__[:-2]: result += separator result += indent result += name + '=' + (_repr(getattr(self, name)).replace('\n', '\n ' + (' ' * (len(name) + len(self.__class__.__name__))))) separator = ',' indent = '\n ' + (' ' * len(self.__class__.__name__)) result += indent + ')' return result def children(self): """ A sequence of all children that are Nodes """ pass def show( self, buf: IO[str] = sys.stdout, offset: int = 0, attrnames: bool = False, showemptyattrs: bool = True, nodenames: bool = False, showcoord: bool = False, _my_node_name: Optional[str] = None, ): """ Pretty print the Node and all its attributes and children (recursively) to a buffer. buf: Open IO buffer into which the Node is printed. offset: Initial offset (amount of leading spaces) attrnames: True if you want to see the attribute names in name=value pairs. False to only see the values. showemptyattrs: False if you want to suppress printing empty attributes. nodenames: True if you want to see the actual node names within their parents. showcoord: Do you want the coordinates of each Node to be displayed. """ lead = ' ' * offset if nodenames and _my_node_name is not None: buf.write(lead + self.__class__.__name__+ ' <' + _my_node_name + '>: ') else: buf.write(lead + self.__class__.__name__+ ': ') if self.attr_names: def is_empty(v): v is None or (hasattr(v, '__len__') and len(v) == 0) nvlist = [(n, getattr(self,n)) for n in self.attr_names \ if showemptyattrs or not is_empty(getattr(self,n))] if attrnames: attrstr = ', '.join(f'{name}={value}' for name, value in nvlist) else: attrstr = ', '.join(f'{value}' for _, value in nvlist) buf.write(attrstr) if showcoord: buf.write(f' (at {self.coord})') buf.write('\n') for (child_name, child) in self.children(): child.show( buf, offset=offset + 2, attrnames=attrnames, showemptyattrs=showemptyattrs, nodenames=nodenames, showcoord=showcoord, _my_node_name=child_name) class NodeVisitor: """ A base NodeVisitor class for visiting c_ast nodes. Subclass it and define your own visit_XXX methods, where XXX is the class name you want to visit with these methods. For example: class ConstantVisitor(NodeVisitor): def __init__(self): self.values = [] def visit_Constant(self, node): self.values.append(node.value) Creates a list of values of all the constant nodes encountered below the given node. To use it: cv = ConstantVisitor() cv.visit(node) Notes: * generic_visit() will be called for AST nodes for which no visit_XXX method was defined. * The children of nodes for which a visit_XXX was defined will not be visited - if you need this, call generic_visit() on the node. You can use: NodeVisitor.generic_visit(self, node) * Modeled after Python's own AST visiting facilities (the ast module of Python 3.0) """ _method_cache = None def visit(self, node: Node): """ Visit a node. """ if self._method_cache is None: self._method_cache = {} visitor = self._method_cache.get(node.__class__.__name__, None) if visitor is None: method = 'visit_' + node.__class__.__name__ visitor = getattr(self, method, self.generic_visit) self._method_cache[node.__class__.__name__] = visitor return visitor(node) def generic_visit(self, node: Node): """ Called if no explicit visitor function exists for a node. Implements preorder visiting of the node. """ for _, c in node.children(): self.visit(c) __main__rzc_ast.pyw)stringrostypingrrrrrr3pathdirnameabspath__file__base_dirrKcfg_pathout_pathast_genr&outrrrrr s"  *c+  "