#!/usr/bin/env python

import pyad2usb.ad2usb
import time
import signal
import traceback
import sys

running = True

def signal_handler(signal, frame):
    global running

    running = False

def handle_open(sender, args):
    print 'opened', args

def handle_close(sender, args):
    print 'closed', args

def handle_read(sender, args):
    print 'read', args

def handle_write(sender, args):
    print 'write', args

def handle_attached(sender, args):
    print 'attached', args

def handle_detached(sender, args):
    print 'detached', args

def handle_firmware(stage):
    if stage == pyad2usb.ad2usb.util.Firmware.STAGE_BOOT:
        print 'Rebooting device..'

        handle_firmware.upload_tick = 0
    elif stage == pyad2usb.ad2usb.util.Firmware.STAGE_LOAD:
        print 'Waiting for boot loader..'
    elif stage == pyad2usb.ad2usb.util.Firmware.STAGE_UPLOADING:
        if handle_firmware.upload_tick == 0:
            sys.stdout.write('Uploading firmware.')

        handle_firmware.upload_tick += 1

        if handle_firmware.upload_tick % 30 == 0:
            sys.stdout.write('.')
            sys.stdout.flush()
    elif stage == pyad2usb.ad2usb.util.Firmware.STAGE_DONE:
        print "\r\nDone!"

signal.signal(signal.SIGINT, signal_handler)

try:
    #overseer = pyad2usb.ad2usb.Overseer(attached_event=handle_attached, detached_event=handle_detached)
    #overseer.start()

    #print pyad2usb.ad2usb.devices.SerialDevice.find_all()

    #a2u = overseer.get_device()

    #a2u = pyad2usb.ad2usb.Overseer.create()

    dev = pyad2usb.ad2usb.devices.SerialDevice(interface='/dev/ttyUSB0')
    #dev = pyad2usb.ad2usb.devices.USBDevice()#serial='A101A429', description='FT232R USB UART')

    #a2u = pyad2usb.ad2usb.AD2USB(dev)
    #a2u.on_open += handle_open
    #a2u.on_close += handle_close
    #a2u.on_read += handle_read
    #a2u.on_write += handle_write

    dev.on_open += handle_open
    dev.on_close += handle_close
    #dev.on_read += handle_read
    #dev.on_write += handle_write


    #a2u.open()
    #a2u.open(baudrate=19200, interface='/dev/ttyUSB0')

    #dev.open(baudrate=115200, interface='/dev/ttyUSB5')
    #dev.open(baudrate=19200, interface='/dev/ttyUSB0')
    dev.open()

    pyad2usb.ad2usb.util.Firmware.upload(dev, 'tmp/ademcoemu_V2_2a_6.hex', handle_firmware)

    while running:
        time.sleep(0.1)

    #a2u.close()
    dev.close()
    #overseer.close()

except Exception, err:
    traceback.print_exc(err)