Пример #1
0
def test_array_of_conditionals():
    test_struct = [
        ("cond", b.uint8),
        ("foos",
         b.array(3, (b.CONDITIONAL, "cond", {
             1: [("foo", b.nibble), b.padding(4)],
             2: [("bar", b.bit), b.padding(7)],
             4: [("baz", b.semi_nibble), b.padding(6)]
         })))
    ]

    test_data = bytearray([1, 0b10000101, 0b01010110, 0b11010101])

    test_parsed = b.parse(test_data, test_struct)

    assert_equal(test_parsed.cond, 1)
    assert_equal(test_parsed.foos[0].foo, 0b1000)
    assert_equal(test_parsed.foos[1].foo, 0b0101)
    assert_equal(test_parsed.foos[2].foo, 0b1101)

    test_parsed.cond = 4

    assert_equal(test_parsed.cond, 4)
    assert_equal(test_parsed.foos[0].baz, 0b10)
    assert_equal(test_parsed.foos[1].baz, 0b01)
    assert_equal(test_parsed.foos[2].baz, 0b11)
Пример #2
0
def test_array_of_conditionals():
    test_struct = [
        ("cond", b.uint8),
        ("foos", b.array(3, (b.CONDITIONAL, "cond", {
            1: [("foo", b.nibble), b.padding(4)],
            2: [("bar", b.bit), b.padding(7)],
            4: [("baz", b.semi_nibble), b.padding(6)]
        })))
    ]

    test_data = bytearray([1, 0b10000101, 0b01010110, 0b11010101])

    test_parsed = b.parse(test_data, test_struct)

    assert_equal(test_parsed.cond, 1)
    assert_equal(test_parsed.foos[0].foo, 0b1000)
    assert_equal(test_parsed.foos[1].foo, 0b0101)
    assert_equal(test_parsed.foos[2].foo, 0b1101)

    test_parsed.cond = 4

    assert_equal(test_parsed.cond, 4)
    assert_equal(test_parsed.foos[0].baz, 0b10)
    assert_equal(test_parsed.foos[1].baz, 0b01)
    assert_equal(test_parsed.foos[2].baz, 0b11)
Пример #3
0
from .vendor.six.moves import range


def padded_hex(pad_count):
    return lambda x: ("0x%%0%dx" % (pad_count)) % (x)


EMPTY_BLOCK = 0xff

# File management structure (starts at 0x8000)
compressed_sav_file = [
    # Up to 32 files (songs) can be saved to one cartridge
    ("filenames", b.array(32, b.string(8))),
    # Each file has a monotonically increasing version number
    ("file_versions", b.array(32, b.byte)),
    b.padding(30 * 8),
    ("sram_init_check", b.string(2)),  # set to 'jk' on init
    # The file that is currently active
    ("active_file", b.byte),
    # Table mapping blocks to files.
    ("block_alloc_table", b.array(191, b.byte))
]

# Should be 0x4000 bytes long
sample_kit = [
    ("magic_number", b.string(2)),  # should be 0x60, 0x40
    # The address of the first byte after each sample, or 0 if the sample is
    # unused
    ("sample_end_addresses", b.array(15, b.string(2))),
    b.padding(2 * 8),
    # For samples with names less than three characters long, sample names are
Пример #4
0
import bread as b
import sys


def hex_array(x):
    return str(map(hex, x))


nsf_header = [('magic_number', b.array(5, b.byte), {
    "str_format": hex_array
}), ('version', b.byte), ('total_songs', b.byte), ('starting_song', b.byte),
              ('load_addr', b.uint16, {
                  "str_format": hex
              }), ('init_addr', b.uint16, {
                  "str_format": hex
              }), ('play_addr', b.uint16, {
                  "str_format": hex
              }), ('title', b.string(32)), ('artist', b.string(32)),
              ('copyright', b.string(32)), ('ntsc_speed', b.uint16),
              ('bankswitch_init', b.array(8, b.byte), {
                  "str_format": hex_array
              }), ('pal_speed', b.uint16), ('ntsc', b.boolean),
              ('pal', b.boolean), ('ntsc_and_pal', b.boolean), (b.padding(6)),
              ('vrc6', b.boolean), ('vrc7', b.boolean), ('fds', b.boolean),
              ('mmc5', b.boolean), ('namco_106', b.boolean),
              ('fme07', b.boolean), (b.padding(2)), (b.padding(32))]

with open(sys.argv[1], 'r') as fp:
    header = b.parse(fp, nsf_header)
    print header
Пример #5
0
import bread as b

import bitstring

# Shared structs for bread struct test

test_struct = [
    {
        "endianness": b.BIG_ENDIAN
    },
    ("flag_one", b.boolean),
    ("flag_two", b.boolean),
    ("flag_three", b.boolean),
    ("flag_four", b.boolean),
    ("first", b.uint8),
    (b.padding(2),),
    b.padding(2),
    ("blah", b.uint16),
    ("second", b.int64),
    ("third", b.uint64),
    ("fourth", b.int8)
]

test_array_struct = [
    {
        "endianness": b.BIG_ENDIAN
    },
    ("first", b.uint8),
    ("flags", b.array(8, b.boolean)),
    ("last", b.uint8)]
Пример #6
0
  [ ('magic_number', b.array(5, b.byte), {"str_format": hex_array})
      , ('version', b.byte)
      , ('total_songs', b.byte)
      , ('starting_song', b.byte)
      , ('load_addr', b.uint16, {"str_format": hex})
      , ('init_addr', b.uint16, {"str_format": hex})
      , ('play_addr', b.uint16, {"str_format": hex})
      , ('title', b.string(32))
      , ('artist', b.string(32))
      , ('copyright', b.string(32))
      , ('ntsc_speed', b.uint16)
      , ('bankswitch_init', b.array(8, b.byte), {"str_format": hex_array})
      , ('pal_speed', b.uint16)
      , ('tv_std', b.boolean, {"str_format": lambda x: "PAL" if x else "NTSC"})
      , ('ntsc_and_pal', b.boolean)
      , (b.padding(6))
      , ('vrc6', b.boolean)
      , ('vrc7', b.boolean)
      , ('fds', b.boolean)
      , ('mmc5', b.boolean)
      , ('namco_106', b.boolean)
      , ('fme07', b.boolean)
      , (b.padding(2))
      , (b.padding(32))
     ]


class Instr:
    def __init__(self, op, amode, opand, size):
        self.op = op
        self.amode = amode
Пример #7
0
     {"str_format": hex_array}),
    ('version', b.byte),
    ('total_songs', b.byte),
    ('starting_song', b.byte),
    ('load_addr', b.uint16, {"str_format": hex}),
    ('init_addr', b.uint16, {"str_format": hex}),
    ('play_addr', b.uint16, {"str_format": hex}),
    ('title', b.string(32)),
    ('artist', b.string(32)),
    ('copyright', b.string(32)),
    ('ntsc_speed', b.uint16),
    ('bankswitch_init', b.array(8, b.byte), {"str_format": hex_array}),
    ('pal_speed', b.uint16),
    ('ntsc', b.boolean),
    ('pal', b.boolean),
    ('ntsc_and_pal', b.boolean),
    (b.padding(6)),
    ('vrc6', b.boolean),
    ('vrc7', b.boolean),
    ('fds', b.boolean),
    ('mmc5', b.boolean),
    ('namco_106', b.boolean),
    ('fme07', b.boolean),
    (b.padding(2)),
    (b.padding(32))
]

with open(sys.argv[1], 'r') as fp:
    header = b.parse(fp, nsf_header)
    print header
Пример #8
0
                 'offset': 1
             }), ('feedback', b.uint8), ('oscillator_sync', b.uint8),
             ('lfo_speed', b.uint8), ('lfo_delay', b.uint8),
             ('lfo_pitch_mod_depth', b.uint8), ('lfo_amp_mod_depth', b.uint8),
             ('lfo_sync', b.uint8),
             ('lfo_waveform', b.enum(8, enums['waveforms'])),
             ('pitch_mod_sensitivity', b.uint8), ('transpose', b.uint8),
             ('name', b.string(10, NAME_ENCODING))]

compressed_operator = [
    ('eg_rates', b.array(4, b.uint8)),
    ('eg_levels', b.array(4, b.uint8)),
    ('keyboard_level_scaling_break_point', b.uint8),  # C3 = 0x27
    ('keyboard_level_scaling_left_depth', b.uint8),
    ('keyboard_level_scaling_right_depth', b.uint8),
    b.padding(4),
    # Byte 11
    ('keyboard_level_scaling_right_curve', b.enum(2, enums['curves'])),
    ('keyboard_level_scaling_left_curve', b.enum(2, enums['curves'])),
    # Byte 12
    b.padding(1),
    ('osc_detune', b.intX(4), {
        'offset': -7
    }),
    ('keyboard_rate_scaling', b.intX(3)),
    # Byte 13
    b.padding(3),
    ('key_velocity_sensitivity', b.intX(3)),
    ('amp_mod_sensitivity', b.intX(2)),
    # Byte 14
    ('output_level', b.uint8),
Пример #9
0
import struct, sys, pprint, unittest, itertools, tempfile, os, json

from nose.tools import assert_equal, assert_not_equal, assert_true, \
    assert_false, assert_raises, raises

import bread as b

import bitstring

# Shared structs for bread struct test

test_struct = [{
    "endianness": b.BIG_ENDIAN
}, ("flag_one", b.boolean), ("flag_two", b.boolean), ("flag_three", b.boolean),
               ("flag_four", b.boolean), ("first", b.uint8), (b.padding(2), ),
               b.padding(2), ("blah", b.uint16), ("second", b.int64),
               ("third", b.uint64), ("fourth", b.int8)]

test_array_struct = [{
    "endianness": b.BIG_ENDIAN
}, ("first", b.uint8), ("flags", b.array(8, b.boolean)), ("last", b.uint8)]

nested_array_struct = [{
    "endianness": b.BIG_ENDIAN
}, ("first", b.uint8), ("matrix", b.array(3, b.array(3, b.uint8))),
                       ("last", b.uint8)]

simple_struct = [("length", b.uint8), ("ok", b.boolean)]

offset_struct = [("length", b.uint8, {"offset": 1})]
Пример #10
0
import bread as b

from .vendor.six.moves import range

def padded_hex(pad_count):
    return lambda x: ("0x%%0%dx" % (pad_count)) % (x)

EMPTY_BLOCK = 0xff

# File management structure (starts at 0x8000)
compressed_sav_file = [
    # Up to 32 files (songs) can be saved to one cartridge
    ("filenames", b.array(32, b.string(8))),
    # Each file has a monotonically increasing version number
    ("file_versions", b.array(32, b.byte)),
    b.padding(30 * 8),
    ("sram_init_check", b.string(2)),  # set to 'jk' on init
    # The file that is currently active
    ("active_file", b.byte),
    # Table mapping blocks to files.
    ("block_alloc_table", b.array(191, b.byte))
]

# Should be 0x4000 bytes long
sample_kit = [
    ("magic_number", b.string(2)),  # should be 0x60, 0x40
    # The address of the first byte after each sample, or 0 if the sample is
    # unused
    ("sample_end_addresses", b.array(15, b.string(2))),
    b.padding(2 * 8),
    # For samples with names less than three characters long, sample names are
Пример #11
0
import bread as b

from .vendor.six.moves import range

def padded_hex(pad_count):
    return lambda x: ("0x%%0%dx" % (pad_count)) % (x)

EMPTY_BLOCK = 0xff

# File management structure (starts at 0x8000)
compressed_sav_file = [
    # Up to 32 files (songs) can be saved to one cartridge
    ("filenames", b.array(32, b.string(8))),
    # Each file has a monotonically increasing version number
    ("file_versions", b.array(32, b.byte)),
    b.padding(30 * 8),
    ("sram_init_check", b.string(2)),  # set to 'jk' on init
    # The file that is currently active
    ("active_file", b.byte),
    # Table mapping blocks to files.
    ("block_alloc_table", b.array(191, b.byte))
]

# Should be 0x4000 bytes long
sample_kit = [
    ("magic_number", b.string(2)),  # should be 0x60, 0x40
    # The address of the first byte after each sample, or 0 if the sample is
    # unused
    ("sample_end_addresses", b.array(15, b.string(2))),
    b.padding(2 * 8),
    # For samples with names less than three characters long, sample names are