1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896 |
- #!/usr/bin/env python3
- """
- # TOP2049 Open Source programming suite
- #
- # Qt-based graphical user interface
- #
- # Copyright (c) 2010-2023 Michael Buesch <m@bues.ch>
- #
- # This program is free software; you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation; either version 2 of the License, or
- # (at your option) any later version.
- #
- # This program is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License along
- # with this program; if not, write to the Free Software Foundation, Inc.,
- # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- """
- from libtoprammer.main import *
- from libtoprammer.util import *
- import sys
- import time
- import html
- import configparser
- try:
- from PyQt6.QtCore import *
- from PyQt6.QtGui import *
- from PyQt6.QtWidgets import *
- Signal = pyqtSignal
- except ImportError as e:
- print("Failed to import PyQt6 modules: %s" % str(e))
- print("Please install PyQt6. On Debian Linux run: apt install python3-pyqt6")
- sys.exit(1)
- EVENT_HWTHREAD = QEvent.Type(QEvent.Type.User + 0)
- def stringRemoveChars(string, chars):
- ret = []
- for c in string:
- if c not in chars:
- ret.append(c)
- return "".join(ret)
- def htmlEscape(plaintext):
- return html.escape(plaintext, True)
- def getIconPath(name):
- return pkg_resources.resource_filename("libtoprammer",
- "icons/" + name + ".png")
- def getIcon(name):
- return QIcon(getIconPath(name))
- class Wrapper(object):
- def __init__(self, obj):
- self.obj = obj
- def __eq__(self, other):
- return self.obj == other.obj
- def __ne__(self, other):
- return self.obj != other.obj
- class ZifPinButton(QWidget):
- TEXT_RIGHT = 0
- TEXT_LEFT = 1
- stateChanged = Signal(bool)
- class Label(QLabel):
- clicked = Signal()
- def __init__(self, text, parent=None):
- QLabel.__init__(self, text, parent)
- def mousePressEvent(self, event):
- self.clicked.emit()
- def __init__(self, text, textPos=TEXT_RIGHT, parent=None):
- QWidget.__init__(self, parent)
- self.setLayout(QHBoxLayout(self))
- self.layout().setContentsMargins(QMargins())
- self.checkbox = QCheckBox(self)
- self.checkbox.stateChanged.connect(self.__cbStateChanged)
- self.label = self.Label(text, self)
- self.label.clicked.connect(self.toggle)
- if textPos == self.TEXT_RIGHT:
- self.label.setAlignment(Qt.AlignmentFlag.AlignLeft)
- self.layout().addWidget(self.checkbox)
- self.layout().addWidget(self.label)
- else:
- self.label.setAlignment(Qt.AlignmentFlag.AlignRight)
- self.layout().addWidget(self.label)
- self.layout().addWidget(self.checkbox)
- def __cbStateChanged(self, newState):
- self.stateChanged.emit(newState == Qt.CheckState.Checked)
- def state(self):
- return self.checkbox.checkState() == Qt.CheckState.Checked
- def setState(self, en):
- self.checkbox.setCheckState(Qt.CheckState.Checked if en else Qt.CheckState.Unchecked)
- def toggle(self):
- self.setState(not self.state())
- class ZifWidget(QGroupBox):
- def __init__(self, unitest, nrZifPins):
- QGroupBox.__init__(self, "ZIF socket", unitest)
- self.unitest = unitest
- self.setLayout(QGridLayout())
- self.nrPins = nrZifPins
- assert(self.nrPins % 2 == 0)
- self.blockedPins = []
- self.ignoreOutChange = False
- label = QLabel("ZIF\nsocket", self)
- label.setFrameShape(QFrame.Shape.Panel)
- label.setFrameShadow(QFrame.Shadow.Sunken)
- label.setAlignment(Qt.AlignmentFlag.AlignHCenter | Qt.AlignmentFlag.AlignVCenter)
- label.setMinimumWidth(80)
- self.layout().addWidget(label, 0, 2, self.nrPins // 2, 1)
- self.pins = [ None ] * self.nrPins
- self.pinOuten = [ None ] * self.nrPins
- for i in range(0, self.nrPins // 2):
- left = i + 1
- right = self.nrPins - i
- self.pins[left - 1] = ZifPinButton(str(left),
- ZifPinButton.TEXT_LEFT, self)
- self.pins[left - 1].stateChanged.connect(self.__outChanged)
- self.pinOuten[left - 1] = ZifPinButton("out",
- ZifPinButton.TEXT_LEFT, self)
- self.pinOuten[left - 1].stateChanged.connect(self.__outEnChanged)
- self.layout().addWidget(self.pinOuten[left - 1], left - 1, 0)
- self.layout().addWidget(self.pins[left - 1], left - 1, 1)
- self.pins[right - 1] = ZifPinButton(str(right),
- ZifPinButton.TEXT_RIGHT, self)
- self.pins[right - 1].stateChanged.connect(self.__outChanged)
- self.pinOuten[right - 1] = ZifPinButton("out",
- ZifPinButton.TEXT_RIGHT, self)
- self.pinOuten[right - 1].stateChanged.connect(self.__outEnChanged)
- self.layout().addWidget(self.pins[right - 1], self.nrPins - right, 3)
- self.layout().addWidget(self.pinOuten[right - 1], self.nrPins - right, 4)
- self.__outEnChanged()
- self.__outChanged()
- def readInputs(self):
- try:
- inputMask = self.unitest.queryTop("top.getChip().getInputs()")
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to fetch input states:\n" +\
- str(e))
- return False
- for i in range(0, self.nrPins):
- if not self.pinOuten[i].state() and\
- i + 1 not in self.blockedPins:
- state = False
- if inputMask & bit(i):
- state = True
- self.ignoreOutChange = True
- self.pins[i].setState(state)
- self.ignoreOutChange = False
- return True
- def __updateInOutStates(self):
- for i in range(0, self.nrPins):
- if i + 1 in self.blockedPins:
- self.pins[i].setEnabled(False)
- self.pins[i].setState(False)
- self.pinOuten[i].setEnabled(False)
- self.pinOuten[i].setState(False)
- else:
- self.pinOuten[i].setEnabled(True)
- if self.pinOuten[i].state():
- self.pins[i].setEnabled(True)
- else:
- self.pins[i].setEnabled(False)
- def __outEnChanged(self, unused=False):
- self.__updateInOutStates()
- outEnMask = self.getOutEnMask()
- try:
- self.unitest.queryTop("top.getChip().setOutputEnableMask(...)",
- outEnMask)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to set output-enable states:\n" +\
- str(e))
- return
- self.readInputs()
- def __outChanged(self, unused=False):
- if self.ignoreOutChange:
- return
- outMask = self.getOutMask()
- try:
- self.unitest.queryTop("top.getChip().setOutputs(...)",
- outMask)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to set output states:\n" +\
- str(e))
- return
- self.readInputs()
- def setBlockedPins(self, blockedPins):
- self.blockedPins = blockedPins
- self.__updateInOutStates()
- self.readInputs()
- def getOutEnMask(self):
- outEnMask = 0
- for i in range(0, self.nrPins):
- if self.pinOuten[i].state():
- outEnMask |= bit(i)
- return outEnMask
- def setOutEnMask(self, mask):
- for i in range(0, self.nrPins):
- if mask & bit(i):
- self.pinOuten[i].setState(True)
- else:
- self.pinOuten[i].setState(False)
- mask &= ~bit(i)
- if mask:
- raise TOPException("ZIF out-en mask has too many bits set")
- def getOutMask(self):
- outMask = 0
- for i in range(0, self.nrPins):
- if self.pins[i].state():
- outMask |= bit(i)
- return outMask
- def setOutMask(self, mask):
- for i in range(0, self.nrPins):
- if mask & bit(i):
- self.pins[i].setState(True)
- else:
- self.pins[i].setState(False)
- mask &= ~bit(i)
- if mask:
- raise TOPException("ZIF out mask has too many bits set")
- class UnitestDialog(QDialog):
- def __init__(self, mainWindow):
- QDialog.__init__(self, mainWindow)
- self.setWindowTitle("Universal logic tester")
- self.mainWindow = mainWindow
- self.setLayout(QGridLayout())
- self.inputPollBlocked = 0
- self.uiChangeBlocked = 0
- # Initialize the unitest chip
- (failed, returnValue) = self.mainWindow.runOperationSync(
- HwThread.TASK_INITCHIP,
- "unitest")
- if failed:
- raise TOPException("Failed to load 'unitest' chip: %s" % str(returnValue))
- self.queryTop("top.getChip().reset()")
- # Query the hardware layer for common parameters
- self.param_topType = self.queryTop("top.getProgrammerType()")
- self.param_gndLayouts = self.queryTop("top.gnd.supportedLayouts()")
- self.param_nrZifPins = self.queryTop("top.gnd.getNrOfPins()")
- self.param_vccLayouts = self.queryTop("top.vcc.supportedLayouts()")
- self.param_minVccVolt = self.queryTop("top.vcc.minVoltage()")
- self.param_maxVccVolt = self.queryTop("top.vcc.maxVoltage()")
- self.param_vppLayouts = self.queryTop("top.vpp.supportedLayouts()")
- self.param_minVppVolt = self.queryTop("top.vpp.minVoltage()")
- self.param_maxVppVolt = self.queryTop("top.vpp.maxVoltage()")
- self.param_oscFreq = self.queryTop("top.getOscillatorHz()")
- assert(self.param_nrZifPins % 2 == 0)
- self.param_vppLayouts.sort(key=lambda layId_layMask: layId_layMask[1])
- self.menuBar = QMenuBar(self)
- self.menuBar.addAction("&Load settings...", self.loadSettings)
- self.menuBar.addAction("&Save settings...", self.saveSettings)
- self.menuBar.addAction("&Raw command...", self.mainWindow.sendRawCommand)
- self.layout().addWidget(self.menuBar, 0, 0, 1, 3)
- self.zifWidget = ZifWidget(self, self.param_nrZifPins)
- self.layout().addWidget(self.zifWidget, 1, 0, 10, 1)
- group = QGroupBox("GND layout", self)
- group.setLayout(QGridLayout())
- self.gndLayout = QComboBox(self)
- self.gndLayout.addItem("Not connected", 0)
- for (layId, layMask) in self.param_gndLayouts:
- if not layMask:
- continue
- descr = "GND on pin "
- for i in range(0, self.param_nrZifPins):
- if layMask & bit(i):
- descr += str(i + 1) + " "
- self.gndLayout.addItem(descr, layId)
- group.layout().addWidget(self.gndLayout, 0, 0)
- self.layout().addWidget(group, 1, 1)
- group = QGroupBox("VCC layout", self)
- group.setLayout(QGridLayout())
- self.vccVoltage = QDoubleSpinBox(self)
- self.vccVoltage.setSuffix(" V")
- self.vccVoltage.setMinimum(self.param_minVccVolt)
- self.vccVoltage.setMaximum(self.param_maxVccVolt)
- self.vccVoltage.setSingleStep(0.1)
- group.layout().addWidget(self.vccVoltage, 0, 0)
- self.vccLayout = QComboBox(self)
- self.vccLayout.addItem("Not connected", 0)
- for (layId, layMask) in self.param_vccLayouts:
- if not layMask:
- continue
- descr = "VCC on pin "
- for i in range(0, self.param_nrZifPins):
- if layMask & bit(i):
- descr += str(i + 1) + " "
- self.vccLayout.addItem(descr, layId)
- group.layout().addWidget(self.vccLayout, 1, 0)
- self.layout().addWidget(group, 2, 1)
- group = QGroupBox("Input polling", self)
- group.setLayout(QGridLayout())
- self.inputPollEn = QCheckBox("Enabled", self)
- self.inputPollEn.setCheckState(Qt.CheckState.Checked)
- group.layout().addWidget(self.inputPollEn, 0, 0)
- self.inputPollInterval = QDoubleSpinBox(self)
- self.inputPollInterval.setPrefix("Interval ")
- self.inputPollInterval.setSuffix(" seconds")
- self.inputPollInterval.setMinimum(0.25)
- self.inputPollInterval.setSingleStep(0.25)
- self.inputPollInterval.setValue(1.0)
- group.layout().addWidget(self.inputPollInterval, 1, 0)
- self.layout().addWidget(group, 3, 1)
- group = QGroupBox("Frequency counter", self)
- group.setLayout(QGridLayout())
- self.fcntEn = QCheckBox("Enabled", self)
- self.fcntEn.setCheckState(Qt.CheckState.Unchecked)
- group.layout().addWidget(self.fcntEn, 0, 0)
- self.fcntPosEdge = QCheckBox("Positive edge", self)
- self.fcntPosEdge.setCheckState(Qt.CheckState.Checked)
- group.layout().addWidget(self.fcntPosEdge, 1, 0)
- self.fcntPin = QComboBox(self)
- group.layout().addWidget(self.fcntPin, 2, 0)
- self.fcntValueLabel = QLabel(self)
- group.layout().addWidget(self.fcntValueLabel, 3, 0)
- self.fcntValue = 0
- self.layout().addWidget(group, 4, 1)
- group = QGroupBox("Oscillator", self)
- group.setLayout(QGridLayout())
- self.oscPin = QComboBox(self)
- self.oscPin.addItem("Disabled", Wrapper(0))
- for i in range(0, self.param_nrZifPins):
- self.oscPin.addItem("On pin %d" % (i + 1),
- Wrapper(bit(i)))
- group.layout().addWidget(self.oscPin, 0, 0)
- self.oscDiv = QSpinBox(self)
- self.oscDiv.setPrefix("Divider ")
- self.oscDiv.setSingleStep(1)
- self.oscDiv.setMinimum(1)
- self.oscDiv.setMaximum(self.param_oscFreq)
- self.oscDiv.setValue(self.param_oscFreq // 1000)
- group.layout().addWidget(self.oscDiv, 1, 0)
- self.oscFreq = QLabel(self)
- group.layout().addWidget(self.oscFreq, 2, 0)
- self.layout().addWidget(group, 5, 1)
- group = QGroupBox("VPP layout", self)
- group.setLayout(QGridLayout())
- self.vppVoltage = QDoubleSpinBox(self)
- self.vppVoltage.setSuffix(" V")
- self.vppVoltage.setMinimum(self.param_minVppVolt)
- self.vppVoltage.setMaximum(self.param_maxVppVolt)
- self.vppVoltage.setSingleStep(0.1)
- group.layout().addWidget(self.vppVoltage, 0, 0, 1, 2)
- self.vppLayouts = {}
- xOffset = 0
- yOffset = 0
- for (layId, layMask) in self.param_vppLayouts:
- if not layMask:
- continue
- for i in range(0, self.param_nrZifPins):
- if layMask & bit(i):
- descr = str(i + 1) + " "
- self.vppLayouts[layId] = QCheckBox(descr, self)
- self.vppLayouts[layId].stateChanged.connect(
- self.vppLayoutChanged)
- group.layout().addWidget(self.vppLayouts[layId],
- yOffset + 1, xOffset)
- yOffset += 1
- if yOffset == len(self.param_vppLayouts) // 2:
- yOffset = 0
- xOffset += 1
- self.layout().addWidget(group, 1, 2, 8, 1)
- self.inputPollTimer = QTimer()
- self.inputPollTimer.setSingleShot(True)
- self.inputPollTimer.timeout.connect(self.doInputPollTimer)
- self.inputPollEn.stateChanged.connect(self.inputPollChanged)
- self.inputPollInterval.valueChanged.connect(self.inputPollChanged)
- self.gndLayout.currentIndexChanged.connect(self.gndLayoutChanged)
- self.vccVoltage.valueChanged.connect(self.vccLayoutChanged)
- self.vccLayout.currentIndexChanged.connect(self.vccLayoutChanged)
- self.vppVoltage.valueChanged.connect(self.vppLayoutChanged)
- self.fcntEn.stateChanged.connect(self.fcntChanged)
- self.fcntPosEdge.stateChanged.connect(self.fcntChanged)
- self.fcntPin.currentIndexChanged.connect(self.fcntChanged)
- self.oscPin.currentIndexChanged.connect(self.oscChanged)
- self.oscDiv.valueChanged.connect(self.oscChanged)
- self.__modeChanged()
- self.gndLayoutChanged()
- self.vccLayoutChanged()
- self.inputPollChanged()
- self.oscChanged()
- def __readFcntValue(self):
- try:
- self.fcntValue = self.queryTop("top.getChip().getFreqCount()")
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to read frequency counter value:\n" +\
- str(e))
- return False
- return True
- def __updateFcntValueLabel(self):
- if self.getMode() != Chip_Unitest.MODE_FCNT:
- self.fcntValueLabel.clear()
- return
- if self.fcntValue:
- hz = "%.3f" % (float(self.param_oscFreq) / self.fcntValue)
- else:
- hz = "~~~"
- self.fcntValueLabel.setText("Measured: %s Hz" % hz)
- def __modeChanged(self):
- mode = self.getMode()
- activePinMask = self.queryTop("top.getChip().getActivePinMask()")
- self.uiChangeBlocked += 1
- # Update freq count UI
- self.fcntValue = 0
- self.fcntPin.clear()
- self.fcntPin.setEnabled(False)
- self.fcntPosEdge.setEnabled(False)
- if mode == Chip_Unitest.MODE_FCNT:
- for i in range(0, self.param_nrZifPins):
- if bit(i) & activePinMask == 0:
- continue
- self.fcntPin.addItem("On pin %d" % (i + 1), i)
- self.fcntPin.setEnabled(True)
- self.fcntPosEdge.setEnabled(True)
- self.__updateFcntValueLabel()
- # Update the ZIF widget
- self.updateZifCheckboxes()
- self.uiChangeBlocked -= 1
- def switchMode(self, mode):
- if self.getMode() == mode:
- return
- self.queryTop("top.getChip().setMode(...)", mode)
- self.__modeChanged()
- def __loadFile(self, filename):
- try:
- p = configparser.ConfigParser()
- p.read((filename,))
- sect = "TOPRAMMER-UNITEST-SETTINGS"
- if not p.has_section(sect):
- raise TOPException("Invalid file format")
- ver = p.getint(sect, "fileVersion")
- verExpected = 1
- if ver != verExpected:
- raise TOPException("Unsupported file version (got %d, expected %d)" %\
- (ver, verExpected))
- pType = p.get(sect, "programmerType")
- if pType.upper() != self.param_topType.upper():
- raise TOPException("Programmer type mismatch (file = %s, connected = %s)" %\
- (pType, self.param_topType))
- layout = p.getint(sect, "gndLayout")
- idx = self.gndLayout.findData(layout)
- if idx < 0:
- raise TOPException("Invalid GND layout")
- self.gndLayout.setCurrentIndex(idx)
- layout = p.getint(sect, "vccLayout")
- idx = self.vccLayout.findData(layout)
- if idx < 0:
- raise TOPException("Invalid VCC layout")
- self.vccLayout.setCurrentIndex(idx)
- voltage = p.getfloat(sect, "vccVoltage")
- self.vccVoltage.setValue(voltage)
- layouts = p.get(sect, "vppLayout").split(",")
- for layId in list(self.vppLayouts.keys()):
- if layId in layouts:
- self.vppLayouts[layId].setCheckState(Qt.CheckState.Checked)
- else:
- self.vppLayouts[layId].setCheckState(Qt.CheckState.Unchecked)
- voltage = p.getfloat(sect, "vppVoltage")
- self.vppVoltage.setValue(voltage)
- interval = p.getfloat(sect, "inputPollInterval")
- self.inputPollInterval.setValue(interval)
- enabled = p.getboolean(sect, "inputPollEnabled")
- if enabled:
- self.inputPollEn.setCheckState(Qt.CheckState.Checked)
- else:
- self.inputPollEn.setCheckState(Qt.CheckState.Unchecked)
- div = p.getint(sect, "oscillatorDiv")
- self.oscDiv.setValue(div)
- mask = int(p.get(sect, "oscillatorMask"), 16)
- for i in range(0, self.oscPin.count()):
- if self.oscPin.itemData(i).obj == mask:
- break
- else:
- raise TOPException("Invalid oscillator mask")
- self.oscPin.setCurrentIndex(i)
- mask = int(p.get(sect, "zifOutEnMask"), 16)
- self.zifWidget.setOutEnMask(mask)
- mask = int(p.get(sect, "zifOutMask"), 16)
- self.zifWidget.setOutMask(mask)
- enabled = p.getboolean(sect, "fcntEnabled")
- if enabled:
- self.fcntEn.setCheckState(Qt.CheckState.Checked)
- else:
- self.fcntEn.setCheckState(Qt.CheckState.Unchecked)
- pos = p.getboolean(sect, "fcntPosEdge")
- if pos:
- self.fcntPosEdge.setCheckState(Qt.CheckState.Checked)
- else:
- self.fcntPosEdge.setCheckState(Qt.CheckState.Unchecked)
- pin = p.getint(sect, "fcntPin")
- if pin >= 0:
- idx = self.fcntPin.findData(pin)
- if idx < 0:
- raise TOPException("Invalid fcntPin")
- self.fcntPin.setCurrentIndex(idx)
- except (configparser.Error, TOPException, ValueError) as e:
- QMessageBox.critical(self, "Failed to load settings",
- "Failed to load settings: %s" % str(e))
- def loadSettings(self):
- (fn, selFltr) = QFileDialog.getOpenFileName(
- self, "Load settings", "",
- "Toprammer-unitest settings (*.tus);;"
- "All files (*)")
- if not fn:
- return
- self.__loadFile(fn)
- def __saveFile(self, filename):
- try:
- fd = open(filename, "w", encoding="UTF-8")
- fd.write("[TOPRAMMER-UNITEST-SETTINGS]\r\n")
- fd.write("fileVersion=%d\r\n" % 1)
- fd.write("programmerType=%s\r\n" % self.param_topType)
- idx = self.gndLayout.currentIndex()
- fd.write("gndLayout=%d\r\n" % self.gndLayout.itemData(idx))
- idx = self.vccLayout.currentIndex()
- fd.write("vccLayout=%d\r\n" % self.vccLayout.itemData(idx))
- fd.write("vccVoltage=%f\r\n" % self.vccVoltage.value())
- vppLayouts = ""
- for layId in list(self.vppLayouts.keys()):
- if self.vppLayouts[layId].checkState() == Qt.CheckState.Checked:
- if vppLayouts:
- vppLayouts += ","
- vppLayouts += str(layId)
- if not vppLayouts:
- vppLayouts = "0"
- fd.write("vppLayout=%s\r\n" % vppLayouts)
- fd.write("vppVoltage=%f\r\n" % self.vppVoltage.value())
- fd.write("inputPollEnabled=%d\r\n" % int(self.inputPollEn.checkState() == Qt.CheckState.Checked))
- fd.write("inputPollInterval=%f\r\n" % self.inputPollInterval.value())
- fd.write("fcntEnabled=%d\r\n" % (int(self.fcntEn.checkState() == Qt.CheckState.Checked)))
- fd.write("fcntPosEdge=%d\r\n" % (int(self.fcntPosEdge.checkState() == Qt.CheckState.Checked)))
- idx = self.fcntPin.currentIndex()
- pin = -1
- if idx >= 0:
- pin = self.fcntPin.itemData(idx)
- fd.write("fcntPin=%d\r\n" % pin)
- idx = self.oscPin.currentIndex()
- fd.write("oscillatorMask=%X\r\n" % self.oscPin.itemData(idx).obj)
- fd.write("oscillatorDiv=%d\r\n" % self.oscDiv.value())
- fd.write("zifOutEnMask=%X\r\n" % self.zifWidget.getOutEnMask())
- fd.write("zifOutMask=%X\r\n" % self.zifWidget.getOutMask())
- except (IOError, UnicodeError) as e:
- QMessageBox.critical(self, "Failed to save settings",
- "Failed to write settings to file: %s" % str(e))
- def saveSettings(self):
- (fn, selFltr) = QFileDialog.getSaveFileName(
- self, "Save settings", "",
- "Toprammer-unitest settings (*.tus)")
- if not fn:
- return
- if not fn.endswith(".tus"):
- fn += ".tus"
- self.__saveFile(fn)
- def closeEvent(self, e):
- self.inputPollTimer.stop()
- def queryTop(self, funcname, *parameters):
- self.inputPollBlocked += 1 # Avoid recursion
- (failed, returnValue) = self.mainWindow.runOperationSync(
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall(funcname, *parameters))
- self.inputPollBlocked -= 1
- if failed:
- raise TOPException("Failed to query TOP %s\n%s" % (funcname, str(returnValue)))
- return returnValue
- def getMode(self):
- return self.queryTop("top.getChip().getMode()")
- def shutdown(self):
- self.inputPollTimer.stop()
- def updateZifCheckboxes(self):
- mode = self.getMode()
- blockedPins = []
- # Basic mask
- activePinMask = self.queryTop("top.getChip().getActivePinMask()")
- for i in range(0, self.zifWidget.nrPins):
- if activePinMask & bit(i) == 0:
- blockedPins.append(i + 1)
- # GND
- idx = self.gndLayout.currentIndex()
- lay = self.gndLayout.itemData(idx)
- blockedPins.extend(self.queryTop("top.gnd.ID2pinlist(...)", lay))
- # VCC
- idx = self.vccLayout.currentIndex()
- lay = self.vccLayout.itemData(idx)
- blockedPins.extend(self.queryTop("top.vcc.ID2pinlist(...)", lay))
- # VPP
- for key in list(self.vppLayouts.keys()):
- if self.vppLayouts[key].checkState() == Qt.CheckState.Checked:
- blockedPins.extend(self.queryTop("top.vpp.ID2pinlist(...)", key))
- # OSC
- idx = self.oscPin.currentIndex()
- mask = self.oscPin.itemData(idx).obj
- for i in range(0, self.zifWidget.nrPins):
- if mask & bit(i):
- blockedPins.append(i + 1)
- # Freq counter
- if mode == Chip_Unitest.MODE_FCNT:
- idx = self.fcntPin.currentIndex()
- pinNumber = self.fcntPin.itemData(idx)
- blockedPins.append(pinNumber + 1)
- # Set blocked pins in ZIF widget
- self.zifWidget.setBlockedPins(tuple(set(blockedPins)))
- def gndLayoutChanged(self, unused=None):
- if self.uiChangeBlocked:
- return
- idx = self.gndLayout.currentIndex()
- selLayout = self.gndLayout.itemData(idx)
- try:
- self.queryTop("top.getChip().setGND(...)", selLayout)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to set GND layout:\n" +\
- str(e))
- return
- self.updateZifCheckboxes()
- def vccLayoutChanged(self, unused=None):
- if self.uiChangeBlocked:
- return
- selVoltage = self.vccVoltage.value()
- idx = self.vccLayout.currentIndex()
- selLayout = self.vccLayout.itemData(idx)
- try:
- self.queryTop("top.getChip().setVCC(...)", selVoltage, selLayout)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to set VCC layout:\n" +\
- str(e))
- return
- self.updateZifCheckboxes()
- def vppLayoutChanged(self, unused=None):
- if self.uiChangeBlocked:
- return
- selVoltage = self.vppVoltage.value()
- selLayouts = []
- for key in list(self.vppLayouts.keys()):
- if self.vppLayouts[key].checkState() == Qt.CheckState.Checked:
- selLayouts.append(key)
- try:
- self.queryTop("top.getChip().setVPP(...)", selVoltage, selLayouts)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed to set VPP layout:\n" +\
- str(e))
- return
- self.updateZifCheckboxes()
- def doInputPollTimer(self):
- if self.inputPollBlocked:
- # Blocked. Reschedule.
- self.inputPollTimer.start(1)
- return
- ok = self.zifWidget.readInputs()
- if ok:
- if self.fcntEn.checkState() == Qt.CheckState.Checked:
- ok = self.__readFcntValue()
- if ok:
- self.__updateFcntValueLabel()
- if not ok:
- # Whoops, error. Disable input polling
- self.inputPollEn.setCheckState(Qt.CheckState.Unchecked)
- return False
- # Reschedule
- inter = int(self.inputPollInterval.value() * 1000)
- self.inputPollTimer.start(inter)
- return True
- def inputPollChanged(self, unused=None):
- if self.uiChangeBlocked:
- return
- if self.inputPollEn.checkState() == Qt.CheckState.Checked:
- self.inputPollInterval.setEnabled(True)
- if self.doInputPollTimer():
- inter = int(self.inputPollInterval.value() * 1000)
- self.inputPollTimer.start(inter)
- else:
- self.inputPollInterval.setEnabled(False)
- self.inputPollTimer.stop()
- def fcntChanged(self, unused=None):
- if self.uiChangeBlocked:
- return
- try:
- if self.fcntEn.checkState() == Qt.CheckState.Checked:
- self.switchMode(Chip_Unitest.MODE_FCNT)
- idx = self.fcntPin.currentIndex()
- pinNumber = self.fcntPin.itemData(idx)
- inv = (self.fcntPosEdge.checkState() != Qt.CheckState.Checked)
- self.queryTop("top.getChip().setFreqCountPin(...)",
- pinNumber, inv)
- self.__readFcntValue()
- self.__updateFcntValueLabel()
- # Force enable input polling
- self.inputPollEn.setCheckState(Qt.CheckState.Checked)
- self.inputPollEn.setEnabled(False)
- else:
- self.switchMode(Chip_Unitest.MODE_UNITEST)
- self.__updateFcntValueLabel()
- self.inputPollEn.setEnabled(True)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed configure frequency counter:\n" +\
- str(e))
- return
- self.updateZifCheckboxes()
- def oscChanged(self, unused=None):
- if self.uiChangeBlocked:
- return
- try:
- div = self.oscDiv.value()
- self.queryTop("top.getChip().setOscDivider(...)", div)
- idx = self.oscPin.currentIndex()
- mask = self.oscPin.itemData(idx).obj
- self.queryTop("top.getChip().setOscMask(...)", mask)
- except (TOPException) as e:
- QMessageBox.critical(self, "TOP communication failed",
- "Failed configure oscillator:\n" +\
- str(e))
- return
- self.oscFreq.setText("%.03f Hz" % (float(self.param_oscFreq) / div))
- self.updateZifCheckboxes()
- class GuiUserInterface(AbstractUserInterface):
- # Global progress
- PROGRESSMETER_USER_GLOBAL = AbstractUserInterface.PROGRESSMETER_USER + 0
- def __init__(self, hwThread):
- self.hwThread = hwThread
- def progressMeterInit(self, meterId, message, nrSteps):
- self.hwThread.appendMessage("progrInit", (meterId, message, nrSteps))
- def progressMeterFinish(self, meterId):
- self.hwThread.appendMessage("progrFinish", meterId)
- def progressMeter(self, meterId, step):
- self.hwThread.appendMessage("progress", (meterId, step))
- def __consoleMessage(self, message):
- self.hwThread.appendMessage("console", message + "\n")
- def warningMessage(self, message):
- self.__consoleMessage("WARNING: " + message)
- def infoMessage(self, message):
- self.__consoleMessage(message)
- def debugMessage(self, message):
- self.__consoleMessage(message)
- class GenericTopCall(object):
- "One call to 'class Top'. Used by TASK_GENERICTOPCALL"
- def __init__(self, methodName, *parameters, **userData):
- self.methodName = methodName
- self.parameters = parameters
- self.result = None
- self.userData = userData
- class HwThread(QThread):
- TASK_SHUTDOWN = 0
- TASK_INITCHIP = 1
- TASK_GENERICTOPCALL = 2
- class CancelException(Exception):
- def __init__(self):
- Exception.__init__(self, "Operation cancelled")
- def __init__(self, mainWindow):
- QThread.__init__(self, mainWindow)
- self.mainWindow = mainWindow
- self.killRequest = False
- self.messageQueue = []
- self.top = None
- self.task = None
- self.opaqueId = None
- self.taskParameter = None
- self.cancel = False
- self.cancellationBlocked = 0
- self.waitCondition = QWaitCondition()
- self.mutex = QMutex()
- self.setTopParameters()
- self.setDevice()
- self.start()
- def killThread(self):
- if self.isRunning():
- self.mutex.lock()
- self.task = self.TASK_SHUTDOWN
- self.killRequest = True
- self.waitCondition.wakeAll()
- self.mutex.unlock()
- self.wait()
- def setDevice(self, devIdentifier=None):
- assert(self.top is None and self.task is None)
- self.param_devIdentifier = devIdentifier
- def setTopParameters(self, verbose=2, forceLevel=0,
- usebroken=True, forceBitfileUpload=False):
- assert(self.top is None and self.task is None)
- self.param_verbose = verbose
- self.param_forceLevel = forceLevel
- self.param_usebroken = usebroken
- self.param_forceBitfileUpload = forceBitfileUpload
- def triggerTask(self, task, taskParameter=None,
- opaqueId=None, mutexIsLocked=False):
- if not mutexIsLocked:
- self.mutex.lock()
- self.cancel = False
- self.task = task
- self.opaqueId = opaqueId
- self.taskParameter = taskParameter
- self.waitCondition.wakeAll()
- if not mutexIsLocked:
- self.mutex.unlock()
- def run(self):
- self.mutex.lock()
- while True:
- if not self.killRequest and self.task is None:
- self.waitCondition.wait(self.mutex)
- self.mutex.unlock()
- self.__taskWorker()
- self.mutex.lock()
- if self.killRequest and self.task is None:
- break
- self.mutex.unlock()
- def cancelTask(self):
- self.mutex.lock()
- self.cancel = True
- self.mutex.unlock()
- def __blockCancellation(self):
- self.cancellationBlocked += 1
- def __unblockCancellation(self):
- self.cancellationBlocked -= 1
- def __cancellationPoint(self):
- # Mutex must be locked
- if not self.cancel:
- return
- if self.cancellationBlocked:
- return
- self.cancel = False
- self.mutex.unlock()
- raise HwThread.CancelException() # Caught in __taskWorker
- def __doCancelTask(self):
- # Make sure the device is in a consistent state.
- self.__blockCancellation()
- if self.top:
- print("Operation cancelled. Resetting chip.")
- self.top.resetChip()
- self.__unblockCancellation()
- def appendMessage(self, message, data, nocancel=False):
- # Append a message to the message queue.
- # This is a cancellation point!
- self.mutex.lock()
- if not nocancel:
- self.__cancellationPoint()
- self.messageQueue.append( (message, data) )
- self.mutex.unlock()
- self.__notifyMainWindow()
- def __taskWorker(self):
- failed = True
- try:
- self.cancellationBlocked = 0
- result = self.__runTask(self.task)
- failed = False
- except (TOPException) as e:
- result = e
- except (HwThread.CancelException) as e:
- self.__doCancelTask()
- result = e
- except (Exception) as e:
- result = e
- self.appendMessage("finished", (self.opaqueId, failed, result),
- nocancel=True)
- self.task = None
- self.taskParameter = None
- self.opaqueId = None
- def __runTask(self, task):
- retval = None
- if task == self.TASK_SHUTDOWN or task == self.TASK_INITCHIP:
- if self.top:
- self.__blockCancellation()
- self.top.shutdownChip()
- self.top.shutdownProgrammer()
- self.top = None
- self.__unblockCancellation()
- if task == self.TASK_SHUTDOWN:
- return None
- if not self.top:
- # Initialize Hardware access
- self.__blockCancellation()
- self.top = TOP(devIdentifier = self.param_devIdentifier,
- verbose = self.param_verbose,
- forceLevel = self.param_forceLevel,
- usebroken = self.param_usebroken,
- forceBitfileUpload = self.param_forceBitfileUpload,
- userInterface = GuiUserInterface(self))
- self.__unblockCancellation()
- if task == self.TASK_INITCHIP:
- self.__blockCancellation()
- self.top.initializeChip(self.taskParameter)
- chip = self.top.getChip()
- asciiArtLayout = None
- layoutGen = chip.getLayoutGenerator()
- if layoutGen:
- asciiArtLayout = layoutGen.zifLayoutAsciiArt()
- self.__unblockCancellation()
- retval = (chip, asciiArtLayout)
- elif task == self.TASK_GENERICTOPCALL:
- # The parameter is one GenericTopCall
- # or a list of calls.
- topCallList = self.taskParameter
- if not isinstance(topCallList, list) and\
- not isinstance(topCallList, tuple):
- # It is only a single call. Make an iterable.
- topCallList = (topCallList, )
- self.__globalProgressInit(len(topCallList))
- # Iterate over the list of GenericTopCalls
- for topCallIndex, topCall in enumerate(topCallList):
- method = topCall.methodName.strip()
- if method.endswith(")"): # strip (...) suffix
- method = method[:method.rfind("(")]
- if method.startswith("top."): # strip 'top.' prefix
- method = method[4:]
- paramList = []
- for i in range(len(topCall.parameters)):
- paramList.append("topCall.parameters[%d]" % i)
- result = eval("self.top.%s(%s)" % (method, ", ".join(paramList)))
- topCall.result = result
- self.__globalProgress(topCallIndex + 1)
- # If there was only one call, return the result.
- # Otherwise return the complete call-list including results.
- if len(topCallList) == 1:
- retval = topCallList[0].result
- else:
- retval = topCallList
- else:
- raise TOPException("INTERNAL ERROR: HwThread: unknown task")
- return retval
- def __globalProgressInit(self, nrSteps):
- self.appendMessage("progrInit",
- (GuiUserInterface.PROGRESSMETER_USER_GLOBAL,
- None, nrSteps))
- def __globalProgress(self, step):
- self.appendMessage("progress",
- (GuiUserInterface.PROGRESSMETER_USER_GLOBAL,
- step))
- return step
- def __notifyMainWindow(self):
- QApplication.postEvent(self.mainWindow, QEvent(EVENT_HWTHREAD))
- def handleMessageQueue(self):
- self.mutex.lock()
- for (message, data) in self.messageQueue:
- if message == "finished":
- self.mainWindow.hardwareTaskFinished(
- opaqueId=data[0], failed=data[1], returnValue=data[2])
- elif message == "console":
- self.mainWindow.console.showMessage(data)
- elif message == "progrInit":
- (meterId, message, nrSteps) = data
- self.mainWindow.console.progressMeterInit(meterId, nrSteps)
- elif message == "progrFinish":
- meterId = data
- self.mainWindow.console.progressMeter(meterId, -1)
- elif message == "progress":
- (meterId, step) = data
- self.mainWindow.console.progressMeter(meterId, step)
- else:
- assert(0)
- self.messageQueue = []
- self.mutex.unlock()
- class Console(QDockWidget):
- STAT_OK = 0
- STAT_ERROR = 1
- STAT_PROGRESS = 2
- def __init__(self, mainWindow):
- QDockWidget.__init__(self, "Console", mainWindow)
- self.addPrefix = True
- self.statusUpdateBlocked = 0
- self.setFeatures(self.DockWidgetFeature.DockWidgetMovable |
- self.DockWidgetFeature.DockWidgetFloatable)
- self.setWidget(QWidget(self))
- self.widget().show()
- self.widget().setLayout(QGridLayout(self.widget()))
- self.consoleMsgs = []
- self.consoleText = QTextEdit(self)
- self.consoleText.setReadOnly(True)
- self.widget().layout().addWidget(self.consoleText, 0, 0, 1, 3)
- self.statusLabel = QLabel(self)
- self.widget().layout().addWidget(self.statusLabel, 1, 0, 2, 1)
- self.chipaccessProgress = QProgressBar(self)
- self.widget().layout().addWidget(self.chipaccessProgress, 1, 1)
- self.globalProgress = QProgressBar(self)
- self.widget().layout().addWidget(self.globalProgress, 2, 1)
- self.cancelButton = QPushButton("Cancel", self)
- self.cancelButton.setEnabled(False)
- self.widget().layout().addWidget(self.cancelButton, 1, 2, 2, 1)
- self.setStatus(self.STAT_OK)
- self.idToProgressBar = {
- GuiUserInterface.PROGRESSMETER_CHIPACCESS : self.chipaccessProgress,
- GuiUserInterface.PROGRESSMETER_USER_GLOBAL : self.globalProgress,
- }
- self.cancelButton.released.connect(mainWindow.cancelHardwareTask)
- def blockStatusUpdate(self):
- self.statusUpdateBlocked += 1
- def unblockStatusUpdate(self):
- self.statusUpdateBlocked -= 1
- def setTaskRunning(self, running, success=True):
- if self.statusUpdateBlocked:
- return
- self.cancelButton.setEnabled(running)
- if running:
- self.setStatus(Console.STAT_PROGRESS)
- # Set progress meters to "busy"
- for meterId in list(self.idToProgressBar.keys()):
- self.progressMeterInit(meterId, 0)
- else:
- if success:
- self.setStatus(Console.STAT_OK)
- else:
- self.setStatus(Console.STAT_ERROR)
- # Reset progress meters to 0%
- for meterId in list(self.idToProgressBar.keys()):
- self.progressMeterInit(meterId, 2)
- def setStatus(self, status):
- if self.statusUpdateBlocked:
- return
- if status == self.STAT_OK:
- path = getIconPath("ok")
- elif status == self.STAT_ERROR:
- path = getIconPath("error")
- elif status == self.STAT_PROGRESS:
- path = getIconPath("progress")
- else:
- assert(0)
- self.statusLabel.setPixmap(QPixmap(path))
- def __commitText(self):
- limit = 100
- if len(self.consoleMsgs) > limit:
- self.consoleMsgs.pop(0)
- assert(len(self.consoleMsgs) == limit)
- html = "<HTML><BODY>" + "".join(self.consoleMsgs) + "</BODY></HTML>"
- self.consoleText.setHtml(html)
- # Scroll to end
- scroll = self.consoleText.verticalScrollBar()
- scroll.setTracking(True)
- scroll.setSliderPosition(scroll.maximum())
- scroll.setValue(scroll.maximum())
- def showMessage(self, message, bold=False):
- message = str(message)
- if not message:
- return
- newline = False
- if message.endswith("\n"):
- newline = True
- message = message[:-1]
- message = htmlEscape(message)
- if bold:
- message = "<B>" + message + "</B>"
- if self.addPrefix:
- time = str(QTime.currentTime().toString("hh:mm:ss"))
- message = "<I>[%s]</I> %s" % (time, message)
- self.addPrefix = newline
- if newline:
- message += "<BR />"
- lastmsg = None
- if self.consoleMsgs:
- lastmsg = self.consoleMsgs[-1]
- if lastmsg and not lastmsg.endswith("<BR />"):
- self.consoleMsgs[-1] = self.consoleMsgs[-1] + message
- else:
- self.consoleMsgs.append(message)
- self.__commitText()
- def progressMeterInit(self, meterId, nrSteps):
- if self.statusUpdateBlocked:
- return
- progress = self.idToProgressBar[meterId]
- progress.setMinimum(0)
- progress.setMaximum(max(0, nrSteps - 1))
- progress.setValue(progress.maximum())
- progress.setValue(0)
- def progressMeter(self, meterId, step):
- if self.statusUpdateBlocked:
- return
- progress = self.idToProgressBar[meterId]
- if progress.maximum() == 0:
- progress.setMaximum(1)
- progress.setValue(1)
- else:
- progress.setValue(step)
- class HexEditWidget(QWidget):
- def __init__(self, scrollArea, parent=None):
- QWidget.__init__(self, parent)
- self.scrollArea = scrollArea
- self.bgColor = QColor("#FFFFFF")
- self.fgColor = QColor("#000000")
- self.cursor0Color = QColor("#A0A07F")
- self.cursor1Color = QColor("#D0D0AF")
- font = self.font()
- font.setFamily("Courier")
- font.setStyleHint(QFont.StyleHint.Courier)
- font.setWeight(QFont.Weight.Normal)
- font.setPointSize(10)
- font.setBold(False)
- font.setFixedPitch(True)
- self.setFont(font)
- self.charWidth = self.fontMetrics().boundingRect("X" * 100).width() / 100
- self.charHeight = self.fontMetrics().height()
- self.bytesPerLine = 16
- self.previousData = b""
- self.setData(b"")
- self.__setCursor(0, False)
- self.setFocusPolicy(Qt.FocusPolicy.StrongFocus)
- def updateText(self):
- lines = []
- if self.data:
- for i in range(0, len(self.data), self.bytesPerLine):
- end = min(i + self.bytesPerLine, len(self.data))
- data = self.data[i : end]
- try:
- prevData = self.previousData[i : end]
- except IndexError:
- prevData = None
- if prevData == data:
- lines.append(self.textLines[i // self.bytesPerLine])
- else:
- text = [ "[%08X]: " % i ]
- for byte in data:
- text.append(" " + byte2hex(byte))
- if len(data) % self.bytesPerLine:
- # padding
- nrLeft = self.bytesPerLine - (len(data) % self.bytesPerLine)
- text.append(" " * nrLeft)
- text.append(" |")
- text.append(bytes2ascii(data))
- text.append("|")
- lines.append("".join(text))
- else:
- lines = [ "<The buffer is empty>", ]
- self.textLines = lines
- self.nrLines = len(lines)
- self.previousData = self.getData()
- width = self.fontMetrics().boundingRect(self.textLines[0]).width()
- height = self.nrLines * self.charHeight
- self.resize(width + 20, height + 20)
- self.repaint()
- def getData(self):
- return bytes(self.data)
- def setData(self, data):
- if not data:
- data = b""
- assert isinstance(data, (bytes, bytearray))
- self.data = bytearray(data)
- self.updateText()
- self.__setCursor(0, False)
- def charInput(self, char):
- if not self.data:
- return
- assert(self.cursorByte < len(self.data))
- if self.cursorOnAscii:
- self.data[self.cursorByte] = ord(char)
- self.__setCursor(self.cursorByte + 1, True)
- else:
- dataByte = self.data[self.cursorByte]
- char = ord(char.upper())
- if char >= ord("0") and char <= ord("9"):
- nibble = char - ord("0")
- elif char >= ord("A") and char <= ord("F"):
- nibble = char - ord("A") + 10
- else:
- return
- if self.cursorNibble == 0:
- dataByte = (dataByte & 0xF0) | nibble
- else:
- dataByte = (dataByte & 0x0F) | (nibble << 4)
- self.data[self.cursorByte] = dataByte
- if self.cursorNibble == 1:
- self.cursorNibble = 0
- else:
- self.__setCursor(self.cursorByte + 1, False)
- self.updateText()
- def keyPressEvent(self, e):
- linesOnScreen = self.scrollArea.viewport().height() // self.charHeight
- if e.matches(QKeySequence.StandardKey.Delete) or\
- e.key() == Qt.Key.Key_Backspace:
- self.__setCursor(self.cursorByte - 1, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToNextChar):
- self.__setCursor(self.cursorByte + 1, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToPreviousChar):
- self.__setCursor(self.cursorByte - 1, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToStartOfLine):
- self.__setCursor(self.cursorByte - self.cursorByte % self.bytesPerLine,
- self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToEndOfLine):
- self.__setCursor(self.cursorByte - self.cursorByte % self.bytesPerLine
- + self.bytesPerLine - 1,
- self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToNextLine):
- self.__setCursor(self.cursorByte + self.bytesPerLine, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToPreviousLine):
- self.__setCursor(self.cursorByte - self.bytesPerLine, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToStartOfDocument):
- self.__setCursor(0, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToEndOfDocument):
- self.__setCursor(len(self.data) - 1, self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToNextPage):
- self.__setCursor(self.cursorByte + int(linesOnScreen * 0.8) * self.bytesPerLine,
- self.cursorOnAscii)
- return
- if e.matches(QKeySequence.StandardKey.MoveToPreviousPage):
- self.__setCursor(self.cursorByte - int(linesOnScreen * 0.8) * self.bytesPerLine,
- self.cursorOnAscii)
- return
- text = str(e.text())
- if text:
- key = ord(text[0])
- if self.cursorOnAscii:
- self.charInput(chr(key))
- else:
- if (key >= ord("0") and key <= ord("9")) or\
- (key >= ord("a") and key <= ord("f")) or\
- (key >= ord("A") and key <= ord("F")):
- self.charInput(chr(key))
- def __setCursor(self, byteNr, asciiArea):
- byteNr = min(byteNr, len(self.data) - 1)
- byteNr = max(byteNr, 0)
- self.cursorOnAscii = asciiArea
- self.cursorByte = byteNr
- self.cursorNibble = 1
- self.repaint()
- (x, y) = self.__cursorUpperLeftEdge()
- self.scrollArea.ensureVisible(x, y)
- def mousePressEvent(self, e):
- x = e.pos().x()
- y = e.pos().y()
- byteAreaX = round(self.charWidth * 13)
- byteAreaY = 3
- asciiAreaX = round(self.charWidth * (13 + 3 * self.bytesPerLine + 3 - 1))
- asciiAreaY = 3
- if x >= byteAreaX and\
- y >= byteAreaY and\
- x <= round(byteAreaX + self.charWidth * 3 * self.bytesPerLine) and\
- y <= byteAreaY + self.charHeight * self.nrLines:
- # Click in byte area.
- x -= byteAreaX
- y -= byteAreaY
- line = y // self.charHeight
- column = round(x // (self.charWidth * 3))
- self.__setCursor(line * self.bytesPerLine + column,
- False)
- return
- if x >= asciiAreaX and\
- y >= asciiAreaY and\
- x <= round(asciiAreaX + self.charWidth * self.bytesPerLine) and\
- y <= asciiAreaY + self.charHeight * self.nrLines:
- # Click on ascii area.
- x -= asciiAreaX
- y -= asciiAreaY
- line = y // self.charHeight
- column = round(x // self.charWidth)
- self.__setCursor(line * self.bytesPerLine + column,
- True)
- return
- def __cursorUpperLeftEdge(self):
- if self.cursorOnAscii:
- x = self.charWidth * (13 + 3 * self.bytesPerLine + 3 - 1)
- x += self.charWidth * (self.cursorByte % self.bytesPerLine)
- y = self.charHeight * (self.cursorByte // self.bytesPerLine)
- y += 3
- else:
- x = self.charWidth * (13 + 3 * (self.cursorByte % self.bytesPerLine))
- y = self.charHeight * (self.cursorByte // self.bytesPerLine)
- y += 3
- return (round(x), round(y))
- def paintEvent(self, e):
- p = QPainter(self)
- # Background
- p.fillRect(0, 0, self.width(), self.height(), self.bgColor)
- # Cursor
- if self.data and self.hasFocus():
- (x, y) = self.__cursorUpperLeftEdge()
- if self.cursorOnAscii:
- p.fillRect(x, y, round(self.charWidth), self.charHeight,
- self.cursor0Color)
- else:
- p.fillRect(x, y, round(self.charWidth * 2), self.charHeight,
- self.cursor1Color)
- if self.cursorNibble == 0:
- x = round(x + self.charWidth)
- p.fillRect(x, y, round(self.charWidth), self.charHeight,
- self.cursor0Color)
- # Text
- p.setPen(QPen(self.fgColor))
- y = self.charHeight
- for line in self.textLines:
- p.drawText(0, y, line)
- y += self.charHeight
- class HexEdit(QScrollArea):
- def __init__(self, parent=None):
- QScrollArea.__init__(self, parent)
- self.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOn)
- self.setWidgetResizable(False)
- self.setWidget(HexEditWidget(self, self))
- def setData(self, data):
- self.widget().setData(data)
- def getData(self):
- return self.widget().getData()
- class BufferWidget(QWidget):
- def __init__(self, mainWindow, name):
- QWidget.__init__(self, mainWindow)
- self.mainWindow = mainWindow
- self.name = name
- self.hide()
- self.setLayout(QGridLayout(self))
- self.setAvailable(False)
- self.setReadOnly()
- def getName(self):
- return self.name
- def setAvailable(self, available):
- self.available = available
- def isAvailable(self):
- return self.available
- def setReadOnly(self, readOnly=True):
- self.readOnly = readOnly
- def isReadOnly(self):
- return self.readOnly
- def getRawData(self):
- return None
- def setRawData(self, data):
- return False
- def setDataWithIHexInterpreter(self, interp, readRaw):
- return False
- def clear(self):
- pass
- class InfoBufferWidget(BufferWidget):
- def __init__(self, mainWindow, name):
- BufferWidget.__init__(self, mainWindow, name)
- self.zifLayout = QLabel(self)
- font = self.zifLayout.font()
- font.setFixedPitch(True)
- font.setFamily("monospace")
- font.setPointSize(7)
- self.zifLayout.setFont(font)
- self.layout().addWidget(self.zifLayout, 0, 0, 6, 1)
- def addLabel(name, y):
- l = QLabel(name, self)
- l.setAlignment(Qt.AlignmentFlag.AlignTop | Qt.AlignmentFlag.AlignLeft)
- self.layout().addWidget(l, y, 1)
- label = QLabel(self)
- label.setAlignment(Qt.AlignmentFlag.AlignTop | Qt.AlignmentFlag.AlignLeft)
- font = label.font()
- font.setBold(True)
- label.setFont(font)
- self.layout().addWidget(label, y, 2)
- return label
- self.chipName = addLabel("Chip name:", 0)
- self.comment = addLabel("Comment:", 1)
- self.packages = addLabel("Packages:", 2)
- self.chipSig = addLabel("Chip signature bytes:", 3)
- self.maintainer = addLabel("Chip maintainer:", 4)
- self.layout().setColumnStretch(3, 99)
- self.layout().setRowStretch(5, 99)
- self.clear()
- def clear(self):
- self.chipName.setText("<No chip selected>")
- self.setComment(None)
- self.setChipSignature(None)
- self.setChipLayout(None)
- self.setPackages(None)
- self.setMaintainer(None)
- def setChipSignature(self, bindata):
- if bindata:
- self.chipSig.setText(bytes2hex(bindata))
- else:
- self.chipSig.setText("None")
- def setComment(self, comment):
- if comment:
- self.comment.setText(comment)
- else:
- self.comment.setText("None")
- def setPackages(self, packages):
- def p2str(param):
- (packageName, description) = param
- text = packageName
- if description:
- text += " (%s)" % description
- return text
- if packages:
- pckgs = list(map(p2str, packages))
- self.packages.setText("\n".join(pckgs))
- else:
- self.packages.setText("Unknown")
- def setMaintainer(self, maintainer):
- if maintainer:
- # Remove email address
- idx = maintainer.find("<")
- if idx >= 0:
- maintainer = maintainer[:idx]
- self.maintainer.setText(maintainer.strip())
- else:
- self.maintainer.setText("None")
- def setupDescription(self, chipDescription):
- self.chipName.setText(chipDescription.description)
- self.setComment(chipDescription.comment)
- self.setPackages(chipDescription.packages)
- self.setMaintainer(chipDescription.maintainer)
- def setChipLayout(self, asciiArtLayout):
- if asciiArtLayout:
- self.zifLayout.setText(asciiArtLayout)
- self.zifLayout.setFrameShape(QFrame.Shape.StyledPanel)
- else:
- self.zifLayout.clear()
- self.zifLayout.setFrameShape(QFrame.Shape.NoFrame)
- class ImageBufferWidget(BufferWidget):
- def __init__(self, mainWindow, name):
- BufferWidget.__init__(self, mainWindow, name)
- self.browser = HexEdit(self)
- self.layout().addWidget(self.browser, 0, 0)
- def loadImage(self, image):
- self.browser.setData(image)
- def getRawData(self):
- return self.browser.getData()
- def setRawData(self, data):
- if self.isReadOnly():
- return False
- self.loadImage(data)
- return True
- def clear(self):
- self.loadImage(None)
- class BitBufferWidget(BufferWidget):
- def __init__(self, mainWindow, name, bitDescriptionsAttr):
- BufferWidget.__init__(self, mainWindow, name)
- self.bitDescriptionsAttr = bitDescriptionsAttr
- self.bitsAreaScroll = QScrollArea(self)
- self.layout().addWidget(self.bitsAreaScroll, 0, 0)
- self.bitsAreaScroll.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOn)
- self.bitsAreaScroll.setWidgetResizable(False)
- self.bitsArea = None
- self.image = None
- self.checkboxes = []
- def __getBitDescription(self, bitNr):
- if not self.mainWindow.currentChipDescription:
- return ""
- bitDescList = getattr(self.mainWindow.currentChipDescription,
- self.bitDescriptionsAttr, None)
- if not bitDescList:
- return ""
- bitDesc = [x for x in bitDescList if x.bitNr == bitNr]
- if not bitDesc:
- return ""
- return bitDesc[0].description
- def loadImage(self, image):
- self.image = image
- for checkbox in self.checkboxes:
- checkbox.deleteLater()
- self.checkboxes = []
- if self.bitsArea:
- self.bitsArea.deleteLater()
- self.bitsArea = None
- if not self.image:
- return
- self.bitsArea = QWidget(self)
- self.bitsArea.setLayout(QGridLayout(self.bitsArea))
- self.bitsArea.layout().addWidget(QLabel("Currently set %s:" % self.getName(),
- self.bitsArea),
- 0, 0)
- self.hexView = QLabel(self.bitsArea)
- font = self.hexView.font()
- font.setBold(True)
- self.hexView.setFont(font)
- self.bitsArea.layout().addWidget(self.hexView, 0, 1)
- for i in range(0, len(image) * 8):
- desc = self.__getBitDescription(i)
- if desc:
- desc = " (%s)" % desc
- checkbox = QCheckBox("bit %d%s" % (i, desc), self.bitsArea)
- if byte2int(image[i // 8]) & bit(i % 8):
- checkbox.setCheckState(Qt.CheckState.Checked)
- self.bitsArea.layout().addWidget(checkbox, 1 + i, 0, 1, 2)
- checkbox.stateChanged.connect(self.__bitStateChanged)
- self.checkboxes.append(checkbox)
- self.__updateHexView()
- def __updateHexView(self):
- self.hexView.setText(bytes2hex(self.image))
- self.bitsAreaScroll.setWidget(self.bitsArea)
- def __bitStateChanged(self, unused):
- self.image = ""
- tmp = 0
- i = 0
- for checkbox in self.checkboxes:
- if checkbox.checkState() == Qt.CheckState.Checked:
- tmp |= bit(i)
- i += 1
- if i == 8:
- self.image += chr(tmp)
- i = 0
- tmp = 0
- assert(i == 0)
- self.__updateHexView()
- def getRawData(self):
- return self.image
- def setRawData(self, data):
- if self.isReadOnly():
- return False
- if len(data) > 32: # Don't want huge files
- return False
- self.loadImage(data)
- return True
- def clear(self):
- self.loadImage(None)
- class ProgmemBufferWidget(ImageBufferWidget):
- def __init__(self, mainWindow):
- ImageBufferWidget.__init__(self, mainWindow, "program memory")
- def setDataWithIHexInterpreter(self, interp, readRaw):
- return self.setRawData(interp.getProgmem(dontInterpretSections = readRaw))
- class EEPROMBufferWidget(ImageBufferWidget):
- def __init__(self, mainWindow):
- ImageBufferWidget.__init__(self, mainWindow, "(E)EPROM memory")
- def setDataWithIHexInterpreter(self, interp, readRaw):
- return self.setRawData(interp.getEEPROM(dontInterpretSections = readRaw))
- class RAMBufferWidget(ImageBufferWidget):
- def __init__(self, mainWindow):
- ImageBufferWidget.__init__(self, mainWindow, "RAM memory")
- def setDataWithIHexInterpreter(self, interp, readRaw):
- return self.setRawData(interp.getRAM(dontInterpretSections = readRaw))
- class FuseBufferWidget(BitBufferWidget):
- def __init__(self, mainWindow):
- BitBufferWidget.__init__(self, mainWindow, "fuse bits", "fuseDesc")
- def setDataWithIHexInterpreter(self, interp, readRaw):
- return self.setRawData(interp.getFusebits(dontInterpretSections = readRaw))
- class LockBufferWidget(BitBufferWidget):
- def __init__(self, mainWindow):
- BitBufferWidget.__init__(self, mainWindow, "lock bits", "lockbitDesc")
- def setDataWithIHexInterpreter(self, interp, readRaw):
- return self.setRawData(interp.getLockbits(dontInterpretSections = readRaw))
- class BufferTabWidget(QTabWidget):
- def __init__(self, mainWindow):
- QTabWidget.__init__(self, mainWindow)
- self.mainWindow = mainWindow
- self.infoBuffer = InfoBufferWidget(mainWindow, "chip info")
- self.progmemBuffer = ProgmemBufferWidget(mainWindow)
- self.eepromBuffer = EEPROMBufferWidget(mainWindow)
- self.fuseBuffer = FuseBufferWidget(mainWindow)
- self.lockBuffer = LockBufferWidget(mainWindow)
- self.ramBuffer = RAMBufferWidget(mainWindow)
- self.setTabPosition(self.TabPosition.South)
- self.__addBufferTab(self.infoBuffer, readOnly=True)
- self.removeAll()
- def __addBufferTab(self, bufferWidget, readOnly=False):
- self.addTab(bufferWidget, bufferWidget.getName())
- bufferWidget.setReadOnly(readOnly)
- bufferWidget.setAvailable(True)
- def __removeBufferTab(self, bufferWidget):
- self.removeTab(self.indexOf(bufferWidget))
- bufferWidget.clear()
- bufferWidget.setAvailable(False)
- def __removeAll(self):
- self.__removeBufferTab(self.progmemBuffer)
- self.__removeBufferTab(self.eepromBuffer)
- self.__removeBufferTab(self.fuseBuffer)
- self.__removeBufferTab(self.lockBuffer)
- self.__removeBufferTab(self.ramBuffer)
- def removeAll(self):
- self.__removeAll()
- def setupBuffers(self, chipSupportFlags):
- prevSelect = self.getCurrentBuffer()
- self.__removeAll()
- for (bufferWidget, flagsMask) in (
- (self.progmemBuffer, Chip.SUPPORT_PROGMEMREAD | Chip.SUPPORT_PROGMEMWRITE),
- (self.eepromBuffer, Chip.SUPPORT_EEPROMREAD | Chip.SUPPORT_EEPROMWRITE),
- (self.fuseBuffer, Chip.SUPPORT_FUSEREAD | Chip.SUPPORT_FUSEWRITE),
- (self.lockBuffer, Chip.SUPPORT_LOCKREAD | Chip.SUPPORT_LOCKWRITE),
- (self.ramBuffer, Chip.SUPPORT_RAMREAD | Chip.SUPPORT_RAMWRITE)):
- if chipSupportFlags & flagsMask:
- self.__addBufferTab(bufferWidget)
- if prevSelect and prevSelect.isAvailable():
- self.setCurrentIndex(self.indexOf(prevSelect))
- def loadBuffers(self, images):
- try:
- self.progmemBuffer.loadImage(images["progmem"])
- except KeyError:
- pass # Ignore
- try:
- self.eepromBuffer.loadImage(images["eeprom"])
- except KeyError:
- pass # Ignore
- try:
- self.fuseBuffer.loadImage(images["fusebits"])
- except KeyError:
- pass # Ignore
- try:
- self.lockBuffer.loadImage(images["lockbits"])
- except KeyError:
- pass # Ignore
- try:
- self.ramBuffer.loadImage(images["ram"])
- except KeyError:
- pass # Ignore
- def verifyBuffers(self, images):
- fail = False
- for (imageName, bufferWidget) in (("progmem", self.progmemBuffer),
- ("eeprom", self.eepromBuffer),
- ("fusebits", self.fuseBuffer),
- ("ram", self.ramBuffer)):
- try:
- image = images[imageName]
- except KeyError:
- continue
- bufImage = bufferWidget.getRawData()
- if not bufImage:
- if not image:
- # Both images are empty. Go on...
- continue
- self.mainWindow.console.showMessage(
- "Chip verify of %s FAILED! "
- "There is no data in the "
- "%s buffer. First load data into "
- "the buffer tab, please.\n" %\
- (bufferWidget.getName(),
- bufferWidget.getName()),
- bold=True)
- fail = True
- continue
- if len(bufImage) > len(image):
- self.mainWindow.console.showMessage(
- "Chip verify of %s FAILED! "
- "Buffer data is larger than "
- "chip memory.\n" %\
- bufferWidget.getName(),
- bold=True)
- fail = True
- continue
- compareSize = min(len(image), len(bufImage))
- if len(image) != len(bufImage):
- self.mainWindow.console.showMessage(
- "%s: Only comparing the first 0x%X bytes due "
- "to a mismatch of the buffer size (0x%X) "
- "with the actual device image size (0x%X).\n" %\
- (bufferWidget.getName(), compareSize,
- len(bufImage), len(image)))
- if bufImage[:compareSize] != image[:compareSize]:
- self.mainWindow.console.showMessage(
- "Chip verify of %s FAILED! "
- "Image data mismatch.\n" %\
- bufferWidget.getName(),
- bold=True)
- fail = True
- continue
- if fail:
- self.mainWindow.console.setStatus(Console.STAT_ERROR)
- else:
- self.mainWindow.console.showMessage("Chip verify succeed\n", bold=True)
- self.mainWindow.console.setStatus(Console.STAT_OK)
- return not fail
- def getCurrentBuffer(self):
- return self.currentWidget()
- class ChipSelectDialog(QDialog):
- ALL_VENDORS = "--- All vendors ---"
- def __init__(self, parent):
- QDialog.__init__(self, parent)
- self.setWindowTitle("Select chip")
- self.setLayout(QGridLayout(self))
- groupBox = QGroupBox("Chip type", self)
- groupBox.setLayout(QGridLayout(groupBox))
- self.allRadio = QRadioButton("All types", groupBox)
- self.allRadio.setChecked(True)
- groupBox.layout().addWidget(self.allRadio, 0, 0)
- self.mcuRadio = QRadioButton("Microcontrollers", groupBox)
- groupBox.layout().addWidget(self.mcuRadio, 1, 0)
- self.epromRadio = QRadioButton("EPROMs", groupBox)
- groupBox.layout().addWidget(self.epromRadio, 2, 0)
- self.eepromRadio = QRadioButton("EEPROMs", groupBox)
- groupBox.layout().addWidget(self.eepromRadio, 3, 0)
- self.galRadio = QRadioButton("PALs / GALs", groupBox)
- groupBox.layout().addWidget(self.galRadio, 4, 0)
- self.sramRadio = QRadioButton("Static RAM", groupBox)
- groupBox.layout().addWidget(self.sramRadio, 5, 0)
- self.logicRadio = QRadioButton("Logic chips", groupBox)
- groupBox.layout().addWidget(self.logicRadio, 6, 0)
- self.showBroken = QCheckBox("Show broken implementations", groupBox)
- groupBox.layout().addWidget(self.showBroken, 7, 0)
- self.layout().addWidget(groupBox, 0, 0, 2, 2)
- l = QLabel("Vendor:", self)
- self.layout().addWidget(l, 0, 2, 1, 2)
- self.vendorList = QListWidget(self)
- self.layout().addWidget(self.vendorList, 1, 2, 1, 2)
- l = QLabel("Chip:", self)
- self.layout().addWidget(l, 0, 4, 1, 2)
- self.chipList = QListWidget(self)
- self.layout().addWidget(self.chipList, 1, 4, 1, 2)
- self.okButton = QPushButton("&Ok", self)
- self.layout().addWidget(self.okButton, 2, 0, 1, 3)
- self.cancelButton = QPushButton("&Cancel", self)
- self.layout().addWidget(self.cancelButton, 2, 3, 1, 3)
- self.okButton.released.connect(self.accept)
- self.cancelButton.released.connect(self.reject)
- self.vendorList.itemSelectionChanged.connect(self.vendorSelectionChanged)
- self.chipList.itemSelectionChanged.connect(self.chipSelectionChanged)
- self.chipList.itemDoubleClicked.connect(self.accept)
- self.showBroken.stateChanged.connect(self.typeToggled)
- for radio in (self.allRadio, self.mcuRadio, self.epromRadio,
- self.eepromRadio, self.galRadio,
- self.sramRadio, self.logicRadio):
- radio.toggled.connect(self.typeToggled)
- self.__updateVendorList()
- self.__updateChipList()
- def __updateVendorList(self):
- previousVendor = self.__getSelectedVendor()
- selType = self.__getSelectedChipType()
- vendors = getRegisteredVendors()
- self.vendorList.clear()
- QListWidgetItem(self.ALL_VENDORS, self.vendorList)
- for vendorName in list(vendors.keys()):
- descriptors = vendors[vendorName]
- if self.showBroken.checkState() != Qt.CheckState.Checked:
- descriptors = [d for d in descriptors if not d.broken]
- descriptors = [d for d in descriptors if (d.chipType == selType or selType == -1) and\
- (d.chipType != ChipDescription.TYPE_INTERNAL)]
- if not descriptors:
- continue
- item = QListWidgetItem(vendorName, self.vendorList)
- item.setData(Qt.ItemDataRole.UserRole, descriptors)
- self.vendorList.sortItems()
- if previousVendor:
- items = self.vendorList.findItems(previousVendor, Qt.MatchFlag.MatchExactly)
- if len(items) == 1:
- self.vendorList.setCurrentItem(items[0])
- else:
- self.vendorList.setCurrentRow(0)
- else:
- self.vendorList.setCurrentRow(0)
- def __descriptorText(self, descriptor):
- text = descriptor.description
- if not descriptor.maintainer:
- text += " (Orphaned)"
- return text
- def __updateChipList(self):
- previousChip = self.getSelectedChip()
- selType = self.__getSelectedChipType()
- selVendor = self.__getSelectedVendor()
- self.chipList.clear()
- for descriptor in getRegisteredChips():
- if descriptor.broken and\
- self.showBroken.checkState() != Qt.CheckState.Checked:
- continue
- if descriptor.chipType != selType and selType != -1:
- continue
- if descriptor.chipType == ChipDescription.TYPE_INTERNAL:
- continue
- if selVendor != self.ALL_VENDORS and selVendor not in descriptor.chipVendors:
- continue
- item = QListWidgetItem(self.__descriptorText(descriptor),
- self.chipList)
- item.setData(Qt.ItemDataRole.UserRole, descriptor)
- self.chipList.sortItems()
- if previousChip:
- items = self.chipList.findItems(self.__descriptorText(previousChip),
- Qt.MatchFlag.MatchExactly)
- if len(items) == 1:
- self.chipList.setCurrentItem(items[0])
- else:
- self.chipList.setCurrentRow(0)
- else:
- self.chipList.setCurrentRow(0)
- def typeToggled(self, unused):
- self.__updateVendorList()
- self.__updateChipList()
- def vendorSelectionChanged(self):
- self.__updateChipList()
- def chipSelectionChanged(self):
- item = self.chipList.currentItem()
- self.okButton.setEnabled(bool(item))
- def __getSelectedVendor(self):
- item = self.vendorList.currentItem()
- if not item:
- return None
- return str(item.text())
- def __getSelectedChipType(self):
- # Returns ChipDescription.TYPE_... or -1 if all are selected
- for (typeId, radioButton) in (
- (ChipDescription.TYPE_MCU, self.mcuRadio),
- (ChipDescription.TYPE_EPROM, self.epromRadio),
- (ChipDescription.TYPE_EEPROM, self.eepromRadio),
- (ChipDescription.TYPE_GAL, self.galRadio),
- (ChipDescription.TYPE_SRAM, self.sramRadio),
- (ChipDescription.TYPE_LOGIC, self.logicRadio)):
- if radioButton.isChecked():
- return typeId
- assert(self.allRadio.isChecked())
- return -1
- def getSelectedChip(self):
- item = self.chipList.currentItem()
- if not item:
- return None
- chipDescription = item.data(Qt.ItemDataRole.UserRole)
- return chipDescription
- class StatusBar(QStatusBar):
- def __init__(self, mainWindow):
- QStatusBar.__init__(self, mainWindow)
- class TopToolBar(QToolBar):
- def __init__(self, mainWindow):
- QToolBar.__init__(self, "Toolbar", mainWindow)
- self.mainWindow = mainWindow
- self.setToolButtonStyle(Qt.ToolButtonStyle.ToolButtonTextUnderIcon)
- self.setup(0)
- def setup(self, chipSupportFlags):
- mainWindow = self.mainWindow
- self.clear()
- a = self.addAction(getIcon("open"), "Load buffer", mainWindow.loadBuffer)
- a.setEnabled(chipSupportFlags != 0)
- a = self.addAction(getIcon("save"), "Save buffer", mainWindow.saveBuffer)
- a.setEnabled(chipSupportFlags != 0)
- self.addSeparator()
- self.addAction(getIcon("chip"), "Run logic tester", mainWindow.startUnitest)
- self.addAction(getIcon("chip"), "Select chip", mainWindow.selectChip)
- self.addSeparator()
- a = self.addAction(getIcon("up"), "Read chip", mainWindow.readChip)
- a.setEnabled(chipSupportFlags != 0)
- a = self.addAction(getIcon("verify"), "Verify all", mainWindow.verifyChip)
- a.setEnabled(chipSupportFlags != 0)
- class RightToolBar(QToolBar):
- def __init__(self, mainWindow):
- QToolBar.__init__(self, "Toolbar", mainWindow)
- self.mainWindow = mainWindow
- self.setToolButtonStyle(Qt.ToolButtonStyle.ToolButtonTextUnderIcon)
- self.setup(0)
- def setup(self, chipSupportFlags):
- mainWindow = self.mainWindow
- self.clear()
- if chipSupportFlags & Chip.SUPPORT_ERASE:
- self.addAction(getIcon("erase"), "Erase", mainWindow.eraseChip)
- if chipSupportFlags & Chip.SUPPORT_PROGMEMWRITE:
- self.addAction(getIcon("memory"), "Write progmem", mainWindow.writeChipProgmem)
- if chipSupportFlags & Chip.SUPPORT_EEPROMWRITE:
- self.addAction(getIcon("memory"), "Write EPROM", mainWindow.writeChipEeprom)
- if chipSupportFlags & Chip.SUPPORT_FUSEWRITE:
- self.addAction(getIcon("memory"), "Write fuses", mainWindow.writeChipFuses)
- if chipSupportFlags & Chip.SUPPORT_RAMWRITE:
- self.addAction(getIcon("memory"), "Write RAM", mainWindow.writeRam)
- if chipSupportFlags & Chip.SUPPORT_LOCKWRITE:
- self.addAction(getIcon("lock"), "Write lockbits", mainWindow.writeChipLockbits)
- if chipSupportFlags & Chip.SUPPORT_TEST:
- self.addAction(getIcon("test"), "Run unit-test", mainWindow.runTest)
- class AutorunWidget(QDockWidget):
- def __init__(self, mainWindow):
- QDockWidget.__init__(self, "Autorun", mainWindow)
- self.mainWindow = mainWindow
- self.running = False
- self.currentAction = -1
- self.setFeatures(self.DockWidgetFeature.DockWidgetMovable |
- self.DockWidgetFeature.DockWidgetFloatable)
- self.setup(0)
- def __addCheckBox(self, row, condition, text, checked):
- if condition:
- checkbox = QCheckBox(text, self.widget())
- if checked:
- checkbox.setCheckState(Qt.CheckState.Checked)
- self.widget().layout().addWidget(checkbox, row, 0)
- row += 1
- return (checkbox, row)
- return (None, row)
- def setup(self, chipSupportFlags):
- self.setWidget(QWidget(self))
- self.widget().show()
- self.widget().setLayout(QGridLayout(self.widget()))
- row = 0
- (self.testCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_TEST,
- "Run unit-test", True)
- (self.eraseCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_ERASE,
- "Erase", True)
- (self.progmemCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_PROGMEMWRITE,
- "Write program memory", True)
- (self.eepromCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_EEPROMWRITE,
- "Write (E)EPROM", True)
- (self.ramCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_RAMWRITE,
- "Write RAM", True)
- (self.fusesCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_FUSEWRITE,
- "Write fuses", True)
- (self.verifyCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags != 0,
- "Verify selected memories", True)
- (self.lockbitsCheckBox, row) = self.__addCheckBox(row,
- chipSupportFlags & Chip.SUPPORT_LOCKWRITE,
- "Write lock bits", False)
- self.runButton = None
- if chipSupportFlags != 0:
- self.runButton = QPushButton("Run", self.widget())
- self.runButton.setIconSize(QSize(32, 32))
- self.runButton.setIcon(getIcon("run"))
- self.widget().layout().addWidget(self.runButton, row, 0)
- row += 1
- self.runButton.released.connect(self.runNextChip)
- self.widget().layout().setRowStretch(row, 99)
- def isRunning(self):
- return self.running
- def abortRunWithMessage(self):
- self.abortRun(withMessage=True)
- def abortRun(self, withMessage=False):
- if withMessage:
- QMessageBox.critical(self, "Aborting autorun",
- "One action failed. Aborting autorun.\n"
- "See console messages for details.")
- self.running = False
- self.currentAction = -1;
- self.mainWindow.guiUpdateEnable()
- def runNextChip(self):
- self.running = True
- self.currentAction = 0
- res = QMessageBox.information(self, "Autorun - Please insert chip",
- "Please insert a new %s chip into the ZIF socket "
- "and press Ok to perform the operations..." %\
- self.mainWindow.currentChipDescription.description,
- QMessageBox.StandardButton.Ok | QMessageBox.StandardButton.Cancel,
- QMessageBox.StandardButton.Ok)
- if res != QMessageBox.StandardButton.Ok:
- self.abortRun()
- return
- self.runNextAction()
- def runNextAction(self):
- assert(self.currentAction >= 0)
- action = 0
- if self.testCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.runTest,
- self.testCheckBox,
- "Run unit-test"):
- return
- action += 1
- if self.eraseCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.eraseChip,
- self.eraseCheckBox,
- "Erase"):
- return
- action += 1
- if self.progmemCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.writeChipProgmem,
- self.progmemCheckBox,
- "Write program memory"):
- return
- action += 1
- if self.eepromCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.writeChipEeprom,
- self.eepromCheckBox,
- "Write (E)EPROM"):
- return
- action += 1
- if self.ramCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.writeRam,
- self.ramCheckBox,
- "Write RAM"):
- return
- action += 1
- if self.fusesCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.writeChipFuses,
- self.fusesCheckBox,
- "Write fuses"):
- return
- action += 1
- if self.verifyCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.__verifySelectedMemories,
- self.verifyCheckBox,
- "Verify selected memories"):
- return
- action += 1
- if self.lockbitsCheckBox:
- if self.currentAction == action:
- if self.__runAction(self.mainWindow.writeChipLockbits,
- self.lockbitsCheckBox,
- "Write lock bits"):
- return
- action += 1
- self.runNextChip()
- def __runAction(self, function, checkbox, actionName):
- self.currentAction += 1
- if checkbox.checkState() == Qt.CheckState.Checked:
- self.mainWindow.console.showMessage(
- "Running action: %s\n" % actionName, bold=True)
- if not function():
- self.abortRun(withMessage=True)
- return True
- return False
- def __verifySelectedMemories(self):
- def checked(cb):
- return cb and (cb.checkState() == Qt.CheckState.Checked)
- return self.mainWindow.verifySelectedMemories(
- verifyProgmem = checked(self.progmemCheckBox),
- verifyEEPROM = checked(self.eepromCheckBox),
- verifyFuse = checked(self.fusesCheckBox),
- verifyRAM = checked(self.ramCheckBox))
- class ProgrammerSelectDialog(QDialog):
- def __init__(self, parent=None):
- QDialog.__init__(self, parent)
- self.setWindowTitle("Select programmer device")
- self.setLayout(QGridLayout(self))
- self.deviceList = QListWidget(self)
- self.layout().addWidget(self.deviceList, 0, 0, 1, 2)
- self.rescanButton = QPushButton("&Rescan USB busses", self)
- self.layout().addWidget(self.rescanButton, 1, 0, 1, 2)
- self.okButton = QPushButton("&OK", self)
- self.layout().addWidget(self.okButton, 2, 0)
- self.cancelButton = QPushButton("&Cancel", self)
- self.layout().addWidget(self.cancelButton, 2, 1)
- self.rescan()
- self.deviceList.currentRowChanged.connect(self.selectionChanged)
- self.deviceList.itemDoubleClicked.connect(self.accept)
- self.okButton.released.connect(self.accept)
- self.cancelButton.released.connect(self.reject)
- self.rescanButton.released.connect(self.rescan)
- def rescan(self):
- self.deviceList.clear()
- for foundDev in TOP.findDevices():
- text = "%s (%s)" % (foundDev.toptype,
- foundDev.devIdentifier)
- item = QListWidgetItem(text, self.deviceList)
- item.setData(Qt.ItemDataRole.UserRole, foundDev.devIdentifier)
- self.deviceList.setCurrentRow(0)
- def getIdentifier(self):
- item = self.deviceList.currentItem()
- if not item:
- return None
- return item.data(Qt.ItemDataRole.UserRole)
- def selectionChanged(self, unused):
- if self.deviceList.currentItem():
- self.okButton.setEnabled(True)
- else:
- self.okButton.setEnabled(False)
- class Operation(object):
- OP_NONE = 0
- OP_SHUTDOWN = 1
- OP_INITCHIP = 2
- OP_READALL = 3
- OP_ERASE = 4
- OP_WRITEPROG = 5
- OP_WRITEEPROM = 6
- OP_WRITEFUSE = 7
- OP_WRITELOCK = 8
- OP_WRITERAM = 9
- OP_TEST = 10
- OP_VERIFY = 11
- OP_RAWCOMMAND = 12
- OP_SYNCHRONOUS = 13
- def __init__(self, op, hwTask, hwTaskParam=None):
- self.op = op
- self.hwTask = hwTask
- self.hwTaskParam = hwTaskParam
- self.returnValue = None
- self.failed = False
- self.finished = False
- class MainWindow(QMainWindow):
- def __init__(self, parent=None):
- QMainWindow.__init__(self, parent)
- self.setWindowTitle("TOPrammer v%s - Open Source programming suite" % VERSION)
- self.guiDisable = 0
- self.chip = None # Chip instance
- self.currentChipDescription = None
- self.previousRawCommand = ""
- self.setStatusBar(StatusBar(self))
- self.topToolBar = TopToolBar(self)
- self.rightToolBar = RightToolBar(self)
- self.addToolBar(Qt.ToolBarArea.TopToolBarArea, self.topToolBar)
- self.addToolBar(Qt.ToolBarArea.RightToolBarArea, self.rightToolBar)
- self.setMenuBar(QMenuBar(self))
- menu = QMenu("&File", self)
- menu.addAction("&Load buffer...", self.loadBuffer)
- menu.addAction("&Save buffer...", self.saveBuffer)
- menu.addSeparator()
- menu.addAction("&Exit", self.close)
- self.menuBar().addMenu(menu)
- self.runMenu = QMenu("&Run", self)
- self.setupRunMenu(0)
- self.menuBar().addMenu(self.runMenu)
- menu = QMenu("&Programmer", self)
- menu.addAction("&Universal logic tester...", self.startUnitest)
- menu.addSeparator()
- menu.addAction("Select &programmer...", self.selectProgrammer)
- menu.addAction("&Send raw command...", self.sendRawCommand)
- self.menuBar().addMenu(menu)
- menu = QMenu("&Help", self)
- menu.addAction("&About", self.showAbout)
- self.menuBar().addMenu(menu)
- self.bufferTab = BufferTabWidget(self)
- self.setCentralWidget(self.bufferTab)
- self.console = Console(self)
- self.addDockWidget(Qt.DockWidgetArea.BottomDockWidgetArea, self.console)
- self.autorun = AutorunWidget(self)
- self.addDockWidget(Qt.DockWidgetArea.LeftDockWidgetArea, self.autorun)
- self.hwThread = HwThread(self)
- self.threadRunning = False
- self.queuedOps = []
- self.runningOps = {}
- def setupRunMenu(self, chipSupportFlags):
- menu = self.runMenu
- menu.clear()
- menu.addAction("&Select chip", self.selectChip)
- menu.addSeparator()
- a = menu.addAction("&Read chip", self.readChip)
- a.setEnabled(chipSupportFlags != 0)
- a = menu.addAction("&Erase", self.eraseChip)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_ERASE))
- a = menu.addAction("Write &program memory", self.writeChipProgmem)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_PROGMEMWRITE))
- a = menu.addAction("Write (E)EP&ROM", self.writeChipEeprom)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_EEPROMWRITE))
- a = menu.addAction("Write &fuses", self.writeChipFuses)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_FUSEWRITE))
- a = menu.addAction("Write &lock bits", self.writeChipLockbits)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_LOCKWRITE))
- a = menu.addAction("Write R&AM", self.writeRam)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_RAMWRITE))
- a = menu.addAction("Run &Unit-test", self.runTest)
- a.setEnabled(bool(chipSupportFlags & Chip.SUPPORT_TEST))
- a = menu.addAction("&Verify whole chip", self.verifyChip)
- a.setEnabled(chipSupportFlags != 0)
- def event(self, e):
- if e.type() == EVENT_HWTHREAD:
- self.hwThread.handleMessageQueue()
- e.accept()
- return True
- return QMainWindow.event(self, e)
- def closeEvent(self, e):
- if self.threadRunning:
- res = QMessageBox.critical(self, "Task running",
- "A hardware access task is running.\n\n"
- "'Cancel' the close request and continue, or\n"
- "'Abort' the task an quit the application?",
- QMessageBox.Cancel | QMessageBox.Abort,
- QMessageBox.Cancel)
- if res == QMessageBox.Abort:
- e.accept()
- # Force quit
- QApplication.exit(1)
- return
- e.ignore()
- return
- self.threadRunning = False
- self.hwThread.killThread()
- e.accept()
- def showAbout(self):
- QMessageBox.information(self, "About TOPrammer",
- "Copyright (c) Michael Büsch <m@bues.ch>")
- def startUnitest(self):
- self.bufferTab.setupBuffers(0)
- self.bufferTab.infoBuffer.clear()
- self.autorun.setup(0)
- self.guiDisable += 1
- self.console.blockStatusUpdate()
- try:
- dlg = UnitestDialog(self)
- dlg.exec()
- except (TOPException) as e:
- self.console.showMessage("Failed to start Unitest: %s\n" % str(e),
- bold=True)
- self.console.unblockStatusUpdate()
- self.guiDisable -= 1
- self.guiUpdateEnable()
- self.runOperation(Operation(Operation.OP_SHUTDOWN,
- HwThread.TASK_SHUTDOWN))
- def selectProgrammer(self):
- dlg = ProgrammerSelectDialog(self)
- if dlg.exec() != QDialog.DialogCode.Accepted:
- return
- devIdentifier = dlg.getIdentifier()
- self.runOperationSync(HwThread.TASK_SHUTDOWN)
- self.hwThread.setDevice(devIdentifier)
- if self.currentChipDescription:
- self.runOperation(Operation(Operation.OP_INITCHIP,
- HwThread.TASK_INITCHIP,
- self.currentChipDescription.chipID))
- def sendRawCommand(self):
- (string, ok) = QInputDialog.getText(self,
- "Send raw command to programmer",
- "Enter raw command to send, in " +\
- "hex format (AABB1122...).\n" +\
- "WARNING: The programmer will malfunction on invalid commands.",
- QLineEdit.EchoMode.Normal,
- self.previousRawCommand)
- if not ok:
- return
- string = str(string).strip().upper()
- string = stringRemoveChars(string, " \t\r\n")
- if stringRemoveChars(string, "0123456789ABCDEF"):
- QMessageBox.critical(self, "Invalid characters",
- "Invalid characters in raw command string.\n" +\
- "Only hex characters 0-9,a-f allowed.")
- return
- if len(string) % 2 != 0 or len(string) // 2 > 64:
- QMessageBox.critical(self, "Invalid length",
- "Invalid command length. Length must be even " +\
- "and smaller or equal to 64 bytes.")
- return
- self.previousRawCommand = string
- command = hex2bin(string)
- if len(command) == 0:
- return
- self.runOperation(Operation(Operation.OP_RAWCOMMAND,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.hw.runCommandSync(...)", command)))
- def loadBuffer(self):
- bufWidget = self.bufferTab.getCurrentBuffer()
- data = None
- if not bufWidget:
- return
- if bufWidget.isReadOnly():
- QMessageBox.critical(self, "Buffer is read only",
- "Cannot load data into the %s buffer.\n"
- "The buffer is read-only." %\
- bufWidget.getName())
- return
- (fn, selectedFilter) = QFileDialog.getOpenFileName(self,
- "%s - open file" % bufWidget.getName(),
- "",
- "Autodetect file format (*);;"
- "Intel hex file (*.ihex *.hex);;"
- "Hex file with ASCII dump (*.ahex);;"
- "Binary file (*.bin)")
- if not fn:
- return
- extensions = str(selectedFilter).split("(")[1].\
- split(")")[0].replace("*", "").strip().split()
- try:
- dataIn = open(fn, "rb").read()
- except (IOError) as e:
- QMessageBox.critical(self, "Failed to read file",
- "Failed to read %s:\n%s" %\
- (str(fn), str(e.strerror)))
- return
- try:
- if ".bin" in extensions:
- handler = IO_binary()
- elif ".ahex" in extensions:
- handler = IO_ahex()
- elif ".ihex" in extensions:
- handler = IO_ihex()
- elif not extensions: # auto
- handler = IO_autodetect(dataIn)()
- else:
- assert(0)
- if isinstance(handler, IO_ihex):
- interp = self.chip.getIHexInterpreter()
- interp.interpret(dataIn)
- if interp.cumulativeSupported():
- res = QMessageBox.question(self,
- "Parse IHEX sections?",
- "This IHEX file might contain sections for "
- "the different memory areas (progmem, eeprom, etc...).\n"
- "\n"
- "Should the sections be interpreted?\n"
- "\n"
- "If 'Yes' is selected, only the section corresponding "
- "to the current buffer is extracted from the IHEX file.\n"
- "If 'No' is selected, the IHEX file will be read "
- "in raw mode and all of its data will be "
- "put into the current buffer.",
- QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
- QMessageBox.StandardButton.Yes)
- readRaw = (res != QMessageBox.StandardButton.Yes)
- else:
- readRaw = True
- ok = bufWidget.setDataWithIHexInterpreter(interp, readRaw)
- else:
- ok = bufWidget.setRawData(handler.toBinary(dataIn))
- if not ok:
- QMessageBox.critical(self, "Failed to load data",
- "Failed to load the file into the buffer")
- except (TOPException) as e:
- QMessageBox.critical(self, "Failed to convert data",
- "Failed to convert the input file data to binary\n%s" % str(e))
- return
- def saveBuffer(self):
- bufWidget = self.bufferTab.getCurrentBuffer()
- if not bufWidget:
- return
- data = bufWidget.getRawData()
- if not data:
- return
- (fn, selectedFilter) = QFileDialog.getSaveFileName(self,
- "%s - save file" % bufWidget.getName(),
- "",
- "Intel hex file (*.ihex *.hex);;"
- "Hex file with ASCII dump (*.ahex);;"
- "Binary file (*)")
- if not fn:
- return
- extensions = str(selectedFilter).split("(")[1].\
- split(")")[0].replace("*", "").strip().split()
- if not extensions:
- extensions = [ "" ]
- if not '.' in fn or\
- not '.' + fn.split('.')[-1] in extensions:
- fn += extensions[0] # Default ext
- if ".ihex" in extensions:
- data = IO_ihex().fromBinary(data)
- elif ".ahex" in extensions:
- data = IO_ahex().fromBinary(data)
- elif not extensions[0]:
- data = IO_binary().fromBinary(data)
- else:
- assert(0)
- try:
- if isinstance(data, str):
- data = data.encode("UTF-8")
- open(fn, "wb").write(data)
- except (IOError, UnicodeError) as e:
- QMessageBox.critical(self, "Failed to write file",
- "Failed to write %s:\n%s" %\
- (str(fn), str(e.strerror)))
- return
- def selectChip(self):
- dlg = ChipSelectDialog(self)
- if dlg.exec() != QDialog.DialogCode.Accepted:
- return
- chipDescription = dlg.getSelectedChip()
- if not chipDescription:
- return
- self.currentChipDescription = chipDescription
- return self.runOperation(Operation(Operation.OP_INITCHIP,
- HwThread.TASK_INITCHIP,
- chipDescription.chipID))
- def __makeReadCallChain(self, mayReadSig=True, mayReadProgmem=True,
- mayReadEEPROM=True, mayReadFuse=True,
- mayReadLockbits=True, mayReadRAM=True):
- callChain = []
- suppFlags = self.chip.getSupportFlags()
- callChain.append(GenericTopCall("top.checkChip()",
- name = "check"))
- if mayReadSig and (suppFlags & Chip.SUPPORT_SIGREAD):
- callChain.append(GenericTopCall("top.readSignature()",
- name = "signature"))
- if mayReadProgmem and (suppFlags & Chip.SUPPORT_PROGMEMREAD):
- callChain.append(GenericTopCall("top.readProgmem()",
- name = "progmem"))
- if mayReadEEPROM and (suppFlags & Chip.SUPPORT_EEPROMREAD):
- callChain.append(GenericTopCall("top.readEEPROM()",
- name = "eeprom"))
- if mayReadFuse and (suppFlags & Chip.SUPPORT_FUSEREAD):
- callChain.append(GenericTopCall("top.readFuse()",
- name = "fusebits"))
- if mayReadLockbits and (suppFlags & Chip.SUPPORT_LOCKREAD):
- callChain.append(GenericTopCall("top.readLockbits()",
- name = "lockbits"))
- if mayReadRAM and (suppFlags & Chip.SUPPORT_RAMREAD):
- callChain.append(GenericTopCall("top.readRAM()",
- name = "ram"))
- return callChain
- def __readCallChain_GetResultImages(self, callChain):
- images = {}
- for topCall in (callChain if callChain else []):
- name = topCall.userData["name"]
- if name in ("signature", "progmem", "eeprom",
- "fusebits", "lockbits", "ram"):
- images[name] = topCall.result
- elif name == "check":
- pass # Nothing to do
- else:
- assert(0)
- return images
- def readChip(self):
- return self.runOperation(Operation(Operation.OP_READALL,
- HwThread.TASK_GENERICTOPCALL,
- self.__makeReadCallChain()))
- def eraseChip(self):
- return self.runOperation(Operation(Operation.OP_ERASE,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.eraseChip()")))
- def writeChipProgmem(self):
- bufferWidget = self.bufferTab.progmemBuffer
- data = bufferWidget.getRawData()
- if not bufferWidget.isAvailable() or not data:
- QMessageBox.critical(self, "No program memory",
- "No program memory available")
- return False
- return self.runOperation(Operation(Operation.OP_WRITEPROG,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.writeProgmem(...)", data)))
- def writeChipEeprom(self):
- bufferWidget = self.bufferTab.eepromBuffer
- data = bufferWidget.getRawData()
- if not bufferWidget.isAvailable() or not data:
- QMessageBox.critical(self, "No (E)EPROM memory",
- "No (E)EPROM memory available")
- return False
- return self.runOperation(Operation(Operation.OP_WRITEEPROM,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.writeEEPROM(...)", data)))
- def writeChipFuses(self):
- bufferWidget = self.bufferTab.fuseBuffer
- data = bufferWidget.getRawData()
- if not bufferWidget.isAvailable() or not data:
- QMessageBox.critical(self, "No fuse bits",
- "No fuse bits available")
- return False
- return self.runOperation(Operation(Operation.OP_WRITEFUSE,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.writeFuse(...)", data)))
- def writeChipLockbits(self):
- bufferWidget = self.bufferTab.lockBuffer
- data = bufferWidget.getRawData()
- if not bufferWidget.isAvailable() or not data:
- QMessageBox.critical(self, "No lock bits",
- "No lock bits available")
- return False
- return self.runOperation(Operation(Operation.OP_WRITELOCK,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.writeLockbits(...)", data)))
- def writeRam(self):
- bufferWidget = self.bufferTab.ramBuffer
- data = bufferWidget.getRawData()
- if not bufferWidget.isAvailable() or not data:
- QMessageBox.critical(self, "No RAM memory",
- "No RAM memory available")
- return False
- return self.runOperation(Operation(Operation.OP_WRITERAM,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.writeRAM(...)", data)))
- def runTest(self):
- return self.runOperation(Operation(Operation.OP_TEST,
- HwThread.TASK_GENERICTOPCALL,
- GenericTopCall("top.testChip()")))
- def verifyChip(self):
- return self.verifySelectedMemories() # defaults to "all".
- def verifySelectedMemories(self, verifyProgmem=True,
- verifyEEPROM=True,
- verifyFuse=True,
- verifyRAM=True):
- callChain = self.__makeReadCallChain(
- mayReadSig = False,
- mayReadProgmem = verifyProgmem,
- mayReadEEPROM = verifyEEPROM,
- mayReadFuse = verifyFuse,
- mayReadLockbits = False,
- mayReadRAM = verifyRAM)
- return self.runOperation(Operation(Operation.OP_VERIFY,
- HwThread.TASK_GENERICTOPCALL,
- callChain))
- def guiUpdateEnable(self):
- enable = not self.guiDisable and\
- not self.threadRunning and\
- not self.autorun.isRunning()
- self.menuBar().setEnabled(enable)
- self.topToolBar.setEnabled(enable)
- self.rightToolBar.setEnabled(enable)
- self.autorun.setEnabled(enable)
- self.bufferTab.setEnabled(enable)
- def cancelHardwareTask(self):
- if self.autorun.isRunning():
- self.autorun.abortRun()
- if self.threadRunning:
- self.hwThread.cancelTask()
- def __triggerOperation(self, op, threadMutexIsLocked=False):
- self.hwThread.triggerTask(op.hwTask, op.hwTaskParam,
- id(op), threadMutexIsLocked)
- self.threadRunning = True
- self.runningOps[id(op)] = op
- self.console.setTaskRunning(True)
- self.guiUpdateEnable()
- def __triggerNextOperation(self, threadMutexIsLocked=False):
- assert(self.queuedOps)
- op = self.queuedOps.pop(0)
- self.__triggerOperation(op, threadMutexIsLocked)
- def runOperation(self, op):
- self.queuedOps.append(op)
- if not self.threadRunning:
- self.__triggerNextOperation()
- return True
- def runOperationSync(self, hwTask, taskParameter=None):
- op = Operation(Operation.OP_SYNCHRONOUS,
- hwTask, taskParameter)
- self.runOperation(op)
- while not op.finished:
- QApplication.processEvents(QEventLoop.ProcessEventsFlag.AllEvents, 50)
- self.runningOps.pop(id(op))
- return (op.failed, op.returnValue)
- def setOperationFinished(self, failed):
- self.console.setTaskRunning(running=False, success=not failed)
- if self.autorun.isRunning():
- if failed:
- QTimer.singleShot(0, self.autorun.abortRunWithMessage)
- else:
- QTimer.singleShot(0, self.autorun.runNextAction)
- def __hardwareTaskFinished(self, op):
- if op.op == Operation.OP_SYNCHRONOUS:
- self.console.setTaskRunning(running=False, success=True)
- return
- self.runningOps.pop(id(op))
- if op.failed:
- self.console.showMessage("[op %d task %d failed] %s\n" %\
- (op.op, op.hwTask, str(op.returnValue)),
- bold=True)
- self.setOperationFinished(failed=True)
- return
- # Task succeed
- if op.op == Operation.OP_INITCHIP:
- assert(op.hwTask == HwThread.TASK_INITCHIP)
- (self.chip, asciiArtLayout) = op.returnValue
- self.bufferTab.setupBuffers(self.chip.getSupportFlags())
- self.bufferTab.infoBuffer.clear()
- self.bufferTab.infoBuffer.setupDescription(self.currentChipDescription)
- self.bufferTab.infoBuffer.setChipLayout(asciiArtLayout)
- self.autorun.setup(self.chip.getSupportFlags())
- self.topToolBar.setup(self.chip.getSupportFlags())
- self.rightToolBar.setup(self.chip.getSupportFlags())
- self.setupRunMenu(self.chip.getSupportFlags())
- elif op.op == Operation.OP_SHUTDOWN:
- pass # Nothing to do
- elif op.op == Operation.OP_READALL:
- assert(op.hwTask == HwThread.TASK_GENERICTOPCALL)
- topCallList = op.returnValue
- images = self.__readCallChain_GetResultImages(topCallList)
- self.bufferTab.setupBuffers(self.chip.getSupportFlags())
- self.bufferTab.loadBuffers(images)
- try:
- self.bufferTab.infoBuffer.setChipSignature(images["signature"])
- except KeyError:
- pass
- elif op.op == Operation.OP_ERASE:
- pass # Nothing to do
- elif op.op == Operation.OP_WRITEPROG:
- pass # Nothing to do
- elif op.op == Operation.OP_WRITEEPROM:
- pass # Nothing to do
- elif op.op == Operation.OP_WRITEFUSE:
- pass # Nothing to do
- elif op.op == Operation.OP_WRITELOCK:
- pass # Nothing to do
- elif op.op == Operation.OP_WRITERAM:
- pass # Nothing to do
- elif op.op == Operation.OP_TEST:
- self.console.showMessage("Unit-test success. The chip seems to be OK.\n",
- bold=True)
- elif op.op == Operation.OP_VERIFY:
- assert(op.hwTask == HwThread.TASK_GENERICTOPCALL)
- topCallList = op.returnValue
- images = self.__readCallChain_GetResultImages(topCallList)
- op.failed = not self.bufferTab.verifyBuffers(images)
- elif op.op == Operation.OP_RAWCOMMAND:
- self.console.showMessage("Successfully sent raw command\n", bold=True)
- else:
- print("ERROR: No handler for op %d task %d" % (op.op, op.hwTask))
- self.setOperationFinished(op.failed)
- def hardwareTaskFinished(self, opaqueId, failed, returnValue):
- if opaqueId is None:
- return
- op = self.runningOps[opaqueId]
- op.returnValue = returnValue
- op.failed = failed
- self.threadRunning = False
- self.__hardwareTaskFinished(op)
- op.op = Operation.OP_NONE
- op.finished = True
- if self.queuedOps:
- self.__triggerNextOperation(threadMutexIsLocked=True)
- return
- self.guiUpdateEnable()
- def main():
- app = QApplication(sys.argv)
- mainwnd = MainWindow()
- mainwnd.show()
- mainwnd.resize(int(mainwnd.width() * 1.6),
- int(mainwnd.height() * 1.2))
- return app.exec()
- if __name__ == "__main__":
- sys.exit(main())
|