o ⹏i @sddlmZmZmZmZddlmZdejdejfddZdej ej Bd eej dd fd d Z d ej ejBdej ejBfddZd ej ejBdeej ejBeffddZd S))AnyListTuplecast)c_ast switch_nodereturncCst|tjsJt|jtjs|Stg|jj}d}|jjp gD].}t|tjtjfr>|j |t ||j|jd}q!|durI|j |q!|j |q!||_|S)a3The 'case' statements in a 'switch' come out of parsing with one child node, so subsequent statements are just tucked to the parent Compound. Additionally, consecutive (fall-through) case statements come out messy. This is a peculiarity of the C grammar. The following: switch (myvar) { case 10: k = 10; p = k + 1; return 10; case 20: case 30: return 20; default: break; } Creates this tree (pseudo-dump): Switch ID: myvar Compound: Case 10: k = 10 p = k + 1 return 10 Case 20: Case 30: return 20 Default: break The goal of this transform is to fix this mess, turning it into the following: Switch ID: myvar Compound: Case 10: k = 10 p = k + 1 return 10 Case 20: Case 30: return 20 Default: break A fixed AST node is returned. The argument may be modified. N) isinstancerSwitchstmtCompoundcoord block_itemsCaseDefaultappend_extract_nested_casestmts)r new_compound last_casechildrU/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/pycparser/ast_transforms.pyfix_switch_casess3   r case_node stmts_listNcCsDt|jdtjtjfr |j}||ttt ||dSdS)z{Recursively extract consecutive Case statements that are made nested by the parser and add them to the stmts_list. rN) r rrrrpoprrrr)rrnestedrrrres   rdeclcCs t|\}}|s nq|}t|tjs*z|j}Wn ty#|YSwt|tjrd|jvr:d|jvr:|jd|jdurC|j |_|S)a:Atomic specifiers like _Atomic(type) are unusually structured, conferring a qualifier upon the contained type. This function fixes a decl with atomic specifiers to have a sane AST structure, by removing spurious Typename->TypeDecl pairs and attaching the _Atomic qualifier in the right place. T_AtomicN) _fix_atomic_specifiers_oncer rTypeDecltypeAttributeErrorqualsrdeclnamename)r foundtyprrrfix_atomic_specifiersqs$       r+cCs|}d}|j}|dur2t|tjrd|jvrnz |}|}|j}Wn ty-|dfYSw|dus t|tjs:J|dus@J|jtt|_d|jjvrT|jj d|dfS)z~Performs one 'fix' round of atomic specifiers. Returns (modified_decl, found) where found is True iff a fix was made. Nr!FT) r$r rTypenamer&r%r#rrr)r parent grandparentnoderrrr"s(     r")typingrrrrrr rrrNoderDeclTypedefr+boolr"rrrrs( V     "