def get_pipeline(self): vistraildata = VistrailManager(self._controller) if vistraildata is None: return None if self.widget() is not None: # Get pipeline info from VisTrails pipelineInfo = self.cellInfo.tab.getCellPipelineInfo( self.cellInfo.row, self.cellInfo.column) version = pipelineInfo[0]['version'] return vistraildata.get_pipeline(version, infer_for_cell=self.cellInfo) else: # Get pipeline info from DAT: we might be building something here return vistraildata.get_pipeline(self.cellInfo)
def show_advanced_config(self): # Get the pipeline info for the cell vistraildata = VistrailManager(self._cell._controller) pipeline = vistraildata.get_pipeline(self._cell.cellInfo) self._cell._set_overlay(pipeline.recipe.plot.configWidget) self._cell._overlay.setup(self._cell, pipeline.recipe.plot)
def _color_version_nodes(node, action, tag, description): pipelineInfo = None if action is not None and VistrailManager.initialized: vistraildata = VistrailManager() # Warning: the first scene might get created before the # VistrailManager gets the 'controller_changed' signal, thus # VistrailManager() might be None if vistraildata is not None: pipelineInfo = vistraildata.get_pipeline(action.id) if tag == 'dat-vars': # Variable root brush = QtGui.QBrush(QtGui.QColor(0, 0, 0)) return dict( VERSION_USER_BRUSH=brush, VERSION_OTHER_BRUSH=brush, VERSION_LABEL_COLOR=QtGui.QColor(255, 255, 255), VERSION_SHAPE='rectangle') elif tag is not None and tag.startswith('dat-var-'): # Variables return dict( VERSION_USER_BRUSH=QtGui.QBrush(QtGui.QColor(27, 27, 75)), VERSION_OTHER_BRUSH=QtGui.QBrush(QtGui.QColor(72, 50, 25)), VERSION_LABEL_COLOR=QtGui.QColor(255, 255, 255), VERSION_SHAPE='rectangle') elif pipelineInfo is not None: return dict( VERSION_USER_BRUSH=QtGui.QBrush(QtGui.QColor(171, 169, 214)), VERSION_OTHER_BRUSH=QtGui.QBrush(QtGui.QColor(219, 198, 179))) else: return dict()
def _color_version_nodes(node, action, tag, description): pipelineInfo = None if action is not None and VistrailManager.initialized: vistraildata = VistrailManager() # Warning: the first scene might get created before the # VistrailManager gets the 'controller_changed' signal, thus # VistrailManager() might be None if vistraildata is not None: pipelineInfo = vistraildata.get_pipeline(action.id) if tag == 'dat-vars': # Variable root brush = QtGui.QBrush(QtGui.QColor(0, 0, 0)) return dict(VERSION_USER_BRUSH=brush, VERSION_OTHER_BRUSH=brush, VERSION_LABEL_COLOR=QtGui.QColor(255, 255, 255), VERSION_SHAPE='rectangle') elif tag is not None and tag.startswith('dat-var-'): # Variables return dict(VERSION_USER_BRUSH=QtGui.QBrush(QtGui.QColor(27, 27, 75)), VERSION_OTHER_BRUSH=QtGui.QBrush(QtGui.QColor(72, 50, 25)), VERSION_LABEL_COLOR=QtGui.QColor(255, 255, 255), VERSION_SHAPE='rectangle') elif pipelineInfo is not None: return dict( VERSION_USER_BRUSH=QtGui.QBrush(QtGui.QColor(171, 169, 214)), VERSION_OTHER_BRUSH=QtGui.QBrush(QtGui.QColor(219, 198, 179))) else: return dict()
def resetClicked(self): mngr = VistrailManager(self.cell._controller) pipeline = mngr.get_pipeline(self.cell.cellInfo) if pipeline.version != self.cell._controller.current_version: self.cell._controller.change_selected_version(pipeline.version) currentTabIndex = self.tabWidget.currentIndex() self.setup(self.cell, self.plot) self.tabWidget.setCurrentIndex(currentTabIndex)
def applyClicked(self): self.okClicked() # Bring this overlay back up self.cell._set_overlay(DefaultPlotConfigOverlay) mngr = VistrailManager(self.cell._controller) pipeline = mngr.get_pipeline(self.cell.cellInfo) self.cell._overlay.setup(self.cell, pipeline.recipe.plot)
def okClicked(self): mngr = VistrailManager(self.cell._controller) pipeline = mngr.get_pipeline(self.cell.cellInfo) if pipeline.version != self.cell._controller.current_version: new_pipeline = PipelineInformation( self.cell._controller.current_version, pipeline.recipe, pipeline.conn_map, pipeline.port_map) mngr.created_pipeline(self.cell.cellInfo, new_pipeline) self.cell.update_pipeline(True) else: self.cell._set_overlay(None)
def _get_custom_version_panels(controller, version): _ = translate("recipe_version_panel") if not VistrailManager.initialized: return [] vistraildata = VistrailManager(controller) if vistraildata is None: return [] pipelineInfo = vistraildata.get_pipeline(version) if pipelineInfo is None: return [] monospace = QtGui.QFont('Monospace') monospace.setStyleHint(QtGui.QFont.TypeWriter) recipe = pipelineInfo.recipe recipe_widget = QtGui.QGroupBox(_("DAT recipe")) recipe_widget.setSizePolicy( recipe_widget.sizePolicy().horizontalPolicy(), QtGui.QSizePolicy.Fixed) layout = QtGui.QVBoxLayout() line = QtGui.QHBoxLayout() line.addWidget(QtGui.QLabel(_("Plot:"))) plot_label = QtGui.QLabel("%s" % recipe.plot.name) plot_label.setFont(monospace) line.addWidget(plot_label) layout.addLayout(line) layout.addWidget(QtGui.QLabel(_("Variables:"))) variable_list = QtGui.QTextEdit() color = variable_list.textColor() variable_list.setEnabled(False) variable_list.setTextColor(color) variable_list.setFont(monospace) variable_list.setLineWrapMode(QtGui.QTextEdit.NoWrap) variable_list.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff) text = [] for p_values in recipe.parameters.itervalues(): for value in p_values: if value.type == RecipeParameterValue.VARIABLE: text.append(value.variable.name) text = '\n'.join(text) variable_list.setPlainText(text) variable_list.setFixedHeight( variable_list.document().size().height()) layout.addWidget(variable_list) recipe_widget.setLayout(layout) return [(-1, recipe_widget)]
def setup(self, cell, plot): self.cell = cell self.plot = plot # Get pipeline of the cell mngr = VistrailManager(cell._controller) pipelineInfo = mngr.get_pipeline(cell.cellInfo) # Clear old tabs self.tabWidget.clear() # Get all of the plot modules in the pipeline plot_modules = get_plot_modules( pipelineInfo, cell._controller.current_pipeline) registry = get_module_registry() getter = registry.get_configuration_widget for module in plot_modules: widgetType = None widget = None # Try to get custom config widget for the module try: widgetType = \ getter(module.package, module.name, module.namespace) except ModuleRegistryException: pass if widgetType: # Use custom widget widget = widgetType(module, cell._controller) self.connect(widget, QtCore.SIGNAL("doneConfigure"), self.configureDone) self.connect(widget, QtCore.SIGNAL("stateChanged"), self.stateChanged) else: # Use PortsList widget, only if module has ports widget = DATPortsList(self) widget.update_module(module) if len(widget.port_spec_items) > 0: widget.set_controller(cell._controller) else: widget = None # Add widget in new tab if widget: self.tabWidget.addTab(widget, module.name)
def _get_custom_version_panels(controller, version): _ = translate("recipe_version_panel") if not VistrailManager.initialized: return [] vistraildata = VistrailManager(controller) if vistraildata is None: return [] pipelineInfo = vistraildata.get_pipeline(version) if pipelineInfo is None: return [] monospace = QtGui.QFont('Monospace') monospace.setStyleHint(QtGui.QFont.TypeWriter) recipe = pipelineInfo.recipe recipe_widget = QtGui.QGroupBox(_("DAT recipe")) recipe_widget.setSizePolicy(recipe_widget.sizePolicy().horizontalPolicy(), QtGui.QSizePolicy.Fixed) layout = QtGui.QVBoxLayout() line = QtGui.QHBoxLayout() line.addWidget(QtGui.QLabel(_("Plot:"))) plot_label = QtGui.QLabel("%s" % recipe.plot.name) plot_label.setFont(monospace) line.addWidget(plot_label) layout.addLayout(line) layout.addWidget(QtGui.QLabel(_("Variables:"))) variable_list = QtGui.QTextEdit() color = variable_list.textColor() variable_list.setEnabled(False) variable_list.setTextColor(color) variable_list.setFont(monospace) variable_list.setLineWrapMode(QtGui.QTextEdit.NoWrap) variable_list.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff) text = [] for p_values in recipe.parameters.itervalues(): for value in p_values: if value.type == RecipeParameterValue.VARIABLE: text.append(value.variable.name) text = '\n'.join(text) variable_list.setPlainText(text) variable_list.setFixedHeight(variable_list.document().size().height()) layout.addWidget(variable_list) recipe_widget.setLayout(layout) return [(-1, recipe_widget)]