diff options
Diffstat (limited to 'df10ch_setup_pkg/device_dlg.py')
| -rw-r--r-- | df10ch_setup_pkg/device_dlg.py | 428 |
1 files changed, 428 insertions, 0 deletions
diff --git a/df10ch_setup_pkg/device_dlg.py b/df10ch_setup_pkg/device_dlg.py new file mode 100644 index 0000000..a473117 --- /dev/null +++ b/df10ch_setup_pkg/device_dlg.py @@ -0,0 +1,428 @@ +# +# Copyright (C) 2010 Andreas Auras +# +# This file is part of the DF10CH Atmolight controller project. +# +# DF10CH Atmolight controller 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. +# +# DF10CH Atmolight controller 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, USA +# +# This file is part of the DF10CH setup program +# + +from Tkinter import * +import tkFont +import tkMessageBox +import tkFileDialog +import usb +import pickle +import string +import time + +import device_drv +import firmware + +class DeviceDialog: + def __init__(self, layoutDlg, master=None, **args): + self.layoutDlg = layoutDlg + self.selectedConfig = None + self.selectedDeviceIdx = -1 + + root = Frame(master, **args) + self.root = root + root.bind("<Map>", self.cbSheetSelected) + + Label(root, text="Device Control", font=tkFont.Font(weight="bold")).grid(row=0, column=0, columnspan=7, padx=5, pady=5) + + self.varDeviceList = StringVar() + self.lbDevices = Listbox(root, selectmode=SINGLE, activestyle=NONE, width=20, height=8, listvariable=self.varDeviceList) + self.lbDevices.grid(row=1, column=0, columnspan=2, rowspan=4, padx=5, pady=5, sticky=N+S+E+W) + self.lbDevices.bind("<ButtonRelease-1>", self.cbSelectDevice) + + self.varVersion = StringVar() + Label(root, text="Version:").grid(row=1, column=2, sticky=E) + self.etVersion = Label(root, textvariable=self.varVersion, anchor=W, relief=SUNKEN) + self.etVersion.grid(row=1, column=3, columnspan=4, padx=5, pady=5, sticky=W+E) + + self.varPWMRes = StringVar() + Label(root, text="PWM Resolution:").grid(row=2, column=2, sticky=E) + self.etPWMRes = Label(root, textvariable=self.varPWMRes, anchor=W, relief=SUNKEN) + self.etPWMRes.grid(row=2, column=3, columnspan=4, padx=5, pady=5, sticky=W+E) + + self.varPWMFreq = IntVar() + Label(root, text="PWM Frequency:").grid(row=3, column=2, sticky=E) + self.varPWMFreq.set(device_drv.MIN_PWM_FREQ) + self.scPWMFreq = Scale(root, length=250, from_=device_drv.MIN_PWM_FREQ, to=device_drv.MAX_PWM_FREQ, resolution=1, tickinterval=50, orient=HORIZONTAL, variable=self.varPWMFreq, command=self.cbSetPWMFreq) + self.scPWMFreq.grid(row=3, column=3, columnspan=4, padx=5, pady=5, sticky="W") + + self.varCommonBright = IntVar() + Label(root, text="Common Brightness:").grid(row=4, column=2, sticky=E) + self.varCommonBright.set(0) + self.scCommonBright = Scale(root, length=250, from_=0, to=255, resolution=1, tickinterval=50, orient=HORIZONTAL, variable=self.varCommonBright, command=self.cbSetCommonBrightness) + self.scCommonBright.grid(row=4, column=3, columnspan=4, padx=5, pady=5, sticky=W) + + self.btScanDevices = Button(root, text="Scan Devices", command=self.cbScanDevices) + self.btScanDevices.grid(row=5, column=0, padx=5, pady=5, ipadx=5) + + self.btResetSetup = Button(root, text="Firmware update", command=self.cbFirmwareUpdate) + self.btResetSetup.grid(row=5, column=1, padx=5, pady=5, ipadx=5) + + self.btStoreSetup = Button(root, text="Backup", command=self.cbBackupSetup) + self.btStoreSetup.grid(row=5, column=2, padx=5, pady=5, ipadx=5, sticky=E) + + self.btStoreSetup = Button(root, text="Restore", command=self.cbRestoreSetup) + self.btStoreSetup.grid(row=5, column=3, padx=5, pady=5, ipadx=5, sticky=W) + + self.btResetSetup = Button(root, text="Reset Setup", command=self.cbResetSetup) + self.btResetSetup.grid(row=5, column=4, padx=5, pady=5, ipadx=5, sticky=E) + + self.btStoreSetup = Button(root, text="Store Setup", command=self.cbStoreSetup) + self.btStoreSetup.grid(row=5, column=5, padx=5, pady=5, ipadx=5, sticky=W) + + self.varStatus = StringVar() + self.lbStatus = Label(root, textvariable=self.varStatus, anchor=W, relief=RIDGE) + self.lbStatus.grid(row=6, column=0, columnspan=6, padx=0, pady=0, sticky=W+E) + + + def cbSheetSelected(self, event): + if self.selectedConfig: + self.loadDeviceValues() + + def cbFirmwareUpdate(self): + if len(device_drv.DeviceList): + self.firmwareUpdate() + + def cbResetSetup(self): + if len(device_drv.ConfigMap): + self.resetSetup() + if self.selectedConfig: + self.loadDeviceValues() + + def cbStoreSetup(self): + if len(device_drv.ConfigMap): + self.storeSetup() + if self.selectedConfig: + self.loadDeviceValues() + + def cbBackupSetup(self): + if self.selectedConfig: + self.backupSetup() + + def cbRestoreSetup(self): + if self.selectedConfig: + self.restoreSetup() + + def cbSetCommonBrightness(self, val): + if self.selectedConfig: + try: + self.selectedConfig.setCommonBright(int(val)) + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + def cbSetPWMFreq(self, val): + if self.selectedConfig: + try: + self.selectedConfig.setPWMFreq(int(val)) + self.varPWMRes.set(self.selectedConfig.pwmRes) + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + def cbSelectDevice(self, event): + s = self.lbDevices.curselection() + if len(s) == 1: + self.selectDevice(int(s[0])) + + def cbScanDevices(self): + self.scanDevices() + + def scanDevices(self): + deviceList = None + self.selectedConfig = None + self.selectedDeviceIdx = -1 + while not deviceList: + try: + device_drv.LoadConfigs() + if len(device_drv.DeviceList): + deviceList = device_drv.DeviceList + except (device_drv.AtmoControllerError, usb.USBError) as err: + if not tkMessageBox.askretrycancel(self.root.winfo_toplevel().title(), "Scanning for controllers fails:" + err.__str__(), icon=tkMessageBox.ERROR): + return False + else: + if not deviceList: + if not tkMessageBox.askretrycancel(self.root.winfo_toplevel().title(), "No Controllers found!", icon=tkMessageBox.ERROR): + return False + + idList = "" + for dev in deviceList: + if len(idList) > 0: + idList = idList + " " + idList = idList + dev.id + self.varDeviceList.set(idList) + + self.layoutDlg.setLayoutFromConfig() + self.selectDevice(0) + return True + + def selectDevice(self, i): + self.selectedDeviceIdx = i + dev = device_drv.DeviceList[i] + id = dev.id + if id in device_drv.ConfigMap: + config = device_drv.ConfigMap[id] + if self.selectedConfig != config: + self.selectedConfig = config + self.scCommonBright.configure(to=self.selectedConfig.commonPWMRes) + self.loadDeviceValues() + else: + if dev.bootloader_mode(): + s = "Bootloader USB:{0}".format(dev.version) + else: + s = "USB:{0} Bootloader PWM:{1:04X}".format(dev.version, dev.get_pwm_version()) + self.varVersion.set(s) + self.lbDevices.selection_clear(0, END) + self.lbDevices.selection_set(i) + self.lbDevices.see(i) + + def loadDeviceValues(self): + self.varVersion.set(self.selectedConfig.version) + self.varPWMRes.set(self.selectedConfig.pwmRes) + self.varPWMFreq.set(self.selectedConfig.pwmFreq) + self.varCommonBright.set(self.selectedConfig.commonBright) + + def storeAndQuit(self, root): + if len(device_drv.ConfigMap): + if tkMessageBox.askokcancel(self.root.winfo_toplevel().title(), "Store Setup?", icon=tkMessageBox.QUESTION): + self.storeSetup() + self.resetDevices() + device_drv.ReleaseDevices() + root.quit() + + def storeSetup(self): + self.layoutDlg.setConfigFromLayout() + try: + for id in device_drv.ConfigMap.keys(): + config = device_drv.ConfigMap[id] + config.write() + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + def resetSetup(self): + if tkMessageBox.askokcancel(self.root.winfo_toplevel().title(), "Really reset setup of all controllers?", default="cancel", icon=tkMessageBox.QUESTION): + try: + for id in device_drv.ConfigMap.keys(): + config = device_drv.ConfigMap[id] + config.reset() + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + def resetDevices(self): + try: + for id in device_drv.ConfigMap.keys(): + config = device_drv.ConfigMap[id] + config.ctrl.reset_pwm_ctrl() + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + def backupSetup(self): + self.layoutDlg.setConfigFromLayout() + ctrl = self.selectedConfig.ctrl + initialfile = ctrl.id + initialfile = string.replace(initialfile, "[", "(") + initialfile = string.replace(initialfile, "]", ")") + initialfile = string.replace(initialfile, ",", "_") + initialfile = initialfile + ".dfc" + self.selectedConfig.ctrl = None + file = None + try: + file = tkFileDialog.asksaveasfile("w", title="Backup Configuration for " + self.selectedConfig.id, defaultextension=".dfc", filetypes=[ ("DF10CH Config", "*.dfc") ], initialfile=initialfile) + if file: + pickle.dump(self.selectedConfig, file) + except IOError: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), "Could not save configuration to file!") + finally: + self.selectedConfig.ctrl = ctrl + if file: + file.close() + + def restoreSetup(self): + file = None + try: + file = tkFileDialog.askopenfile("r", title="Restore Configuration for " + self.selectedConfig.id, defaultextension=".dfc", filetypes=[ ("DF10CH Config", "*.dfc") ]) + if file: + config = pickle.load(file) + else: + return + except IOError: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), "Could not read configuration from file!") + return + finally: + if file: + file.close() + + config.id = self.selectedConfig.id + config.ctrl = self.selectedConfig.ctrl + + try: + config.ctrl.set_common_brightness(config.commonBright) + config.ctrl.set_pwm_freq(config.pwmFreq) + config.write() + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + device_drv.ConfigMap[config.id] = config + self.selectedConfig = config + self.layoutDlg.setLayoutFromConfig() + self.scCommonBright.configure(to=self.selectedConfig.commonPWMRes) + self.loadDeviceValues() + + def firmwareUpdate(self): + try: + filename = tkFileDialog.askopenfilename(title="Select firmware", defaultextension=".dff", filetypes=[ ("DF10CH Firmware", "*.dff") ]) + if not filename or len(filename) == 0: + return + fw = firmware.FlashMem(filename, 64, True) + except (IOError, firmware.FirmwareFlashError) as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + return + + target = fw.target + if target != "DF10CH-USB" and target != "DF10CH-PWM": + tkMessageBox.showerror(self.root.winfo_toplevel().title(), "Unknown firmware target '{0}'!".format(target)) + return + + if target == "DF10CH-PWM": + for ctrl in device_drv.DeviceList: + if ctrl.bootloader_mode(): + tkMessageBox.showerror(self.root.winfo_toplevel().title(), "{0}: Firmware of USB controller must be installed first!".format(ctrl.id)) + return + + if not tkMessageBox.askokcancel(self.root.winfo_toplevel().title(), "Target controller: {0}, Firmware version: {1}. Really start firmware update?".format(target, fw.version), default="cancel", icon=tkMessageBox.QUESTION): + return + + self.displayStatus("Start bootloaders...") + + doFlash = True + doRefresh = False + try: + for ctrl in device_drv.DeviceList: + if target == "DF10CH-USB": + if not ctrl.bootloader_mode(): + doRefresh = True + ctrl.start_bootloader() + else: + if not ctrl.get_pwm_bootloader_mode(): + ctrl.start_pwm_ctrl_bootloader() + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + doFlash = False + + if doRefresh: + time.sleep(5.0) + try: + device_drv.FindDevices() + except (device_drv.AtmoControllerError, usb.USBError) as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + doFlash = False + + for ctrl in device_drv.DeviceList: + try: + if target == "DF10CH-USB": + if ctrl.bootloader_mode(): + if doFlash: + pageSize = ctrl.get_flash_page_size() + if pageSize != fw.pageSize: + fw = firmware.FlashMem(filename, pageSize) + firstPage = fw.getPageForAddr(0x0000) + i = 1 + if firstPage: + fp = firmware.FlashPage(0x0000, pageSize) + ctrl.write_flash_page(fp.baseAddr, fp.data) + i = 2 + n = len(fw.pageList) + for fp in fw.pageList: + if fp != firstPage: + self.displayStatus("Flashing USB controller {0}: {1}%".format(ctrl.id, i * 100 / n)) + ctrl.write_flash_page(fp.baseAddr, fp.data) + i = i + 1 + if firstPage: + ctrl.write_flash_page(firstPage.baseAddr, firstPage.data) + time.sleep(1.0) + + i = 1 + for fp in fw.pageList: + self.displayStatus("Verifying USB controller {0}: {1}%".format(ctrl.id, i * 100 / n)) + data = ctrl.read_flash(fp.baseAddr, fp.pageSize) + fp.verify(data) + i = i + 1 + time.sleep(1.0) + else: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), "{0}: Bootloader of USB controller does not start!".format(ctrl.id)) + doFlash = False + else: + if ctrl.get_pwm_bootloader_mode(): + if doFlash: + pageSize = ctrl.get_pwm_flash_page_size() + if pageSize != fw.pageSize: + fw = firmware.FlashMem(filename, pageSize) + firstPage = fw.getPageForAddr(0x0000) + i = 1 + if firstPage: + fp = firmware.FlashPage(0x0000, pageSize) + ctrl.write_pwm_flash_page(fp.baseAddr, fp.data) + i = 2 + n = len(fw.pageList) + for fp in fw.pageList: + if fp != firstPage: + self.displayStatus("Flashing PWM controller {0}: {1}%".format(ctrl.id, i * 100 / n)) + ctrl.write_pwm_flash_page(fp.baseAddr, fp.data) + i = i + 1 + if firstPage: + ctrl.write_pwm_flash_page(firstPage.baseAddr, firstPage.data) + time.sleep(1.0) + + i = 1 + for fp in fw.pageList: + self.displayStatus("Verifying PWM controller {0}: {1}%".format(ctrl.id, i * 100 / n)) + data = ctrl.read_pwm_flash(fp.baseAddr, fp.pageSize) + fp.verify(data) + i = i + 1 + time.sleep(1.0) + else: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), "{0}: Bootloader of PWM controller does not start!".format(ctrl.id)) + doFlash = False + except (device_drv.AtmoControllerError, firmware.FirmwareFlashError) as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + doFlash = False + + doRefresh = False + for ctrl in device_drv.DeviceList: + try: + if target == "DF10CH-USB": + if ctrl.bootloader_mode(): + doRefresh = True + ctrl.start_appl() + else: + if ctrl.get_pwm_bootloader_mode(): + ctrl.reset_pwm_ctrl() + except device_drv.AtmoControllerError as err: + tkMessageBox.showerror(self.root.winfo_toplevel().title(), err.__str__()) + + if doRefresh: + self.displayStatus("Start firmware...") + time.sleep(5.0) + self.displayStatus("") + self.scanDevices() + + def displayStatus(self, msg): + self.varStatus.set(msg) + self.root.update_idletasks() |