class Chip8CPU(object):
def __init__(self, screen):
self.registers = {
'v': [],
'index': 0,
'pc': 0,
'sp': 0,
'rpl': []
}
self.timers = {
'delay': 0,
'sound': 0,
} self.operation_lookup = {
0x0: self.screen_return,
0x1: self.jump_to_address,
0x2: self.jump_to_subroutine,
0x3: self.skip_if_reg_equal_val,
0x4: self.skip_if_reg_not_equal_val,
0x5: self.skip_if_reg_equal_reg,
0x6: self.move_value_to_reg,
0x7: self.add_value_to_reg,
0x8: self.execute_logical_instruction,
0x9: self.skip_if_reg_not_equal_reg,
0xA: self.load_index_reg_with_value,
0xB: self.jump_to_index_plus_value,
0xC: self.generate_random_number_to_reg,
0xD: self.draw_sprite,
0xE: self.keyboard_routines,
0xF: self.misc_routines, } self.logical_operation_lookup = {
0x0: self.move_reg_into_reg,
0x1: self.logical_or,
0x2: self.logical_and,
0x3: self.exclusive_or,
0x4: self.add_reg_to_reg,
0x5: self.subtract_reg_from_reg,
0x6: self.right_shift_reg,
0x7: self.substract_reg_from_reg1,
0xE: self.left_shift_reg,
} self.misc_routine_lookup = {
0x07: self.move_delay_timer_into_reg,
0x0A: self.wait_for_keypress,
0x15: self.move_reg_into_delay_timer,
0x18: self.move_reg_into_sound_timer,
0x1E: self.add_reg_into_index,
0x29: self.load_index_with_reg_sprite,
0x30: self.load_index_with_extended_reg_sprite,
0x33: self.store_bcd_in_memory,
0x55: self.store_regs_in_memory,
0x65: self.read_regs_from_memory,
0x75: self.store_regs_in_rpl,
0x85: self.read_regs_from_rpl
} self.operand = 0
self.mode = MODE_NORMAL
self.screen = screen self.memory = bytearray(MAX_MEMORY)
self.reset()
self.running = True def execute_instruction(self, operand=None):
if operand:
self.operand = operand
else:
self.operand = int(self.memory[self.registers['pc']])
self.operand = self.operand << 8
self.operand += int(self.memory[self.registers['pc'] + 1])
self.registers['pc'] += 2
operation = (self.operand & 0xF000) >> 12
self.operation_lookup[operation]()
return self.operand def screen_return(self):
operation = self.operand & 0x00FF
sub_operation = operation & 0x00F0
if sub_operation == 0x00C0:
num_lines = self.operand & 0x000F
self.screen.scroll_down(num_lines)
elif operation == 0x00E0:
self.screen.clear_screen()
elif operation == 0x00EE:
self.return_from_subroutine()
elif operation == 0x00FB:
self.screen.scroll_right()
elif operation == 0x00FC:
self.screen.scroll_left()
elif operation == 0x00FD:
self.running = False
elif operation == 0x00FE:
self.disable_extended_mode()
elif operation == 0x00FF:
self.enable_extended_mode() def return_from_subroutine(self):
self.registers['sp'] -= 1
self.registers['pc'] = self.memory[self.registers['sp']] << 8
self.registers['sp'] -= 1
self.registers['pc'] += self.memory[self.registers['sp']] def enable_extended_mode(self):
self.screen.set_extended()
self.mode = MODE_EXTENDED def disable_extended_mode(self):
self.screen.set_normal()
self.mode = MODE_NORMAL def jump_to_address(self):
self.registers['pc'] = self.operand & 0x0FFF def jump_to_subroutine(self):
self.memory[self.registers['sp']] = self.registers['pc'] & 0x00FF
self.registers['sp'] += 1
self.memory[self.registers['sp']] = (self.registers['pc'] & 0xFF00) >> 8
self.registers['sp'] += 1
self.registers['pc'] = self.operand & 0x0FFF def skip_if_reg_equal_val(self):
source = (self.operand & 0x0F00) >> 8
if self.registers['v'][source] == (self.operand & 0x00FF):
self.registers['pc'] += 2 def skip_if_reg_not_equal_val(self):
source = (self.operand & 0x0F00) >> 8
if self.registers['v'][source] != (self.operand & 0x00FF):
self.registers['pc'] += 2 def skip_if_reg_equal_reg(self):
source = (self.operand & 0x0F00) >> 8
target = (self.operand & 0x00F0) >> 4
if self.registers['v'][source] == self.registers['v'][target]:
self.registers['pc'] += 2 def move_value_to_reg(self):
target = (self.operand & 0x0F00) >> 8
self.registers['v'][target] = self.operand & 0x00FF def add_value_to_reg(self):
target = (self.operand & 0x0F00) >> 8
temp = self.registers['v'][target] + (self.operand & 0x00FF)
self.registers['v'][target] = temp if temp < 256 else temp - 256 def skip_if_reg_not_equal_reg(self):
source = (self.operand & 0x0F00) >> 8
target = (self.operand & 0x00F0) >> 4
if self.registers['v'][source] != self.registers['v'][target]:
self.registers['pc'] += 2 def load_index_reg_with_value(self):
self.registers['index'] = self.operand & 0x0FFF def jump_to_index_plus_value(self):
self.registers['pc'] = self.registers['index'] + (self.operand & 0x0FFF) def generate_random_number_to_reg(self):
value = self.operand & 0x00FF
target = (self.operand & 0x0F00) >> 8
self.registers['v'][target] = value & randint(0, 255) def draw_sprite(self):
x_source = (self.operand & 0x0F00) >> 8
y_source = (self.operand & 0x00F0) >> 4
x_pos = self.registers['v'][x_source]
y_pos = self.registers['v'][y_source]
num_bytes = self.operand & 0x000F
self.registers['v'][0xF] = 0 if self.mode == MODE_EXTENDED and num_bytes == 0:
self.draw_extended(x_pos, y_pos, 16)
else:
self.draw_normal(x_pos, y_pos, num_bytes) def draw_normal(self, x_pos, y_pos, num_bytes):
for y_index in range(num_bytes):
color_byte = bin(self.memory[self.registers['index'] + y_index])
color_byte = color_byte[2:].zfill(8)
y_coord = y_pos + y_index
y_coord = y_coord % self.screen.get_height() for x_index in range(8):
x_coord = x_pos + x_index
x_coord = x_coord % self.screen.get_width() color = int(color_byte[x_index])
current_color = self.screen.get_pixel(x_coord, y_coord) if color == 1 and current_color == 1:
self.registers['v'][0xF] = 1
color = 0
elif color == 0 and current_color == 1:
color = 1 self.screen.draw_pixel(x_coord, y_coord, color) self.screen.update() def draw_extended(self, x_pos, y_pos, num_bytes):
for y_index in range(num_bytes):
for x_byte in range(2):
color_byte = bin(self.memory[self.registers['index'] + (y_index * 2) + x_byte])
color_byte = color_byte[2:].zfill(8)
y_coord = y_pos + y_index
y_coord = y_coord % self.screen.height for x_index in range(8):
x_coord = x_pos + x_index + (x_byte * 8)
x_coord = x_coord % self.screen.width
color = int(color_byte[x_index])
current_color = self.screen.get_pixel(x_coord, y_coord) if color == 1 and current_color == 1:
self.registers['v'][0xF] = 1
color = 0 elif color == 0 and current_color == 1:
color = 1 self.screen.draw_pixel(x_coord, y_coord, color) self.screen.update() def keyboard_routines(self):
operation = self.operand & 0x00FF
source = (self.operand & 0x0F00) >> 8 key_to_check = self.registers['v'][source]
keys_pressed = key.get_pressed()
if operation == 0x9E:
if keys_pressed[KEY_MAPPINGS[key_to_check]]:
self.registers['pc'] += 2
elif operation == 0xA1:
if not keys_pressed[KEY_MAPPINGS[key_to_check]]:
self.registers['pc'] += 2 def execute_logical_instruction(self):
operation = self.operand & 0x000F
self.logical_operation_lookup[operation]() def move_reg_into_reg(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
self.registers['v'][target] = self.registers['v'][source] def logical_or(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
self.registers['v'][target] |= self.registers['v'][source] def logical_and(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
self.registers['v'][target] &= self.registers['v'][source] def exclusive_or(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
self.registers['v'][target] ^= self.registers['v'][source] def add_reg_to_reg(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
temp = self.registers['v'][target] + self.registers['v'][source]
if temp > 255:
self.registers['v'][target] = temp - 256
self.registers['v'][0xF] = 1
else:
self.registers['v'][target] = temp
self.registers['v'][0xF] = 0 def subtract_reg_from_reg(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
source_reg = self.registers['v'][source]
target_reg = self.registers['v'][target]
if target_reg > source_reg:
target_reg -= source_reg
self.registers['v'][0xF] = 1
else:
target_reg = 256 + target_reg - source_reg
self.registers['v'][0xF] = 0
self.registers['v'][target] = target_reg def right_shift_reg(self):
source = (self.operand & 0x0F00) >> 8
target = (self.operand & 0x00F0) >> 4
bit_zero = self.registers['v'][source] & 0x1
self.registers['v'][target] = self.registers['v'][source] >> 1
self.registers['v'][0xF] = bit_zero def substract_reg_from_reg1(self):
target = (self.operand & 0x0F00) >> 8
source = (self.operand & 0x00F0) >> 4
source_reg = self.registers['v'][source]
target_reg = self.registers['v'][target]
if source_reg > target_reg:
target_reg = source_reg - target_reg
self.registers['v'][0xF] = 1
else:
target_reg = 256 + source_reg - target_reg
self.registers['v'][0xF] = 0
self.registers['v'][target] = target_reg def left_shift_reg(self):
source = (self.operand & 0x0F00) >> 8
target = (self.operand & 0x00F0) >> 4
bit_seven = (self.registers['v'][source] & 0x80) >> 8
self.registers['v'][target] = self.registers['v'][source] << 1
self.registers['v'][0xF] = bit_seven def misc_routines(self):
operation = self.operand & 0x00FF
self.misc_routine_lookup[operation]() def move_delay_timer_into_reg(self):
target = (self.operand & 0x0F00) >> 8
self.registers['v'][target] = self.timers['delay'] def wait_for_keypress(self):
target = (self.operand & 0x0F00) >> 8
key_pressed = False
while not key_pressed:
event = pygame.event.wait()
if event.type == pygame.KEYDOWN:
keys_pressed = key.get_pressed()
for keyval, lookup_key in KEY_MAPPINGS.items():
if keys_pressed[lookup_key]:
self.registers['v'][target] = keyval
key_pressed = True
break def move_reg_into_delay_timer(self):
source = (self.operand & 0x0F00) >> 8
self.timers['delay'] = self.registers['v'][source] def move_reg_into_sound_timer(self):
source = (self.operand & 0x0F00) >> 8
self.timers['sound'] = self.registers['v'][source] def load_index_with_reg_sprite(self):
source = (self.operand & 0x0F00) >> 8
self.registers['index'] = self.registers['v'][source] * 5 def load_index_with_extended_reg_sprite(self):
source = (self.operand & 0x0F00) >> 8
self.registers['index'] = self.registers['v'][source] * 10 def add_reg_into_index(self):
source = (self.operand & 0x0F00) >> 8
self.registers['index'] += self.registers['v'][source] def store_bcd_in_memory(self):
source = (self.operand & 0x0F00) >> 8
bcd_value = '{:03d}'.format(self.registers['v'][source])
self.memory[self.registers['index']] = int(bcd_value[0])
self.memory[self.registers['index'] + 1] = int(bcd_value[1])
self.memory[self.registers['index'] + 2] = int(bcd_value[2]) def store_regs_in_memory(self):
source = (self.operand & 0x0F00) >> 8
for counter in range(source + 1):
self.memory[self.registers['index'] + counter] = \
self.registers['v'][counter] def read_regs_from_memory(self):
source = (self.operand & 0x0F00) >> 8
for counter in range(source + 1):
self.registers['v'][counter] = \
self.memory[self.registers['index'] + counter] def store_regs_in_rpl(self):
source = (self.operand & 0x0F00) >> 8
for counter in range(source + 1):
self.registers['rpl'][counter] = self.registers['v'][counter] def read_regs_from_rpl(self):
source = (self.operand & 0x0F00) >> 8
for counter in range(source + 1):
self.registers['v'][counter] = self.registers['rpl'][counter] def reset(self):
self.registers['v'] = [0] * NUM_REGISTERS
self.registers['pc'] = PROGRAM_COUNTER_START
self.registers['sp'] = STACK_POINTER_START
self.registers['index'] = 0
self.registers['rpl'] = [0] * NUM_REGISTERS
self.timers['delay'] = 0
self.timers['sound'] = 0 def load_rom(self, filename, offset=PROGRAM_COUNTER_START):
rom_data = open(filename, 'rb').read()
for index, val in enumerate(rom_data):
self.memory[offset + index] = val def decrement_timers(self):
if self.timers['delay'] != 0:
self.timers['delay'] -= 1 if self.timers['sound'] != 0:
self.timers['delay'] -= 1

最新文章

  1. scikit-learn一般实例之六:构建评估器之前进行缺失值填充
  2. AppBox升级进行时 - Any与All的用法(Entity Framework)
  3. subverison的安装与注意事项
  4. Greedy:Yogurt factory(POJ 2393)
  5. HDU:过山车(二分图最大匹配)
  6. Java SSH远程执行Shell脚本实现(转)
  7. 追踪CPU跑满 堆栈调试
  8. Nginx 代理 jira 和 confluence
  9. Java增强的泛型
  10. JavaScript之call()和apply()方法详解
  11. outlook 无法搜索邮件的解决方法
  12. 《实战Nginx》读书笔记
  13. Learn HTML5 in 5 Minutes!
  14. Greenplum hostname和address不一致导致配置文件无法加载
  15. 最少拦截系统 HDU - 1257
  16. NOIP2018题解
  17. Java多线程系列——深入重入锁ReentrantLock
  18. MapReduce中的Shuffle和Sort分析
  19. Codeforces Round #302 (Div. 1) D - Road Improvement 树形dp
  20. Web监听器导图详解

热门文章

  1. 解决vue中百度地图覆盖物引用本地图片问题
  2. Element-ui中为上传组件添加表单校验
  3. TypeScript 学习笔记(二)
  4. python获得多个输入值
  5. C学习笔记(11)--- 可变参数,浅谈内存管理 【C基础概念系列完结】
  6. 9. Go语言—流程控制
  7. is ==的区别 编码和解码
  8. Oracle SQL日期及日期格式获取命令
  9. [C14] 总结(Conclusion)
  10. Velocity与Jsp、Freemarker的对比