diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 24 |
1 files changed, 13 insertions, 11 deletions
@@ -57,18 +57,13 @@ code.append('mov rdi, 1') code.append('lea rsi, [int_buffer + rcx]') code.append('mov rdx, rbx') code.append('syscall') -code.append('mov [int_buffer], 10') -code.append('mov rax, 1') -code.append('mov rdi, 1') -code.append('mov rsi, int_buffer') -code.append('mov rdx, 1') -code.append('syscall') code.append('mov rbx, [rsp - 8]') code.append('ret') code.append('start:') loop_stack = [] loop_index = -1 + with open('input.stk', 'r') as f: text = f.read() words = text.replace('\n', ' ').strip().split(' ') @@ -96,20 +91,27 @@ with open('input.stk', 'r') as f: elif word == '*': code.append('pop rax') code.append('pop rdi') - code.append('mul rdi') + code.append('imul rdi') code.append('push rax') - elif word == '!': + elif word == 'pop': code.append('pop rax') - elif word == '~': + elif word == 'dup': code.append('pop rax') code.append('push rax') code.append('push rax') - elif word == '.': + elif word == 'print_number': code.append('call print') code.append('pop rax') - elif word == ',': + elif word == 'read_number': code.append('call read') code.append('push rax') + elif word == 'newline': + code.append('mov [int_buffer], 10') + code.append('mov rax, 1') + code.append('mov rdi, 1') + code.append('mov rsi, int_buffer') + code.append('mov rdx, 1') + code.append('syscall') elif word == '[': loop_index += 1 loop_stack.append(loop_index) |