wizard/BToolchainPage.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3 #
   4 # Copyright 2008 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 import collections
  14
  15 from BWizardPage import *
  16 import BToolchainSearch
  17 import bertos_utils
  18 import qvariant_converter
  19
  20 from const import *
  21
  22 class BToolchainPage(BWizardPage):
  23
  24     def __init__(self):
  25         BWizardPage.__init__(self, UI_LOCATION + "/toolchain_select.ui")
  26         self.setTitle(self.tr("Select toolchain"))
  27         self._validation_process = None
  28         self._updateUi()
  29         #self._populateToolchainList()
  30         self._connectSignals()
  31
  32     def _updateUi(self):
  33         self.pageContent.infoLabel.setVisible(False)
  34
  35     def _populateToolchainList(self):
  36         toolchains = self.toolchains()
  37         sel_toolchain = self._projectInfoRetrieve("TOOLCHAIN")
  38         for key, value in toolchains.items():
  39             item = QListWidgetItem(key)
  40             item.setData(Qt.UserRole, qvariant_converter.convertStringDict({"path": key}))
  41             self.pageContent.toolchainList.addItem(item)
  42             if sel_toolchain is not None and sel_toolchain["path"] == key:
  43                 self.pageContent.toolchainList.setCurrentItem(item)
  44             if value:
  45                 self.validateToolchain(self.pageContent.toolchainList.row(item))
  46
  47     def _clearList(self):
  48         self.pageContent.toolchainList.clear()
  49
  50     def _selectionChanged(self):
  51         if self.pageContent.toolchainList.currentRow() != -1:
  52             infos = collections.defaultdict(lambda: unicode("not defined"))
  53             infos.update(qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
  54             self.pageContent.infoLabel.setText("GCC " + infos["version"] + " (" + infos["build"] + ")\nTarget: " + infos["target"] + "\nPath: " + os.path.normpath(infos["path"]))
  55             self.pageContent.infoLabel.setVisible(True)
  56             self.emit(SIGNAL("completeChanged()"))
  57
  58     def _search(self):
  59         dir_list = self.searchDirList()
  60         if self.pathSearch():
  61             dir_list += [element for element in bertos_utils.getSystemPath()]
  62         toolchain_list = bertos_utils.findToolchains(dir_list)
  63         stored_toolchains = self.toolchains()
  64         for element in toolchain_list:
  65             if not element in stored_toolchains.keys():
  66                 item = QListWidgetItem(element)
  67                 item.setData(Qt.UserRole, qvariant_converter.convertStringDict({"path": element}))
  68                 self.pageContent.toolchainList.addItem(item)
  69                 stored_toolchains[element] = False
  70         self.setToolchains(stored_toolchains)
  71
  72     def _connectSignals(self):
  73         self.connect(self.pageContent.toolchainList, SIGNAL("itemSelectionChanged()"), self._selectionChanged)
  74         self.connect(self.pageContent.addButton, SIGNAL("clicked()"), self.addToolchain)
  75         self.connect(self.pageContent.removeButton, SIGNAL("clicked()"), self.removeToolchain)
  76         self.connect(self.pageContent.searchButton, SIGNAL("clicked()"), self.searchToolchain)
  77         self.connect(self.pageContent.checkButton, SIGNAL("clicked()"), self.validateAllToolchains)
  78
  79     def _validItem(self, index, infos):
  80         item = self.pageContent.toolchainList.item(index)
  81         new_data = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(index).data(Qt.UserRole))
  82         new_data.update(infos)
  83         item.setData(Qt.UserRole, qvariant_converter.convertStringDict(new_data))
  84         needed = self._projectInfoRetrieve("CPU_INFOS")
  85         if "target" in infos.keys() and infos["target"].find(needed["TOOLCHAIN"]) != -1:
  86             item.setIcon(QIcon(":/images/ok.png"))
  87         else:
  88             item.setIcon(QIcon(":/images/warning.png"))
  89         if "version" in infos.keys() and "target" in infos.keys():
  90             item.setText("GCC " + infos["version"] + " - " + infos["target"])
  91
  92     def _invalidItem(self, index):
  93         item = self.pageContent.toolchainList.item(index)
  94         item.setIcon(QIcon(":/images/error.png"))
  95
  96     def addToolchain(self):
  97         sel_toolchain = QFileDialog.getOpenFileName(self, self.tr("Choose the toolchain"), "")
  98         if not sel_toolchain.isEmpty():
  99             item = QListWidgetItem(sel_toolchain)
 100             item.setData(Qt.UserRole, qvariant_converter.convertString({"path": sel_toolchain}))
 101             self.pageContent.toolchainList.addItem(item)
 102             toolchains = self.toolchains()
 103             toolchains[sel_toolchain] = False
 104             self.setToolchains(toolchains)
 105
 106     def removeToolchain(self):
 107         if self.pageContent.toolchainList.currentRow() != -1:
 108             item = self.pageContent.toolchainList.takeItem(self.pageContent.toolchainList.currentRow())
 109             toolchain = qvariant_converter.getStringDict(item.data(Qt.UserRole))["path"]
 110             toolchains = self.toolchains()
 111             del toolchains[toolchain]
 112             self.setToolchains(toolchains)
 113
 114     def searchToolchain(self):
 115         search = BToolchainSearch.BToolchainSearch()
 116         self.connect(search, SIGNAL("accepted()"), self._search)
 117         search.exec_()
 118
 119     def validateAllToolchains(self):
 120         QApplication.instance().setOverrideCursor(Qt.WaitCursor)
 121         for i in range(self.pageContent.toolchainList.count()):
 122             self.validateToolchain(i)
 123         QApplication.instance().restoreOverrideCursor()
 124
 125     def validateToolchain(self, i):
 126         filename = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(i).data(Qt.UserRole))["path"]
 127         valid = False
 128         info = {}
 129         ## Check for the other tools of the toolchain
 130         for tool in TOOLCHAIN_ITEMS:
 131             if os.path.exists(filename.replace("gcc", tool)):
 132                 valid = True
 133             else:
 134                 valid = False
 135                 break
 136         ## Try to retrieve the informations about the toolchain only for the valid toolchains
 137         if valid:
 138             self._validation_process = QProcess()
 139             self._validation_process.start(filename, ["-v"])
 140             self._validation_process.waitForStarted(1000)
 141             if self._validation_process.waitForFinished(200):
 142                 description = str(self._validation_process.readAllStandardError())
 143                 info = bertos_utils.getToolchainInfo(description)
 144                 if len(info.keys()) >= 4:
 145                     valid = True
 146             else:
 147                 self._validation_process.kill()
 148         ## Add the item in the list with the appropriate associate data.
 149         if valid:
 150             self._validItem(i, info)
 151         else:
 152             self._invalidItem(i)
 153         toolchains = self.toolchains()
 154         toolchains[filename] = True
 155         self.setToolchains(toolchains)
 156
 157     def isComplete(self):
 158         if self.pageContent.toolchainList.currentRow() != -1:
 159             self._projectInfoStore("TOOLCHAIN",
 160                 qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
 161             return True
 162         else:
 163             return False
 164
 165     def reloadData(self):
 166         self._clearList()
 167         self._updateUi()
 168         self._populateToolchainList()