U Ú²g;ã@s0ddlmZdd„Zdd„Zdd„Zdd „Zd S) é)Úc_astcCsªt|tjƒst‚t|jtjƒs"|St g|jj¡}d}|jjp@gD]\}t|tjtj fƒr||j  |¡t ||jƒ|jd}qB|dkr’|j  |¡qB|j   |¡qB||_|S)aÜ The '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éÿÿÿÿ) Ú isinstancerZSwitchÚAssertionErrorZstmtZCompoundZcoordZ block_itemsÚCaseÚDefaultÚappendÚ_extract_nested_caseÚstmts)Z switch_nodeZ new_compoundZ last_caseÚchild©r úTypeDecl pairs and attaching the _Atomic qualifier in the right place. Ú_AtomicN) Ú_fix_atomic_specifiers_oncerrÚTypeDeclÚtypeÚAttributeErrorÚqualsrZdeclnameÚname)ÚdeclÚfoundÚtypr r r Úfix_atomic_specifiersls      rcCsš|}d}|j}|dk r`t|tjƒr.d|jkr.q`z|}|}|j}Wqtk r\|dfYSXqt|tjƒspt‚|j|_d|jjkr’|jj d¡|dfS)zƒ Performs one 'fix' round of atomic specifiers. Returns (modified_decl, found) where found is True iff a fix was made. NrFT) rrrZTypenamerrrrr)rÚparentZ grandparentÚnoder r r rŒs"  rN)Úrrr rrr r r r Ú s V