def translate(lang=None):
global app, t, t1
if lang is None:
lang = QtCore.QLocale.system().name()
t = QtCore.QTranslator()
t.load("lang/" + lang, os.path.dirname(__file__))
app.installTranslator(t)
t1 = QtCore.QTranslator()
t1.load("qt_" + lang,
QtCore.QLibraryInfo.location(QtCore.QLibraryInfo.TranslationsPath))
app.installTranslator(t1)
def newRegister(self):
reg = dio.REGEDIT(self.commandQ)
#self.registerLayout.addWidget(reg)
#self.registers.append(reg)
#TODO: Convert layout to listwidget to enable re-ordering
regItem = QtWidgets.QListWidgetItem()
regItem.setSizeHint(QtCore.QSize(200, 40))
self.registerList.addItem(regItem)
self.registerList.setItemWidget(regItem, reg)
self.registers.append(regItem)
def translators(langDir, lang=None):
"""
create a list of translators
@param langDir a path containing .qm translation
@param lang the preferred locale, like en_IN.UTF-8, fr_FR.UTF-8, etc.
@result a list of QtCore.QTranslator instances
"""
if lang==None:
lang=QtCore.QLocale.system().name()
result=[]
qtTranslator=QtCore.QTranslator()
qtTranslator.load("qt_" + lang,
QtCore.QLibraryInfo.location(QtCore.QLibraryInfo.TranslationsPath))
result.append(qtTranslator)
# path to the translation files (.qm files)
sparkTranslator=QtCore.QTranslator()
sparkTranslator.load(lang, langDir);
result.append(sparkTranslator)
return result
def exportSvg(self):
from utilities.Qt import QtSvg
path, _filter = QtWidgets.QFileDialog.getSaveFileName(self, 'Save File', '~/')
if path:
generator = QtSvg.QSvgGenerator()
generator.setFileName(path)
target_rect = QtCore.QRectF(0, 0, 800, 600)
generator.setSize(target_rect.size().toSize())#self.size())
generator.setViewBox(self.rect())
generator.setTitle("Your title")
generator.setDescription("some description")
p = QtGui.QPainter()
p.begin(generator)
self.render(p)
p.end()
def abort(self):
if self.codeThread.isRunning():
self.log.append(
'''<span style="color:red;">----------Kill Signal(Doesn't work yet. Restart the application)-----------</span>'''
)
self.codeThread.quit()
self.codeThread.terminate()
del self.codeThread
self.codeThread = QtCore.QThread()
self.codeEval = self.codeObject(self.REGISTERS)
self.codeEval.moveToThread(self.codeThread)
self.codeEval.finished.connect(self.codeThread.quit)
self.codeEval.logThis.connect(
self.appendLog) #Connect to the log window
self.codeThread.started.connect(self.codeEval.execute)
self.codeThread.finished.connect(self.codeFinished)
class uploadObject(QtCore.QObject):
finished = QtCore.pyqtSignal()
logThis = QtCore.pyqtSignal(str)
logThisPlain = QtCore.pyqtSignal(bytes)
fname = ''
p = None
def __init__(self):
super(AppWindow.uploadObject, self).__init__()
def config(self, mode, p, fname):
self.p = p
self.fname = fname
self.mode = mode
def execute(self):
if self.mode == 'compileupload':
try:
import subprocess
fname = '.'.join(self.fname.split('.')[:-1])
cmd = 'avr-gcc -Wall -O2 -mmcu=%s -o "%s" "%s"' % (
'atmega328p', fname, self.fname)
self.logThis.emit(
'''<span style="color:green;">Compiling for Atmega328p (Uno)</span>'''
)
print(cmd)
res = subprocess.getstatusoutput(cmd)
if res[0] != 0:
self.logThis.emit(
'''<span style="color:red;">Compile Error: %s</span>'''
% res[1])
self.finished.emit()
return
else:
self.logThis.emit(
'''<span style="color:white;">%s</span><br>''' %
res[1])
cmd = 'avr-objcopy -j .text -j .data -O ihex "%s" "%s.hex"' % (
fname, fname)
res = subprocess.getstatusoutput(cmd)
self.logThis.emit(
'''<span style="color:white;">%s</span><br>''' %
res[1])
cmd = 'avr-objdump -S "%s" > "%s.lst"' % (fname, fname)
res = subprocess.getstatusoutput(cmd)
self.logThis.emit(
'''<span style="color:white;">%s</span><br>''' %
res[1])
if self.fname[-2:] in ['.c', '.C']:
self.fname = self.fname[:-2] + '.hex' #Replace .c with .hex
self.mode = 'upload'
self.logThis.emit(
'''<span style="color:green;">Generated Hex File</span>'''
)
self.logThis.emit(
'''<span style="color:green;">Finished Compiling: Generated Hex File</span>'''
)
except Exception as err:
self.logThis.emit(
'''<span style="color:red;">Failed to Compile:%s</span>'''
% str(err))
if self.p.connected:
if self.mode == 'upload':
try:
self.p.fd.setRTS(0)
time.sleep(0.01)
self.p.fd.setRTS(1)
time.sleep(0.4)
dude = uploader.Uploader(self.p.fd,
hexfile=self.fname,
logger=self.logThis)
dude.program()
dude.verify()
self.p.fd.setRTS(0)
time.sleep(0.01)
self.p.fd.setRTS(1)
time.sleep(0.2)
self.p.get_version()
self.logThis.emit(
'''<span style="color:green;">Finished upload</span>'''
)
except Exception as err:
self.logThis.emit(
'''<span style="color:red;">Failed to upload</span>'''
)
self.finished.emit()
def __init__(self, parent=None, **kwargs):
super(AppWindow, self).__init__(parent)
self.setupUi(self)
self.VERSION = REGISTERS.VERSIONNUM
self.SPECIALS = REGISTERS.SPECIALS
self.REGISTERS = REGISTERS.REGISTERS
self.EXAMPLES_DIR = REGISTERS.EXAMPLES
self.ADC_PINS = REGISTERS.ADC
self.docks = [self.leftdock, self.rightdock]
self.sensorList = []
self.controllerList = []
self.monitoring = True
self.logRegisters = True
self.userHexRunning = False
self.uploadingHex = False
self.autoUpdateUserRegisters = False
self.CFile = None #'~/kuttyPy.c'
self.ipy = None
self.setTheme("material")
examples = [
a for a in os.listdir(
os.path.join(path["examples"], self.EXAMPLES_DIR))
if ('.py' in a) and a is not 'kuttyPy.py'
] #.py files except the library
self.exampleList.addItems(examples)
blinkindex = self.exampleList.findText('blink.py')
if blinkindex != -1: #default example. blink.py present in examples directory
self.exampleList.setCurrentIndex(blinkindex)
######## PYTHON CODE
self.codeThread = QtCore.QThread()
self.codeEval = self.codeObject(self.REGISTERS)
self.codeEval.moveToThread(self.codeThread)
self.codeEval.finished.connect(self.codeThread.quit)
self.codeEval.logThis.connect(
self.appendLog) #Connect to the log window
self.logThis.connect(self.appendLog) #Connect to the log window
self.logThisPlain.connect(
self.appendLogPlain) #Connect to the log window
self.serialGaugeSignal.connect(self.setSerialgauge)
self.codeThread.started.connect(self.codeEval.execute)
self.codeThread.finished.connect(self.codeFinished)
######### C CODE UPLOADER
self.uploadThread = QtCore.QThread()
self.UploadObject = self.uploadObject()
self.UploadObject.moveToThread(self.uploadThread)
self.UploadObject.finished.connect(self.uploadThread.quit)
self.UploadObject.logThis.connect(
self.appendLog) #Connect to the log window
self.UploadObject.logThisPlain.connect(
self.appendLogPlain) #Connect to the log window. add plain text
self.logThis.connect(self.appendLog) #Connect to the log window
self.uploadThread.started.connect(self.UploadObject.execute)
self.uploadThread.finished.connect(self.codeFinished)
self.commandQ = []
self.btns = {}
self.registers = []
self.addPins()
self.statusBar = self.statusBar()
self.makeBottomMenu()
global app
self.initializeCommunications()
self.pending = {
'status': myTimer(constants.STATUS_UPDATE_INTERVAL),
'update': myTimer(constants.AUTOUPDATE_INTERVAL),
}
serialgaugeoptions = {
'name': 'Serial Monitor',
'init': print,
'read': None,
'fields': ['Value'],
'min': [0],
'max': [255]
}
self.serialGauge = dio.DIOSENSOR(self, serialgaugeoptions)
self.startTime = time.time()
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateEverything)
self.timer.start(20)
#Auto-Detector
self.shortlist = KuttyPyLibUno.getFreePorts()
class AppWindow(QtWidgets.QMainWindow, layout.Ui_MainWindow):
p = None
logThis = QtCore.pyqtSignal(str)
logThisPlain = QtCore.pyqtSignal(bytes)
serialGaugeSignal = QtCore.pyqtSignal(int)
def __init__(self, parent=None, **kwargs):
super(AppWindow, self).__init__(parent)
self.setupUi(self)
self.VERSION = REGISTERS.VERSIONNUM
self.SPECIALS = REGISTERS.SPECIALS
self.REGISTERS = REGISTERS.REGISTERS
self.EXAMPLES_DIR = REGISTERS.EXAMPLES
self.ADC_PINS = REGISTERS.ADC
self.docks = [self.leftdock, self.rightdock]
self.sensorList = []
self.controllerList = []
self.monitoring = True
self.logRegisters = True
self.userHexRunning = False
self.uploadingHex = False
self.autoUpdateUserRegisters = False
self.CFile = None #'~/kuttyPy.c'
self.ipy = None
self.setTheme("material")
examples = [
a for a in os.listdir(
os.path.join(path["examples"], self.EXAMPLES_DIR))
if ('.py' in a) and a is not 'kuttyPy.py'
] #.py files except the library
self.exampleList.addItems(examples)
blinkindex = self.exampleList.findText('blink.py')
if blinkindex != -1: #default example. blink.py present in examples directory
self.exampleList.setCurrentIndex(blinkindex)
######## PYTHON CODE
self.codeThread = QtCore.QThread()
self.codeEval = self.codeObject(self.REGISTERS)
self.codeEval.moveToThread(self.codeThread)
self.codeEval.finished.connect(self.codeThread.quit)
self.codeEval.logThis.connect(
self.appendLog) #Connect to the log window
self.logThis.connect(self.appendLog) #Connect to the log window
self.logThisPlain.connect(
self.appendLogPlain) #Connect to the log window
self.serialGaugeSignal.connect(self.setSerialgauge)
self.codeThread.started.connect(self.codeEval.execute)
self.codeThread.finished.connect(self.codeFinished)
######### C CODE UPLOADER
self.uploadThread = QtCore.QThread()
self.UploadObject = self.uploadObject()
self.UploadObject.moveToThread(self.uploadThread)
self.UploadObject.finished.connect(self.uploadThread.quit)
self.UploadObject.logThis.connect(
self.appendLog) #Connect to the log window
self.UploadObject.logThisPlain.connect(
self.appendLogPlain) #Connect to the log window. add plain text
self.logThis.connect(self.appendLog) #Connect to the log window
self.uploadThread.started.connect(self.UploadObject.execute)
self.uploadThread.finished.connect(self.codeFinished)
self.commandQ = []
self.btns = {}
self.registers = []
self.addPins()
self.statusBar = self.statusBar()
self.makeBottomMenu()
global app
self.initializeCommunications()
self.pending = {
'status': myTimer(constants.STATUS_UPDATE_INTERVAL),
'update': myTimer(constants.AUTOUPDATE_INTERVAL),
}
serialgaugeoptions = {
'name': 'Serial Monitor',
'init': print,
'read': None,
'fields': ['Value'],
'min': [0],
'max': [255]
}
self.serialGauge = dio.DIOSENSOR(self, serialgaugeoptions)
self.startTime = time.time()
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateEverything)
self.timer.start(20)
#Auto-Detector
self.shortlist = KuttyPyLibUno.getFreePorts()
def newRegister(self):
reg = dio.REGEDIT(self.commandQ)
#self.registerLayout.addWidget(reg)
#self.registers.append(reg)
#TODO: Convert layout to listwidget to enable re-ordering
regItem = QtWidgets.QListWidgetItem()
regItem.setSizeHint(QtCore.QSize(200, 40))
self.registerList.addItem(regItem)
self.registerList.setItemWidget(regItem, reg)
self.registers.append(regItem)
def addPins(self):
'''
This function adds pins in the right order. this should be moved to a configuration file!
'''
self.btns['C'] = dio.REGVALS('C')
self.leftlayout.addWidget(self.btns['C'])
s = QtWidgets.QSpacerItem(20, 250, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Minimum)
self.leftlayout.addItem(s)
for name in ['PC0', 'PC1', 'PC2', 'PC3', 'PC4', 'PC5']: #left dock
checkbox = dio.widget(name,
self.commandQ,
extra=self.SPECIALS.get(name, ''))
self.leftlayout.addWidget(checkbox)
self.btns[name] = checkbox
self.btns['B'] = dio.REGVALS('B')
self.rightlayout.addWidget(self.btns['B'])
for name in ['PB5', 'PB4', 'PB3', 'PB2', 'PB1', 'PB0']: #right dock
checkbox = dio.widget(name,
self.commandQ,
extra=self.SPECIALS.get(name, ''))
self.rightlayout.addWidget(checkbox)
self.btns[name] = checkbox
s = QtWidgets.QSpacerItem(20, 20, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Minimum)
self.rightlayout.addItem(s)
self.btns['D'] = dio.REGVALS('D')
self.rightlayout.addWidget(self.btns['D'])
for name in ['PD7', 'PD6', 'PD5', 'PD4', 'PD3', 'PD2', 'PD1',
'PD0']: #right dock
checkbox = dio.widget(name,
self.commandQ,
extra=self.SPECIALS.get(name, ''))
self.rightlayout.addWidget(checkbox)
self.btns[name] = checkbox
def tabChanged(self, index):
if index != 0: #examples/editor tab. disable monitoring
self.monitoring = False
for a in self.docks:
a.hide()
else: #Playground . enable monitoring and control.
self.monitoring = True
self.autoRefreshUserRegisters.setChecked(False)
self.userRegistersAutoRefresh(False)
for a in self.docks:
a.show()
################USER CODE SECTION####################
class codeObject(QtCore.QObject):
finished = QtCore.pyqtSignal()
logThis = QtCore.pyqtSignal(str)
code = ''
def __init__(self, REGISTERS):
super(AppWindow.codeObject, self).__init__()
self.PORTMAP = {v: k
for k, v in REGISTERS.items()
} #Lookup port name based on number
self.compiled = ''
self.SR = None
self.GR = None
self.evalGlobals = {}
self.evalGlobals['getReg'] = self.getReg
self.evalGlobals['setReg'] = self.setReg
self.evalGlobals['print'] = self.printer
def setCode(self, code, **kwargs):
try:
self.compiled = compile(code.encode(), '<string>', mode='exec')
except SyntaxError as err:
error_class = err.__class__.__name__
detail = err.args[0]
line_number = err.lineno
return '''<span style="color:red;">%s at line %d : %s</span>''' % (
error_class, line_number, detail)
except Exception as err:
error_class = err.__class__.__name__
detail = err.args[0]
cl, exc, tb = sys.exc_info()
line_number = traceback.extract_tb(tb)[-1][1]
return '''<span style="color:red;">%s at line %d: %s</span>''' % (
error_class, line_number, detail)
self.SR = kwargs.get('setReg')
self.GR = kwargs.get('getReg')
self.evalGlobals = kwargs
self.evalGlobals[
'getReg'] = self.getReg #Overwrite these three. They will be wrapped.
self.evalGlobals['setReg'] = self.setReg
self.evalGlobals['print'] = self.printer
return ''
def printer(self, *args):
self.logThis.emit('''<span style="color:cyan;">%s</span>''' %
(' '.join([str(a) for a in args])))
def setReg(self, reg, value):
html = u'''<pre><span>W\u2193</span>{0:s}\t{1:d}\t0x{1:02x} / 0b{1:08b}</pre>'''.format(
self.PORTMAP.get(reg, ''), value)
self.logThis.emit(html)
self.SR(reg, value)
def getReg(self, reg):
value = self.GR(reg)
html = u'''<pre><span>R\u2191</span>{0:s}\t{1:d}\t0x{1:02x} / 0b{1:08b}</pre>'''.format(
self.PORTMAP.get(reg, ''), value)
self.logThis.emit(html)
return value
def execute(self):
#old = sys.stdout
#olde = sys.stderr
#sys.stdout = self.toLog(self.logThis)
#sys.stderr = self.toLog(self.logThis)
try:
exec(self.compiled, {}, self.evalGlobals)
except SyntaxError as err:
error_class = err.__class__.__name__
detail = err.args[0]
line_number = err.lineno
self.logThis.emit(
'''<span style="color:red;">%s at line %d : %s</span>''' %
(error_class, line_number, detail))
except Exception as err:
error_class = err.__class__.__name__
detail = err.args[0]
cl, exc, tb = sys.exc_info()
line_number = traceback.extract_tb(tb)[-1][1]
self.logThis.emit(
'''<span style="color:red;">%s at line %d: %s</span>''' %
(error_class, line_number, detail))
#sys.stdout = old
#sys.stderr = olde
self.logThis.emit("Finished executing user code")
self.finished.emit()
def runCode(self):
#if self.p:
# try:
# self.p.fd.read() #Clear junk
# self.p.fd.close()
# except:pass
if self.codeThread.isRunning():
print('one code is already running')
return
self.log.clear() #clear the log window
self.log.setText(
'''<span style="color:green;">----------User Code Started-----------</span>'''
)
kwargs = {}
for a in dir(self.p):
attr = getattr(self.p, a)
if inspect.ismethod(attr) and a[:2] != '__':
kwargs[a] = attr
compilemsg = self.codeEval.setCode(
'{0:s}'.format(self.userCode.toPlainText()), **kwargs)
if len(compilemsg):
self.log.append(compilemsg)
return
self.codeThread.start()
self.userCode.setStyleSheet("border: 3px dashed #53ffff;")
self.tabs.setTabEnabled(0, False)
def codeFinished(self):
print('finished')
self.tabs.setTabEnabled(0, True)
self.userCode.setStyleSheet("")
self.uploadingHex = False
def abort(self):
if self.codeThread.isRunning():
self.log.append(
'''<span style="color:red;">----------Kill Signal(Doesn't work yet. Restart the application)-----------</span>'''
)
self.codeThread.quit()
self.codeThread.terminate()
del self.codeThread
self.codeThread = QtCore.QThread()
self.codeEval = self.codeObject(self.REGISTERS)
self.codeEval.moveToThread(self.codeThread)
self.codeEval.finished.connect(self.codeThread.quit)
self.codeEval.logThis.connect(
self.appendLog) #Connect to the log window
self.codeThread.started.connect(self.codeEval.execute)
self.codeThread.finished.connect(self.codeFinished)
def getReg(self, reg, record=True):
val = self.p.getReg(reg)
if record:
self.updatedRegs[reg] = [0, val]
return val
def setReg(self, reg, val, record=True):
self.p.setReg(reg, val)
if record:
self.updatedRegs[reg] = [1, val]
return val
def appendLog(self, txt):
self.log.append(txt)
def appendLogPlain(self, txt):
self.log.moveCursor(QtGui.QTextCursor.End)
self.log.insertPlainText(txt.decode('ascii'))
def setSerialgauge(self, val):
self.serialGauge.setValue([val])
def genLog(self):
html = '''<table border="1" align="center" cellpadding="1" cellspacing="0" style="font-family:arial,helvetica,sans-serif;font-size:9pt;">
<tbody><tr><td colspan="4">%s</td></tr>''' % (time.ctime())
#html+='''<tr><td style="background-color:#77cfbb;">R/W</td><td style="background-color:#77cfbb;">REGISTER</td>
#<td style="background-color:#77cfbb;">Value</td><td style="background-color:#77cfbb;">Hex/Binary</td></tr>'''
for a in self.updatedRegs:
row = self.updatedRegs[a]
html += u'''
<tr>
<td>{0:s}</td>
<td>{1:s}</td>
<td>{2:d}</td>
<td>0b{2:08b} | 0x{2:02x}</td>
</tr>
'''.format(u'W \u2193' if row[0] else u'R \u2191', a, row[1])
html += "</tbody></table>"
self.log.setHtml(html)
def userRegistersAutoRefresh(self, state):
self.autoUpdateUserRegisters = state
def updateEverything(self):
self.locateDevices()
if not self.checkConnectionStatus(): return
#KuttyPy monitor has handed over control to native code. act as serial monitor/ debug window
if self.uploadingHex:
return
if self.userHexRunning:
t = self.p.fd.read(self.p.fd.in_waiting)
if len(t):
self.serialGaugeSignal.emit(t[0])
self.logThisPlain.emit(t)
return
#self.setTheme('material')
if self.codeThread.isRunning():
return
if self.autoUpdateUserRegisters:
for a in range(self.registerList.count()):
self.registerList.itemWidget(
self.registerList.item(a)).execute()
while len(self.commandQ):
if not self.centralWidget().isEnabled():
return
a = self.commandQ.pop(0)
if a[0] == 'DSTATE': #Digital Out ['DSTATE','Pxx',state]
pname = 'PORT' + a[1][1].upper()
bit = int(a[1][2])
reg = self.getReg(pname)
reg &= ~(1 << bit)
if (a[2]): reg |= (1 << bit)
self.setReg(pname, reg)
elif a[0] == 'DTYPE': #Digital pin I/O ['DTYPE','Pxx',state]
pname = 'DDR' + a[1][1].upper()
bit = int(a[1][2])
reg = self.getReg(pname)
reg &= ~(1 << bit)
if (a[2]): reg |= (1 << bit)
self.setReg(pname, reg)
elif a[0] == 'WRITE': #['WRITE','REGNAME',val]
self.setReg(a[1], a[2])
elif a[0] == 'READ': #['READ','REGNAME',function]
val = self.getReg(a[1])
a[2](val)
elif a[0] == 'CNTR1': #['CNTR1',output with setValue function]
cl = self.getReg('TCNT1L', False)
ch = self.getReg('TCNT1H', False)
a[1].setValue(cl | (ch << 8))
elif a[0] == 'ADC': #['ADC',ADMUX,output with setValue function, to log, or not to log]
self.setReg('ADMUX', a[1], a[3])
self.setReg('ADCSRA', 196 | 1, a[3])
adcl = self.getReg('ADCL', a[3])
adch = self.getReg('ADCH', a[3])
a[2].setValue(adcl | (adch << 8))
for a in self.sensorList:
if a[0].isVisible():
a[0].setValue(a[0].read())
if self.enableLog.isChecked():
if self.clearLog.isChecked() and len(self.updatedRegs):
self.log.clear()
if len(self.updatedRegs):
self.genLog()
self.updatedRegs = OrderedDict()
if self.pending['status'].ready() and self.monitoring:
val = self.p.getReg(self.getRegs[self.currentRegister][0])
self.getRegs[self.currentRegister][1](
self.getRegs[self.currentRegister][0], val)
self.currentRegister += 1
if self.currentRegister == len(self.getRegs):
self.currentRegister = 0
for a in ['B', 'C', 'D']:
self.btns[a].setRegs(self.p.REGSTATES)
if self.pending['update'].ready():
pass
#print('update')
def updateInputs(self, port, value):
portchar = port[3]
for a in range(8):
name = 'P' + portchar + str(a)
btn = self.btns.get(name, None)
if btn is None: continue
btn.nameIn.setChecked((value >> a) & 1)
if name in self.ADC_PINS: #ADC
if btn.currentPage == 2: #ADC Page displayed
self.commandQ.append(['ADC', btn.ADMUX, btn, btn.logstate])
elif type(btn) == dio.DIOCNTR and btn.currentPage == 2: # CNTR
self.commandQ.append(['CNTR1', btn.slider])
def newStepperController(self):
if self.p.connected:
dialog = dio.DIOSTEPPER(self, total=200, device=self.p)
dialog.launch()
self.sensorList.append([dialog, None])
############ I2C SENSORS #################
def I2CScan(self):
if self.p.connected:
x = self.p.I2CScan()
print('Responses from: ', x)
for a in self.sensorList:
a[0].setParent(None)
a[1].setParent(None)
self.sensorList = []
for a in self.controllerList:
a[0].setParent(None)
a[1].setParent(None)
self.controllerList = []
for a in x:
s = self.p.sensors.get(a, None)
if s is not None:
btn = QtWidgets.QPushButton(s['name'] + ':' + hex(a))
dialog = dio.DIOSENSOR(self, s)
btn.clicked.connect(dialog.launch)
self.sensorLayout.addWidget(btn)
self.sensorList.append([dialog, btn])
continue
s = self.p.controllers.get(a, None)
if s is not None:
btn = QtWidgets.QPushButton(s['name'] + ':' + hex(a))
dialog = dio.DIOCONTROL(self, s)
btn.clicked.connect(dialog.launch)
self.sensorLayout.addWidget(btn)
self.controllerList.append([dialog, btn])
continue
s = self.p.special.get(a, None)
if s is not None:
btn = QtWidgets.QPushButton(s['name'] + ':' + hex(a))
dialog = dio.DIOROBOT(self, s)
btn.clicked.connect(dialog.launch)
self.sensorLayout.addWidget(btn)
self.controllerList.append([dialog, btn])
continue
def loadExample(self, filename):
self.userCode.setPlainText(
open(os.path.join(path["examples"], self.EXAMPLES_DIR, filename),
"r").read())
########################### UPLOAD HEX FILE #######################
class uploadObject(QtCore.QObject):
finished = QtCore.pyqtSignal()
logThis = QtCore.pyqtSignal(str)
logThisPlain = QtCore.pyqtSignal(bytes)
fname = ''
p = None
def __init__(self):
super(AppWindow.uploadObject, self).__init__()
def config(self, mode, p, fname):
self.p = p
self.fname = fname
self.mode = mode
def execute(self):
if self.mode == 'compileupload':
try:
import subprocess
fname = '.'.join(self.fname.split('.')[:-1])
cmd = 'avr-gcc -Wall -O2 -mmcu=%s -o "%s" "%s"' % (
'atmega328p', fname, self.fname)
self.logThis.emit(
'''<span style="color:green;">Compiling for Atmega328p (Uno)</span>'''
)
print(cmd)
res = subprocess.getstatusoutput(cmd)
if res[0] != 0:
self.logThis.emit(
'''<span style="color:red;">Compile Error: %s</span>'''
% res[1])
self.finished.emit()
return
else:
self.logThis.emit(
'''<span style="color:white;">%s</span><br>''' %
res[1])
cmd = 'avr-objcopy -j .text -j .data -O ihex "%s" "%s.hex"' % (
fname, fname)
res = subprocess.getstatusoutput(cmd)
self.logThis.emit(
'''<span style="color:white;">%s</span><br>''' %
res[1])
cmd = 'avr-objdump -S "%s" > "%s.lst"' % (fname, fname)
res = subprocess.getstatusoutput(cmd)
self.logThis.emit(
'''<span style="color:white;">%s</span><br>''' %
res[1])
if self.fname[-2:] in ['.c', '.C']:
self.fname = self.fname[:-2] + '.hex' #Replace .c with .hex
self.mode = 'upload'
self.logThis.emit(
'''<span style="color:green;">Generated Hex File</span>'''
)
self.logThis.emit(
'''<span style="color:green;">Finished Compiling: Generated Hex File</span>'''
)
except Exception as err:
self.logThis.emit(
'''<span style="color:red;">Failed to Compile:%s</span>'''
% str(err))
if self.p.connected:
if self.mode == 'upload':
try:
self.p.fd.setRTS(0)
time.sleep(0.01)
self.p.fd.setRTS(1)
time.sleep(0.4)
dude = uploader.Uploader(self.p.fd,
hexfile=self.fname,
logger=self.logThis)
dude.program()
dude.verify()
self.p.fd.setRTS(0)
time.sleep(0.01)
self.p.fd.setRTS(1)
time.sleep(0.2)
self.p.get_version()
self.logThis.emit(
'''<span style="color:green;">Finished upload</span>'''
)
except Exception as err:
self.logThis.emit(
'''<span style="color:red;">Failed to upload</span>'''
)
self.finished.emit()
def uploadHex(self):
filename = QtWidgets.QFileDialog.getOpenFileName(
self, " Open a hex file to upload to your KuttyPy", "",
"Hex Files (*.hex)")
if len(filename[0]):
#self.userCode.setStyleSheet("border: 3px dashed #53ffff;")
#self.tabs.setTabEnabled(0,False)
self.uploadingHex = True
self.log.clear()
self.log.setText(
'''<span style="color:cyan;">-- Uploading Code --</span><br>'''
)
self.UploadObject.config('upload', self.p, filename[0])
self.uploadThread.start()
def openFile(self):
filename = QtWidgets.QFileDialog.getOpenFileName(
self, " Open a C file to edit", path["examples"],
"C Files (*.c *.C)")
if len(filename[0]):
self.filenameLabel.setText(filename[0])
self.CFile = filename[0]
self.log.clear()
self.log.setText(
'''<span style="color:cyan;">-- Opened File: %s --</span><br>'''
% filename[0])
if 'inux' in platform.system(): #Linux based system
os.system('%s "%s"' % ('xdg-open', filename[0]))
else:
os.system('%s "%s"' % ('open', filename[0]))
def compileAndUpload(self):
if self.CFile:
self.uploadingHex = True
self.log.clear()
self.log.setText(
'''<span style="color:cyan;">-- Compiling and Uploading Code --</span><br>'''
)
self.UploadObject.config('compileupload', self.p, self.CFile)
self.uploadThread.start()
##############################
def setTheme(self, theme):
self.setStyleSheet("")
self.setStyleSheet(
open(os.path.join(path["themes"], theme + ".qss"), "r").read())
def initializeCommunications(self, port=False):
if self.p:
try:
self.p.fd.close()
except:
pass
if port:
self.p = KuttyPyLibUno.connect(port=port)
else:
self.p = KuttyPyLibUno.connect(autoscan=True)
if self.p.connected:
self.userApplication.setChecked(False)
self.setWindowTitle('KuttyPy Interactive Console [{0:s}]'.format(
self.p.portname))
self.updatedRegs = OrderedDict()
self.currentRegister = 0
self.getRegs = [
('PINB', self.updateInputs),
('PINC', self.updateInputs),
('PIND', self.updateInputs),
]
else:
self.setWindowTitle(
'KuttyPy Interactive Console [ Hardware not detected ]')
def jumpToApplication(self, state):
if self.p:
if state:
self.userHexRunning = True
self.p.fd.write(b'j') #Skip to application (Bootloader resets)
for a in self.docks:
a.setEnabled(False)
self.tabs.setEnabled(False)
self.log.clear()
self.log.setText(
'''<span style="color:cyan;">-- Serial Port Monitor --</span><br>'''
)
self.serialGauge.show()
else:
self.p.fd.setRTS(0) #Trigger a reset
time.sleep(0.01)
self.p.fd.setRTS(1)
time.sleep(0.1)
while self.p.fd.in_waiting:
self.p.fd.read()
self.p.get_version()
for a in self.docks:
a.setEnabled(True)
self.userHexRunning = False
self.tabs.setEnabled(True)
self.serialGauge.hide()
else:
if self.isChecked():
self.setChecked(False)
def makeBottomMenu(self):
try:
self.pushbutton.setParent(None)
except:
pass
self.pushbutton = QtWidgets.QPushButton('Menu')
self.pushbutton.setStyleSheet(
"height: 13px;padding:3px;color: #FFFFFF;")
menu = QtWidgets.QMenu()
menu.addAction('Save Window as Svg', self.exportSvg)
menu.addAction('Open Stepper Controller', self.newStepperController)
#Theme
self.themeAction = QtWidgets.QWidgetAction(menu)
themes = [
a.split('.qss')[0] for a in os.listdir(path["themes"])
if '.qss' in a
]
self.themeBox = QtWidgets.QComboBox()
self.themeBox.addItems(themes)
self.themeBox.currentIndexChanged['QString'].connect(self.setTheme)
self.themeAction.setDefaultWidget(self.themeBox)
menu.addAction(self.themeAction)
self.pushbutton.setMenu(menu)
self.userApplication = QtWidgets.QCheckBox("User App")
self.userApplication.toggled['bool'].connect(self.jumpToApplication)
self.statusBar.addPermanentWidget(self.userApplication)
self.hexUploadButton = QtWidgets.QPushButton("Upload Hex")
self.hexUploadButton.clicked.connect(self.uploadHex)
self.statusBar.addPermanentWidget(self.hexUploadButton)
self.speedbutton = QtWidgets.QComboBox()
self.speedbutton.addItems(['Slow', 'Fast', 'Ultra'])
self.speedbutton.setCurrentIndex(1)
self.speedbutton.currentIndexChanged['int'].connect(self.setSpeed)
self.statusBar.addPermanentWidget(self.speedbutton)
self.statusBar.addPermanentWidget(self.pushbutton)
def setSpeed(self, index):
self.timer.setInterval([100, 20, 5][index])
def locateDevices(self):
try:
L = KuttyPyLibUno.getFreePorts(self.p.portname)
except Exception as e:
print(e)
total = len(L)
menuChanged = False
if L != self.shortlist:
menuChanged = True
if self.p.connected:
if self.p.portname not in L:
self.setWindowTitle('Error : Device Disconnected')
QtWidgets.QMessageBox.warning(
self, 'Connection Error',
'Device Disconnected. Please check the connections')
try:
self.p.fd.close()
self.p.portname = None
except:
pass
self.p.connected = False
self.setWindowTitle(
'KuttyPy Interactive Console [ Hardware not detected ]'
)
elif True in L.values():
reply = QtWidgets.QMessageBox.question(
self, 'Connection', 'Device Available. Connect?',
QtWidgets.QMessageBox.Yes, QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.Yes:
self.initializeCommunications()
#update the shortlist
self.shortlist = L
def checkConnectionStatus(self, dialog=False):
if self.p.connected: return True
else:
if dialog:
QtWidgets.QMessageBox.warning(
self, 'Connection Error',
'Device not connected. Please connect a KuttyPy to the USB port'
)
return False
def updateStatus(self):
if not self.checkConnectionStatus():
self.countLabel.setText('Not Connected')
return
try:
state, cnt = self.p.getStatus()
self.currentState = state
self.countLabel.setText('%s: %d' %
("Running" if state else "Paused", cnt))
except:
self.countLabel.setText('Disconnect!')
self.p.fd.close()
######## WINDOW EXPORT SVG
def exportSvg(self):
from utilities.Qt import QtSvg
path, _filter = QtWidgets.QFileDialog.getSaveFileName(
self, 'Save File', '~/')
if path:
generator = QtSvg.QSvgGenerator()
generator.setFileName(path)
target_rect = QtCore.QRectF(0, 0, 800, 600)
generator.setSize(target_rect.size().toSize()) #self.size())
generator.setViewBox(self.rect())
generator.setTitle("Your title")
generator.setDescription("some description")
p = QtGui.QPainter()
p.begin(generator)
self.render(p)
p.end()
def ipython(
self): #Experimental feature. Import ipython and launch console
if not self.p.connected:
return
from utilities import ipy
if not self.ipy:
self.ipy = ipy.AppWindow(self, kp=self.p)
self.ipy.show()
self.ipy.updateDevice(self.p)
class codeObject(QtCore.QObject):
finished = QtCore.pyqtSignal()
logThis = QtCore.pyqtSignal(str)
code = ''
def __init__(self, REGISTERS):
super(AppWindow.codeObject, self).__init__()
self.PORTMAP = {v: k
for k, v in REGISTERS.items()
} #Lookup port name based on number
self.compiled = ''
self.SR = None
self.GR = None
self.evalGlobals = {}
self.evalGlobals['getReg'] = self.getReg
self.evalGlobals['setReg'] = self.setReg
self.evalGlobals['print'] = self.printer
def setCode(self, code, **kwargs):
try:
self.compiled = compile(code.encode(), '<string>', mode='exec')
except SyntaxError as err:
error_class = err.__class__.__name__
detail = err.args[0]
line_number = err.lineno
return '''<span style="color:red;">%s at line %d : %s</span>''' % (
error_class, line_number, detail)
except Exception as err:
error_class = err.__class__.__name__
detail = err.args[0]
cl, exc, tb = sys.exc_info()
line_number = traceback.extract_tb(tb)[-1][1]
return '''<span style="color:red;">%s at line %d: %s</span>''' % (
error_class, line_number, detail)
self.SR = kwargs.get('setReg')
self.GR = kwargs.get('getReg')
self.evalGlobals = kwargs
self.evalGlobals[
'getReg'] = self.getReg #Overwrite these three. They will be wrapped.
self.evalGlobals['setReg'] = self.setReg
self.evalGlobals['print'] = self.printer
return ''
def printer(self, *args):
self.logThis.emit('''<span style="color:cyan;">%s</span>''' %
(' '.join([str(a) for a in args])))
def setReg(self, reg, value):
html = u'''<pre><span>W\u2193</span>{0:s}\t{1:d}\t0x{1:02x} / 0b{1:08b}</pre>'''.format(
self.PORTMAP.get(reg, ''), value)
self.logThis.emit(html)
self.SR(reg, value)
def getReg(self, reg):
value = self.GR(reg)
html = u'''<pre><span>R\u2191</span>{0:s}\t{1:d}\t0x{1:02x} / 0b{1:08b}</pre>'''.format(
self.PORTMAP.get(reg, ''), value)
self.logThis.emit(html)
return value
def execute(self):
#old = sys.stdout
#olde = sys.stderr
#sys.stdout = self.toLog(self.logThis)
#sys.stderr = self.toLog(self.logThis)
try:
exec(self.compiled, {}, self.evalGlobals)
except SyntaxError as err:
error_class = err.__class__.__name__
detail = err.args[0]
line_number = err.lineno
self.logThis.emit(
'''<span style="color:red;">%s at line %d : %s</span>''' %
(error_class, line_number, detail))
except Exception as err:
error_class = err.__class__.__name__
detail = err.args[0]
cl, exc, tb = sys.exc_info()
line_number = traceback.extract_tb(tb)[-1][1]
self.logThis.emit(
'''<span style="color:red;">%s at line %d: %s</span>''' %
(error_class, line_number, detail))
#sys.stdout = old
#sys.stderr = olde
self.logThis.emit("Finished executing user code")
self.finished.emit()
def __init__(self, parent=None,**kwargs):
super(AppWindow, self).__init__(parent)
self.setupUi(self)
self.setTheme("default")
#self.setTheme("material2")
self.statusBar = self.statusBar()
self.splash = kwargs.get('splash',None)
global app
self.fileBrowser = fileBrowser(thumbnail_directory = 'MCA_thumbnails',app=app, clickCallback = self.loadPlot,recordToHistory = self.recordToHistory, loadList = self.loadList)
self.saveLayout.addWidget(self.fileBrowser)
#Calibration Menu & storage
self.calibrationEnabled = False
self.calPoly = np.poly1d([1,0])
self.calPolyInv = np.poly1d([1,0])
self.calibrationMenu = QtWidgets.QMenu()
self.y=[];
self.y2=[];
self.spectrumTime = time.time()
self.offlineData = True
self._browserPath = '.'
self.saved={}
self.thumbList={}
self.markers=[]
# PLOT creation
self.createMainPlot()
#self.createMainPlot('w','k')
#Region Widget
#self.regionLayout.setAlignment(QtCore.Qt.AlignTop)
self.regionWindow = regionPopup.AppWindow(self,insertRegion=self.insertRegion,getData = self.getData,hltime = self.halflifeTime,changeDirectory = self.changeDirectory,enablePeriodicSpectrumSaving = self.enablePeriodicSpectrumSaving )
self.regionWindow.widgetLayout.setAlignment(QtCore.Qt.AlignTop)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.regionWindow)
#self.regionWindow.setFloating(True)
self.regionWindow.close()
#Spectrum History Widget
self.historyWindow = historyPopup.AppWindow(self,regions = self.regionWindow.regions)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.historyWindow)
self.historyWindow.setFloating(True)
#self.historyWindow.close()
#Define some keyboard shortcuts for ease of use
self.shortcutActions={}
self.shortcuts={"+":self.summation,"s":self.start,"f":self.fit,"u":self.load,"r":self.insertRegion,"g":self.fitWithTail,"t":self.fitWithTail,"h":self.historyWindow.show,"Ctrl+S":self.save,"o":self.selectDevice}
for a in self.shortcuts:
shortcut = QtWidgets.QShortcut(QtGui.QKeySequence(a), self)
shortcut.activated.connect(self.shortcuts[a])
self.shortcutActions[a] = shortcut
# exporter
self.exporter = PQG_ImageExporter(self.plot.plotItem)# pg.exporters.ImageExporter(self.plot.plotItem)
#Auto-Update Menu
self.autoUpdateMenu = QtGui.QMenu()
#self.autoUpdateMenu.addAction("Interval",self.setAutoUpdateTime)
self.autoUpdateTimerAction = QtWidgets.QWidgetAction(self.autoUpdateMenu)
self.autoUpdateTimerInterval = QtWidgets.QSpinBox()
self.autoUpdateTimerInterval.setMinimum(5);self.autoUpdateTimerInterval.setMaximum(3600);self.autoUpdateTimerInterval.setValue(constants.AUTOUPDATE_INTERVAL); self.autoUpdateTimerInterval.setSuffix(' S');
self.autoUpdateTimerInterval.valueChanged['int'].connect(self.setAutoUpdateInterval)
self.autoUpdateTimerAction.setDefaultWidget(self.autoUpdateTimerInterval)
self.autoUpdateMenu.addAction(self.autoUpdateTimerAction)
self.autoUpdateSettings.setMenu(self.autoUpdateMenu)
self.calibWindow = calPopup.AppWindow(self,application=self.setCalibration)
self.splash.showMessage("<h2><font color='Black'>Connecting...</font></h2>", QtCore.Qt.AlignLeft, QtCore.Qt.black)
self.splash.pbar.setValue(7)
self.initializeCommunications()
self.pending = {
'status':myTimer(constants.STATUS_UPDATE_INTERVAL),
'update':myTimer(constants.AUTOUPDATE_INTERVAL),
'current':myTimer(constants.CURRENT_UPDATE_INTERVAL),
'temperature':myTimer(constants.TEMPERATURE_UPDATE_INTERVAL),
'halflife':myTimer(self.regionWindow.decayInterval.value()),
'datadump':myTimer(self.regionWindow.saveAllInterval.value()*60) #convert minutes to seconds
}
self.temperature = decayTools.temperatureHandler()
self.startTime = time.time()
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateEverything)
self.timer.start(200)
#Auto-Detector
self.shortlist=MCALib.getFreePorts(None)
self.deviceSelector.setList(self.shortlist,self.p)
#self.loadPlot('DATA/212Bi.csv')
#self.showGammaMarkers('137Cs')
#self.loadPlot('DATA/eu152.dat')
#self.loadList('DATA/list_sample.csv')
self.splash.showMessage("<h2><font color='Black'>Ready!</font></h2>", QtCore.Qt.AlignLeft, QtCore.Qt.black)
self.splash.pbar.setValue(8)
class AppWindow(QtWidgets.QMainWindow, layout.Ui_MainWindow):
total_bins = 1024
version_number=0
p=None
menu_entries=[]
currentState = False
dataDumpPath = None
calibrationChanged = QtCore.pyqtSignal(object,object, name='calibrationChanged')
plot = None
vLine = None
switchingPlot = False
surfacePlot = None
def __init__(self, parent=None,**kwargs):
super(AppWindow, self).__init__(parent)
self.setupUi(self)
self.setTheme("default")
#self.setTheme("material2")
self.statusBar = self.statusBar()
self.splash = kwargs.get('splash',None)
global app
self.fileBrowser = fileBrowser(thumbnail_directory = 'MCA_thumbnails',app=app, clickCallback = self.loadPlot,recordToHistory = self.recordToHistory, loadList = self.loadList)
self.saveLayout.addWidget(self.fileBrowser)
#Calibration Menu & storage
self.calibrationEnabled = False
self.calPoly = np.poly1d([1,0])
self.calPolyInv = np.poly1d([1,0])
self.calibrationMenu = QtWidgets.QMenu()
self.y=[];
self.y2=[];
self.spectrumTime = time.time()
self.offlineData = True
self._browserPath = '.'
self.saved={}
self.thumbList={}
self.markers=[]
# PLOT creation
self.createMainPlot()
#self.createMainPlot('w','k')
#Region Widget
#self.regionLayout.setAlignment(QtCore.Qt.AlignTop)
self.regionWindow = regionPopup.AppWindow(self,insertRegion=self.insertRegion,getData = self.getData,hltime = self.halflifeTime,changeDirectory = self.changeDirectory,enablePeriodicSpectrumSaving = self.enablePeriodicSpectrumSaving )
self.regionWindow.widgetLayout.setAlignment(QtCore.Qt.AlignTop)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.regionWindow)
#self.regionWindow.setFloating(True)
self.regionWindow.close()
#Spectrum History Widget
self.historyWindow = historyPopup.AppWindow(self,regions = self.regionWindow.regions)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.historyWindow)
self.historyWindow.setFloating(True)
#self.historyWindow.close()
#Define some keyboard shortcuts for ease of use
self.shortcutActions={}
self.shortcuts={"+":self.summation,"s":self.start,"f":self.fit,"u":self.load,"r":self.insertRegion,"g":self.fitWithTail,"t":self.fitWithTail,"h":self.historyWindow.show,"Ctrl+S":self.save,"o":self.selectDevice}
for a in self.shortcuts:
shortcut = QtWidgets.QShortcut(QtGui.QKeySequence(a), self)
shortcut.activated.connect(self.shortcuts[a])
self.shortcutActions[a] = shortcut
# exporter
self.exporter = PQG_ImageExporter(self.plot.plotItem)# pg.exporters.ImageExporter(self.plot.plotItem)
#Auto-Update Menu
self.autoUpdateMenu = QtGui.QMenu()
#self.autoUpdateMenu.addAction("Interval",self.setAutoUpdateTime)
self.autoUpdateTimerAction = QtWidgets.QWidgetAction(self.autoUpdateMenu)
self.autoUpdateTimerInterval = QtWidgets.QSpinBox()
self.autoUpdateTimerInterval.setMinimum(5);self.autoUpdateTimerInterval.setMaximum(3600);self.autoUpdateTimerInterval.setValue(constants.AUTOUPDATE_INTERVAL); self.autoUpdateTimerInterval.setSuffix(' S');
self.autoUpdateTimerInterval.valueChanged['int'].connect(self.setAutoUpdateInterval)
self.autoUpdateTimerAction.setDefaultWidget(self.autoUpdateTimerInterval)
self.autoUpdateMenu.addAction(self.autoUpdateTimerAction)
self.autoUpdateSettings.setMenu(self.autoUpdateMenu)
self.calibWindow = calPopup.AppWindow(self,application=self.setCalibration)
self.splash.showMessage("<h2><font color='Black'>Connecting...</font></h2>", QtCore.Qt.AlignLeft, QtCore.Qt.black)
self.splash.pbar.setValue(7)
self.initializeCommunications()
self.pending = {
'status':myTimer(constants.STATUS_UPDATE_INTERVAL),
'update':myTimer(constants.AUTOUPDATE_INTERVAL),
'current':myTimer(constants.CURRENT_UPDATE_INTERVAL),
'temperature':myTimer(constants.TEMPERATURE_UPDATE_INTERVAL),
'halflife':myTimer(self.regionWindow.decayInterval.value()),
'datadump':myTimer(self.regionWindow.saveAllInterval.value()*60) #convert minutes to seconds
}
self.temperature = decayTools.temperatureHandler()
self.startTime = time.time()
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateEverything)
self.timer.start(200)
#Auto-Detector
self.shortlist=MCALib.getFreePorts(None)
self.deviceSelector.setList(self.shortlist,self.p)
#self.loadPlot('DATA/212Bi.csv')
#self.showGammaMarkers('137Cs')
#self.loadPlot('DATA/eu152.dat')
#self.loadList('DATA/list_sample.csv')
self.splash.showMessage("<h2><font color='Black'>Ready!</font></h2>", QtCore.Qt.AlignLeft, QtCore.Qt.black)
self.splash.pbar.setValue(8)
def setTheme(self,theme):
self.setStyleSheet("")
self.setStyleSheet(open(os.path.join(path["themes"],theme+".qss"), "r").read())
def plotRangeChanged(self,val):
Y = self.plot.plotItem.vb.viewRange()[1]
H = (Y[1]-Y[0])*.95 + Y[0]
for _,a in self.markers:
a.setPos(a.x(), H)
def showAlphaMarkers(self,state):
self.showMarkers(constants.ALPHAS,state)
def showGammaMarkers(self,state):
self.showMarkers(constants.GAMMAS,state)
def showMarkers(self,markerList,state):
for a,b in self.markers:
self.plot.removeItem(a)
self.plot.removeItem(b)
self.markers=[]
H = self.plot.plotItem.vb.viewRange()[1][1]*.95
energies = markerList.get(state,[])
for a in energies:
line = pg.InfiniteLine(angle=90, movable=False)
line.setPos(a)
self.plot.addItem(line, ignoreBounds=True)
text = pg.TextItem(html='<div style="text-align: center"><span style="color: #FFF;font-size: 7pt;">%.1fkeV</span><br><span style="color: #FF0; font-size: 8pt;">%s</span></div>'%(a,energies[a]), anchor=(-0.1,0),border='w', fill=(0, 0, 100, 100))
self.plot.addItem(text)
text.setPos(a, H)
self.markers.append([line,text])
def createMainPlot(self,bg=QtGui.QColor(0,0,0),fg = QtGui.QColor(200,200,200)):
#destroy any old plot
if self.plot:
if self.vLine:
self.plot.removeItem(self.vLine)
self.plot.removeItem(self.chanLabel)
self.plot.removeItem(self.arrow)
self.plot.removeItem(self.markerarrow)
self.plot_area.removeWidget(self.plot)
self.plot.destroy()
# Go about creating a new plot
pg.setConfigOptions(antialias=True, background=bg,foreground=fg)
self.plot=pg.PlotWidget()
self.plot.setMinimumHeight(250)
self.plot_area.addWidget(self.plot)
self.plot.getAxis('left').setGrid(170)
self.plot.getAxis('bottom').setGrid(170); self.plot.getAxis('bottom').setLabel('Channel Number')
self.plot.sigYRangeChanged.connect(self.plotRangeChanged)
if bg=='w': #Light background
self.pen = pg.mkPen((0,0,0), width=1)
self.pen2 = pg.mkPen((255,100,100), width=1)
self.brush = pg.mkBrush((26, 197, 220,150))
else:
self.pen = pg.mkPen((255,255,255), width=1)
self.pen2 = pg.mkPen((255,0,0), width=1)
self.brush = pg.mkBrush((26, 197, 220,100))
self.curve = pg.PlotCurveItem(name = 'Data')
self.curve2 = pg.PlotCurveItem(name = 'Data2')
self.trendline = pg.PlotCurveItem(name = 'background')
self.plot.addItem(self.trendline);
self.fitcurves = []
self.plot.addItem(self.curve);
self.plot.addItem(self.curve2);
self.plot.scene().sigMouseClicked.connect(self.onClick)
self.moveproxy = pg.SignalProxy(self.plot.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
self.arrow = pg.ArrowItem(angle=-60,tipAngle = 90, headLen=7, tailLen=9, tailWidth=5, pen={'color': 'g', 'width': 1})
self.plot.addItem(self.arrow)
self.arrow.setPos(0,0)
#self.vLine = pg.InfiniteLine(angle=90, movable=False)
#self.plot.addItem(self.vLine, ignoreBounds=True)
## Create text object, use HTML tags to specify color/size
self.currentPeak = None
self.markerarrow = pg.ArrowItem(angle=-160,tipAngle = 90, headLen=7, tailLen=9, tailWidth=5, pen={'color': 'r', 'width': 1})
self.plot.addItem(self.markerarrow)
self.markerarrow.setPos(0,0)
self.toggleLog(self.logBox.isChecked()) #Change to log scale if necessary
def enableCalibration(self):
self.calibrateOnOff.setChecked(True)
def enableTemperatureMonitor(self,state):
if state:
self.temperatureLabel.show()
else:
self.temperatureLabel.hide()
def enableCurrentMonitor(self,state):
if state:
self.currentFrame.show()
self.currentMonitorAvailable = True
else:
self.currentFrame.hide()
self.currentMonitorAvailable = False
def getTotalBins(self):
return self.total_bins # self.channelList[self.channelBox.currentIndex()]
def setTotalBins(self,bins):
self.total_bins = bins
self.binLabel.setText('Bins: %d'%(bins))
#self.channelBox.setCurrentIndex(self.channelList.index(bins))
def initializeCommunications(self,port=False):
if self.p:
try:self.p.fd.close()
except:pass
if port:
self.p = MCALib.connect(port = port)
else:
self.p = MCALib.connect(autoscan=True)
self.decayHandler = decayTools.decayHandler()
if self.p.connected:
self.listFrame.hide()
if self.p.activeSpectrum.datatype=='list':
traces = []
for b in range(self.p.activeSpectrum.parameters):
traces.append('%s:%d'%(self.p.portname,b+1))
self.updateTraceList(traces)
if self.p.activeSpectrum.parameters==2: #Dual list mode. Open 2D plots
self.listFrame.show()
from utilities import plot3DTools
if self.surfacePlot:
self.surfacePlot.close()
del self.surfacePlot
self.surfacePlot = plot3DTools.surface3d(self,self.p.activeSpectrum.HISTOGRAM2D,self.p.activeSpectrum.BINS2D)
else:
self.updateTraceList(['%s'%self.p.portname])
try:
self.setTotalBins(self.p.total_bins)
self.version_number = float(self.p.version[-3:])
self.decayHandler.interval = self.regionWindow.decayInterval.value()
#self.p.setSqr1(2000,10) #TODO : REMOVE in production
except Exception as e:
print(e)
self.enableCurrentMonitor(False)
self.enableTemperatureMonitor(False)
if self.p.connected==False:
self.showStatus("System Status | Device not found. Dummy mode.",True)
self.setWindowTitle('MCA : Error : device not found')
else:
self.showStatus("System Status | Connected to device. Version : %s"%(self.p.version))
self.setWindowTitle("CSpark Research : %s"%(self.p.name))
if self.p.VOLTMETER_ENABLED:
self.enableTemperatureMonitor(True)
if self.p.CCS_ENABLED: #Current source monitoring is only available from version 2.0 onwards, and in alpha detector integrated boards only
self.enableCurrentMonitor(True)
self.makeBottomMenu()
self.plot.setLimits(xMin=0,xMax=self.total_bins,yMin=0,yMax=(1<<32));self.plot.setXRange(0,self.total_bins)
def makeBottomMenu(self):
try:self.pushbutton.setParent(None)
except:pass
self.pushbutton = QtWidgets.QPushButton('Menu')
menu = QtWidgets.QMenu()
menu.addAction(self.controlDock.toggleViewAction())
menu.addAction(self.historyWindow.toggleViewAction())
menu.addAction('Set Square Wave', self.sqr1)
if self.version_number>=2.0:
menu.addAction('Set Threshold', self.set_threshold)
menu.addSeparator()
self.plotColorAction = QtWidgets.QAction('Light Theme', menu, checkable=True)
self.plotColorAction.triggered.connect(self.setPlotColor)
menu.addAction(self.plotColorAction)
self.removeCalBox = QtWidgets.QAction('File Load: Remove Calibration', menu, checkable=True)
self.removeCalBox.triggered.connect(menu.show)
menu.addAction(self.removeCalBox)
#self.pileUpAction = QtWidgets.QAction('PileUp Reject', menu, checkable=True)
#self.pileUpAction.triggered.connect(self.pileUpRejection)
#self.pileUpAction.triggered.connect(menu.show)
#menu.addAction(self.pileUpAction)
self.coincidencegate = QtWidgets.QAction('External Gate', menu, checkable=True)
self.coincidencegate.triggered.connect(self.externalGate)
self.coincidencegate.triggered.connect(menu.show)
menu.addAction(self.coincidencegate)
menu.addAction('Set Window Opacity', self.setOpacity)
menu.addAction('Save Window as Svg', self.exportSvg)
#Theme
self.themeAction = QtWidgets.QWidgetAction(menu)
themes = [a.split('.qss')[0] for a in os.listdir(path["themes"]) if '.qss' in a]
self.themeBox = QtWidgets.QComboBox(); self.themeBox.addItems(themes)
self.themeBox.currentIndexChanged['QString'].connect(self.setTheme)
self.themeAction.setDefaultWidget(self.themeBox)
menu.addAction(self.themeAction)
#Alpha Markers
self.markerAction = QtWidgets.QWidgetAction(menu)
self.markerBox = QtWidgets.QComboBox(); self.markerBox.addItems(['Add Alpha Energy Guides']+list(constants.ALPHAS.keys()))
self.markerBox.currentIndexChanged['QString'].connect(self.showAlphaMarkers)
self.markerAction.setDefaultWidget(self.markerBox)
menu.addAction(self.markerAction)
#Gamma Markers
self.markerActionG = QtWidgets.QWidgetAction(menu)
self.markerBoxG = QtWidgets.QComboBox(); self.markerBoxG.addItems(['Add Gamma Energy Guides']+list(constants.GAMMAS.keys()))
self.markerBoxG.currentIndexChanged['QString'].connect(self.showGammaMarkers)
self.markerActionG.setDefaultWidget(self.markerBoxG)
menu.addAction(self.markerActionG)
#Graph Colour
self.traceRow = traceRowWidget(self.curve)
self.colAction = QtWidgets.QWidgetAction(menu)
self.colAction.setDefaultWidget(self.traceRow)
menu.addAction(self.colAction)
#TRACE2
self.traceRow2 = traceRowWidget(self.curve2)
self.colAction2 = QtWidgets.QWidgetAction(menu)
self.colAction2.setDefaultWidget(self.traceRow2)
menu.addAction(self.colAction2)
# Quit
menu.addAction('Exit', self.askBeforeQuit)
self.pushbutton.setMenu(menu)
self.extraLayout.addWidget(self.pushbutton)
#self.statusBar.addPermanentWidget(self.pushbutton)
self.deviceSelector = self.portSelectionDialog()
def changeActiveTrace(self,s):
if self.p.activeSpectrum.datatype=='list':
if self.p.activeSpectrum.parameters==2:
try:
num = int(str(s).split(':')[-1])
self.p.activeSpectrum.selectDataset(num-1)
except:
pass
def updateTraceList(self,traces):
self.activeTrace.clear()
for a in traces:
self.activeTrace.addItem(a)
'''
def pileUpRejection(self,state):
if not self.checkConnectionStatus():return
if self.p.PILEUP_REJECT_ENABLED:
self.p.pileupRejection(state)
'''
def externalGate(self,state):
if not self.checkConnectionStatus():return
if self.p.EXTERNAL_TRIGGER_ENABLED:
self.p.externalGate(state)
def setPlotColor(self,state):
self.switchingPlot = True
if state: #Light background
self.createMainPlot('w','k')
else:
self.createMainPlot()
self.switchingPlot = False
def showTrendline(self):
print('generating trendline')
order = 2
x= self.p.activeSpectrum.xaxis(self.calibrationEnabled)
z = np.polyfit(x, self.y, order)
p= np.poly1d(z)
pen=pg.mkPen('r', width=5)
self.trendline.setData(x,p(x),pen=pen)
def setAutoUpdateInterval(self,val):
self.showStatus("Updated auto-refresh interval to %d seconds"%val,False)
self.pending['update'] = myTimer(val)
def autoUpdateEnable(self,state):
self.progressBar.setValue(0)
self.progressBar.setEnabled(state)
if state:
self.pending['update'].reset()
def updateEverything(self):
# For diagnostics
#if self.regionWindow.saveAllCheckbox.isChecked():
# if self.pending['datadump'].ready():
# self.dataDump()
self.locateDevices()
#self.setTheme("default")
if not self.checkConnectionStatus():return
if self.progressBar.isEnabled():
self.progressBar.setValue(self.pending['update'].progress())
if self.pending['status'].ready():
self.updateStatus()
if self.currentMonitorAvailable:
if self.pending['current'].ready():
self.updateCurrent()
if self.p.VOLTMETER_ENABLED:
if self.pending['temperature'].ready():
self.updateTemperature()
if self.switchingPlot: #Skip this if the plot is being regenerated with new colours
return
if self.autoUpdateBox.isChecked() and self.currentState:#Only do this if it's running
if self.pending['update'].ready():
self.load()
if self.pending['halflife'].ready():
vals = self.summationRaw()
T = self.decayHandler.getElapsedTime()
for a in vals:
a[0].appendPoint (T,a[3])
if self.regionWindow.saveAllCheckbox.isChecked():
if self.pending['datadump'].ready():
self.dataDump()
def halflifeTime(self):
self.pending['halflife'] = myTimer(self.regionWindow.decayInterval.value())
self.showStatus('Count logging interval changed to %s Seconds'%(self.regionWindow.decayInterval.value()) )
def load(self):
if not self.checkConnectionStatus(True):return
try:
self.p.sync() #Get latest data from the hardware. List/Hist.
if self.p.activeSpectrum.datatype=='list':
self.y = self.p.activeSpectrum.getHistogram(trace=0)
if self.p.activeSpectrum.parameters==2: #fetch second one as well
self.y2 = self.p.activeSpectrum.getHistogram(trace=1)
else: #Single parameter list / histogram.
self.y = self.p.activeSpectrum.getHistogram(trace=0)
#self.showStatus("System Status | Data Refreshed : %s"%time.ctime())
m, s = divmod(time.time() - self.startTime, 60)
h, m = divmod(m, 60)
ST = "%d:%02d:%02d" % (h, m, s)
self.clearButton.setText('CLEAR %s'%(ST))
self.clearFits()
self.offlineData = False
self.refreshPlot()
self.recordToHistory(resetTemperature=True)
except Exception as e:
self.showStatus("System Status | Load Error %s"%str(e),True)
def recordToHistory(self,**kwargs):
self.spectrumTime = time.time()- self.startTime
TIME = kwargs.get('time',self.spectrumTime)
TEMP = kwargs.get('temp',self.temperature.get())
if kwargs.get('resetTemperature',False):self.temperature.clear()
self.historyWindow.addSpectrum(np.copy(self.y), time = TIME,temp = TEMP)
def insertRegion(self):
R = decayTools.regionWidget(self.plot,self.deleteRegion,self.p.activeSpectrum.calPolyInv if self.calibrationEnabled else np.poly1d([1,0]),len(self.regionWindow.regions))
self.calibrationChanged.connect(R.updateCalibration)
self.regionWindow.widgetLayout.addWidget(R)
self.regionWindow.regions.append(R)
def removeRegion(self):
#simply switch to the second tab which shows the region list
if len(self.regionWindow.regions):
self.regionWindow.show() #tabWidget.setCurrentWidget(self.additionalParameters)
self.regionWindow.activateWindow() #tabWidget.setCurrentWidget(self.additionalParameters)
else:
self.showStatus("No more regions available for deletion",True)
def showMainTab(self):
self.tabWidget.setCurrentWidget(self.mainTab)
def deleteRegion(self,R):
#R.setParent(None)
self.regionWindow.regions.remove(R)
self.showStatus("Deleted Region",True)
def locateDevices(self):
try:L = MCALib.getFreePorts(self.p.portname)
except Exception as e:print(e)
total = len(L)
menuChanged = False
if L != self.shortlist:
menuChanged = True
self.deviceSelector.setList(L,self.p)
#Check for, and handle disconnect event
if self.p.connected:
if self.p.portname not in L:
self.showStatus('Device Disconnected',True)
self.setWindowTitle('Error : Device Disconnected')
QtWidgets.QMessageBox.warning(self, 'Connection Error', 'Device Disconnected. Please check the connections')
try:
self.p.portname = None
self.p.fd.close()
except:pass
self.p.connected = False
elif True in L.values():
reply = QtWidgets.QMessageBox.question(self, 'Connection', 'Device Available. Connect?', QtWidgets.QMessageBox.Yes, QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.Yes:
self.initializeCommunications()
#update the shortlist
self.shortlist=L
####################
def selectDevice(self):
if self.deviceSelector.exec_():
self.initializeCommunications(port = self.deviceSelector.getSelection())
class portSelectionDialog(QtWidgets.QDialog):
def __init__(self,parent=None):
super(AppWindow.portSelectionDialog, self).__init__(parent)
self.button_layout = QtGui.QVBoxLayout()
self.setLayout(self.button_layout)
self.btns=[]
self.doneButton = QtWidgets.QPushButton("Done")
self.button_layout.addWidget(self.doneButton)
self.doneButton.clicked.connect(self.finished)
def setList(self,L,handler):
for a in self.btns:
a.setParent(None)
del a
self.btns=[]
self.button_group = QtWidgets.QButtonGroup()
#moods[0].setChecked(True)
pos=0
for i in L:
# Add each radio button to the button layout
btn = QtWidgets.QRadioButton(i)
self.button_layout.addWidget(btn)
self.btns.append(btn)
if handler:
if handler.connected:
if handler.portname == i:
btn.setStyleSheet("color:green;")
if not L[i]: #Port in use
btn.setEnabled(False)
self.button_group.addButton(btn, pos)
pos+=1
# Set the layout of the group box to the button layout
#Print out the ID & text of the checked radio button
def finished(self):
if self.button_group.checkedId()!= -1:
self.done(QtWidgets.QDialog.Accepted)
def getSelection(self):
if self.button_group.checkedId()!= -1:
return self.button_group.checkedButton().text()
else:
return False
def checkConnectionStatus(self,dialog=False):
if self.p.connected:return True
else:
if dialog: QtWidgets.QMessageBox.warning(self, 'Connection Error', 'Device not connected. Please connect an MCA to the USB port')
return False
def sqr1(self):
if not self.checkConnectionStatus(True):return
val,ok = QtWidgets.QInputDialog.getDouble(self,"Set Square Wave", 'Enter Frequency for 10% square wave(10Hz - 100KHz)',20e3,0,100e3)
if ok :
if val<8e6:
self.p.setSqr1(val,10)
def set_threshold(self):
if not self.checkConnectionStatus(True):return
val,ok = QtWidgets.QInputDialog.getInt(self,"Set Threshold", 'Enter Number of initial channels to reject [0 -> x].',self.p.threshold,0,1000)
if ok :
self.p.setThreshold(val)
def setOpacity(self):
val,ok = QtWidgets.QInputDialog.getDouble(self,"Set Opacity", 'Enter Opacity (in %)',100,20,100)
if ok :
self.setWindowOpacity(val/100.)
def closeEvent(self, evnt):
evnt.ignore()
self.askBeforeQuit()
def askBeforeQuit(self):
self.calibWindow.close()
reply = QtWidgets.QMessageBox.question(self, 'Warning', 'Really quit?', QtWidgets.QMessageBox.Yes, QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.Yes:
global app
#self.timer.stop()
app.quit()
#sys.exit()
def updateStatus(self):
if not self.checkConnectionStatus():
self.countLabel.setText('Not Connected')
return
try:
if self.p.activeSpectrum.datatype=='list':
self.p.sync()
state,cnt = self.p.getStatus()
self.currentState = state
if self.p.activeSpectrum.datatype=='list':
T = time.time()-self.startTime
if self.p.activeSpectrum.parameters==2:
A = np.sum(self.p.activeSpectrum.spectra[0].data)
B = np.sum(self.p.activeSpectrum.spectra[1].data)
C = self.p.activeSpectrum.totalCoincidences
if self.surfacePlot.isVisible():
self.surfacePlot.setData(self.p.activeSpectrum.HISTOGRAM2D)
self.surfacePlot.countA.display(A)
self.surfacePlot.countB.display(B)
self.surfacePlot.countC.display(C)
self.surfacePlot.labelA.setText('/%d [%.2f%%]'%(A+B,100*A/(A+B)))
self.surfacePlot.labelB.setText('/%d [%.2f%%]'%(A+B,100*B/(A+B)))
self.coincidenceLabel.setText('C: %d [%d , %d]'%(C,A,B) )
self.coincidenceLabel.setText('C: %d [%d , %d]'%(self.p.activeSpectrum.totalCoincidences,A,B) )
self.countLabel.setText('%s: %d'%("%d in %dS"%(self.p.activeSpectrum.totalCoincidences,T) if state else "Paused",cnt))
else:
self.countLabel.setText('%s: %d'%("Running" if state else "Paused",cnt))
except Exception as e:
self.countLabel.setText('Disconnect!')
print(e)
#self.p.fd.close()
def updateCurrent(self):
if not self.checkConnectionStatus():
self.currentLabel.setText('Device not Connected')
return
V = self.p.getCCS()
self.currentLabel.setText('Source Preparation: %.3fV'%(V))
if V>2.5 or V<0.1:self.currentLabel.setStyleSheet("QLabel{color:red}");
else:self.currentLabel.setStyleSheet("QLabel{color:#8F8}");
def updateTemperature(self):
if not self.checkConnectionStatus():
self.temperatureLabel.setText('Device not Connected')
return
T = self.p.getTemperature()
self.temperature.add(T)
self.temperatureLabel.setText('Temperature: %.1f C(%.2f)'%(T,self.temperature.get()))
#if V>2.5 or V<0.1:self.currentLabel.setStyleSheet("QLabel{color:red}");
#else:self.currentLabel.setStyleSheet("QLabel{color:#8F8}");
def clearCount(self):
'''
Reset the total pulse counter
'''
if not self.checkConnectionStatus(True):return
self.p.startCount()
def stateHighlight(self,state):
self.startButton.setProperty("class", "active" if state else "") ; self.startButton.style().unpolish(self.startButton); self.startButton.style().polish(self.startButton)
self.pauseButton.setProperty("class", "" if state else "active") ; self.pauseButton.style().unpolish(self.pauseButton); self.pauseButton.style().polish(self.pauseButton)
def start(self):
if not self.checkConnectionStatus(True):return
self.p.setThreshold(self.p.threshold)
self.showStatus("System Status | Acquisition Started : %s"%time.ctime())
self.p.startHistogram()
self.stateHighlight(True)
def pause(self):
if not self.checkConnectionStatus(True):return
self.showStatus("System Status | Acquisition Stopped : %s"%time.ctime())
self.p.stopHistogram()
self.stateHighlight(False)
def clear(self):
reply = QtWidgets.QMessageBox.question(self, 'Warning', 'Clear the entire histogram?\nThis will erase data stored in the hardware also.', QtWidgets.QMessageBox.No, QtWidgets.QMessageBox.Yes)
if reply == QtWidgets.QMessageBox.Yes:
self.clearPlot()
self.clearAllSums()
self.p.activeSpectrum.clearData()
self.showStatus("System Status | Data cleared : %s"%time.ctime())
if not self.checkConnectionStatus():return
self.p.clearHistogram()
self.pending['update'].reset()
self.historyWindow.clear()
self.temperature.clear()
def clearPlot(self):
#if not self.checkConnectionStatus(True):return
self.y=[];self.y2=[];
self.curve.clear();self.curve2.clear();
self.trendline.clear();
self.clearFits()
self.markerarrow.setPos(0,0)
self.startTime = time.time()
self.clearButton.setText('CLEAR')
def clearFits(self):
for a in self.fitcurves:
a.clear()
self.plot.removeItem(a)
self.fitcurves=[]
def showStatus(self,msg,error=None):
if error: self.statusBar.setStyleSheet("color:#F77")
else: self.statusBar.setStyleSheet("color:#000000")
self.statusBar.showMessage(msg)
def toggleLog(self,state):
if not self.p: return
if not self.p.activeSpectrum: return
self.p.activeSpectrum.setLog(state)
if state:
self.plot.getAxis('left').setLabel('Log(Count)')
else:
self.plot.getAxis('left').setLabel('Count')
if self.p.activeSpectrum.datatype=='list':
self.y = self.p.activeSpectrum.getHistogram(trace=0)
if self.p.activeSpectrum.parameters==2:
self.y2 = self.p.activeSpectrum.getHistogram(trace=1)
else: #Single parameter list / histogram.
self.y = self.p.activeSpectrum.getHistogram(trace=0)
self.refreshPlot()
def launch3D(self):
self.surfacePlot.show()
def refreshPlot(self):
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled)
if self.plotAEnabled.isChecked():
self.curve.setData(x,self.y[:-1], stepMode=True, fillLevel=0,pen = self.pen,brush=self.brush)#, brush=brush,pen = pen)
else:
self.curve.clear()
yMax = max(self.y[:-2])*1.05
if self.p.activeSpectrum.datatype=='list':
if self.p.activeSpectrum.parameters ==2:
brush=(126, 97, 220,100)
if self.plotBEnabled.isChecked():
self.curve2.setData(x,self.y2[:-1], stepMode=True, fillLevel=0, brush=brush,pen = self.pen2)
else:
self.curve2.clear()
yMax = max( max(self.y[:-2]), max(self.y2[:-2]))*1.05
self.plot.setLimits(xMax=max(x)*1.1,yMax = max(10,yMax)*1.05)
self.plot.setYRange(0,max(10,yMax)*1.05)
self.plot.getAxis('bottom').setLabel(self.p.activeSpectrum.get_xlabel())
################## SUMMATION ROUTINES #################
def showRegionWindow(self):
self.regionWindow.show()
self.regionWindow.activateWindow() #tabWidget.setCurrentWidget(self.additionalParameters)
def getData(self):
return self.p.activeSpectrum.xaxis(self.calibrationEnabled),self.y
def summationRaw(self):
sums = []
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled)
for a in self.regionWindow.regions:
start,end=a.region.getRegion()
start = self.closestIndex(x,start)
end = self.closestIndex(x,end)
sums.append([a,x[start],x[end],sum(self.y[start:end])])
return sums
def summation(self):
vals = self.summationRaw()
msg = ''
if vals == []: #No regions present / x is empty
msg = ' Data Unavailable. Please select a peak using the region utility. '
for a in vals:
region,start,end,count = a
msg += u'\u2211[%.2f:%.2f] : %d\n'%(start,end,count)
QtWidgets.QMessageBox.information(self, 'Summation (Region) ', msg)
def clearAllSums(self):
for a in self.regionWindow.regions:
a.clearData()
def saveAllSums(self):
for a in self.regionWindow.regions:
print(a)
def fit(self,**kwargs):
#self.calibWindow.show()
if not self.p.activeSpectrum.hasData():
QtWidgets.QMessageBox.information(self, 'Data Unavailable ', 'please acquire a spectrum')
return
elif len(self.regionWindow.regions)==0:
QtWidgets.QMessageBox.information(self, 'Regions Unavailable ', 'Please insert regions and center them around the peaks to be fitted.')
#self.regionLabel.color_anim.start()
return
tail = kwargs.get('fitTail',self.lowTailBox.isChecked())
fitres = []
if not len(self.regionWindow.regions):return
self.clearFits()
YDATA = self.p.activeSpectrum.getHistogram()
for R in self.regionWindow.regions:
try:
if not tail: # REGULAR GAUSSIAN
try:
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled)
res = fitting.gaussfit(x,YDATA,R.region.getRegion())
except:
res = None
if res is None:
continue
Xmore,Y,par,FIT = res
fitcurve = pg.PlotCurveItem(name = 'Fit',pen = [255,0,0]);self.plot.addItem(fitcurve)
self.fitcurves.append(fitcurve)
fitcurve.setData(Xmore,Y, stepMode=False,fillLevel=0, brush=(126, 197, 220,100)) #Curve
#QtWidgets.QMessageBox.critical(self, 'Fit Results', msg)
msg = 'Amplitude= %5.1f Centroid= %5.2f sigma = %5.2f'%(par[0], par[1], par[2])
if self.calibrationEnabled: FIT['channel'] = self.p.activeSpectrum.calPolyInv(par[1])
else: FIT['channel'] = par[1]
fitres.append(FIT)
else:
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled)
res = fitting.gausstailfit(x,YDATA,R.region.getRegion())
if res is None:
continue
Xmore,Y,par,FIT = res
fitcurve = pg.PlotCurveItem(name = 'Fit',pen = [255,0,0]);self.plot.addItem(fitcurve)
self.fitcurves.append(fitcurve)
fitcurve.setData(Xmore,Y, stepMode=False,fillLevel=0, brush=(126, 197, 220,100)) #Curve
msg = 'Amplitude= %5.1f Centroid= %5.2f S = %5.2f G = %5.2f'%(par[0], par[1], par[2], par[3])
if self.calibrationEnabled: FIT['channel'] = self.p.activeSpectrum.calPolyInv(par[1])
else: FIT['channel'] = par[1]
fitres.append(FIT)
except Exception as e:
QtWidgets.QMessageBox.critical(self, 'Fit Failed', str(e))
self.calibWindow.show()
self.calibWindow.setFitList(fitres)
def fitWithTail(self):
self.fit(fitTail=True)
def showCalibrationEditor(self):
self.calibWindow.show()
def toggleCalibration(self,state):
self.calibrationEnabled = state
self.currentPeak = None
self.clearFits()
self.markerarrow.setPos(0,0)
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled,trace=0) #first trace
self.plot.setLimits(xMax=max(x) );self.plot.setXRange(0,max(x) )
if state: #Calibration needs to be applied
self.plot.getAxis('bottom').setLabel('Energy (KeV)')
self.calibrationChanged.emit(self.p.activeSpectrum.calPoly,self.p.activeSpectrum.calPolyInv)
else:
'''
Calibration reset on the graph. but user defined calibration is not reset to original values.
'''
self.calibrationChanged.emit(np.poly1d([1,0]),np.poly1d([1,0])) #reset calibration
self.plot.getAxis('bottom').setLabel('Channel Number')
self.plot.setLimits(xMax=max(x)*1.05 ); self.plot.setXRange(0,max(x)*1.05 ); self.plot.setYRange(0,max(self.y)*1.05)
self.curve.setData(np.array(x),self.y[:-1], stepMode=True, fillLevel=0,pen=self.pen, brush=self.brush)
if self.p.activeSpectrum.datatype=='list':
if self.p.activeSpectrum.parameters==2:
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled,trace=1) #xaxis for second plot.
self.curve2.setData(np.array(x),self.y2[:-1], stepMode=True, fillLevel=0, brush=(0,0,255,150))
def addSelectedPoint(self):
if not self.currentPeak:
QtWidgets.QMessageBox.critical(self, 'Peak not selected', 'Please locate a peak first, by clicking on it<br>Then apply the calibration after typing its known energy')
return
msg=''
def setCalibration(self,**kwargs):
if not self.p.activeSpectrum.hasData():
return
###
points = self.calibWindow.getCalibrationPoints()
if len(points)>1:
self.p.setCalibration(points)
self.calibrateOnOff.setToolTip(_translate("MainWindow",'''
<html><head/><body>
<p><span style=\" font-size:14pt;color:#939\">Polynomial:%s</span></p>
<p><span style=\" font-size:12pt;\">Enable/Disable the calibration polynomial.</span></p>
<p><span style=\" font-size:12pt;\">Switch to the calibration tab for details</span></p>
</body></html>'''%(self.p.activeSpectrum.calPoly)))
###
self.calibrateOnOff.setChecked(True)
self.toggleCalibration(True) # refresh plots etc.
msg = kwargs.get('msg','Calibration Enabled')
self.showStatus(msg,True)
def onClick(self,event):
if not self.p.activeSpectrum.hasData():return
if not len(self.y): return
pos = event.scenePos()
if self.plot.sceneBoundingRect().contains(pos):
mousePoint = self.plot.plotItem.vb.mapSceneToView(pos)
index = mousePoint.x()
#XR = self.plot.plotItem.vb.viewRange()[0]
#XR = (XR[1]-XR[0])/30.
#if(len(self.regions)):
# self.regions[-1].region.setRegion([index-XR,index+XR])
if self.p.activeSpectrum.hasData:
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled)
P = (np.abs(x - index ) ).argmin()
self.markerarrow.setPos(x[P],self.y[P])
self.currentPeak = x[P]
self.showStatus("Selected Peak > X : %.2f(%d) , Y : %.2f"%(x[P],P,self.y[P]))
else:
self.markerarrow.setPos(0,0)
self.currentPeak = None
self.showStatus("Unselected Peak")
def closestIndex(self,arr,val):
return (np.abs(arr-val)).argmin()
def mouseMoved(self,event):
if not self.p.activeSpectrum.hasData():return
if not len(self.y): return
pos = event[0]
if self.plot.sceneBoundingRect().contains(pos):
mousePoint = self.plot.plotItem.vb.mapSceneToView(pos)
index = mousePoint.x()
x = self.p.activeSpectrum.xaxis(self.calibrationEnabled)
P = self.closestIndex(x,index)
self.arrow.setPos(x[P],self.y[P])
if self.vLine:
self.vLine.setPos(mousePoint.x())
self.chanLabel.setText('[x=%0.1f, %d]' % (mousePoint.x(), self.y[P]))
#self.chanLabel.setPos(mousePoint.x(),self.y[P]+1)
def save(self):
if self.p.activeSpectrum.datatype=='list':
from utilities import plotSaveWindow
comments = 'Histogram (X, Y)\n'
info = plotSaveWindow.AppWindow(self,[[self.p.activeSpectrum.xaxis(self.calibrationEnabled),self.p.activeSpectrum.getHistogram()]],self.plot,comments = comments)
else:
if not self.p.activeSpectrum.hasData():
QtWidgets.QMessageBox.critical(self, 'Acquire Data', 'Please acquire some data first!')
return
from utilities import plotSaveWindow
#info = plotSaveWindow.AppWindow(self,[self.curve,self.fitcurve],self.plot)
comments = ''
if len(self.fitcurves):
for a in self.calibWindow.fitList:
if a['centroid'] == a['channel']:
centroidString = '%.2f'%(a['centroid'])
else:
centroidString = '%.2f keV (channel:%.2f)'%(a['centroid'],a['channel'])
comments+="Region: %s\nAmplitude: %.2f\nCentroid: %s\nFWHM: %.3f(%.2f%%)\nArea: %.2f\n-----------"%(a['region'],a['amplitude'],centroidString,a['fwhm'],100*a['fwhm']/a['centroid'],a['area'])
#info = plotSaveWindow.AppWindow(self,[[self.x,self.y]]+[a.getData() for a in self.fitcurves],self.plot,comments = comments)
info = plotSaveWindow.AppWindow(self,[[self.p.activeSpectrum.xaxis(self.calibrationEnabled),self.y]],self.plot,comments = comments)
info.show()
def loadPlot(self,fname):
self.offlineData = True
self.showStatus("Loaded File | %s (Click UPDATE to return)"%fname,True)
self.loadFromFile( self.plot,[self.curve],fname ,True)
self.tabWidget.setCurrentIndex(0)
self.calibrateOnOff.setChecked(False)
def loadFromFile(self,plot,curves,filename,histMode=True):
self.p.loadFile(filename,self.removeCalBox.isChecked())
#self.y = self.p.activeSpectrum.getHistogram(self.logBox.isChecked()) # No need. The togglelog function will set it.
self.toggleLog(self.logBox.isChecked())
self.clearFits()
head, tail = os.path.split(filename)
self.updateTraceList(['%s'%tail])
#self.recordToHistory(time = self.spectrumTime)
def loadList(self,fname):
self.offlineData = True
self.showStatus("Loaded File | %s (Click UPDATE to return)"%fname,True)
self.p.loadListFile(fname,self.removeCalBox.isChecked())
traces = []
head, tail = os.path.split(self.p.activeSpectrum.filename)
for b in range(self.p.activeSpectrum.parameters):
traces.append('%s:%d'%(tail,b+1))
self.updateTraceList(traces)
self.clearPlot()
self.toggleLog(self.logBox.isChecked())
self.tabWidget.setCurrentIndex(0)
self.calibrateOnOff.setChecked(False)
from utilities import plot3DTools
if self.surfacePlot:
self.surfacePlot.close()
del self.surfacePlot
self.surfacePlot = plot3DTools.surface3d(self,self.p.activeSpectrum.HISTOGRAM2D,self.p.activeSpectrum.BINS2D)
A = np.sum(self.p.activeSpectrum.spectra[0].data)
B = np.sum(self.p.activeSpectrum.spectra[1].data)
C = np.sum(self.p.activeSpectrum.HISTOGRAM2D)
self.surfacePlot.countA.display(A)
self.surfacePlot.countB.display(B)
self.surfacePlot.countC.display(C)
self.surfacePlot.labelA.setText('/%d [%.2f%%]'%(A+B,100*A/(A+B)))
self.surfacePlot.labelB.setText('/%d [%.2f%%]'%(A+B,100*B/(A+B)))
self.coincidenceLabel.setText('C: %d [%d , %d]'%(C,A,B) )
self.listFrame.show()
self.surfacePlot.show()
def setListSaveFilename(self):
path, _filter = QtWidgets.QFileDialog.getSaveFileName(self, 'Specify filename to dump list data', '~/')
if path:
if self.p.activeSpectrum.datatype == 'list':
self.p.activeSpectrum.setOutputFilename(path)
self.showStatus("List data being saved to:%s"%(path))
def autoScale(self):
xMin,xMax = self.p.activeSpectrum.getCalibratedRange()
self.plot.setLimits(xMin = xMin, xMax=xMax,yMax = max(20,max(self.y)*1.1))
self.plot.setXRange(xMin, xMax);self.plot.setYRange(0,max(20,max(self.y)*1.1))
############ DATA DUMPING #############
def changeDirectory(self):
dirname = QtWidgets.QFileDialog.getExistingDirectory(self,"Select a folder in which spectra will be periodically saved", os.path.expanduser("./"), QtWidgets.QFileDialog.ShowDirsOnly)
if not dirname:return
self.dataDumpPath=str(dirname)
self.pathLabel.setText(self.dataDumpPath)
def dataDumpTime(self):
self.pending['datadump'] = myTimer(self.regionWindow.saveAllInterval.value()*60) #in minutes
self.showStatus('Spectrum save interval changed to %s Minutes'%(self.regionWindow.saveAllInterval.value()) )
def dataDump(self):
dt = time.time() - self.startTime
m, s = divmod(dt, 60)
h, m = divmod(m, 60)
ST = "%d:%02d:%02d" % (h, m, s)
np.savetxt(os.path.join(self.dataDumpPath,'data_%.1fmins'%dt),np.column_stack([self.p.activeSpectrum.xaxis(self.calibrationEnabled),self.y]))
self.regionWindow.savedSpectraCounter.setValue(self.regionWindow.savedSpectraCounter.value()+1)
def enablePeriodicSpectrumSaving(self):
if self.dataDumpPath is None:
self.changeDirectory()
if self.dataDumpPath is None:
self.regionWindow.saveAllCheckbox.setChecked(False)
######## WINDOW EXPORT SVG
def exportSvg(self):
from utilities.Qt import QtSvg
path, _filter = QtWidgets.QFileDialog.getSaveFileName(self, 'Save File', '~/')
if path:
generator = QtSvg.QSvgGenerator()
generator.setFileName(path)
target_rect = QtCore.QRectF(0, 0, 800, 600)
generator.setSize(target_rect.size().toSize())#self.size())
generator.setViewBox(self.rect())
generator.setTitle("Your title")
generator.setDescription("some description")
p = QtGui.QPainter()
p.begin(generator)
self.render(p)
p.end()
def __init__(self, parent=None, **kwargs):
super(AppWindow, self).__init__(parent)
self.setupUi(self)
self.VERSION = REGISTERS.VERSION_ATMEGA32 #This needs to be dynamically changed when hardware is connected
self.SPECIALS = REGISTERS.VERSIONS[self.VERSION]['SPECIALS']
self.REGISTERS = REGISTERS.VERSIONS[self.VERSION]['REGISTERS']
self.EXAMPLES_DIR = REGISTERS.VERSIONS[
self.VERSION]['examples directory']
self.docks = [self.padock, self.pbdock, self.pcdock, self.pddock]
self.sensorList = []
self.controllerList = []
self.monitoring = True
self.logRegisters = True
self.userHexRunning = False
self.uploadingHex = False
self.autoUpdateUserRegisters = False
self.CFile = None #'~/kuttyPy.c'
self.ipy = None
self.setTheme("material")
examples = [
a for a in os.listdir(
os.path.join(path["examples"], self.EXAMPLES_DIR))
if ('.py' in a) and a is not 'kuttyPy.py'
] #.py files except the library
self.exampleList.addItems(examples)
blinkindex = self.exampleList.findText('blink.py')
if blinkindex != -1: #default example. blink.py present in examples directory
self.exampleList.setCurrentIndex(blinkindex)
#Define some keyboard shortcuts for ease of use
self.shortcutActions = {}
self.shortcuts = {
"f": partial(self.setLanguage, 'fr_FR'),
"e": partial(self.setLanguage, 'en_IN'),
"m": partial(self.setLanguage, 'ml_IN')
}
for a in self.shortcuts:
shortcut = QtWidgets.QShortcut(QtGui.QKeySequence(a), self)
shortcut.activated.connect(self.shortcuts[a])
self.shortcutActions[a] = shortcut
######## PYTHON CODE
self.codeThread = QtCore.QThread()
self.codeEval = self.codeObject(self.REGISTERS)
self.codeEval.moveToThread(self.codeThread)
self.codeEval.finished.connect(self.codeThread.quit)
self.codeEval.logThis.connect(
self.appendLog) #Connect to the log window
self.logThis.connect(self.appendLog) #Connect to the log window
self.logThisPlain.connect(
self.appendLogPlain) #Connect to the log window
self.serialGaugeSignal.connect(self.setSerialgauge)
self.codeThread.started.connect(self.codeEval.execute)
self.codeThread.finished.connect(self.codeFinished)
######### C CODE UPLOADER
self.uploadThread = QtCore.QThread()
self.UploadObject = self.uploadObject()
self.UploadObject.moveToThread(self.uploadThread)
self.UploadObject.finished.connect(self.uploadThread.quit)
self.UploadObject.logThis.connect(
self.appendLog) #Connect to the log window
self.UploadObject.logThisPlain.connect(
self.appendLogPlain) #Connect to the log window. add plain text
self.logThis.connect(self.appendLog) #Connect to the log window
self.uploadThread.started.connect(self.UploadObject.execute)
self.uploadThread.finished.connect(self.codeFinished)
self.commandQ = []
self.btns = {}
self.registers = []
self.addPins()
self.statusBar = self.statusBar()
self.makeBottomMenu()
global app
self.initializeCommunications()
self.pending = {
'status': myTimer(constants.STATUS_UPDATE_INTERVAL),
'update': myTimer(constants.AUTOUPDATE_INTERVAL),
}
serialgaugeoptions = {
'name':
'Serial Monitor',
'init':
print,
'read':
None,
'fields': ['Value'],
'min': [0],
'max': [1000],
'config': [{
'name': 'Data Type',
'options': ['byte', 'ASCII'],
'function': self.configSerialGauge
}]
}
self.serialGauge = dio.DIOSENSOR(self, serialgaugeoptions)
self.startTime = time.time()
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateEverything)
self.timer.start(20)
#Auto-Detector
self.shortlist = KuttyPyLib.getFreePorts()