wizard/BProjectPresets.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3 #
   4 # This file is part of slimqc.
   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 2010 Develer S.r.l. (http://www.develer.com/)
  30 #
  31 # $Id$
  32 #
  33 # Author: Lorenzo Berni <duplo@develer.com>
  34 #
  35
  36 import os
  37
  38 from PyQt4 import uic
  39
  40 from PyQt4.QtCore import *
  41 from PyQt4.QtGui import *
  42
  43 from BWizardPage import BWizardPage
  44
  45 from BOutputPage import BOutputPage
  46 from BToolchainPage import BToolchainPage
  47
  48 from bertos_utils import _cmp
  49 from toolchain_manager import ToolchainManager
  50
  51 import const
  52 import qvariant_converter
  53
  54 class BProjectPresetsPage(QWidget):
  55     def __init__(self, preset_data, parent=None):
  56         QWidget.__init__(self, parent)
  57         self.pageContent = uic.loadUi(os.path.join(const.DATA_DIR, const.UI_LOCATION, "preset_page.ui"), None)
  58         self.project = QApplication.instance().project
  59         self.settings = QApplication.instance().settings
  60         self.preset_data = preset_data
  61         layout = QVBoxLayout()
  62         layout.addWidget(self.pageContent)
  63         self.setLayout(layout)
  64         self.setupUi()
  65         self.connectSignals()
  66
  67     def setupUi(self):
  68         self.pageContent.presetList.clear()
  69         self.pageContent.categoryDescription.setText(self.preset_data["info"].get("description", ""))
  70         for preset in sorted(self.preset_data["children"].values(), _cmp):
  71             item_name = preset["info"].get("name", preset["info"]["filename"])
  72             item_icon = os.path.join(preset["info"]["path"], const.PREDEFINED_BOARD_ICON_FILE)
  73             if not os.path.exists(item_icon):
  74                 item_icon = const.PREDEFINED_BOARD_DEFAULT_PROJECT_ICON
  75             item_icon = QIcon(item_icon)
  76             item = QListWidgetItem(item_icon, item_name)
  77             item.setData(Qt.UserRole, qvariant_converter.convertString(preset["info"]["path"]))
  78             self.pageContent.presetList.addItem(item)
  79         self.pageContent.presetList.setCurrentRow(0)
  80         self.updateUi()
  81
  82     def connectSignals(self):
  83         self.connect(self.pageContent.presetList, SIGNAL("itemSelectionChanged()"), self.updateUi)
  84         self.connect(self.pageContent.presetList, SIGNAL("itemSelectionChanged()"), self, SIGNAL("completeChanged()"))
  85
  86     def updateUi(self):
  87         if self.selected:
  88             preset_path = qvariant_converter.getString(self.selected.data(Qt.UserRole))
  89             preset = self.preset_data["children"][preset_path]
  90             description = preset["info"].get("description", "")
  91             path = unicode(QUrl.fromLocalFile(preset_path).toString())
  92             description = description.replace("$path", path)
  93             self.pageContent.descriptionArea.setHtml(description)
  94
  95     @property
  96     def selected(self):
  97         return self.pageContent.presetList.currentItem()
  98
  99
 100 class BProjectPresets(BWizardPage):
 101     def __init__(self):
 102         BWizardPage.__init__(self, const.UI_LOCATION + "/project_presets.ui")
 103
 104     ## Overloaded QWizardPage methods ##
 105
 106     def isComplete(self):
 107         preset_path = self.selected_path
 108         if preset_path:
 109             self.setProjectInfo("PROJECT_PRESET", preset_path)
 110             self.setProjectInfo("BASE_MODE", not self.advanced)
 111             # TODO: find a better place for preset loading...
 112             try:
 113                 QApplication.instance().setOverrideCursor(Qt.WaitCursor)
 114                 self.project.loadProjectFromPreset(preset_path)
 115             finally:
 116                 QApplication.instance().restoreOverrideCursor()
 117             return True
 118         else:
 119             self.setProjectInfo("PROJECT_PRESET", None)
 120             return False
 121
 122     def nextId(self):
 123         """
 124         Overload of the QWizardPage nextId method.
 125         """
 126         # Route to Toolchain page if the user select advanced
 127         # or to Output page if the user select base
 128         if self.advanced:
 129             return self.wizard().pageIndex(BToolchainPage)
 130         else:
 131             cpu_info = self.projectInfo("CPU_INFOS")
 132             if cpu_info:
 133                 target = cpu_info["TOOLCHAIN"]
 134                 # Try to find a suitable toolchain automatically
 135                 tm = ToolchainManager()
 136                 suitable_toolchains = tm.suitableToolchains(target)
 137                 if len(suitable_toolchains) == 1:
 138                     toolchain = suitable_toolchains.pop()
 139                     toolchain_info = tm._validateToolchain(toolchain)
 140                     toolchain_info["path"] = toolchain
 141                     self.setProjectInfo("TOOLCHAIN", toolchain_info)
 142                     return self.wizard().pageIndex(BOutputPage)
 143                 else:
 144                     return self.wizard().pageIndex(BToolchainPage)
 145             else:
 146                 # It seems that the nextId method is called before the
 147                 # reloadData one (that is called after the page changing.
 148                 #
 149                 # TODO: fix this awful code lines
 150                 target = None
 151                 return self.wizard().pageIndex(BToolchainPage)
 152
 153     ####
 154
 155     ## Overloaded BWizardPage methods ##
 156
 157     def reloadData(self):
 158         preset_path = self.projectInfo("PROJECT_BOARD")
 159         preset_tree = self.projectInfo("PRESET_TREE")
 160         preset_list = preset_tree["children"][preset_path]["children"]
 161         preset_list = sorted(preset_list.values(), _cmp)
 162         self.setTitle(self.tr("Select the template/demo for %1").arg(preset_tree["children"][preset_path]["info"].get("name", preset_tree["children"][preset_path]["info"]["filename"])))
 163         self.setupTabs(preset_list)
 164
 165     def connectSignals(self):
 166         self.connect(self.pageContent.boardTabWidget, SIGNAL("currentChanged(int)"), self, SIGNAL("completeChanged()"))
 167
 168     ####
 169
 170     ## Slots ##
 171     ####
 172
 173     def setupTabs(self, preset_list):
 174         self.pageContent.boardTabWidget.clear()
 175         for preset in preset_list:
 176             icon = os.path.join(preset["info"]["path"], ".icon.png")
 177             preset_page = BProjectPresetsPage(preset)
 178             if os.path.exists(icon):
 179                 self.pageContent.boardTabWidget.addTab(preset_page, QIcon(icon), preset["info"].get("name", preset["info"]["filename"]))
 180             else:
 181                 self.pageContent.boardTabWidget.addTab(preset_page, preset["info"].get("name", preset["info"]["filename"]))
 182             self.connect(preset_page, SIGNAL("completeChanged()"), self, SIGNAL("completeChanged()"))
 183
 184     @property
 185     def advanced(self):
 186         if self.selected_data:
 187             return self.selected_data["info"].get("advanced", False)
 188         else:
 189             return None
 190
 191     @property
 192     def selected_path(self):
 193         current_widget = self.pageContent.boardTabWidget.currentWidget()
 194         preset_path = None
 195         if current_widget:
 196             current_item = current_widget.pageContent.presetList.currentItem()
 197             if current_item:
 198                 preset_path = current_item.data(Qt.UserRole)
 199                 preset_path = qvariant_converter.getString(preset_path)
 200         return preset_path
 201
 202     @property
 203     def selected_data(self):
 204         if self.selected_path:
 205             current_widget = self.pageContent.boardTabWidget.currentWidget()
 206             return current_widget.preset_data["children"][self.selected_path]
 207         else:
 208             return None