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