o hj@sdZddlmZddlmZmZGdddZGdddeZeGdd d eZeGd d d eZ eGd d d eZ eGdddeZ eGdddeZ eGdddeZ eGdddeZeGdddeZGdddeZeGdddeZeGdddeZeGdddeZeGd d!d!eZeGd"d#d#eZeGd$d%d%eZeGd&d'd'eZeGd(d)d)eZeGd*d+d+eZeGd,d-d-eZeGd.d/d/eZeGd0d1d1eZeGd2d3d3eZeGd4d5d5eZeZ eGd6d7d7eZ!eGd8d9d9eZ"eGd:d;d;eZ#eGdd?d?eZ%eGd@dAdAeZ&eGdBdCdCeZ'eGdDdEdEeZ(dFS)GaUAST node definitions for MicroBasic. Shape follows the confirmed grammar in docs/reverse_engineering.md plus the manual's documented syntax for constructs whose bytecode we didn't need to special-case at the AST level (e.g. compound assignment desugars to a plain AssignStmt at parse time -- see parser.py -- so there's no AugAssign node). ) annotations) dataclassfieldc@eZdZdZdS)NodezBase class for all AST nodes.N__name__ __module__ __qualname____doc__r r P/sessions/nice-awesome-ritchie/mnt/compiler/src/microbasic_compiler/ast_nodes.pyrsrc@ eZdZdS)ExprNrr r r r r r rrc@eZdZUded<dS) IntLiteralintvalueNrr r __annotations__r r r r r rc@r) BoolLiteralboolrNrr r r r rrrc@r) StringLiteralstrrNrr r r r r"rrc@s"eZdZUded<dZded<dS)VarRefrnameN Expr | Noneindex)rr r rr r r r r r's rc@seZdZUded<ded<dS)UnaryOproproperandNrr r r r r!-s  r!c@s&eZdZUded<ded<ded<dS)BinaryOprr"rleftrightNrr r r r r$3s  r$c@,eZdZUdZded<eedZded<dS)CallzA function call used as an expression, e.g. GetValue(_TEMP, 1). Also doubles as a statement (SetCommand(...), SetConfig(...), Wait(...)) when wrapped in ExprStmt -- see below. rrdefault_factory list[Expr]argsN)rr r r rrlistr,r r r r r(:s r(c@s*eZdZUdZded<ded<ded<dS) IncDecExprz8++x / --x / x++ / x-- used where the result is consumed.rvarr"rprefixNrr r r rr r r r r.Es  r.c@r)StmtNrr r r r r2Orr2c@ eZdZUeedZded<dS)Programr) list[Stmt] statementsN)rr r rr-r6rr r r r r4S r4c@r)LabelrrNrr r r r r8Xrr8c@s*eZdZUded<ded<dZded<dS)DimStmtrrvar_typeNz int | None array_len)rr r rr;r r r r r9]s r9c@r)OptionExplicitStmtzx`Option Explicit` directive. Accepted and currently a no-op -- codegen doesn't yet enforce declare-before-use. TODO.Nrr r r r r<dsr<c@"eZdZUdZded<ded<dS) AssignStmta0Always a plain '=' by the time this reaches codegen -- compound assignment operators (+=, -=, *=, /=, <<=, >>=) are desugared into this (target = target op value) during parsing, since that's confirmed to be exactly how the real compiler treats them (see reverse_engineering.md, batch 3).rtargetrrNr1r r r r r>js  r>c@s<eZdZUded<ded<eedZded<dZd ed <dS) IfStmtr conditionr5 then_blockr)zlist[tuple[Expr, list[Stmt]]]elseif_clausesNzlist[Stmt] | None else_block)rr r rrr-rCrDr r r r r@us r@c@r)GotoStmtrlabelNrr r r r rE}rrEc@r) GosubStmtrrFNrr r r r rGrrGc@r) ReturnStmtNrr r r r rHrHc@r) TerminateStmtNrr r r r rJrIrJc@r)WaitStmtr millisecondsNrr r r r rKrrKc@r3) PrintStmtr)r+r,N)rr r rr-r,rr r r r rMr7rMc@seZdZUdZded<dS)ExprStmtzHA bare call used for side effects, e.g. SetCommand(...), SetConfig(...).r(callNr1r r r r rNs  rNc@s"eZdZUdZded<ded<dS) IncDecStmtzBare x++ / x-- / ++x / --x as a full statement (result discarded). Compiles to the compact 3-instruction form (LOAD, INC/DEC, STORE) rather than IncDecExpr's 4-instruction consumed form.rr/r"Nr1r r r r rPs  rPc@r)ExitStmtLoopKindkindNrr r r r rQrrQc@r) ContinueStmtrRrSNrr r r r rTrrTc@r') WhileStmtzCovers both `While...End While` and `Do While...Loop` -- confirmed byte-identical, so the parser normalizes both source forms to this node.rrAr)r5bodyNrr r r rrr-rVr r r r rU rUc@r') DoUntilStmtz`Do Until ... Loop` -- pre-test, exits when condition becomes true (mirror image of WhileStmt, which exits when condition is false).rrAr)r5rVNrWr r r r rYrXrYc@r=)DoLoopWhileStmtzl`Do ... Loop While ` -- post-test, body always runs >= 1 time, loops back while condition is true.r5rVrrANr1r r r r rZ  rZc@r=)DoLoopUntilStmtz`Do ... Loop Until ` -- post-test, body always runs >= 1 time, loops back while condition is false (exits once true).r5rVrrANr1r r r r r\r[r\c@DeZdZUdZded<ded<ded<ded<eed Zd ed <d S) ForTraditionalStmtzTraditional `For var = init To limit [Step step] ... Next`. Confirmed to allocate two hidden Integer slots (limit, step) and use a step-sign-generic exit test -- see codegen.py.rr/rinitlimitrstepr)r5rVNrWr r r r r^ r^c@r]) ForCStyleStmtz`For var = init AndWhile cond [Evaluate stmt] ... Next`. Confirmed: if no Evaluate clause is given, the compiler auto-increments var by 1 each iteration (reverse_engineering.md, batch 3).rr/rr_rAz Stmt | Noneevaluater)r5rVNrWr r r r rcrbrcN))r __future__r dataclassesrrrrrrrrr!r$r(r.r2r4r8r9r<r>r@rErGrHrJrKrMrNrPrrRrQrTrUrYrZr\r^rcr r r r s