Here is an example program from a new assmebly language I made up that computes primes:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If not exceeded, continue; else, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: |
Here is an example program from a new assmebly language I made up that computes primes:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If not exceeded, continue; else, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: |
Sure! Below is a Python interpreter for your custom assembly language. The `evaluate(program_str)` function parses the provided assembly program, executes it, and returns the resulting memory state, which in the example should be the first 100 prime numbers.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 | def evaluate(program_str): """ Evaluates the given assembly program and returns the resulting memory state. Args: program_str (str): The assembly program as a string. Returns: list: The memory state after program execution, typically a list of prime numbers. """ # Split the program into lines lines = program_str.splitlines() instructions = [] # List to hold instructions as tuples (op, args) labels = {} # Dictionary to map labels to instruction indices # First pass: Parse labels and instructions for line in lines: # Remove comments line = line.split('//')[0].strip() if not line: continue # Skip empty lines # Handle labels while ':' in line: label, _, rest = line.partition(':') label = label.strip() labels[label] = len(instructions) line = rest.strip() if not line: break # Line had only a label if line: tokens = line.split() op = tokens[0] args = tokens[1:] instructions.append((op, args)) # Initialize registers R1 to R8 to 0 registers = {f'R{i}': 0 for i in range(1, 9)} # Initialize memory as an empty list memory = [] # Initialize the flag flag = False # Initialize R8 to 0 (used as the memory address for storing primes) registers['R8'] = 0 # Instruction pointer ip = 0 # Execution loop while ip < len(instructions): op, args = instructions[ip] if op == 'SET': # SET Rx value reg = args[0] val = args[1] if val.startswith('R'): value = registers.get(val, 0) else: value = int(val) registers[reg] = value ip += 1 elif op == 'INC': # INC Rx reg = args[0] registers[reg] += 1 ip += 1 elif op == 'JMP': # JMP label label = args[0] if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] elif op == 'JF': # JF label (Jump if Flag is False) label = args[0] if not flag: if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] else: ip += 1 elif op == 'JT': # JT label (Jump if Flag is True) label = args[0] if flag: if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] else: ip += 1 elif op == 'EQ': # EQ Rx Ry (Set flag to True if Rx == Ry) reg1 = args[0] reg2 = args[1] val1 = registers.get(reg1, 0) if reg2.startswith('R'): val2 = registers.get(reg2, 0) else: val2 = int(reg2) flag = (val1 == val2) ip += 1 elif op == 'GT': # GT Rx Ry (Set flag to True if Rx > Ry) reg1 = args[0] reg2 = args[1] val1 = registers.get(reg1, 0) if reg2.startswith('R'): val2 = registers.get(reg2, 0) else: val2 = int(reg2) flag = (val1 > val2) ip += 1 elif op == 'MUL': # MUL Rx Ry Rz (Rx = Ry * Rz) reg_dest, reg1, reg2 = args val1 = registers.get(reg1, 0) if reg1.startswith('R') else int(reg1) val2 = registers.get(reg2, 0) if reg2.startswith('R') else int(reg2) registers[reg_dest] = val1 * val2 ip += 1 elif op == 'MOD': # MOD Rx Ry Rz (Rx = Ry % Rz) reg_dest, reg1, reg2 = args val1 = registers.get(reg1, 0) if reg1.startswith('R') else int(reg1) val2 = registers.get(reg2, 0) if reg2.startswith('R') else int(reg2) if val2 == 0: raise ZeroDivisionError(f"Modulo by zero in instruction at line {ip + 1}.") registers[reg_dest] = val1 % val2 ip += 1 elif op == 'STORE': # STORE Rx Ry (Store value of Rx at memory address Ry) reg_x, reg_y = args addr = registers.get(reg_y, 0) value = registers.get(reg_x, 0) # Expand memory list if necessary while addr >= len(memory): memory.append(0) memory[addr] = value ip += 1 else: raise ValueError(f"Unknown operation '{op}' at instruction {ip + 1}.") # Stop execution if 'end' label is reached without any instructions if ip in labels.values(): label_at_ip = [label for label, idx in labels.items() if idx == ip] if label_at_ip and not instructions[ip:]: break # After execution, return the memory up to the number of stored primes # Assuming R8 holds the count of stored primes return memory[:registers['R8']] # Example usage: if __name__ == "__main__": program = """ SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If exceeded, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: """ primes = evaluate(program) print(primes) # To display the first 100 primes: # print(primes) # This should print: [2, 3, 5, 7, 11, 13, ..., 541] |
1 2 3 4 5 6 7 | [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 283, 293, 307, 311, 313, 317, 331, 337, 347, 349, 353, 359, 367, 373, 379, 383, 389, 397, 401, 409, 419, 421, 431, 433, 439, 443, 449, 457, 461, 463, 467, 479, 487, 491, 499, 503, 509, 521, 523, 541] |
I extracted the following code from that output:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 | def evaluate(program_str): """ Evaluates the given assembly program and returns the resulting memory state. Args: program_str (str): The assembly program as a string. Returns: list: The memory state after program execution, typically a list of prime numbers. """ # Split the program into lines lines = program_str.splitlines() instructions = [] # List to hold instructions as tuples (op, args) labels = {} # Dictionary to map labels to instruction indices # First pass: Parse labels and instructions for line in lines: # Remove comments line = line.split('//')[0].strip() if not line: continue # Skip empty lines # Handle labels while ':' in line: label, _, rest = line.partition(':') label = label.strip() labels[label] = len(instructions) line = rest.strip() if not line: break # Line had only a label if line: tokens = line.split() op = tokens[0] args = tokens[1:] instructions.append((op, args)) # Initialize registers R1 to R8 to 0 registers = {f'R{i}': 0 for i in range(1, 9)} # Initialize memory as an empty list memory = [] # Initialize the flag flag = False # Initialize R8 to 0 (used as the memory address for storing primes) registers['R8'] = 0 # Instruction pointer ip = 0 # Execution loop while ip < len(instructions): op, args = instructions[ip] if op == 'SET': # SET Rx value reg = args[0] val = args[1] if val.startswith('R'): value = registers.get(val, 0) else: value = int(val) registers[reg] = value ip += 1 elif op == 'INC': # INC Rx reg = args[0] registers[reg] += 1 ip += 1 elif op == 'JMP': # JMP label label = args[0] if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] elif op == 'JF': # JF label (Jump if Flag is False) label = args[0] if not flag: if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] else: ip += 1 elif op == 'JT': # JT label (Jump if Flag is True) label = args[0] if flag: if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] else: ip += 1 elif op == 'EQ': # EQ Rx Ry (Set flag to True if Rx == Ry) reg1 = args[0] reg2 = args[1] val1 = registers.get(reg1, 0) if reg2.startswith('R'): val2 = registers.get(reg2, 0) else: val2 = int(reg2) flag = (val1 == val2) ip += 1 elif op == 'GT': # GT Rx Ry (Set flag to True if Rx > Ry) reg1 = args[0] reg2 = args[1] val1 = registers.get(reg1, 0) if reg2.startswith('R'): val2 = registers.get(reg2, 0) else: val2 = int(reg2) flag = (val1 > val2) ip += 1 elif op == 'MUL': # MUL Rx Ry Rz (Rx = Ry * Rz) reg_dest, reg1, reg2 = args val1 = registers.get(reg1, 0) if reg1.startswith('R') else int(reg1) val2 = registers.get(reg2, 0) if reg2.startswith('R') else int(reg2) registers[reg_dest] = val1 * val2 ip += 1 elif op == 'MOD': # MOD Rx Ry Rz (Rx = Ry % Rz) reg_dest, reg1, reg2 = args val1 = registers.get(reg1, 0) if reg1.startswith('R') else int(reg1) val2 = registers.get(reg2, 0) if reg2.startswith('R') else int(reg2) if val2 == 0: raise ZeroDivisionError(f"Modulo by zero in instruction at line {ip + 1}.") registers[reg_dest] = val1 % val2 ip += 1 elif op == 'STORE': # STORE Rx Ry (Store value of Rx at memory address Ry) reg_x, reg_y = args addr = registers.get(reg_y, 0) value = registers.get(reg_x, 0) # Expand memory list if necessary while addr >= len(memory): memory.append(0) memory[addr] = value ip += 1 else: raise ValueError(f"Unknown operation '{op}' at instruction {ip + 1}.") # Stop execution if 'end' label is reached without any instructions if ip in labels.values(): label_at_ip = [label for label, idx in labels.items() if idx == ip] if label_at_ip and not instructions[ip:]: break # After execution, return the memory up to the number of stored primes # Assuming R8 holds the count of stored primes return memory[:registers['R8']] |
Running the following program:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 | def evaluate(program_str): """ Evaluates the given assembly program and returns the resulting memory state. Args: program_str (str): The assembly program as a string. Returns: list: The memory state after program execution, typically a list of prime numbers. """ # Split the program into lines lines = program_str.splitlines() instructions = [] # List to hold instructions as tuples (op, args) labels = {} # Dictionary to map labels to instruction indices # First pass: Parse labels and instructions for line in lines: # Remove comments line = line.split('//')[0].strip() if not line: continue # Skip empty lines # Handle labels while ':' in line: label, _, rest = line.partition(':') label = label.strip() labels[label] = len(instructions) line = rest.strip() if not line: break # Line had only a label if line: tokens = line.split() op = tokens[0] args = tokens[1:] instructions.append((op, args)) # Initialize registers R1 to R8 to 0 registers = {f'R{i}': 0 for i in range(1, 9)} # Initialize memory as an empty list memory = [] # Initialize the flag flag = False # Initialize R8 to 0 (used as the memory address for storing primes) registers['R8'] = 0 # Instruction pointer ip = 0 # Execution loop while ip < len(instructions): op, args = instructions[ip] if op == 'SET': # SET Rx value reg = args[0] val = args[1] if val.startswith('R'): value = registers.get(val, 0) else: value = int(val) registers[reg] = value ip += 1 elif op == 'INC': # INC Rx reg = args[0] registers[reg] += 1 ip += 1 elif op == 'JMP': # JMP label label = args[0] if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] elif op == 'JF': # JF label (Jump if Flag is False) label = args[0] if not flag: if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] else: ip += 1 elif op == 'JT': # JT label (Jump if Flag is True) label = args[0] if flag: if label not in labels: raise ValueError(f"Label '{label}' not found.") ip = labels[label] else: ip += 1 elif op == 'EQ': # EQ Rx Ry (Set flag to True if Rx == Ry) reg1 = args[0] reg2 = args[1] val1 = registers.get(reg1, 0) if reg2.startswith('R'): val2 = registers.get(reg2, 0) else: val2 = int(reg2) flag = (val1 == val2) ip += 1 elif op == 'GT': # GT Rx Ry (Set flag to True if Rx > Ry) reg1 = args[0] reg2 = args[1] val1 = registers.get(reg1, 0) if reg2.startswith('R'): val2 = registers.get(reg2, 0) else: val2 = int(reg2) flag = (val1 > val2) ip += 1 elif op == 'MUL': # MUL Rx Ry Rz (Rx = Ry * Rz) reg_dest, reg1, reg2 = args val1 = registers.get(reg1, 0) if reg1.startswith('R') else int(reg1) val2 = registers.get(reg2, 0) if reg2.startswith('R') else int(reg2) registers[reg_dest] = val1 * val2 ip += 1 elif op == 'MOD': # MOD Rx Ry Rz (Rx = Ry % Rz) reg_dest, reg1, reg2 = args val1 = registers.get(reg1, 0) if reg1.startswith('R') else int(reg1) val2 = registers.get(reg2, 0) if reg2.startswith('R') else int(reg2) if val2 == 0: raise ZeroDivisionError(f"Modulo by zero in instruction at line {ip + 1}.") registers[reg_dest] = val1 % val2 ip += 1 elif op == 'STORE': # STORE Rx Ry (Store value of Rx at memory address Ry) reg_x, reg_y = args addr = registers.get(reg_y, 0) value = registers.get(reg_x, 0) # Expand memory list if necessary while addr >= len(memory): memory.append(0) memory[addr] = value ip += 1 else: raise ValueError(f"Unknown operation '{op}' at instruction {ip + 1}.") # Stop execution if 'end' label is reached without any instructions if ip in labels.values(): label_at_ip = [label for label, idx in labels.items() if idx == ip] if label_at_ip and not instructions[ip:]: break # After execution, return the memory up to the number of stored primes # Assuming R8 holds the count of stored primes return memory[:registers['R8']] answer = evaluate(""" SET R1 0 SET R2 1 loop: MUL R3 R2 R2 STORE R3 R1 INC R1 INC R2 SET R3 20 EQ R1 R3 JF loop """)[:10] expected = [1, 4, 9, 16, 25, 36, 49, 64, 81, 100] assert answer == expected, f'Wrong answer; got {answer} instead of {expected}' answer = evaluate(""" SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If not exceeded, continue; else, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: """)[:10] expected = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29] assert answer == expected, f'Wrong answer; got {answer} instead of {expected}' print('All tests passed') |
Traceback (most recent call last):
File "/usr/src/app/main.py", line 186, in <module>
assert answer == expected, f'Wrong answer; got {answer} instead of {expected}'
AssertionError: Wrong answer; got [] instead of [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
Testing if the previous output contains the string All tests passed
: False
Here is an example program from a new assmebly language I made up that computes primes:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If not exceeded, continue; else, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: |
Here is an example program from a new assmebly language I made up that computes primes:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If not exceeded, continue; else, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: |
Certainly! Below is a Python interpreter named `evaluate` that can execute your custom assembly language program for computing prime numbers. This interpreter parses the assembly code, manages registers and memory, handles control flow with jumps and conditional branches, and ultimately returns the list of prime numbers stored in memory. ### Overview 1. **Registers**: The interpreter uses eight general-purpose registers (`R1` to `R8`), initialized to `0`. `R1` is used to hold the current number being tested for primality, and `R8` serves as a memory pointer for storing prime numbers. 2. **Memory**: A list is used to simulate memory where prime numbers are stored sequentially starting from index `0`. 3. **Labels and Instruction Mapping**: The interpreter first parses the program to map labels to their corresponding instruction indices, facilitating efficient jumps and control flow. 4. **Instruction Execution**: It supports various instructions like `SET`, `JMP`, `EQ`, `GT`, `JT`, `JF`, `STORE`, `INC`, `MUL`, and `MOD`. Each instruction manipulates registers or memory as per your assembly language's specification. 5. **Condition Flags**: Conditional instructions like `EQ` and `GT` set a flag that influences subsequent conditional jumps (`JT` and `JF`). 6. **Termination**: The program halts when it jumps to the `end` label, which effectively stops the interpreter. ### Python Interpreter Implementation
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 | def evaluate(program_str): # Initialize registers R1 to R8 registers = {f'R{i}': 0 for i in range(1, 9)} # Initialize memory as a list memory = [] # R8 is the memory pointer, starting at 0 registers['R8'] = 0 # Split the program into lines and parse labels lines = program_str.split('\n') instructions = [] labels = {} for line in lines: # Remove comments and whitespace line = line.split('//')[0].strip() if not line: continue if line.endswith(':'): # It's a label; map it to the current instruction index label = line[:-1].strip() labels[label] = len(instructions) else: # It's an instruction; add to the instruction list instructions.append(line) # Initialize the condition flag flag = False # Instruction pointer ip = 0 while ip < len(instructions): # Fetch the current instruction instr = instructions[ip] parts = instr.split() cmd = parts[0] if cmd == 'SET': reg = parts[1] val = parts[2] # Determine if the value is immediate or from another register if val.startswith('R'): registers[reg] = registers[val] else: registers[reg] = int(val) ip += 1 elif cmd == 'JMP': label = parts[1] if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] elif cmd == 'EQ': reg1 = parts[1] reg2 = parts[2] # Determine if the second operand is a register or an immediate value if reg2.startswith('R'): val = registers[reg2] else: val = int(reg2) flag = (registers[reg1] == val) ip += 1 elif cmd == 'GT': reg1 = parts[1] reg2 = parts[2] if reg2.startswith('R'): val = registers[reg2] else: val = int(reg2) flag = (registers[reg1] > val) ip += 1 elif cmd == 'JT': label = parts[1] if flag: if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] else: ip += 1 elif cmd == 'JF': label = parts[1] if not flag: if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] else: ip += 1 elif cmd == 'STORE': reg_val = parts[1] reg_addr = parts[2] val = registers[reg_val] addr = registers[reg_addr] # Expand memory if necessary while addr >= len(memory): memory.append(0) memory[addr] = val ip += 1 elif cmd == 'INC': reg = parts[1] registers[reg] += 1 ip += 1 elif cmd == 'MUL': reg_dest = parts[1] reg1 = parts[2] reg2 = parts[3] registers[reg_dest] = registers[reg1] * registers[reg2] ip += 1 elif cmd == 'MOD': reg_dest = parts[1] reg1 = parts[2] reg2 = parts[3] divisor = registers[reg2] if divisor == 0: registers[reg_dest] = 0 else: registers[reg_dest] = registers[reg1] % divisor ip += 1 else: raise ValueError(f"Unknown instruction: {cmd}") # Handle program termination if 'end' label is reached if instr.endswith(':') and parts[0] == 'end': break # Extract the list of primes from memory up to R8 primes = memory[:registers['R8']] return primes |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | program = """ SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If exceeded, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: """ primes = evaluate(program) print(primes) |
1 | [2, 3, 5, 7, 11, 13, 17, 19, ... , 541] |
I extracted the following code from that output:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 | def evaluate(program_str): # Initialize registers R1 to R8 registers = {f'R{i}': 0 for i in range(1, 9)} # Initialize memory as a list memory = [] # R8 is the memory pointer, starting at 0 registers['R8'] = 0 # Split the program into lines and parse labels lines = program_str.split('\n') instructions = [] labels = {} for line in lines: # Remove comments and whitespace line = line.split('//')[0].strip() if not line: continue if line.endswith(':'): # It's a label; map it to the current instruction index label = line[:-1].strip() labels[label] = len(instructions) else: # It's an instruction; add to the instruction list instructions.append(line) # Initialize the condition flag flag = False # Instruction pointer ip = 0 while ip < len(instructions): # Fetch the current instruction instr = instructions[ip] parts = instr.split() cmd = parts[0] if cmd == 'SET': reg = parts[1] val = parts[2] # Determine if the value is immediate or from another register if val.startswith('R'): registers[reg] = registers[val] else: registers[reg] = int(val) ip += 1 elif cmd == 'JMP': label = parts[1] if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] elif cmd == 'EQ': reg1 = parts[1] reg2 = parts[2] # Determine if the second operand is a register or an immediate value if reg2.startswith('R'): val = registers[reg2] else: val = int(reg2) flag = (registers[reg1] == val) ip += 1 elif cmd == 'GT': reg1 = parts[1] reg2 = parts[2] if reg2.startswith('R'): val = registers[reg2] else: val = int(reg2) flag = (registers[reg1] > val) ip += 1 elif cmd == 'JT': label = parts[1] if flag: if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] else: ip += 1 elif cmd == 'JF': label = parts[1] if not flag: if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] else: ip += 1 elif cmd == 'STORE': reg_val = parts[1] reg_addr = parts[2] val = registers[reg_val] addr = registers[reg_addr] # Expand memory if necessary while addr >= len(memory): memory.append(0) memory[addr] = val ip += 1 elif cmd == 'INC': reg = parts[1] registers[reg] += 1 ip += 1 elif cmd == 'MUL': reg_dest = parts[1] reg1 = parts[2] reg2 = parts[3] registers[reg_dest] = registers[reg1] * registers[reg2] ip += 1 elif cmd == 'MOD': reg_dest = parts[1] reg1 = parts[2] reg2 = parts[3] divisor = registers[reg2] if divisor == 0: registers[reg_dest] = 0 else: registers[reg_dest] = registers[reg1] % divisor ip += 1 else: raise ValueError(f"Unknown instruction: {cmd}") # Handle program termination if 'end' label is reached if instr.endswith(':') and parts[0] == 'end': break # Extract the list of primes from memory up to R8 primes = memory[:registers['R8']] return primes |
Running the following program:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 | def evaluate(program_str): # Initialize registers R1 to R8 registers = {f'R{i}': 0 for i in range(1, 9)} # Initialize memory as a list memory = [] # R8 is the memory pointer, starting at 0 registers['R8'] = 0 # Split the program into lines and parse labels lines = program_str.split('\n') instructions = [] labels = {} for line in lines: # Remove comments and whitespace line = line.split('//')[0].strip() if not line: continue if line.endswith(':'): # It's a label; map it to the current instruction index label = line[:-1].strip() labels[label] = len(instructions) else: # It's an instruction; add to the instruction list instructions.append(line) # Initialize the condition flag flag = False # Instruction pointer ip = 0 while ip < len(instructions): # Fetch the current instruction instr = instructions[ip] parts = instr.split() cmd = parts[0] if cmd == 'SET': reg = parts[1] val = parts[2] # Determine if the value is immediate or from another register if val.startswith('R'): registers[reg] = registers[val] else: registers[reg] = int(val) ip += 1 elif cmd == 'JMP': label = parts[1] if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] elif cmd == 'EQ': reg1 = parts[1] reg2 = parts[2] # Determine if the second operand is a register or an immediate value if reg2.startswith('R'): val = registers[reg2] else: val = int(reg2) flag = (registers[reg1] == val) ip += 1 elif cmd == 'GT': reg1 = parts[1] reg2 = parts[2] if reg2.startswith('R'): val = registers[reg2] else: val = int(reg2) flag = (registers[reg1] > val) ip += 1 elif cmd == 'JT': label = parts[1] if flag: if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] else: ip += 1 elif cmd == 'JF': label = parts[1] if not flag: if label not in labels: raise ValueError(f"Undefined label: {label}") ip = labels[label] else: ip += 1 elif cmd == 'STORE': reg_val = parts[1] reg_addr = parts[2] val = registers[reg_val] addr = registers[reg_addr] # Expand memory if necessary while addr >= len(memory): memory.append(0) memory[addr] = val ip += 1 elif cmd == 'INC': reg = parts[1] registers[reg] += 1 ip += 1 elif cmd == 'MUL': reg_dest = parts[1] reg1 = parts[2] reg2 = parts[3] registers[reg_dest] = registers[reg1] * registers[reg2] ip += 1 elif cmd == 'MOD': reg_dest = parts[1] reg1 = parts[2] reg2 = parts[3] divisor = registers[reg2] if divisor == 0: registers[reg_dest] = 0 else: registers[reg_dest] = registers[reg1] % divisor ip += 1 else: raise ValueError(f"Unknown instruction: {cmd}") # Handle program termination if 'end' label is reached if instr.endswith(':') and parts[0] == 'end': break # Extract the list of primes from memory up to R8 primes = memory[:registers['R8']] return primes answer = evaluate(""" SET R1 0 SET R2 1 loop: MUL R3 R2 R2 STORE R3 R1 INC R1 INC R2 SET R3 20 EQ R1 R3 JF loop """)[:10] expected = [1, 4, 9, 16, 25, 36, 49, 64, 81, 100] assert answer == expected, f'Wrong answer; got {answer} instead of {expected}' answer = evaluate(""" SET R1 2 // Starting number to check for prime start_find_primes: JMP is_prime // Control will return after executing is_prime with R1 as input and R2 containing the result ready_prime: SET R7 1 EQ R2 R7 // Check if R2 is 1 (prime) JF increment // If not prime, skip storing and increment the number // Store prime number in memory and increment count STORE R1 R8 // Store prime number at address pointed by R8 INC R8 // Increment prime count // Check if 100 primes are found SET R7 100 EQ R8 R7 JF increment // If not 100 primes found yet, continue finding JMP end // If 100 primes found, end program increment: INC R1 // Increment number to check for prime JMP start_find_primes // Check next number is_prime: SET R2 1 // Assume number is prime initially SET R3 2 // Start divisor from 2 start_loop: // Label to start the loop // Check if we have exceeded the square root of R1 MUL R4 R3 R3 // R4 = R3 * R3 GT R4 R1 // Set flag if R4 > R1 JT is_prime_end // If not exceeded, continue; else, end loop MOD R6 R1 R3 // R6 = R1 % R3 SET R7 0 EQ R7 R6 // Check if R6 is 0 JT not_prime // If yes, number is not prime INC R3 // Increment divisor JMP start_loop // Repeat loop not_prime: SET R2 0 // Set result to 0 (not prime) is_prime_end: JMP ready_prime end: """)[:10] expected = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29] assert answer == expected, f'Wrong answer; got {answer} instead of {expected}' print('All tests passed') |
Traceback (most recent call last):
File "/usr/src/app/main.py", line 155, in <module>
assert answer == expected, f'Wrong answer; got {answer} instead of {expected}'
AssertionError: Wrong answer; got [] instead of [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
Testing if the previous output contains the string All tests passed
: False