Pi *ddlmZmZmZmZddlmZdejdejfdZdej ej zdeej dd fd Z d ej ejzdej ejzfd Zd ej ejzdeej ejzeffd Zd S))AnyListTuplecast)c_ast switch_nodereturnc&t|tjsJt|jtjs|Stjg|jj}d}|jjpgD]}t|tjtjfr=|j |t||j|jd}e||j ||j |||_|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_casechilds l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pycparser/ast_transforms.pyfix_switch_casesrsf k5< 0 000 0 k& 7 7>"k&6&<==L48I !-3.. eej%-8 9 9 .  $ + +E 2 2 2  (@ A A A$04II (//6666&&u----#K  case_node stmts_listNct|jdtjtjfrS|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)rrnesteds rrresw )/!$uz5=&ABB<$$&&&!!!T#v.. ;;;;;<TypeDecl pairs and attaching the _Atomic qualifier in the right place. T_Atomic) _fix_atomic_specifiers_oncer rTypeDecltypeAttributeErrorqualsrdeclnamename)r#foundtyps rfix_atomic_specifiersr/qs1$77 e  Cen-- (CC   KKK en-- CI)4:"="= )$$$ |y Ks= A  A c|}d}|j}|Gt|tjr d|jvrn# |}|}|j}n#t $r|dfcYSwxYw|Gt|tjsJ|J|jtt|_d|jjvr|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 grandparentnodes rr&r&s FK D   dEN + +  TZ0G0G   KF9DD   ;        fen - --- -  " " ""&)Dk '' y))) :s ?AA)typingrrrrrrrrrNoderDeclTypedefr/boolr&rrr<s'*)))))))))))S%,S5<SSSSl <zEM) <7;EJ7G <  < < < < *u} $ Z%-D *u} $ 5: %t +,r