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.currentIndex != -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 infos["target"].find(needed["TOOLCHAIN"]) != -1:
  83             item.setIcon(QIcon(":/images/ok.png"))
  84         else:
  85             item.setIcon(QIcon(":/images/warning.png"))
  86         item.setText("GCC " + infos["version"] + " - " + infos["target"])
  87
  88     def _invalidItem(self, index):
  89         item = self.pageContent.toolchainList.item(index)
  90         item.setIcon(QIcon(":/images/error.png"))
  91
  92     def addToolchain(self):
  93         sel_toolchain = QFileDialog.getOpenFileName(self, self.tr("Choose the toolchain"), "")
  94         if not sel_toolchain.isEmpty():
  95             item = QListWidgetItem(sel_toolchain)
  96             item.setData(Qt.UserRole, qvariant_converter.convertString({"path": sel_toolchain}))
  97             self.pageContent.toolchainList.addItem(item)
  98             toolchains = self.toolchains()
  99             toolchains[sel_toolchain] = False
 100             self.setToolchains(toolchains)
 101
 102     def removeToolchain(self):
 103         if self.pageContent.toolchainList.currentRow() != -1:
 104             item = self.pageContent.toolchainList.takeItem(self.pageContent.toolchainList.currentRow())
 105             toolchain = qvariant_converter.getStringDict(item.data(Qt.UserRole))["path"]
 106             toolchains = self.toolchains()
 107             del toolchains[toolchain]
 108             self.setToolchains(toolchains)
 109
 110     def searchToolchain(self):
 111         search = BToolchainSearch.BToolchainSearch()
 112         self.connect(search, SIGNAL("accepted()"), self._search)
 113         search.exec_()
 114
 115     def validateAllToolchains(self):
 116         for i in range(self.pageContent.toolchainList.count()):
 117             self.validateToolchain(i)
 118
 119     def validateToolchain(self, i):
 120         filename = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(i).data(Qt.UserRole))["path"]
 121         valid = False
 122         info = {}
 123         ## Check for the other tools of the toolchain
 124         for tool in TOOLCHAIN_ITEMS:
 125             if os.path.exists(filename.replace("gcc", tool)):
 126                 valid = True
 127             else:
 128                 valid = False
 129                 break
 130         ## Try to retrieve the informations about the toolchain only for the valid toolchains
 131         if valid:
 132             self._validationProcess = QProcess()
 133             self._validationProcess.start(filename, ["-v"])
 134             self._validationProcess.waitForStarted(1000)
 135             if self._validationProcess.waitForFinished(200):
 136                 description = str(self._validationProcess.readAllStandardError())
 137                 info = bertos_utils.getToolchainInfo(description)
 138                 if len(info.keys()) >= 4:
 139                     valid = True
 140             else:
 141                 self._validationProcess.kill()
 142         ## Add the item in the list with the appropriate associate data.
 143         if valid:
 144             self._validItem(i, info)
 145         else:
 146             self._invalidItem(i)
 147         toolchains = self.toolchains()
 148         toolchains[filename] = True
 149         self.setToolchains(toolchains)
 150
 151     def isComplete(self):
 152         if self.pageContent.toolchainList.currentRow() != -1:
 153             self._projectInfoStore("TOOLCHAIN",
 154                 qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
 155             return True
 156         else:
 157             return False
 158
 159     def reloadData(self):
 160         self._clearList()
 161         self._updateUi()
 162         self._populateToolchainList()