If the target of a toolchain is not defined the wizard use the warning icon
[bertos.git] / wizard / BToolchainPage.py
index 4b2b4c5e21f1063833c0456d9402556eebd87e2a..100abb5666488b290da0f2e3c8d3085f21e0ace5 100644 (file)
 # Author: Lorenzo Berni <duplo@develer.com>
 #
 
+import os
+import collections
+
 from BWizardPage import *
+import BToolchainSearch
 import bertos_utils
+import qvariant_converter
+
+from const import *
 
 class BToolchainPage(BWizardPage):
     
     def __init__(self):
-        BWizardPage.__init__(self, "toolchain_select.ui")
+        BWizardPage.__init__(self, UI_LOCATION + "/toolchain_select.ui")
         self.setTitle(self.tr("Select toolchain"))
-        self._setupUi()
-        self._populateDirList()
-        self._connectSignals()
-    
-    def _setupUi(self):
-        path = self._settingsRetrieve("path_search").toBool()
-        if not path is None:
-            self.pageContent.pathBox.setChecked(path)
-        else:
-            self.pageContent.pathBox.setChecked(False)
-        customPath = self._settingsRetrieve("custom_dir").toBool()
-        if not customPath is None:
-            self.pageContent.customDirBox.setChecked(customPath)
-        else:
-            self.pageContent.customDirBox.setChecked(False)
+        self._validationProcess = None
         self._updateUi()
+        #self._populateToolchainList()
+        self._connectSignals()
     
     def _updateUi(self):
-        if self.pageContent.customDirBox.isChecked():
-            self._enableCustomDir()
-        else:
-            self._disableCustomDir()
-        if self.pageContent.pathBox.isChecked() or (self.pageContent.customDirBox.isChecked() and self.pageContent.customDirList.count() != 0):
-            self.pageContent.doSearchButton.setEnabled(True)
-        else:
-            self.pageContent.doSearchButton.setEnabled(False)
-    
-    def _populateDirList(self):
-        search_dir_list = self._settingsRetrieve("search_dir_list").toList()
-        for element in search_dir_list:
-            item = QListWidgetItem(element.toString())
-            item.setData(Qt.UserRole, element)
-            self.pageContent.customDirList.addItem(item)
-        
-    def _disableCustomDir(self):
-        self.pageContent.customDirList.setEnabled(False)
-        self.pageContent.addDirButton.setEnabled(False)
-        self.pageContent.removeDirButton.setEnabled(False)
-    
-    def _enableCustomDir(self):
-        self.pageContent.customDirList.setEnabled(True)
-        self.pageContent.addDirButton.setEnabled(True)
-        self.pageContent.removeDirButton.setEnabled(True)
+        self.pageContent.infoLabel.setVisible(False)
     
+    def _populateToolchainList(self):
+        toolchains = self.toolchains()
+        for key, value in toolchains.items():
+            item = QListWidgetItem(key)
+            item.setData(Qt.UserRole, qvariant_converter.convertStringDict({"path": key}))
+            self.pageContent.toolchainList.addItem(item)
+            if value:
+                self.validateToolchain(self.pageContent.toolchainList.row(item))
+
     def _clearList(self):
         self.pageContent.toolchainList.clear()
     
+    def _selectionChanged(self):
+        if self.pageContent.toolchainList.currentIndex != -1:
+            infos = collections.defaultdict(lambda: unicode("not defined"))
+            infos.update(qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
+            self.pageContent.infoLabel.setText("GCC " + infos["version"] + " (" + infos["build"] + ")\nTarget: " + infos["target"] + "\nPath: " + os.path.normpath(infos["path"]))
+            self.pageContent.infoLabel.setVisible(True)
+            self.emit(SIGNAL("completeChanged()"))
+    
+    def _search(self):
+        dirList = self.searchDirList()
+        if self.pathSearch():
+            dirList += [element for element in bertos_utils.getSystemPath()]
+        toolchainList = bertos_utils.findToolchains(dirList)
+        storedToolchains = self.toolchains()
+        for element in toolchainList:
+            if not element in storedToolchains.keys():
+                item = QListWidgetItem(element)
+                item.setData(Qt.UserRole, qvariant_converter.convertStringDict({"path": element}))
+                self.pageContent.toolchainList.addItem(item)
+                storedToolchains[element] = False
+        self.setToolchains(storedToolchains)
+        
     def _connectSignals(self):
-        self.connect(self.pageContent.pathBox, SIGNAL("clicked()"), self._updateUi)
-        self.connect(self.pageContent.customDirBox, SIGNAL("clicked()"), self._updateUi)
-        self.connect(self.pageContent.searchButton, SIGNAL("clicked()"), self.toSearchSubpage)
-        self.connect(self.pageContent.cancelButton, SIGNAL("clicked()"), self.toSelectionSubpage)
-        self.connect(self.pageContent.doSearchButton, SIGNAL("clicked()"), self.doSearch)
-        self.connect(self.pageContent.addDirButton, SIGNAL("clicked()"), self.addDir)
-        self.connect(self.pageContent.removeDirButton, SIGNAL("clicked()"), self.removeDir)
+        self.connect(self.pageContent.toolchainList, SIGNAL("itemSelectionChanged()"), self._selectionChanged)
+        self.connect(self.pageContent.addButton, SIGNAL("clicked()"), self.addToolchain)
+        self.connect(self.pageContent.removeButton, SIGNAL("clicked()"), self.removeToolchain)
+        self.connect(self.pageContent.searchButton, SIGNAL("clicked()"), self.searchToolchain)
+        self.connect(self.pageContent.checkButton, SIGNAL("clicked()"), self.validateAllToolchains)
     
-    def toSearchSubpage(self):
-        self.pageContent.pageSelector.setCurrentIndex(1)
-
-    def toSelectionSubpage(self):
-        self.pageContent.pageSelector.setCurrentIndex(0)    
-
-    def doSearch(self):
-        path = []
-        if self.pageContent.pathBox.isChecked():
-            path += bertos_utils.getSystemPath()
-        if self.pageContent.customDirBox.isChecked():
-            for element in range(self.pageContent.customDirList.count()):
-                path += [unicode(self.pageContent.customDirList.item(element).data(Qt.UserRole).toString())]
-        toolchains = bertos_utils.findToolchains(path)
-        self._clearList()
-        for toolchain in toolchains:
-            self.pageContent.toolchainList.addItem(QListWidgetItem(toolchain))
-        self.toSelectionSubpage()
+    def _validItem(self, index, infos):
+        item = self.pageContent.toolchainList.item(index)
+        newData = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(index).data(Qt.UserRole))
+        newData.update(infos)
+        item.setData(Qt.UserRole, qvariant_converter.convertStringDict(newData))
+        needed = self._projectInfoRetrieve("CPU_INFOS")
+        if "target" in infos.keys() and infos["target"].find(needed["TOOLCHAIN"]) != -1:
+            item.setIcon(QIcon(":/images/ok.png"))
+        else:
+            item.setIcon(QIcon(":/images/warning.png"))
+        item.setText("GCC " + infos["version"] + " - " + infos["target"])
+    
+    def _invalidItem(self, index):
+        item = self.pageContent.toolchainList.item(index)
+        item.setIcon(QIcon(":/images/error.png"))
+    
+    def addToolchain(self):
+        sel_toolchain = QFileDialog.getOpenFileName(self, self.tr("Choose the toolchain"), "")
+        if not sel_toolchain.isEmpty():
+            item = QListWidgetItem(sel_toolchain)
+            item.setData(Qt.UserRole, qvariant_converter.convertString({"path": sel_toolchain}))
+            self.pageContent.toolchainList.addItem(item)
+            toolchains = self.toolchains()
+            toolchains[sel_toolchain] = False
+            self.setToolchains(toolchains)
+    
+    def removeToolchain(self):
+        if self.pageContent.toolchainList.currentRow() != -1:
+            item = self.pageContent.toolchainList.takeItem(self.pageContent.toolchainList.currentRow())
+            toolchain = qvariant_converter.getStringDict(item.data(Qt.UserRole))["path"]
+            toolchains = self.toolchains()
+            del toolchains[toolchain]
+            self.setToolchains(toolchains)
     
-    def addDir(self):
-        directory = QFileDialog.getExistingDirectory(self, self.tr("Open Directory"), "", QFileDialog.ShowDirsOnly)
-        if not directory.isEmpty():
-            item = QListWidgetItem(directory)
-            item.setData(Qt.UserRole, QVariant(directory))
-            self.pageContent.customDirList.addItem(item)
-            search_dir_list = self._settingsRetrieve("search_dir_list").toList()
-            search_dir_list = set([d.toString() for d in search_dir_list] + [directory])
-            self._settingsStore("search_dir_list", list(search_dir_list))
+    def searchToolchain(self):
+        search = BToolchainSearch.BToolchainSearch()
+        self.connect(search, SIGNAL("accepted()"), self._search)
+        search.exec_()
     
-    def removeDir(self):
-        item = self.pageContent.customDirList.takeItem(self.pageContent.customDirList.currentRow())
-        search_dir_list = self._settingsRetrieve("search_dir_list").toList()
-        search_dir_list = set([d.toString() for d in search_dir_list])
-        search_dir_list.remove(item.data(Qt.UserRole).toString())
-        self._settingsStore("search_dir_list", list(search_dir_list))
\ No newline at end of file
+    def validateAllToolchains(self):
+        for i in range(self.pageContent.toolchainList.count()):
+            self.validateToolchain(i)
+    
+    def validateToolchain(self, i):
+        filename = qvariant_converter.getStringDict(self.pageContent.toolchainList.item(i).data(Qt.UserRole))["path"]
+        valid = False
+        info = {}
+        ## Check for the other tools of the toolchain
+        for tool in TOOLCHAIN_ITEMS:
+            if os.path.exists(filename.replace("gcc", tool)):
+                valid = True
+            else:
+                valid = False
+                break
+        ## Try to retrieve the informations about the toolchain only for the valid toolchains
+        if valid:
+            self._validationProcess = QProcess()
+            self._validationProcess.start(filename, ["-v"])
+            self._validationProcess.waitForStarted(1000)
+            if self._validationProcess.waitForFinished(200):
+                description = str(self._validationProcess.readAllStandardError())
+                info = bertos_utils.getToolchainInfo(description)
+                if len(info.keys()) >= 4:
+                    valid = True
+            else:
+                self._validationProcess.kill()
+        ## Add the item in the list with the appropriate associate data.
+        if valid:
+            self._validItem(i, info)
+        else:
+            self._invalidItem(i)
+        toolchains = self.toolchains()
+        toolchains[filename] = True
+        self.setToolchains(toolchains)
+    
+    def isComplete(self):
+        if self.pageContent.toolchainList.currentRow() != -1:
+            self._projectInfoStore("TOOLCHAIN", 
+                qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
+            return True
+        else:
+            return False
+    
+    def reloadData(self):
+        self._clearList()
+        self._updateUi()
+        self._populateToolchainList()
\ No newline at end of file