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',
locator=controller.locator,
current_version=versionId,
reason='Drop Version')
else:
event.ignore()
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(self, 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)
for mId in inspector.spreadsheet_cells:
name = pipeline.modules[mId].name
if inspector.annotated_modules.has_key(mId):
idx = inspector.annotated_modules[mId]
else:
idx = -1
(vRow, vCol) = cells[(name, idx)]
decodedCells.append((mId, vRow, vCol))
return decodedCells
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 pipeline:
self.inspector.inspect_spreadsheet_cells(pipeline)
self.inspector.inspect_ambiguous_modules(pipeline)
cells = []
for mId in self.inspector.spreadsheet_cells:
name = pipeline.modules[mId].name
if self.inspector.annotated_modules.has_key(mId):
cells.append((name, self.inspector.annotated_modules[mId]))
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 decodeConfiguration(self, 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)
for mId in inspector.spreadsheet_cells:
name = pipeline.modules[mId].name
if inspector.annotated_modules.has_key(mId):
idx = inspector.annotated_modules[mId]
else:
idx = -1
(vRow, vCol) = cells[(name, idx)]
decodedCells.append((mId, vRow, vCol))
return decodedCells
def positionPipelines(self, sheetPrefix, sheetCount, rowCount, colCount,
pipelines):
""" positionPipelines(sheetPrefix: str, sheetCount: int, rowCount: int,
colCount: int, pipelines: list of Pipeline)
-> list of Pipelines
Apply the virtual cell location to a list of pipelines in a
parameter exploration given that pipelines has multiple chunk
of sheetCount x rowCount x colCount cells
"""
(vRCount, vCCount, cells) = self.getConfiguration()
modifiedPipelines = []
for pId in xrange(len(pipelines)):
pipeline = copy.copy(pipelines[pId])
col = pId % colCount
row = (pId / colCount) % rowCount
sheet = (pId / (colCount*rowCount)) % sheetCount
decodedCells = self.decodeConfiguration(pipeline, cells)
for (mId, vRow, vCol) in decodedCells:
# Walk through all connection and remove all
# CellLocation connected to this spreadsheet cell
# delConn = DeleteConnectionAction()
action_list = []
for (cId,c) in self.pipeline.connections.iteritems():
if (c.destinationId==mId and
pipeline.modules[c.sourceId].name=="CellLocation"):
action_list.append(('delete', c))
# delConn.addId(cId)
# delConn.perform(pipeline)
action = db.services.action.create_action(action_list)
# FIXME: this should go to dbservice
Action.convert(action)
pipeline.perform_action(action)
# Add a sheet reference with a specific name
param_id = -pipeline.tmp_id.getNewId(ModuleParam.vtType)
sheetNameParam = ModuleParam(id=param_id,
pos=0,
name="",
val="%s %d" % (sheetPrefix, sheet),
type="String",
alias="",
)
function_id = -pipeline.tmp_id.getNewId(ModuleFunction.vtType)
sheetNameFunction = ModuleFunction(id=function_id,
pos=0,
name="SheetName",
parameters=[sheetNameParam],
)
param_id = -pipeline.tmp_id.getNewId(ModuleParam.vtType)
minRowParam = ModuleParam(id=param_id,
pos=0,
name="",
val=str(rowCount*vRCount),
type="Integer",
alias="",
)
function_id = -pipeline.tmp_id.getNewId(ModuleFunction.vtType)
minRowFunction = ModuleFunction(id=function_id,
pos=1,
name="MinRowCount",
parameters=[minRowParam],
)
param_id = -pipeline.tmp_id.getNewId(ModuleParam.vtType)
minColParam = ModuleParam(id=param_id,
pos=0,
name="",
val=str(colCount*vCCount),
type="Integer",
alias="",
)
function_id = -pipeline.tmp_id.getNewId(ModuleFunction.vtType)
minColFunction = ModuleFunction(id=function_id,
pos=2,
name="MinColumnCount",
parameters=[minColParam],
)
module_id = -pipeline.tmp_id.getNewId(module.Module.vtType)
sheetReference = module.Module(id=module_id,
name="SheetReference",
package="edu.utah.sci.vistrails.spreadsheet",
functions=[sheetNameFunction,
minRowFunction,
minColFunction])
action = db.services.action.create_action([('add',
sheetReference)])
# FIXME: this should go to dbservice
Action.convert(action)
pipeline.perform_action(action)
# sheetReference.id = pipeline.fresh_module_id()
# sheetReference.name = "SheetReference"
# addModule = AddModuleAction()
# addModule.module = sheetReference
# addModule.perform(pipeline)
#
# addParam = ChangeParameterAction()
# addParam.addParameter(sheetReference.id, 0, 0,
# "SheetName", "",
# '%s %d' % (sheetPrefix, sheet),
# "String", "" )
# addParam.addParameter(sheetReference.id, 1, 0,
# "MinRowCount", "",
# str(rowCount*vRCount), "Integer", "" )
# addParam.addParameter(sheetReference.id, 2, 0,
# "MinColumnCount", "",
# str(colCount*vCCount), "Integer", "" )
# addParam.perform(pipeline)
# Add a cell location module with a specific row and column
param_id = -pipeline.tmp_id.getNewId(ModuleParam.vtType)
rowParam = ModuleParam(id=param_id,
pos=0,
name="",
val=str(row*vRCount+vRow+1),
type="Integer",
alias="",
)
function_id = -pipeline.tmp_id.getNewId(ModuleFunction.vtType)
rowFunction = ModuleFunction(id=function_id,
pos=0,
name="Row",
parameters=[rowParam],
)
param_id = -pipeline.tmp_id.getNewId(ModuleParam.vtType)
colParam = ModuleParam(id=param_id,
pos=0,
name="",
val=str(col*vCCount+vCol+1),
type="Integer",
alias="",
)
function_id = -pipeline.tmp_id.getNewId(ModuleFunction.vtType)
colFunction = ModuleFunction(id=function_id,
pos=1,
name="Column",
parameters=[colParam],
)
module_id = -pipeline.tmp_id.getNewId(module.Module.vtType)
cellLocation = module.Module(id=module_id,
name="CellLocation",
package="edu.utah.sci.vistrails.spreadsheet",
functions=[rowFunction,
colFunction])
action = db.services.action.create_action([('add',
cellLocation)])
# FIXME: this should go to dbservice
Action.convert(action)
pipeline.perform_action(action)
# cellLocation.id = pipeline.fresh_module_id()
# cellLocation.name = "CellLocation"
# addModule = AddModuleAction()
# addModule.module = cellLocation
# addModule.perform(pipeline)
#
# addParam = ChangeParameterAction()
# addParam.addParameter(cellLocation.id, 0, 0,
# "Row", "", str(row*vRCount+vRow+1),
# "Integer", "" )
# addParam.addParameter(cellLocation.id, 1, 0,
# "Column", "", str(col*vCCount+vCol+1),
# "Integer", "" )
# addParam.perform(pipeline)
# Then connect the SheetReference to the CellLocation
port_id = -pipeline.tmp_id.getNewId(Port.vtType)
source = Port(id=port_id,
type='source',
moduleId=sheetReference.id,
moduleName=sheetReference.name)
source.name = "self"
source.spec = copy.copy(registry.get_output_port_spec(sheetReference,
"self"))
port_id = -pipeline.tmp_id.getNewId(Port.vtType)
destination = Port(id=port_id,
type='destination',
moduleId=cellLocation.id,
moduleName=cellLocation.name)
destination.name = "SheetReference"
destination.spec = copy.copy(registry.get_input_port_spec(cellLocation,
"SheetReference"))
c_id = -pipeline.tmp_id.getNewId(connection.Connection.vtType)
conn = connection.Connection(id=c_id,
ports=[source, destination])
action = db.services.action.create_action([('add',
conn)])
# FIXME: this should go to dbservice
Action.convert(action)
pipeline.perform_action(action)
# conn = connection.Connection()
# conn.id = pipeline.fresh_connection_id()
# conn.source.moduleId = sheetReference.id
# conn.source.moduleName = sheetReference.name
# conn.source.name = "self"
# conn.source.spec = registry.getOutputPortSpec(
# sheetReference, "self")
# conn.connectionId = conn.id
# conn.destination.moduleId = cellLocation.id
# conn.destination.moduleName = cellLocation.name
# conn.destination.name = "SheetReference"
# conn.destination.spec = registry.getInputPortSpec(
# cellLocation, "SheetReference")
# addConnection = AddConnectionAction()
# addConnection.connection = conn
# addConnection.perform(pipeline)
# Then connect the CellLocation to the spreadsheet cell
port_id = -pipeline.tmp_id.getNewId(Port.vtType)
source = Port(id=port_id,
type='source',
moduleId=cellLocation.id,
moduleName=cellLocation.name)
source.name = "self"
source.spec = registry.get_output_port_spec(cellLocation, "self")
port_id = -pipeline.tmp_id.getNewId(Port.vtType)
cell_module = pipeline.get_module_by_id(mId)
destination = Port(id=port_id,
type='destination',
moduleId=mId,
moduleName=pipeline.modules[mId].name)
destination.name = "Location"
destination.spec = registry.get_input_port_spec(cell_module,
"Location")
c_id = -pipeline.tmp_id.getNewId(connection.Connection.vtType)
conn = connection.Connection(id=c_id,
ports=[source, destination])
action = db.services.action.create_action([('add',
conn)])
# FIXME: this should go to dbservice
Action.convert(action)
pipeline.perform_action(action)
# conn = connection.Connection()
# conn.id = pipeline.fresh_connection_id()
# conn.source.moduleId = cellLocation.id
# conn.source.moduleName = cellLocation.name
# conn.source.name = "self"
# conn.source.spec = registry.getOutputPortSpec(
# cellLocation, "self")
# conn.connectionId = conn.id
# conn.destination.moduleId = mId
# conn.destination.moduleName = pipeline.modules[mId].name
# conn.destination.name = "Location"
# conn.destination.spec = registry.getInputPortSpec(
# cellLocation, "Location")
# addConnection = AddConnectionAction()
# addConnection.connection = conn
# addConnection.perform(pipeline)
modifiedPipelines.append(pipeline)
return modifiedPipelines
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 positionPipelines(self, sheetPrefix, sheetCount, rowCount, colCount,
pipelines, controller):
""" positionPipelines(sheetPrefix: str, sheetCount: int, rowCount: int,
colCount: int, pipelines: list of Pipeline)
-> list of Pipelines
Apply the virtual cell location to a list of pipelines in a
parameter exploration given that pipelines has multiple chunk
of sheetCount x rowCount x colCount cells
"""
return _positionPipelines(sheetPrefix, sheetCount,
rowCount, colCount, pipelines,
self.getConfiguration(), self.pipeline,
controller)
def assignPipelineCellLocations(pipeline,
sheetName,
row,
col,
cellIds=None,
minRowCount=None,
minColCount=None):
reg = get_module_registry()
spreadsheet_cell_desc = \
reg.get_descriptor_by_name('edu.utah.sci.vistrails.spreadsheet', \
'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,
"edu.utah.sci.vistrails.spreadsheet",
"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,
"edu.utah.sci.vistrails.spreadsheet",
"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
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 positionPipelines(self, sheetPrefix, sheetCount, rowCount, colCount,
pipelines, controller):
""" positionPipelines(sheetPrefix: str, sheetCount: int, rowCount: int,
colCount: int, pipelines: list of Pipeline)
-> list of Pipelines
Apply the virtual cell location to a list of pipelines in a
parameter exploration given that pipelines has multiple chunk
of sheetCount x rowCount x colCount cells
"""
return _positionPipelines(sheetPrefix, sheetCount, rowCount, colCount,
pipelines, self.getConfiguration(),
self.pipeline, controller)
def assignPipelineCellLocations(pipeline, sheetName,
row, col, cellIds=None,
minRowCount=None, minColCount=None):
reg = get_module_registry()
spreadsheet_cell_desc = \
reg.get_descriptor_by_name('edu.utah.sci.vistrails.spreadsheet', \
'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,
"edu.utah.sci.vistrails.spreadsheet",
"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,
"edu.utah.sci.vistrails.spreadsheet",
"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
