Ѯh;@sDddlmZddZddZddZdd Zd S) )c_astcCst|tjstt|jtjs1|Stjg|jj}d}x|jjpagD]}t|tjtj fr|jj |t ||j|jd}qb|dkr|jj |qb|j j |qbW||_|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. Nr) isinstancerSwitchAssertionErrorstmtCompoundcoord block_itemsCaseDefaultappend_extract_nested_casestmts)Z switch_nodeZ new_compoundZ last_casechildr../pycparser/ast_transforms.pyfix_switch_cases s3  rcCsMt|jdtjtjfrI|j|jjt|d|dS)z Recursively extract consecutive Case statements that are made nested by the parser and add them to the stmts_list. rNr)rrrr r r popr)Z case_nodeZ stmts_listrrrrcs"rc Csxt|\}}|sPqW|}x<t|tjsdy |j}Wq)tk r`|SYq)Xq)Wd|jkrd|jkr|jjd|jdkr|j |_|S)aK 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. _AtomicN) _fix_atomic_specifiers_oncerrTypeDecltypeAttributeErrorqualsr declnamename)declfoundtyprrrfix_atomic_specifiersls     r!c Cs|}d}|j}xj|dk rt|tjrFd|jkrFPy|}|}|j}Wqtk r}|dfSYqXqWt|tjst|j|_d|jjkr|jjjd|dfS)z Performs one 'fix' round of atomic specifiers. Returns (modified_decl, found) where found is True iff a fix was made. NrFT) rrrTypenamerrrrr )rparentZ grandparentnoderrrrs" !   rN)rrrr!rrrrr s V