Examples with Node org.drools.workbench.models.guided.dtree.shared.model.nodes.Node used on opensource projects

Example 11 with Node

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.Node in project drools by kiegroup.

the class GuidedDecisionTreeModelUnmarshallingVisitor method getTypesOnPath.

private List<TypeNode> getTypesOnPath(final List<Node> nodes) {
    final List<TypeNode> types = new ArrayList<TypeNode>();
    for (Node node : nodes) {
        if (node instanceof TypeNode) {
            final TypeNode tn = (TypeNode) node;
            types.add(tn);
        }
    }
    return types;
}

Example 12 with Node

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.Node in project drools by kiegroup.

the class GuidedDecisionTreeModelUnmarshallingVisitor method visit.

private GuidedDecisionTree visit(final String drl, final String baseFileName, final List<String> globals, final PackageDataModelOracle dmo) {
    PortablePreconditions.checkNotNull("drl", drl);
    PortablePreconditions.checkNotNull("baseFileName", baseFileName);
    PortablePreconditions.checkNotNull("globals", globals);
    PortablePreconditions.checkNotNull("dmo", dmo);
    final GuidedDecisionTree model = new GuidedDecisionTree();
    model.setTreeName(baseFileName);
    // De-serialize Package name
    final String packageName = PackageNameParser.parsePackageName(drl);
    model.setPackageName(packageName);
    // De-serialize imports
    final Imports imports = ImportsParser.parseImports(drl);
    for (Import item : imports.getImports()) {
        model.getImports().addImport(item);
    }
    // Split DRL into separate rules
    final List<String> rules = new ArrayList<String>();
    final Pattern pattern = Pattern.compile("\\s?rule\\s(.+?)\\send\\s?", Pattern.DOTALL);
    final Matcher matcher = pattern.matcher(drl);
    while (matcher.find()) {
        rules.add(matcher.group());
    }
    // Build a linear Path of Nodes for each rule
    final List<GuidedDecisionTreeParserExtendedError> rulesParserContent = new ArrayList<GuidedDecisionTreeParserExtendedError>();
    for (String rule : rules) {
        final GuidedDecisionTreeParserExtendedError ruleParserContent = new GuidedDecisionTreeParserExtendedError();
        rulesParserContent.add(ruleParserContent);
        try {
            final RuleModel rm = RuleModelDRLPersistenceImpl.getInstance().unmarshal(rule, globals, dmo);
            ruleParserContent.setOriginalRuleName(rm.name);
            ruleParserContent.setOriginalDrl(rule);
            for (IPattern p : rm.lhs) {
                ruleParserContent.getNodes().addAll(visit(p, model, dmo, ruleParserContent.getMessages()));
            }
            for (IAction a : rm.rhs) {
                ruleParserContent.getNodes().addAll(visit(a, getTypesOnPath(ruleParserContent.getNodes()), model, dmo, ruleParserContent.getMessages()));
            }
        } catch (Exception e) {
            ruleParserContent.getMessages().add(new DefaultParserMessage(e.getMessage()));
        }
    }
    // Combine Paths into a single tree.
    for (GuidedDecisionTreeParserExtendedError ruleParserContent : rulesParserContent) {
        Node activeModelNode = null;
        boolean error = !ruleParserContent.getMessages().isEmpty();
        for (int index = 0; !error && index < ruleParserContent.getNodes().size(); index++) {
            final Node node = ruleParserContent.getNodes().get(index);
            switch(index) {
                case 0:
                    if (!(node instanceof TypeNode)) {
                        ruleParserContent.getMessages().add(new InvalidRootParserMessage());
                        error = true;
                        break;
                    }
                    final TypeNode root = (TypeNode) node;
                    if (model.getRoot() == null) {
                        model.setRoot(root);
                    } else if (!root.equals(model.getRoot())) {
                        ruleParserContent.getMessages().add(new AmbiguousRootParserMessage(root.getClassName()));
                        error = true;
                        break;
                    }
                    activeModelNode = model.getRoot();
                    break;
                default:
                    if (!activeModelNode.getChildren().contains(node)) {
                        // If the active node in the Model doesn't contain the child add it as a new child
                        activeModelNode.addChild(node);
                        activeModelNode = node;
                    } else {
                        // Otherwise swap out the Node in the Path for the existing one in the Model
                        activeModelNode = activeModelNode.getChildren().get(activeModelNode.getChildren().indexOf(node));
                    }
            }
        }
        if (!ruleParserContent.getMessages().isEmpty()) {
            model.getParserErrors().add(new GuidedDecisionTreeParserError(ruleParserContent.getOriginalRuleName(), ruleParserContent.getOriginalDrl(), ruleParserContent.getMessages()));
        }
    }
    return model;
}

Example 13 with Node

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.Node in project drools-wb by kiegroup.

the class GuidedDecisionTreeModelVisitor method visitNode.

private Set<String> visitNode(final Node node) {
    if (node == null) {
        return Collections.EMPTY_SET;
    }
    final Set<String> factTypes = new HashSet<String>();
    if (node instanceof TypeNode) {
        final TypeNode tn = (TypeNode) node;
        factTypes.add(tn.getClassName());
    } else if (node instanceof ActionInsertNode) {
        final ActionInsertNode an = (ActionInsertNode) node;
        factTypes.add(an.getClassName());
    }
    for (Node child : node.getChildren()) {
        factTypes.addAll(visitNode(child));
    }
    return factTypes;
}

Example 14 with Node

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.Node in project drools-wb by kiegroup.

the class ActionRetractShape method setParentNode.

@Override
public void setParentNode(final WiresBaseTreeNode parent) {
    super.setParentNode(parent);
    // Set binding to first bound parent TypeNode
    if (parent instanceof BaseGuidedDecisionTreeShape) {
        Node node = ((BaseGuidedDecisionTreeShape) parent).getModelNode();
        while (node != null) {
            if (node instanceof TypeNode) {
                final TypeNode tn = (TypeNode) node;
                if (tn.isBound()) {
                    getModelNode().setBoundNode(tn);
                    setNodeLabel(getNodeLabel());
                    break;
                }
            }
            node = node.getParent();
        }
    }
}