Teacup_Firmware/configtool/heaterspage.py

import wx
from configtool.page import Page
from configtool.data import pinNames, BSIZESMALL
from configtool.heaterlist import HeaterList
from configtool.addheaterdlg import AddHeaterDlg


class HeatersPage(wx.Panel, Page):
    def __init__(self, parent, nb, idPg, font):
        wx.Panel.__init__(self, nb, wx.ID_ANY)
        Page.__init__(self, font)
        self.parent = parent
        self.font = font
        self.id = idPg

        self.labels = {"FORCE_SOFTWARE_PWM": "Force software PWM"}

        sz = wx.GridBagSizer()
        sz.Add((30, 30), pos=(0, 0))

        self.heaters = []
        self.validPins = pinNames

        self.lb = HeaterList(self, font)
        sz.Add(self.lb, pos=(1, 1))
        sz.Add((20, 20), pos=(1, 2))

        k = "FORCE_SOFTWARE_PWM"
        cb = self.addCheckBox(k, self.onCheckBox)

        sz.Add(cb, pos=(2, 1))
        sz.Add((20, 20), pos=(2, 2))

        bsz = wx.BoxSizer(wx.VERTICAL)

        self.bAdd = wx.Button(self, wx.ID_ANY, "Add", size=BSIZESMALL)
        self.bAdd.SetBackgroundColour(self.deco.getBackgroundColour())
        self.bAdd.SetFont(font)
        self.Bind(wx.EVT_BUTTON, self.doAdd, self.bAdd)
        self.bAdd.SetToolTip("Add a heater to the configuration.")
        bsz.Add(self.bAdd)

        bsz.Add((10, 10))

        self.bModify = wx.Button(self, wx.ID_ANY, "Modify", size=BSIZESMALL)
        self.bModify.SetBackgroundColour(self.deco.getBackgroundColour())
        self.bModify.SetFont(font)
        self.bModify.Enable(False)
        self.Bind(wx.EVT_BUTTON, self.doModify, self.bModify)
        self.bModify.SetToolTip("Modify the selected heater.")
        bsz.Add(self.bModify)

        bsz.Add((10, 10))

        self.bDelete = wx.Button(self, wx.ID_ANY, "Delete", size=BSIZESMALL)
        self.bDelete.SetBackgroundColour(self.deco.getBackgroundColour())
        self.bDelete.SetFont(font)
        self.bDelete.Enable(False)
        self.Bind(wx.EVT_BUTTON, self.doDelete, self.bDelete)
        self.bDelete.SetToolTip("Remove the selected heater from the " "configuration.")
        bsz.Add(self.bDelete)

        sz.Add(bsz, pos=(1, 3))

        self.SetSizer(sz)
        self.enableAll(False)

    def enableAll(self, flag=True):
        self.bAdd.Enable(flag)
        Page.enableAll(self, flag)

    def setItemSelected(self, n):
        self.selection = n
        if n is None:
            self.bDelete.Enable(False)
            self.bModify.Enable(False)
        else:
            self.bDelete.Enable(True)
            self.bModify.Enable(True)

    def getFreePins(self):
        freePins = [] + self.validPins
        usedPins = []
        for s in self.heaters:
            usedPins.append(s[1])

        for p in usedPins:
            if p in freePins:
                freePins.remove(p)

        return freePins

    def doAdd(self, evt):
        nm = []
        for s in self.heaters:
            nm.append(s[0])

        dlg = AddHeaterDlg(self, nm, self.getFreePins(), self.font)
        rc = dlg.ShowModal()
        if rc == wx.ID_OK:
            ht = dlg.getValues()

        dlg.Destroy()

        if rc != wx.ID_OK:
            return

        self.heaters.append(ht)
        self.lb.updateList(self.heaters)
        self.validateTable()
        self.parent.setHeaters(self.heaters)
        self.assertModified(True)

    def doModify(self, evt):
        if self.selection is None:
            return

        nm = []
        for s in self.heaters:
            nm.append(s[0])

        h = self.heaters[self.selection]

        dlg = AddHeaterDlg(
            self,
            nm,
            [h[1]] + self.getFreePins(),
            self.font,
            name=h[0],
            pin=h[1],
            invert=h[2],
            pwm=h[3],
            max_pwm=h[4],
        )
        rc = dlg.ShowModal()
        if rc == wx.ID_OK:
            ht = dlg.getValues()

        dlg.Destroy()

        if rc != wx.ID_OK:
            return

        self.heaters[self.selection] = ht
        self.lb.updateList(self.heaters)
        self.validateTable()
        self.parent.setHeaters(self.heaters)
        self.assertModified(True)

    def doDelete(self, evt):
        if self.selection is None:
            return

        self.assertModified(True)

        del self.heaters[self.selection]
        self.lb.updateList(self.heaters)
        self.validateTable()
        self.parent.setHeaters(self.heaters)
        self.assertModified(True)

    def setHeaters(self, heaters):
        self.heaters = heaters
        self.lb.updateList(self.heaters)
        self.validateTable()
        self.parent.setHeaters(self.heaters)

    def setCandidatePins(self, plist):
        if not plist or len(plist) == 0:
            self.validPins = pinNames
        else:
            self.validPins = plist

        self.validateTable()

    def heaterNames(self):
        heaterNames = []
        for heater in self.heaters:
            heaterNames.append(heater[0])

        return heaterNames

    def validateTable(self):
        self.lb.setTableValidity(True)
        self.setFieldValidity("HEATERLIST", True)
        for i in range(len(self.heaters)):
            if self.heaters[i][1] not in self.validPins:
                self.lb.setRowValidity(i, False)
                self.setFieldValidity("HEATERLIST", False)

    def setHelpText(self, ht):
        Page.setHelpText(self, ht)

        k = "DEFINE_HEATER"
        if k in ht.keys():
            self.bAdd.SetToolTip(ht[k])