diff options
Diffstat (limited to 'gentests.py')
| -rwxr-xr-x | gentests.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/gentests.py b/gentests.py index 3e8f083..9aa6a92 100755 --- a/gentests.py +++ b/gentests.py @@ -23,10 +23,21 @@ class Program(object): def write_rom_test(self, outfile): outfile.write('\tdc.l $0, start\n') + needdivzero = self.inst.name.startswith('div') for i in xrange(0x8, 0x100, 0x4): - outfile.write('\tdc.l empty_handler\n') - outfile.write('\tdc.b "SEGA"\nempty_handler:\n\trte\nstart:\n') - outfile.write('\tmove #0, CCR\n') + if needdivzero and i == 0x14: + outfile.write('\tdc.l div_zero_handler\n') + else: + outfile.write('\tdc.l empty_handler\n') + outfile.write('\tdc.b "SEGA"\nempty_handler:\n\trte\n') + if needdivzero: + outfile.write('div_zero_handler:\n') + div_zero_count = self.get_dreg() + outfile.write('\taddq #1, ' + str(div_zero_count) + '\n') + outfile.write('\trte\n') + outfile.write('start:\n\tmove #0, CCR\n') + if needdivzero: + outfile.write('\tmoveq #0, ' + str(div_zero_count) + '\n') already = {} self.inst.write_init(outfile, already) if 'label' in already: |