o jP@s2ddlmZmZmZddlmZGdddZdS))CallableListOptional)c_astc @seZdZUdZeed<eed<ddeddfddZdefd d Z d e j defd d Z d e e j defddZde jdefddZde jdefddZde jdefddZde jdefddZde jdefddZde jdefddZde jdefddZidd d!d"d#d$d%d&d'd(d)d*d+d*d,d-d.d-d/d-d0d-d1d2d3d2d4d5d6d5d7d8d9d8d:d8iZde jdefd;d<Zde j defd=d>Z!de j"defd?d@Z#de j defdAdBZ$dde j%dCedefdDdEZ&de j'defdFdGZ(de j)defdHdIZ*de j+defdJdKZ,de j-defdLdMZ.de j/defdNdOZ0de j1defdPdQZ2de j3defdRdSZ4de j5defdTdUZ6de j7defdVdWZ8de j9defdXdYZ:de j;defdZd[Zde j?defd^d_Z@de jAdefd`daZBde jCdefdbdcZDde jEdefdddeZFde jGdefdfdgZHde jIdefdhdiZJde jKdefdjdkZLde jMdefdldmZNde jOdefdndoZPde jQdefdpdqZRde jSdefdrdsZTde jUdefdtduZVde jWdefdvdwZXde jYdefdxdyZZde j[defdzd{Z\de j]defd|d}Z^de j_defd~dZ`de jadefddZbde jcdefddZdde jedefddZfde jgdefddZhde jidefddZjde jkdefddZlde jmdefddZnde jodefddZpde jae jeBe j1BdedefddZqdere j defddZsdere j5defddZtdde j dedefddZude j%defddZvgdfde j dere j dedefddZwde j dexe j gefdefddZyde j defddZzde j defddZ{dS) CGeneratorzUses the same visitor pattern as c_ast.NodeVisitor, but modified to return a value from each visit method, using string accumulation in generic_visit. indent_levelreduce_parenthesesFreturnNcCsd|_||_dS)zConstructs C-code generator reduce_parentheses: if True, eliminates needless parentheses on binary operators rN)rr )selfr r /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/pycparser/c_generator.py__init__s zCGenerator.__init__cCs d|jSN )rr r r r _make_indent" zCGenerator._make_indentnodecCsd|jj}t|||j|S)Nvisit_) __class____name__getattr generic_visit)r rmethodr r r visit%s zCGenerator.visitcs(|durdSdfdd|DS)Nc3s|] \}}|VqdSNr).0c_namecrr r -sz+CGenerator.generic_visit..)joinchildren)r rr rr r)szCGenerator.generic_visitncC|jSr)valuer r%r r r visit_Constant/zCGenerator.visit_ConstantcCr&rnamer(r r r visit_ID2r*zCGenerator.visit_IDcCsd}|jr |d|j7}|S)Nz#pragmar)string)r r%retr r r visit_Pragma5szCGenerator.visit_PragmacCs$||j}|d||jdS)N[])_parenthesize_unless_simpler,r subscript)r r%arrrefr r r visit_ArrayRef;s zCGenerator.visit_ArrayRefcCs"||j}||j||jSr)r3r,typerfield)r r%srefr r r visit_StructRef?s zCGenerator.visit_StructRefcCs6||j}|jdur||jnd}|d|dS)Nr())r3r,argsr)r r%frefr=r r r visit_FuncCallCs zCGenerator.visit_FuncCallcCsz|jdkrd||jdSdkr!||j}|dSdkr/||j}|dS ||j}|j|S)Nsizeofzsizeof(r<zp++z++zp--z--)oprexprr3)r r%operandr r r visit_UnaryOpHs       zCGenerator.visit_UnaryOpz||rz&&r|^&z==z!=>z>=>z<<+-* /%csFjfdd}jfdd}|djd|S)Ncs6|pjot|tjoj|jjjk Sr_is_simple_noder isinstancerBinaryOpprecedence_maprAdr%r r r |  z+CGenerator.visit_BinaryOp..cs6|pjot|tjoj|jjjk SrrWr\r^r r r_r`r)_parenthesize_ifleftrightrA)r r%lval_strrval_strr r^r visit_BinaryOpps    zCGenerator.visit_BinaryOpcCs0||jdd}||jd|jd|S)NcSs t|tjSr)rYr Assignment)r%r r r r_ z-CGenerator.visit_Assignment..r)rarvaluerlvaluerA)r r%rer r r visit_Assignments zCGenerator.visit_AssignmentcCs d|jSr)r#namesr(r r r visit_IdentifierType zCGenerator.visit_IdentifierTypecCsn|tjdrd||dStjdrn tjdr$nn d||dS ||S)Nr {}r;r<)rInitListrExprListCompoundr(r r r _visit_exprs* zCGenerator._visit_exprno_typecCsL|r|jn||}|jr|d||j7}|jr$|d||j7}|S)Nz :  = )r,_generate_declbitsizerinitrt)r r%rusr r r visit_Decls zCGenerator.visit_DeclcsL|jd}t|jdkr$|ddfdd|jddD7}|S)Nrr, c3s|] }j|ddVqdS)T)ruN)r{rdeclrr r r"s z,CGenerator.visit_DeclList..)rdeclslenr#r r%rzr rr visit_DeclLists   zCGenerator.visit_DeclListcCs2d}|jr|d|jd7}|||j7}|S)Nrr)storager#_generate_typer7rr r r visit_Typedefs zCGenerator.visit_TypedefcCs,d|j|jddd}|d||jS)Nr;F emit_declnamer<r)rto_typer3rBrr r r visit_CastszCGenerator.visit_CastcC*g}|jD] }|||qd|SNr|exprsappendrtr#r r%visited_subexprsrBr r r visit_ExprList  zCGenerator.visit_ExprListcCrrrrr r r visit_InitListrzCGenerator.visit_InitListcC|j|ddS)Nenumr+_generate_struct_union_enumr(r r r visit_EnumzCGenerator.visit_EnumcCsd||jS)Nz _Alignas({}))formatr alignmentr(r r r visit_AlignasszCGenerator.visit_AlignascCs8|js dj||jdSdj||j||jdS)Nz{indent}{name}, )indentr,z{indent}{name} = {value}, )rr,r')r'rrr,rr(r r r visit_Enumerators zCGenerator.visit_Enumeratorcsf|j}d_|j}|jr+dfdd|jD}|d|d|dS|d|dS)Nr; c3|]}|VqdSrr)rprr r r"z+CGenerator.visit_FuncDef.. )rr~rbody param_declsr#)r r%r~rknrdeclsr rr visit_FuncDefs  zCGenerator.visit_FuncDefcCsnd}|jD]/}|tjdr|||7}ntjdr)|||d7}n  |||d7}q|S)Nrr rr)extrFuncDefrPragma)r r%rzrr r r visit_FileASTs  zCGenerator.visit_FileASTcs`d}jd7_|jr|dfdd|jD7}jd8_|d7}|S)N{ rFrc3rr_generate_stmt)rstmtrr r r"rz,CGenerator.visit_Compound..z} )rr block_itemsr#rr rr visit_Compounds zCGenerator.visit_CompoundcCs$d||jd||jdS)Nr;z){rp)rr7ryr(r r r visit_CompoundLiterals$z CGenerator.visit_CompoundLiteralcCdS)N;r r(r r r visit_EmptyStatementzCGenerator.visit_EmptyStatementcsdfdd|jDS)Nr|c3rrr)rparamrr r r" rz-CGenerator.visit_ParamList..)r#paramsr(r rr visit_ParamList szCGenerator.visit_ParamListcCs&d}|jr|d||j7}|dS)Nr rr)rBrrr r r visit_Return szCGenerator.visit_ReturncCr)Nzbreak;r r(r r r visit_BreakrzCGenerator.visit_BreakcCr)Nz continue;r r(r r r visit_ContinuerzCGenerator.visit_ContinuecCsHd||jd}|d||jd7}|d||jd7}|S)Nr;z) ? z) : r<)rtcondiftrueiffalserr r r visit_TernaryOpszCGenerator.visit_TernaryOpcCsdd}|jr |||j7}|d7}||j|jdd7}|jr0||d7}||j|jdd7}|S)Nzif () T add_indentzelse )rrrrrrrr r r visit_IfszCGenerator.visit_IfcCs~d}|jr |||j7}|d7}|jr|d||j7}|d7}|jr/|d||j7}|d7}||j|jdd7}|S)Nzfor (rrrTr)ryrrnextrrrr r r visit_For)szCGenerator.visit_ForcCs:d}|jr |||j7}|d7}||j|jdd7}|S)Nwhile (rTr)rrrrrr r r visit_While7s zCGenerator.visit_WhilecCsJd}||j|jdd7}||d7}|jr|||j7}|d7}|S)Nzdo Trrz);)rrrrrrr r r visit_DoWhile?szCGenerator.visit_DoWhilecCs>d}|||j7}|jr|d7}|||j7}|d7}|S)Nz_Static_assert(,r<)rrmessagerr r r visit_StaticAssertHszCGenerator.visit_StaticAssertcCs,d||jd}||j|jdd7}|S)Nzswitch (rTr)rrrrrr r r visit_SwitchQszCGenerator.visit_SwitchcCs6d||jd}|jD] }||j|dd7}q |S)Nzcase : Tr)rrBstmtsrr r%rzrr r r visit_CaseVs zCGenerator.visit_CasecCs&d}|jD] }||j|dd7}q|S)Nz default: Tr)rrrr r r visit_Default\s zCGenerator.visit_DefaultcCs|jd||jS)Nr)r,rrr(r r r visit_LabelbszCGenerator.visit_LabelcCsd|jdS)Nzgoto rr+r(r r r visit_GotoerzCGenerator.visit_GotocCr)Nz...r r(r r r visit_EllipsisParamhrzCGenerator.visit_EllipsisParamcC ||dS)Nstructrr(r r r visit_StructkrnzCGenerator.visit_StructcCs ||jSr)rr7r(r r r visit_TypenamenrnzCGenerator.visit_TypenamecCr)Nunionrr(r r r visit_UnionqrnzCGenerator.visit_UnioncCsZd}|jD]}t|tjr|d|j7}q|d||d7}q|d||j7}|S)Nr.r1r2rv)r,rYrIDrrtrB)r r%rzr,r r r visit_NamedInitializerts  z!CGenerator.visit_NamedInitializercCs ||Srrr(r r r visit_FuncDecl~rzCGenerator.visit_FuncDeclcCrNFrrr(r r r visit_ArrayDeclrzCGenerator.visit_ArrayDeclcCrrrr(r r r visit_TypeDeclrzCGenerator.visit_TypeDeclcCrrrr(r r r visit_PtrDeclrzCGenerator.visit_PtrDeclr,cCs|dvrt|tjtjfsJ|j}|j}n|dksJt|tjs$J|jdur+dn|jj}|j }|d|j p9d}|duri|d7}|| 7}|j d7_ |d7}|||7}|j d8_ || d 7}|S) zlGenerates code for structs, unions, and enums. name should be 'struct', 'union', or 'enum'. )rrrNrrrrFrrp) rYrStructUnionr_generate_struct_union_bodyEnumvalues enumerators_generate_enum_bodyr,rr)r r%r,members body_functionrzr r r rs$   z&CGenerator._generate_struct_union_enumrcsdfdd|DS)Nrc3rrrr}rr r r"rz9CGenerator._generate_struct_union_body..r#r rr rr rsz&CGenerator._generate_struct_union_bodycs$dfdd|DdddS)Nrc3rrr)rr'rr r r"rz1CGenerator._generate_enum_body..rrrr rr rs$zCGenerator._generate_enum_bodyrcCs|r |jd7_|}|r|jd8_|tjdr notjdr)nftjdr2n]tjdr;nTtjdrDnKtjdrMnBtj drVn9tj dr_n0tj drhn'tj drqntj drzntjdrn tjdrnn |||dStjdr||Stjdr|||S |||dS)zGeneration from a statement node. This method exists as a wrapper for individual visit_* methods to handle different treatment of some statements in this context. rFr rr)rrrDeclrgCastUnaryOprZ TernaryOpFuncCallArrayRef StructRefConstantrTypedefrrrrsIf)r r%rrr r r rs8  zCGenerator._generate_stmtcCsfd}|jr d|jd}|jr|d|jd7}|jr)|||jdd7}|||j7}|S)zGeneration from a Decl node.rrr)funcspecr#ralignrrr7rr r r rwszCGenerator._generate_declT modifiersrc Csz|tjdrd}|jr|d|jd7}|||j7}|jr(|r(|jnd}t|D]\}}|tjdrr|dkrOt ||dtj rOd|d}|d7}|j r`|d|j d7}|j d urm|||j 7}|d 7}nZtj dr|dkrt ||dtj rd|d}|jd ur||jnd}|d|d7}n'tj dr|jrd|j} |rd|nd} d | | }nd |}nq.|r|d|7}|Stjdr||jStjdr|j|j|d Stjdr d|jdStjdrntj drn tj dr'nn |j|j||g|d S ||S)zRecursive 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 rrrrr;r<r1Nr2z* rSr)rTypeDeclqualsr#rr7declname enumerate ArrayDeclrYPtrDecl dim_qualsdimFuncDeclr=rrwTypenamerIdentifierTyperl) r r%rrrznstrimodifierr=rsuffixr r r rsn            ) B zCGenerator._generate_type conditioncCs"||}||rd|dS|S)zVisits 'n' and returns its string representation, parenthesized if the condition function applied to the node returns True. r;r<)rt)r r%r rzr r r ra&s  zCGenerator._parenthesize_ifcs|fddS)z$Common use case for _parenthesize_ifcs | Sr)rXr\rr r r_4rhz8CGenerator._parenthesize_unless_simple..)rar(r rr r32sz&CGenerator._parenthesize_unless_simplecCst|tjtjtjtjtjfS)zyReturns True for nodes that are "simple" - i.e. nodes that always have higher precedence than operators. )rYrrrrrrr(r r r rX6szCGenerator._is_simple_node)F)|r __module__ __qualname____doc__int__annotations__boolrstrrrNoderrrrr)rr-rr0rr6rr:rr?rrDr[rZrfrgrkrrmrtrr{DeclListrrrrrrrrrqrrrAlignasr EnumeratorrrrFileASTrrsrCompoundLiteralrEmptyStatementr ParamListrReturnrBreakrContinuerrrrrForrWhilerDoWhiler StaticAssertrSwitchrCaserDefaultrLabelrGotor EllipsisParamrrrrrrrNamedInitializerrrrrrrrrrrrrrrrwrrrar3rXr r r r rs       %           * C  rN)typingrrrrrrr r r r s