class FactorySelector(qt.QtGui.QDialog, ui_tofactory.Ui_FactorySelector):
""" Dialog to select a particular CompositeNode factory """
def __init__(self, default_factory=None, parent=None):
"""
Construtor
@param default_factory : default selected factory
"""
qt.QtGui.QDialog.__init__(self, parent)
ui_tofactory.Ui_FactorySelector.__init__(self)
self.setupUi(self)
self.pkgmanager = PackageManager()
self.factorymap = {}
self.default_factory = default_factory
cfactories = []
# Get all composite node factories in writable packages
for pkg in self.pkgmanager.get_user_packages():
pname = pkg.name
for f in pkg.itervalues():
if (isinstance(f, CompositeNodeFactory)):
name = "%s.%s" % (pname, f.name)
cfactories.append(name)
self.factorymap[name] = f
cfactories.sort()
self.comboBox.addItems(cfactories)
i = -1
if (default_factory and default_factory.package):
name = "%s.%s" % (default_factory.package.name,
default_factory.name)
i = self.comboBox.findText(name)
self.comboBox.setCurrentIndex(i)
self.connect(self.newFactoryButton, qt.QtCore.SIGNAL("clicked()"),
self.new_factory)
def accept(self):
""" Accept dialog result """
# Test if name is correct
text = self.comboBox.currentText()
if (not text):
mess = qt.QtGui.QMessageBox.warning(self, "Error",
"Invalid Choice.")
return
qt.QtGui.QDialog.accept(self)
def new_factory(self):
""" Create a new composite node """
if (self.default_factory and self.default_factory.package):
pkg_id = self.default_factory.package.name
else:
pkg_id = None
dialog = NewGraph("New Composite Node",
self.pkgmanager,
self,
io=False,
pkg_id=pkg_id)
ret = dialog.exec_()
if (ret > 0):
newfactory = dialog.create_cnfactory(self.pkgmanager)
name = "%s.%s" % (newfactory.package.name, newfactory.name)
self.comboBox.addItem(name)
self.factorymap[name] = newfactory
i = self.comboBox.findText(name)
self.comboBox.setCurrentIndex(i)
self.accept()
def get_factory(self):
""" Return the selected factory """
text = unicode(self.comboBox.currentText()).encode('latin1')
return self.factorymap[str(text)]
class FactorySelector(qt.QtGui.QDialog, ui_tofactory.Ui_FactorySelector):
""" Dialog to select a particular CompositeNode factory """
def __init__(self, default_factory=None, parent=None):
"""
Construtor
@param default_factory : default selected factory
"""
qt.QtGui.QDialog.__init__(self, parent)
ui_tofactory.Ui_FactorySelector.__init__(self)
self.setupUi(self)
self.pkgmanager = PackageManager()
self.factorymap = {}
self.default_factory = default_factory
cfactories = []
# Get all composite node factories in writable packages
for pkg in self.pkgmanager.get_user_packages():
pname = pkg.name
for f in pkg.itervalues():
if isinstance(f, CompositeNodeFactory):
name = "%s.%s" % (pname, f.name)
cfactories.append(name)
self.factorymap[name] = f
cfactories.sort()
self.comboBox.addItems(cfactories)
i = -1
if default_factory and default_factory.package:
name = "%s.%s" % (default_factory.package.name, default_factory.name)
i = self.comboBox.findText(name)
self.comboBox.setCurrentIndex(i)
self.connect(self.newFactoryButton, qt.QtCore.SIGNAL("clicked()"), self.new_factory)
def accept(self):
""" Accept dialog result """
# Test if name is correct
text = self.comboBox.currentText()
if not text:
mess = qt.QtGui.QMessageBox.warning(self, "Error", "Invalid Choice.")
return
qt.QtGui.QDialog.accept(self)
def new_factory(self):
""" Create a new composite node """
if self.default_factory and self.default_factory.package:
pkg_id = self.default_factory.package.name
else:
pkg_id = None
dialog = NewGraph("New Composite Node", self.pkgmanager, self, io=False, pkg_id=pkg_id)
ret = dialog.exec_()
if ret > 0:
newfactory = dialog.create_cnfactory(self.pkgmanager)
name = "%s.%s" % (newfactory.package.name, newfactory.name)
self.comboBox.addItem(name)
self.factorymap[name] = newfactory
i = self.comboBox.findText(name)
self.comboBox.setCurrentIndex(i)
self.accept()
def get_factory(self):
""" Return the selected factory """
text = unicode(self.comboBox.currentText()).encode("latin1")
return self.factorymap[str(text)]
