// File: baziotis[P]minijava-cpp/typecheck.cpp
// Function: track_while_assignment_for_stmt(Statement*, Array<WhileAssign, (MEM)4>)
bool track_while_assignment_for_stmt(Statement *stmt, FuncArr<WhileAssign> track_buf) {
    // IMPORTANT: Be sure to first visit the usage, then
    // the assignment in assigments. For example, in
    // AssignmentStatement, we first want to visit the RHS,
    // then track the assignment because if we have say:
    // a = a + 1;
    // then, it counts as a usage before an assignment.
    switch (stmt->kind) {
    case STMT::UNDEFINED: assert(0);
    case STMT::BLOCK:
    {
        BlockStatement *block = (BlockStatement *)stmt;
        for (Statement *s : block->block) {
            track_while_assignment_for_stmt(s, track_buf);
        }
    } break;
    case STMT::ASGN:
    {
        AssignmentStatement *asgn = (AssignmentStatement *)stmt;
        track_while_assignment_for_expr(asgn->rhs, track_buf);
        for (WhileAssign& wa : track_buf) {
            Local *local = wa.local;
            if (asgn->id == local->id) {
                if (!wa.assigned) {
                    if (!wa.used) {
                        wa.assigned = 1;
                    } else {
                        wa.assigned = 2;
                    }
                }
                break;
            }
        }
    } break;
    case STMT::ARR_ASGN:
    {
        ArrayAssignmentStatement *arr_asgn = (ArrayAssignmentStatement *)stmt;
        track_while_assignment_for_expr(arr_asgn->index, track_buf);
        track_while_assignment_for_expr(arr_asgn->rhs, track_buf);

        for (WhileAssign& wa : track_buf) {
            Local *local = wa.local;
            if (arr_asgn->id == local->id) {
                if (!wa.assigned) {
                    if (!wa.used) {
                        wa.assigned = 1;
                    } else {
                        wa.assigned = 2;
                    }
                }
                break;
            }
        }
    } break;
    case STMT::IF:
    {
        IfStatement *if_stmt = (IfStatement *)stmt;
        track_while_assignment_for_expr(if_stmt->cond, track_buf);
        track_while_assignment_for_stmt(if_stmt->then, track_buf);
        track_while_assignment_for_stmt(if_stmt->else_, track_buf);
    } break;
    case STMT::WHILE:
    {
        WhileStatement *while_stmt = (WhileStatement *)stmt;
        track_while_assignment_for_expr(while_stmt->cond, track_buf);
        track_while_assignment_for_stmt(while_stmt->body, track_buf);
    } break;
    case STMT::PRINT:
    {
        PrintStatement *print_stmt = (PrintStatement *)stmt;
        track_while_assignment_for_expr(print_stmt->to_print, track_buf);
    } break;
    }
    return false;
}

// Assembly:
pushq %rbp
movq %rsp, %rbp
subq $0x1c0, %rsp            # imm = 0x1C0
leaq 0x10(%rbp), %rax
movq %rax, -0x1a8(%rbp)
movq %rdi, -0x8(%rbp)
movq -0x8(%rbp), %rax
movl 0x8(%rax), %eax
movq %rax, -0x1a0(%rbp)
subq $0x6, %rax
ja 0x13bc2
movq -0x1a0(%rbp), %rax
leaq 0x8100(%rip), %rcx      # 0x1b700
movslq (%rcx,%rax,4), %rax
addq %rcx, %rax
jmpq *%rax
leaq 0x8372(%rip), %rdi      # 0x1b982
leaq 0x8147(%rip), %rsi      # 0x1b75e
movl $0x796, %edx            # imm = 0x796
leaq 0x91f4(%rip), %rcx      # 0x1c817
callq 0x20d0
movq -0x8(%rbp), %rax
movq %rax, -0x10(%rbp)
movq -0x10(%rbp), %rax
addq $0x10, %rax
movq %rax, -0x18(%rbp)
movq -0x18(%rbp), %rdi
callq 0x16450
movq %rax, -0x20(%rbp)
movq -0x18(%rbp), %rdi
callq 0x16470
movq %rax, -0x28(%rbp)
movq -0x20(%rbp), %rax
cmpq -0x28(%rbp), %rax
je 0x136bd
movq -0x1a8(%rbp), %rax
movq -0x20(%rbp), %rcx
movq (%rcx), %rcx
movq %rcx, -0x30(%rbp)
movq -0x30(%rbp), %rdi
movq (%rax), %rcx
movq %rcx, -0x48(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x40(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x38(%rbp)
leaq -0x48(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x135c0
movq -0x20(%rbp), %rax
addq $0x8, %rax
movq %rax, -0x20(%rbp)
jmp 0x13656
jmp 0x13bc2
movq -0x1a8(%rbp), %rax
movq -0x8(%rbp), %rcx
movq %rcx, -0x50(%rbp)
movq -0x50(%rbp), %rcx
movq 0x18(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x68(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x60(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x58(%rbp)
leaq -0x68(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x13290
movq -0x1a8(%rbp), %rax
movq %rax, -0x70(%rbp)
movq -0x70(%rbp), %rdi
callq 0x169e0
movq %rax, -0x78(%rbp)
movq -0x70(%rbp), %rdi
callq 0x16a00
movq %rax, -0x80(%rbp)
movq -0x78(%rbp), %rax
cmpq -0x80(%rbp), %rax
je 0x137df
movq -0x78(%rbp), %rax
movq %rax, -0x88(%rbp)
movq -0x88(%rbp), %rax
movq (%rax), %rax
movq %rax, -0x90(%rbp)
movq -0x50(%rbp), %rax
movq 0x10(%rax), %rax
movq -0x90(%rbp), %rcx
cmpq (%rcx), %rax
jne 0x137cc
movq -0x88(%rbp), %rax
movb 0x8(%rax), %al
andb $0x3, %al
movzbl %al, %eax
cmpl $0x0, %eax
jne 0x137ca
movq -0x88(%rbp), %rax
movb 0x8(%rax), %al
shrb $0x2, %al
andb $0x3, %al
movzbl %al, %eax
cmpl $0x0, %eax
jne 0x137b5
movq -0x88(%rbp), %rax
movb 0x8(%rax), %cl
andb $-0x4, %cl
orb $0x1, %cl
movb %cl, 0x8(%rax)
jmp 0x137c8
movq -0x88(%rbp), %rax
movb 0x8(%rax), %cl
andb $-0x4, %cl
orb $0x2, %cl
movb %cl, 0x8(%rax)
jmp 0x137ca
jmp 0x137df
jmp 0x137ce
movq -0x78(%rbp), %rax
addq $0x10, %rax
movq %rax, -0x78(%rbp)
jmp 0x13737
jmp 0x13bc2
movq -0x1a8(%rbp), %rax
movq -0x8(%rbp), %rcx
movq %rcx, -0x98(%rbp)
movq -0x98(%rbp), %rcx
movq 0x18(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0xb0(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0xa8(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0xa0(%rbp)
leaq -0xb0(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x13290
movq -0x1a8(%rbp), %rax
movq -0x98(%rbp), %rcx
movq 0x20(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0xc8(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0xc0(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0xb8(%rbp)
leaq -0xc8(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x13290
movq -0x1a8(%rbp), %rax
movq %rax, -0xd0(%rbp)
movq -0xd0(%rbp), %rdi
callq 0x169e0
movq %rax, -0xd8(%rbp)
movq -0xd0(%rbp), %rdi
callq 0x16a00
movq %rax, -0xe0(%rbp)
movq -0xd8(%rbp), %rax
cmpq -0xe0(%rbp), %rax
je 0x1398b
movq -0xd8(%rbp), %rax
movq %rax, -0xe8(%rbp)
movq -0xe8(%rbp), %rax
movq (%rax), %rax
movq %rax, -0xf0(%rbp)
movq -0x98(%rbp), %rax
movq 0x10(%rax), %rax
movq -0xf0(%rbp), %rcx
cmpq (%rcx), %rax
jne 0x13972
movq -0xe8(%rbp), %rax
movb 0x8(%rax), %al
andb $0x3, %al
movzbl %al, %eax
cmpl $0x0, %eax
jne 0x13970
movq -0xe8(%rbp), %rax
movb 0x8(%rax), %al
shrb $0x2, %al
andb $0x3, %al
movzbl %al, %eax
cmpl $0x0, %eax
jne 0x1395b
movq -0xe8(%rbp), %rax
movb 0x8(%rax), %cl
andb $-0x4, %cl
orb $0x1, %cl
movb %cl, 0x8(%rax)
jmp 0x1396e
movq -0xe8(%rbp), %rax
movb 0x8(%rax), %cl
andb $-0x4, %cl
orb $0x2, %cl
movb %cl, 0x8(%rax)
jmp 0x13970
jmp 0x1398b
jmp 0x13974
movq -0xd8(%rbp), %rax
addq $0x10, %rax
movq %rax, -0xd8(%rbp)
jmp 0x138d1
jmp 0x13bc2
movq -0x1a8(%rbp), %rax
movq -0x8(%rbp), %rcx
movq %rcx, -0xf8(%rbp)
movq -0xf8(%rbp), %rcx
movq 0x10(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x110(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x108(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x100(%rbp)
leaq -0x110(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x13290
movq -0x1a8(%rbp), %rax
movq -0xf8(%rbp), %rcx
movq 0x18(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x128(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x120(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x118(%rbp)
leaq -0x128(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x135c0
movq -0x1a8(%rbp), %rax
movq -0xf8(%rbp), %rcx
movq 0x20(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x140(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x138(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x130(%rbp)
leaq -0x140(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x135c0
jmp 0x13bc2
movq -0x1a8(%rbp), %rax
movq -0x8(%rbp), %rcx
movq %rcx, -0x148(%rbp)
movq -0x148(%rbp), %rcx
movq 0x10(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x160(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x158(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x150(%rbp)
leaq -0x160(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x13290
movq -0x1a8(%rbp), %rax
movq -0x148(%rbp), %rcx
movq 0x18(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x178(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x170(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x168(%rbp)
leaq -0x178(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x135c0
jmp 0x13bc2
movq -0x1a8(%rbp), %rax
movq -0x8(%rbp), %rcx
movq %rcx, -0x180(%rbp)
movq -0x180(%rbp), %rcx
movq 0x10(%rcx), %rdi
movq (%rax), %rcx
movq %rcx, -0x198(%rbp)
movq 0x8(%rax), %rcx
movq %rcx, -0x190(%rbp)
movq 0x10(%rax), %rax
movq %rax, -0x188(%rbp)
leaq -0x198(%rbp), %rax
movq (%rax), %rcx
movq %rcx, (%rsp)
movq 0x8(%rax), %rcx
movq %rcx, 0x8(%rsp)
movq 0x10(%rax), %rax
movq %rax, 0x10(%rsp)
callq 0x13290
xorl %eax, %eax
andb $0x1, %al
addq $0x1c0, %rsp            # imm = 0x1C0
popq %rbp
retq
nop