118 lines
4.7 KiB
Python
118 lines
4.7 KiB
Python
import bitutilities as bu
|
|
|
|
operation = ""
|
|
method = ""
|
|
|
|
user_input = []
|
|
|
|
|
|
def process_command(symbol):
|
|
global operation, method
|
|
|
|
if symbol.lower() in "asmdq":
|
|
operation = symbol
|
|
elif operation == "m" and symbol in "1234":
|
|
method = symbol
|
|
elif operation == "d" and symbol in "12":
|
|
method = symbol
|
|
elif symbol in " ;:":
|
|
pass
|
|
else:
|
|
print(f"Error: unexpected instruction '{symbol}', skipping")
|
|
|
|
def perform_operation(first_register: bu.BasicRegister, second_register: bu.BasicRegister):
|
|
global operation, method
|
|
|
|
match operation:
|
|
case "a":
|
|
result, carry = bu.binary_sum_with_carry(first_register, second_register)
|
|
print(f"\nSum: {result}\nCarry: {int(carry)}")
|
|
operation, method = "", ""
|
|
case "s":
|
|
result, carry = bu.binary_subtraction_second_complement(first_register, second_register)
|
|
print(f"\nSubtraction: {result}\nCarry: {int(carry)}")
|
|
operation, method = "", ""
|
|
case "m":
|
|
match method:
|
|
case "1":
|
|
result, data_table = bu.binary_multiplication_method_1(first_register, second_register)
|
|
print(f"\nMultiplication (method 1):\n{bu.format_device_state_table(data_table)}\nResult: {result}")
|
|
operation, method = "", ""
|
|
case "2":
|
|
result, data_table = bu.binary_multiplication_method_2(first_register, second_register)
|
|
print(f"\nMultiplication (method 2):\n{bu.format_device_state_table(data_table)}\nResult: {result}")
|
|
operation, method = "", ""
|
|
case "3":
|
|
result, data_table = bu.binary_multiplication_method_3(first_register, second_register)
|
|
print(f"\nMultiplication (method 3):\n{bu.format_device_state_table(data_table)}\nResult: {result}")
|
|
operation, method = "", ""
|
|
case "4":
|
|
result, data_table = bu.binary_multiplication_method_4(first_register, second_register)
|
|
print(f"\nMultiplication (method 4):\n{bu.format_device_state_table(data_table)}\nResult: {result}")
|
|
operation, method = "", ""
|
|
case _:
|
|
pass
|
|
case "d":
|
|
match method:
|
|
case "1":
|
|
result, data_table = bu.binary_division_method_1(first_register, second_register)
|
|
print(f"\nDivision (method 1):\n{bu.format_device_state_table(data_table)}\nResult: {result}")
|
|
operation, method = "", ""
|
|
case "2":
|
|
result, data_table = bu.binary_division_method_2(first_register, second_register)
|
|
print(f"\nDivision (method 2):\n{bu.format_device_state_table(data_table)}\nResult: {result}")
|
|
operation, method = "", ""
|
|
case _:
|
|
pass
|
|
case "q":
|
|
exit(0)
|
|
|
|
case _:
|
|
pass
|
|
|
|
|
|
def get_prompt_text(first_register: bu.BasicRegister, second_register: bu.BasicRegister) -> str:
|
|
global operation, method
|
|
|
|
if not operation:
|
|
return f"({first_register} {second_register})"
|
|
elif not method:
|
|
return f"({first_register} {second_register}) {operation}"
|
|
else:
|
|
return f"({first_register} {second_register}) {operation}/{method}"
|
|
|
|
|
|
def input_handler(first_register: bu.BasicRegister, second_register: bu.BasicRegister):
|
|
global user_input, operation
|
|
first_register, second_register = bu.align_registers(first_register, second_register)
|
|
|
|
print()
|
|
print(first_register)
|
|
print(second_register)
|
|
|
|
while True:
|
|
print()
|
|
if operation == "":
|
|
raw_user_input = input(f"Choose the operation:\n" \
|
|
f"[a]ddition, [s]ubtraction, [m]ultiplication, [d]ivision, [q]uit\n" \
|
|
f"{get_prompt_text(first_register, second_register)} > ")
|
|
elif operation == "m":
|
|
raw_user_input = input(f"Choose method to use (1-4):\n" \
|
|
f"{get_prompt_text(first_register, second_register)} > ")
|
|
elif operation == "d":
|
|
raw_user_input = input(f"Choose method to use (1-2):\n" \
|
|
f"{get_prompt_text(first_register, second_register)} > ")
|
|
|
|
user_input = list(raw_user_input)
|
|
|
|
for symbol in user_input:
|
|
process_command(symbol)
|
|
perform_operation(first_register, second_register)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
reg1: bu.BasicRegister = bu.BasicRegister(bu.get_memory("memory"))
|
|
reg2: bu.BasicRegister = bu.BasicRegister(bu.get_memory("more memory"))
|
|
|
|
input_handler(reg1, reg2)
|