j ddlmZddlmZmZddlmZdejdejfdZdejejzd e ejdd fd Z d ejejzej zdejejzej zfd Zd ejejzej zdeejejzej zeffdZy ))deepcopy)Anycast)c_ast switch_nodereturncBt|tjsJt|jtjrB|j}|j|_t ||j|_||_|St|jtj s|Stj g|jj}d}|jjxsgD]}t|tjtjfrA|jj|t||j|jd}n||jj||jj|||_|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) isinstancerSwitchstmtAttributedStatementfix_switch_casesCompoundcoord block_itemsCaseDefaultappend_extract_nested_casestmts)r attributed new_compound last_casechilds e/home/oem/CyberBrain26-unified/dev-env/.venv/lib/python3.12/site-packages/pycparser/ast_transforms.pyrrsUf k5<< 00 0+""E$=$=> %% %?? %%** %  k&& 7>>"k&6&6&<&<=L48I !!--3. eejj%--8 9  $ $ + +E 2  (@(@ A$004I ((//6&&u-."$K  case_node stmts_listNct|jdtjtjfrF|jj }|j |ttt||yy)z{Recursively extract consecutive Case statements that are made nested by the parser and add them to the stmts_list. rN) r rrrrpoprrrr)rr nesteds rrrms[ )//!$uzz5==&AB$$&&!T#v. ;Crdeclc8 t|\}}|sn|}t|tjs( |j}t|tjs(|j dd|_|j|j|_|S#t $r|cYSwxYw)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. N) _fix_atomic_specifiers_oncer rTypeDecltypeAttributeErrorqualsdeclnamename)r$foundtyps rfix_atomic_specifiersr/ys 1$7 e  Cenn- ((Cenn- 1DJ ||yy K  K s B BBc|}d}|j}|=t|tjrd|jvrn |}|}|j}|=t|tj sJ|Jt|jtjtjfrddl m }||jdt|j}|j|j|_ |j|jz|_d|jvr|jjd|tt|_|dfS#t $r|dfcYSwxYw)z~Performs one 'fix' round of atomic specifiers. Returns (modified_decl, found) where found is True iff a fix was made. N_AtomicFr) ParseErrorz5: _Atomic cannot be applied to array or function typeT)r(r rTypenamer*r)r' ArrayDeclFuncDeclc_parserr2rrrrr)r$parent grandparentnoder2r.s rr&r&sH FK D   dENN + TZZ0G   KF99D   fenn -- -  "" "$))eoou~~>?(zzlO P  499 C yyLL  syy(CI ! #"%Dk :? ;   sE E! E!)copyrtypingrrrr rrrlistNoderDeclTypedefr3r/tupleboolr&rrrDsZ%,,Z5<<Zz <zzEMM) <7;EJJ7G <  < **u}} $u~~ 5 ZZ%--%..0B/ **u}} $u~~ 5/ 5:: % 6 <=/r