Add try/finally for every setOverrideCursor/restoreOverrideCursor.
[bertos.git] / wizard / BEditingDialog.py
1 #!/usr/bin/env python
2 # encoding: utf-8
3 #
4 # This file is part of BeRTOS.
5 #
6 # Bertos is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19 #
20 # As a special exception, you may use this file as part of a free software
21 # library without restriction.  Specifically, if other files instantiate
22 # templates or use macros or inline functions from this file, or you compile
23 # this file and link it with other files to produce an executable, this
24 # file does not by itself cause the resulting executable to be covered by
25 # the GNU General Public License.  This exception does not however
26 # invalidate any other reasons why the executable file might be covered by
27 # the GNU General Public License.
28 #
29 # Copyright 2008 Develer S.r.l. (http://www.develer.com/)
30 #
31 # $Id$
32 #
33 # Author: Lorenzo Berni <duplo@develer.com>
34 #
35
36 import sys
37 import os
38
39 from PyQt4.QtCore import *
40 from PyQt4.QtGui import *
41
42 from bertos_utils import loadBertosProject, bertosVersion, getToolchainName, createBertosProject
43 from BToolchainPage import BToolchainPage
44 from BVersionPage import BVersionPage
45 import qvariant_converter
46 import BModulePage
47 import bertos_utils
48
49 class BEditingDialog(QDialog):
50
51     def __init__(self, parent=None):
52         QDialog.__init__(self, parent)
53         self.setupUi()
54         self.connectSignals()
55         self.module_page.reloadData()
56         self.setFrequency()
57     
58     def setupUi(self):
59         layout = QVBoxLayout()
60         self.module_page = BModulePage.BModulePage()
61         layout.addWidget(self.module_page)
62         frequency_layout = QHBoxLayout()
63         frequency_layout.addWidget(QLabel(self.tr("CPU frequency")))
64         self.cpu_frequency_spinbox = QDoubleSpinBox()
65         self.cpu_frequency_spinbox.setSuffix("Hz")
66         self.cpu_frequency_spinbox.setRange(1, 1000000000)
67         self.cpu_frequency_spinbox.setSingleStep(1000)
68         self.cpu_frequency_spinbox.setDecimals(0)
69         frequency_layout.addWidget(self.cpu_frequency_spinbox)
70         frequency_layout.addStretch()
71         layout.addLayout(frequency_layout)
72         button_layout = QHBoxLayout()
73         self.advanced_button = QToolButton()
74         self.setupMenu()
75         self.advanced_button.setMenu(self.menu)
76         self.advanced_button.setPopupMode(QToolButton.InstantPopup)
77         self.advanced_button.setText(self.tr("Advanced"))
78         button_layout.addWidget(self.advanced_button)
79         button_layout.addStretch()
80         self.cancel_button = QPushButton(self.tr("Cancel"))
81         button_layout.addWidget(self.cancel_button)
82         self.apply_button = QPushButton(self.tr("Apply"))
83         button_layout.addWidget(self.apply_button)
84         layout.addLayout(button_layout)
85         self.setLayout(layout)
86         self.setWindowTitle(self.tr("Edit \"%1\" project").arg(os.path.basename(self.module_page.projectInfo("PROJECT_PATH"))))
87         self.setWindowIcon(QIcon(":/images/appicon.png"))
88
89     def setupMenu(self):
90         self.menu = QMenu(self.tr("Advanced options"))
91         self.change_toolchain = QAction(self.tr("Change toolchain"), self)
92         self.change_bertos_version = QAction(self.tr("Change BeRTOS version"), self)
93         self.menu.addAction(self.change_toolchain)
94         self.menu.addAction(self.change_bertos_version)
95
96     def connectSignals(self):
97         self.connect(self.change_toolchain, SIGNAL("triggered(bool)"), self.changeToolchain)
98         self.connect(self.change_bertos_version, SIGNAL("triggered(bool)"), self.changeBertosVersion)
99         self.connect(self.apply_button, SIGNAL("clicked()"), self.apply)
100         self.connect(self.cancel_button, SIGNAL("clicked()"), self.reject)
101         self.connect(self.cpu_frequency_spinbox, SIGNAL("valueChanged(double)"), self.frequencyChanged)
102     
103     def setFrequency(self):
104         frequency = long(self.module_page.projectInfo("SELECTED_FREQ"))
105         self.cpu_frequency_spinbox.setValue(frequency)
106
107     def frequencyChanged(self, frequency):
108         frequency = unicode(long(frequency))
109         self.module_page.setProjectInfo("SELECTED_FREQ", frequency)
110
111     def changeToolchain(self):
112         dialog = BToolchainDialog()
113         if dialog.exec_():
114             toolchain = qvariant_converter.getStringDict(dialog.toolchain_page.currentItem().data(Qt.UserRole))
115             dialog.toolchain_page.setProjectInfo("TOOLCHAIN", toolchain)
116     
117     def changeBertosVersion(self):
118         current_version = self.module_page.projectInfo("SOURCES_PATH")
119         dialog = BVersionDialog()
120         if dialog.exec_():
121             version = qvariant_converter.getString(dialog.version_page.currentItem().data(Qt.UserRole))
122             if QMessageBox.question(
123                 dialog.version_page,
124                 self.tr("BeRTOS version update"),
125                 self.tr("Changing the BeRTOS version will destroy all the modification done on the BeRTOS sources"),
126                 QMessageBox.Ok | QMessageBox.Cancel
127             ) == QMessageBox.Ok:
128                 try:
129                     qApp.setOverrideCursor(QCursor(Qt.WaitCursor))
130                     dialog.version_page.setProjectInfo("SOURCES_PATH", version)
131                     dialog.version_page.setProjectInfo("OLD_SOURCES_PATH", current_version)
132                     enabled_modules = bertos_utils.enabledModules(dialog.version_page.project())
133                     old_configuration = dialog.version_page.projectInfo("CONFIGURATIONS")
134                     dialog.version_page.project().loadSourceTree()
135                     bertos_utils.loadModuleData(dialog.version_page.project())
136                     new_configuration = dialog.version_page.projectInfo("CONFIGURATIONS")
137                     merged_configuration = {}
138                     for conf in new_configuration:
139                         if conf in old_configuration:
140                             configuration = bertos_utils.updateConfigurationValues(new_configuration[conf], old_configuration[conf])
141                         else:
142                             configuration = new_configuration[conf]
143                         merged_configuration[conf] = configuration
144                     dialog.version_page.setProjectInfo("CONFIGURATIONS", merged_configuration)
145                     bertos_utils.setEnabledModules(dialog.version_page.project(), enabled_modules)
146                     self.module_page.fillModuleTree()
147                 finally:
148                     qApp.restoreOverrideCursor()
149             else:
150                 # Rollback version to the previous selected one.
151                 dialog.version_page.setProjectInfo("SOURCES_PATH", current_version)
152
153     def apply(self):
154         try:
155             qApp.setOverrideCursor(QCursor(Qt.WaitCursor))
156             createBertosProject(self.module_page.project(), edit=True)
157         finally:
158             qApp.restoreOverrideCursor()
159         self.accept()
160
161     def toolchains(self):
162         return self.module_page.toolchains()
163     
164     def currentToolchain(self):
165         return self.module_page.projectInfo("TOOLCHAIN")
166     
167     def setCurrentToolchain(self, toolchain):
168         self.module_page.setProjectInfo("TOOLCHAIN", toolchain)
169
170     def versions(self):
171         return self.module_page.versions()
172
173     def currentVersion(self):
174         return self.module_page.projectInfo("SOURCES_PATH")
175     
176     def setCurrentVersion(self, version):
177         self.module_page.setProjectInfo("SOURCES_PATH", version)
178
179 class BToolchainDialog(QDialog):
180     def __init__(self):
181         QDialog.__init__(self)
182         self.setWindowIcon(QIcon(":/images/appicon.png"))
183         layout = QVBoxLayout()
184         toolchain_page = BToolchainPage()
185         current_toolchain = toolchain_page.projectInfo("TOOLCHAIN")
186         toolchain_page.reloadData()
187         # TODO: to be moved in BToolchainPage
188         for toolchain_row in range(toolchain_page.pageContent.toolchainList.count()):
189             toolchain = qvariant_converter.getStringDict(toolchain_page.pageContent.toolchainList.item(toolchain_row).data(Qt.UserRole))
190             if current_toolchain and toolchain["path"] == current_toolchain["path"]:
191                 toolchain_page.pageContent.toolchainList.setCurrentRow(toolchain_row)
192                 toolchain_page.selectionChanged()
193                 break
194         self.toolchain_page = toolchain_page
195         layout.addWidget(toolchain_page)
196         button_layout = QHBoxLayout()
197         button_layout.addStretch()
198         cancel_button = QPushButton(self.tr("Cancel")) 
199         button_layout.addWidget(cancel_button)
200         ok_button = QPushButton(self.tr("Ok"))
201         button_layout.addWidget(ok_button)
202         self.connect(cancel_button, SIGNAL("clicked()"), self.reject)
203         layout.addLayout(button_layout)
204         self.setLayout(layout)
205         self.connect(ok_button, SIGNAL("clicked()"), self.accept)
206         self.setWindowTitle(self.tr("Change toolchain"))
207
208 class BVersionDialog(QDialog):
209     def __init__(self):
210         QDialog.__init__(self)
211         self.setWindowIcon(QIcon(":/images/appicon.png"))
212         layout = QVBoxLayout()
213         version_page = BVersionPage(edit=True)
214         version_page.reloadData()
215         self.version_page = version_page
216         layout.addWidget(version_page)
217         button_layout = QHBoxLayout()
218         button_layout.addStretch()
219         cancel_button = QPushButton(self.tr("Cancel")) 
220         button_layout.addWidget(cancel_button)
221         ok_button = QPushButton(self.tr("Ok"))
222         button_layout.addWidget(ok_button)
223         self.connect(cancel_button, SIGNAL("clicked()"), self.reject)
224         layout.addLayout(button_layout)
225         self.setLayout(layout)
226         self.connect(ok_button, SIGNAL("clicked()"), self.accept)
227         current_version = version_page.projectInfo("SOURCES_PATH")
228         self.setWindowTitle(self.tr("Change BeRTOS version"))
229
230
231 def main():
232     if len(sys.argv) > 1:
233         project_file = sys.argv[1]
234     else:
235         print "Invalid usage: use <command> project_file"
236         sys.exit()
237     app = QApplication([])
238     app.project = loadBertosProject(project_file)
239     app.settings = QSettings("Develer", "Bertos Configurator")
240     dialog = BEditingDialog()
241     dialog.show()
242     sys.exit(app.exec_())
243
244 if __name__ == "__main__":
245     main()