class AVRProgrammerDialog(QtFixes.QDialog):
def __init__(self):
super(AVRProgrammerDialog, self).__init__(CWMainGUI.getInstance())
# self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
self.avr = AVRProgrammer()
self.setWindowTitle("ChipWhisperer-Lite AVR Programmer")
layout = QVBoxLayout()
layoutFW = QHBoxLayout()
self.flashLocation = QtFixes.QLineEdit()
flashFileButton = QPushButton("Find")
flashFileButton.clicked.connect(self.findFlash)
layoutFW.addWidget(QLabel("FLASH File"))
layoutFW.addWidget(self.flashLocation)
layoutFW.addWidget(flashFileButton)
layout.addLayout(layoutFW)
self.flashLocation.setText(QSettings().value("avr-flash-location"))
# Add buttons
readSigBut = QPushButton("Check Signature")
readSigBut.clicked.connect(self.readSignature)
verifyFlashBut = QPushButton("Verify FLASH")
verifyFlashBut.clicked.connect(self.verifyFlash)
verifyFlashBut.setEnabled(False)
progFlashBut = QPushButton("Erase/Program/Verify FLASH")
progFlashBut.clicked.connect(self.writeFlash)
layoutBut = QHBoxLayout()
layoutBut.addWidget(readSigBut)
layoutBut.addWidget(verifyFlashBut)
layoutBut.addWidget(progFlashBut)
layout.addLayout(layoutBut)
layoutFuse = QHBoxLayout()
readFuseBut = QPushButton("Read Fuses")
readFuseBut.clicked.connect(self.readFuses)
writeFuseBut = QPushButton("Write Fuses")
writeFuseBut.clicked.connect(self.writeFuses)
self.lowfuseLine = QtFixes.QLineEdit("?")
self.lowfuseLine.setMaxLength(2)
self.lowfuseLine.setFixedWidth(25)
self.highfuseLine = QtFixes.QLineEdit("?")
self.highfuseLine.setMaxLength(2)
self.highfuseLine.setFixedWidth(25)
self.extfuseLine = QtFixes.QLineEdit("?")
self.extfuseLine.setMaxLength(2)
self.extfuseLine.setFixedWidth(25)
# Don't allow use to change these fuses
self.highfuseLine.setReadOnly(True)
self.extfuseLine.setReadOnly(True)
layoutFuse.addWidget(readFuseBut)
layoutFuse.addWidget(QLabel("LOW:"))
layoutFuse.addWidget(self.lowfuseLine)
layoutFuse.addWidget(QLabel("HIGH:"))
layoutFuse.addWidget(self.highfuseLine)
layoutFuse.addWidget(QLabel("EXT:"))
layoutFuse.addWidget(self.extfuseLine)
layoutFuse.addWidget(writeFuseBut)
layout.addLayout(layoutFuse)
layoutExtra = QHBoxLayout()
self.clockMode = QPushButton("Enable Slow Clock Mode")
self.clockMode.setCheckable(True)
self.clockMode.clicked.connect(self.toggleSlowClock)
layoutExtra.addWidget(self.clockMode)
layoutExtra.addStretch()
layout.addLayout(layoutExtra)
# Add status stuff
self.statusLine = QPlainTextEdit()
self.statusLine.setReadOnly(True)
# self.statusLine.setFixedHeight(QFontMetrics(self.statusLine.font()).lineSpacing() * 5 + 10)
layout.addWidget(self.statusLine)
self.avr.newTextLog.connect(self.append)
# Set dialog layout
self.setLayout(layout)
def append(self, text):
self.statusLine.appendPlainText(text)
def toggleSlowClock(self):
if self.clockMode.isChecked():
self.avr.enableSlowClock(True)
self.clockMode.setText("Disable Slow Clock Mode")
else:
self.avr.enableSlowClock(False)
self.clockMode.setText("Enable Slow Clock Mode")
def reject(self):
#Ensure we disable slow-clock mode
if self.clockMode.isChecked():
self.clockMode.setChecked(False)
self.toggleSlowClock()
#Release AVR
self.avr.close()
QDialog.reject(self)
def findFlash(self):
fname, _ = QFileDialog.getOpenFileName(
self, 'Find FLASH File',
QSettings().value("avr-flash-location"), '*.hex')
if fname:
self.flashLocation.setText(fname)
QSettings().setValue("avr-flash-location", fname)
def readSignature(self, close=True):
self.avr.find()
if close:
self.avr.close()
def readFuses(self):
try:
self.readSignature(close=False)
self.statusLine.appendPlainText("Reading fuses")
lfuse = self.avr.readFuse("low")
hfuse = self.avr.readFuse("high")
efuse = self.avr.readFuse("extended")
self.statusLine.appendPlainText("OK: %02x %02x %02x" %
(lfuse, hfuse, efuse))
self.lowfuseLine.setText("%02x" % lfuse)
self.highfuseLine.setText("%02x" % hfuse)
self.extfuseLine.setText("%02x" % efuse)
except IOError:
self.statusLine.appendPlainText("Reading fuses failed")
self.lowfuseLine.setText("?")
self.highfuseLine.setText("?")
self.extfuseLine.setText("?")
self.avr.close()
def writeFuses(self):
lfuse = int(self.lowfuseLine.text(), 16)
# hfuse = int(self.highfuseLine.text(), 16)
# efuse = int(self.extfuseLine.text(), 16)
self.statusLine.appendPlainText(
"Writing fuse: not only 'low' fuse is written, as hfuse/efuse can disable device\n"
)
try:
self.readSignature(close=False)
self.avr.writeFuse("low", lfuse)
# self.avr.writeFuse("high", hfuse)
# self.avr.writeFuse("extended", efuse)
# print("%x %x %x" % (lfuse, hfuse, efuse))
self.avr.close()
except:
self.avr.close()
raise
def verifyFlash(self):
self.statusLine.appendPlainText("Verify not implemented")
def writeFlash(self, erase=True, verify=True):
self.avr.autoProgram(self.flashLocation.text(), erase, verify,
self.statusLine.appendPlainText,
QCoreApplication.processEvents)
def setUSBInterface(self, iface):
self.avr.setUSBInterface(iface)
class XMEGAProgrammerDialog(QtFixes.QDialog):
def __init__(self):
super(XMEGAProgrammerDialog, self).__init__(CWMainGUI.getInstance())
# self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
self.xmega = XMEGAProgrammer()
self.setWindowTitle("ChipWhisperer-Lite XMEGA Programmer")
layout = QVBoxLayout()
layoutFW = QHBoxLayout()
self.flashLocation = QtFixes.QLineEdit()
flashFileButton = QPushButton("Find")
flashFileButton.clicked.connect(self.findFlash)
layoutFW.addWidget(QLabel("FLASH File"))
layoutFW.addWidget(self.flashLocation)
layoutFW.addWidget(flashFileButton)
layout.addLayout(layoutFW)
self.flashLocation.setText(QSettings().value("xmega-flash-location"))
# Add buttons
readSigBut = QPushButton("Check Signature")
readSigBut.clicked.connect(self.readSignature)
verifyFlashBut = QPushButton("Verify FLASH")
verifyFlashBut.clicked.connect(self.verifyFlash)
verifyFlashBut.setEnabled(False)
progFlashBut = QPushButton("Erase/Program/Verify FLASH")
progFlashBut.clicked.connect(self.writeFlash)
layoutBut = QHBoxLayout()
layoutBut.addWidget(readSigBut)
layoutBut.addWidget(verifyFlashBut)
layoutBut.addWidget(progFlashBut)
layout.addLayout(layoutBut)
# Add status stuff
self.statusLine = QPlainTextEdit()
self.statusLine.setReadOnly(True)
# self.statusLine.setFixedHeight(QFontMetrics(self.statusLine.font()).lineSpacing() * 5 + 10)
self.xmega.newTextLog.connect(self.append)
layout.addWidget(self.statusLine)
# Set dialog layout
self.setLayout(layout)
def append(self, text):
self.statusLine.appendPlainText(text)
def findFlash(self):
fname, _ = QFileDialog.getOpenFileName(
self, 'Find FLASH File',
QSettings().value("xmega-flash-location"), '*.hex')
if fname:
self.flashLocation.setText(fname)
QSettings().setValue("xmega-flash-location", fname)
def readSignature(self, close=True):
self.xmega.find()
if close:
self.xmega.close()
def verifyFlash(self):
pass
def writeFlash(self, erase=True, verify=True):
self.xmega.autoProgram(self.flashLocation.text(), erase, verify,
self.statusLine.appendPlainText,
QCoreApplication.processEvents)
def setUSBInterface(self, iface):
self.xmega.setUSBInterface(iface)
class STM32FProgrammerDialog(QtFixes.QDialog):
def __init__(self):
super(STM32FProgrammerDialog, self).__init__(CWMainGUI.getInstance())
# self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
self.stm32f = STM32FProgrammer()
self.setWindowTitle("Serial STM32F Programmer")
layout = QVBoxLayout()
layoutFW = QHBoxLayout()
self.flashLocation = QtFixes.QLineEdit()
flashFileButton = QPushButton("Find")
flashFileButton.clicked.connect(self.findFlash)
layoutFW.addWidget(QLabel("FLASH File"))
layoutFW.addWidget(self.flashLocation)
layoutFW.addWidget(flashFileButton)
layout.addLayout(layoutFW)
self.flashLocation.setText(QSettings().value("stm32f-flash-location"))
# Add buttons
readSigBut = QPushButton("Check Signature")
readSigBut.clicked.connect(self.readSignature)
verifyFlashBut = QPushButton("Verify FLASH")
verifyFlashBut.clicked.connect(self.verifyFlash)
verifyFlashBut.setEnabled(False)
progFlashBut = QPushButton("Erase/Program/Verify FLASH")
progFlashBut.clicked.connect(self.writeFlash)
layoutBut = QHBoxLayout()
layoutBut.addWidget(readSigBut)
layoutBut.addWidget(verifyFlashBut)
layoutBut.addWidget(progFlashBut)
layout.addLayout(layoutBut)
layoutsetting = QHBoxLayout()
mode = QComboBox()
mode.addItem("fast", False)
mode.addItem("slow", True)
mode.currentIndexChanged.connect(self.modechanged)
layoutsetting.addWidget(QLabel("Speed:"))
layoutsetting.addWidget(mode)
blocksize = QComboBox()
blocksize.addItem("256", False)
blocksize.addItem("64", True)
blocksize.currentIndexChanged.connect(self.blocksizechanged)
layoutsetting.addStretch()
layoutsetting.addWidget(QLabel("Read Block Size:"))
layoutsetting.addWidget(blocksize)
layout.addLayout(layoutsetting)
# Add status stuff
self.statusLine = QPlainTextEdit()
self.statusLine.setReadOnly(True)
# self.statusLine.setFixedHeight(QFontMetrics(self.statusLine.font()).lineSpacing() * 5 + 10)
self.stm32f.newTextLog.connect(self.append)
layout.addWidget(self.statusLine)
# Set dialog layout
self.setLayout(layout)
def modechanged(self, newmode):
self.stm32f.slow_speed = newmode
def blocksizechanged(self, use_small_block):
self.stm32f.small_blocks = use_small_block
def append(self, text):
self.statusLine.appendPlainText(text)
def findFlash(self):
fname, _ = QFileDialog.getOpenFileName(
self, 'Find FLASH File',
QSettings().value("stm32f-flash-location"), '*.hex')
if fname:
self.flashLocation.setText(fname)
QSettings().setValue("stm32f-flash-location", fname)
def readSignature(self, close=True):
self.stm32f.open_and_find(log_func=self.statusLine.appendPlainText)
if close:
self.stm32f.close()
def verifyFlash(self):
pass
def writeFlash(self, erase=True, verify=True):
self.stm32f.autoProgram(self.flashLocation.text(), erase, verify,
self.statusLine.appendPlainText,
QCoreApplication.processEvents)
def setUSBInterface(self, iface):
self.stm32f.setUSBInterface(iface)
class Truss(QMainWindow):
def __init__(self, parent=None):
super(Truss, self).__init__(parent)
self.resize(800, 600)
self.filename = None
self.filetuple = None
self.dirty = False # Refers to Data Page only.
centralwidget = QWidget(self)
gridLayout = QGridLayout(centralwidget)
self.tabWidget = QTabWidget(centralwidget)
self.tab = QWidget()
font = QFont()
font.setFamily("Courier 10 Pitch")
font.setPointSize(12)
self.tab.setFont(font)
gridLayout_3 = QGridLayout(self.tab)
self.plainTextEdit = QPlainTextEdit(self.tab)
gridLayout_3.addWidget(self.plainTextEdit, 0, 0, 1, 1)
self.tabWidget.addTab(self.tab, "")
self.tab_2 = QWidget()
self.tab_2.setFont(font)
gridLayout_2 = QGridLayout(self.tab_2)
self.plainTextEdit_2 = QPlainTextEdit(self.tab_2)
gridLayout_2.addWidget(self.plainTextEdit_2, 0, 0, 1, 1)
self.tabWidget.addTab(self.tab_2, "")
gridLayout.addWidget(self.tabWidget, 0, 0, 1, 1)
self.setCentralWidget(centralwidget)
menubar = QMenuBar(self)
menubar.setGeometry(QRect(0, 0, 800, 29))
menu_File = QMenu(menubar)
self.menu_Solve = QMenu(menubar)
self.menu_Help = QMenu(menubar)
self.setMenuBar(menubar)
self.statusbar = QStatusBar(self)
self.setStatusBar(self.statusbar)
self.action_New = QAction(self)
self.actionSave_As = QAction(self)
self.action_Save = QAction(self)
self.action_Open = QAction(self)
self.action_Quit = QAction(self)
self.action_About = QAction(self)
self.actionShow_CCPL = QAction(self)
self.action_Solve = QAction(self)
self.action_CCPL = QAction(self)
self.action_Help = QAction(self)
menu_File.addAction(self.action_New)
menu_File.addAction(self.action_Open)
menu_File.addAction(self.actionSave_As)
menu_File.addAction(self.action_Save)
menu_File.addSeparator()
menu_File.addAction(self.action_Quit)
self.menu_Solve.addAction(self.action_Solve)
self.menu_Help.addAction(self.action_About)
self.menu_Help.addAction(self.action_CCPL)
self.menu_Help.addAction(self.action_Help)
menubar.addAction(menu_File.menuAction())
menubar.addAction(self.menu_Solve.menuAction())
menubar.addAction(self.menu_Help.menuAction())
self.setWindowTitle("Main Window")
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab),\
"Data Page")
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_2),\
"Solution Page")
menu_File.setTitle("&File")
self.menu_Solve.setTitle("&Solve")
self.menu_Help.setTitle("&Help")
self.tabWidget.setCurrentIndex(0)
self.action_New.setText("&New")
self.action_Open.setText("&Open")
self.actionSave_As.setText("Save &As")
self.action_Save.setText("&Save")
self.action_Quit.setText("&Quit")
self.action_Solve.setText("&Solve")
self.action_About.setText("&About")
self.action_CCPL.setText("&CCPL")
self.action_Help.setText("&Help")
self.action_Quit.triggered.connect(self.close)
allToolBar = self.addToolBar("AllToolBar")
allToolBar.setObjectName("AllToolBar")
self.addActions(allToolBar, (self.action_Open, self.actionSave_As,\
self.action_Save, self.action_Solve,\
self.action_Quit ))
self.action_New.triggered.connect(self.fileNew)
self.action_Open.triggered.connect(self.fileOpen)
self.actionSave_As.triggered.connect(self.fileSaveAs)
self.action_Save.triggered.connect(self.fileSave)
self.action_Solve.triggered.connect(self.trussSolve)
self.action_About.triggered.connect(self.aboutBox)
self.action_CCPL.triggered.connect(self.displayCCPL)
self.action_Help.triggered.connect(self.help)
self.plainTextEdit.textChanged.connect(self.setDirty)
self.action_New = self.editAction(self.action_New, None,\
'ctrl+N', 'filenew', 'New File.')
self.action_Open = self.editAction(self.action_Open, None, 'ctrl+O',
'fileopen', 'Open File.')
self.actionSave_As = self.editAction(self.actionSave_As,\
None, 'ctrl+A', 'filesaveas',\
'Save and Name File.')
self.action_Save = self.editAction(self.action_Save, None, 'ctrl+S',
'filesave', 'Save File.')
self.action_Solve = self.editAction(self.action_Solve, None, 'ctrl+L',
'solve', 'Solve Structure.')
self.action_About = self.editAction(self.action_About, None, 'ctrl+B',
'about', 'Pop About Box.')
self.action_CCPL = self.editAction(self.action_CCPL, None, 'ctrl+G',
'licence', 'Show Licence')
self.action_Help = self.editAction(self.action_Help, None, 'ctrl+H',
'help', 'Show Help Page.')
self.action_Quit = self.editAction(self.action_Quit, None, 'ctrl+Q',
'quit', 'Quit the program.')
self.plainTextEdit_2.setReadOnly(True)
def setDirty(self):
'''On change of text in textEdit window, set the flag
"dirty" to True'''
index = self.tabWidget.currentIndex()
if index is not 0:
return
if self.dirty:
return True
self.dirty = True
self.updateStatus('self.dirty set to True')
def clearDirty(self):
'Clear dirty flag'
self.dirty = False
def fileNew(self):
'''Clear both Data Page and Solution Page.'''
self.plainTextEdit.setPlainText(' ')
self.plainTextEdit_2.setPlainText(' ')
self.clearDirty(self)
def okToContinue(self):
if self.dirty:
reply = QMessageBox.question(
self, "Data Loader - Unsaved Changes", "Save unsaved changes?",
QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel)
if reply == QMessageBox.Cancel:
return False
elif reply == QMessageBox.Yes:
self.clearDirty()
return self.fileSave()
return True
def okRead(self):
'Pop-up a warning message.'
reply = QMessageBox.warning(
self, "Warning",
'''\nFile Open and Save is possible only in Data Page!
\n\(Use SaveAs for Solution Page)''', QMessageBox.Ok)
return True
def fileOpen(self):
'''Open a file in Data Page (with index == 0)'''
if self.tabWidget.currentIndex():
self.okRead()
return
if not self.okToContinue():
return
dir = (os.path.dirname(self.filename)
if self.filename is not None else ".")
self.filetuple = QFileDialog.getOpenFileName(self,\
"Open File", dir, \
"Data (*.dat *.txt)\nAll Files (*.*)")
self.filename = self.filetuple[0]
fname = self.filename
# QFileDialog returns a tuple x with x[0] = file name and
# x[1] = type of filter.
if fname:
self.loadFile(fname)
self.filename = fname
self.updateStatus('New file opened.')
def loadFile(self, fname=None):
fl = open(fname)
text = fl.read()
self.plainTextEdit.setPlainText(text)
self.dirty = False
def fileSave(self):
'''Save file with current file name.'''
if self.tabWidget.currentIndex():
self.okRead()
return
if self.filename is None:
return self.fileSaveAs()
else:
flname = self.filename
if flname:
try:
with open(flname, 'w') as fl:
fl.write(tempText)
self.dirty = False
self.updateStatus('File saved.')
except IOError:
self.dirty = True
self.updateStatus('File not saved.')
return False
return True
else:
self.updateStatus('Failed to save... ')
return False
self.filename = None
self.dirty = False
def fileSaveAs(self):
'''Save file with a new name.'''
qpr = self.qprintline
fname = self.filename if self.filename is not None else\
"NoName"
self.filetuple = QFileDialog.getSaveFileName(
self, "Truss program - Save File", fname, "Data File (*.*)")
flname = self.filetuple[0]
index = self.tabWidget.currentIndex()
if index == 0:
self.filename = flname
if flname:
fl = open(flname, 'w')
tempText = self.plainTextEdit.toPlainText()
fl.write(tempText)
fl.close()
self.dirty = False
self.updateStatus('File saved.')
elif index == 1:
if flname:
fl = open(flname, 'w')
tempText = self.plainTextEdit_2.toPlainText()
fl.write(tempText)
fl.close()
def trussSolve(self):
'''Solve a statically determinate truss, specified in
Data Page and display the results in the Solution Page.
To start, make a copy of the Data Page with a header all
shown on the Data Page.'''
printline = self.qprintline
dataBall = self.plainTextEdit.toPlainText()
self.plainTextEdit_2.clear()
printline('================================')
flbase = os.path.basename(self.filename)
printline('SOLUTION FOR ' + flbase)
printline('================================')
dataBall = self.plainTextEdit.toPlainText()
ncrunch.main(printline, self.filename, dataBall)
def aboutBox(self):
'''Popup a box with about message.'''
QMessageBox.about(self, "About PySide, Platform and the like",
"""<b>Part of Structural Analysis.</b> v %s
<p>Copyright © 2011 Algis Kabaila.
All rights reserved in accordance with
Creative Commons Attribution Licence (CCPL) v3
or later - NO WARRANTIES!
<p>This progam finds bar forces in
statically determinate trusses.
<p>Python %s - PySide version %s - Qt version %s on\
%s""" % (__version__, platform.python_version(),\
PySide.__version__, PySide.QtCore.__version__,
platform.system()))
def displayCCPL(self):
'''Read and display CCPL licence.'''
self.plainTextEdit.setPlainText(open('CCPL.txt').read())
self.dirty = False
self.filename = 'COPYING.txt'
self.updateStatus('CCPL displayed.')
def help(self):
'''Read and display a help file- currently the README.txt.'''
self.plainTextEdit.setPlainText(open('README.md').read())
self.dirty = False
self.filename = 'README.txt'
self.updateStatus('README displayed.')
def addActions(self, target, actions):
'''Actions are added to Tool Bar.'''
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def editAction(self,
action,
slot=None,
shortcut=None,
icon=None,
tip=None):
'''This method adds to action: icon, shortcut, ToolTip,\
StatusTip and can connect triggered action to slot '''
if icon is not None:
action.setIcon(QIcon(":/%s.png" % (icon)))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
action.triggered.connect(slot)
return action
def qreadline(self, lineNo):
'''Read one line from Data Page (lineNo starts with 0)'''
return str(self.plainTextEdit.document().\
findBlockByLineNumber(lineNo).text()).rstrip()
def qprintline(self, line):
'''Append one line to Solution Page.'''
self.plainTextEdit_2.appendPlainText(line.rstrip())
def updateStatus(self, message):
'''Keep status current.'''
if self.filename is not None:
flbase = os.path.basename(self.filename)
self.setWindowTitle(str("Truss Analysis - " +\
flbase + "[*]") )
self.statusBar().showMessage(message, 5000)
self.setWindowModified(self.dirty)
class ClassyWidget(QWidget):
def __init__(self, parent=None):
QWidget.__init__(self, parent)
layout = QVBoxLayout(self)
horiz_layout = QHBoxLayout()
self.conditional_legend_widget = EdgeWidget(self, True)
self.conditional_legend_widget.setMinimumHeight(15)
horiz_layout.addWidget(self.conditional_legend_widget)
self.conditional_legend_label = QLabel("Conditional transition", self)
horiz_layout.addWidget(self.conditional_legend_label)
self.unconditional_legend_widget = EdgeWidget(self, False)
self.unconditional_legend_widget.setMinimumHeight(15)
horiz_layout.addWidget(self.unconditional_legend_widget)
self.unconditional_legend_label = QLabel("Non-conditional transition",
self)
horiz_layout.addWidget(self.unconditional_legend_label)
layout.addLayout(horiz_layout)
self.splitter = QSplitter(self)
layout.addWidget(self.splitter)
self.view = ClassyView(self.splitter)
# layout.addWidget(self.view)
self.scene = ClassyScene(self)
self.view.setScene(self.scene)
self._menu_bar = QMenuBar(self)
self._menu = QMenu("&File")
self._menu_bar.addMenu(self._menu)
layout.setMenuBar(self._menu_bar)
self.open_action = QAction("O&pen", self)
self.exit_action = QAction("E&xit", self)
self._menu.addAction(self.open_action)
self._menu.addAction(self.exit_action)
self.connect(self.open_action, SIGNAL("triggered()"), self.open_file)
self.connect(self.exit_action, SIGNAL("triggered()"), self.close)
self.settings = QSettings("CD Projekt RED", "TweakDB")
self.log_window = QPlainTextEdit(self.splitter)
self.splitter.setOrientation(Qt.Vertical)
self.setWindowTitle("Classy nodes")
@Slot()
def open_file(self):
last_dir = self.settings.value("last_dir")
if last_dir is None:
last_dir = ''
# noinspection PyCallByClass
file_path = QFileDialog.getOpenFileName(self, "Select tweakdb file",
last_dir, "*.tweak")
if file_path[0]:
tweak_file = file_path[0]
self.settings.setValue("last_dir", os.path.dirname(tweak_file))
else:
return
try:
self._process_db_file(file_path[0])
except Exception as e:
# noinspection PyCallByClass
QMessageBox.critical(
self, "Error",
"Error while loading {0}: {1}".format(file_path, str(e)))
self.setWindowTitle("Classy nodes - {0}".format(file_path[0]))
@Slot(str)
def log(self, value):
self.log_window.appendPlainText(str(value))
self.log_window.verticalScrollBar().scroll(
0,
self.log_window.verticalScrollBar().maximum())
def _process_db_file(self, file_path):
class TweakClass(object):
def __init__(self, name, base=""):
self.name = name
self.base = base
self.transitions_to = []
self.transitions_conditions = []
def has_transition(self, name):
return name in self.transitions_to
def transition_conditional(self, name):
if len(self.transitions_to) != len(
self.transitions_conditions):
raise ValueError(
"Transitions length does not match conditions length in {0}"
.format(self.name))
if name not in name:
raise ValueError(
"Class {0} has no transition to {1}".format(
self.name, name))
return self.transitions_conditions[self.transitions_to.index(
name)]
with open(file_path, "r") as f:
contents = f.read()
lines = contents.split("\n")
is_in_class = False
is_class_declared = False
current_class = None
classes = {}
parsed_line = 0
for line in lines:
parsed_line += 1
if line.startswith("package"):
continue
if line.startswith("using"):
continue
if not line.strip():
continue
if line.startswith("//"):
continue
if not is_in_class and not is_class_declared:
if line.find("=") > 0: # static variable
continue
spl = [x.strip() for x in line.strip().split(":")]
if len(spl) == 1:
c = TweakClass(spl[0])
else:
c = TweakClass(spl[0], spl[1])
is_class_declared = True
current_class = c
classes[c.name] = c
continue
if not is_in_class and not line.find("{") >= 0:
raise ValueError(
"Failed to parse the file {0} in line: {1}".format(
file_path, parsed_line))
else:
is_in_class = True
if line.find("}") >= 0:
is_in_class = False
is_class_declared = False
continue
if re.match(".+transitionTo( +||\t+)+=( +||\t+)+\[.+\]$",
line.strip()):
spl = line.split("=")[1].strip()[1:-1]
transitions = [x.strip().strip("\"") for x in spl.split(",")]
current_class.transitions_to = transitions
continue
if re.match(".+transitionCondition( +||\t+)+=( +||\t+)+\[.+\]$",
line.strip()):
spl = "=".join(line.split("=")[1:]).strip()[1:-1]
transitions = [x.strip().strip("\"") for x in spl.split(",")]
current_class.transitions_conditions = [
True if x.strip() == '=' else False for x in transitions
]
continue
self.scene.clear()
# --- Create nodes
for node_key in classes:
# g.add_node(str(node_key),
# shape_fill="#aaaa33",
# shape="roundrectangle",
# font_style="bolditalic",
# label=classes.get(node_key).name)
self.scene.add_node(node_key)
edges_created = []
# --- Create edges
for node_key in classes:
# --- Get each node
current_node = classes.get(node_key)
# --- Iterate it's "transition_to" list
for t in current_node.transitions_to:
# --- Find node to create transition edge to
transition_to_node = classes.get(t)
if transition_to_node is None:
self.log("Invalid transition in class {0} -> {1}".format(
node_key, t))
continue
# --- Check if it's a two way transition
two_way = False
if transition_to_node.has_transition(node_key):
two_way = True
# --- make sure that these edges were not created yet
edges_to_create = ["{0}!@#{1}".format(node_key, t)]
if two_way:
edges_to_create = ["{0}!@#{1}".format(t, node_key)]
edge_already_created = False
for e in edges_to_create:
if e in edges_created:
edge_already_created = True
break
if edge_already_created:
continue
# --- Check if the transitions are conditional or not
try:
conditional_to = current_node.transition_conditional(t)
except ValueError as e:
self.log(str(e))
continue
conditional_from = False
if two_way:
conditional_from = transition_to_node.transition_conditional(
node_key)
self.scene.add_edge(current_node.name,
transition_to_node.name,
conditional_to=conditional_to,
two_way=two_way,
conditional_from=conditional_from)
edges_created += edges_to_create
self.scene.layout_nodes()
class XMEGAProgrammerDialog(QtFixes.QDialog):
def __init__(self):
super(XMEGAProgrammerDialog, self).__init__(CWMainGUI.getInstance())
# self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
self.xmega = XMEGAProgrammer()
self.setWindowTitle("ChipWhisperer-Lite XMEGA Programmer")
layout = QVBoxLayout()
layoutFW = QHBoxLayout()
self.flashLocation = QtFixes.QLineEdit()
flashFileButton = QPushButton("Find")
flashFileButton.clicked.connect(self.findFlash)
layoutFW.addWidget(QLabel("FLASH File"))
layoutFW.addWidget(self.flashLocation)
layoutFW.addWidget(flashFileButton)
layout.addLayout(layoutFW)
self.flashLocation.setText(QSettings().value("xmega-flash-location"))
# Add buttons
readSigBut = QPushButton("Check Signature")
readSigBut.clicked.connect(self.readSignature)
verifyFlashBut = QPushButton("Verify FLASH")
verifyFlashBut.clicked.connect(self.verifyFlash)
verifyFlashBut.setEnabled(False)
progFlashBut = QPushButton("Erase/Program/Verify FLASH")
progFlashBut.clicked.connect(self.writeFlash)
layoutBut = QHBoxLayout()
layoutBut.addWidget(readSigBut)
layoutBut.addWidget(verifyFlashBut)
layoutBut.addWidget(progFlashBut)
layout.addLayout(layoutBut)
# Add status stuff
self.statusLine = QPlainTextEdit()
self.statusLine.setReadOnly(True)
# self.statusLine.setFixedHeight(QFontMetrics(self.statusLine.font()).lineSpacing() * 5 + 10)
self.xmega.newTextLog.connect(self.append)
layout.addWidget(self.statusLine)
# Set dialog layout
self.setLayout(layout)
def append(self, text):
self.statusLine.appendPlainText(text)
def findFlash(self):
fname, _ = QFileDialog.getOpenFileName(self, 'Find FLASH File', QSettings().value("xmega-flash-location"), '*.hex')
if fname:
self.flashLocation.setText(fname)
QSettings().setValue("xmega-flash-location", fname)
def readSignature(self, close=True):
self.xmega.find()
if close:
self.xmega.close()
def verifyFlash(self):
pass
def writeFlash(self, erase=True, verify=True):
status = "FAILED"
fname = self.flashLocation.text()
self.statusLine.appendPlainText("***Starting FLASH program process at %s***" % datetime.now().strftime('%H:%M:%S'))
if (os.path.isfile(fname)):
self.statusLine.appendPlainText("File %s last changed on %s" % (fname, time.ctime(os.path.getmtime(fname))))
QCoreApplication.processEvents()
try:
self.statusLine.appendPlainText("Entering Programming Mode")
QCoreApplication.processEvents()
self.readSignature(close=False)
if erase:
try:
self.xmega.erase()
except IOError:
self.statusLine.appendPlainText("**chip-erase timeout, erasing application only**")
QCoreApplication.processEvents()
self.xmega.xmega.enablePDI(False)
self.xmega.xmega.enablePDI(True)
self.xmega.erase("app")
QCoreApplication.processEvents()
self.xmega.program(self.flashLocation.text(), memtype="flash", verify=verify)
QCoreApplication.processEvents()
self.statusLine.appendPlainText("Exiting programming mode")
self.xmega.close()
QCoreApplication.processEvents()
status = "SUCCEEDED"
except IOError, e:
self.statusLine.appendPlainText("FAILED: %s" % str(e))
try:
self.xmega.close()
except IOError:
pass
else:
class AVRProgrammerDialog(QtFixes.QDialog):
def __init__(self):
super(AVRProgrammerDialog, self).__init__(CWMainGUI.getInstance())
# self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
self.avr = AVRProgrammer()
self.setWindowTitle("ChipWhisperer-Lite AVR Programmer")
layout = QVBoxLayout()
layoutFW = QHBoxLayout()
self.flashLocation = QtFixes.QLineEdit()
flashFileButton = QPushButton("Find")
flashFileButton.clicked.connect(self.findFlash)
layoutFW.addWidget(QLabel("FLASH File"))
layoutFW.addWidget(self.flashLocation)
layoutFW.addWidget(flashFileButton)
layout.addLayout(layoutFW)
self.flashLocation.setText(QSettings().value("avr-flash-location"))
# Add buttons
readSigBut = QPushButton("Check Signature")
readSigBut.clicked.connect(self.readSignature)
verifyFlashBut = QPushButton("Verify FLASH")
verifyFlashBut.clicked.connect(self.verifyFlash)
verifyFlashBut.setEnabled(False)
progFlashBut = QPushButton("Erase/Program/Verify FLASH")
progFlashBut.clicked.connect(self.writeFlash)
layoutBut = QHBoxLayout()
layoutBut.addWidget(readSigBut)
layoutBut.addWidget(verifyFlashBut)
layoutBut.addWidget(progFlashBut)
layout.addLayout(layoutBut)
layoutFuse = QHBoxLayout()
readFuseBut = QPushButton("Read Fuses")
readFuseBut.clicked.connect(self.readFuses)
writeFuseBut = QPushButton("Write Fuses")
writeFuseBut.clicked.connect(self.writeFuses)
self.lowfuseLine = QtFixes.QLineEdit("?")
self.lowfuseLine.setMaxLength(2)
self.lowfuseLine.setFixedWidth(25)
self.highfuseLine = QtFixes.QLineEdit("?")
self.highfuseLine.setMaxLength(2)
self.highfuseLine.setFixedWidth(25)
self.extfuseLine = QtFixes.QLineEdit("?")
self.extfuseLine.setMaxLength(2)
self.extfuseLine.setFixedWidth(25)
# Don't allow use to change these fuses
self.highfuseLine.setReadOnly(True)
self.extfuseLine.setReadOnly(True)
layoutFuse.addWidget(readFuseBut)
layoutFuse.addWidget(QLabel("LOW:"))
layoutFuse.addWidget(self.lowfuseLine)
layoutFuse.addWidget(QLabel("HIGH:"))
layoutFuse.addWidget(self.highfuseLine)
layoutFuse.addWidget(QLabel("EXT:"))
layoutFuse.addWidget(self.extfuseLine)
layoutFuse.addWidget(writeFuseBut)
layout.addLayout(layoutFuse)
layoutExtra = QHBoxLayout()
self.clockMode = QPushButton("Enable Slow Clock Mode")
self.clockMode.setCheckable(True)
self.clockMode.clicked.connect(self.toggleSlowClock)
layoutExtra.addWidget(self.clockMode)
layoutExtra.addStretch()
layout.addLayout(layoutExtra)
# Add status stuff
self.statusLine = QPlainTextEdit()
self.statusLine.setReadOnly(True)
# self.statusLine.setFixedHeight(QFontMetrics(self.statusLine.font()).lineSpacing() * 5 + 10)
layout.addWidget(self.statusLine)
self.avr.newTextLog.connect(self.append)
# Set dialog layout
self.setLayout(layout)
def append(self, text):
self.statusLine.appendPlainText(text)
def toggleSlowClock(self):
if self.clockMode.isChecked():
self.avr.avr.enableSlowClock(True)
self.clockMode.setText("Disable Slow Clock Mode")
else:
self.avr.avr.enableSlowClock(False)
self.clockMode.setText("Enable Slow Clock Mode")
def reject(self):
#Ensure we disable slow-clock mode
if self.clockMode.isChecked():
self.clockMode.setChecked(False)
self.toggleSlowClock()
#Release AVR
self.avr.close()
QDialog.reject(self)
def findFlash(self):
fname, _ = QFileDialog.getOpenFileName(self, 'Find FLASH File', QSettings().value("avr-flash-location"), '*.hex')
if fname:
self.flashLocation.setText(fname)
QSettings().setValue("avr-flash-location", fname)
def readSignature(self, close=True):
self.avr.find()
if close:
self.avr.close()
def readFuses(self):
try:
self.readSignature(close=False)
self.statusLine.appendPlainText("Reading fuses")
lfuse = self.avr.avr.readFuse("low")
hfuse = self.avr.avr.readFuse("high")
efuse = self.avr.avr.readFuse("extended")
self.statusLine.appendPlainText("OK: %02x %02x %02x" % (lfuse, hfuse, efuse))
self.lowfuseLine.setText("%02x" % lfuse)
self.highfuseLine.setText("%02x" % hfuse)
self.extfuseLine.setText("%02x" % efuse)
except IOError:
self.statusLine.appendPlainText("Reading fuses failed")
self.lowfuseLine.setText("?")
self.highfuseLine.setText("?")
self.extfuseLine.setText("?")
self.avr.close()
def writeFuses(self):
lfuse = int(self.lowfuseLine.text(), 16)
# hfuse = int(self.highfuseLine.text(), 16)
# efuse = int(self.extfuseLine.text(), 16)
self.statusLine.appendPlainText("Writing fuse: not only 'low' fuse is written, as hfuse/efuse can disable device\n")
try:
self.readSignature(close=False)
self.avr.avr.writeFuse("low", lfuse)
# self.avr.avr.writeFuse("high", hfuse)
# self.avr.avr.writeFuse("extended", efuse)
# print("%x %x %x" % (lfuse, hfuse, efuse))
self.avr.close()
except:
self.avr.close()
raise
def verifyFlash(self):
self.statusLine.appendPlainText("Verify not implemented")
def writeFlash(self, erase=True, verify=True):
status = "FAILED"
fname = self.flashLocation.text()
self.statusLine.appendPlainText("***Starting FLASH program process at %s***" % datetime.now().strftime('%H:%M:%S'))
if (os.path.isfile(fname)):
self.statusLine.appendPlainText("File %s last changed on %s" % (fname, time.ctime(os.path.getmtime(fname))))
QCoreApplication.processEvents()
try:
self.statusLine.appendPlainText("Entering Programming Mode")
QCoreApplication.processEvents()
self.readSignature(close=False)
if erase:
self.avr.erase()
self.avr.close()
self.readSignature(close=False)
QCoreApplication.processEvents()
self.avr.program(self.flashLocation.text(), memtype="flash", verify=verify)
QCoreApplication.processEvents()
self.statusLine.appendPlainText("Exiting programming mode")
self.avr.close()
QCoreApplication.processEvents()
status = "SUCCEEDED"
except IOError as e:
self.statusLine.appendPlainText("FAILED: %s" % str(e))
try:
self.avr.close()
except IOError:
pass
else:
self.statusLine.appendPlainText("%s does not appear to be a file, check path" % fname)
self.statusLine.appendPlainText("***FLASH Program %s at %s***" % (status, datetime.now().strftime('%H:%M:%S')))
def setUSBInterface(self, iface):
self.avr.setUSBInterface(iface)
def __init__(self):
QMainWindow.__init__(self)
# the user interface, consisting of a progress bar above a plain
# text edit, which logs all actions.
container = QWidget(self)
container.setLayout(QVBoxLayout(container))
self.setCentralWidget(container)
progressbar = QProgressBar(container)
container.layout().addWidget(progressbar)
log = QPlainTextEdit(container)
container.layout().addWidget(log)
# the actual worker thread
counter = Counter(100, self)
# an action to quit the windows
exit = QAction(QIcon.fromTheme('application-exit'), 'exit', self)
# add two actions to start and stop the worker to the toolbar of the
# main window
start_counting = QAction(QIcon.fromTheme('media-playback-start'),
'Start counting', self)
stop_counting = QAction(QIcon.fromTheme('media-playback-stop'),
'Stop counting', self)
# initially no counter runs, so we can disable the stop action
stop_counting.setEnabled(False)
# add all actions to a toolbar
actions = self.addToolBar('Actions')
actions.addAction(exit)
actions.addSeparator()
actions.addAction(start_counting)
actions.addAction(stop_counting)
# quit the application, if the quit action is triggered
exit.triggered.connect(QApplication.instance().quit)
# start and stop the counter, if the corresponding actions are
# triggered.
start_counting.triggered.connect(counter.start)
stop_counting.triggered.connect(counter.stop)
# adjust the minimum and the maximum of the progress bar, if
# necessary. Not really required in this snippet, but added for the
# purpose of demonstrating it
counter.minimumChanged.connect(progressbar.setMinimum)
counter.maximumChanged.connect(progressbar.setMaximum)
# switch the enabled states of the actions according to whether the
# worker is running or not
counter.started.connect(partial(start_counting.setEnabled, False))
counter.started.connect(partial(stop_counting.setEnabled, True))
counter.finished.connect(partial(start_counting.setEnabled, True))
counter.finished.connect(partial(stop_counting.setEnabled, False))
# update the progess bar continuously
counter.progress.connect(progressbar.setValue)
# log all actions in our logging widget
counter.started.connect(
partial(self.statusBar().showMessage, 'Counting'))
counter.finished.connect(partial(self.statusBar().showMessage, 'Done'))
# log a forced stop
stop_counting.triggered.connect(partial(log.appendPlainText,
'Stopped'))
# log all counted values
counter.progress.connect(lambda v: log.appendPlainText(str(v)))
# and finally show the current state in the status bar.
counter.started.connect(partial(log.appendPlainText, 'Counting'))
counter.finished.connect(partial(log.appendPlainText, 'Done'))
class ConsoleWidget(QMainWindow):
def __init__(self):
super(ConsoleWidget, self).__init__()
self.setWindowTitle('1c query')
self._connection = None
self._home = os.path.expanduser('~/%s' % QApplication.applicationName())
if not os.path.isdir(self._home):
os.mkdir(self._home)
self.queryToolBar = self.addToolBar('Query')
self.queryAction = self.queryToolBar.addAction('Run', self.executeQuery)
self.queryAction.setDisabled(True)
uri_history = list()
path = os.path.join(self._home, 'uri_history.txt')
if os.path.isfile(path):
uri_history = open(path, 'r').read().split('\n')
self.connectionToolBar = self.addToolBar('Connection')
self.connectionUriCombo = QComboBox(self)
self.connectionUriCombo.setEditable(True)
if not uri_history:
self.connectionUriCombo.addItem('File="";usr="";pwd="";')
self.connectionUriCombo.addItem('Srvr="{host}";Ref="{ref}";Usr="******";Pwd="{password}";')
else:
self.connectionUriCombo.addItems(uri_history)
self.connectionUriCombo.setCurrentIndex(len(uri_history) - 1)
self.connectionUriCombo.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Maximum)
self.connectionToolBar.addWidget(self.connectionUriCombo)
self.onesVersionCombo = QComboBox(self)
self.onesVersionCombo.addItems(['8.3', '8.2', '8.1', '8.0'])
self.onesVersionCombo.setCurrentIndex(0)
self.connectionToolBar.addWidget(self.onesVersionCombo)
self.connectAction = self.connectionToolBar.addAction('Connect', self.connectOneS)
self.disconnectAction = self.connectionToolBar.addAction('Disconnect', self.disconnectOneS)
self.disconnectAction.setDisabled(True)
self.logEdit = QPlainTextEdit(self)
self.logDock = QDockWidget('Log', self)
self.logDock.setWidget(self.logEdit)
self.addDockWidget(Qt.BottomDockWidgetArea, self.logDock, Qt.Horizontal)
self.splitter = QSplitter(Qt.Vertical, self)
self.setCentralWidget(self.splitter)
self.sqlEdit = QTextEdit(self)
self.sqlEdit.setLineWrapMode(QTextEdit.NoWrap)
path = os.path.join(self._home, 'last-sql.txt')
if os.path.isfile(path):
sql = open(path, 'r').read()
self.sqlEdit.setText(sql)
self.model = QStandardItemModel(self)
self.tableView = QTableView(self)
self.tableView.setModel(self.model)
self.splitter.addWidget(self.sqlEdit)
self.splitter.addWidget(self.tableView)
self.splitter.setStretchFactor(0, 3)
self.splitter.setStretchFactor(1, 2)
def query(self, sql):
if not self._connection:
self.logEdit.appendPlainText('No connection')
return None
try:
query = self._connection.NewObject('Query', sql)
result = query.Execute()
except Exception as e:
self.logEdit.appendPlainText(str(e))
return None
return result
def refresh(self, result):
self.model.clear()
columns = list()
result_columns = result.Columns
for index in range(result_columns.Count()):
name = result_columns.Get(index).Name
columns.append(name)
self.model.setColumnCount(len(columns))
for section, name in enumerate(columns):
self.model.setHeaderData(section, Qt.Horizontal, name)
select = result.Choose()
self.logEdit.appendPlainText('Selected %d records' % select.Count())
while select.Next():
items = list()
for index in range(len(columns)):
value = select.Get(index)
item = QStandardItem('')
if isinstance(value, bool):
item.setText(value and 'Yes' or 'No')
elif isinstance(value, (int, str)):
item.setText(str(value))
elif isinstance(value, datetime.datetime):
item.setText(value.strftime('%Y.%m.%d %H:%M:%S'))
else:
item.setText(str(value))
items.append(item)
self.model.appendRow(items)
@Slot()
def executeQuery(self):
sql = self.sqlEdit.toPlainText()
result = self.query(sql)
if result:
path = os.path.join(self._home, 'last-sql.txt')
open(path, 'w').write(sql)
self.refresh(result)
@Slot()
def connectOneS(self):
uri = self.connectionUriCombo.currentText().strip()
if not uri:
self.logEdit.appendPlainText('Need a connection string')
return
version = self.onesVersionCombo.currentText()
comName = "V%s.COMConnector" % str(version).replace('.', '')
pythoncom.CoInitialize()
try:
obj = win32com.client.Dispatch(comName)
self._connection = obj.Connect(uri)
except Exception as e:
self.logEdit.appendPlainText(str(e))
return
self.connectAction.setDisabled(True)
self.disconnectAction.setEnabled(True)
self.queryAction.setEnabled(True)
uri_history = list()
for i in range(self.connectionUriCombo.count()):
uri_history.append(self.connectionUriCombo.itemText(i))
if uri not in uri_history:
self.connectionUriCombo.clearEditText()
self.connectionUriCombo.addItem(uri)
self.connectionUriCombo.setCurrentIndex(len(uri_history))
uri_history.append(uri)
path = os.path.join(self._home, 'uri_history.txt')
open(path, 'w').write('\n'.join(uri_history))
@Slot()
def disconnectOneS(self):
pythoncom.CoUninitialize()
self._connection = None
self.connectAction.setEnabled(True)
self.disconnectAction.setDisabled(True)
self.queryAction.setDisabled(True)
def __init__(self):
QMainWindow.__init__(self)
# the user interface, consisting of a progress bar above a plain
# text edit, which logs all actions.
container = QWidget(self)
container.setLayout(QVBoxLayout(container))
self.setCentralWidget(container)
progressbar = QProgressBar(container)
container.layout().addWidget(progressbar)
log = QPlainTextEdit(container)
container.layout().addWidget(log)
# the actual worker thread
counter = Counter(100, self)
# an action to quit the windows
exit = QAction(QIcon.fromTheme('application-exit'), 'exit', self)
# add two actions to start and stop the worker to the toolbar of the
# main window
start_counting = QAction(QIcon.fromTheme('media-playback-start'),
'Start counting', self)
stop_counting = QAction(QIcon.fromTheme('media-playback-stop'),
'Stop counting', self)
# initially no counter runs, so we can disable the stop action
stop_counting.setEnabled(False)
# add all actions to a toolbar
actions = self.addToolBar('Actions')
actions.addAction(exit)
actions.addSeparator()
actions.addAction(start_counting)
actions.addAction(stop_counting)
# quit the application, if the quit action is triggered
exit.triggered.connect(QApplication.instance().quit)
# start and stop the counter, if the corresponding actions are
# triggered.
start_counting.triggered.connect(counter.start)
stop_counting.triggered.connect(counter.stop)
# adjust the minimum and the maximum of the progress bar, if
# necessary. Not really required in this snippet, but added for the
# purpose of demonstrating it
counter.minimumChanged.connect(progressbar.setMinimum)
counter.maximumChanged.connect(progressbar.setMaximum)
# switch the enabled states of the actions according to whether the
# worker is running or not
counter.started.connect(partial(start_counting.setEnabled, False))
counter.started.connect(partial(stop_counting.setEnabled, True))
counter.finished.connect(partial(start_counting.setEnabled, True))
counter.finished.connect(partial(stop_counting.setEnabled, False))
# update the progess bar continuously
counter.progress.connect(progressbar.setValue)
# log all actions in our logging widget
counter.started.connect(
partial(self.statusBar().showMessage, 'Counting'))
counter.finished.connect(
partial(self.statusBar().showMessage, 'Done'))
# log a forced stop
stop_counting.triggered.connect(
partial(log.appendPlainText, 'Stopped'))
# log all counted values
counter.progress.connect(lambda v: log.appendPlainText(str(v)))
# and finally show the current state in the status bar.
counter.started.connect(partial(log.appendPlainText, 'Counting'))
counter.finished.connect(partial(log.appendPlainText, 'Done'))
class XMEGAProgrammerDialog(QtFixes.QDialog):
def __init__(self):
super(XMEGAProgrammerDialog, self).__init__(CWMainGUI.getInstance())
# self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
self.xmega = XMEGAProgrammer()
self.setWindowTitle("ChipWhisperer-Lite XMEGA Programmer")
layout = QVBoxLayout()
layoutFW = QHBoxLayout()
self.flashLocation = QtFixes.QLineEdit()
flashFileButton = QPushButton("Find")
flashFileButton.clicked.connect(self.findFlash)
layoutFW.addWidget(QLabel("FLASH File"))
layoutFW.addWidget(self.flashLocation)
layoutFW.addWidget(flashFileButton)
layout.addLayout(layoutFW)
self.flashLocation.setText(QSettings().value("xmega-flash-location"))
# Add buttons
readSigBut = QPushButton("Check Signature")
readSigBut.clicked.connect(self.readSignature)
verifyFlashBut = QPushButton("Verify FLASH")
verifyFlashBut.clicked.connect(self.verifyFlash)
verifyFlashBut.setEnabled(False)
progFlashBut = QPushButton("Erase/Program/Verify FLASH")
progFlashBut.clicked.connect(self.writeFlash)
layoutBut = QHBoxLayout()
layoutBut.addWidget(readSigBut)
layoutBut.addWidget(verifyFlashBut)
layoutBut.addWidget(progFlashBut)
layout.addLayout(layoutBut)
# Add status stuff
self.statusLine = QPlainTextEdit()
self.statusLine.setReadOnly(True)
# self.statusLine.setFixedHeight(QFontMetrics(self.statusLine.font()).lineSpacing() * 5 + 10)
self.xmega.newTextLog.connect(self.append)
layout.addWidget(self.statusLine)
# Set dialog layout
self.setLayout(layout)
def append(self, text):
self.statusLine.appendPlainText(text)
def findFlash(self):
fname, _ = QFileDialog.getOpenFileName(
self, 'Find FLASH File',
QSettings().value("xmega-flash-location"), '*.hex')
if fname:
self.flashLocation.setText(fname)
QSettings().setValue("xmega-flash-location", fname)
def readSignature(self, close=True):
self.xmega.find()
if close:
self.xmega.close()
def verifyFlash(self):
pass
def writeFlash(self, erase=True, verify=True):
status = "FAILED"
fname = self.flashLocation.text()
self.statusLine.appendPlainText(
"***Starting FLASH program process at %s***" %
datetime.now().strftime('%H:%M:%S'))
if (os.path.isfile(fname)):
self.statusLine.appendPlainText(
"File %s last changed on %s" %
(fname, time.ctime(os.path.getmtime(fname))))
QCoreApplication.processEvents()
try:
self.statusLine.appendPlainText("Entering Programming Mode")
QCoreApplication.processEvents()
self.readSignature(close=False)
if erase:
try:
self.xmega.erase()
except IOError:
self.statusLine.appendPlainText(
"**chip-erase timeout, erasing application only**")
QCoreApplication.processEvents()
self.xmega.xmega.enablePDI(False)
self.xmega.xmega.enablePDI(True)
self.xmega.erase("app")
QCoreApplication.processEvents()
self.xmega.program(self.flashLocation.text(),
memtype="flash",
verify=verify)
QCoreApplication.processEvents()
self.statusLine.appendPlainText("Exiting programming mode")
self.xmega.close()
QCoreApplication.processEvents()
status = "SUCCEEDED"
except IOError, e:
self.statusLine.appendPlainText("FAILED: %s" % str(e))
try:
self.xmega.close()
except IOError:
pass
else:
class Truss(QMainWindow):
def __init__(self, parent=None):
super(Truss, self).__init__(parent)
self.resize(800, 600)
self.filename = None
self.filetuple = None
self.dirty = False # Refers to Data Page only.
centralwidget = QWidget(self)
gridLayout = QGridLayout(centralwidget)
self.tabWidget = QTabWidget(centralwidget)
self.tab = QWidget()
font = QFont()
font.setFamily("Courier 10 Pitch")
font.setPointSize(12)
self.tab.setFont(font)
gridLayout_3 = QGridLayout(self.tab)
self.plainTextEdit = QPlainTextEdit(self.tab)
gridLayout_3.addWidget(self.plainTextEdit, 0, 0, 1, 1)
self.tabWidget.addTab(self.tab, "")
self.tab_2 = QWidget()
self.tab_2.setFont(font)
gridLayout_2 = QGridLayout(self.tab_2)
self.plainTextEdit_2 = QPlainTextEdit(self.tab_2)
gridLayout_2.addWidget(self.plainTextEdit_2, 0, 0, 1, 1)
self.tabWidget.addTab(self.tab_2, "")
gridLayout.addWidget(self.tabWidget, 0, 0, 1, 1)
self.setCentralWidget(centralwidget)
menubar = QMenuBar(self)
menubar.setGeometry(QRect(0, 0, 800, 29))
menu_File = QMenu(menubar)
self.menu_Solve = QMenu(menubar)
self.menu_Help = QMenu(menubar)
self.setMenuBar(menubar)
self.statusbar = QStatusBar(self)
self.setStatusBar(self.statusbar)
self.action_New = QAction(self)
self.actionSave_As = QAction(self)
self.action_Save = QAction(self)
self.action_Open = QAction(self)
self.action_Quit = QAction(self)
self.action_About = QAction(self)
self.actionShow_CCPL = QAction(self)
self.action_Solve = QAction(self)
self.action_CCPL = QAction(self)
self.action_Help = QAction(self)
menu_File.addAction(self.action_New)
menu_File.addAction(self.action_Open)
menu_File.addAction(self.actionSave_As)
menu_File.addAction(self.action_Save)
menu_File.addSeparator()
menu_File.addAction(self.action_Quit)
self.menu_Solve.addAction(self.action_Solve)
self.menu_Help.addAction(self.action_About)
self.menu_Help.addAction(self.action_CCPL)
self.menu_Help.addAction(self.action_Help)
menubar.addAction(menu_File.menuAction())
menubar.addAction(self.menu_Solve.menuAction())
menubar.addAction(self.menu_Help.menuAction())
self.setWindowTitle("Main Window")
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab),\
"Data Page")
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_2),\
"Solution Page")
menu_File.setTitle("&File")
self.menu_Solve.setTitle("&Solve")
self.menu_Help.setTitle("&Help")
self.tabWidget.setCurrentIndex(0)
self.action_New.setText("&New")
self.action_Open.setText("&Open")
self.actionSave_As.setText("Save &As")
self.action_Save.setText("&Save")
self.action_Quit.setText("&Quit")
self.action_Solve.setText("&Solve")
self.action_About.setText("&About")
self.action_CCPL.setText("&CCPL")
self.action_Help.setText("&Help")
self.action_Quit.triggered.connect(self.close)
allToolBar = self.addToolBar("AllToolBar")
allToolBar.setObjectName("AllToolBar")
self.addActions(allToolBar, (self.action_Open, self.actionSave_As,\
self.action_Save, self.action_Solve,\
self.action_Quit ))
self.action_New.triggered.connect(self.fileNew)
self.action_Open.triggered.connect(self.fileOpen)
self.actionSave_As.triggered.connect(self.fileSaveAs)
self.action_Save.triggered.connect(self.fileSave)
self.action_Solve.triggered.connect(self.trussSolve)
self.action_About.triggered.connect(self.aboutBox)
self.action_CCPL.triggered.connect(self.displayCCPL)
self.action_Help.triggered.connect(self.help)
self.plainTextEdit.textChanged.connect(self.setDirty)
self.action_New = self.editAction(self.action_New, None,\
'ctrl+N', 'filenew', 'New File.')
self.action_Open = self.editAction(self.action_Open, None,
'ctrl+O', 'fileopen', 'Open File.')
self.actionSave_As = self.editAction(self.actionSave_As,\
None, 'ctrl+A', 'filesaveas',\
'Save and Name File.')
self.action_Save = self.editAction(self.action_Save, None,
'ctrl+S', 'filesave', 'Save File.')
self.action_Solve = self.editAction(self.action_Solve, None,
'ctrl+L', 'solve', 'Solve Structure.')
self.action_About = self.editAction(self.action_About, None,
'ctrl+B', 'about','Pop About Box.')
self.action_CCPL = self.editAction(self.action_CCPL, None,
'ctrl+G', 'licence', 'Show Licence')
self.action_Help = self.editAction(self.action_Help, None,
'ctrl+H', 'help', 'Show Help Page.')
self.action_Quit = self.editAction(self.action_Quit, None,
'ctrl+Q', 'quit', 'Quit the program.')
self.plainTextEdit_2.setReadOnly(True)
def setDirty(self):
'''On change of text in textEdit window, set the flag
"dirty" to True'''
index = self.tabWidget.currentIndex()
if index is not 0:
return
if self.dirty:
return True
self.dirty = True
self.updateStatus('self.dirty set to True')
def clearDirty(self):
'Clear dirty flag'
self.dirty = False
def fileNew(self):
'''Clear both Data Page and Solution Page.'''
self.plainTextEdit.setPlainText(' ')
self.plainTextEdit_2.setPlainText(' ')
self.clearDirty(self)
def okToContinue(self):
if self.dirty:
reply = QMessageBox.question(self,
"Data Loader - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes|QMessageBox.No|QMessageBox.Cancel)
if reply == QMessageBox.Cancel:
return False
elif reply == QMessageBox.Yes:
self.clearDirty()
return self.fileSave()
return True
def okRead(self):
'Pop-up a warning message.'
reply = QMessageBox.warning(self,
"Warning",
'''\nFile Open and Save is possible only in Data Page!
\n\(Use SaveAs for Solution Page)''', QMessageBox.Ok)
return True
def fileOpen(self):
'''Open a file in Data Page (with index == 0)'''
if self.tabWidget.currentIndex():
self.okRead()
return
if not self.okToContinue():
return
dir = (os.path.dirname(self.filename)
if self.filename is not None else ".")
self.filetuple = QFileDialog.getOpenFileName(self,\
"Open File", dir, \
"Data (*.dat *.txt)\nAll Files (*.*)")
self.filename = self.filetuple[0]
fname = self.filename
# QFileDialog returns a tuple x with x[0] = file name and
# x[1] = type of filter.
if fname:
self.loadFile(fname)
self.filename = fname
self.updateStatus('New file opened.')
def loadFile(self, fname=None):
fl = open(fname)
text = fl.read()
self.plainTextEdit.setPlainText(text)
self.dirty = False
def fileSave(self):
'''Save file with current file name.'''
if self.tabWidget.currentIndex():
self.okRead()
return
if self.filename is None:
return self.fileSaveAs()
else:
flname = self.filename
if flname:
with open(flname, 'w') as fl:
fl.write(tempText)
self.dirty = False
self.updateStatus('File saved.')
return True
else:
self.updateStatus('Failed to save... ')
return False
self.filename = None
self.dirty = False
def fileSaveAs(self):
'''Save file with a new name.'''
qpr = self.qprintline
fname = self.filename if self.filename is not None else\
"NoName"
self.filetuple = QFileDialog.getSaveFileName(self,
"Truss program - Save File", fname, "Data File (*.*)")
flname = self.filetuple[0]
index = self.tabWidget.currentIndex()
if index == 0:
self.filename = flname
if flname:
fl = open(flname, 'w')
tempText = self.plainTextEdit.toPlainText()
fl.write(tempText)
fl.close()
self.dirty = False
self.updateStatus('File saved.')
elif index == 1:
if flname:
fl = open(flname, 'w')
tempText = self.plainTextEdit_2.toPlainText()
fl.write(tempText)
fl.close()
def trussSolve(self):
'''Solve a statically determinate truss, specified in
Data Page and display the results in the Solution Page.
To start, make a copy of the Data Page with a header all
shown on the Data Page.'''
printline = self.qprintline
dataBall = self.plainTextEdit.toPlainText()
self.plainTextEdit_2.clear()
printline('================================')
flbase = os.path.basename(self.filename)
printline('SOLUTION FOR ' + flbase)
printline('================================')
dataBall = self.plainTextEdit.toPlainText()
ncrunch.main(printline, self.filename, dataBall)
def aboutBox(self):
'''Popup a box with about message.'''
QMessageBox.about(self, "About PySide, Platform and the like",
"""<b>Part of Structural Analysis.</b> v %s
<p>Copyright © 2011 Algis Kabaila.
All rights reserved in accordance with
Creative Commons Attribution Licence (CCPL) v3
or later - NO WARRANTIES!
<p>This progam finds bar forces in
statically determinate trusses.
<p>Python %s - PySide version %s - Qt version %s on\
%s""" % (__version__, platform.python_version(),\
PySide.__version__, PySide.QtCore.__version__,
platform.system()))
def displayCCPL(self):
'''Read and display CCPL licence.'''
self.plainTextEdit.setPlainText(open('CCPL.txt').read())
self.dirty = False
self.filename = 'COPYING.txt'
self.updateStatus('CCPL displayed.')
def help(self):
'''Read and display a help file- currently the README.txt.'''
self.plainTextEdit.setPlainText(open('README.md').read())
self.dirty = False
self.filename = 'README.txt'
self.updateStatus('README displayed.')
def addActions(self, target, actions):
'''Actions are added to Tool Bar.'''
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def editAction(self, action, slot=None, shortcut=None, icon=None,
tip=None):
'''This method adds to action: icon, shortcut, ToolTip,\
StatusTip and can connect triggered action to slot '''
if icon is not None:
action.setIcon(QIcon(":/%s.png" % (icon)))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
action.triggered.connect(slot)
return action
def qreadline(self, lineNo):
'''Read one line from Data Page (lineNo starts with 0)'''
return unicode(self.plainTextEdit.document().\
findBlockByLineNumber(lineNo).text()).rstrip()
def qprintline(self, line):
'''Append one line to Solution Page.'''
self.plainTextEdit_2.appendPlainText(line.rstrip())
def updateStatus(self, message):
'''Keep status current.'''
if self.filename is not None:
flbase = os.path.basename(self.filename)
self.setWindowTitle(unicode("Truss Analysis - " +\
flbase + "[*]") )
self.statusBar().showMessage(message, 5000)
self.setWindowModified(self.dirty)
