def from_module(module):
from core.uvcdat.plot_pipeline_helper import PlotPipelineHelper
var = Variable()
var.filename = PlotPipelineHelper.get_value_from_function_as_str(module, 'file')
var.file = var.filename
var.url = PlotPipelineHelper.get_value_from_function(module, 'url')
var.name = PlotPipelineHelper.get_value_from_function(module, 'name')
var.load = PlotPipelineHelper.get_value_from_function(module, 'load')
var.relativizePaths()
return var
def from_module(module):
from core.uvcdat.plot_pipeline_helper import PlotPipelineHelper
var = Variable()
var.filename = PlotPipelineHelper.get_value_from_function_as_str(
module, 'file')
var.file = var.filename
var.url = PlotPipelineHelper.get_value_from_function(module, 'url')
var.name = PlotPipelineHelper.get_value_from_function(module, 'name')
var.load = PlotPipelineHelper.get_value_from_function(module, 'load')
var.relativizePaths()
return var
def from_module(module):
from core.uvcdat.plot_pipeline_helper import PlotPipelineHelper
var = Variable.from_module(module)
readerParameters = PlotPipelineHelper.get_value_from_function_as_str(module,
'readerParameters')
var.set_reader_parameters(readerParameters)
return var
def find_plot_modules(pipeline):
#// Find plot modules in the order they appear in the Cell
res = []
cell = PlotPipelineHelper.find_module_by_name(pipeline, 'PVGenericCell')
plots = pipeline.get_inputPort_modules(cell.id, 'representation')
for plot in plots:
res.append(pipeline.modules[plot])
return res
def update_functions(self, module_name, functions):
#// @todo: Check with Ben if this is the right way to do it:
import api
controller = api.get_current_controller()
module = PlotPipelineHelper.find_module_by_name(controller.current_pipeline, module_name)
action = controller.update_functions(module, functions)
if action is not None:
window = get_vistrails_application().uvcdatWindow
window.get_current_project_controller().cell_was_changed(action)
def build_plot_pipeline_action(controller, version, var_modules, plots,row, col):
# FIXME want to make sure that nothing changes if var_module
# or plot_module do not change
#plot_type = plots[0].parent
#plot_gm = plots[0].name
if controller is None:
controller = api.get_current_controller()
version = 0L
reg = get_module_registry()
ops = []
cell_module = None
pipeline = controller.vistrail.getPipeline(version)
var_module = var_modules[0]
try:
temp_var_module = pipeline.get_module_by_id(var_module.id)
except KeyError:
temp_var_module = None
if temp_var_module is not None:
var_module = temp_var_module
else:
ops.append(('add',var_module))
for plot in plots:
plot_type = plot.parent
plot_gm = plot.name
import re
plotname = re.sub(r'\s', '', plot_gm)
plot_module = PlotPipelineHelper.find_module_by_name(pipeline, plotname)
if plot_module is not None:
continue
plot_descriptor = reg.get_descriptor_by_name('gov.lbl.visit', 'VisItCell')
plot_module = controller.create_module_from_descriptor(plot_descriptor)
ops.append(('add',plot_module))
#if cell_module is None:
# cell_module = PlotPipelineHelper.find_module_by_name(pipeline, "VisItCell")
if cell_module is None:
#cell_desc = reg.get_descriptor_by_name('gov.lbl.visit', "VisItCell")
#cell_module = controller.create_module_from_descriptor(cell_desc)
#ops.append(('add', cell_module))
cell_module = plot_module
if issubclass(var_modules[0].module_descriptor.module, CDMSVariable):
conn = controller.create_connection(var_module, 'self',
plot_module, 'variable')
else:
conn = controller.create_connection(var_module, 'self',
cell_module, 'variable')
ops.append(('add', conn))
#print 'connection source id is ', conn.sourceId
#print 'connection dest id is ', conn.destinationId
loc_module = controller.create_module_from_descriptor(
reg.get_descriptor_by_name('edu.utah.sci.vistrails.spreadsheet',
'CellLocation'))
functions = controller.create_functions(loc_module,
[('Row', [str(row+1)]), ('Column', [str(col+1)])])
for f in functions:
loc_module.add_function(f)
loc_conn = controller.create_connection(loc_module, 'self',
cell_module, 'Location')
ops.extend([('add', loc_module),
('add', loc_conn)])
type_of_plot = str(plot_gm)
param_module = controller.create_module_from_descriptor(
reg.get_descriptor_by_name('gov.lbl.visit', 'VisItParams'))
functions = controller.create_functions(param_module,
[('renderType', [type_of_plot])])
for f in functions:
param_module.add_function(f)
param_conn = controller.create_connection(param_module, 'self',
plot_module, 'visitparams')
ops.extend([('add', param_module),
('add', param_conn)])
# Create connection between the cell and the representation
#conn = controller.create_connection(plot_module, 'self',
# cell_module, 'representation')
# Add the connection to the pipeline operations
#ops.append(('add', conn))
action = core.db.action.create_action(ops)
controller.change_selected_version(version)
controller.add_new_action(action)
controller.perform_action(action)
return action
def build_plot_pipeline_action(controller, version, var_modules, plots,row, col, template=None):
#// @todo: want to make sure that nothing changes if var_module
#// or plot_module do not change
# Get controller
if controller is None:
controller = api.get_current_controller()
version = 0L
# Get module registry
reg = get_module_registry()
ops = []
cell_module = None
#// First get the pipeline
pipeline = controller.vistrail.getPipeline(version)
#// @todo: Add support for multiple variables per plot
#// Use only the first var module for now
var_module = var_modules[0]
#// @Aashish: As of now, very first time var module is being added to the pipeline by the project controller
#// but second time on the same cell, it gets removed and hence we needed to add var module again to pipeline.
#// I need to put this code under try catch because as of now looking for an id that does not exists
#// results in exception.
try:
temp_var_module = pipeline.get_module_by_id(var_module.id)
except KeyError:
temp_var_module = None
if temp_var_module is not None:
var_module = temp_var_module
else:
# This time we need to add var module to the pipeline
ops.append(('add', var_module))
for plot in plots:
plot_gm = plot.name
#// Create plot module from the descriptor
#// Is there a better way? I looked around and found none
import re
plotUsableName = re.sub(r'\s', '', plot_gm)
plot_module = PlotPipelineHelper.find_module_by_name(pipeline, plotUsableName)
if plot_module is not None:
continue
plot_descriptor = reg.get_descriptor_by_name('com.kitware.pvclimate', plotUsableName)
#// @Aashish: Looks like the pipeline is using the cached representation
plot_module = controller.create_module_from_descriptor(plot_descriptor)
ops.append(('add', plot_module))
#// @Aashish: This is no longer required as of this commit
#// e13bb034ceb302afe3aad3caf20153e1525586db
#// I am not sure though why we still need to add plot module
#ops.append(('add', var_modules[0]))
#// Create cell - representation linkage
#// Check to see if a cell already exits, if yes then set the input (representation on it) or else
#// create a new one and then set the representation on it.
if cell_module is None:
cell_module = PlotPipelineHelper.find_module_by_name(pipeline, "PVGenericCell")
#// If cell module is None, then create a new one
if cell_module is None:
cell_desc = reg.get_descriptor_by_name('com.kitware.pvclimate', "PVGenericCell")
cell_module = controller.create_module_from_descriptor(cell_desc)
ops.append(('add', cell_module))
#// Create a connection between the cell and the variable
#// @Aashish: I am expecting that every time we drop a variable, we will get a
#// pipeline that does not have modules from the previous execution. I need to verify
#// this but for now this assumption seems to hold.
oport = CDMSPipelineHelper.get_output_port_name(
var_module.module_descriptor.module)
conn = controller.create_connection(var_module, oport,
plot_module, 'cdms_variable')
ops.append(('add', conn))
loc_module = controller.create_module_from_descriptor(
reg.get_descriptor_by_name('edu.utah.sci.vistrails.spreadsheet',
'CellLocation'))
functions = controller.create_functions(loc_module,
[('Row', [str(row+1)]), ('Column', [str(col+1)])])
for f in functions:
loc_module.add_function(f)
loc_conn = controller.create_connection(loc_module, 'self',
cell_module, 'Location')
ops.extend([('add', loc_module),
('add', loc_conn)])
#// Create connection between the cell and the representation
conn = controller.create_connection(plot_module, 'self',
cell_module, 'representation')
#// Add the connection to the pipeline operations
ops.append(('add', conn))
action = core.db.action.create_action(ops)
controller.change_selected_version(version)
controller.add_new_action(action)
controller.perform_action(action)
return action
def build_plot_pipeline_action(controller, version, var_modules, plots,
row, col):
# FIXME want to make sure that nothing changes if var_module
# or plot_module do not change
#plot_type = plots[0].parent
#plot_gm = plots[0].name
if controller is None:
controller = api.get_current_controller()
version = 0L
reg = get_module_registry()
ops = []
cell_module = None
pipeline = controller.vistrail.getPipeline(version)
var_module = var_modules[0]
try:
temp_var_module = pipeline.get_module_by_id(var_module.id)
except KeyError:
temp_var_module = None
if temp_var_module is not None:
var_module = temp_var_module
else:
ops.append(('add', var_module))
for plot in plots:
plot_type = plot.parent
plot_gm = plot.name
import re
plotname = re.sub(r'\s', '', plot_gm)
plot_module = PlotPipelineHelper.find_module_by_name(
pipeline, plotname)
if plot_module is not None:
continue
plot_descriptor = reg.get_descriptor_by_name(
'gov.lbl.visit', 'VisItCell')
plot_module = controller.create_module_from_descriptor(
plot_descriptor)
ops.append(('add', plot_module))
#if cell_module is None:
# cell_module = PlotPipelineHelper.find_module_by_name(pipeline, "VisItCell")
if cell_module is None:
#cell_desc = reg.get_descriptor_by_name('gov.lbl.visit', "VisItCell")
#cell_module = controller.create_module_from_descriptor(cell_desc)
#ops.append(('add', cell_module))
cell_module = plot_module
if issubclass(var_modules[0].module_descriptor.module,
CDMSVariable):
conn = controller.create_connection(
var_module, 'self', plot_module, 'variable')
else:
conn = controller.create_connection(
var_module, 'self', cell_module, 'variable')
ops.append(('add', conn))
#print 'connection source id is ', conn.sourceId
#print 'connection dest id is ', conn.destinationId
loc_module = controller.create_module_from_descriptor(
reg.get_descriptor_by_name(
'edu.utah.sci.vistrails.spreadsheet', 'CellLocation'))
functions = controller.create_functions(
loc_module, [('Row', [str(row + 1)]),
('Column', [str(col + 1)])])
for f in functions:
loc_module.add_function(f)
loc_conn = controller.create_connection(
loc_module, 'self', cell_module, 'Location')
ops.extend([('add', loc_module), ('add', loc_conn)])
type_of_plot = str(plot_gm)
param_module = controller.create_module_from_descriptor(
reg.get_descriptor_by_name('gov.lbl.visit', 'VisItParams'))
functions = controller.create_functions(
param_module, [('renderType', [type_of_plot])])
for f in functions:
param_module.add_function(f)
param_conn = controller.create_connection(
param_module, 'self', plot_module, 'visitparams')
ops.extend([('add', param_module), ('add', param_conn)])
# Create connection between the cell and the representation
#conn = controller.create_connection(plot_module, 'self',
# cell_module, 'representation')
# Add the connection to the pipeline operations
#ops.append(('add', conn))
action = core.db.action.create_action(ops)
controller.change_selected_version(version)
controller.add_new_action(action)
controller.perform_action(action)
return action