Teacup_Firmware/configtool/addsensordlg.py

import wx
from configtool.data import pinNames, BSIZESMALL


class AddSensorDlg(wx.Dialog):
  def __init__(self, parent, names, sl, pins):
    wx.Dialog.__init__(self, parent, wx.ID_ANY, "Add temperature sensor",
                       size = (400, 204))
    self.Bind(wx.EVT_CLOSE, self.onCancel)

    self.names = names

    self.nameValid = False

    sz = wx.BoxSizer(wx.VERTICAL)

    csz = wx.GridBagSizer()
    csz.AddSpacer((10, 10), pos = (0, 0))

    b = wx.StaticBox(self, wx.ID_ANY, "Sensor Type")
    sbox = wx.StaticBoxSizer(b, wx.VERTICAL)
    sbox.AddSpacer((5, 5))
    style = wx.RB_GROUP
    self.rbs = {}
    for k in sl:
      rb = wx.RadioButton(self, wx.ID_ANY, k, style = style)
      self.rbs[k] = rb
      self.Bind(wx.EVT_RADIOBUTTON, self.onSensorType, rb)
      style = 0

      sbox.Add(rb, 1, wx.LEFT + wx.RIGHT, 16)
      sbox.AddSpacer((5, 5))

    csz.Add(sbox, pos = (1, 3))

    vsz = wx.BoxSizer(wx.VERTICAL)
    vsz.AddSpacer((10, 10))

    lsz = wx.BoxSizer(wx.HORIZONTAL)
    st = wx.StaticText(self, wx.ID_ANY, "Sensor Name:", size = (80, -1),
                       style = wx.ALIGN_RIGHT)
    lsz.Add(st)

    self.tcName = wx.TextCtrl(self, wx.ID_ANY, "")
    self.tcName.SetBackgroundColour("pink")
    self.tcName.Bind(wx.EVT_TEXT, self.onNameEntry)
    lsz.Add(self.tcName)
    self.tcName.SetToolTipString("Enter a unique name for this sensor.")

    vsz.Add(lsz)
    vsz.AddSpacer((10, 10))

    lsz = wx.BoxSizer(wx.HORIZONTAL)
    st = wx.StaticText(self, wx.ID_ANY, "Pin:", size = (80, -1),
                       style = wx.ALIGN_RIGHT)
    lsz.Add(st)

    self.choiceList = pinNames
    self.chPin = wx.Choice(self, wx.ID_ANY, choices = pins)
    self.chPin.Bind(wx.EVT_CHOICE, self.onChoice)
    self.chPin.SetSelection(0)
    lsz.Add(self.chPin)
    self.chPin.SetToolTipString("Choose a pin name for this sensor.")

    vsz.Add(lsz)
    vsz.AddSpacer((10, 10))

    lsz = wx.BoxSizer(wx.HORIZONTAL)
    st = wx.StaticText(self, wx.ID_ANY, "Additional:", size = (80, -1),
                       style = wx.ALIGN_RIGHT)
    lsz.Add(st)

    self.tcAddtl = wx.TextCtrl(self, wx.ID_ANY, "")
    self.tcAddtl.Bind(wx.EVT_TEXT, self.onAddtlEntry)
    self.selectSensorType(sl[0])
    lsz.Add(self.tcAddtl)
    self.tcAddtl.SetToolTipString("Enter additional information required "
                                  "by the sensor type.")

    vsz.Add(lsz)
    csz.Add(vsz, pos = (1, 1))

    csz.AddSpacer((10, 10), pos = (1, 4))

    sz.Add(csz)
    sz.AddSpacer((30, 30))

    bsz = wx.BoxSizer(wx.HORIZONTAL)

    self.bSave = wx.Button(self, wx.ID_ANY, "Save", size = BSIZESMALL)
    self.bSave.Bind(wx.EVT_BUTTON, self.onSave)
    bsz.Add(self.bSave)
    self.bSave.Enable(False)

    bsz.AddSpacer(30, 100)

    self.bCancel = wx.Button(self, wx.ID_ANY, "Cancel", size = BSIZESMALL)
    self.bCancel.Bind(wx.EVT_BUTTON, self.onCancel)
    bsz.Add(self.bCancel)

    sz.Add(bsz, flag = wx.ALIGN_CENTER_HORIZONTAL)
    self.SetSizer(sz)

  def onNameEntry(self, evt):
    tc = evt.GetEventObject()
    w = tc.GetValue().strip()
    if w == "":
      self.nameValid = False
    else:
      if w in self.names:
        self.nameValid = False
      else:
        self.nameValid = True

    if self.nameValid:
      tc.SetBackgroundColour(wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
    else:
      tc.SetBackgroundColour("pink")
    tc.Refresh()

    self.checkDlgValidity()
    evt.Skip()

  def checkDlgValidity(self):
    if self.nameValid:
      self.bSave.Enable(True)
    else:
      self.bSave.Enable(False)

  def onAddtlEntry(self, evt):
    evt.Skip()

  def selectSensorType(self, lbl):
    if lbl == 'TT_THERMISTOR':
      self.tcAddtl.Enable(True);
    else:
      self.tcAddtl.Enable(False);

  def onChoice(self, evt):
    pass

  def onSensorType(self, evt):
    rb = evt.GetEventObject()
    label = rb.GetLabel()

    self.selectSensorType(label)

    evt.Skip()

  def getValues(self):
    nm = self.tcName.GetValue()
    pin = pinNames[self.chPin.GetSelection()]
    addtl = self.tcAddtl.GetValue()

    for k in self.rbs:
      if self.rbs[k].GetValue():
        stype = k
        break

    if stype is None:
      stype = "??"

    if stype in ['TT_THERMISTOR']:
      return (nm, stype, pin, addtl)
    else:
      return (nm, stype, pin)


  def onSave(self, evt):
    self.EndModal(wx.ID_OK)

  def onCancel(self, evt):
    self.EndModal(wx.ID_CANCEL)