def decodeConfiguration(pipeline, cells):
""" decodeConfiguration(pipeline: Pipeline,
cells: configuration) -> decoded cells
Convert cells of type [{(type,id): (row, column)}) to
(mId, row, col) in a particular pipeline
"""
decodedCells = []
inspector = PipelineInspector()
inspector.inspect_spreadsheet_cells(pipeline)
inspector.inspect_ambiguous_modules(pipeline)
orig_pipeline = pipeline
for id_list in inspector.spreadsheet_cells:
pipeline = orig_pipeline
id_iter = iter(id_list)
m = pipeline.modules[id_iter.next()]
m_id = m.id
for m_id in id_iter:
pipeline = m.pipeline
m = pipeline.modules[m_id]
name = m.name
if len(id_list) == 1 and m_id in inspector.annotated_modules:
idx = inspector.annotated_modules[m_id]
elif tuple(id_list) in inspector.annotated_modules:
idx = inspector.annotated_modules[tuple(id_list)]
else:
idx = -1
(vRow, vCol) = cells[(name, idx)]
if len(id_list) == 1:
decodedCells.append((m_id, vRow, vCol))
else:
decodedCells.append((tuple(id_list), vRow, vCol))
return decodedCells
def decodeConfiguration(pipeline, cells):
""" decodeConfiguration(pipeline: Pipeline,
cells: configuration) -> decoded cells
Convert cells of type [{(type,id): (row, column)}) to
(mId, row, col) in a particular pipeline
"""
decodedCells = []
inspector = PipelineInspector()
inspector.inspect_spreadsheet_cells(pipeline)
inspector.inspect_ambiguous_modules(pipeline)
orig_pipeline = pipeline
for id_list in inspector.spreadsheet_cells:
pipeline = orig_pipeline
id_iter = iter(id_list)
m = pipeline.modules[id_iter.next()]
m_id = m.id
for m_id in id_iter:
pipeline = m.pipeline
m = pipeline.modules[m_id]
name = m.name
if len(id_list) == 1 and m_id in inspector.annotated_modules:
idx = inspector.annotated_modules[m_id]
elif tuple(id_list) in inspector.annotated_modules:
idx = inspector.annotated_modules[tuple(id_list)]
else:
idx = -1
(vRow, vCol) = cells[(name, idx)]
if len(id_list) == 1:
decodedCells.append((m_id, vRow, vCol))
else:
decodedCells.append((tuple(id_list), vRow, vCol))
return decodedCells
def dropEvent(self, event):
""" Execute the pipeline at the particular location """
mimeData = event.mimeData()
if (hasattr(mimeData, 'versionId') and
hasattr(mimeData, 'controller')):
event.accept()
versionId = mimeData.versionId
controller = mimeData.controller
pipeline = controller.vistrail.getPipeline(versionId)
inspector = PipelineInspector()
inspector.inspect_spreadsheet_cells(pipeline)
inspector.inspect_ambiguous_modules(pipeline)
if len(inspector.spreadsheet_cells)==1:
localPos = self.sheet.viewport().mapFromGlobal(QtGui.QCursor.pos())
row = self.sheet.rowAt(localPos.y())
col = self.sheet.columnAt(localPos.x())
if (row!=-1 and col!=-1):
pipeline = self.setPipelineToLocateAt(row, col, pipeline)
executePipelineWithProgress(pipeline, 'Execute Cell',
controller=controller,
locator=controller.locator,
current_version=versionId,
reason='Drop Version')
else:
event.ignore()
def dropEvent(self, event):
""" Execute the pipeline at the particular location """
mimeData = event.mimeData()
if (hasattr(mimeData, 'versionId')
and hasattr(mimeData, 'controller')):
event.accept()
versionId = mimeData.versionId
controller = mimeData.controller
pipeline = controller.vistrail.getPipeline(versionId)
inspector = PipelineInspector()
inspector.inspect_spreadsheet_cells(pipeline)
inspector.inspect_ambiguous_modules(pipeline)
if len(inspector.spreadsheet_cells) == 1:
localPos = self.sheet.viewport().mapFromGlobal(
QtGui.QCursor.pos())
row = self.sheet.rowAt(localPos.y())
col = self.sheet.columnAt(localPos.x())
if (row != -1 and col != -1):
pipeline = self.setPipelineToLocateAt(row, col, pipeline)
executePipelineWithProgress(pipeline,
'Execute Cell',
controller=controller,
locator=controller.locator,
current_version=versionId,
reason='Drop Version')
else:
event.ignore()
class QVirtualCellWindow(QtGui.QFrame, QToolWindowInterface):
"""
QVirtualCellWindow contains a caption, a virtual cell
configuration
"""
def __init__(self, parent=None):
""" QVirtualCellWindow(parent: QWidget) -> QVirtualCellWindow
Initialize the widget
"""
QtGui.QFrame.__init__(self, parent)
self.setFrameShape(QtGui.QFrame.StyledPanel)
self.setWindowTitle('Spreadsheet Virtual Cell')
vLayout = QtGui.QVBoxLayout(self)
vLayout.setMargin(2)
vLayout.setSpacing(0)
self.setLayout(vLayout)
label = QtGui.QLabel('Arrange the cell(s) below to construct'
' a virtual cell')
font = QtGui.QFont(label.font())
label.setFont(font)
label.setWordWrap(True)
vLayout.addWidget(label)
hLayout = QtGui.QVBoxLayout()
hLayout.setMargin(0)
hLayout.setSpacing(0)
vLayout.addLayout(hLayout)
self.config = QVirtualCellConfiguration()
self.config.setSizePolicy(QtGui.QSizePolicy.Maximum,
QtGui.QSizePolicy.Maximum)
hLayout.addWidget(self.config)
hPadWidget = QtGui.QWidget()
hLayout.addWidget(hPadWidget)
hPadWidget.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Ignored)
vPadWidget = QtGui.QWidget()
vLayout.addWidget(vPadWidget)
vPadWidget.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding)
self.inspector = PipelineInspector()
self.pipeline = None
def updateVirtualCell(self, pipeline):
""" updateVirtualCell(pipeline: QPipeline) -> None
Setup the virtual cells given a pipeline
"""
self.pipeline = pipeline
if self.pipeline and self.pipeline.is_valid:
self.inspector.inspect_spreadsheet_cells(self.pipeline)
self.inspector.inspect_ambiguous_modules(self.pipeline)
cells = []
for id_list in self.inspector.spreadsheet_cells:
pipeline = self.pipeline
id_iter = iter(id_list)
m = pipeline.modules[id_iter.next()]
m_id = m.id
for m_id in id_iter:
pipeline = m.pipeline
m = pipeline.modules[m_id]
name = m.name
if len(id_list) == 1 and \
m_id in self.inspector.annotated_modules:
cells.append((name, self.inspector.annotated_modules[m_id]))
elif tuple(id_list) in self.inspector.annotated_modules:
cells.append((name, self.inspector.annotated_modules[ \
tuple(id_list)]))
else:
cells.append((name, -1))
self.config.configVirtualCells(cells)
else:
self.config.clear()
def getConfiguration(self):
""" getConfiguration() -> info (see below)
Return the current configuration of the virtual cell. The
information is:
info = (rowCount, columnCount,
{(type,id): (row, column)})
"""
return self.config.getConfiguration()
def setConfiguration(self, info):
""" setConfiguration(info) -> None (see below)
Set the configuration of the virtual cell. The
information is:
info = {(type, id): (row, column)}
or
info = (rowCount, columnCount,
{(type, id): (row, column)})
The second form is allowed so that the output of
getConfiguration could be passed directly to
setConfiguration (the dimensions aren't used).
"""
self.config.setConfiguration(info)
def assignPipelineCellLocations(pipeline, sheetName,
row, col, cellIds=None,
minRowCount=None, minColCount=None):
reg = get_module_registry()
spreadsheet_cell_desc = reg.get_descriptor_by_name(spreadsheet_pkg,
'SpreadsheetCell')
output_module_desc = reg.get_descriptor_by_name(
'org.vistrails.vistrails.basic', 'OutputModule')
create_module = VistrailController.create_module_static
create_function = VistrailController.create_function_static
create_connection = VistrailController.create_connection_static
pipeline = copy.copy(pipeline)
root_pipeline = pipeline
if cellIds is None:
inspector = PipelineInspector()
inspector.inspect_spreadsheet_cells(pipeline)
inspector.inspect_ambiguous_modules(pipeline)
cellIds = inspector.spreadsheet_cells
def fix_cell_module(pipeline, mId):
# Delete connections to 'Location' input port
conns_to_delete = []
for c in pipeline.connection_list:
if c.destinationId == mId and c.destination.name == 'Location':
conns_to_delete.append(c.id)
for c_id in conns_to_delete:
pipeline.delete_connection(c_id)
# a hack to first get the id_scope to the local pipeline scope
# then make them negative by hacking the getNewId method
# all of this is reset at the end of this block
id_scope = pipeline.tmp_id
orig_getNewId = pipeline.tmp_id.__class__.getNewId
def getNewId(self, objType):
return -orig_getNewId(self, objType)
pipeline.tmp_id.__class__.getNewId = getNewId
# Add a sheet reference with a specific name
sheetReference = create_module(id_scope, spreadsheet_pkg,
"SheetReference")
sheetNameFunction = create_function(id_scope, sheetReference,
"SheetName", [str(sheetName)])
# ["%s %d" % (sheetPrefix, sheet)])
sheetReference.add_function(sheetNameFunction)
if minRowCount is not None:
minRowFunction = create_function(id_scope, sheetReference,
"MinRowCount", [str(minRowCount)])
# [str(rowCount*vRCount)])
sheetReference.add_function(minRowFunction)
if minColCount is not None:
minColFunction = create_function(id_scope, sheetReference,
"MinColumnCount",
[str(minColCount)])
# [str(colCount*vCCount)])
sheetReference.add_function(minColFunction)
# Add a cell location module with a specific row and column
cellLocation = create_module(id_scope, spreadsheet_pkg,
"CellLocation")
rowFunction = create_function(id_scope, cellLocation, "Row", [str(row)])
# [str(row*vRCount+vRow+1)])
colFunction = create_function(id_scope, cellLocation, "Column",
[str(col)])
# [str(col*vCCount+vCol+1)])
cellLocation.add_function(rowFunction)
cellLocation.add_function(colFunction)
# Then connect the SheetReference to the CellLocation
sheet_conn = create_connection(id_scope, sheetReference, "value",
cellLocation, "SheetReference")
# Then connect the CellLocation to the spreadsheet cell
cell_module = pipeline.get_module_by_id(mId)
cell_conn = create_connection(id_scope, cellLocation, "value",
cell_module, "Location")
pipeline.add_module(sheetReference)
pipeline.add_module(cellLocation)
pipeline.add_connection(sheet_conn)
pipeline.add_connection(cell_conn)
# replace the getNewId method
pipeline.tmp_id.__class__.getNewId = orig_getNewId
def fix_output_module(pipeline, mId):
# Remove all connections to 'configuration' input port
conns_to_delete = []
for c in pipeline.connection_list:
if (c.destinationId == mId and
c.destination.name == 'configuration'):
conns_to_delete.append(c.id)
for c_id in conns_to_delete:
pipeline.delete_connection(c_id)
m = pipeline.modules[mId]
# Remove all functions on 'configuration' input port
funcs_to_delete = []
for f in m.functions:
if f.name == 'configuration':
funcs_to_delete.append(f.real_id)
for f_id in funcs_to_delete:
m.delete_function_by_real_id(f_id)
# a hack to first get the id_scope to the local pipeline scope
# then make them negative by hacking the getNewId method
# all of this is reset at the end of this block
id_scope = pipeline.tmp_id
orig_getNewId = pipeline.tmp_id.__class__.getNewId
def getNewId(self, objType):
return -orig_getNewId(self, objType)
pipeline.tmp_id.__class__.getNewId = getNewId
config = {'row': row - 1, 'col': col - 1}
if minRowCount is not None:
config['sheetRowCount'] = minRowCount
if minColCount is not None:
config['sheetColCount'] = minColCount
if sheetName is not None:
config['sheetName']= sheetName
config = {'spreadsheet': config}
config_function = create_function(id_scope, m,
'configuration', [repr(config)])
m.add_function(config_function)
# replace the getNewId method
pipeline.tmp_id.__class__.getNewId = orig_getNewId
for id_list in cellIds:
cell_pipeline = pipeline
# find at which depth we need to be working
if isinstance(id_list, (int, long)):
mId = id_list
m = cell_pipeline.modules[mId]
else:
id_iter = iter(id_list)
mId = next(id_iter)
m = cell_pipeline.modules[mId]
for mId in id_iter:
cell_pipeline = m.pipeline
m = cell_pipeline.modules[mId]
if reg.is_descriptor_subclass(m.module_descriptor,
spreadsheet_cell_desc):
fix_cell_module(cell_pipeline, mId)
elif reg.is_descriptor_subclass(m.module_descriptor,
output_module_desc):
fix_output_module(cell_pipeline, mId)
return root_pipeline
class QVirtualCellWindow(QtGui.QFrame, QToolWindowInterface):
"""
QVirtualCellWindow contains a caption, a virtual cell
configuration
"""
def __init__(self, parent=None):
""" QVirtualCellWindow(parent: QWidget) -> QVirtualCellWindow
Initialize the widget
"""
QtGui.QFrame.__init__(self, parent)
self.setFrameShape(QtGui.QFrame.StyledPanel)
self.setWindowTitle('Spreadsheet Virtual Cell')
vLayout = QtGui.QVBoxLayout(self)
vLayout.setMargin(2)
vLayout.setSpacing(0)
self.setLayout(vLayout)
label = QtGui.QLabel('Arrange the cell(s) below to construct'
' a virtual cell')
font = QtGui.QFont(label.font())
label.setFont(font)
label.setWordWrap(True)
vLayout.addWidget(label)
hLayout = QtGui.QVBoxLayout()
hLayout.setMargin(0)
hLayout.setSpacing(0)
vLayout.addLayout(hLayout)
self.config = QVirtualCellConfiguration()
self.config.setSizePolicy(QtGui.QSizePolicy.Maximum,
QtGui.QSizePolicy.Maximum)
hLayout.addWidget(self.config)
hPadWidget = QtGui.QWidget()
hLayout.addWidget(hPadWidget)
hPadWidget.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Ignored)
vPadWidget = QtGui.QWidget()
vLayout.addWidget(vPadWidget)
vPadWidget.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding)
self.inspector = PipelineInspector()
self.pipeline = None
def updateVirtualCell(self, pipeline):
""" updateVirtualCell(pipeline: QPipeline) -> None
Setup the virtual cells given a pipeline
"""
self.pipeline = pipeline
if self.pipeline and self.pipeline.is_valid:
self.inspector.inspect_spreadsheet_cells(self.pipeline)
self.inspector.inspect_ambiguous_modules(self.pipeline)
cells = []
for id_list in self.inspector.spreadsheet_cells:
pipeline = self.pipeline
id_iter = iter(id_list)
m = pipeline.modules[id_iter.next()]
m_id = m.id
for m_id in id_iter:
pipeline = m.pipeline
m = pipeline.modules[m_id]
name = m.name
if len(id_list) == 1 and \
m_id in self.inspector.annotated_modules:
cells.append(
(name, self.inspector.annotated_modules[m_id]))
elif tuple(id_list) in self.inspector.annotated_modules:
cells.append((name, self.inspector.annotated_modules[ \
tuple(id_list)]))
else:
cells.append((name, -1))
self.config.configVirtualCells(cells)
else:
self.config.clear()
def getConfiguration(self):
""" getConfiguration() -> info (see below)
Return the current configuration of the virtual cell. The
information is:
info = (rowCount, columnCount,
{(type,id): (row, column)})
"""
return self.config.getConfiguration()
def setConfiguration(self, info):
""" setConfiguration(info) -> None (see below)
Set the configuration of the virtual cell. The
information is:
info = {(type, id): (row, column)}
or
info = (rowCount, columnCount,
{(type, id): (row, column)})
The second form is allowed so that the output of
getConfiguration could be passed directly to
setConfiguration (the dimensions aren't used).
"""
self.config.setConfiguration(info)
def assignPipelineCellLocations(pipeline, sheetName,
row, col, cellIds=None,
minRowCount=None, minColCount=None):
reg = get_module_registry()
spreadsheet_cell_desc = \
reg.get_descriptor_by_name(spreadsheet_pkg, 'SpreadsheetCell')
create_module = VistrailController.create_module_static
create_function = VistrailController.create_function_static
create_connection = VistrailController.create_connection_static
pipeline = copy.copy(pipeline)
root_pipeline = pipeline
if cellIds is None:
inspector = PipelineInspector()
inspector.inspect_spreadsheet_cells(pipeline)
inspector.inspect_ambiguous_modules(pipeline)
cellIds = inspector.spreadsheet_cells
for id_list in cellIds:
# find at which depth we need to be working
try:
id_iter = iter(id_list)
m = pipeline.modules[id_iter.next()]
for mId in id_iter:
pipeline = m.pipeline
m = pipeline.modules[mId]
except TypeError:
mId = id_list
m = pipeline.modules[mId]
if not reg.is_descriptor_subclass(m.module_descriptor,
spreadsheet_cell_desc):
continue
# Walk through all connections and remove all CellLocation
# modules connected to this spreadsheet cell
conns_to_delete = []
for (cId,c) in pipeline.connections.iteritems():
if (c.destinationId==mId and
pipeline.modules[c.sourceId].name=="CellLocation"):
conns_to_delete.append(c.id)
for c_id in conns_to_delete:
pipeline.delete_connection(c_id)
# a hack to first get the id_scope to the local pipeline scope
# then make them negative by hacking the getNewId method
# all of this is reset at the end of this block
id_scope = pipeline.tmp_id
orig_getNewId = pipeline.tmp_id.__class__.getNewId
def getNewId(self, objType):
return -orig_getNewId(self, objType)
pipeline.tmp_id.__class__.getNewId = getNewId
# Add a sheet reference with a specific name
sheetReference = create_module(id_scope, spreadsheet_pkg,
"SheetReference")
sheetNameFunction = create_function(id_scope, sheetReference,
"SheetName", [str(sheetName)])
# ["%s %d" % (sheetPrefix, sheet)])
sheetReference.add_function(sheetNameFunction)
if minRowCount is not None:
minRowFunction = create_function(id_scope, sheetReference,
"MinRowCount", [str(minRowCount)])
# [str(rowCount*vRCount)])
sheetReference.add_function(minRowFunction)
if minColCount is not None:
minColFunction = create_function(id_scope, sheetReference,
"MinColumnCount",
[str(minColCount)])
# [str(colCount*vCCount)])
sheetReference.add_function(minColFunction)
# Add a cell location module with a specific row and column
cellLocation = create_module(id_scope, spreadsheet_pkg,
"CellLocation")
rowFunction = create_function(id_scope, cellLocation, "Row", [str(row)])
# [str(row*vRCount+vRow+1)])
colFunction = create_function(id_scope, cellLocation, "Column",
[str(col)])
# [str(col*vCCount+vCol+1)])
cellLocation.add_function(rowFunction)
cellLocation.add_function(colFunction)
# Then connect the SheetReference to the CellLocation
sheet_conn = create_connection(id_scope, sheetReference, "self",
cellLocation, "SheetReference")
# Then connect the CellLocation to the spreadsheet cell
cell_module = pipeline.get_module_by_id(mId)
cell_conn = create_connection(id_scope, cellLocation, "self",
cell_module, "Location")
pipeline.add_module(sheetReference)
pipeline.add_module(cellLocation)
pipeline.add_connection(sheet_conn)
pipeline.add_connection(cell_conn)
# replace the getNewId method
pipeline.tmp_id.__class__.getNewId = orig_getNewId
return root_pipeline
