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         infos = collections.defaultdict(lambda: unicode("not defined"))
  49         infos.update(qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
  50         self.pageContent.infoLabel.setText("GCC " + infos["version"] + " (" + infos["build"] + ")\nTarget: " + infos["target"] + "\nPath: " + infos["path"])
  51         self.pageContent.infoLabel.setVisible(True)
  52         self.emit(SIGNAL("completeChanged()"))
  53
  54     def _search(self):
  55         dirList = self.searchDirList()
  56         if(self.pathSearch()):
  57             dirList += [element for element in bertos_utils.getSystemPath()]
  58         toolchainList = bertos_utils.findToolchains(dirList)
  59         storedToolchains = self.toolchains()
  60         for element in toolchainList:
  61             if not element in storedToolchains.keys():
  62                 item = QListWidgetItem(element)
  63                 item.setData(Qt.UserRole, qvariant_converter.convertStringDict({"path": element}))
  64                 self.pageContent.toolchainList.addItem(item)
  65                 storedToolchains[element] = False
  66         self.setToolchains(storedToolchains)
  67
  68     def _connectSignals(self):
  69         self.connect(self.pageContent.toolchainList, SIGNAL("itemSelectionChanged()"), self._selectionChanged)
  70         self.connect(self.pageContent.addButton, SIGNAL("clicked()"), self.addToolchain)
  71         self.connect(self.pageContent.removeButton, SIGNAL("clicked()"), self.removeToolchain)
  72         self.connect(self.pageContent.searchButton, SIGNAL("clicked()"), self.searchToolchain)
  73         self.connect(self.pageContent.checkButton, SIGNAL("clicked()"), self.validateAllToolchains)
  74
  75     def _validItem(self, index, infos):
  76         item = self.pageContent.toolchainList.item(index)
  77         newData = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(index).data(Qt.UserRole))
  78         newData.update(infos)
  79         item.setData(Qt.UserRole, qvariant_converter.convertStringDict(newData))
  80         needed = self._projectInfoRetrieve("CPU_INFOS")
  81         if infos["target"].find(needed["TOOLCHAIN"]) != -1:
  82             item.setIcon(QIcon(":/images/ok.png"))
  83         else:
  84             item.setIcon(QIcon(":/images/warning.png"))
  85         item.setText("GCC " + infos["version"] + " - " + infos["target"])
  86
  87     def _invalidItem(self, index):
  88         item = self.pageContent.toolchainList.item(index)
  89         item.setIcon(QIcon(":/images/error.png"))
  90
  91     def addToolchain(self):
  92         sel_toolchain = QFileDialog.getOpenFileName(self, self.tr("Choose the toolchain"), "")
  93         if not sel_toolchain.isEmpty():
  94             item = QListWidgetItem(sel_toolchain)
  95             item.setData(Qt.UserRole, qvariant_converter.convertString(sel_toolchain))
  96             self.pageContent.toolchainList.addItem(item)
  97             toolchains = self.toolchains()
  98             toolchains[sel_toolchain] = False
  99             self.setToolchains(toolchains)
 100
 101     def removeToolchain(self):
 102         if self.pageContent.toolchainList.currentRow() != -1:
 103             item = self.pageContent.toolchainList.takeItem(self.pageContent.toolchainList.currentRow())
 104             toolchain = qvariant_converter.getStringDict(item.data(Qt.UserRole))["path"]
 105             toolchains = self.toolchains()
 106             del toolchains[toolchain]
 107             self.setToolchains(toolchains)
 108
 109     def searchToolchain(self):
 110         search = BToolchainSearch.BToolchainSearch()
 111         self.connect(search, SIGNAL("accepted()"), self._search)
 112         search.exec_()
 113
 114     def validateAllToolchains(self):
 115         for i in range(self.pageContent.toolchainList.count()):
 116             self.validateToolchain(i)
 117
 118     def validateToolchain(self, i):
 119         filename = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(i).data(Qt.UserRole))["path"]
 120         self._validationProcess = QProcess()
 121         self._validationProcess.start(filename, ["-v"])
 122         self._validationProcess.waitForStarted(1000)
 123         if self._validationProcess.waitForFinished(200):
 124             description = str(self._validationProcess.readAllStandardError())
 125             infos = bertos_utils.getToolchainInfo(description)
 126             if len(infos.keys()) >= 4:
 127                 self._validItem(i, infos)
 128             else:
 129                 self._invalidItem(i)
 130         else:
 131             self._validationProcess.kill()
 132             self._invalidItem(i)
 133         toolchains = self.toolchains()
 134         toolchains[filename] = True
 135         self.setToolchains(toolchains)
 136
 137     def isComplete(self):
 138         if self.pageContent.toolchainList.currentRow() != -1:
 139             self._projectInfoStore("TOOLCHAIN",
 140                 qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole))["path"])
 141             return True
 142         else:
 143             return False
 144
 145     def reloadData(self):
 146         self._clearList()
 147         self._updateUi()
 148         self._populateToolchainList()