wizard/BModulePage.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3 #
   4 # Copyright 2009 Develer S.r.l. (http://www.develer.com/)
   5 # All rights reserved.
   6 #
   7 # $Id:$
   8 #
   9 # Author: Lorenzo Berni <duplo@develer.com>
  10 #
  11
  12 import os
  13
  14 from PyQt4.QtGui import *
  15 from BWizardPage import *
  16 import bertos_utils
  17
  18
  19 class BModulePage(BWizardPage):
  20
  21     def __init__(self):
  22         BWizardPage.__init__(self, "module_select.ui")
  23         self.setTitle(self.tr("Configure the BeRTOS modules"))
  24         self._setupUi()
  25         self._connectSignals()
  26
  27     def reloadData(self):
  28         self._setupButtonGroup()
  29         self._loadModuleData()
  30         self._fillModuleTable()
  31
  32     def _setupButtonGroup(self):
  33         self._buttonGroup = QButtonGroup()
  34         self._buttonGroup.setExclusive(False)
  35         self.connect(self._buttonGroup, SIGNAL("buttonClicked(int)"), self._moduleSelectionChanged)
  36
  37     def _loadModuleData(self):
  38         modules = bertos_utils.loadModuleInfosDict(self._projectInfoRetrieve("SOURCES_PATH"))
  39         lists = bertos_utils.loadDefineListsDict(self._projectInfoRetrieve("SOURCES_PATH"))
  40         configurations = {}
  41         for module, informations in modules.items():
  42             configurations[informations["configuration"]] = bertos_utils.loadConfigurationInfos(self._projectInfoRetrieve("SOURCES_PATH") +
  43                                                                                                 "/" + informations["configuration"])
  44         self._projectInfoStore("MODULES", modules)
  45         self._projectInfoStore("LISTS", lists)
  46         self._projectInfoStore("CONFIGURATIONS", configurations)
  47
  48     def _fillModuleTable(self):
  49         modules = self._projectInfoRetrieve("MODULES")
  50         self.pageContent.moduleTable.setRowCount(len(modules))
  51         for index, module in enumerate(modules):
  52             self.pageContent.moduleTable.setItem(index, 1, QTableWidgetItem(module))
  53             checkBox = QCheckBox()
  54             self._buttonGroup.addButton(checkBox, index)
  55             self.pageContent.moduleTable.setCellWidget(index, 0, checkBox)
  56             checkBox.setChecked(modules[module]["enabled"])
  57
  58     def _fillPropertyTable(self):
  59         module = self._currentModule()
  60         configuration = self._projectInfoRetrieve("MODULES")[module]["configuration"]
  61         configurations = self._projectInfoRetrieve("CONFIGURATIONS")[configuration]
  62         self.pageContent.propertyTable.clear()
  63         self.pageContent.propertyTable.setRowCount(len(configurations))
  64         for index, property in enumerate(configurations):
  65             item = QTableWidgetItem(property)
  66             item.setData(Qt.UserRole, qvariant_converter.convertString(property))
  67             self.pageContent.propertyTable.setItem(index, 0, item)
  68             if "type" in configurations[property]["informations"].keys() and configurations[property]["informations"]["type"] == "boolean":
  69                 ## boolean property
  70                 checkBox = QCheckBox()
  71                 self.pageContent.propertyTable.setCellWidget(index, 1, checkBox)
  72                 if configurations[property]["value"] == "1":
  73                     checkBox.setChecked(True)
  74                 else:
  75                     checkBox.setChecked(False)
  76             elif "type" in configurations[property]["informations"].keys() and configurations[property]["informations"]["type"] == "enum":
  77                 ## enum property
  78                 comboBox = QComboBox()
  79                 self.pageContent.propertyTable.setCellWidget(index, 1, comboBox)
  80                 enum = self._projectInfoRetrieve("LISTS")[configurations[property]["informations"]["value_list"]]
  81                 for index, element in enumerate(enum):
  82                     comboBox.addItem(element)
  83                     if element == configurations[property]["value"]:
  84                         comboBox.setCurrentIndex(index)
  85             else:
  86                 ## int, long or undefined type property
  87                 spinBox = QSpinBox()
  88                 self.pageContent.propertyTable.setCellWidget(index, 1, spinBox)
  89                 if "min" in configurations[property]["informations"].keys():
  90                     minimum = int(configurations[property]["informations"]["min"])
  91                 else:
  92                     minimum = -32768
  93                 spinBox.setMinimum(minimum)
  94                 if "max" in configurations[property]["informations"].keys():
  95                     maximum = int(configurations[property]["informations"]["max"])
  96                 else:
  97                     maximum = 32767
  98                 spinBox.setMaximum(maximum)
  99                 if "long" in configurations[property]["informations"].keys() and configurations[property]["informations"]["long"] == "True":
 100                     spinBox.setSuffix("L")
 101                 spinBox.setValue(int(configurations[property]["value"].replace("L", "")))
 102
 103     def _saveData(self, previousRow, previousColumn):
 104         module = self._module(previousRow)
 105         moduleConfigurations = self._configurations(module)
 106         for index in range(self.pageContent.propertyTable.rowCount()):
 107             parameter = qvariant_converter.getString(self.pageContent.propertyTable.item(index, 0).data(Qt.UserRole))
 108             if "type" not in moduleConfigurations[parameter]["informations"].keys() or moduleConfigurations[parameter]["informations"]["type"] == "int":
 109                 moduleConfigurations[parameter]["value"] = str(self.pageContent.propertyTable.cellWidget(index, 1).value())
 110             elif moduleConfigurations[parameter]["informations"]["type"] == "enum":
 111                 moduleConfigurations[parameter]["value"] = unicode(self.pageContent.propertyTable.cellWidget(index, 1).currentText())
 112             elif moduleConfigurations[parameter]["informations"]["type"] == "boolean":
 113                 if self.pageContent.propertyTable.cellWidget(index, 1).isChecked():
 114                     moduleConfigurations[parameter]["value"] = "1"
 115                 else:
 116                     moduleConfigurations[parameter]["value"] = "0"
 117
 118     def _pageChanged(self, row, column, previousRow, previousColumn):
 119         if previousRow != -1 and previousColumn != -1:
 120             self._saveData(previousRow, previousColumn)
 121         self._fillPropertyTable()
 122
 123     def _currentModule(self):
 124         return unicode(self.pageContent.moduleTable.item(self.pageContent.moduleTable.currentRow(), 1).text())
 125
 126     def _currentModuleConfigurations(self):
 127         return self._configurations(self._currentModule())
 128
 129     def _currentProperty(self):
 130         return qvariant_converter.getString(self.pageContent.propertyTable.item(self.pageContent.propertyTable.currentRow(), 0).data(Qt.UserRole))
 131
 132     def _currentPropertyItem(self):
 133         return self.pageContent.propertyTable.item(self.pageContent.propertyTable.currentRow(), 0)
 134
 135     def _module(self, row):
 136         return unicode(self.pageContent.moduleTable.item(row, 1).text())
 137
 138     def _configurations(self, module):
 139         configuration = self._projectInfoRetrieve("MODULES")[module]["configuration"]
 140         return self._projectInfoRetrieve("CONFIGURATIONS")[configuration]
 141
 142     def _resetPropertyDescription(self):
 143         for index in range(self.pageContent.propertyTable.rowCount()):
 144             propertyName = qvariant_converter.getString(self.pageContent.propertyTable.item(index, 0).data(Qt.UserRole))
 145             self.pageContent.propertyTable.item(index, 0).setText(propertyName)
 146
 147     def _showPropertyDescription(self):
 148         self._resetPropertyDescription()
 149         configurations = self._currentModuleConfigurations()
 150         if self._currentProperty() in configurations.keys():
 151             description = configurations[self._currentProperty()]["description"]
 152             name = self._currentProperty()
 153             self._currentPropertyItem().setText(name + "\n" + description)
 154
 155     def _setupUi(self):
 156         self.pageContent.moduleTable.horizontalHeader().setResizeMode(QHeaderView.ResizeToContents)
 157         self.pageContent.moduleTable.horizontalHeader().setStretchLastSection(True)
 158         self.pageContent.moduleTable.horizontalHeader().setVisible(False)
 159         self.pageContent.moduleTable.verticalHeader().setResizeMode(QHeaderView.ResizeToContents)
 160         self.pageContent.moduleTable.verticalHeader().setVisible(False)
 161         self.pageContent.moduleTable.setColumnCount(2)
 162         self.pageContent.moduleTable.setRowCount(0)
 163         self.pageContent.propertyTable.horizontalHeader().setResizeMode(QHeaderView.Stretch)
 164         self.pageContent.propertyTable.horizontalHeader().setVisible(False)
 165         self.pageContent.propertyTable.verticalHeader().setResizeMode(QHeaderView.ResizeToContents)
 166         self.pageContent.propertyTable.verticalHeader().setVisible(False)
 167         self.pageContent.propertyTable.setColumnCount(2)
 168         self.pageContent.propertyTable.setRowCount(0)
 169
 170     def _connectSignals(self):
 171         self.connect(self.pageContent.moduleTable, SIGNAL("currentCellChanged(int, int, int, int)"), self._pageChanged)
 172         self.connect(self.pageContent.propertyTable, SIGNAL("itemSelectionChanged()"), self._showPropertyDescription)
 173
 174     def _moduleSelectionChanged(self, index):
 175         module = unicode(self.pageContent.moduleTable.item(index, 1).text())
 176         if self._buttonGroup.button(index).isChecked():
 177             self._moduleSelected(module)
 178         else:
 179             self._moduleUnselected(module)
 180
 181     def _moduleSelected(self, selectedModule):
 182         modules = self._projectInfoRetrieve("MODULES")
 183         modules[selectedModule]["enabled"] = True
 184         self._projectInfoStore("MODULES", modules)
 185         depends = self._projectInfoRetrieve("MODULES")[selectedModule]["depends"]
 186         unsatisfied = self.selectDependencyCheck(selectedModule)
 187         if len(unsatisfied) > 0:
 188             message = self.tr("The module %1 needs the following modules:\n%2.\n\nDo you want to resolve automatically the problem?")
 189             message = message.arg(selectedModule).arg(", ".join(unsatisfied))
 190             choice = QMessageBox.warning(self, self.tr("Dependency error"), message, QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
 191             if choice == QMessageBox.Yes:
 192                 for module in unsatisfied:
 193                     modules = self._projectInfoRetrieve("MODULES")
 194                     modules[module]["enabled"] = True
 195                 for index in range(self.pageContent.moduleTable.rowCount()):
 196                     if unicode(self.pageContent.moduleTable.item(index, 1).text()) in unsatisfied:
 197                         self._buttonGroup.button(index).setChecked(True)
 198
 199     def _moduleUnselected(self, unselectedModule):
 200         modules = self._projectInfoRetrieve("MODULES")
 201         modules[unselectedModule]["enabled"] = False
 202         self._projectInfoStore("MODULES", modules)
 203         unsatisfied = self.unselectDependencyCheck(unselectedModule)
 204         if len(unsatisfied) > 0:
 205             message = self.tr("The module %1 is needed by the following modules:\n%2.\n\nDo you want to resolve automatically the problem?")
 206             message = message.arg(unselectedModule).arg(", ".join(unsatisfied))
 207             choice = QMessageBox.warning(self, self.tr("Dependency error"), message, QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
 208             if choice == QMessageBox.Yes:
 209                 for module in unsatisfied:
 210                     modules = self._projectInfoRetrieve("MODULES")
 211                     modules[module]["enabled"] = False
 212                 for index in range(self.pageContent.moduleTable.rowCount()):
 213                     if unicode(self.pageContent.moduleTable.item(index, 1).text()) in unsatisfied:
 214                         self._buttonGroup.button(index).setChecked(False)
 215
 216
 217     def selectDependencyCheck(self, module):
 218         unsatisfied = set()
 219         modules = self._projectInfoRetrieve("MODULES")
 220         for dependency in modules[module]["depends"]:
 221             if not modules[dependency]["enabled"]:
 222                 unsatisfied |= set([dependency])
 223                 if dependency not in unsatisfied:
 224                     unsatisfied |= self.selectDependencyCheck(dependency)
 225         return unsatisfied
 226
 227     def unselectDependencyCheck(self, dependency):
 228         unsatisfied = set()
 229         modules = self._projectInfoRetrieve("MODULES")
 230         for module, informations in modules.items():
 231             if dependency in informations["depends"] and informations["enabled"]:
 232                 unsatisfied |= set([module])
 233                 if dependency not in unsatisfied:
 234                     unsatisfied |= self.unselectDependencyCheck(module)
 235         return unsatisfied