87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
import sys
|
|
from io import TextIOWrapper
|
|
from blessed import Terminal
|
|
|
|
# create memory buffer and pointer
|
|
class Memory:
|
|
BUFFER_SIZE = 30_000
|
|
buffer = [0 for _ in range(BUFFER_SIZE)]
|
|
pointer = 0
|
|
loop_stack = []
|
|
|
|
@classmethod
|
|
def get_pointer(cls) -> int:
|
|
# TODO add logic to roll around buffer
|
|
return cls.pointer
|
|
|
|
# define bf commands
|
|
class Operation:
|
|
SHIFT_LEFT = "<"
|
|
SHIFT_RIGHT = ">"
|
|
INCREMENT = "+"
|
|
DECREMENT = "-"
|
|
OUTPUT = "."
|
|
INPUT = ","
|
|
OPEN_LOOP = "["
|
|
CLOSE_LOOP = "]"
|
|
|
|
def main() -> None:
|
|
# get bf file location
|
|
file_location = None
|
|
if len(sys.argv) != 2:
|
|
print("Invalid or missing arguments...")
|
|
sys.exit()
|
|
else:
|
|
file_location = sys.argv[1]
|
|
|
|
# open file
|
|
file = open(file_location, "r")
|
|
while True:
|
|
operation = file.read(1)
|
|
if not operation:
|
|
break
|
|
|
|
# perform operation
|
|
perform_operation(operation, file)
|
|
|
|
file.close()
|
|
|
|
def perform_operation(operation: str, file: TextIOWrapper) -> None:
|
|
if operation == Operation.SHIFT_LEFT:
|
|
Memory.pointer -= 1
|
|
elif operation == Operation.SHIFT_RIGHT:
|
|
Memory.pointer += 1
|
|
elif operation == Operation.INCREMENT:
|
|
Memory.buffer[Memory.get_pointer()] += 1
|
|
elif operation == Operation.DECREMENT:
|
|
Memory.buffer[Memory.get_pointer()] -= 1
|
|
elif operation == Operation.OUTPUT:
|
|
print(chr(Memory.buffer[Memory.get_pointer()]), end="")
|
|
elif operation == Operation.INPUT:
|
|
term = Terminal()
|
|
with term.cbreak():
|
|
Memory.buffer[Memory.get_pointer()] = ord(term.inkey())
|
|
elif operation == Operation.OPEN_LOOP:
|
|
if Memory.buffer[Memory.get_pointer()] == 0:
|
|
while True:
|
|
operation = file.read(1)
|
|
if operation == Operation.CLOSE_LOOP:
|
|
break
|
|
elif not operation:
|
|
print("Mismatched bracket...") # TODO add line number and column on line
|
|
sys.exit()
|
|
else:
|
|
Memory.loop_stack.append(file.tell()-1)
|
|
elif operation == Operation.CLOSE_LOOP:
|
|
if len(Memory.loop_stack) == 0:
|
|
print("Mismatched bracket...") # TODO add line number and column on line
|
|
sys.exit()
|
|
|
|
open_loop_position = Memory.loop_stack.pop()
|
|
if Memory.buffer[Memory.get_pointer()] != 0:
|
|
file.seek(open_loop_position, 0)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|