def nameChanged(self):
old_alias = self.alias.name
new_alias = str(self.name_edit.text())
if old_alias == new_alias:
return
if new_alias in self.table.aliases.keys():
show_warning(
"Mashup",
"""Label name %s already exists.
Please type a different name. """
% new_alias,
)
self.name_edit.setText(old_alias)
self.name_edit.setFocus()
elif new_alias == "":
show_warning(
"Mashup",
"""Variables with empty name are not allowed.
Please type a unique name. """
% new_alias,
)
self.name_edit.setText(old_alias)
self.name_edit.setFocus()
else:
self.table.aliases[new_alias] = self.table.aliases[old_alias]
# self.table.alias_cache[new_alias] = self.table.alias_cache[old_alias]
del self.table.aliases[old_alias]
# del self.table.alias_cache[old_alias]
self.alias.name = new_alias
self.aliasChanged.emit(self.alias)
def updateCurrentTag(self, name):
if BaseController.updateCurrentTag(self, name) == False:
show_warning('Name Exists',
"There is already another version named '%s'.\n"
"Please enter a different one." % name)
return False
else:
return True
def updateCurrentTag(self, name):
if BaseController.updateCurrentTag(self, name) == False:
show_warning(
'Name Exists', "There is already another version named '%s'.\n"
"Please enter a different one." % name)
return False
else:
return True
def compute(self):
# if server, grab local file using checksum id
if self.is_server:
self.check_input('checksum')
self.checksum = self.get_input("checksum")
# get file path
path_url = "%s/datasets/path/%s/"%(self.base_url, self.checksum)
try:
dataset_path_request = urllib2.urlopen(url=path_url)
dataset_path = dataset_path_request.read()
except urllib2.HTTPError:
pass
if os.path.isfile(dataset_path):
out_file = PathObject(dataset_path)
self.set_output("file", out_file)
else: # is client
self.check_input('file')
self.in_file = self.get_input("file")
if os.path.isfile(self.in_file.name):
# do size check
size = os.path.getsize(self.in_file.name)
if size > 26214400:
show_warning("File is too large",
"file is larger than 25MB, "
"unable to sync with web repository")
self.set_output("file", self.in_file)
else:
# compute checksum
f = open(self.in_file.name, 'r')
self.checksum = hashlib.sha1()
block = 1
while block:
block = f.read(128)
self.checksum.update(block)
f.close()
self.checksum = self.checksum.hexdigest()
# upload/download file
self.data_sync()
# set checksum param in module
if not self.has_input('checksum'):
self.change_parameter('checksum', [self.checksum])
else:
# local file not present
if self.has_input('checksum'):
self.checksum = self.get_input("checksum")
# download file
self.data_sync()
def compute(self):
# if server, grab local file using checksum id
if self.is_server:
self.check_input('checksum')
self.checksum = self.get_input("checksum")
# get file path
path_url = "%s/datasets/path/%s/"%(self.base_url, self.checksum)
dataset_path_request = urllib2.urlopen(url=path_url)
dataset_path = dataset_path_request.read()
if os.path.isfile(dataset_path):
out_file = PathObject(dataset_path)
self.set_output("file", out_file)
else: # is client
self.check_input('file')
self.in_file = self.get_input("file")
if os.path.isfile(self.in_file.name):
# do size check
size = os.path.getsize(self.in_file.name)
if size > 26214400:
show_warning("File is too large",
"file is larger than 25MB, "
"unable to sync with web repository")
self.set_output("file", self.in_file)
else:
# compute checksum
f = open(self.in_file.name, 'r')
self.checksum = hashlib.sha1()
block = 1
while block:
block = f.read(128)
self.checksum.update(block)
f.close()
self.checksum = self.checksum.hexdigest()
# upload/download file
self.data_sync()
# set checksum param in module
if not self.has_input('checksum'):
self.change_parameter('checksum', [self.checksum])
else:
# local file not present
if self.has_input('checksum'):
self.checksum = self.get_input("checksum")
# download file
self.data_sync()
def thumbs_cache_directory_changed(self):
""" thumbs_cache_changed(v: int) -> None
"""
value = str(self._thumbs_cache_directory_edt.text())
old_folder = self._configuration.thumbs.cacheDirectory
if os.path.exists(value):
self._configuration.thumbs.cacheDirectory = value
self._temp_configuration.thumbs.cacheDirectory = value
self.emit(QtCore.SIGNAL('configuration_changed'), None, value)
self._cache.move_cache_directory(old_folder, value)
else:
show_warning('VisTrails',
'The directory specified does not exist.')
self._thumbs_cache_directory_edt.setText(old_folder)
def thumbs_cache_directory_changed(self):
""" thumbs_cache_changed(v: int) -> None
"""
value = str(self._thumbs_cache_directory_edt.text())
old_folder = self._configuration.thumbs.cacheDirectory
if os.path.exists(value):
self._configuration.thumbs.cacheDirectory = value
self._temp_configuration.thumbs.cacheDirectory = value
self.emit(QtCore.SIGNAL('configuration_changed'),
None, value)
self._cache.move_cache_directory(old_folder,value)
else:
show_warning('VisTrails', 'The directory specified does not exist.')
self._thumbs_cache_directory_edt.setText(old_folder)
def addParameter(self, paramInfo):
""" addParameter(paramInfo: (str, [ParameterInfo]) -> None
Add a parameter to the table. The parameter info is specified
in QParameterTreeWidgetItem
"""
# Check to see paramInfo is not a subset of some other parameter set
params = paramInfo[1]
for i in xrange(self.layout().count()):
pEditor = self.layout().itemAt(i).widget()
if pEditor and isinstance(pEditor, QParameterSetEditor):
subset = True
for p in params:
if not (p in pEditor.info[1]):
subset = False
break
if subset:
show_warning(
'Parameter Exists',
'The parameter you are trying to add is '
'already in the list.')
return
self.showPrompt(False)
newEditor = QParameterSetEditor(paramInfo, self)
# Make sure to disable all duplicated parameter
for p in xrange(len(params)):
for i in xrange(self.layout().count()):
pEditor = self.layout().itemAt(i).widget()
if pEditor and isinstance(pEditor, QParameterSetEditor):
if params[p] in pEditor.info[1]:
widget = newEditor.paramWidgets[p]
widget.setDimension(4)
widget.setDuplicate(True)
widget.setEnabled(False)
break
self.layout().addWidget(newEditor)
newEditor.show()
self.setMinimumHeight(self.layout().minimumSize().height())
self.emit(QtCore.SIGNAL('exploreChange(bool)'),
self.layout().count() > 3)
return newEditor
def exportMashupGUI(self):
if self.controller:
dialog = QMashupExportDialog(self)
if dialog.exec_() == QtGui.QDialog.Accepted:
result = dialog.btnPressed
fileName = QtGui.QFileDialog.getSaveFileName(
self,
"Export Mashup...",
vistrails.core.system.vistrails_file_directory(),
"VisTrail link files (*.vtl)")
if fileName:
filename = str(fileName)
res = MashupsManager.exportMashup(filename,
self.controller.originalController,
self.controller.mshptrail,
self.controller.currentVersion,
result)
if not res:
show_warning('VisTrails - Exporting Mashup',
'There was an error and the mashup could not be exported.')
def exportMashupGUI(self):
if self.controller:
dialog = QMashupExportDialog(self)
if dialog.exec_() == QtGui.QDialog.Accepted:
result = dialog.btnPressed
fileName = QtGui.QFileDialog.getSaveFileName(
self, "Export Mashup...",
vistrails.core.system.vistrails_file_directory(),
"VisTrail link files (*.vtl)")
if fileName:
filename = str(fileName)
res = MashupsManager.exportMashup(
filename, self.controller.originalController,
self.controller.mshptrail,
self.controller.currentVersion, result)
if not res:
show_warning(
'VisTrails - Exporting Mashup',
'There was an error and the mashup could not be exported.'
)
def addParameter(self, paramInfo):
""" addParameter(paramInfo: (str, [ParameterInfo]) -> None
Add a parameter to the table. The parameter info is specified
in QParameterTreeWidgetItem
"""
# Check to see paramInfo is not a subset of some other parameter set
params = paramInfo[1]
for i in xrange(self.layout().count()):
pEditor = self.layout().itemAt(i).widget()
if pEditor and isinstance(pEditor, QParameterSetEditor):
subset = True
for p in params:
if not (p in pEditor.info[1]):
subset = False
break
if subset:
show_warning('Parameter Exists',
'The parameter you are trying to add is '
'already in the list.')
return
self.showPrompt(False)
newEditor = QParameterSetEditor(paramInfo, self)
# Make sure to disable all duplicated parameter
for p in xrange(len(params)):
for i in xrange(self.layout().count()):
pEditor = self.layout().itemAt(i).widget()
if pEditor and isinstance(pEditor, QParameterSetEditor):
if params[p] in pEditor.info[1]:
widget = newEditor.paramWidgets[p]
widget.setDimension(4)
widget.setDuplicate(True)
widget.setEnabled(False)
break
self.layout().addWidget(newEditor)
newEditor.show()
self.setMinimumHeight(self.layout().minimumSize().height())
self.emit(QtCore.SIGNAL('exploreChange(bool)'), self.layout().count() > 3)
return newEditor
def get_values(self, count):
""" get_values(count) -> []
Convert the list values into a list
count should be an integer with the expected size of the list (given
by the dimension 'size' in the exploration)
"""
param_info = self._param_info
module = param_info.spec.descriptor.module
result = [module.translate_to_python(m) for m in self._str_values]
if len(result) != count:
show_warning(
'Inconsistent Size', 'One of the <i>%s</i>\'s list '
'interpolated '
'values has a different '
'size from the step count. '
'Expected %d, got %d instead. '
'Parameter Exploration aborted.' %
(self.type, count, len(result)))
return None
return result
def nameChanged(self):
old_alias = self.alias.name
new_alias = str(self.name_edit.text())
if old_alias == new_alias:
return
if new_alias in self.table.aliases.keys():
show_warning(
"Mashup", "Label name %s already exists. "
"Please type a different name." % new_alias)
self.name_edit.setText(old_alias)
self.name_edit.setFocus()
elif new_alias == '':
show_warning(
"Mashup", "Variables with empty name are not allowed. "
"Please type a unique name.")
self.name_edit.setText(old_alias)
self.name_edit.setFocus()
else:
self.table.aliases[new_alias] = self.table.aliases[old_alias]
#self.table.alias_cache[new_alias] = self.table.alias_cache[old_alias]
del self.table.aliases[old_alias]
#del self.table.alias_cache[old_alias]
self.alias.name = new_alias
self.aliasChanged.emit(self.alias)
def get_values(self, count):
""" get_values(count) -> []
Convert the list values into a list
count should be an integer with the expected size of the list (given
by the dimension 'size' in the exploration)
"""
param_info = self._param_info
module = param_info.spec.descriptor.module
result = [module.translate_to_python(m)
for m in self._str_values]
if len(result) != count:
show_warning('Inconsistent Size',
'One of the <i>%s</i>\'s list '
'interpolated '
'values has a different '
'size from the step count. '
'Expected %d, got %d instead. '
'Parameter Exploration aborted.'
% (self.type, count, len(result)))
return None
return result
def data_sync(self):
""" downloads/uploads/uses the local file depending on availability """
self.checksum_lookup()
# local file not on repository, so upload
if not self.on_server and os.path.isfile(self.in_file.name):
cookiejar = vistrails.gui.repository.QRepositoryDialog.cookiejar
if cookiejar:
register_openers(cookiejar=cookiejar)
params = {
'dataset_file': open(self.in_file.name, 'rb'),
'name': self.in_file.name.split('/')[-1],
'origin': 'vistrails',
'checksum': self.checksum
}
upload_url = "%s/datasets/upload/" % self.base_url
datagen, headers = multipart_encode(params)
request = urllib2.Request(upload_url, datagen, headers)
try:
result = urllib2.urlopen(request)
if result.code != 200:
show_warning("Upload Failure",
"Data failed to upload to repository")
# make temporarily uncachable
self.is_cacheable = self.invalidate_cache
else:
debug.warning("Push to repository was successful")
# make sure module caches
self.is_cacheable = self.validate_cache
except Exception, e:
show_warning("Upload Failure",
"Data failed to upload to repository")
# make temporarily uncachable
self.is_cacheable = self.invalidate_cache
debug.warning('RepoSync uploaded %s to the repository' % \
self.in_file.name)
else:
show_warning("Please login", ("You must be logged into the web"
" repository in order to upload "
"data. No data was synced"))
# make temporarily uncachable
self.is_cacheable = self.invalidate_cache
# use local data
self.set_output("file", self.in_file)
def data_sync(self):
""" downloads/uploads/uses the local file depending on availability """
self.checksum_lookup()
# local file not on repository, so upload
if not self.on_server and os.path.isfile(self.in_file.name):
cookiejar = vistrails.gui.repository.QRepositoryDialog.cookiejar
if cookiejar:
register_openers(cookiejar=cookiejar)
params = {'dataset_file': open(self.in_file.name, 'rb'),
'name': self.in_file.name.split('/')[-1],
'origin': 'vistrails',
'checksum': self.checksum}
upload_url = "%s/datasets/upload/" % self.base_url
datagen, headers = multipart_encode(params)
request = urllib2.Request(upload_url, datagen, headers)
try:
result = urllib2.urlopen(request)
if result.code != 200:
show_warning("Upload Failure",
"Data failed to upload to repository")
# make temporarily uncachable
self.is_cacheable = self.invalidate_cache
else:
debug.warning("Push to repository was successful")
# make sure module caches
self.is_cacheable = self.validate_cache
except Exception, e:
show_warning("Upload Failure",
"Data failed to upload to repository")
# make temporarily uncachable
self.is_cacheable = self.invalidate_cache
debug.warning('RepoSync uploaded %s to the repository' % \
self.in_file.name)
else:
show_warning("Please login", ("You must be logged into the web"
" repository in order to upload "
"data. No data was synced"))
# make temporarily uncachable
self.is_cacheable = self.invalidate_cache
# use local data
self.set_output("file", self.in_file)
if v == None:
return module.default_value
return v
except Exception, e:
debug.unexpected_exception(e)
return debug.format_exception(e)
return [evaluate(i) for i in xrange(self.size)]
result = get()
if not all(module.validate(x) for x in result):
show_warning(
'Failed Validation', 'One of the <i>%s</i>\'s user defined '
'functions has failed validation, '
'which usually means it generated a '
'value of a type different '
'than that specified by the '
'parameter. Parameter Exploration '
'aborted.' % param_info.spec.descriptor.name)
return None
return result
def getValuesString(self):
""" getValuesString() -> str
Return a string representation of the parameter list
"""
r = self.get_values(self.size)
if r is None:
return '{ERROR}'
else:
def __init__(self,
parent=None,
vistrail_view=None,
dumpcells=False,
controller=None,
version=-1):
""" QMashupAppMainWindow()
Initialize an app window from a mashup.
"""
# Constructing the main widget
QtGui.QMainWindow.__init__(self, parent)
self.vtkCells = []
self.setStatusBar(QtGui.QStatusBar(self))
# Central widget
centralWidget = QtGui.QWidget()
self.mainLayout = QtGui.QVBoxLayout()
self.mainLayout.setMargin(0)
self.mainLayout.setSpacing(5)
centralWidget.setLayout(self.mainLayout)
self.setCentralWidget(centralWidget)
self.numberOfCells = 0
self.is_executing = False
self.sequenceOption = False
self.steps = []
self.isLooping = False
#self.resize(100,100)
self.dumpcells = dumpcells
self.view = vistrail_view
if controller:
self.controller = controller
self.mshptrail = controller.mshptrail
if version == -1:
self.currentMashup = self.controller.currentMashup
else:
self.currentMashup = self.mshptrail.getMashup(version)
self.setWindowTitle('%s Mashup' %
self.controller.getMashupName(version))
else:
self.setWindowTitle('Mashup')
# Assign "hidden" shortcut
self.editingModeAct = QtGui.QAction(
"Chang&e Layout",
self,
shortcut="Ctrl+E",
statusTip="Change the layout of the widgets",
triggered=self.toggleEditingMode)
#self.editingModeShortcut = QtGui.QShortcut(QtGui.QKeySequence('Ctrl+E'), self)
#self.connect(self.editingModeShortcut, QtCore.SIGNAL('activated()'),
# self.toggleEditingMode)
self.editing = False
# Constructing alias controls
self.controlDocks = []
# Show here to make sure XDisplay info is correct (for VTKCell)
self.show()
spreadsheetController.setEchoMode(True)
#will run to get Spreadsheet Cell events
(cellEvents, errors) = self.runAndGetCellEvents(useDefaultValues=True)
if cellEvents:
self.numberOfCells = len(cellEvents)
self.initCells(cellEvents)
if len(errors) > 0:
show_warning(
"VisTrails::Mashup Preview",
"There was a problem executing the pipeline: %s." % errors)
# Construct the controllers for aliases
self.controlDocks = {}
self.cellControls = {}
self.aliasWidgets = {}
self.initControls()
if self.currentMashup.layout is not None:
self.restoreState(
QtCore.QByteArray.fromPercentEncoding(
QtCore.QByteArray(self.currentMashup.layout)))
if self.currentMashup.geometry is not None:
self.restoreGeometry(
QtCore.QByteArray.fromPercentEncoding(
QtCore.QByteArray(self.currentMashup.geometry)))
else:
self.resize(self.sizeHint())
# Constructing buttons
buttonDock = QCustomDockWidget('Control Buttons', self)
buttonWidget = QtGui.QWidget(buttonDock)
buttonWidget.setSizePolicy(QtGui.QSizePolicy.Preferred,
QtGui.QSizePolicy.Preferred)
buttonLayout = QtGui.QGridLayout()
buttonWidget.setLayout(buttonLayout)
buttonLayout.setMargin(5)
self.cb_auto_update = QtGui.QCheckBox("Turn on auto-update",
self.centralWidget())
self.cb_auto_update.setChecked(False)
self.cb_loop_sequence = QtGui.QCheckBox("Render all steps in '%s'" % \
(self.sequenceOption.alias.name if self.sequenceOption else 'None'),
self.centralWidget())
self.cb_loop_sequence.setChecked(False)
self.cb_loop_sequence.setVisible(self.sequenceOption is not False)
self.cb_loop_sequence.setToolTip(
"Render each step of this stepper for fast switching")
self.cb_loop_int = QtGui.QCheckBox("Interactive Steps",
self.centralWidget())
self.cb_loop_int.setChecked(False)
self.cb_loop_int.setVisible(False)
self.cb_loop_int.setToolTip(
"Show complete result of each step instead of static images")
self.cb_loop_sequence.clicked.connect(self.cb_loop_int.setVisible)
self.cb_keep_camera = QtGui.QCheckBox("Keep camera position",
self.centralWidget())
self.cb_keep_camera.setChecked(True)
self.connect(self.cb_auto_update, QtCore.SIGNAL("stateChanged(int)"),
self.auto_update_changed)
self.connect(self.cb_loop_int, QtCore.SIGNAL("stateChanged(int)"),
self.loop_int_changed)
self.loopButton = QtGui.QPushButton("&Loop", self.centralWidget())
self.loopButton.setToolTip("Loop automatically through steps")
self.loopButton.setCheckable(True)
self.loopButton.setVisible(self.sequenceOption is not False)
self.updateButton = QtGui.QPushButton("&Update", self.centralWidget())
if self.dumpcells:
self.quitButton = QtGui.QPushButton("&Save", self.centralWidget())
self.connect(self.quitButton, QtCore.SIGNAL('clicked(bool)'),
self.saveAndExport)
else:
self.quitButton = QtGui.QPushButton("&Quit", self.centralWidget())
self.connect(self.quitButton, QtCore.SIGNAL('clicked(bool)'),
self.close)
buttonLayout.setColumnStretch(0, 1)
if self.sequenceOption:
sequenceLayout = QtGui.QHBoxLayout()
sequenceLayout.setMargin(5)
sequenceLayout.addWidget(self.cb_loop_int)
sequenceLayout.addWidget(self.cb_loop_sequence)
buttonLayout.addLayout(sequenceLayout, 0, 0, QtCore.Qt.AlignRight)
buttonLayout.addWidget(self.cb_auto_update, 0, 1, QtCore.Qt.AlignLeft)
buttonLayout.addWidget(self.cb_keep_camera, 0, 2, 1, 2,
QtCore.Qt.AlignLeft)
if self.sequenceOption:
buttonLayout.addWidget(self.loopButton, 1, 1, QtCore.Qt.AlignRight)
self.loopButton.setEnabled(False)
buttonLayout.addWidget(self.updateButton, 1, 2, QtCore.Qt.AlignRight)
buttonLayout.addWidget(self.quitButton, 1, 3, QtCore.Qt.AlignRight)
self.connect(self.updateButton, QtCore.SIGNAL('clicked(bool)'),
self.updateButtonClick)
if self.sequenceOption:
self.connect(self.loopButton, QtCore.SIGNAL('clicked(bool)'),
self.loopButtonClick)
buttonDock.setWidget(buttonWidget)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, buttonDock)
self.controlDocks["__buttons__"] = buttonDock
self.saveAllAct = QtGui.QAction(
"S&ave Combined",
self,
shortcut=QtGui.QKeySequence.SelectAll,
statusTip="Save combined images to disk",
triggered=self.saveAllEvent)
self.saveAct = QtGui.QAction("&Save Each",
self,
shortcut=QtGui.QKeySequence.Save,
statusTip="Save separate images to disk",
triggered=self.saveEventAction)
self.showBuilderAct = QtGui.QAction(
"VisTrails Main Window",
self,
statusTip="Show VisTrails Main Window",
triggered=self.showBuilderWindow)
self.createMenus()
self.lastExportPath = ''
try:
v = d['value'](i)
if v is None:
return module.default_value
return v
except Exception, e:
debug.unexpected_exception(e)
return debug.format_exception(e)
return [evaluate(i) for i in xrange(self.size)]
result = get()
if not all(module.validate(x) for x in result):
show_warning('Failed Validation',
'One of the <i>%s</i>\'s user defined '
'functions has failed validation, '
'which usually means it generated a '
'value of a type different '
'than that specified by the '
'parameter. Parameter Exploration '
'aborted.' % param_info.spec.descriptor.name)
return None
return result
def getValuesString(self):
""" getValuesString() -> str
Return a string representation of the parameter list
"""
r = self.get_values(self.size)
if r is None:
return '{ERROR}'
def saveSpreadsheet(self, fileName=None):
""" saveSpreadsheet(fileName: str) -> None
Save the current spreadsheet to a file if fileName is not
None. Else, pop up a dialog to ask for a file name.
"""
def serialize_locator(locator):
wrapper = XMLWrapper()
dom = wrapper.create_document('spreadsheet_locator')
root = dom.documentElement
root.setAttribute("version", "1.0")
locator.serialize(dom,root)
return dom.toxml()
def need_save():
from vistrails.gui.vistrails_window import _app
need_save_vt = False
for t in self.tabWidgets:
dim = t.getDimension()
for r in xrange(dim[0]):
for c in xrange(dim[1]):
info = t.getCellPipelineInfo(r,c)
if info:
controller = info[0]['controller']
view = _app.ensureController(controller)
if view:
controller = view.get_controller()
if controller.changed:
need_save_vt = True
return need_save_vt
if need_save():
show_warning('Save Spreadsheet', 'Please save your vistrails and try again.')
return
if fileName==None:
fileName = self.spreadsheetFileName
if fileName:
indexFile = open(fileName, 'w')
indexFile.write(str(len(self.tabWidgets))+'\n')
for t in self.tabWidgets:
dim = t.getDimension()
sheet = spreadsheetRegistry.getSheetByType(type(t))
indexFile.write('%s\n'%str((str(t.windowTitle()),
sheet,
dim[0], dim[1])))
for r in xrange(dim[0]):
for c in xrange(dim[1]):
info = t.getCellPipelineInfo(r,c)
if info:
newinfo0 = copy.copy(info[0])
newinfo0['pipeline'] = None
newinfo0['actions'] = []
newinfo0['locator'] = \
serialize_locator(newinfo0['controller'].locator)
newinfo0['controller'] = None
indexFile.write('%s\n'
%str((r, c,
newinfo0,
info[1], info[2])))
indexFile.write('---\n')
indexFile.write(str(len(self.executedPipelines[0]))+'\n')
for vistrail in self.executedPipelines[0]:
indexFile.write('%s\n'%str((serialize_locator(vistrail[0].locator),
vistrail[1])))
self.changeSpreadsheetFileName(fileName)
indexFile.close()
else:
self.saveSpreadsheetAs()
def saveSpreadsheet(self, fileName=None):
""" saveSpreadsheet(fileName: str) -> None
Save the current spreadsheet to a file if fileName is not
None. Else, pop up a dialog to ask for a file name.
"""
def serialize_locator(locator):
wrapper = XMLWrapper()
dom = wrapper.create_document('spreadsheet_locator')
root = dom.documentElement
root.setAttribute("version", "1.0")
locator.serialize(dom, root)
return dom.toxml()
def need_save():
from vistrails.gui.vistrails_window import _app
need_save_vt = False
for t in self.tabWidgets:
dim = t.getDimension()
for r in xrange(dim[0]):
for c in xrange(dim[1]):
info = t.getCellPipelineInfo(r, c)
if info:
controller = info[0]['controller']
view = _app.ensureController(controller)
if view:
controller = view.get_controller()
if controller.changed:
need_save_vt = True
return need_save_vt
if need_save():
show_warning('Save Spreadsheet',
'Please save your vistrails and try again.')
return
if fileName == None:
fileName = self.spreadsheetFileName
if fileName:
indexFile = open(fileName, 'w')
indexFile.write(str(len(self.tabWidgets)) + '\n')
for t in self.tabWidgets:
dim = t.getDimension()
sheet = spreadsheetRegistry.getSheetByType(type(t))
indexFile.write('%s\n' % str(
(str(t.windowTitle()), sheet, dim[0], dim[1])))
for r in xrange(dim[0]):
for c in xrange(dim[1]):
info = t.getCellPipelineInfo(r, c)
if info:
newinfo0 = copy.copy(info[0])
newinfo0['pipeline'] = None
newinfo0['actions'] = []
newinfo0['locator'] = \
serialize_locator(newinfo0['controller'].locator)
newinfo0['controller'] = None
indexFile.write('%s\n' % str(
(r, c, newinfo0, info[1], info[2])))
indexFile.write('---\n')
indexFile.write(str(len(self.executedPipelines[0])) + '\n')
for vistrail in self.executedPipelines[0]:
indexFile.write('%s\n' % str(
(serialize_locator(vistrail[0].locator), vistrail[1])))
self.changeSpreadsheetFileName(fileName)
indexFile.close()
else:
self.saveSpreadsheetAs()
def __init__(self, parent=None, vistrail_view=None, dumpcells=False, controller=None, version=-1):
""" QMashupAppMainWindow()
Initialize an app window from a mashup.
"""
# Constructing the main widget
QtGui.QMainWindow.__init__(self, parent)
self.vtkCells = []
self.setStatusBar(QtGui.QStatusBar(self))
# Central widget
centralWidget = QtGui.QWidget()
self.mainLayout = QtGui.QVBoxLayout()
self.mainLayout.setMargin(0)
self.mainLayout.setSpacing(5)
centralWidget.setLayout(self.mainLayout)
self.setCentralWidget(centralWidget)
self.numberOfCells = 0
self.is_executing = False
self.sequenceOption = False
self.steps = []
self.isLooping = False
# self.resize(100,100)
self.dumpcells = dumpcells
self.view = vistrail_view
if controller:
self.controller = controller
self.mshptrail = controller.mshptrail
if version == -1:
self.currentMashup = self.controller.currentMashup
else:
self.currentMashup = self.mshptrail.getMashup(version)
self.setWindowTitle("%s Mashup" % self.controller.getMashupName(version))
else:
self.setWindowTitle("Mashup")
# Assign "hidden" shortcut
self.editingModeAct = QtGui.QAction(
"Chang&e Layout",
self,
shortcut="Ctrl+E",
statusTip="Change the layout of the widgets",
triggered=self.toggleEditingMode,
)
# self.editingModeShortcut = QtGui.QShortcut(QtGui.QKeySequence('Ctrl+E'), self)
# self.connect(self.editingModeShortcut, QtCore.SIGNAL('activated()'),
# self.toggleEditingMode)
self.editing = False
# Constructing alias controls
self.controlDocks = []
# Show here to make sure XDisplay info is correct (for VTKCell)
self.show()
spreadsheetController.setEchoMode(True)
# will run to get Spreadsheet Cell events
(cellEvents, errors) = self.runAndGetCellEvents(useDefaultValues=True)
if cellEvents:
self.numberOfCells = len(cellEvents)
self.initCells(cellEvents)
if len(errors) > 0:
show_warning("VisTrails::Mashup Preview", "There was a problem executing the pipeline: %s." % errors)
# Construct the controllers for aliases
self.controlDocks = {}
self.cellControls = {}
self.aliasWidgets = {}
self.initControls()
if self.currentMashup.layout is not None:
self.restoreState(QtCore.QByteArray.fromPercentEncoding(QtCore.QByteArray(self.currentMashup.layout)))
if self.currentMashup.geometry is not None:
self.restoreGeometry(QtCore.QByteArray.fromPercentEncoding(QtCore.QByteArray(self.currentMashup.geometry)))
else:
self.resize(self.sizeHint())
# Constructing buttons
buttonDock = QCustomDockWidget("Control Buttons", self)
buttonWidget = QtGui.QWidget(buttonDock)
buttonWidget.setSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Preferred)
buttonLayout = QtGui.QGridLayout()
buttonWidget.setLayout(buttonLayout)
buttonLayout.setMargin(5)
self.cb_auto_update = QtGui.QCheckBox("Turn on auto-update", self.centralWidget())
self.cb_auto_update.setChecked(False)
self.cb_loop_sequence = QtGui.QCheckBox(
"Render all steps in '%s'" % (self.sequenceOption.alias.name if self.sequenceOption else "None"),
self.centralWidget(),
)
self.cb_loop_sequence.setChecked(False)
self.cb_loop_sequence.setVisible(self.sequenceOption is not False)
self.cb_loop_sequence.setToolTip("Render each step of this stepper for fast switching")
self.cb_loop_int = QtGui.QCheckBox("Interactive Steps", self.centralWidget())
self.cb_loop_int.setChecked(False)
self.cb_loop_int.setVisible(False)
self.cb_loop_int.setToolTip("Show complete result of each step instead of static images")
self.cb_loop_sequence.clicked.connect(self.cb_loop_int.setVisible)
self.cb_keep_camera = QtGui.QCheckBox("Keep camera position", self.centralWidget())
self.cb_keep_camera.setChecked(True)
self.connect(self.cb_auto_update, QtCore.SIGNAL("stateChanged(int)"), self.auto_update_changed)
self.connect(self.cb_loop_int, QtCore.SIGNAL("stateChanged(int)"), self.loop_int_changed)
self.loopButton = QtGui.QPushButton("&Loop", self.centralWidget())
self.loopButton.setToolTip("Loop automatically through steps")
self.loopButton.setCheckable(True)
self.loopButton.setVisible(self.sequenceOption is not False)
self.updateButton = QtGui.QPushButton("&Update", self.centralWidget())
if self.dumpcells:
self.quitButton = QtGui.QPushButton("&Save", self.centralWidget())
self.connect(self.quitButton, QtCore.SIGNAL("clicked(bool)"), self.saveAndExport)
else:
self.quitButton = QtGui.QPushButton("&Quit", self.centralWidget())
self.connect(self.quitButton, QtCore.SIGNAL("clicked(bool)"), self.close)
buttonLayout.setColumnStretch(0, 1)
if self.sequenceOption:
sequenceLayout = QtGui.QHBoxLayout()
sequenceLayout.setMargin(5)
sequenceLayout.addWidget(self.cb_loop_int)
sequenceLayout.addWidget(self.cb_loop_sequence)
buttonLayout.addLayout(sequenceLayout, 0, 0, QtCore.Qt.AlignRight)
buttonLayout.addWidget(self.cb_auto_update, 0, 1, QtCore.Qt.AlignLeft)
buttonLayout.addWidget(self.cb_keep_camera, 0, 2, 1, 2, QtCore.Qt.AlignLeft)
if self.sequenceOption:
buttonLayout.addWidget(self.loopButton, 1, 1, QtCore.Qt.AlignRight)
self.loopButton.setEnabled(False)
buttonLayout.addWidget(self.updateButton, 1, 2, QtCore.Qt.AlignRight)
buttonLayout.addWidget(self.quitButton, 1, 3, QtCore.Qt.AlignRight)
self.connect(self.updateButton, QtCore.SIGNAL("clicked(bool)"), self.updateButtonClick)
if self.sequenceOption:
self.connect(self.loopButton, QtCore.SIGNAL("clicked(bool)"), self.loopButtonClick)
buttonDock.setWidget(buttonWidget)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, buttonDock)
self.controlDocks["__buttons__"] = buttonDock
self.saveAllAct = QtGui.QAction(
"S&ave Combined",
self,
shortcut=QtGui.QKeySequence.SelectAll,
statusTip="Save combined images to disk",
triggered=self.saveAllEvent,
)
self.saveAct = QtGui.QAction(
"&Save Each",
self,
shortcut=QtGui.QKeySequence.Save,
statusTip="Save separate images to disk",
triggered=self.saveEventAction,
)
self.showBuilderAct = QtGui.QAction(
"VisTrails Main Window", self, statusTip="Show VisTrails Main Window", triggered=self.showBuilderWindow
)
self.createMenus()
self.lastExportPath = ""