#!/usr/bin/python
#
#  spreadspace pic utils
#
#
#  Copyright (C) 2011 Christian Pointner <equinox@spreadspace.org>
#
#  This file is part of spreadspace pic utils.
#
#  spreadspace pic utils is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  any later version.
#
#  spreadspace pic utils is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with spreadspace pic utils. If not, see <http://www.gnu.org/licenses/>.
#

'''spreadspace simple pic downloader.'''

VERSION = '0.1'

def open_serial(device, baud):
    import os
    import tty
    import termios

    baudrates = { 50: termios.B50, 75: termios.B75, 110: termios.B110, 134: termios.B134,
                  150: termios.B150, 200: termios.B200, 300: termios.B300, 600: termios.B600,
                  1200: termios.B1200, 1800: termios.B1800, 2400: termios.B2400, 4800: termios.B4800,
                  9600: termios.B9600, 19200: termios.B19200, 38400: termios.B38400,
                  57600: termios.B57600, 115200: termios.B115200, 230400: termios.B230400 }

    baudreate = termios.B19200
    try:
        baudrate = baudrates[int(baud)]
    except (KeyError, ValueError):
        print "ERROR: invalid baudrate"
        sys.exit(3)

    try:
        dev = os.open(device, os.O_RDWR | os.O_NOCTTY)
        tty.setraw(dev, termios.TCSAFLUSH)
        tio = termios.tcgetattr(dev)
        tio[4] = tio[5] = baudrate
        termios.tcsetattr(dev, termios.TCSAFLUSH, tio)
        termios.tcflush(dev, termios.TCIFLUSH)
        return dev

    except OSError, msg:
        print "ERROR: opening serial device: %s" % msg
        sys.exit(3)
    except termios.error, msg:
        print "ERROR: configuring serial device: %s" % msg
        sys.exit(3)


def load_hex(file):
    import os

    fin = file
    if fin == '-':
        fin = sys.stdin
    elif not os.path.isfile(fin):
        print "ERROR: File not found: %s" % fin
        sys.exit(1)

    data = {}
    ih = IntelHex(fin)
    for a in ih.addresses():
        if a/2 not in data.keys():
            data[a/2] = 0
        if a%2 == 0:
            data[a/2] += ih[a]
        else:
            data[a/2] += (ih[a] << 8)

    return data


if __name__ == '__main__':
    import getopt
    import sys

    from intelhex import IntelHex

    usage = '''spreadspace simple pic downloader.
Usage:
    python downloader.py [options] INFILE

Arguments:
    INFILE      name of hex file for downloading.
                Use '-' for reading from stdin.

Options:
    -h, --help              this help message.
    -v, --version           version info.
    --device=N              the serial port to use (default: /dev/ttyUSB0).
    --baud=N                baudrate to use (default: 19200).
'''

    device = "/dev/ttyUSB0"
    baudrate = 19200

    try:
        opts, args = getopt.getopt(sys.argv[1:], "hv",
                                  ["help", "version", "device=", "baud="])

        for o, a in opts:
            if o in ("-h", "--help"):
                print(usage)
                sys.exit(0)
            elif o in ("-v", "--version"):
                print(VERSION)
                sys.exit(0)
            elif o in ("--device"):
                device = a
            elif o in ("--baud"):
                baudrate = a

        if not args:
            raise getopt.GetoptError('Input file is not specified')

        if len(args) > 1:
            raise getopt.GetoptError('Too many arguments')

    except getopt.GetoptError, msg:
        print "ERROR: %s" % msg
        print usage
        sys.exit(2)

    dev = open_serial(device, baudrate)
    data = load_hex(args[0])

    addrs = data.keys()
    addrs.sort()
    for a in addrs:
        print "%05d: %04X" % (a, data[a])