o iP@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 [C:\Users\psant\Downloads\PAF backend\school\venv\Lib\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).0Zc_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+rZ subscript)r r$Zarrrefr r r visit_ArrayRef;s zCGenerator.visit_ArrayRefcCs"||j}||j||jSr)r2r+typerfield)r r$Zsrefr r r visit_StructRef?s zCGenerator.visit_StructRefcCs6||j}|jdur||jnd}|d|dS)Nr())r2r+argsr)r r$Zfrefr9r r r visit_FuncCallCs zCGenerator.visit_FuncCallcCsz|jdkrd||jdSdkr!||j}|dSdkr/||j}|dS ||j}|j|S)Nsizeofzsizeof(r8zp++z++zp--z--)oprexprr2)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_mapr<dr$r r r |  z+CGenerator.visit_BinaryOp..cs6|pjot|tjoj|jjjk SrrRrWrYr r rZr[r)_parenthesize_ifleftrightr<)r r$Zlval_strrval_strr rYr visit_BinaryOpps    zCGenerator.visit_BinaryOpcCs0||jdd}||jd|jd|S)NcSs t|tjSr)rTr Assignment)r$r r r rZ z-CGenerator.visit_Assignment..r)r\ZrvaluerZlvaluer<)r r$r_r 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 {}r7r8)rInitListrExprListCompoundr'r r r _visit_exprs* zCGenerator._visit_exprno_typecCsL|r|jn||}|jr|d||j7}|jr$|d||j7}|S)Nz :  = )r+_generate_declZbitsizerinitrl)r r$rmsr r r visit_Decls zCGenerator.visit_DeclcsL|jd}t|jdkr$|ddfdd|jddD7}|S)Nrr, c3s|] }j|ddVqdS)T)rmN)rrrdeclrr r r!s z,CGenerator.visit_DeclList..)rdeclslenr"r r$rqr rr visit_DeclLists   zCGenerator.visit_DeclListcCs2d}|jr|d|jd7}|||j7}|S)Nrr)storager"_generate_typer4rxr r r visit_Typedefs zCGenerator.visit_TypedefcCs,d|j|jddd}|d||jS)Nr7F emit_declnamer8r)r{Zto_typer2r=rxr r r visit_CastszCGenerator.visit_CastcC*g}|jD] }|||qd|SNrsexprsappendrlr"r r$Zvisited_subexprsr=r 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.. )rrurbodyZ param_declsr")r r$rurZknrdeclsr rr visit_FuncDefs  zCGenerator.visit_FuncDefcCsnd}|jD]/}|tjdr|||7}qtjdr)|||d7}q |||d7}q|S)Nrr rr)extrFuncDefrPragma)r r$rqrr r r visit_FileASTs  zCGenerator.visit_FileASTcs`d}jd7_|jr|dfdd|jD7}jd8_|d7}|S)N{ rArc3rr_generate_stmt)rstmtrr r r!rz,CGenerator.visit_Compound..z} )rrZ block_itemsr"rxr rr visit_Compounds zCGenerator.visit_CompoundcCs$d||jd||jdS)Nr7z){rh)rr4rpr'r r r visit_CompoundLiterals$z CGenerator.visit_CompoundLiteralcCdS)N;r r'r r r visit_EmptyStatementzCGenerator.visit_EmptyStatementcsdfdd|jDS)Nrsc3rrr)rparamrr r r! rz-CGenerator.visit_ParamList..)r"paramsr'r rr visit_ParamList szCGenerator.visit_ParamListcCs&d}|jr|d||j7}|dS)Nr rr)r=rrxr 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)Nr7z) ? z) : r8)rlcondiftrueiffalserxr r r visit_TernaryOpszCGenerator.visit_TernaryOpcCsdd}|jr |||j7}|d7}||j|jdd7}|jr0||d7}||j|jdd7}|S)Nzif () T add_indentzelse )rrrrrrrxr r r visit_IfszCGenerator.visit_IfcCs~d}|jr |||j7}|d7}|jr|d||j7}|d7}|jr/|d||j7}|d7}||j|jdd7}|S)Nzfor (rrrTr)rprrnextrrrxr r r visit_For)szCGenerator.visit_ForcCs:d}|jr |||j7}|d7}||j|jdd7}|S)Nwhile (rTr)rrrrrxr r r visit_While7s zCGenerator.visit_WhilecCsJd}||j|jdd7}||d7}|jr|||j7}|d7}|S)Nzdo Trrz);)rrrrrrxr r r visit_DoWhile?szCGenerator.visit_DoWhilecCs>d}|||j7}|jr|d7}|||j7}|d7}|S)Nz_Static_assert(,r8)rrmessagerxr r r visit_StaticAssertHszCGenerator.visit_StaticAssertcCs,d||jd}||j|jdd7}|S)Nzswitch (rTr)rrrrrxr r r visit_SwitchQszCGenerator.visit_SwitchcCs6d||jd}|jD] }||j|dd7}q |S)Nzcase : Tr)rr=stmtsrr r$rqrr 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_StructkrfzCGenerator.visit_StructcCs ||jSr)r{r4r'r r r visit_TypenamenrfzCGenerator.visit_TypenamecCr)Nunionrr'r r r visit_UnionqrfzCGenerator.visit_UnioncCsZd}|jD]}t|tjr|d|j7}q|d||d7}q|d||j7}|S)Nr.r0r1rn)r+rTrIDrrlr=)r r$rqr+r r r visit_NamedInitializerts  z!CGenerator.visit_NamedInitializercCs ||Srr{r'r r r visit_FuncDecl~rzCGenerator.visit_FuncDeclcCrNFr}rr'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'. )rrrNrrrrArrh) rTrStructUnionrv_generate_struct_union_bodyEnumvaluesZ enumerators_generate_enum_bodyr+rr)r r$r+membersZ body_functionrqr r r rs$   z&CGenerator._generate_struct_union_enumrcsdfdd|DS)Nrc3rrrrtrr 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. rAr rrN)rrrDeclraCastUnaryOprU TernaryOpFuncCallArrayRef StructRefConstantrTypedefrjrrkIf)r r$rrr r r rs8  zCGenerator._generate_stmtcCsfd}|jr d|jd}|jr|d|jd7}|jr)|||jdd7}|||j7}|S)zGeneration from a Decl node.rrrN)Zfuncspecr"rzZalignrr{r4rxr r r roszCGenerator._generate_declT modifiersr~c 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}q.tj dr|dkrt ||dtj rd|d}|jd ur||jnd}|d|d7}q.tj dr|jrd|j} |rd|nd} d | | }q.d |}q.q.|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 rrrrr7r8r0Nr1z* rNr})rTypeDeclqualsr"rr4Zdeclname enumerate ArrayDeclrTPtrDeclZ dim_qualsZdimFuncDeclr9rroTypenamer{IdentifierTyperd) r r$rr~rqZnstrimodifierr9rsuffixr r r r{sl          ) 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. r7r8N)rl)r r$rrqr r r r\&s  zCGenerator._parenthesize_ifcs|fddS)z$Common use case for _parenthesize_ifcs | Sr)rSrWrr r rZ4rbz8CGenerator._parenthesize_unless_simple..N)r\r'r rr r22sz&CGenerator._parenthesize_unless_simplecCst|tjtjtjtjtjfS)zyReturns True for nodes that are "simple" - i.e. nodes that always have higher precedence than operators. N)rTrrrrrrr'r r r rS6szCGenerator._is_simple_node)F)|r __module__ __qualname____doc__int__annotations__boolrstrrrNoderrrrr(rr,rr/rr3rr6rr:rr?rVrUr`rarcrrerlrrrZDeclListryrr|rrrjrrirrrZAlignasrZ EnumeratorrrrZFileASTrrkrZCompoundLiteralrZEmptyStatementrZ ParamListrReturnrBreakrContinuerrrrrForrWhilerZDoWhilerZ StaticAssertrZSwitchrZCaserZDefaultrZLabelrZGotorZ EllipsisParamrrrrrrrZNamedInitializerrrrrrrrrrrrrrrror{rr\r2rSr r r r rs       %           * C  rN)typingrrrrrrr r r r s