U
|
L ±dlE ã @ s d dl mZ G dd deZdS )é )Úc_astc @ s6 e Zd ZdZdddZdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdddd d d d d d d d d d d!d!d"d"d"d#Zd$d% Zd&d' Zd(d) Zd*d+ Zdd,d-Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Zd<d= Z d>d? Z d@dA Z dBdC Z dDdE Z dFdG Z!dHdI Z"dJdK Z#dLdM Z$dNdO Z%dPdQ Z&dRdS Z'dTdU Z(dVdW Z)dXdY Z*dZd[ Z+d\d] Z,d^d_ Z-d`da Z.dbdc Z/ddde Z0dfdg Z1dhdi Z2djdk Z3dldm Z4dndo Z5dpdq Z6drds Z7dtdu Z8dvdw Z9dxdy Z:dzd{ Z;d|d} Z<dd~dZ=dd Z>g dfddZ?d
d Z@dd ZAdd ZBdS )Ú
|
CGeneratorz¬ Uses the same visitor pattern as c_ast.NodeVisitor, but modified to
|
return a value from each visit method, using string accumulation in
|
generic_visit.
|
Fc C s d| _ || _dS )z Constructs C-code generator
|
|
reduce_parentheses:
|
if True, eliminates needless parentheses on binary operators
|
é N)Ú indent_levelÚreduce_parentheses)Úselfr © r úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pycparser/c_generator.pyÚ__init__ s zCGenerator.__init__c C s
|
d| j S ©Nú )r ©r r r r Ú _make_indent s zCGenerator._make_indentc C s d|j j }t| || j|S )NZvisit_)Ú __class__Ú__name__ÚgetattrÚ generic_visit)r ÚnodeÚmethodr r r Úvisit s zCGenerator.visitc s, |d kr dS d fdd| ¡ D ¡S d S )NÚ c 3 s | ]\}} |¡V qd S ©N©r )Ú.0Zc_nameÚcr r r Ú <genexpr>' s z+CGenerator.generic_visit.<locals>.<genexpr>)ÚjoinÚchildren)r r r r r r # s zCGenerator.generic_visitc C s |j S r )Úvalue©r Únr r r Úvisit_Constant) s zCGenerator.visit_Constantc C s |j S r ©Únamer r r r Úvisit_ID, s zCGenerator.visit_IDc C s d}|j r|d|j 7 }|S )Nz#pragmar )Ústring)r r Úretr r r Ú visit_Pragma/ s zCGenerator.visit_Pragmac C s$ | |j¡}|d | |j¡ d S )Nú[ú])Ú_parenthesize_unless_simpler# r Z subscript)r r Zarrrefr r r Úvisit_ArrayRef5 s zCGenerator.visit_ArrayRefc C s" | |j¡}||j | |j¡ S r )r* r# Útyper Úfield)r r Zsrefr r r Úvisit_StructRef9 s zCGenerator.visit_StructRefc C s$ | |j¡}|d | |j¡ d S )Nú(ú))r* r# r Úargs)r r Zfrefr r r Úvisit_FuncCall= s zCGenerator.visit_FuncCallc C s\ |j dkrd| |j¡ S | |j¡}|j dkr8d| S |j dkrJd| S d|j |f S d S )NÚsizeofz
|
sizeof(%s)zp++z%s++zp--z%s--z%s%s)Úopr Úexprr* )r r Zoperandr r r Ú visit_UnaryOpA s
|
|
|
zCGenerator.visit_UnaryOpr r é é é é é é é é )z||z&&ú|ú^ú&z==z!=ú>z>=ú<z<=z>>z<<ú+ú-Ú*ú/ú%c s@ j fdd¡} j fdd¡}d| j|f S )Nc s6 | ¡p2jo2t| tjo2j| j j j k S r ©Ú_is_simple_noder Ú
|
isinstancer ÚBinaryOpÚprecedence_mapr4 ©Úd©r r r r Ú<lambda>k s
|
þz+CGenerator.visit_BinaryOp.<locals>.<lambda>c s6 | ¡p2jo2t| tjo2j| j j j k S r rI rN rP r r rQ w s
|
þú%s %s %s)Ú_parenthesize_ifÚleftÚrightr4 )r r Zlval_strÚrval_strr rP r Úvisit_BinaryOp_ s
|
þ þzCGenerator.visit_BinaryOpc C s* | |jdd ¡}d| |j¡|j|f S )Nc S s t | tjS r )rK r Ú
|
Assignment)r r r r rQ ó z-CGenerator.visit_Assignment.<locals>.<lambda>rR )rS Zrvaluer Zlvaluer4 )r r rV r r r Úvisit_Assignment| s
|
þzCGenerator.visit_Assignmentc C s d |j¡S r )r Únamesr r r r Úvisit_IdentifierType s z CGenerator.visit_IdentifierTypec C sJ t |tjr d| |¡ d S t |tjr<d| |¡ d S | |¡S d S )NÚ{Ú}r/ r0 )rK r ZInitListr ÚExprListr r r r Ú _visit_expr
s
|
zCGenerator._visit_exprc C sL |r
|
|j n| |¡}|jr.|d| |j¡ 7 }|jrH|d| |j¡ 7 }|S )Nz : ú = )r# Ú_generate_declZbitsizer Úinitr` )r r Úno_typeÚsr r r Ú
|
visit_Decl s zCGenerator.visit_Declc sL |jd ¡}t|jdkrH|dd fdd|jdd
D ¡ 7 }|S )Nr r ú, c 3 s | ]} j |d dV qdS )T)rd N)rf ©r Údeclr r r r s ÿz,CGenerator.visit_DeclList.<locals>.<genexpr>)r ÚdeclsÚlenr ©r r re r r r Úvisit_DeclList s ÿ zCGenerator.visit_DeclListc C s2 d}|j r |d |j ¡d 7 }|| |j¡7 }|S )Nr r )Ústorager Ú_generate_typer, rl r r r Ú visit_Typedef s
|
zCGenerator.visit_Typedefc C s, d| j |jdd d }|d | |j¡ S )Nr/ F©Ú emit_declnamer0 r )ro Zto_typer* r5 rl r r r Ú
|
visit_Cast¤ s zCGenerator.visit_Castc C s* g }|j D ]}| | |¡¡ q
|
d |¡S ©Nrg ©ÚexprsÚappendr` r ©r r Zvisited_subexprsr5 r r r Úvisit_ExprList¨ s
|
zCGenerator.visit_ExprListc C s* g }|j D ]}| | |¡¡ q
|
d |¡S rt ru rx r r r Úvisit_InitList® s
|
zCGenerator.visit_InitListc C s | j |ddS )NÚenumr" ©Ú_generate_struct_union_enumr r r r Ú
|
visit_Enum´ s zCGenerator.visit_Enumc C s d | |j¡¡S )Nz _Alignas({}))Úformatr Ú alignmentr r r r Ú visit_Alignas· s zCGenerator.visit_Alignasc C s< |j sdj| ¡ |jdS dj| ¡ |j| |j ¡dS d S )Nz{indent}{name},
|
)Úindentr# z{indent}{name} = {value},
|
)r r# r )r r r r# r r r r r Úvisit_Enumeratorº s þ
|
ýzCGenerator.visit_Enumeratorc sj |j¡}d _ |j¡}|jrVd fdd|jD ¡}|d | d | d S |d | d S d S )Nr ú;
|
c 3 s | ]} |¡V qd S r r )r Úpr r r r Ì s z+CGenerator.visit_FuncDef.<locals>.<genexpr>Ú
|
)r ri r ÚbodyZ param_declsr )r r ri r Zknrdeclsr r r Ú visit_FuncDefÇ s zCGenerator.visit_FuncDefc C sb d}|j D ]R}t|tjr*|| |¡7 }q
|
t|tjrJ|| |¡d 7 }q
|
|| |¡d 7 }q
|
|S )Nr r r )ÚextrK r ZFuncDefr ZPragma)r r re r r r r Ú visit_FileASTÑ s
|
zCGenerator.visit_FileASTc s` ¡ d } jd7 _|jr>|d fdd|jD ¡7 } jd8 _| ¡ d 7 }|S )Nú{
|
r7 r c 3 s | ]} |¡V qd S r ©Ú_generate_stmt)r Ústmtr r r r à s z,CGenerator.visit_Compound.<locals>.<genexpr>z}
|
)r r Z block_itemsr rl r r r Úvisit_CompoundÜ s zCGenerator.visit_Compoundc C s$ d| |j¡ d | |j¡ d S )Nr/ z){r^ )r r, rc r r r r Úvisit_CompoundLiteralå s z CGenerator.visit_CompoundLiteralc C s dS )Nú;r r r r r Úvisit_EmptyStatementé s z CGenerator.visit_EmptyStatementc s d fdd|jD ¡S )Nrg c 3 s | ]} |¡V qd S r r )r Úparamr r r r í s z-CGenerator.visit_ParamList.<locals>.<genexpr>)r Úparamsr r r r Úvisit_ParamListì s zCGenerator.visit_ParamListc C s& d}|j r |d| |j ¡ 7 }|d S )NÚreturnr r )r5 r rl r r r Ú visit_Returnï s zCGenerator.visit_Returnc C s dS )Nzbreak;r r r r r Ú visit_Breakô s zCGenerator.visit_Breakc C s dS )Nz continue;r r r r r Úvisit_Continue÷ s zCGenerator.visit_Continuec C sH d| |j¡ d }|d| |j¡ d 7 }|d| |j¡ d 7 }|S )Nr/ z) ? z) : r0 )r` ÚcondÚiftrueÚiffalserl r r r Úvisit_TernaryOpú s zCGenerator.visit_TernaryOpc C sd d}|j r|| |j ¡7 }|d7 }|| j|jdd7 }|jr`|| ¡ d 7 }|| j|jdd7 }|S )Nzif (ú)
|
T©Ú
|
add_indentzelse
|
)r r r r r r rl r r r Úvisit_If s zCGenerator.visit_Ifc C s~ d}|j r|| |j ¡7 }|d7 }|jr<|d| |j¡ 7 }|d7 }|jr^|d| |j¡ 7 }|d7 }|| j|jdd7 }|S )Nzfor (r r r Tr )rc r r Únextr r rl r r r Ú visit_For
|
s zCGenerator.visit_Forc C s: d}|j r|| |j ¡7 }|d7 }|| j|jdd7 }|S )Núwhile (r Tr )r r r r rl r r r Ú visit_While s zCGenerator.visit_Whilec C sJ d}|| j |jdd7 }|| ¡ d 7 }|jr>|| |j¡7 }|d7 }|S )Nzdo
|
Tr r¤ z);)r r r r r rl r r r Ú visit_DoWhile s zCGenerator.visit_DoWhilec C s> d}|| |j¡7 }|jr2|d7 }|| |j¡7 }|d7 }|S )Nz_Static_assert(ú,r0 )r r Úmessagerl r r r Úvisit_StaticAssert$ s z CGenerator.visit_StaticAssertc C s, d| |j¡ d }|| j|jdd7 }|S )Nzswitch (r Tr )r r r r rl r r r Ú visit_Switch- s zCGenerator.visit_Switchc C s6 d| |j¡ d }|jD ]}|| j|dd7 }q|S )Nzcase ú:
|
Tr )r r5 Ústmtsr ©r r re r r r r Ú
|
visit_Case2 s
|
zCGenerator.visit_Casec C s& d}|j D ]}|| j|dd7 }q
|
|S )Nz default:
|
Tr )r¬ r r r r r Ú visit_Default8 s
|
zCGenerator.visit_Defaultc C s |j d | |j¡ S )Nr« )r# r r r r r r Ú visit_Label> s zCGenerator.visit_Labelc C s d|j d S )Nzgoto r r" r r r r Ú
|
visit_GotoA s zCGenerator.visit_Gotoc C s dS )Nz...r r r r r Úvisit_EllipsisParamD s z CGenerator.visit_EllipsisParamc C s | |d¡S )NÚstructr| r r r r Ú visit_StructG s zCGenerator.visit_Structc C s | |j¡S r )ro r, r r r r Úvisit_TypenameJ s zCGenerator.visit_Typenamec C s | |d¡S )NÚunionr| r r r r Ú visit_UnionM s zCGenerator.visit_Unionc C sZ d}|j D ]6}t|tjr*|d|j 7 }q
|
|d| |¡ d 7 }q
|
|d| |j¡ 7 }|S )Nr Ú.r( r) ra )r# rK r ÚIDr r` r5 )r r re r# r r r Úvisit_NamedInitializerP s
|
z!CGenerator.visit_NamedInitializerc C s
|
| |¡S r ©ro r r r r Úvisit_FuncDeclZ s zCGenerator.visit_FuncDeclc C s | j |ddS ©NFrq r» r r r r Úvisit_ArrayDecl] s zCGenerator.visit_ArrayDeclc C s | j |ddS r½ r» r r r r Úvisit_TypeDecl` s zCGenerator.visit_TypeDeclc C s | j |ddS r½ r» r r r r Ú visit_PtrDeclc s zCGenerator.visit_PtrDeclc C s° |dkr|j }| j}n(|dks"t|jdkr0dn|jj}| j}|d |jpLd }|dk r¬|d7 }|| ¡ 7 }| jd7 _|d7 }|||7 }| jd8 _|| ¡ d 7 }|S )
|
zq Generates code for structs, unions, and enums. name should be
|
'struct', 'union', or 'enum'.
|
)r³ r¶ r{ Nr r r r7 r r^ ) rj Ú_generate_struct_union_bodyÚAssertionErrorÚvaluesZ enumeratorsÚ_generate_enum_bodyr# r r )r r r# ÚmembersZ body_functionre r r r r} f s z&CGenerator._generate_struct_union_enumc s d fdd|D ¡S )Nr c 3 s | ]} |¡V qd S r r rh r r r r s z9CGenerator._generate_struct_union_body.<locals>.<genexpr>©r ©r rÅ r r r rÁ ~ s z&CGenerator._generate_struct_union_bodyc s$ d fdd|D ¡d d
d S )Nr c 3 s | ]} |¡V qd S r r )r r r r r r s z1CGenerator._generate_enum_body.<locals>.<genexpr>éþÿÿÿr rÆ rÇ r r r rÄ s z CGenerator._generate_enum_bodyc C sÈ t |}|r| jd7 _| ¡ }|r4| jd8 _|tjtjtjtjtjtj tj
|
tj tj tj tjtjtjf kr|| |¡ d S |tjfkr| |¡S |tjfkr²|| |¡ S || |¡ d S dS )zÄ Generation from a statement node. This method exists as a wrapper
|
for individual visit_* methods to handle different treatment of
|
some statements in this context.
|
r7 r r N)r, r r r ÚDeclrX ZCastZUnaryOprL Z TernaryOpÚFuncCallÚArrayRefÚ StructRefÚConstantr¹ ZTypedefr_ r ZCompoundZIf)r r r Útypr r r r r
s6 ü
|
zCGenerator._generate_stmtc C sf d}|j rd |j ¡d }|jr4|d |j¡d 7 }|jrR|| |jd ¡d 7 }|| |j¡7 }|S )z& Generation from a Decl node.
|
r r r )Zfuncspecr rn Zalignr ro r, rl r r r rb £ s zCGenerator._generate_declTc C s t |}|tjkrd}|jr2|d |j¡d 7 }|| |j ¡7 }|jrR|rR|jnd}t|D ]\}}t|tj rÔ|dkrt||d tj
|
rd| d }|d7 }|j r¾|d |j ¡d 7 }|| |j ¡d 7 }q^t|tj r(|dkrt||d tj
|
rd| d }|d| |j¡ d 7 }q^t|tj
|
r^|jr`d d |j¡|rVd| ndf }q^d
|
| }q^|r||d| 7 }|S |tjkr| |j ¡S |tjkr´| j|j |d S |tjkrÐd |j¡d S |tj tj
|
tj fkrþ| j|j ||g |d S | |¡S d S ) zø Recursive generation from a type node. n is the type node.
|
modifiers collects the PtrDecl, ArrayDecl and FuncDecl modifiers
|
encountered on the way down to a TypeDecl, to allow proper
|
generation from it.
|
r r r r r/ r0 r( r) z* %s%srF rq N)r, r ZTypeDeclZqualsr r ZdeclnameÚ enumeraterK Z ArrayDeclZPtrDeclZ dim_qualsZdimZFuncDeclr1 rÉ rb ZTypenamero ZIdentifierTyper[ ) r r Ú modifiersrr rÎ re ZnstrÚiÚmodifierr r r ro sV ÿ
|
ÿ ÿ
|
ÿzCGenerator._generate_typec C s& | |¡}||r d| d S |S dS )z Visits 'n' and returns its string representation, parenthesized
|
if the condition function applied to the node returns True.
|
r/ r0 N)r` )r r Ú conditionre r r r rS â s
|
zCGenerator._parenthesize_ifc s | fdd¡S )z. Common use case for _parenthesize_if
|
c s | ¡ S r )rJ rN r r r rQ ï rY z8CGenerator._parenthesize_unless_simple.<locals>.<lambda>)rS r r r r r* ì s z&CGenerator._parenthesize_unless_simplec C s t |tjtjtjtjtjfS )z~ Returns True for nodes that are "simple" - i.e. nodes that always
|
have higher precedence than operators.
|
)rK r rÍ r¹ rË rÌ rÊ r r r r rJ ñ s ÿzCGenerator._is_simple_nodeN)F)F)F)Cr Ú
|
__module__Ú __qualname__Ú__doc__r
|
r r r r! r$ r' r+ r. r2 r6 rM rW rZ r\ r` rf rm rp rs ry 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º r¼ r¾ r¿ rÀ r} rÁ rÄ r rb ro rS r* rJ r r r r r s¤
|
ô
|
|
|
|
|
|
|
5
|
r N)r r Úobjectr r r r r Ú<module> s
|
