pixelpulse/ceeVendorReq.py at master · meeg/pixelpulse · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#!/usr/bin/env python2
# -*- coding: utf8 -*-

import sys
import usb.core
import atexit
import pixelpulse
from optparse import OptionParser
import random, time
from cee import *

def clip(v, min, max):
	if v>max: return max
	if v<min: return min
	return v

class CEEChannel(object):
	def __init__(self, cee, index, name, show):
		self.cee = cee
		cee.setA()
		cee.setB()
		self.i=0
		self.v=0
		self.voltageGain = 1
		self.currentGain = 1
		self.driving='v'
		stateOpts = ['source', 'measure']
		self.voltageChan = pixelpulse.AnalogChannel('Voltage '+name,     'V',  0,  5,   'source',
		                            stateOptions=stateOpts, showGraph=show, onSet=self.setVoltage)
		self.currentChan = pixelpulse.AnalogChannel('Current '+name,     'mA', -200, 200,  'measure',
		                            stateOptions=stateOpts, showGraph=show, onSet=self.setCurrent)
		self.gainChan = pixelpulse.AnalogChannel("Gain "+name, 'x', 1, 64, 'output', showGraph=False, onSet=self.setGain)
		self.channels = [self.voltageChan, self.currentChan, self.gainChan]
		self.index = index
		self.name = name

	def setDriving(self, driving):
		if self.driving != driving:
			self.updateNeeded = True
			self.driving = driving
			if self.driving == 'v':
				self.currentChan.setState('measure')
				self.voltageChan.setState('source')
			else:
				self.currentChan.setState('source')
				self.voltageChan.setState('measure')


	def setVoltage(self, chan, volts, state=None):
		if state is not None:
			self.setDriving('v' if state=='source' else 'i')
		if volts is not None:
			self.v = clip(volts, 0, 5)
			self.cee.set(self.index, v=self.v)

	def setCurrent(self, chan, ma, state=None):
		if state is not None:
			self.setDriving('i' if state=='source' else 'v')
		if ma is not None:
			self.i = clip(ma, -200, 200)
			self.cee.set(self.index, i=self.i/1000.0)

	def getChanData(self, replypkt):
		return [(self.voltageChan, replypkt[self.name.lower()+'_v']),
				(self.currentChan, replypkt[self.name.lower()+'_i']*1000),
				(self.gainChan, self.getGain(self.name.lower()))]

	closest=lambda self,a,l:min(l,key=lambda x:abs(x-a))

	def getGain(self, name):
		name = self.name.lower()
		driving = self.driving
		mapping = {'a':{'v':1, 'i':0}, 'b':{'v':2,'i':3}}
		return self.cee.gains[mapping[name][driving]]

	def setGain(self, chan, gain, state=None):
		gains = [16, 1, 2, 4, 0.5, 32, 8, 64]
		gain = self.closest(gain, gains)
		name = self.name.lower()
		driving = self.driving
		mapping = {'a':{'v':1, 'i':0}, 'b':{'v':2,'i':3}}
		chan = mapping[name][driving]
		oldGain = self.cee.gains[mapping[name][driving]]
		if driving == 'v':
			self.voltageChan.max = float(self.voltageChan.max) * oldGain / gain
		elif driving == 'i':
			self.currentChan.max = float(self.currentChan.max) * oldGain / gain
			self.currentChan.min = float(self.currentChan.min) * oldGain / gain
		self.cee.setGain(chan, gain)
		server.updateConfig()

class CEE_vendor_req(pixelpulse.Device):
	def __init__(self):
		stateOpts = ['source', 'measure']

		self.cee = CEE()

		self.chanA = CEEChannel(self.cee, 0, 'A', True)
		self.chanB = CEEChannel(self.cee, 1, 'B', True)
		self.ceechannels = [self.chanA, self.chanB]

		self.channels = sum([x.channels for x in self.ceechannels], [])

	def start(self, server):
		server.poll(self.poll)

	def poll(self):
		data = self.cee.readADC()
		return sum([x.getChanData(data) for x in self.ceechannels], [])

if __name__ == '__main__':
	dev = CEE_vendor_req()
	server = pixelpulse.DataServer(dev)
	server.start(openWebBrowser=False)