''' ''' #!/usr/bin/python3 import os,sys,time,re,traceback,platform sys.path.append(os.path.dirname(os.path.abspath(__file__))) from PyQt5 import QtGui, QtCore, QtWidgets import KuttyPyLib from utilities.templates import ui_layout as layout from utilities import dio,REGISTERS,uploader,syntax import constants import inspect from functools import partial from collections import OrderedDict # translation stuff 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) class myTimer(): def __init__(self,interval): self.interval = interval self.reset() def reset(self): self.timeout = time.time()+self.interval def ready(self): T = time.time() dt = T - self.timeout if dt>0: #timeout is ahead of current time #if self.interval>5:print('reset',self.timeout,dt) self.timeout = T - dt%self.interval + self.interval #if self.interval>5:print(self.timeout) return True return False def progress(self): return 100*(self.interval - self.timeout + time.time())/(self.interval) class AppWindow(QtWidgets.QMainWindow, layout.Ui_MainWindow): p=None ports = ['A','B','C','D'] logThis = QtCore.pyqtSignal(str) logThisPlain = QtCore.pyqtSignal(bytes) serialGaugeSignal = QtCore.pyqtSignal(bytes) serialGaugeConvert = 'bytes' serialStream = b'' 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.defaultDirectory = path["examples"] self.setTheme("material") examples = [a for a in os.listdir(os.path.join(path["examples"],self.EXAMPLES_DIR)) if ('.py' in a) and a != '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) self.serialGuageButton.hide() #Hide the serial guage button #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) ########## C CODE EDITOR SYNTAX HIGHLIGHTER self.highlight = syntax.PythonHighlighter(self.editor.document()) 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() 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): for port,dock in zip(self.ports,[self.palayout,self.pblayout,self.pclayout,self.pdlayout]): checkbox = dio.REGVALS(port) dock.addWidget(checkbox) self.btns[port] = checkbox seq = range(7,-1,-1) if port == 'C':seq = reversed(seq) #PORTC pins are ordered top to bottom for a in seq: name = 'P'+port+str(a) checkbox = dio.widget(name,self.commandQ,extra = self.SPECIALS.get(name,'')) dock.addWidget(checkbox) self.btns[name] = checkbox def tabChanged(self,index): if self.userHexRunning: #Firmware was running. Stop it. self.launchFirmwareButton.setChecked(False) self.jumpToApplication(False) 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) self.setLogType('playground') for a in self.docks: a.show() def setLogType(self,tp): self.log.clear() if tp == 'playground': self.logLabel.setText("Monitor registers being read and set during each operation") elif tp == 'avr': self.logLabel.setText("Avr-Gcc compile/upload messages") elif tp == 'monitor': self.logLabel.setText("Monitor data input from the serial port") elif tp == 'python': self.logLabel.setText("Python Code operations monitor") ################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(), '', mode='exec') except SyntaxError as err: error_class = err.__class__.__name__ detail = err.args[0] line_number = err.lineno return '''%s at line %d : %s''' % (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 '''%s at line %d: %s''' % (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('''%s'''%(' '.join([str(a) for a in args]))) def setReg(self,reg,value): html=u'''
W\u2193{0:s}\t{1:d}\t0x{1:02x} / 0b{1:08b}
'''.format(self.PORTMAP.get(reg,''),value) self.logThis.emit(html) self.SR(reg,value) def getReg(self,reg): value = self.GR(reg) html=u'''
R\u2191{0:s}\t{1:d}\t0x{1:02x} / 0b{1:08b}
'''.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('''%s at line %d : %s''' % (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('''%s at line %d: %s''' % (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.codeThread.isRunning(): print('one code is already running') return self.setLogType('python') self.log.clear() #clear the log window self.log.setText('''----------User Code Started-----------''') 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('''----------Kill Signal(Doesn't work yet. Restart the application)-----------''') 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 configSerialGauge(self,val): if val == 0: #'byte' print('Byte Mode(0-255)') for a in self.serialGauge.gauges: a.set_MaxValue(255) self.serialGaugeConvert = 'bytes' if val == 1: #'ascii' print('ASCII(0-10000)') for a in self.serialGauge.gauges: a.set_MaxValue(10000) self.serialGaugeConvert = 'ascii' def setSerialgauge(self,vals): if self.serialGaugeConvert == 'bytes': self.serialGauge.setValue([vals[0]]) elif self.serialGaugeConvert == 'ascii': self.serialStream += vals while b'\n' in self.serialStream: val,_,self.serialStream = self.serialStream.partition(b'\n') self.serialGauge.setValue([float(val)]) def genLog(self): html=''''''%(time.ctime()) #html+=''' #''' for a in self.updatedRegs: row = self.updatedRegs[a] html+=u''' '''.format(u'W \u2193' if row[0] else u'R \u2191',a,row[1]) html+="
%s
R/WREGISTERValueHex/Binary
{0:s} {1:s} {2:d} 0b{2:08b} | 0x{2:02x}
" 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) 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<>a)&1) if portchar == 'A': #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 'compile' in self.mode: try: import subprocess if self.fname[-2:] in ['.s','.S']: action = 'Assembl' else:#elif self.fname[-2:] in ['.c','.C']: action = 'Compil' fname = '.'.join(self.fname.split('.')[:-1]) if self.p.version == REGISTERS.VERSION_ATMEGA32 or self.p.connected==False: #by default, compile for ATMEGA32. cmd = 'avr-gcc -Wall -O2 -mmcu=%s -o "%s" "%s"' %('atmega32',fname,self.fname) self.logThis.emit('''%sing for Atmega32'''%(action)) elif self.p.version == REGISTERS.VERSION_ATMEGA328P: cmd = 'avr-gcc -Wall -O2 -mmcu=%s -o "%s" "%s"' %('atmega328p',fname,self.fname) self.logThis.emit('''%sing for Atmega328p (Nano)'''%(action)) else: self.logThis.emit('''%ser UNAVAILABLE'''%(action)) return print(cmd) res = subprocess.getstatusoutput(cmd) if res[0] != 0: self.logThis.emit('''%se Error: %s'''%(action,res[1].replace('\n','
')) ) self.finished.emit() return else: self.logThis.emit('''%s
'''%res[1]) cmd = 'avr-objcopy -j .text -j .data -O ihex "%s" "%s.hex"' %(fname,fname) res = subprocess.getstatusoutput(cmd) self.logThis.emit(res[1])#'''%s
'''%res[1]) cmd = 'avr-objdump -S "%s" > "%s.lst"'%(fname,fname) res = subprocess.getstatusoutput(cmd) self.logThis.emit(res[1])#'''%s
'''%res[1]) self.logThis.emit('Finished %sing: Generated Hex File'%action) #'''Finished %sing: Generated Hex File'''%(action)) except Exception as err: self.logThis.emit('''Failed to %se:%s'''%str(action,err)) if self.p.connected: if 'upload' in self.mode: try: if self.fname[-2:] in ['.c','.C','.s','.S']: self.fname = self.fname[:-2]+'.hex' #Replace .c with .hex self.logThis.emit('''Upload Code... Trigger Reset...''') self.p.fd.setRTS(0);time.sleep(0.01);self.p.fd.setRTS(1);time.sleep(0.1) self.p.get_version() 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('''Finished upload''') except Exception as err: print('upload error',err) 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('''Failed to upload''') #self.jumpToApplication(False) #Force a reset 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('''-- Uploading Code --
''') self.UploadObject.config('upload',self.p,filename[0]) self.uploadThread.start() def openFile(self): filename = QtWidgets.QFileDialog.getOpenFileName(self," Open a C/Asm file to edit", self.defaultDirectory, "C/ASM Files (*.c *.C *.s *.S);; Asm Files (*.s *.S);; C Files (*.c *.C)") if len(filename[0]): self.defaultDirectory = '' self.filenameLabel.setText(filename[0]) self.CFile = filename[0] self.log.clear() infile = open(filename[0], 'r') self.editor.setPlainText(infile.read()) self.log.setText('''-- Opened File: %s --
'''%filename[0]) def compileAndUpload(self): self.setLogType('avr') self.saveFile() if self.CFile: self.uploadingHex = True self.log.setText('''-- Compiling and Uploading Code --
''') self.UploadObject.config('compileupload',self.p,self.CFile) self.uploadThread.start() def compile(self): self.setLogType('avr') self.saveFile() if self.CFile: self.log.setText('''-- Compiling Code --
''') self.UploadObject.config('compile',self.p,self.CFile) self.uploadThread.start() def upload(self): self.setLogType('avr') if self.CFile: self.uploadingHex = True self.log.setText('''-- Uploading Code --
''') self.UploadObject.config('upload',self.p,self.CFile) self.uploadThread.start() def saveFile(self): if not self.CFile: name = QtGui.QFileDialog.getSaveFileName(self, 'Save File') print('created new file:',name) self.CFile = name[0] self.log.setText('''-- Created new file: %s --
'''%self.CFile) fn = open(self.CFile,'w') fn.write(self.editor.toPlainText()) fn.close() self.log.setText('''-- Saved to: %s --
'''%self.CFile) ############################## 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 = KuttyPyLib.connect(port = port) else: self.p = KuttyPyLib.connect(autoscan=True) if self.p.connected: self.launchFirmwareButton.setChecked(False) self.setWindowTitle('KuttyPy Interactive Console [{0:s}]'.format(self.p.portname)) self.updatedRegs=OrderedDict() self.currentRegister = 0 self.VERSION = self.p.version self.SPECIALS = REGISTERS.VERSIONS[self.VERSION]['SPECIALS'] self.REGISTERS = REGISTERS.VERSIONS[self.VERSION]['REGISTERS'] self.EXAMPLES_DIR = REGISTERS.VERSIONS[self.VERSION]['examples directory'] if self.p.version == REGISTERS.VERSION_ATMEGA328P: self.getRegs=[ ('PINB',self.updateInputs), ('PINC',self.updateInputs), ('PIND',self.updateInputs), ] elif self.p.version == REGISTERS.VERSION_ATMEGA32: self.getRegs=[ ('PINA',self.updateInputs), ('PINB',self.updateInputs), ('PINC',self.updateInputs), ('PIND',self.updateInputs), ] else: self.setWindowTitle('KuttyPy Interactive Console [ Hardware not detected ]') def setLanguage(self,lang = 'fr_FR'): translate(lang) self.retranslateUi(self) def showSerialGauge(self): self.serialGauge.show() def jumpToApplication(self,state): if self.p: if state: self.serialGuageButton.show() 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.setLogType('monitor') self.log.setText('''-- Serial Port Monitor --
''') #self.serialGauge.show() else: self.serialGuageButton.hide() 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: #7c7;") 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) ''' 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.statusBar.addPermanentWidget(self.themeBox) 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 = KuttyPyLib.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.close() 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 PyQt5 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) 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 firstExistingPath(l): """ Returns the first existing path taken from a list of possible paths. @param l a list of paths @return the first path which exists in the filesystem, or None """ for p in l: if os.path.exists(p): return p return None def common_paths(): """ Finds common paths @result a dictionary of common paths """ path={} curPath = os.path.dirname(os.path.realpath(__file__)) path["current"] = curPath sharedPath = "/usr/share/kuttypy" path["translation"] = firstExistingPath( [os.path.join(p, "lang") for p in (curPath, sharedPath,)]) path["utilities"] = firstExistingPath( [os.path.join(p,'utilities') for p in (curPath, sharedPath,)]) path["templates"] = firstExistingPath( [os.path.join(p,'utilities','templates') for p in (curPath, sharedPath,)]) path["themes"] = firstExistingPath( [os.path.join(p,'utilities','themes') for p in (curPath, sharedPath,)]) path["examples"] = firstExistingPath( [os.path.join(p,'examples') for p in (curPath, sharedPath,)]) path["editor"] = firstExistingPath( [os.path.join(p,'editor') for p in (curPath, sharedPath,)]) lang=str(QtCore.QLocale.system().name()) shortLang=lang[:2] return path def run(): global path, app, myapp path = common_paths() print('QT Version',QtWidgets.__file__) app = QtWidgets.QApplication(sys.argv) myapp = AppWindow(app=app, path=path) myapp.show() r = app.exec_() ''' if myapp.p.connected: myapp.p.fd.write(b'j') #myapp.p.fd.setRTS(0) #time.sleep(0.01) #myapp.p.fd.setRTS(1) ''' app.deleteLater() sys.exit(r) if __name__ == "__main__": run()