HwCfgSelectorImpl.py

Go to the documentation of this file.

#!/usr/local/bin/python
#
#                               Copyright 2005
#                                     by
#                        The Board of Trustees of the
#                     Leland Stanford Junior University.
#                            All rights reserved.
#

__facility__ = "Online"
__abstract__ = "Hardware Configuration Selector"
__author__   = "S. Tuvi <STuvi@SLAC.Stanford.edu>"
__date__     = "2005/09/15 23:45:02"
__updated__  = "$Date: 2006/04/07 01:17:13 $"
__version__  = "$Revision: 1.3 $"
__release__  = "$Name: HEAD $"
__credits__  = "SLAC"

import LICOS.copyright_SLAC

from qt import *
import logging as log
import time
import sys
import types

from ConfigParser import ConfigParser, NoOptionError

from LICOS.scriptEngine.HwCfgSelector         import HwCfgSelector
from LICOS.util.gOptions                      import Options

class HwCfgSelectorImpl(HwCfgSelector):
  """!\brief Hardware Configuration Selector.

  """

  CFG_PREFIX  = 'lat-config'
  CROSS_STRAP = ['AA','AB','BA','BB']
  MAX_CONFIG  = 9

  def __init__(self, common, parent = None, name = None, fl = 0):
    """!\brief HwCfgSelectorImpl constructor.

    \param common    ScriptEngineCommon instance
    \param parent    The parent GUI (typically the Run Control GUI)
    \param name      Name of the GUI
    \param fl        Qt flags
    """
    HwCfgSelector.__init__(self, parent, name, fl)
    self.connect(self.buttonOK, SIGNAL("clicked()"), self.screenToPrefs)
    self.connect(self.buttonCancel, SIGNAL("clicked()"), self.cancel)
    self.connect(self.bgPresetConfig, SIGNAL("clicked(int)"), self.configToScreen)
    self.connect(self.checkUnlock, SIGNAL("clicked()"), self.allowDisallow)
    self.connect(self.bgSiuSelect, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgDaqSelect, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgEpuAselect, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgEpuBselect, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgGasuFeed, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgGasuPriRed, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgSiuPriRed, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgPduPriRed, SIGNAL("clicked(int)"), self.updatePresetConfig)
    self.connect(self.bgAcdHvbs, SIGNAL("clicked(int)"), self.updatePresetConfig)
    if type(common) is types.StringType:
      # For debugging purposes
      self.__vscConfig = common
      self.__common = common
      self.initialize()
    else:
      self.__vscConfig = common.options().vscConfig
      self.__common = common
      self.initialize()
      self.prefsToScreen()

  def initialize(self):
    config = ConfigParser()
    config.readfp(file(self.__vscConfig))
    self.setButtonGroup(self.bgPresetConfig, 0)
    self.disallowChanges()
    for section in config.sections():
      if section.startswith(self.CFG_PREFIX):
        cfgId = int(section[len(self.CFG_PREFIX):])
        button = self.bgPresetConfig.find(cfgId-1)
        button.setEnabled(1)
    self.__config = config

  def allowDisallow(self):
    if self.checkUnlock.isOn():
      self.allowChanges()
    else:
      self.disallowChanges()

  def allowChanges(self):
    self.setButtonGroup(self.bgSiuSelect, 1)
    self.setButtonGroup(self.bgDaqSelect, 1)
    self.setButtonGroup(self.bgEpuAselect, 1)
    self.setButtonGroup(self.bgEpuBselect, 1)
    self.setButtonGroup(self.bgGasuFeed, 1)
    self.setButtonGroup(self.bgGasuPriRed, 1)
    self.setButtonGroup(self.bgSiuPriRed, 1)
    self.setButtonGroup(self.bgPduPriRed, 1)
    self.setButtonGroup(self.bgAcdHvbs, 1)

  def disallowChanges(self):
    self.setButtonGroup(self.bgSiuSelect, 0)
    self.setButtonGroup(self.bgDaqSelect, 0)
    self.setButtonGroup(self.bgEpuAselect, 0)
    self.setButtonGroup(self.bgEpuBselect, 0)
    self.setButtonGroup(self.bgGasuFeed, 0)
    self.setButtonGroup(self.bgGasuPriRed, 0)
    self.setButtonGroup(self.bgSiuPriRed, 0)
    self.setButtonGroup(self.bgPduPriRed, 0)
    self.setButtonGroup(self.bgAcdHvbs, 0)

  def setButtonGroup(self, buttonGroup, enable):
    for i in range(buttonGroup.count()):
      button = buttonGroup.find(i)
      button.setEnabled(enable)

  def configToScreen(self, id):
    section = "%s%d" % (self.CFG_PREFIX, id+1)
    siuSelect = self.__config.get(section, 'siuSelect')
    self.bgSiuSelect.find(self.CROSS_STRAP.index(siuSelect)).setOn(1)
    daqSelect = self.__config.get(section, 'daqSelect')
    self.bgDaqSelect.find(self.CROSS_STRAP.index(daqSelect)).setOn(1)
    siuIdPhysical = int(self.__config.get(section, 'siuIdPhysical'))
    self.bgSiuPriRed.find(siuIdPhysical ^ 1).setOn(1)
    pduId = int(self.__config.get(section, 'pduId'))
    self.bgPduPriRed.find(pduId).setOn(1)
    primaryFeed = int(self.__config.get(section, 'primaryFeed'))
    self.bgGasuFeed.find(primaryFeed ^ 1).setOn(1)
    gasuPriRed = int(self.__config.get(section, 'primaryGasu'))
    self.bgGasuPriRed.find(gasuPriRed ^ 1).setOn(1)
    epuAid = int(self.__config.get(section, 'firstEpuId'))
    self.bgEpuAselect.find(epuAid).setOn(1)
    epuBid = int(self.__config.get(section, 'secondEpuId'))
    self.bgEpuBselect.find(epuBid).setOn(1)
    acdHvbs = int(self.__config.get(section, 'hvbs'))
    self.bgAcdHvbs.find(acdHvbs-1).setOn(1)

  def prefsToScreen(self):
    if 'hwConfig' in self.__common.preferences():
      hwConfig = self.__common.preferences()['hwConfig']
      self.bgSiuSelect.find(self.CROSS_STRAP.index(hwConfig['siuSelect'])).setOn(1)
      self.bgDaqSelect.find(self.CROSS_STRAP.index(hwConfig['daqSelect'])).setOn(1)
      self.bgSiuPriRed.find(hwConfig['siuIdPhysical'] ^ 1).setOn(1)
      self.bgPduPriRed.find(hwConfig['pduId']).setOn(1)
      self.bgGasuFeed.find(hwConfig['primaryFeed'] ^ 1).setOn(1)
      self.bgGasuPriRed.find(hwConfig['primaryGasu'] ^ 1).setOn(1)
      self.bgEpuAselect.find(hwConfig['firstEpuId']).setOn(1)
      self.bgEpuBselect.find(hwConfig['secondEpuId']).setOn(1)
      self.bgAcdHvbs.find(hwConfig['hvbs'] - 1).setOn(1)
      self.updatePresetConfig()

  def matchStandardConfig(self):
    for id in range(1, self.MAX_CONFIG+1):
      match = True
      section = "%s%d" % (self.CFG_PREFIX, id)
      siuSelect = self.CROSS_STRAP.index(self.__config.get(section, 'siuSelect'))
      match &= (self.bgSiuSelect.selectedId() == siuSelect)
      daqSelect = self.CROSS_STRAP.index(self.__config.get(section, 'daqSelect'))
      match &= (self.bgDaqSelect.selectedId() == daqSelect)
      siuIdPhysical = int(self.__config.get(section, 'siuIdPhysical')) ^ 1
      match &= (self.bgSiuPriRed.selectedId() == siuIdPhysical)
      pduId = int(self.__config.get(section, 'pduId'))
      match &= (self.bgPduPriRed.selectedId() == pduId)
      primaryFeed = int(self.__config.get(section, 'primaryFeed')) ^ 1
      match &= (self.bgGasuFeed.selectedId() == primaryFeed)
      primaryGasu = int(self.__config.get(section, 'primaryGasu')) ^ 1
      match &= (self.bgGasuPriRed.selectedId() == primaryGasu)
      firstEpuId = int(self.__config.get(section, 'firstEpuId'))
      match &= (self.bgEpuAselect.selectedId() == firstEpuId)
      secondEpuId = int(self.__config.get(section, 'secondEpuId'))
      match &= (self.bgEpuBselect.selectedId() == secondEpuId)
      acdHvbs = int(self.__config.get(section, 'hvbs')) - 1
      match &= (self.bgAcdHvbs.selectedId() == acdHvbs)
      if match:
        return id

  def updatePresetConfig(self):
    cfgId = self.matchStandardConfig()
    if cfgId is None:
      for i in range(self.MAX_CONFIG):
        self.bgPresetConfig.find(i).setOn(0)
    else:
      self.bgPresetConfig.find(cfgId-1).setOn(1)

  def screenToPrefs(self):
    hwConfig = {}
    siuSelect = self.bgSiuSelect.selectedId()
    if siuSelect == -1:
      QMessageBox.critical(self, "Error",
                            "SIU cross-strapping not selected")
      return
    hwConfig['siuSelect'] = self.CROSS_STRAP[siuSelect]
    daqSelect = self.bgDaqSelect.selectedId()
    if daqSelect == -1:
      QMessageBox.critical(self, "Error",
                            "DAQ cross-strapping not selected")
      return
    hwConfig['daqSelect'] = self.CROSS_STRAP[daqSelect]
    siuIdPhysical = self.bgSiuPriRed.selectedId()
    if siuIdPhysical == -1:
      QMessageBox.critical(self, "Error",
                            "Primary or Redundant SIU not selected")
      return
    hwConfig['siuIdPhysical'] = siuIdPhysical ^ 1
    pduId = self.bgPduPriRed.selectedId()
    if pduId == -1:
      QMessageBox.critical(self, "Error",
                            "Primary or Redundant PDU not selected")
      return
    hwConfig['pduId'] = pduId
    primaryFeed = self.bgGasuFeed.selectedId()
    if primaryFeed == -1:
      QMessageBox.critical(self, "Error",
                            "Primary or Redundant GASU feed not selected")
      return
    hwConfig['primaryFeed'] = primaryFeed ^ 1
    primaryGasu = self.bgGasuPriRed.selectedId()
    if primaryGasu == -1:
      QMessageBox.critical(self, "Error",
                            "Primary or Redundant GASU not selected")
      return
    hwConfig['primaryGasu'] = primaryGasu ^ 1
    firstEpuId = self.bgEpuAselect.selectedId()
    if firstEpuId == -1:
      QMessageBox.critical(self, "Error",
                            "EPU-A not selected")
      return
    hwConfig['firstEpuId'] = firstEpuId
    secondEpuId = self.bgEpuBselect.selectedId()
    if secondEpuId == -1:
      QMessageBox.critical(self, "Error",
                            "EPU-B not selected")
      return
    hwConfig['secondEpuId'] = secondEpuId
    hvbs = self.bgAcdHvbs.selectedId()
    if hvbs == -1:
      QMessageBox.critical(self, "Error",
                            "ACD HVBS not selected")
      return
    hwConfig['hvbs'] = hvbs + 1
    hvbs = self.bgAcdHvbs.selectedId()
    preset = self.bgPresetConfig.selectedId()
    if preset == -1:
      result = QMessageBox.warning(self, "Warning",
                          "You have selected a non-standard configuration.\n" + \
                          "Are you sure you want to continue?",
                          "&Yes", "&No")
      if result == 1:
        return
    else:
      preset += 1
    hwConfig['preset'] = preset

    if type(self.__common) is not types.StringType:
      self.__common.prefMan().setPref('hwConfig', hwConfig)
    else:
      print hwConfig
    QDialog.done(self, QDialog.Accepted)

  def cancel(self):
    """!\brief Cancel button is clicked.

    """
    QDialog.done(self, QDialog.Rejected)


def usage():
  """!\brief Output command line usage.

  """
  return """

HwCfgSelectorImpl Usage:

Mandatory options:
--config    specify the VSC config file

Example:
$ HwCfgSelectorImpl --config vsc_mcr.cfg

"""

if __name__ == "__main__":
  log.getLogger().setLevel(log.INFO)

  options = Options(['config'])
  try:
    options.parse()
  except Exception, msg:
    options.usage(usage())
    raise Exception, msg

  a = QApplication(sys.argv)
  dm = HwCfgSelectorImpl(options.config)
  a.setMainWidget(dm)
  dm.show()
  a.exec_loop()

---