class RunConfigDialog(BaseRunConfigDialog):
"""Run configuration dialog box: multiple file version"""
def __init__(self, parent=None):
BaseRunConfigDialog.__init__(self, parent)
self.file_to_run = None
self.combo = None
self.stack = None
def run_btn_clicked(self):
"""Run button was just clicked"""
self.file_to_run = unicode(self.combo.currentText())
def setup(self, fname):
"""Setup Run Configuration dialog with filename *fname*"""
combo_label = QLabel(_("Select a run configuration:"))
self.combo = QComboBox()
self.combo.setMaxVisibleItems(20)
self.combo.setSizeAdjustPolicy(QComboBox.AdjustToMinimumContentsLength)
self.combo.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
self.stack = QStackedWidget()
configurations = _get_run_configurations()
for index, (filename, options) in enumerate(configurations):
if fname == filename:
break
else:
# There is no run configuration for script *fname*:
# creating a temporary configuration that will be kept only if
# dialog changes are accepted by the user
configurations.insert(0, (fname, RunConfiguration(fname).get()))
index = 0
for filename, options in configurations:
widget = RunConfigOptions(self)
widget.set(options)
self.combo.addItem(filename)
self.stack.addWidget(widget)
self.connect(self.combo, SIGNAL("currentIndexChanged(int)"),
self.stack.setCurrentIndex)
self.combo.setCurrentIndex(index)
self.add_widgets(combo_label, self.combo, 10, self.stack)
self.add_button_box(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
self.setWindowTitle(_("Run Settings"))
def accept(self):
"""Reimplement Qt method"""
configurations = []
for index in range(self.stack.count()):
filename = unicode(self.combo.itemText(index))
runconfigoptions = self.stack.widget(index)
if index == self.stack.currentIndex() and\
not runconfigoptions.is_valid():
return
options = runconfigoptions.get()
configurations.append((filename, options))
_set_run_configurations(configurations)
QDialog.accept(self)
class RunConfigDialog(BaseRunConfigDialog):
"""Run configuration dialog box: multiple file version"""
def __init__(self, parent=None):
BaseRunConfigDialog.__init__(self, parent)
self.file_to_run = None
self.combo = None
self.stack = None
def run_btn_clicked(self):
"""Run button was just clicked"""
self.file_to_run = unicode(self.combo.currentText())
def setup(self, fname):
"""Setup Run Configuration dialog with filename *fname*"""
combo_label = QLabel(_("Select a run configuration:"))
self.combo = QComboBox()
self.combo.setMaxVisibleItems(20)
self.combo.setSizeAdjustPolicy(QComboBox.AdjustToMinimumContentsLength)
self.combo.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
self.stack = QStackedWidget()
configurations = _get_run_configurations()
for index, (filename, options) in enumerate(configurations):
if fname == filename:
break
else:
# There is no run configuration for script *fname*:
# creating a temporary configuration that will be kept only if
# dialog changes are accepted by the user
configurations.insert(0, (fname, RunConfiguration(fname).get()))
index = 0
for filename, options in configurations:
widget = RunConfigOptions(self)
widget.set(options)
self.combo.addItem(filename)
self.stack.addWidget(widget)
self.connect(self.combo, SIGNAL("currentIndexChanged(int)"),
self.stack.setCurrentIndex)
self.combo.setCurrentIndex(index)
self.add_widgets(combo_label, self.combo, 10, self.stack)
self.add_button_box(QDialogButtonBox.Ok|QDialogButtonBox.Cancel)
self.setWindowTitle(_("Run Settings"))
def accept(self):
"""Reimplement Qt method"""
configurations = []
for index in range(self.stack.count()):
filename = unicode(self.combo.itemText(index))
runconfigoptions = self.stack.widget(index)
if index == self.stack.currentIndex() and\
not runconfigoptions.is_valid():
return
options = runconfigoptions.get()
configurations.append( (filename, options) )
_set_run_configurations(configurations)
QDialog.accept(self)
def createEditor(self,parent,option,index):
"""create widgets to edit values"""
if index.column() == 1:
cbox = QComboBox(parent)
cbox.setMaxVisibleItems(10)
self.updateEditorGeometry(cbox, option,index )
for pos,value in enumerate([0,20,30,40,50,60,70,80,90,100]):
icon = QPixmap(20,10)
icon.fill(self._getColorForProgress(value))
cbox.insertItem(pos, QIcon(icon),"",value)
cbox.showPopup()
return cbox
else:
return QItemDelegate.createEditor(self,parent,option,index)
class ConnectedComponentParamEditorWidget(QWidget):
def __init__(self, parent=None):
super(ConnectedComponentParamEditorWidget, self).__init__(parent)
hbox = QHBoxLayout()
hbox.addWidget(QLabel('Scene : '))
self._sceneQCB = QComboBox(self)
self._sceneQCB.setMaxVisibleItems(5)
hbox.addWidget(self._sceneQCB)
self._sceneQCB.clear()
self.setLayout(hbox)
def initSceneQCB(self):
try:
self._sceneQCB.currentIndexChanged.disconnect(self.changeScene)
except TypeError:
pass
sceneOfCC = self._selectedConnectedComponent.scene()
self._sceneQCB.clear()
i = 0
for scene in self.parent().parent().scenes():
self._sceneQCB.addItem(scene.getName())
if scene == sceneOfCC:
self._sceneQCB.setCurrentIndex(i)
i += 1
self._sceneQCB.currentIndexChanged.connect(self.changeScene)
def setSelectedConnectedComponent(self, cc):
self._selectedConnectedComponent = cc
self.initSceneQCB()
def changeScene(self, sceneIndex):
self._sceneQCB.clear()
self._selectedConnectedComponent.scene().changeConnectedComponentSceneByIndex(
self._selectedConnectedComponent, sceneIndex)
class GUI(QWidget):
def __init__(self, parent=None):
global f
f = open(filename, "a")
f.write("Widget init.\n")
f.close()
QWidget.__init__(self, parent, Qt.WindowStaysOnTopHint)
self.__setup_gui__(self)
self._flag = False
self._change = False
f = open(filename, "a")
f.write("End of widget init.\n")
f.close()
def closeEvent(self, event):
reply = QMessageBox.question(self, "Confirm",
"Are you sure You want to quit?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if reply == QMessageBox.Yes: event.accept()
else: event.ignore()
def __setup_gui__(self, Dialog):
global f
f = open(filename, "a")
f.write("Setup of gui.\n")
f.close()
Dialog.setObjectName("Dialog")
Dialog.resize(270, 145)
self.setWindowTitle("Map Layer")
screen = QDesktopWidget().screenGeometry()
size = self.geometry()
self.move((screen.width() - size.width()) / 2,
(screen.height() - size.height()) / 2)
self.Render = QPushButton("Render", Dialog)
self.Render.setGeometry(QRect(85, 90, 100, 25))
self.Render.setObjectName("Render")
self.comboBox = QComboBox(Dialog)
self.comboBox.setGeometry(QRect(100, 34, 115, 18))
self.comboBox.setEditable(False)
self.comboBox.setMaxVisibleItems(11)
self.comboBox.setInsertPolicy(QComboBox.InsertAtBottom)
self.comboBox.setObjectName("comboBox")
self.comboBox.addItems([
"Google Roadmap", "Google Terrain", "Google Satellite",
"Google Hybrid", "Yahoo Roadmap", "Yahoo Satellite",
"Yahoo Hybrid", "Bing Roadmap", "Bing Satellite", "Bing Hybrid",
"Open Street Maps"
])
self.comboBox.setCurrentIndex(10)
self.label1 = QLabel("Source:", Dialog)
self.label1.setGeometry(QRect(55, 35, 35, 16))
self.label1.setObjectName("label1")
self.slider = QSlider(Dialog)
self.slider.setOrientation(Qt.Horizontal)
self.slider.setMinimum(1)
self.slider.setMaximum(12)
self.slider.setValue(4)
self.slider.setGeometry(QRect(110, 61, 114, 16))
self.label2 = QLabel("Quality: " + str(self.slider.value()), Dialog)
self.label2.setGeometry(QRect(47, 61, 54, 16))
self.label2.setObjectName("label2")
self.doubleSpinBox = QDoubleSpinBox(Dialog)
self.doubleSpinBox.setGeometry(QRect(160, 5, 40, 20))
self.doubleSpinBox.setDecimals(0)
self.doubleSpinBox.setObjectName("doubleSpinBox")
self.doubleSpinBox.setMinimum(10.0)
self.doubleSpinBox.setValue(20.0)
self.doubleSpinBox.setEnabled(False)
self.checkBox = QCheckBox("Auto refresh", Dialog)
self.checkBox.setGeometry(QRect(50, 6, 100, 20))
self.checkBox.setLayoutDirection(Qt.RightToLeft)
self.checkBox.setObjectName("checkBox")
self.progressBar = QProgressBar(Dialog)
self.progressBar.setGeometry(QRect(5, 130, 260, 10))
self.progressBar.setProperty("value", 0)
self.progressBar.setTextVisible(False)
self.progressBar.setObjectName("progressBar")
self.progressBar.setVisible(False)
QObject.connect(self.Render, SIGNAL("clicked()"), Dialog.__repaint__)
QMetaObject.connectSlotsByName(Dialog)
QObject.connect(self.slider, SIGNAL("valueChanged(int)"),
self.__update_slider_label__)
QObject.connect(self.comboBox, SIGNAL("activated(int)"),
self.__combobox_changed__)
self.timerRepaint = QTimer()
QObject.connect(self.checkBox, SIGNAL("clicked()"),
self.__activate_timer__)
QObject.connect(self.timerRepaint, SIGNAL("timeout()"),
self.__on_timer__)
f = open(filename, "a")
f.write("End of setup of gui.\n")
f.close()
def __combobox_changed__(self):
self._change = True
def __activate_timer__(self):
self.doubleSpinBox.setEnabled(self.checkBox.isChecked())
if self.checkBox.isChecked():
self.timerRepaint.start(self.doubleSpinBox.value() * 1000)
self.Render.setEnabled(False)
if _progress == 0: self.__repaint__()
else:
self.timerRepaint.stop()
self.Render.setEnabled(True)
def __get_net_size__(self):
global f
f = open(filename, "a")
f.write("Geting net size...\n")
f.close()
if not os.path.exists(Paths["Screenshot"]):
Visum.Graphic.Screenshot(Paths["Screenshot"])
size = Image.open(Paths["Screenshot"]).size
f = open(filename, "a")
f.write("Read net size:" + str(size) + ".\n")
f.close()
return size
def __on_timer__(self):
global _paramGlobal
self._flag = False
Visum.Graphic.MaximizeNetWindow()
param = _paramGlobal
_paramGlobal = Visum.Graphic.GetWindow()
shift = abs((param[0] - _paramGlobal[0]) / (param[2] - param[0]))
zoom = abs((param[2] - param[0]) /
(_paramGlobal[2] - _paramGlobal[0]) - 1)
print _windowSizeGlobal
if _windowSizeGlobal[2:4] != Visum.Graphic.GetMainWindowPos()[2:4]:
self.__get_net_size__()
self._flag = True
elif shift > 0.4 or zoom > 0.2:
self._flag = True
if self._flag or self._change and _progress == 0:
self.__repaint__()
self._change = False
def __update_slider_label__(self, value):
self.label2.setText("Quality: " + str(value))
self._change = True
def __update_progress_bar__(self):
if _progress != 0:
self.progressBar.setVisible(True)
self.progressBar.setValue(_progress)
else:
self.progressBar.setVisible(False)
def __rebuild_paths__(self):
global Paths
Paths["Images"] = []
list = os.listdir(Paths["ScriptFolder"])
imageList = []
for i in range(len(list)):
if list[i][-3:] == "png": imageList.append(list[i])
for i in range(len(imageList)):
try:
Visum.Graphic.Backgrounds.ItemByKey(imageList[i])
Paths["Images"].append(Paths["ScriptFolder"] + "\\" +
imageList[i])
except:
pass
def __repaint__(self):
global _progress, f
if len(Visum.Graphic.Backgrounds.GetAll) != len(Paths["Images"]):
self.__rebuild_paths__()
if _progress == 0:
f = open(filename, "a")
f.write("Doing repaint...\n")
f.close()
QWebSettings.clearMemoryCaches()
timer = QTimer()
timer.start(100)
QObject.connect(timer, SIGNAL("timeout()"),
self.__update_progress_bar__)
Main(self.comboBox.currentIndex(), Visum.Graphic.GetWindow(),
self.slider.value() / 4.0, self.__get_net_size__())
Visum.Graphic.Draw()
self.__update_progress_bar__()
_progress = 0
QTimer().singleShot(1500, self.__update_progress_bar__)
f = open(filename, "a")
f.write("End of doing repaint.\n")
f.close()
class ArcParamEditorWidget(QWidget):
def __init__(self, parent=None):
super(ArcParamEditorWidget, self).__init__(parent)
hbox = QHBoxLayout()
vbox = QVBoxLayout()
hbox2 = QHBoxLayout()
self._lb1 = QLabel('Arc index : ', self)
self._indexQCB = QComboBox(self)
self._indexQCB.setMaxVisibleItems(5)
hbox2.addWidget(self._lb1)
hbox2.addWidget(self._indexQCB)
self._labelTE = QTextEdit(self)
self._labelTE.setUndoRedoEnabled(True)
vbox.addLayout(hbox2)
vbox.addWidget(self._labelTE)
self._formulaTE = QTextEdit(self)
self._formulaTE.setUndoRedoEnabled(True)
self._consequencesTE = QTextEdit(self)
self._consequencesTE.setUndoRedoEnabled(True)
hbox.addLayout(vbox)
hbox.addWidget(self._formulaTE)
hbox.addWidget(self._consequencesTE)
self.setLayout(hbox)
self._selectedArc = None
self._labelTE.textChanged.connect(self.labelChanged)
self._formulaTE.textChanged.connect(self.formulaChanged)
self._consequencesTE.textChanged.connect(self.consequencesChanged)
self._indexQCB.currentIndexChanged.connect(self.window().setModified)
self._labelTE.textChanged.connect(self.window().setModified)
self._formulaTE.textChanged.connect(self.window().setModified)
self._consequencesTE.textChanged.connect(self.window().setModified)
def init(self):
self._indexQCB.clear()
self.setLabel('Arc label.')
self.setFormula('Arc boolean formula.')
self.setConsequences('Arc consequeneces.')
def setIndexes(self, maxIndex, index):
self._indexQCB.clear()
for i in xrange(maxIndex):
self._indexQCB.addItem(str(i))
self._indexQCB.setCurrentIndex(index)
def setLabel(self, label):
self._labelTE.setText(label)
def setFormula(self, formula):
self._formulaTE.setText(formula)
def setConsequences(self, consequences):
self._consequencesTE.setText(consequences)
def labelChanged(self):
try:
self._selectedArc.setLabel(str(self._labelTE.toPlainText()))
except AttributeError:
pass
def formulaChanged(self):
try:
self._selectedArc.setFormula(str(self._formulaTE.toPlainText()))
except AttributeError:
pass
def consequencesChanged(self):
try:
self._selectedArc.setConsequences(str(self._consequencesTE.toPlainText()))
except AttributeError:
pass
def setSelectedArc(self, a):
try:
self._indexQCB.currentIndexChanged.disconnect(self._selectedArc.setIndex)
except (AttributeError, TypeError):
pass
self._selectedArc = a
try:
self.setIndexes(a.getMaxIndex(), a.getIndex())
self.setLabel(a.getLabel())
self.setFormula(a.getFormula())
self.setConsequences(a.getConsequencesStr())
self._indexQCB.currentIndexChanged.connect(a.setIndex)
except AttributeError:
self.init()
