Store the entire toolchain information in the project
[bertos.git] / 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: " + 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         self._validationProcess = QProcess()
122         self._validationProcess.start(filename, ["-v"])
123         self._validationProcess.waitForStarted(1000)
124         if self._validationProcess.waitForFinished(200):
125             description = str(self._validationProcess.readAllStandardError())
126             infos = bertos_utils.getToolchainInfo(description)
127             if len(infos.keys()) >= 4:
128                 self._validItem(i, infos)
129             else:
130                 self._invalidItem(i)
131         else:
132             self._validationProcess.kill()
133             self._invalidItem(i)
134         toolchains = self.toolchains()
135         toolchains[filename] = True
136         self.setToolchains(toolchains)
137     
138     def isComplete(self):
139         if self.pageContent.toolchainList.currentRow() != -1:
140             self._projectInfoStore("TOOLCHAIN", 
141                 qvariant_converter.getStringDict(self.pageContent.toolchainList.currentItem().data(Qt.UserRole)))
142             return True
143         else:
144             return False
145     
146     def reloadData(self):
147         self._clearList()
148         self._updateUi()
149         self._populateToolchainList()