Search in sources :

Example 11 with NodeFactory

use of org.knime.core.node.NodeFactory in project knime-core by knime.

the class AbstractRepositoryView method enrichWithAdditionalInfo.

/**
 * This methods recursively retrieves and enriches the repository objects with additional information,
 * e.g. number of ports, whether the node is streamable and/or distributable, etc.
 * Should be called only after the repository content was already loaded with {@link #readRepository(Composite, IProgressMonitor)}.
 */
protected void enrichWithAdditionalInfo(final IRepositoryObject parent, final IProgressMonitor monitor, final boolean updateTreeStructure) {
    if (monitor.isCanceled()) {
        return;
    }
    if (!m_additionalInfoAvailable) {
        if (parent instanceof IContainerObject) {
            IRepositoryObject[] children = ((IContainerObject) parent).getChildren();
            for (IRepositoryObject child : children) {
                enrichWithAdditionalInfo(child, monitor, updateTreeStructure);
            }
        } else if (parent instanceof NodeTemplate) {
            NodeTemplate nodeTemplate = (NodeTemplate) parent;
            try {
                NodeFactory<? extends NodeModel> nf = nodeTemplate.createFactoryInstance();
                NodeModel nm = nf.createNodeModel();
                // check whether the current node model overrides the #createStreamableOperator-method
                Method m = nm.getClass().getMethod("createStreamableOperator", PartitionInfo.class, PortObjectSpec[].class);
                if (m.getDeclaringClass() != NodeModel.class) {
                    // method has been overriden -> node is probably streamable or distributable
                    nodeTemplate.addAdditionalInfo(KEY_INFO_STREAMABLE, "streamable");
                }
            // possible TODO: parse xml description and get some more additional information (e.g. short description, ...)
            // nodeTemplate.addAdditionalInfo(KEY_INFO_SHORT_DESCRIPTION,
            // "this could be the short description, number of ports etc.");
            } catch (Throwable t) {
                LOGGER.error("Unable to instantiate the node " + nodeTemplate.getFactory().getName(), t);
                return;
            }
            if (!updateTreeStructure) {
                Display.getDefault().asyncExec(new Runnable() {

                    @Override
                    public void run() {
                        if (!m_viewer.getControl().isDisposed()) {
                            m_viewer.update(parent, null);
                        }
                    }
                });
            } else {
                Display.getDefault().asyncExec(new Runnable() {

                    @Override
                    public void run() {
                        if (!m_viewer.getControl().isDisposed()) {
                            m_viewer.update(parent, null);
                            TreeViewerUpdater.update(m_viewer, true, false);
                        }
                    }
                });
            }
        }
    }
}
Also used : NodeModel(org.knime.core.node.NodeModel) NodeTemplate(org.knime.workbench.repository.model.NodeTemplate) MetaNodeTemplate(org.knime.workbench.repository.model.MetaNodeTemplate) NodeFactory(org.knime.core.node.NodeFactory) IContainerObject(org.knime.workbench.repository.model.IContainerObject) Method(java.lang.reflect.Method) PartitionInfo(org.knime.core.node.streamable.PartitionInfo) IRepositoryObject(org.knime.workbench.repository.model.IRepositoryObject)

Example 12 with NodeFactory

use of org.knime.core.node.NodeFactory in project knime-core by knime.

the class RepositoryFactory method createNodeSet.

/**
 * Creates the set of dynamic node templates.
 *
 * @param root the root to add the missing categories in
 * @param element from the extension points
 * @return the created dynamic node templates
 */
public static Collection<DynamicNodeTemplate> createNodeSet(final Root root, final IConfigurationElement element) {
    String iconPath = element.getAttribute("default-category-icon");
    // Try to load the node set factory class...
    NodeSetFactory nodeSet;
    // this ensures that the class is loaded by the correct eclipse
    // classloaders
    GlobalClassCreator.lock.lock();
    try {
        nodeSet = (NodeSetFactory) element.createExecutableExtension("factory-class");
    } catch (Throwable e) {
        throw new IllegalArgumentException("Can't load factory class for node: " + element.getAttribute("factory-class"), e);
    } finally {
        GlobalClassCreator.lock.unlock();
    }
    Collection<DynamicNodeTemplate> dynamicNodeTemplates = new ArrayList<DynamicNodeTemplate>();
    // for all nodes in the node set
    for (String factoryId : nodeSet.getNodeFactoryIds()) {
        @SuppressWarnings("unchecked") Class<NodeFactory<? extends NodeModel>> factoryClass = (Class<NodeFactory<? extends NodeModel>>) nodeSet.getNodeFactory(factoryId);
        // Try to load the node factory class...
        NodeFactory<? extends NodeModel> factory;
        // this ensures that the class is loaded by the correct eclipse
        // classloaders
        GlobalClassCreator.lock.lock();
        try {
            factory = DynamicNodeTemplate.createFactoryInstance(factoryClass, nodeSet, factoryId);
        } catch (Throwable e) {
            throw new IllegalArgumentException("Can't load factory class for node: " + factoryClass.getName() + "-" + factoryId, e);
        } finally {
            GlobalClassCreator.lock.unlock();
        }
        // DynamicNodeFactory implementations can set deprecation independently from extension
        if (factory.isDeprecated()) {
            continue;
        }
        DynamicNodeTemplate node = new DynamicNodeTemplate(factoryClass, factoryId, nodeSet, factory.getNodeName());
        node.setAfterID(nodeSet.getAfterID(factoryId));
        if (!Boolean.getBoolean("java.awt.headless")) {
            Image icon = ImageRepository.getIconImage(factory);
            node.setIcon(icon);
        }
        node.setCategoryPath(nodeSet.getCategoryPath(factoryId));
        dynamicNodeTemplates.add(node);
        String pluginID = element.getDeclaringExtension().getNamespaceIdentifier();
        // 
        // Insert in proper location, create all categories on
        // the path
        // if not already there
        // 
        String path = node.getCategoryPath();
        if (path.startsWith("/")) {
            path = path.substring(1);
        }
        // split the path
        String[] segments = path.split("/");
        // start at root
        IContainerObject container = root;
        String currentPath = "";
        for (int i = 0; i < segments.length; i++) {
            IRepositoryObject obj = container.getChildByID(segments[i], false);
            currentPath += segments[i];
            if (obj == null) {
                Category cat = createCategory(pluginID, segments[i], "", segments[i], "", iconPath, currentPath);
                // append the newly created category to the container
                container.addChild(cat);
                obj = cat;
            }
            currentPath += "/";
            // continue at this level
            container = (IContainerObject) obj;
        }
    }
    return dynamicNodeTemplates;
}
Also used : Category(org.knime.workbench.repository.model.Category) ArrayList(java.util.ArrayList) Image(org.eclipse.swt.graphics.Image) DynamicNodeTemplate(org.knime.workbench.repository.model.DynamicNodeTemplate) IRepositoryObject(org.knime.workbench.repository.model.IRepositoryObject) NodeModel(org.knime.core.node.NodeModel) NodeFactory(org.knime.core.node.NodeFactory) DynamicNodeFactory(org.knime.core.node.DynamicNodeFactory) NodeSetFactory(org.knime.core.node.NodeSetFactory) IContainerObject(org.knime.workbench.repository.model.IContainerObject)

Example 13 with NodeFactory

use of org.knime.core.node.NodeFactory in project knime-core by knime.

the class CovarianceMatrixCalculatorTest method setUp.

/**
 * @throws java.lang.Exception
 */
@Before
public void setUp() throws Exception {
    @SuppressWarnings({ "unchecked", "rawtypes" }) NodeFactory<NodeModel> dummyFactory = (NodeFactory) new VirtualParallelizedChunkPortObjectInNodeFactory(new PortType[0]);
    m_exec = new ExecutionContext(new DefaultNodeProgressMonitor(), new Node(dummyFactory), SingleNodeContainer.MemoryPolicy.CacheOnDisc, new HashMap<Integer, ContainerTable>());
}
Also used : NodeModel(org.knime.core.node.NodeModel) ExecutionContext(org.knime.core.node.ExecutionContext) VirtualParallelizedChunkPortObjectInNodeFactory(org.knime.core.node.workflow.virtual.parchunk.VirtualParallelizedChunkPortObjectInNodeFactory) NodeFactory(org.knime.core.node.NodeFactory) HashMap(java.util.HashMap) VirtualParallelizedChunkPortObjectInNodeFactory(org.knime.core.node.workflow.virtual.parchunk.VirtualParallelizedChunkPortObjectInNodeFactory) DefaultNodeProgressMonitor(org.knime.core.node.DefaultNodeProgressMonitor) Node(org.knime.core.node.Node) PortType(org.knime.core.node.port.PortType) Before(org.junit.Before)

Example 14 with NodeFactory

use of org.knime.core.node.NodeFactory in project knime-core by knime.

the class LearnerTest method setUp.

/**
 * @throws java.lang.Exception
 */
@Before
public void setUp() throws Exception {
    NodeFactory nodeFactory = new LogRegLearnerNodeFactory();
    Node node = new Node(nodeFactory);
    m_exec = new ExecutionContext(new DefaultNodeProgressMonitor(), node, SingleNodeContainer.MemoryPolicy.CacheOnDisc, new HashMap<Integer, ContainerTable>());
}
Also used : ExecutionContext(org.knime.core.node.ExecutionContext) NodeFactory(org.knime.core.node.NodeFactory) HashMap(java.util.HashMap) Node(org.knime.core.node.Node) DefaultNodeProgressMonitor(org.knime.core.node.DefaultNodeProgressMonitor) Before(org.junit.Before)

Example 15 with NodeFactory

use of org.knime.core.node.NodeFactory in project knime-core by knime.

the class JoinerTest method setUp.

/**
 * @throws java.lang.Exception
 */
@Before
public void setUp() throws Exception {
    NodeFactory<NodeModel> dummyFactory = (NodeFactory) new VirtualParallelizedChunkPortObjectInNodeFactory(new PortType[0]);
    m_exec = new ExecutionContext(new DefaultNodeProgressMonitor(), new Node(dummyFactory), SingleNodeContainer.MemoryPolicy.CacheOnDisc, new HashMap<Integer, ContainerTable>());
}
Also used : NodeModel(org.knime.core.node.NodeModel) ExecutionContext(org.knime.core.node.ExecutionContext) VirtualParallelizedChunkPortObjectInNodeFactory(org.knime.core.node.workflow.virtual.parchunk.VirtualParallelizedChunkPortObjectInNodeFactory) NodeFactory(org.knime.core.node.NodeFactory) HashMap(java.util.HashMap) VirtualParallelizedChunkPortObjectInNodeFactory(org.knime.core.node.workflow.virtual.parchunk.VirtualParallelizedChunkPortObjectInNodeFactory) DefaultNodeProgressMonitor(org.knime.core.node.DefaultNodeProgressMonitor) Node(org.knime.core.node.Node) PortType(org.knime.core.node.port.PortType) Before(org.junit.Before)

Aggregations

NodeFactory (org.knime.core.node.NodeFactory)17 Node (org.knime.core.node.Node)11 NodeModel (org.knime.core.node.NodeModel)11 HashMap (java.util.HashMap)7 PortType (org.knime.core.node.port.PortType)7 VirtualParallelizedChunkPortObjectInNodeFactory (org.knime.core.node.workflow.virtual.parchunk.VirtualParallelizedChunkPortObjectInNodeFactory)7 Before (org.junit.Before)6 DefaultNodeProgressMonitor (org.knime.core.node.DefaultNodeProgressMonitor)6 ExecutionContext (org.knime.core.node.ExecutionContext)6 InvalidSettingsException (org.knime.core.node.InvalidSettingsException)3 MissingNodeFactory (org.knime.core.node.missing.MissingNodeFactory)3 IContainerObject (org.knime.workbench.repository.model.IContainerObject)3 IRepositoryObject (org.knime.workbench.repository.model.IRepositoryObject)3 IOException (java.io.IOException)2 Image (org.eclipse.swt.graphics.Image)2 CanceledExecutionException (org.knime.core.node.CanceledExecutionException)2 ConfigurableNodeFactory (org.knime.core.node.ConfigurableNodeFactory)2 FileNodePersistor (org.knime.core.node.FileNodePersistor)2 NodeSettingsRO (org.knime.core.node.NodeSettingsRO)2 DialogNode (org.knime.core.node.dialog.DialogNode)2