U
��_([ � @ s d dl Z G dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�Z G dd� de�Z
G dd� de�Z G dd� de�Z G dd� de�Z
G dd� de�ZG d d � d e�ZG d d � d e�ZG d d!� d!e�ZG d"d#� d#e�ZG d$d%� d%e�ZG d&d'� d'e�ZG d(d)� d)e�ZG d*d+� d+e�ZG d,d-� d-e�ZG d.d/� d/e�ZG d0d1� d1e�ZG d2d3� d3e�ZG d4d5� d5e�ZG d6d7� d7e�Z G d8d9� d9e�Z G d:d;� d;e�Z G d<d=� d=e�Z G d>d?� d?e�Z G d@dA� dAe�Z!G dBdC� dCe�Z"G dDdE� dEe�Z#G dFdG� dGe�Z$G dHdI� dIe�Z%G dJdK� dKe�Z&G dLdM� dMe�Z'G dNdO� dOe�Z(G dPdQ� dQe�Z)G dRdS� dSe�Z*G dTdU� dUe�Z+G dVdW� dWe�Z,G dXdY� dYe�Z-G dZd[� d[e�Z.G d\d]� d]e�Z/G d^d_� d_e�Z0G d`da� dae�Z1dS )b� Nc @ s0 e Zd ZdZdd� Zejdddddfdd�ZdS ) �Node� c C s dS )z3 A sequence of all children that are Nodes
Nr ��selfr r � /c_ast.py�children s z
Node.childrenr FNc
s� d| }|r4|dk r4|� |� jj d | d � n|� |� jj d � � jr�|r~� fdd�� jD �}d�d d
� |D ��} n(� fd d�� jD �}
d�d d
� |
D ��} |� | � |r�|� d
� j � |� d� � �� D ]"\} } | j||d |||| d� q�dS )a� 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.
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.
� Nz <z>: z: c s g | ]}|t � |�f�qS r ��getattr��.0�nr r r �
<listcomp>= s z Node.show.<locals>.<listcomp>z, c s s | ]}d | V qdS )z%s=%sNr )r Znvr r r � <genexpr>> s z Node.show.<locals>.<genexpr>c s g | ]}t � |��qS r r r r r r r @ s c s s | ]}d | V qdS )z%sNr )r �vr r r r A s z (at %s)�
� )�offset� attrnames� nodenames� showcoord�
_my_node_name)�write� __class__�__name__�
attr_names�join�coordr �show)
r Zbufr r r r r ZleadZnvlistZattrstrZvlistZ
child_name�childr r r r s.
�z Node.show)r �
__module__� __qualname__� __slots__r �sys�stdoutr r r r r r s r c @ s e Zd ZdZdd� Zdd� ZdS )� NodeVisitora- 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)
c C s" d|j j }t| || j�}||�S )z Visit a node.
Zvisit_)r r r
�
generic_visit)r �node�methodZvisitorr r r �visits s zNodeVisitor.visitc C s |� � D ]\}}| �|� qdS )zy Called if no explicit visitor function exists for a
node. Implements preorder visiting of the node.
N)r r) )r r' Zc_name�cr r r r& z s zNodeVisitor.generic_visitN)r r r! �__doc__r) r& r r r r r% R s r% c @ s& e Zd ZdZddd�Zdd� ZdZdS ) � ArrayDecl)�type�dim� dim_qualsr � __weakref__Nc C s || _ || _|| _|| _d S �N)r- r. r/ r )r r- r. r/ r r r r �__init__� s zArrayDecl.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr- r. )r- �appendr. �tuple�r �nodelistr r r r � s
zArrayDecl.children)r/ )N�r r r! r" r2 r r r r r r r, � s
r, c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �ArrayRef)�name� subscriptr r0 Nc C s || _ || _|| _d S r1 )r9 r: r )r r9 r: r r r r r2 � s zArrayRef.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr9 r: )r9 r3 r: r4 r5 r r r r � s
zArrayRef.childrenr )Nr7 r r r r r8 � s
r8 c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �
Assignment)�op�lvalue�rvaluer r0 Nc C s || _ || _|| _|| _d S r1 )r< r= r> r )r r< r= r> r r r r r2 � s zAssignment.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr= r> )r= r3 r> r4 r5 r r r r � s
zAssignment.children�r< )Nr7 r r r r r; � s
r; c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �BinaryOp)r< �left�rightr r0 Nc C s || _ || _|| _|| _d S r1 )r< rA rB r )r r< rA rB r r r r r2 � s zBinaryOp.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )NrA rB )rA r3 rB r4 r5 r r r r � s
zBinaryOp.childrenr? )Nr7 r r r r r@ � s
r@ c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Break�r r0 Nc C s
|| _ d S r1 �r �r r r r r r2 � s zBreak.__init__c C s dS �Nr r r r r r r � s zBreak.childrenr )Nr7 r r r r rC � s
rC c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Case)�expr�stmtsr r0 Nc C s || _ || _|| _d S r1 )rI rJ r )r rI rJ r r r r r2 � s z
Case.__init__c C sP g }| j d k r |�d| j f� t| jp(g �D ]\}}|�d| |f� q,t|�S )NrI � stmts[%d])rI r3 � enumeraterJ r4 �r r6 �ir r r r r � s
z
Case.childrenr )Nr7 r r r r rH � s
rH c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Cast)�to_typerI r r0 Nc C s || _ || _|| _d S r1 )rP rI r )r rP rI r r r r r2 � s z
Cast.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )NrP rI )rP r3 rI r4 r5 r r r r � s
z
Cast.childrenr )Nr7 r r r r rO � s
rO c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Compound)� block_itemsr r0 Nc C s || _ || _d S r1 )rR r )r rR r r r r r2 � s zCompound.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S )Nzblock_items[%d])rL rR r3 r4 rM r r r r � s zCompound.childrenr )Nr7 r r r r rQ � s
rQ c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �CompoundLiteral)r- �initr r0 Nc C s || _ || _|| _d S r1 )r- rT r )r r- rT r r r r r2 � s zCompoundLiteral.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr- rT )r- r3 rT r4 r5 r r r r � s
zCompoundLiteral.childrenr )Nr7 r r r r rS � s
rS c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Constant)r- �valuer r0 Nc C s || _ || _|| _d S r1 )r- rV r )r r- rV r r r r r2 s zConstant.__init__c C s g }t |�S r1 �r4 r5 r r r r s zConstant.children)r- rV )Nr7 r r r r rU s
rU c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �ContinuerD Nc C s
|| _ d S r1 rE rF r r r r2 s zContinue.__init__c C s dS rG r r r r r r s zContinue.childrenr )Nr7 r r r r rX s
rX c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Decl) r9 �quals�storage�funcspecr- rT �bitsizer r0 Nc C s4 || _ || _|| _|| _|| _|| _|| _|| _d S r1 )r9 rZ r[ r\ r- rT r] r ) r r9 rZ r[ r\ r- rT r] r r r r r2 s z
Decl.__init__c C sZ g }| j d k r |�d| j f� | jd k r8|�d| jf� | jd k rR|�d| jf� t|�S )Nr- rT r] )r- r3 rT r] r4 r5 r r r r * s
z
Decl.children)r9 rZ r[ r\ )Nr7 r r r r rY s
rY c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �DeclList)�declsr r0 Nc C s || _ || _d S r1 )r_ r )r r_ r r r r r2 5 s zDeclList.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S �Nz decls[%d]�rL r_ r3 r4 rM r r r r 9 s zDeclList.childrenr )Nr7 r r r r r^ 3 s
r^ c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Default)rJ r r0 Nc C s || _ || _d S r1 )rJ r )r rJ r r r r r2 C s zDefault.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S )NrK )rL rJ r3 r4 rM r r r r G s zDefault.childrenr )Nr7 r r r r rb A s
rb c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �DoWhile��cond�stmtr r0 Nc C s || _ || _|| _d S r1 �re rf r �r re rf r r r r r2 Q s zDoWhile.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S �Nre rf �re r3 rf r4 r5 r r r r V s
zDoWhile.childrenr )Nr7 r r r r rc O s
rc c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �
EllipsisParamrD Nc C s
|| _ d S r1 rE rF r r r r2 ` s zEllipsisParam.__init__c C s dS rG r r r r r r c s zEllipsisParam.childrenr )Nr7 r r r r rk ^ s
rk c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �EmptyStatementrD Nc C s
|| _ d S r1 rE rF r r r r2 j s zEmptyStatement.__init__c C s dS rG r r r r r r m s zEmptyStatement.childrenr )Nr7 r r r r rl h s
rl c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Enum)r9 �valuesr r0 Nc C s || _ || _|| _d S r1 )r9 rn r )r r9 rn r r r r r2 t s z
Enum.__init__c C s& g }| j d k r |�d| j f� t|�S )Nrn )rn r3 r4 r5 r r r r y s
z
Enum.children�r9 )Nr7 r r r r rm r s
rm c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �
Enumerator)r9 rV r r0 Nc C s || _ || _|| _d S r1 )r9 rV r )r r9 rV r r r r r2 � s zEnumerator.__init__c C s& g }| j d k r |�d| j f� t|�S )NrV )rV r3 r4 r5 r r r r � s
zEnumerator.childrenro )Nr7 r r r r rp � s
rp c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �EnumeratorList)� enumeratorsr r0 Nc C s || _ || _d S r1 )rr r )r rr r r r r r2 � s zEnumeratorList.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S )Nzenumerators[%d])rL rr r3 r4 rM r r r r � s zEnumeratorList.childrenr )Nr7 r r r r rq � s
rq c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �ExprList��exprsr r0 Nc C s || _ || _d S r1 �ru r �r ru r r r r r2 � s zExprList.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S �Nz exprs[%d]�rL ru r3 r4 rM r r r r � s zExprList.childrenr )Nr7 r r r r rs � s
rs c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �FileAST)�extr r0 Nc C s || _ || _d S r1 )r{ r )r r{ r r r r r2 � s zFileAST.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S )Nzext[%d])rL r{ r3 r4 rM r r r r � s zFileAST.childrenr )Nr7 r r r r rz � s
rz c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �For)rT re �nextrf r r0 Nc C s" || _ || _|| _|| _|| _d S r1 )rT re r} rf r )r rT re r} rf r r r r r2 � s
z For.__init__c C st g }| j d k r |�d| j f� | jd k r8|�d| jf� | jd k rR|�d| jf� | jd k rl|�d| jf� t|�S )NrT re r} rf )rT r3 re r} rf r4 r5 r r r r � s
z For.childrenr )Nr7 r r r r r| � s
r| c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �FuncCall)r9 �argsr r0 Nc C s || _ || _|| _d S r1 )r9 r r )r r9 r r r r r r2 � s zFuncCall.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr9 r )r9 r3 r r4 r5 r r r r � s
zFuncCall.childrenr )Nr7 r r r r r~ � s
r~ c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �FuncDecl)r r- r r0 Nc C s || _ || _|| _d S r1 )r r- r )r r r- r r r r r2 � s zFuncDecl.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr r- )r r3 r- r4 r5 r r r r � s
zFuncDecl.childrenr )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �FuncDef)�decl� param_decls�bodyr r0 Nc C s || _ || _|| _|| _d S r1 )r� r� r� r )r r� r� r� r r r r r2 � s zFuncDef.__init__c C sj g }| j d k r |�d| j f� | jd k r8|�d| jf� t| jpBg �D ]\}}|�d| |f� qFt|�S )Nr� r� zparam_decls[%d])r� r3 r� rL r� r4 rM r r r r � s
zFuncDef.childrenr )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Goto�r9 r r0 Nc C s || _ || _d S r1 �r9 r �r r9 r r r r r2 � s z
Goto.__init__c C s g }t |�S r1 rW r5 r r r r s z
Goto.childrenro )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �IDr� Nc C s || _ || _d S r1 r� r� r r r r2 s z ID.__init__c C s g }t |�S r1 rW r5 r r r r
s z ID.childrenro )Nr7 r r r r r� s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �IdentifierType)�namesr r0 Nc C s || _ || _d S r1 )r� r )r r� r r r r r2 s zIdentifierType.__init__c C s g }t |�S r1 rW r5 r r r r s zIdentifierType.children)r� )Nr7 r r r r r� s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �If�re �iftrue�iffalser r0 Nc C s || _ || _|| _|| _d S r1 �re r� r� r �r re r� r� r r r r r2 ! s z If.__init__c C sZ g }| j d k r |�d| j f� | jd k r8|�d| jf� | jd k rR|�d| jf� t|�S �Nre r� r� �re r3 r� r� r4 r5 r r r r ' s
z If.childrenr )Nr7 r r r r r� s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �InitListrt Nc C s || _ || _d S r1 rv rw r r r r2 2 s zInitList.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S rx ry rM r r r r 6 s zInitList.childrenr )Nr7 r r r r r� 0 s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Label)r9 rf r r0 Nc C s || _ || _|| _d S r1 )r9 rf r )r r9 rf r r r r r2 @ s zLabel.__init__c C s& g }| j d k r |�d| j f� t|�S )Nrf )rf r3 r4 r5 r r r r E s
zLabel.childrenro )Nr7 r r r r r� > s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �NamedInitializer)r9 rI r r0 Nc C s || _ || _|| _d S r1 )r9 rI r )r r9 rI r r r r r2 N s zNamedInitializer.__init__c C sP g }| j d k r |�d| j f� t| jp(g �D ]\}}|�d| |f� q,t|�S )NrI zname[%d])rI r3 rL r9 r4 rM r r r r S s
zNamedInitializer.childrenr )Nr7 r r r r r� L s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) � ParamList)�paramsr r0 Nc C s || _ || _d S r1 )r� r )r r� r r r r r2 ^ s zParamList.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S )Nz
params[%d])rL r� r3 r4 rM r r r r b s zParamList.childrenr )Nr7 r r r r r� \ s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �PtrDecl)rZ r- r r0 Nc C s || _ || _|| _d S r1 )rZ r- r )r rZ r- r r r r r2 l s zPtrDecl.__init__c C s& g }| j d k r |�d| j f� t|�S �Nr- �r- r3 r4 r5 r r r r q s
zPtrDecl.children)rZ )Nr7 r r r r r� j s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Return)rI r r0 Nc C s || _ || _d S r1 )rI r )r rI r r r r r2 z s zReturn.__init__c C s& g }| j d k r |�d| j f� t|�S �NrI �rI r3 r4 r5 r r r r ~ s
zReturn.childrenr )Nr7 r r r r r� x s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Struct�r9 r_ r r0 Nc C s || _ || _|| _d S r1 �r9 r_ r �r r9 r_ r r r r r2 � s zStruct.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S r` ra rM r r r r � s zStruct.childrenro )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) � StructRef)r9 r- �fieldr r0 Nc C s || _ || _|| _|| _d S r1 )r9 r- r� r )r r9 r- r� r r r r r2 � s zStructRef.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S )Nr9 r� )r9 r3 r� r4 r5 r r r r � s
zStructRef.children)r- )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Switchrd Nc C s || _ || _|| _d S r1 rg rh r r r r2 � s zSwitch.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S ri rj r5 r r r r � s
zSwitch.childrenr )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) � TernaryOpr� Nc C s || _ || _|| _|| _d S r1 r� r� r r r r2 � s zTernaryOp.__init__c C sZ g }| j d k r |�d| j f� | jd k r8|�d| jf� | jd k rR|�d| jf� t|�S r� r� r5 r r r r � s
zTernaryOp.childrenr )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �TypeDecl)�declnamerZ r- r r0 Nc C s || _ || _|| _|| _d S r1 )r� rZ r- r )r r� rZ r- r r r r r2 � s zTypeDecl.__init__c C s& g }| j d k r |�d| j f� t|�S r� r� r5 r r r r � s
zTypeDecl.children)r� rZ )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Typedef)r9 rZ r[ r- r r0 Nc C s" || _ || _|| _|| _|| _d S r1 )r9 rZ r[ r- r )r r9 rZ r[ r- r r r r r2 � s
zTypedef.__init__c C s& g }| j d k r |�d| j f� t|�S r� r� r5 r r r r � s
zTypedef.children)r9 rZ r[ )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Typename)r9 rZ r- r r0 Nc C s || _ || _|| _|| _d S r1 )r9 rZ r- r )r r9 rZ r- r r r r r2 � s zTypename.__init__c C s& g }| j d k r |�d| j f� t|�S r� r� r5 r r r r � s
zTypename.children)r9 rZ )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �UnaryOp)r< rI r r0 Nc C s || _ || _|| _d S r1 )r< rI r )r r< rI r r r r r2 � s zUnaryOp.__init__c C s& g }| j d k r |�d| j f� t|�S r� r� r5 r r r r � s
zUnaryOp.childrenr? )Nr7 r r r r r� � s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Unionr� Nc C s || _ || _|| _d S r1 r� r� r r r r2 s zUnion.__init__c C s6 g }t | jpg �D ]\}}|�d| |f� qt|�S r` ra rM r r r r s zUnion.childrenro )Nr7 r r r r r� s
r� c @ s& e Zd ZdZddd�Zdd� ZdZdS ) �Whilerd Nc C s || _ || _|| _d S r1 rg rh r r r r2 s zWhile.__init__c C s@ g }| j d k r |�d| j f� | jd k r8|�d| jf� t|�S ri rj r5 r r r r s
zWhile.childrenr )Nr7 r r r r r� s
r� )2r# �objectr r% r, r8 r; r@ rC rH rO rQ rS rU rX rY r^ rb rc rk rl rm rp rq rs rz r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s` <0