mirror of
https://github.com/brmlab/edubrm.git
synced 2025-06-08 21:03:59 +02:00
139 lines
4.1 KiB
Python
139 lines
4.1 KiB
Python
from PyQt4.QtGui import QWidget
|
|
from PyQt4.QtGui import QGraphicsScene
|
|
from PyQt4.QtGui import QPainterPath
|
|
from PyQt4.QtGui import QPen
|
|
from PyQt4.QtGui import QColor
|
|
from PyQt4.QtCore import pyqtSlot
|
|
from PyQt4.QtCore import SIGNAL
|
|
from PyQt4.QtCore import QObject
|
|
from PyQt4.QtCore import QTimer
|
|
from ModuleCUi import Ui_ModuleC
|
|
from device import Device
|
|
|
|
class ModuleCWidget(QWidget):
|
|
|
|
def __init__(self):
|
|
QWidget.__init__(self)
|
|
self.ui = Ui_ModuleC()
|
|
self.ui.setupUi(self)
|
|
|
|
self.timer = QTimer()
|
|
QObject.connect(self.timer, SIGNAL("timeout()"), self.read_inputs)
|
|
|
|
self.data1 = 200*[0.0]
|
|
self.data2 = 200*[0.0]
|
|
|
|
self.mode = ['DC', 'L']
|
|
|
|
@pyqtSlot()
|
|
def on_radioAC_clicked(self):
|
|
self.mode[0] = 'AC'
|
|
self.setup_hw()
|
|
|
|
@pyqtSlot()
|
|
def on_radioDC_clicked(self):
|
|
self.mode[0] = 'DC'
|
|
self.setup_hw()
|
|
|
|
@pyqtSlot()
|
|
def on_radioL_clicked(self):
|
|
self.mode[1] = 'L'
|
|
self.setup_hw()
|
|
|
|
@pyqtSlot()
|
|
def on_radioC_clicked(self):
|
|
self.mode[1] = 'C'
|
|
self.setup_hw()
|
|
|
|
@pyqtSlot()
|
|
def on_radioLC_clicked(self):
|
|
self.mode[1] = 'LC'
|
|
self.setup_hw()
|
|
|
|
@pyqtSlot()
|
|
def on_btnStart_clicked(self):
|
|
self.ui.btnStart.setEnabled(False)
|
|
self.ui.btnStop.setEnabled(True)
|
|
if self.mode == ['DC', 'L']:
|
|
self.dev.setout(1, 1)
|
|
if self.mode == ['DC', 'C']:
|
|
self.dev.setout(2, 1)
|
|
if self.mode == ['DC', 'LC']:
|
|
self.dev.setout(3, 1)
|
|
|
|
@pyqtSlot()
|
|
def on_btnStop_clicked(self):
|
|
self.ui.btnStart.setEnabled(True)
|
|
self.ui.btnStop.setEnabled(False)
|
|
if self.mode == ['DC', 'L']:
|
|
self.dev.setout(1, 0)
|
|
if self.mode == ['DC', 'C']:
|
|
self.dev.setout(2, 0)
|
|
if self.mode == ['DC', 'LC']:
|
|
self.dev.setout(3, 0)
|
|
|
|
def setup_hw(self):
|
|
if self.mode == ['DC', 'L']:
|
|
self.dev.switches(1<<0)
|
|
if self.mode == ['DC', 'C']:
|
|
self.dev.switches(1<<1)
|
|
if self.mode == ['DC', 'LC']:
|
|
self.dev.switches(1<<2)
|
|
|
|
def setup_scene(self, scene):
|
|
scene.addLine(-5, 200-0, 610, 200-0)
|
|
scene.addLine(0, 200+5, 0, 200-205)
|
|
scene.addLine(-5, 200- 50, 10, 200- 50)
|
|
scene.addLine(-5, 200-100, 10, 200-100)
|
|
scene.addLine(-5, 200-150, 10, 200-150)
|
|
scene.addSimpleText('4.0').moveBy(-40, 0-10)
|
|
scene.addSimpleText('3.0').moveBy(-40, 50-10)
|
|
scene.addSimpleText('2.0').moveBy(-40, 100-10)
|
|
scene.addSimpleText('1.0').moveBy(-40, 150-10)
|
|
scene.addSimpleText('0.0').moveBy(-40, 200-10)
|
|
|
|
def tick_DC(self, u, i):
|
|
self.data1.pop(0)
|
|
self.data2.pop(0)
|
|
self.data1.append(u)
|
|
self.data2.append(i)
|
|
self.scene1 = QGraphicsScene()
|
|
self.scene2 = QGraphicsScene()
|
|
self.setup_scene(self.scene1)
|
|
self.setup_scene(self.scene2)
|
|
self.scene1.addSimpleText('[U]').moveBy(-39, 220-10)
|
|
self.scene2.addSimpleText('[I]').moveBy(-39, 220-10)
|
|
path = QPainterPath()
|
|
path.moveTo(0,200-self.data1[0]*50)
|
|
for i in xrange(1,200):
|
|
path.lineTo(3*(i+1), 200-self.data1[i]*50)
|
|
self.scene1.addPath(path, QPen(QColor(0,0,255), 3))
|
|
path = QPainterPath()
|
|
path.moveTo(0,200-self.data2[0]*50)
|
|
for i in xrange(1,200):
|
|
path.lineTo(3*(i+1), 200-self.data2[i]*50)
|
|
self.scene2.addPath(path, QPen(QColor(0,0,255), 3))
|
|
self.ui.graph1.setScene(self.scene1)
|
|
self.ui.graph2.setScene(self.scene2)
|
|
|
|
def read_inputs(self):
|
|
r = self.dev.read()
|
|
|
|
if self.mode[0] == 'DC':
|
|
u = r[0]/1023.0 * 3.3 # TODO: change this if we change opamp
|
|
i = 1 # TODO: read real current
|
|
self.tick_DC(u, i)
|
|
|
|
class ModuleC():
|
|
|
|
def __init__(self):
|
|
self.title = 'L/C/LC circuit'
|
|
self.widget = ModuleCWidget()
|
|
|
|
def start(self):
|
|
self.widget.dev = Device()
|
|
self.widget.setup_hw()
|
|
self.widget.timer.start(25)
|
|
|
|
def stop(self):
|
|
self.widget.timer.stop()
|