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