3 gwU5@s ddlmZGdddeZdS))c_astc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZdldd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"dAdBZ#dCdDZ$dEdFZ%dGdHZ&dIdJZ'dKdLZ(dMdNZ)dOdPZ*dQdRZ+dSdTZ,dUdVZ-dWdXZ.dYdZZ/d[d\Z0d]d^Z1dmd_d`Z2dadbZ3gfdcddZ4dedfZ5dgdhZ6didjZ7dkS)n 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. cCs d|_dS)N) indent_level)selfr!/usr/lib/python3.6/c_generator.py__init__szCGenerator.__init__cCs d|jS)N )r)rrrr _make_indentszCGenerator._make_indentcCsd|jj}t|||j|S)NZvisit_) __class____name__getattr generic_visit)rnodemethodrrrvisits zCGenerator.visitcs,|dkr dSdjfdd|jDSdS)Nc3s|]\}}j|VqdS)N)r).0Zc_namec)rrr #sz+CGenerator.generic_visit..)joinZchildren)rrr)rrrszCGenerator.generic_visitcCs|jS)N)value)rnrrrvisit_Constant%szCGenerator.visit_ConstantcCs|jS)N)name)rrrrrvisit_ID(szCGenerator.visit_IDcCs$|j|j}|d|j|jdS)N[])_parenthesize_unless_simplerrZ subscript)rrZarrrefrrrvisit_ArrayRef+s zCGenerator.visit_ArrayRefcCs"|j|j}||j|j|jS)N)rrtyperZfield)rrZsrefrrrvisit_StructRef/s zCGenerator.visit_StructRefcCs$|j|j}|d|j|jdS)N())rrrargs)rrZfrefrrrvisit_FuncCall3s zCGenerator.visit_FuncCallcCs\|j|j}|jdkrd|S|jdkr0d|S|jdkrJd|j|jSd|j|fSdS)Nzp++z%s++zp--z%s--Zsizeofz sizeof(%s)z%s%s)rexpropr)rrZoperandrrr visit_UnaryOp7s    zCGenerator.visit_UnaryOpcs<j|jfdd}j|jfdd}d||j|fS)Ncs j| S)N)_is_simple_node)d)rrrFsz+CGenerator.visit_BinaryOp..cs j| S)N)r*)r+)rrrr,Hsz%s %s %s)_parenthesize_ifleftrightr()rrZlval_strrval_strr)rrvisit_BinaryOpDs zCGenerator.visit_BinaryOpcCs*|j|jdd}d|j|j|j|fS)NcSs t|tjS)N) isinstancer Assignment)rrrrr,Nsz-CGenerator.visit_Assignment..z%s %s %s)r-ZrvaluerZlvaluer()rrr0rrrvisit_AssignmentKs zCGenerator.visit_AssignmentcCs dj|jS)Nr )rnames)rrrrrvisit_IdentifierTypeQszCGenerator.visit_IdentifierTypecCsJt|tjrd|j|dSt|tjr.)rdeclslenr)rrr>r)rrvisit_DeclListfs zCGenerator.visit_DeclListcCs2d}|jr|dj|jd7}||j|j7}|S)Nrr )storager_generate_typer!)rrr>rrr visit_Typedefms zCGenerator.visit_TypedefcCs(d|j|jd}|d|j|jS)Nr#r$r )rEZto_typerr')rrr>rrr visit_CastsszCGenerator.visit_CastcCs.g}x|jD]}|j|j|q Wdj|S)Nz, )exprsappendr:r)rrvisited_subexprsr'rrrvisit_ExprListws zCGenerator.visit_ExprListcCs.g}x|jD]}|j|j|q Wdj|S)Nz, )rHrIr:r)rrrJr'rrrvisit_InitList}s zCGenerator.visit_InitListcCsd}|jr|d|j7}|jr|d7}xXt|jjD]H\}}||j7}|jr`|d|j|j7}|t|jjdkr4|d7}q4W|d7}|S)Nenumr z {z = rz, r8)rvalues enumerateZ enumeratorsrrrB)rrr>iZ enumeratorrrr visit_Enums  zCGenerator.visit_Enumcsjj|j}d_j|j}|jrVdjfdd|jD}|d|d|dS|d|dSdS)Nrz; c3s|]}j|VqdS)N)r)rp)rrrrsz+CGenerator.visit_FuncDef.. )rr@rbodyZ param_declsr)rrr@rTZknrdeclsr)rr visit_FuncDefs  zCGenerator.visit_FuncDefcCsFd}x<|jD]2}t|tjr,||j|7}q ||j|d7}q W|S)Nrz; )extr2rZFuncDefr)rrr>rVrrr visit_FileASTs   zCGenerator.visit_FileASTcs`jd}jd7_|jr>|djfdd|jD7}jd8_|jd7}|S)Nz{ rc3s|]}j|VqdS)N)_generate_stmt)rstmt)rrrrsz,CGenerator.visit_Compound..z} )r rZ block_itemsr)rrr>r)rrvisit_Compounds zCGenerator.visit_CompoundcCsdS)N;r)rrrrrvisit_EmptyStatementszCGenerator.visit_EmptyStatementcsdjfdd|jDS)Nz, c3s|]}j|VqdS)N)r)rZparam)rrrrsz-CGenerator.visit_ParamList..)rZparams)rrr)rrvisit_ParamListszCGenerator.visit_ParamListcCs&d}|jr|d|j|j7}|dS)Nreturnr r\)r'r)rrr>rrr visit_ReturnszCGenerator.visit_ReturncCsdS)Nzbreak;r)rrrrr visit_BreakszCGenerator.visit_BreakcCsdS)Nz continue;r)rrrrrvisit_ContinueszCGenerator.visit_ContinuecCs8|j|jd}||j|jd7}||j|j7}|S)Nz ? z : )r:condiftrueiffalse)rrr>rrrvisit_TernaryOpszCGenerator.visit_TernaryOpcCsdd}|jr||j|j7}|d7}||j|jdd7}|jr`||jd7}||j|jdd7}|S)Nzif (z) T) add_indentzelse )rcrrYrdrer )rrr>rrrvisit_IfszCGenerator.visit_IfcCs~d}|jr||j|j7}|d7}|jr<|d|j|j7}|d7}|jr^|d|j|j7}|d7}||j|jdd7}|S)Nzfor (r\r z) T)rg)r<rrcnextrYrZ)rrr>rrr visit_ForszCGenerator.visit_ForcCs:d}|jr||j|j7}|d7}||j|jdd7}|S)Nzwhile (z) T)rg)rcrrYrZ)rrr>rrr visit_Whiles zCGenerator.visit_WhilecCsJd}||j|jdd7}||jd7}|jr>||j|j7}|d7}|S)Nzdo T)rgzwhile (z);)rYrZr rcr)rrr>rrr visit_DoWhileszCGenerator.visit_DoWhilecCs,d|j|jd}||j|jdd7}|S)Nzswitch (z) T)rg)rrcrYrZ)rrr>rrr visit_SwitchszCGenerator.visit_SwitchcCs:d|j|jd}x |jD]}||j|dd7}qW|S)Nzcase z: T)rg)rr'stmtsrY)rrr>rZrrr visit_Cases zCGenerator.visit_CasecCs*d}x |jD]}||j|dd7}q W|S)Nz default: T)rg)rnrY)rrr>rZrrr visit_Defaults zCGenerator.visit_DefaultcCs|jd|j|jS)Nz: )rrYrZ)rrrrr visit_LabelszCGenerator.visit_LabelcCsd|jdS)Nzgoto r\)r)rrrrr visit_GotoszCGenerator.visit_GotocCsdS)Nz...r)rrrrrvisit_EllipsisParamszCGenerator.visit_EllipsisParamcCs |j|dS)Nstruct)_generate_struct_union)rrrrr visit_StructszCGenerator.visit_StructcCs |j|jS)N)rEr!)rrrrrvisit_TypenameszCGenerator.visit_TypenamecCs |j|dS)Nunion)ru)rrrrr visit_UnionszCGenerator.visit_UnioncCsfd}xH|jD]>}t|tjr,|d|j7}q t|tjr |d|jd7}q W|d|j|j7}|S)Nr.rrz = )rr2rIDConstantrrr')rrr>rrrrvisit_NamedInitializer s   z!CGenerator.visit_NamedInitializercCs |j|S)N)rE)rrrrrvisit_FuncDeclszCGenerator.visit_FuncDeclcCs|d|jpd}|jr~|d7}||j7}|jd7_|d7}x|jD]}||j|7}qJW|jd8_||jd7}|S)ze Generates code for structs and unions. name should be either 'struct' or union. r rrSrXz{ r8)rrAr rrY)rrrr>r@rrrrus  z!CGenerator._generate_struct_unioncCst|}|r|jd7_|j}|r4|jd8_|tjtjtjtjtjtj tj tj tj tj tjtjtjf kr||j|dS|tjfkr|j|S||j|dSdS)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. rXz; rSN)r!rr rDeclr3ZCastZUnaryOpZBinaryOpZ TernaryOpFuncCallArrayRef StructRefr|r{ZTypedefr9rZCompound)rrrgtypindentrrrrY(s   zCGenerator._generate_stmtcCsHd}|jrdj|jd}|jr4|dj|jd7}||j|j7}|S)z& Generation from a Decl node. rr )ZfuncspecrrDrEr!)rrr>rrrr;DszCGenerator._generate_declcCst|}|tjkrNd}|jr2|dj|jd7}||j|j7}|jrN|jnd}xt|D]\}}t|tj r|dkrt||dtj rd|d}|d|j|j d7}q\t|tj r|dkrt||dtj rd|d}|d|j|j d7}q\t|tj r\|jr,d dj|j|f}q\d |}q\W|rJ|d|7}|S|tjkrf|j|jS|tjkr~|j|jS|tjkrdj|jdS|tj tj tj fkr|j|j||gS|j|Sd 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. rr rrr#r$rrz* %s %s*N)r!rZTypeDeclZqualsrrZdeclnamerOr2Z ArrayDeclZPtrDeclZdimZFuncDeclr%rr;ZTypenamerEZIdentifierTyper5)rrZ modifiersrr>ZnstrrPZmodifierrrrrEMs@            zCGenerator._generate_typecCs&|j|}||rd|dS|SdS)z Visits 'n' and returns its string representation, parenthesized if the condition function applied to the node returns True. r#r$N)r:)rrZ conditionr>rrrr-{s  zCGenerator._parenthesize_ifcsj|fddS)z. Common use case for _parenthesize_if cs j| S)N)r*)r+)rrrr,sz8CGenerator._parenthesize_unless_simple..)r-)rrr)rrrsz&CGenerator._parenthesize_unless_simplecCst|tjtjtjtjtjfS)z~ Returns True for nodes that are "simple" - i.e. nodes that always have higher precedence than operators. )r2rr|r{rrr)rrrrrr*szCGenerator._is_simple_nodeN)F)F)8r __module__ __qualname____doc__r r rrrrr r"r&r)r1r4r6r:r?rCrFrGrKrLrQrUrWr[r]r^r`rarbrfrhrjrkrlrmrorprqrrrsrvrwryr}r~rurYr;rEr-rr*rrrrr sj          . rN)rrobjectrrrrr s