pyinst: usb488.py annotate

annotate usb488.py @ 8:381ed6f3d2d7

Save handle after using it, makes the code look nicer.

author	Daniel O'Connor <darius@dons.net.au>
date	Sat, 16 May 2009 23:32:33 +0930
parents	813e183cfd49
children	a6b746654ae2

rev	line source
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	1 #!/usr/bin/env python
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	2
7 813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	3 # Copyright (c) 2009
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	4 # Daniel O'Connor <darius@dons.net.au>. All rights reserved.
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	5 #
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	6 # Redistribution and use in source and binary forms, with or without
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	7 # modification, are permitted provided that the following conditions
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	8 # are met:
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	9 # 1. Redistributions of source code must retain the above copyright
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	10 # notice, this list of conditions and the following disclaimer.
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	11 # 2. Redistributions in binary form must reproduce the above copyright
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	12 # notice, this list of conditions and the following disclaimer in the
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	13 # documentation and/or other materials provided with the distribution.
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	14 #
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	15 # THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	16 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	17 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	18 # ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	19 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	20 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	21 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	22 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	23 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	24 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	25 # SUCH DAMAGE.
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	26 #
813e183cfd49 Add 2 clause BSDL. Daniel O'Connor <darius@dons.net.au> parents: 5 diff changeset	27
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	28 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	29 # Spec/info..
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	30 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	31 # http://www.usb.org/developers/devclass_docs/USBTMC_1_006a.zip
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	32 # http://svn.openmoko.org/developers/werner/ahrt/host/tmc/README
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	33 # http://www.home.agilent.com/agilent/redirector.jspx?action=ref&cname=AGILENT_EDITORIAL&ckey=1189335&lc=eng&cc=US&nfr=-35560.0.00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	34 # linux-2.6.29.3/drivers/usb/class/usbtmc.c
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	35 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	36
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	37 import usb
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	38
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	39 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	40 # The usual SCPI commands are wrapped before being sent.
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	41 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	42 # Write:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	43 # Offset Field Size Value Description
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	44 # 0 MsgID 1 0x01 DEV_DEP_MSG_OUT
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	45 # 1 bTag 1 0x01 Varies with each transfer
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	46 # 2 bTagInverse 1 0xfe Inverse of previous field
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	47 # 3 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	48 # 4 TransferSize 4 0x06
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	49 # 5 .. 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	50 # 6 .. 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	51 # 7 .. 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	52 # 8 bmTransferAttr 1 0x01 1 == end of msg
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	53 # 9 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	54 # 10 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	55 # 11 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	56 # 12 Msg itself 1 0x2a '*'
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	57 # 13 1 0x49 'I'
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	58 # 14 1 0x44 'D'
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	59 # 15 1 0x4e 'N'
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	60 # 16 1 0x3f '?'
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	61 # 17 1 0x0a '\n'
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	62 # 18-19 Alignment 2 0x0000 Bring into 4 byte alignment
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	63 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	64 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	65 # Send a read request:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	66 # Offset Field Size Value Description
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	67 # 0 MsgID 1 0x02 REQUEST_DEV_DEP_MSG_IN
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	68 # 1 bTag 1 0x02 Varies with each transfer
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	69 # 2 bTagInverse 1 0xfd Inverse of previous field
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	70 # 3 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	71 # 4 TransferSize 4 0x64
8 381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	72 # 5 .. 0x00# 6 .. 0x00
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	73 # 7 .. 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	74 # 8 bmTransferAttr 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	75 # 9 Term char 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	76 # 10 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	77 # 11 Reserved 1 0x00
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	78
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	79 # No libusb versions of these available
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	80 USB_CLASS_APP_SPECIFIC = 254
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	81 USB_SUBCLASS_TMC = 3
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	82 USB_PROTOCOL_488 = 1
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	83
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	84 # USB488 message IDs
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	85 DEV_DEP_MSG_OUT = 1
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	86 REQUEST_DEV_DEP_MSG_IN = 2
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	87 DEV_DEP_MSG_IN = 2
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	88
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	89 class USB488Device(object):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	90 def __init__(self, vendor = None, product = None, serial = None, path = None):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	91 """Search for a USB488 class device, if specified vendor,
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	92 product, serial and path will refine the search"""
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	93
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	94 busses = usb.busses()
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	95
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	96 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	97 # Search for the device we want
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	98 #
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	99 found = False
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	100 for bus in busses:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	101 for dev in bus.devices:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	102 # Skip ones that don't match
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	103 if vendor != None and dev.idVendor != vendor:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	104 continue
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	105 if product != None and dev.idProduct != product:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	106 continue
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	107 if serial != None and dev.idSerialNumber != serial:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	108 continue
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	109 if path != None and dev.filename != path:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	110 continue
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	111
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	112 # The libusb examples say you can check for device
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	113 # class and then open, however in that case you can't
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	114 # find the endpoint number which seems pretty useless
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	115 # unless you want to hard code everything.
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	116 for confidx in xrange(len(dev.configurations)):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	117 for iface in dev.configurations[confidx].interfaces:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	118 for altif in iface:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	119 # Check if this is a USB488 capable interface
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	120 if altif.interfaceClass == USB_CLASS_APP_SPECIFIC and \
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	121 altif.interfaceSubClass == USB_SUBCLASS_TMC and \
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	122 altif.interfaceProtocol == USB_PROTOCOL_488:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	123 found = True
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	124 break
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	125 if found:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	126 break
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	127 if found:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	128 break
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	129 if found:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	130 break
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	131 if found:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	132 break
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	133 if not found:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	134 raise "Could not find a suitable USB device"
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	135
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	136 # Open the device and claim the USB interface that supports the spec
8 381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	137 handle = dev.open()
381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	138 handle.setConfiguration(dev.configurations[confidx].value)
381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	139 handle.claimInterface(altif.interfaceNumber)
381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	140 handle.setAltInterface(altif.alternateSetting)
381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	141 self.dev = dev
381ed6f3d2d7 Save handle after using it, makes the code look nicer. Daniel O'Connor <darius@dons.net.au> parents: 7 diff changeset	142 self.handle = handle
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	143
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	144 # Get some info for humans
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	145 self.vendname = self.handle.getString(dev.iManufacturer, 1024)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	146 self.prodname = self.handle.getString(dev.iProduct, 1024)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	147 self.serial = self.handle.getString(dev.iSerialNumber, 1024)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	148
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	149 # Determine the endpoints for each operation type
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	150 self.intrep = self.bulkinep = self.bulkoutep = None
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	151
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	152 for ep in altif.endpoints:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	153 if ep.type == usb.ENDPOINT_TYPE_INTERRUPT and \
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	154 ep.address & usb.ENDPOINT_IN == usb.ENDPOINT_IN:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	155 self.intrep = ep.address
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	156
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	157 if ep.type == usb.ENDPOINT_TYPE_BULK:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	158 if ep.address & usb.ENDPOINT_IN == usb.ENDPOINT_IN:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	159 self.bulkinep = ep.address
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	160 else:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	161 self.bulkoutep = ep.address
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	162 self.maxPacket = ep.maxPacketSize
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	163
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	164 # Required for 488.2 devices, optional otherwise
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	165 if self.intrep == None:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	166 print "Can't find interrupt endpoint"
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	167
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	168 # Data from the scope (mandatory)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	169 if self.bulkinep == None:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	170 raise "Can't find bulk-in endpoint"
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	171
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	172 # Data to the scope (mandatory)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	173 if self.bulkoutep == None:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	174 raise "Can't find bulk-out endpoint"
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	175
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	176 self.tag = 1
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	177
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	178 def __str__(self):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	179 rtn = "Mfg: %s Prod: %s" % (self.vendname, self.prodname)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	180 if self.serial != "":
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	181 rtn += " S/N: " + self.serial
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	182
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	183 return rtn
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	184
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	185 def incrtag(self):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	186 self.tag += 1
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	187 if self.tag == 0:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	188 self.tag += 1
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	189
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	190 def write(self, data):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	191 """Send data (string) to the scope"""
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	192
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	193 orddata = map(ord, data)
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	194 # The device needs a \n at the end, enfore this
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	195 if orddata[-1] != '\n':
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	196 orddata += [ord('\n')]
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	197 datalen = len(orddata)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	198
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	199 # Build the packet
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	200 pkt = [ DEV_DEP_MSG_OUT, self.tag, ~self.tag & 0xff, 0x00,
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	201 datalen & 0xff, datalen >> 8 & 0xff, datalen >> 16 & 0xff,
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	202 datalen >> 24 & 0xff, 1, 0, 0, 0 ]
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	203
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	204 # Add the data
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	205 pkt = pkt + orddata
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	206
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	207 # Align to 4 bytes
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	208 alignlen = ((len(pkt) / 4) + 1) * 4
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	209 pkt = pkt + [0] * (alignlen - len(pkt))
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	210
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	211 # Bump the tag
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	212 self.incrtag()
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	213
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	214 # Split it up into maxPacket sized chunks and send..
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	215 while len(pkt) > 0:
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	216 chunk = pkt[0:self.maxPacket]
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	217 pkt = pkt[self.maxPacket:]
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	218
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	219 #print "Sending %s bytes of data: %s" % (len(chunk), chunk)
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	220 wrote = self.handle.bulkWrite(self.bulkoutep, chunk)
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	221 if wrote != len(chunk):
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	222 raise "Short write, got %d, expected %d" % (wrote, len(chunk))
a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	223
5 51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	224 def read(self, timeout = 0.2):
51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	225 """Read data from the device, waits for up to timeout seconds for each USB transaction"""
51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	226
51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	227 # Mangle into milliseconds
51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	228 _timeout = int(timeout * 1000.0)
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	229
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	230 # Maximum we accept at once
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	231 # Was 2^31 - 1 but that seems to make things take too long to
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	232 # read (perhaps libusb tries to malloc it..)
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	233 datalen = 10240
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	234 data = []
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	235
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	236 while True:
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	237 # Ask the device to send us something
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	238 pkt = [ REQUEST_DEV_DEP_MSG_IN, self.tag, ~self.tag & 0xff, 0x00,
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	239 datalen & 0xff, datalen >> 8 & 0xff, datalen >> 16 & 0xff,
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	240 datalen >> 24 & 0xff, 0, 0, 0, 0]
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	241
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	242 # Expected tag
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	243 exptag = self.tag
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	244
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	245 # Bump tag
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	246 self.incrtag()
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	247
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	248 # Send it
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	249 #print "Sending " + str(pkt)
5 51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	250 wrote = self.handle.bulkWrite(self.bulkoutep, pkt, _timeout)
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	251 if wrote != len(pkt):
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	252 print "Short write, got %d, expected %d" % (wrote, len(pkt))
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	253
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	254 #print "Reading.."
5 51d1fc44a753 Add variable timeout for reads. Daniel O'Connor <darius@dons.net.au> parents: 4 diff changeset	255 read = self.handle.bulkRead(self.bulkinep, datalen, _timeout)
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	256 #print "Read %s bytes: %s" % (len(read), str(read))
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	257
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	258 if read[0] != DEV_DEP_MSG_IN:
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	259 raise "Unexpected Msg ID, got %s expected %d" % (read[0], DEV_DEP_MSG_IN)
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	260 if read[1] != exptag:
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	261 raise "Unexpected tag, got %d expected %d" % (read[1], exptag)
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	262 if read[2] != ~exptag & 0xff:
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	263 raise "Unexpected tag inverse, got %d expected %d" % (read[1], ~exptag & 0xff)
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	264
1 e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	265 actualdata = read[4] \| read[5] << 8 \| read[6] << 16 \| read[7] << 24
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	266 #print "Computed datalen is %d" % (actualdata)
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	267 data += read[12:12 + actualdata]
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	268 if read[8] & 0x01:
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	269 #print "Breaking out due to EOM"
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	270 break
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	271
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	272 # Stringify result for easier consumption
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	273 result = reduce(lambda x, y: x+y, map(chr, data))
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	274 # Trim off \n if present
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	275 if result[-1] == '\n':
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	276 result = result[0:-1]
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	277
e2089824735a Make the read support work properly & check what the device gives us. Daniel O'Connor <darius@dons.net.au> parents: 0 diff changeset	278 return result
0 a43a47dfc902 First stab at code that actually works! Daniel O'Connor <darius@dons.net.au> parents: diff changeset	279

Mercurial > ~darius > hgwebdir.cgi > pyinst

annotate usb488.py @ 8:381ed6f3d2d7