4 # Copyright 2009 Develer S.r.l. (http://www.develer.com/)
9 # Author: Lorenzo Berni <duplo@develer.com>
14 from PyQt4.QtGui import *
15 from BWizardPage import *
19 class BModulePage(BWizardPage):
22 BWizardPage.__init__(self, "module_select.ui")
23 self.setTitle(self.tr("Configure the BeRTOS modules"))
25 self._controlGroup = QControlGroup()
26 self._connectSignals()
29 self._setupButtonGroup()
30 self._loadModuleData()
31 self._fillModuleTable()
33 def _setupButtonGroup(self):
34 self._buttonGroup = QButtonGroup()
35 self._buttonGroup.setExclusive(False)
36 self.connect(self._buttonGroup, SIGNAL("buttonClicked(int)"), self._moduleSelectionChanged)
38 def _loadModuleData(self):
39 modules = bertos_utils.loadModuleInfosDict(self._projectInfoRetrieve("SOURCES_PATH"))
40 lists = bertos_utils.loadDefineListsDict(self._projectInfoRetrieve("SOURCES_PATH"))
42 for module, informations in modules.items():
43 configurations[informations["configuration"]] = bertos_utils.loadConfigurationInfos(self._projectInfoRetrieve("SOURCES_PATH") +
44 "/" + informations["configuration"])
45 self._projectInfoStore("MODULES", modules)
46 self._projectInfoStore("LISTS", lists)
47 self._projectInfoStore("CONFIGURATIONS", configurations)
49 def _fillModuleTable(self):
50 modules = self._projectInfoRetrieve("MODULES")
51 self.pageContent.moduleTable.setRowCount(len(modules))
52 for index, module in enumerate(modules):
53 self.pageContent.moduleTable.setItem(index, 1, QTableWidgetItem(module))
54 checkBox = QCheckBox()
55 self._buttonGroup.addButton(checkBox, index)
56 self.pageContent.moduleTable.setCellWidget(index, 0, checkBox)
57 checkBox.setChecked(modules[module]["enabled"])
59 def _fillPropertyTable(self):
60 module = self._currentModule()
61 self._controlGroup.clear()
62 configuration = self._projectInfoRetrieve("MODULES")[module]["configuration"]
63 configurations = self._projectInfoRetrieve("CONFIGURATIONS")[configuration]
64 self.pageContent.propertyTable.clear()
65 self.pageContent.propertyTable.setRowCount(len(configurations))
66 for index, property in enumerate(configurations):
67 item = QTableWidgetItem(property)
68 item.setData(Qt.UserRole, qvariant_converter.convertString(property))
69 self.pageContent.propertyTable.setItem(index, 0, item)
70 if "type" in configurations[property]["informations"].keys() and configurations[property]["informations"]["type"] == "boolean":
72 checkBox = QCheckBox()
73 self.pageContent.propertyTable.setCellWidget(index, 1, checkBox)
74 if configurations[property]["value"] == "1":
75 checkBox.setChecked(True)
77 checkBox.setChecked(False)
78 self._controlGroup.addControl(index, checkBox)
79 elif "type" in configurations[property]["informations"].keys() and configurations[property]["informations"]["type"] == "enum":
81 comboBox = QComboBox()
82 self.pageContent.propertyTable.setCellWidget(index, 1, comboBox)
83 enum = self._projectInfoRetrieve("LISTS")[configurations[property]["informations"]["value_list"]]
84 for i, element in enumerate(enum):
85 comboBox.addItem(element)
86 if element == configurations[property]["value"]:
87 comboBox.setCurrentIndex(i)
88 self._controlGroup.addControl(index, comboBox)
90 ## int, long or undefined type property
92 self.pageContent.propertyTable.setCellWidget(index, 1, spinBox)
93 if "min" in configurations[property]["informations"].keys():
94 minimum = int(configurations[property]["informations"]["min"])
97 spinBox.setMinimum(minimum)
98 if "max" in configurations[property]["informations"].keys():
99 maximum = int(configurations[property]["informations"]["max"])
102 spinBox.setMaximum(maximum)
103 if "long" in configurations[property]["informations"].keys() and configurations[property]["informations"]["long"] == "True":
104 spinBox.setSuffix("L")
105 spinBox.setValue(int(configurations[property]["value"].replace("L", "")))
106 self._controlGroup.addControl(index, spinBox)
108 def _currentModule(self):
109 return unicode(self.pageContent.moduleTable.item(self.pageContent.moduleTable.currentRow(), 1).text())
111 def _currentModuleConfigurations(self):
112 return self._configurations(self._currentModule())
114 def _currentProperty(self):
115 return qvariant_converter.getString(self.pageContent.propertyTable.item(self.pageContent.propertyTable.currentRow(), 0).data(Qt.UserRole))
117 def _currentPropertyItem(self):
118 return self.pageContent.propertyTable.item(self.pageContent.propertyTable.currentRow(), 0)
120 def _module(self, row):
121 return unicode(self.pageContent.moduleTable.item(row, 1).text())
123 def _configurations(self, module):
124 configuration = self._projectInfoRetrieve("MODULES")[module]["configuration"]
125 return self._projectInfoRetrieve("CONFIGURATIONS")[configuration]
127 def _resetPropertyDescription(self):
128 for index in range(self.pageContent.propertyTable.rowCount()):
129 propertyName = qvariant_converter.getString(self.pageContent.propertyTable.item(index, 0).data(Qt.UserRole))
130 self.pageContent.propertyTable.item(index, 0).setText(propertyName)
132 def _showPropertyDescription(self):
133 self._resetPropertyDescription()
134 configurations = self._currentModuleConfigurations()
135 if self._currentProperty() in configurations.keys():
136 description = configurations[self._currentProperty()]["description"]
137 name = self._currentProperty()
138 self._currentPropertyItem().setText(name + "\n" + description)
141 self.pageContent.moduleTable.horizontalHeader().setResizeMode(QHeaderView.ResizeToContents)
142 self.pageContent.moduleTable.horizontalHeader().setStretchLastSection(True)
143 self.pageContent.moduleTable.horizontalHeader().setVisible(False)
144 self.pageContent.moduleTable.verticalHeader().setResizeMode(QHeaderView.ResizeToContents)
145 self.pageContent.moduleTable.verticalHeader().setVisible(False)
146 self.pageContent.moduleTable.setColumnCount(2)
147 self.pageContent.moduleTable.setRowCount(0)
148 self.pageContent.propertyTable.horizontalHeader().setResizeMode(QHeaderView.Stretch)
149 self.pageContent.propertyTable.horizontalHeader().setVisible(False)
150 self.pageContent.propertyTable.verticalHeader().setResizeMode(QHeaderView.ResizeToContents)
151 self.pageContent.propertyTable.verticalHeader().setVisible(False)
152 self.pageContent.propertyTable.setColumnCount(2)
153 self.pageContent.propertyTable.setRowCount(0)
155 def _connectSignals(self):
156 self.connect(self.pageContent.moduleTable, SIGNAL("itemSelectionChanged()"), self._fillPropertyTable)
157 self.connect(self.pageContent.propertyTable, SIGNAL("itemSelectionChanged()"), self._showPropertyDescription)
158 self.connect(self._controlGroup, SIGNAL("stateChanged"), self._saveValue)
160 def _saveValue(self, index):
161 property = qvariant_converter.getString(self.pageContent.propertyTable.item(index, 0).data(Qt.UserRole))
162 configuration = self._projectInfoRetrieve("MODULES")[self._currentModule()]["configuration"]
163 configurations = self._projectInfoRetrieve("CONFIGURATIONS")
164 if "type" not in configurations[configuration][property]["informations"].keys() or configurations[configuration][property]["informations"]["type"] == "int":
165 configurations[configuration][property]["value"] = str(self.pageContent.propertyTable.cellWidget(index, 1).value())
166 elif configurations[configuration][property]["informations"]["type"] == "enum":
167 configurations[configuration][property]["value"] = unicode(self.pageContent.propertyTable.cellWidget(index, 1).currentText())
168 elif configurations[configuration][property]["informations"]["type"] == "boolean":
169 if self.pageContent.propertyTable.cellWidget(index, 1).isChecked():
170 configurations[configuration][property]["value"] = "1"
172 configurations[configuration][property]["value"] = "0"
173 self._projectInfoStore("CONFIGURATIONS", configurations)
175 def _moduleSelectionChanged(self, index):
176 module = unicode(self.pageContent.moduleTable.item(index, 1).text())
177 if self._buttonGroup.button(index).isChecked():
178 self._moduleSelected(module)
180 self._moduleUnselected(module)
182 def _moduleSelected(self, selectedModule):
183 modules = self._projectInfoRetrieve("MODULES")
184 modules[selectedModule]["enabled"] = True
185 self._projectInfoStore("MODULES", modules)
186 depends = self._projectInfoRetrieve("MODULES")[selectedModule]["depends"]
187 unsatisfied = self.selectDependencyCheck(selectedModule)
188 if len(unsatisfied) > 0:
189 message = self.tr("The module %1 needs the following modules:\n%2.\n\nDo you want to resolve automatically the problem?")
190 message = message.arg(selectedModule).arg(", ".join(unsatisfied))
191 choice = QMessageBox.warning(self, self.tr("Dependency error"), message, QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
192 if choice == QMessageBox.Yes:
193 for module in unsatisfied:
194 modules = self._projectInfoRetrieve("MODULES")
195 modules[module]["enabled"] = True
196 for index in range(self.pageContent.moduleTable.rowCount()):
197 if unicode(self.pageContent.moduleTable.item(index, 1).text()) in unsatisfied:
198 self._buttonGroup.button(index).setChecked(True)
200 def _moduleUnselected(self, unselectedModule):
201 modules = self._projectInfoRetrieve("MODULES")
202 modules[unselectedModule]["enabled"] = False
203 self._projectInfoStore("MODULES", modules)
204 unsatisfied = self.unselectDependencyCheck(unselectedModule)
205 if len(unsatisfied) > 0:
206 message = self.tr("The module %1 is needed by the following modules:\n%2.\n\nDo you want to resolve automatically the problem?")
207 message = message.arg(unselectedModule).arg(", ".join(unsatisfied))
208 choice = QMessageBox.warning(self, self.tr("Dependency error"), message, QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
209 if choice == QMessageBox.Yes:
210 for module in unsatisfied:
211 modules = self._projectInfoRetrieve("MODULES")
212 modules[module]["enabled"] = False
213 for index in range(self.pageContent.moduleTable.rowCount()):
214 if unicode(self.pageContent.moduleTable.item(index, 1).text()) in unsatisfied:
215 self._buttonGroup.button(index).setChecked(False)
218 def selectDependencyCheck(self, module):
220 modules = self._projectInfoRetrieve("MODULES")
221 for dependency in modules[module]["depends"]:
222 if not modules[dependency]["enabled"]:
223 unsatisfied |= set([dependency])
224 if dependency not in unsatisfied:
225 unsatisfied |= self.selectDependencyCheck(dependency)
228 def unselectDependencyCheck(self, dependency):
230 modules = self._projectInfoRetrieve("MODULES")
231 for module, informations in modules.items():
232 if dependency in informations["depends"] and informations["enabled"]:
233 unsatisfied |= set([module])
234 if dependency not in unsatisfied:
235 unsatisfied |= self.unselectDependencyCheck(module)
238 class QControlGroup(QObject):
240 QObject.__init__(self)
243 def addControl(self, id, control):
244 self._controls[id] = control
245 if type(control) == QCheckBox:
246 self.connect(control, SIGNAL("stateChanged(int)"), lambda: self._stateChanged(id))
247 elif type(control) == QSpinBox:
248 self.connect(control, SIGNAL("valueChanged(int)"), lambda: self._stateChanged(id))
249 elif type(control) == QComboBox:
250 self.connect(control, SIGNAL("currentIndexChanged(int)"), lambda: self._stateChanged(id))
255 def _stateChanged(self, id):
256 self.emit(SIGNAL("stateChanged"), id)