uhd

def hex_to_bytes (s):

    if len (s) & 0x1:

        raise ValueError, "Length must be even"

    r = []

    for i in range (0, len(s), 2):

        r.append (int (s[i:i+2], 16))

    return r

class ihx_rec (object):

    def __init__ (self, addr, type, data):

        self.addr = addr

        self.type = type

        self.data = data

class ihx_file (object):

    def __init__ (self):

        self.pat = re.compile (r':[0-9A-F]{10,}')

    def read (self, file):

        r = []

        for line in file:

            line = line.strip().upper ()

            if not self.pat.match (line):

                raise ValueError, "Invalid hex record format"

            bytes = hex_to_bytes (line[1:])

            sum = reduce (lambda x, y: x + y, bytes, 0) % 256

            if sum != 0:

                raise ValueError, "Bad hex checksum"

            lenx = bytes[0]

            addr = (bytes[1] << 8) + bytes[2]

            type = bytes[3]

            data = bytes[4:-1]

            if lenx != len (data):

                raise ValueError, "Invalid hex record (bad length)"

            if type != 0:

                break;

            r.append (ihx_rec (addr, type, data))

        return r

def get_code (filename):

    """Read the intel hex format file FILENAME and return a tuple

    of the code starting address and a list of bytes to load there.

    """

    f = open (filename, 'r')

    ifx = ihx_file ()

    r = ifx.read (f)

    r.sort (lambda a,b: a.addr - b.addr)

    code_start = r[0].addr

    code_end = r[-1].addr + len (r[-1].data)

    code_len = code_end - code_start

    code = [0] * code_len

    for x in r:

        a = x.addr

        l = len (x.data)

        code[a-code_start:a-code_start+l] = x.data

    return (code_start, code)

    (start_addr, bytes) = get_code (filename)

    image = rom_header + code_header + bytes + trailer

    return image

def build_shell_script (out, ihx_filename, rev):

    image = build_eeprom_image (ihx_filename, rev)

    out.write ('#!/bin/sh\n')

    out.write ('usrper -x load_firmware /usr/local/share/usrp/rev%d/std.ihx\n' % rev)

    out.write ('sleep 1\n')

    # print "len(image) =", len(image)

    i2c_addr = 0x50

    rom_addr = 0x00

    hex_image = map (lambda x : "%02x" % (x,), image)

    while (len (hex_image) > 0):

        l = min (len (hex_image), 16)

        out.write ('usrper i2c_write 0x%02x %02x%s\n' %

                   (i2c_addr, rom_addr, ''.join (hex_image[0:l])))

        hex_image = hex_image[l:]

        rom_addr = rom_addr + l

        out.write ('sleep 1\n')

    usage = "usage: %prog -r REV [options] bootfile.ihx"

    if len (args) != 1:

    ihx_filename = args[0]

    build_shell_script (sys.stdout, ihx_filename, options.rev)