Teacup_Firmware/configtool/page.py

import wx

from configtool.data import reInteger, reFloat


class Page:
  def __init__(self, font):
    self.modified = False
    self.valid = True
    self.fieldValid = {}
    self.textControls = {}
    self.checkBoxes = {}
    self.radioButtons = {}
    self.choices = {}
    self.font = font


  def enableAll(self, flag = True):
    for c in self.textControls.keys():
      self.textControls[c].Enable(flag)
    for c in self.checkBoxes.keys():
      self.checkBoxes[c].Enable(flag)
    for c in self.radioButtons.keys():
      self.radioButtons[c].Enable(flag)
    for c in self.choices.keys():
      self.choices[c].Enable(flag)

  def addTextCtrl(self, name, labelWidth, validator):
    lsz = wx.BoxSizer(wx.HORIZONTAL)
    st = wx.StaticText(self, wx.ID_ANY, self.labels[name] + " ",
                       size = (labelWidth, -1), style = wx.ALIGN_RIGHT)
    st.SetFont(self.font)
    lsz.Add(st)

    tc = wx.TextCtrl(self, wx.ID_ANY, "", style = wx.TE_RIGHT, name = name)
    tc.SetFont(self.font)
    self.fieldValid[name] = True
    tc.Bind(wx.EVT_TEXT, validator)
    self.textControls[name] = tc
    lsz.Add(tc)

    return lsz

  def addCheckBox(self, name, validator):
    if name in self.labels.keys():
      lbl = self.labels[name]
    else:
      lbl = name
    cb = wx.CheckBox(self, wx.ID_ANY, lbl)
    cb.SetFont(self.font)
    cb.Bind(wx.EVT_CHECKBOX, validator)
    self.checkBoxes[name] = cb

    return cb

  def addRadioButton(self, name, style, validator):
    rb = wx.RadioButton(self, wx.ID_ANY, self.labels[name], style = style)
    rb.SetFont(self.font)
    self.Bind(wx.EVT_RADIOBUTTON, validator, rb)
    self.radioButtons[name] = rb

    return rb

  def addChoice(self, name, choices, selection, labelWidth, validator):
    lsz = wx.BoxSizer(wx.HORIZONTAL)
    st = wx.StaticText(self, wx.ID_ANY, self.labels[name],
                       size = (labelWidth, -1), style = wx.ALIGN_RIGHT)
    st.SetFont(self.font)
    lsz.Add(st)

    ch = wx.Choice(self, wx.ID_ANY, choices = choices, name = name)
    ch.SetFont(self.font)
    ch.Bind(wx.EVT_CHOICE, validator)
    ch.SetSelection(selection)
    lsz.Add(ch)
    self.choices[name] = ch

    return lsz

  def addPinChoice(self, name, choiceVal, pins, allowBlank , labelWidth):
    lsz = wx.BoxSizer(wx.HORIZONTAL)
    st = wx.StaticText(self, wx.ID_ANY, self.labels[name],
                       size = (labelWidth, -1), style = wx.ALIGN_RIGHT)
    st.SetFont(self.font)
    lsz.Add(st)

    if allowBlank:
      opts = ["-"] + pins
    else:
      opts = pins

    ch = wx.Choice(self, wx.ID_ANY, choices = opts, name = name,
                   style = wx.CB_SORT)
    ch.SetFont(self.font)
    ch.Bind(wx.EVT_CHOICE, self.onChoice)
    self.choices[name] = ch
    try:
      sv = self.pinNames.index(choiceVal)
    except:
      sv = 0
    ch.SetSelection(sv)
    lsz.Add(ch)

    return lsz

  def setChoice(self, name, cfgValues, default):
    if name in cfgValues.keys():
      bv = cfgValues[name]
    else:
      bv = default

    s = self.choices[name].FindString(bv)
    if s < 0:
      s = self.choices[name].FindString(default)
      if s < 0:
        s = 0

    self.choices[name].SetSelection(s)

  def onTextCtrlInteger(self, evt):
    self.assertModified(True)
    tc = evt.GetEventObject()
    name = tc.GetName()
    w = tc.GetValue().strip()
    if w == "":
      valid = True
    else:
      m = reInteger.match(w)
      if m:
        valid = True
      else:
        valid = False

    self.setFieldValidity(name, valid)

    if valid:
      tc.SetBackgroundColour(wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
    else:
      tc.SetBackgroundColour("pink")
    tc.Refresh()
    evt.Skip()

  def onTextCtrlFloat(self, evt):
    self.assertModified(True)
    tc = evt.GetEventObject()
    name = tc.GetName()
    w = tc.GetValue().strip()
    if w == "":
      valid = True
    else:
      m = reFloat.match(w)
      if m:
        valid = True
      else:
        valid = False

    self.setFieldValidity(name, valid)

    if valid:
      tc.SetBackgroundColour(wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
    else:
      tc.SetBackgroundColour("pink")
    tc.Refresh()
    evt.Skip()

  def onTextCtrlPin(self, evt):
    self.assertModified(True)
    tc = evt.GetEventObject()
    self.validatePin(tc)
    evt.Skip()

  def onTextCtrl(self, evt):
    self.assertModified(True)
    evt.Skip()

  def onChoice(self, evt):
    self.assertModified(True)
    evt.Skip()

  def onCheckBox(self, evt):
    self.assertModified(True)
    evt.Skip()

  def setHelpText(self, ht):
    for k in self.textControls.keys():
      if k in ht.keys():
        self.textControls[k].SetToolTipString(ht[k])

    for k in self.checkBoxes.keys():
      if k in ht.keys():
        self.checkBoxes[k].SetToolTipString(ht[k])

    for k in self.radioButtons.keys():
      if k in ht.keys():
        self.radioButtons[k].SetToolTipString(ht[k])

    for k in self.choices.keys():
      if k in ht.keys():
        self.choices[k].SetToolTipString(ht[k])

  def getValues(self):
    self.assertModified(False)
    result = {}
    for k in self.checkBoxes.keys():
      cb = self.checkBoxes[k]
      result[k] = cb.IsChecked()

    for k in self.textControls.keys():
      v = self.textControls[k].GetValue()
      result[k] = v

    for k in self.radioButtons.keys():
      result[k] = self.radioButtons[k].GetValue()

    for k in self.choices.keys():
      v = self.choices[k].GetSelection()
      result[k] = self.choices[k].GetString(v)

    return result

  def assertModified(self, flag):
    if flag != self.modified:
      self.parent.assertModified(self.id, flag)
      self.modified = flag

  def setFieldValidity(self, name, flag):
    self.fieldValid[name] = flag

    pgValid = True
    for k in self.fieldValid.keys():
      if not self.fieldValid[k]:
        pgValid = False
        break

    self.assertValid(pgValid)

  def assertValid(self, flag):
    if flag != self.valid:
      self.parent.assertValid(self.id, flag)
      self.valid = flag