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

Example 51 with TypeNode

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

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_MultiplePatterns.

@Test
public void testSingleRule_MultiplePatterns() throws Exception {
    final String drl = "rule \"test_0\"\n" + "when \n" + "  Person( name == \"Michael\" )\n" + "  Address( country == \"England\" )\n" + "then \n" + "end";
    final GuidedDecisionTree expected = new GuidedDecisionTree();
    expected.setTreeName("test");
    final TypeNode type1 = new TypeNodeImpl("Person");
    final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
    final TypeNode type2 = new TypeNodeImpl("Address");
    final ConstraintNode c2 = new ConstraintNodeImpl("Address", "country", "==", new StringValue("England"));
    expected.setRoot(type1);
    type1.addChild(c1);
    c1.addChild(type2);
    type2.addChild(c2);
    addModelField("Person", "this", "Person", DataType.TYPE_THIS);
    addModelField("Person", "name", String.class.getName(), DataType.TYPE_STRING);
    addModelField("Address", "this", "Address", DataType.TYPE_THIS);
    addModelField("Address", "country", String.class.getName(), DataType.TYPE_STRING);
    final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
    assertEquals(expected.getTreeName(), model.getTreeName());
    assertNotNull(model.getRoot());
    final TypeNode _t1 = model.getRoot();
    assertEquals(type1.getClassName(), _t1.getClassName());
    assertFalse(_t1.isBound());
    assertEquals(1, _t1.getChildren().size());
    assertNotNull(_t1.getChildren().get(0));
    assertTrue(_t1.getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c1 = (ConstraintNode) _t1.getChildren().get(0);
    assertEquals(c1.getClassName(), _c1.getClassName());
    assertEquals(c1.getFieldName(), _c1.getFieldName());
    assertEquals(c1.getOperator(), _c1.getOperator());
    assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
    assertEquals(1, _c1.getChildren().size());
    assertNotNull(_c1.getChildren().get(0));
    assertTrue(_c1.getChildren().get(0) instanceof TypeNode);
    final TypeNode _t2 = (TypeNode) _c1.getChildren().get(0);
    assertEquals(type2.getClassName(), _t2.getClassName());
    assertFalse(_t2.isBound());
    assertEquals(1, _t2.getChildren().size());
    assertNotNull(_t2.getChildren().get(0));
    assertTrue(_t2.getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c2 = (ConstraintNode) _t2.getChildren().get(0);
    assertEquals(c2.getClassName(), _c2.getClassName());
    assertEquals(c2.getFieldName(), _c2.getFieldName());
    assertEquals(c2.getOperator(), _c2.getOperator());
    assertEquals(c2.getValue().getValue().toString(), _c2.getValue().getValue().toString());
    assertEquals(0, _c2.getChildren().size());
}

Example 52 with TypeNode

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

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testValue_Double.

@Test
public void testValue_Double() throws Exception {
    final String drl = "rule \"test_0\"\n" + "when \n" + "  Person( doubleField == 1000.56 )\n" + "then \n" + "end";
    final GuidedDecisionTree expected = new GuidedDecisionTree();
    expected.setTreeName("test");
    final TypeNode type = new TypeNodeImpl("Person");
    final ConstraintNode c1 = new ConstraintNodeImpl("Person", "doubleField", "==", new DoubleValue(1000.56));
    expected.setRoot(type);
    type.addChild(c1);
    addModelField("Person", "this", "Person", DataType.TYPE_THIS);
    addModelField("Person", "doubleField", Double.class.getName(), DataType.TYPE_NUMERIC_DOUBLE);
    final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
    assertEquals(expected.getTreeName(), model.getTreeName());
    assertNotNull(model.getRoot());
    assertEquals(type.getClassName(), model.getRoot().getClassName());
    assertFalse(model.getRoot().isBound());
    assertEquals(1, model.getRoot().getChildren().size());
    assertNotNull(model.getRoot().getChildren().get(0));
    assertTrue(model.getRoot().getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c1 = (ConstraintNode) model.getRoot().getChildren().get(0);
    assertEquals(c1.getClassName(), _c1.getClassName());
    assertEquals(c1.getFieldName(), _c1.getFieldName());
    assertEquals(c1.getOperator(), _c1.getOperator());
    assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
}

Example 53 with TypeNode

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

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_WithImport.

@Test
public void testSingleRule_WithImport() throws Exception {
    final String drl = "package smurf; \n" + "import org.drools.workbench.models.guided.dtree.backend.Person; \n" + "rule \"test_0\"\n" + "when \n" + "  Person( name == \"Michael\" )\n" + "then \n" + "end";
    final GuidedDecisionTree expected = new GuidedDecisionTree();
    expected.setTreeName("test");
    final TypeNode type = new TypeNodeImpl("Person");
    final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
    expected.setRoot(type);
    type.addChild(c1);
    addModelField("org.drools.workbench.models.guided.dtree.backend.Person", "this", "org.drools.workbench.models.guided.dtree.backend.Person", DataType.TYPE_THIS);
    addModelField("org.drools.workbench.models.guided.dtree.backend.Person", "name", String.class.getName(), DataType.TYPE_STRING);
    final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
    assertEquals(expected.getTreeName(), model.getTreeName());
    assertNotNull(model.getRoot());
    assertEquals(type.getClassName(), model.getRoot().getClassName());
    assertFalse(model.getRoot().isBound());
    assertEquals(1, model.getRoot().getChildren().size());
    assertNotNull(model.getRoot().getChildren().get(0));
    assertTrue(model.getRoot().getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c1 = (ConstraintNode) model.getRoot().getChildren().get(0);
    assertEquals(c1.getClassName(), _c1.getClassName());
    assertEquals(c1.getFieldName(), _c1.getFieldName());
    assertEquals(c1.getOperator(), _c1.getOperator());
    assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
}

Example 54 with TypeNode

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

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_ActionModifyMultipleFields.

@Test
public void testSingleRule_ActionModifyMultipleFields() throws Exception {
    final String drl = "rule \"test_0\"\n" + "when \n" + "  $p : Person( )\n" + "then \n" + "  modify( $p ) {\n" + "    setAge( 25 ), \n" + "    setName( \"Michael\" )\n" + "  }\n" + "end";
    final GuidedDecisionTree expected = new GuidedDecisionTree();
    expected.setTreeName("test");
    final TypeNode type = new TypeNodeImpl("Person");
    type.setBinding("$p");
    expected.setRoot(type);
    final ActionUpdateNode action = new ActionUpdateNodeImpl(type);
    action.setModify(true);
    action.getFieldValues().add(new ActionFieldValueImpl("age", new IntegerValue(25)));
    action.getFieldValues().add(new ActionFieldValueImpl("name", new StringValue("Michael")));
    type.addChild(action);
    addModelField("Person", "this", "Person", DataType.TYPE_THIS);
    addModelField("Person", "name", String.class.getName(), DataType.TYPE_STRING);
    addModelField("Person", "age", Integer.class.getName(), DataType.TYPE_NUMERIC_INTEGER);
    final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
    assertEquals(expected.getTreeName(), model.getTreeName());
    assertNotNull(model.getRoot());
    assertEquals(type.getClassName(), model.getRoot().getClassName());
    assertTrue(model.getRoot().isBound());
    assertEquals(type.getBinding(), model.getRoot().getBinding());
    assertEquals(1, model.getRoot().getChildren().size());
    assertNotNull(model.getRoot().getChildren().get(0));
    assertTrue(model.getRoot().getChildren().get(0) instanceof ActionUpdateNode);
    final ActionUpdateNode _action = (ActionUpdateNode) model.getRoot().getChildren().get(0);
    assertEquals(action.getBoundNode().getBinding(), _action.getBoundNode().getBinding());
    assertEquals(action.isModify(), _action.isModify());
    assertEquals(action.getFieldValues().size(), _action.getFieldValues().size());
    assertEquals(2, _action.getFieldValues().size());
    assertEquals(action.getFieldValues().get(0).getFieldName(), _action.getFieldValues().get(0).getFieldName());
    assertEquals(action.getFieldValues().get(0).getValue().getValue().toString(), _action.getFieldValues().get(0).getValue().getValue().toString());
    assertEquals(action.getFieldValues().get(1).getFieldName(), _action.getFieldValues().get(1).getFieldName());
    assertEquals(action.getFieldValues().get(1).getValue().getValue().toString(), _action.getFieldValues().get(1).getValue().getValue().toString());
    assertEquals(0, _action.getChildren().size());
}

Example 55 with TypeNode

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

the class GuidedDecisionTreeWidget method processChildren.

private void processChildren(final Node node, final WiresBaseTreeNode uiNode, final boolean isReadOnly) {
    uiNode.setSelectionManager(this);
    uiNode.setShapesManager(this);
    uiNode.setLayoutManager(layoutManager);
    if (uiNode instanceof BaseGuidedDecisionTreeShape) {
        ((BaseGuidedDecisionTreeShape) uiNode).setPresenter(presenter);
    }
    canvasLayer.add(uiNode);
    shapesInCanvas.add(uiNode);
    final Iterator<Node> itr = node.iterator();
    while (itr.hasNext()) {
        final Node child = itr.next();
        WiresBaseTreeNode uiChildNode = null;
        if (child instanceof TypeNode) {
            uiChildNode = typeNodeFactory.getShape((TypeNode) child, isReadOnly);
        } else if (child instanceof ConstraintNode) {
            uiChildNode = constraintNodeFactory.getShape((ConstraintNode) child, isReadOnly);
        } else if (child instanceof ActionInsertNode) {
            uiChildNode = actionInsertNodeFactory.getShape((ActionInsertNode) child, isReadOnly);
        } else if (child instanceof ActionUpdateNode) {
            uiChildNode = actionUpdateNodeFactory.getShape((ActionUpdateNode) child, isReadOnly);
        } else if (child instanceof ActionRetractNode) {
            uiChildNode = actionRetractNodeFactory.getShape((ActionRetractNode) child, isReadOnly);
        }
        if (uiChildNode != null) {
            uiNode.addChildNode(uiChildNode);
            processChildren(child, uiChildNode, isReadOnly);
        }
    }
}