3 L]L9@s ddlmZGdddeZdS))c_astc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZdvd!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^Z1d_d`Z2dadbZ3dcddZ4dedfZ5dgdhZ6dwdidjZ7dkdlZ8gfdmdnZ9dodpZ:dqdrZ;dsdtZ#sz+CGenerator.generic_visit..)joinchildren)rrr)rrrszCGenerator.generic_visitcCs|jS)N)value)rnrrrvisit_Constant%szCGenerator.visit_ConstantcCs|jS)N)name)rrrrrvisit_ID(szCGenerator.visit_IDcCsd}|jr|d|j7}|S)Nz#pragmar )string)rrretrrr visit_Pragma+szCGenerator.visit_PragmacCs$|j|j}|d|j|jdS)N[])_parenthesize_unless_simplerrZ subscript)rrZarrrefrrrvisit_ArrayRef1s zCGenerator.visit_ArrayRefcCs"|j|j}||j|j|jS)N)r$rtyperfield)rrZsrefrrrvisit_StructRef5s zCGenerator.visit_StructRefcCs$|j|j}|d|j|jdS)N())r$rrargs)rrZfrefrrrvisit_FuncCall9s zCGenerator.visit_FuncCallcCs\|j|j}|jdkrd|S|jdkr0d|S|jdkrJd|j|jSd|j|fSdS)Nzp++z%s++zp--z%s--sizeofz sizeof(%s)z%s%s)r$expropr)rroperandrrr visit_UnaryOp=s    zCGenerator.visit_UnaryOpcs<j|jfdd}j|jfdd}d||j|fS)Ncs j| S)N)_is_simple_node)d)rrrLsz+CGenerator.visit_BinaryOp..cs j| S)N)r2)r3)rrrr4Nsz%s %s %s)_parenthesize_ifleftrightr/)rrZlval_strrval_strr)rrvisit_BinaryOpJs zCGenerator.visit_BinaryOpcCs*|j|jdd}d|j|j|j|fS)NcSs t|tjS)N) isinstancer Assignment)rrrrr4Tsz-CGenerator.visit_Assignment..z%s %s %s)r5ZrvaluerZlvaluer/)rrr8rrrvisit_AssignmentQs zCGenerator.visit_AssignmentcCs dj|jS)Nr )rnames)rrrrrvisit_IdentifierTypeWszCGenerator.visit_IdentifierTypecCsJt|tjrd|j|dSt|tjr.)rdeclslenr)rrrFr)rrvisit_DeclListls zCGenerator.visit_DeclListcCs2d}|jr|dj|jd7}||j|j7}|S)Nrr )storager_generate_typer&)rrrFrrr visit_Typedefss zCGenerator.visit_TypedefcCs(d|j|jd}|d|j|jS)Nr)r*r )rMZto_typer$r.)rrrFrrr visit_CastyszCGenerator.visit_CastcCs.g}x|jD]}|j|j|q Wdj|S)Nz, )exprsappendrBr)rrvisited_subexprsr.rrrvisit_ExprList}s zCGenerator.visit_ExprListcCs.g}x|jD]}|j|j|q Wdj|S)Nz, )rPrQrBr)rrrRr.rrrvisit_InitLists zCGenerator.visit_InitListcCs|j|ddS)Nenum)r)_generate_struct_union_enum)rrrrr visit_EnumszCGenerator.visit_EnumcCs<|jsdj|j|jdSdj|j|j|j|jdSdS)Nz{indent}{name}, )indentrz{indent}{name} = {value}, )rXrr)rformatr rr)rrrrrvisit_Enumerators zCGenerator.visit_Enumeratorcsjj|j}d_j|j}|jrVdjfdd|jD}|d|d|dS|d|dSdS)Nrz; c3s|]}j|VqdS)N)r)rp)rrrrsz+CGenerator.visit_FuncDef.. )rrHrbodyZ param_declsr)rrrHr]Zknrdeclsr)rr visit_FuncDefs  zCGenerator.visit_FuncDefcCsfd}x\|jD]R}t|tjr,||j|7}q t|tjrL||j|d7}q ||j|d7}q W|S)Nrr\z; )extr:rZFuncDefrZPragma)rrrFr_rrr 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)rrrFr)rrvisit_Compounds zCGenerator.visit_CompoundcCs$d|j|jd|j|jdS)Nr)z){r@)rr&rD)rrrrrvisit_CompoundLiteralsz CGenerator.visit_CompoundLiteralcCsdS)N;r)rrrrrvisit_EmptyStatementszCGenerator.visit_EmptyStatementcsdjfdd|jDS)Nz, c3s|]}j|VqdS)N)r)rparam)rrrrsz-CGenerator.visit_ParamList..)rparams)rrr)rrvisit_ParamListszCGenerator.visit_ParamListcCs&d}|jr|d|j|j7}|dS)Nreturnr rf)r.r)rrrFrrr visit_ReturnszCGenerator.visit_ReturncCsdS)Nzbreak;r)rrrrr visit_BreakszCGenerator.visit_BreakcCsdS)Nz continue;r)rrrrrvisit_ContinueszCGenerator.visit_ContinuecCsHd|j|jd}|d|j|jd7}|d|j|jd7}|S)Nr)z) ? z) : r*)rBcondiftrueiffalse)rrrFrrrvisit_TernaryOpszCGenerator.visit_TernaryOpcCsdd}|jr||j|j7}|d7}||j|jdd7}|jr`||jd7}||j|jdd7}|S)Nzif (z) T) add_indentzelse )rorrbrprqr )rrrFrrrvisit_IfszCGenerator.visit_IfcCs~d}|jr||j|j7}|d7}|jr<|d|j|j7}|d7}|jr^|d|j|j7}|d7}||j|jdd7}|S)Nzfor (rfr z) T)rs)rDrronextrbrc)rrrFrrr visit_ForszCGenerator.visit_ForcCs:d}|jr||j|j7}|d7}||j|jdd7}|S)Nzwhile (z) T)rs)rorrbrc)rrrFrrr visit_Whiles zCGenerator.visit_WhilecCsJd}||j|jdd7}||jd7}|jr>||j|j7}|d7}|S)Nzdo T)rszwhile (z);)rbrcr ror)rrrFrrr visit_DoWhileszCGenerator.visit_DoWhilecCs,d|j|jd}||j|jdd7}|S)Nzswitch (z) T)rs)rrorbrc)rrrFrrr visit_SwitchszCGenerator.visit_SwitchcCs:d|j|jd}x |jD]}||j|dd7}qW|S)Nzcase z: T)rs)rr.stmtsrb)rrrFrcrrr visit_Cases zCGenerator.visit_CasecCs*d}x |jD]}||j|dd7}q W|S)Nz default: T)rs)rzrb)rrrFrcrrr visit_Defaults zCGenerator.visit_DefaultcCs|jd|j|jS)Nz: )rrbrc)rrrrr visit_LabelszCGenerator.visit_LabelcCsd|jdS)Nzgoto rf)r)rrrrr visit_Goto szCGenerator.visit_GotocCsdS)Nz...r)rrrrrvisit_EllipsisParam szCGenerator.visit_EllipsisParamcCs |j|dS)Nstruct)rV)rrrrr visit_StructszCGenerator.visit_StructcCs |j|jS)N)rMr&)rrrrrvisit_TypenameszCGenerator.visit_TypenamecCs |j|dS)Nunion)rV)rrrrr visit_UnionszCGenerator.visit_UnioncCs^d}x@|jD]6}t|tjr,|d|j7}q |d|j|d7}q W|d|j|j7}|S)Nr.r"r#z = )rr:rIDrrBr.)rrrFrrrrvisit_NamedInitializers  z!CGenerator.visit_NamedInitializercCs |j|S)N)rM)rrrrrvisit_FuncDecl#szCGenerator.visit_FuncDeclcCs|d kr|j}|j}n(|dks"t|jdkr0dn|jj}|j}|d|jpLd}|dk r|d7}||j7}|jd7_|d 7}|||7}|jd8_||jd 7}|S) zq Generates code for structs, unions, and enums. name should be 'struct', 'union', or 'enum'. rrrUNr rr\raz{ r@)rr) rI_generate_struct_union_bodyAssertionErrorvaluesZ enumerators_generate_enum_bodyrr r)rrrmembersZ body_functionrFrrrrV&s    z&CGenerator._generate_struct_union_enumcsdjfdd|DS)Nrc3s|]}j|VqdS)N)rb)rrH)rrrr?sz9CGenerator._generate_struct_union_body..)r)rrr)rrr>sz&CGenerator._generate_struct_union_bodycs$djfdd|DdddS)Nrc3s|]}j|VqdS)N)r)rr)rrrrCsz1CGenerator._generate_enum_body..rar\)r)rrr)rrrAszCGenerator._generate_enum_bodycCst|}|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. raz; r\N)r&rr rDeclr;ZCastUnaryOpZBinaryOpZ TernaryOpFuncCallArrayRef StructRefConstantrZTypedefrArZCompound)rrrstyprXrrrrbEs   zCGenerator._generate_stmtcCsHd}|jrdj|jd}|jr4|dj|jd7}||j|j7}|S)z& Generation from a Decl node. rr )ZfuncspecrrLrMr&)rrrFrrrrCaszCGenerator._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*r"r#z* %s %s*N)r&rZTypeDeclZqualsrrZdeclname enumerater:Z ArrayDeclZPtrDeclZdimZFuncDeclr+rrCZTypenamerMZIdentifierTyper=)rr modifiersrrFZnstrimodifierrrrrMjs@            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)rB)rr conditionrFrrrr5s  zCGenerator._parenthesize_ifcsj|fddS)z. Common use case for _parenthesize_if cs j| S)N)r2)r3)rrrr4sz8CGenerator._parenthesize_unless_simple..)r5)rrr)rrr$sz&CGenerator._parenthesize_unless_simplecCst|tjtjtjtjtjfS)z~ Returns True for nodes that are "simple" - i.e. nodes that always have higher precedence than operators. )r:rrrrrr)rrrrrr2szCGenerator._is_simple_nodeN)F)F)=r __module__ __qualname____doc__r r rrrrr!r%r(r,r1r9r<r>rBrGrKrNrOrSrTrWrZr^r`rdrergrjrlrmrnrrrtrvrwrxryr{r|r}r~rrrrrrrVrrrbrCrMr5r$r2rrrrr st           . rN)rrobjectrrrrr s