Mercurial > lcfOS

comparison python/utils/devices.py @ 292:534b94b40aa8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fixup reorganize
author Windel Bouwman
date Wed, 27 Nov 2013 08:06:42 +0100
parents python/devices.py@9e350a7dde98
children
comparison
equal deleted inserted replaced
290:7b38782ed496 292:534b94b40aa8
1 import sys
2 import usb
3
4 # Global device list to which devices are registered.
5 devices = {}
6
7 def registerDevice(chipId):
8 """ Decorator to register a device """
9 def wrapper(dev):
10 devices[chipId] = dev
11 return dev
12 return wrapper
13
14 # Global interface dictionary.
15 interfaces = {}
16
17 def registerInterface(vid_pid):
18 def wrapper(iface):
19 interfaces[vid_pid] = iface
20 return iface
21 return wrapper
22
23 def createInterfaces():
24 """ Create a list of detected interfaces """
25 ctx = usb.UsbContext()
26
27 # Retrieve all usb devices:
28 devs = ctx.DeviceList
29 keys = interfaces.keys()
30
31 # Filter function to filter only registered interfaces:
32 def filt(usbiface):
33 return (usbiface.VendorId, usbiface.ProductId) in keys
34 def buildInterface(usbiface):
35 key = (usbiface.VendorId, usbiface.ProductId)
36 iface = interfaces[key]
37 return iface(usbiface)
38 return [buildInterface(uif) for uif in filter(filt, devs)]
39
40 class Device:
41 """
42 Base class for a device possibly connected via an interface.
43 """
44 def __init__(self, iface):
45 # Store the interface through which this device is connected:
46 assert isinstance(iface, Interface)
47 self.iface = iface
48
49 class Interface:
50 """
51 Generic interface class. Connected via Usb to a JTAG interface.
52 Possibly is connected with a certain chip.
53 """
54 def createDevice(self):
55 """ Try to get the device connected to this interface """
56 if self.ChipId in devices:
57 return devices[self.ChipId](self)
58 raise STLinkException('No device found!')
59
60 class STLinkException(Exception):
61 """ Exception used for interfaces and devices """
62 pass
63