בjnddlmZddlmZmZddlmZdejdejfdZdej ej zd e ej dd fd Z d ejejzejzdejejzejzfd Zd ejejzejzdeejejzejzeffdZd S))deepcopy)Anycast)c_ast switch_nodereturnct|tjsJt|jtjr7|j}|j|_t ||j|_||_|St|jtjs|Stjg|jj}d}|jjpgD]}t|tj tj fr=|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) isinstancerSwitchstmtAttributedStatementfix_switch_casesCompoundcoord block_itemsCaseDefaultappend_extract_nested_casestmts)r attributed new_compound last_casechilds /Users/tedpeters/Documents/Projects/Claude Projects/TPC Houses/house-viewer/.venv/lib/python3.11/site-packages/pycparser/ast_transforms.pyrrsef k5< 0 000 0+"E$=>> % %? %%%%* %  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)rr nesteds rrrmsw )/!$uz5=&ABB<$$&&&!!!T#v.. ;;;;;<TypeDecl pairs and attaching the _Atomic qualifier in the right place. TN) _fix_atomic_specifiers_oncer rTypeDecltypeAttributeErrorqualsdeclnamename)r$foundtyps rfix_atomic_specifiersr/ys1$77 e  Cen-- (CC   KKK en-- 111DJ |y Ks= A  A cj|}d}|j}|Gt|tjr d|jvrn# |}|}|j}n#t $r|dfcYSwxYw|Gt|tjsJ|Jt|jtjtjfrddl m }||j dt|j}|j |j |_ |j|jz|_d|jvr|j d|tt|_|dfS)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&sy FK D   dEN + +  TZ0G0G   KF9DD   ;        fen - --- -  " " "$)eou~>?? ((((((jz P P P   49  C yL  sy(CI !! ###"%Dk :s ?AA)copyrtypingrrrr rrrlistNoderDeclTypedefr3r/tupleboolr&rrrDsIZ%,Z5<ZZZZz <zEM) <7;EJ7G <  < < < < *u} $u~ 5 Z%-%.0B/ *u} $u~ 5/ 5: % 6 <=//////r