use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceMarshallingTest method testValue_Byte.
@Test
public void testValue_Byte() throws Exception {
final String expected = "rule \"test_0\"" + "when\n" + " Person( byteField == 100 )\n" + "then\n" + "end";
final GuidedDecisionTree model = new GuidedDecisionTree();
model.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "byteField", "==", new ByteValue(new Byte("100")));
model.setRoot(type);
type.addChild(c1);
final String drl = GuidedDecisionTreeDRLPersistence.getInstance().marshal(model);
assertEqualsIgnoreWhitespace(expected, drl);
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testMultipleRules_MultiplePatterns.
@Test
public void testMultipleRules_MultiplePatterns() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( name == \"Michael\" )\n" + " Address( country == \"England\" )\n" + "then \n" + "end \n" + "rule \"test_1\"\n" + "when \n" + " Person( name == \"Michael\" )\n" + " Address( country == \"Norway\" )\n" + "then \n" + "end \n" + "rule \"test_2\"\n" + "when \n" + " Person( name == \"Fred\" )\n" + "then \n" + "end \n";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type1 = new TypeNodeImpl("Person");
final ConstraintNode c1a = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
final ConstraintNode c1b = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Fred"));
final TypeNode type2 = new TypeNodeImpl("Address");
final ConstraintNode c2a = new ConstraintNodeImpl("Address", "country", "==", new StringValue("England"));
final ConstraintNode c2b = new ConstraintNodeImpl("Address", "country", "==", new StringValue("Norway"));
expected.setRoot(type1);
type1.addChild(c1a);
type1.addChild(c1b);
c1a.addChild(type2);
type2.addChild(c2a);
type2.addChild(c2b);
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(2, _t1.getChildren().size());
assertNotNull(_t1.getChildren().get(0));
assertTrue(_t1.getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c1a = (ConstraintNode) _t1.getChildren().get(0);
assertEquals(c1a.getClassName(), _c1a.getClassName());
assertEquals(c1a.getFieldName(), _c1a.getFieldName());
assertEquals(c1a.getOperator(), _c1a.getOperator());
assertEquals(c1a.getValue().getValue().toString(), _c1a.getValue().getValue().toString());
assertNotNull(_t1.getChildren().get(1));
assertTrue(_t1.getChildren().get(1) instanceof ConstraintNode);
final ConstraintNode _c1b = (ConstraintNode) _t1.getChildren().get(1);
assertEquals(c1b.getClassName(), _c1b.getClassName());
assertEquals(c1b.getFieldName(), _c1b.getFieldName());
assertEquals(c1b.getOperator(), _c1b.getOperator());
assertEquals(c1b.getValue().getValue().toString(), _c1b.getValue().getValue().toString());
assertEquals(1, _c1a.getChildren().size());
assertNotNull(_c1a.getChildren().get(0));
assertTrue(_c1a.getChildren().get(0) instanceof TypeNode);
final TypeNode _t2 = (TypeNode) _c1a.getChildren().get(0);
assertEquals(type2.getClassName(), _t2.getClassName());
assertFalse(_t2.isBound());
assertEquals(2, _t2.getChildren().size());
assertNotNull(_t2.getChildren().get(0));
assertTrue(_t2.getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c2a = (ConstraintNode) _t2.getChildren().get(0);
assertEquals(c2a.getClassName(), _c2a.getClassName());
assertEquals(c2a.getFieldName(), _c2a.getFieldName());
assertEquals(c2a.getOperator(), _c2a.getOperator());
assertEquals(c2a.getValue().getValue().toString(), _c2a.getValue().getValue().toString());
assertNotNull(_t2.getChildren().get(1));
assertTrue(_t2.getChildren().get(1) instanceof ConstraintNode);
final ConstraintNode _c2b = (ConstraintNode) _t2.getChildren().get(1);
assertEquals(c2b.getClassName(), _c2b.getClassName());
assertEquals(c2b.getFieldName(), _c2b.getFieldName());
assertEquals(c2b.getOperator(), _c2b.getOperator());
assertEquals(c2b.getValue().getValue().toString(), _c2b.getValue().getValue().toString());
assertEquals(0, _c2a.getChildren().size());
assertEquals(0, _c2b.getChildren().size());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode 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);
}
}
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools-wb by kiegroup.
the class EditConstraintPopup method cloneNode.
// Clone node whilst editing to preserve original node should User cancel the edit
private ConstraintNode cloneNode(final ConstraintNode node) {
final ConstraintNode clone = new ConstraintNodeImpl(node.getClassName(), node.getFieldName());
if (node.getOperator() != null) {
clone.setOperator(node.getOperator());
}
if (node.getValue() != null) {
clone.setValue(node.getValue());
}
clone.setParent(node.getParent());
clone.setBinding(node.getBinding());
return clone;
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools-wb by kiegroup.
the class GuidedDecisionTreePalette method makeStencils.
private Widget makeStencils(final String className, final boolean isReadOnly) {
final GuidedDecisionTreePaletteGroup paletteGroup = new GuidedDecisionTreePaletteGroup();
if (className == null) {
return paletteGroup;
}
oracle.getFieldCompletions(className, new Callback<ModelField[]>() {
@Override
public void callback(final ModelField[] mfs) {
if (mfs == null || mfs.length == 0) {
return;
}
final TypeNode tn = new TypeNodeImpl(className);
paletteGroup.addStencil(typeNodeFactory, stencilBuilder, new TypeFactoryHelper(tn, isReadOnly), isReadOnly);
for (ModelField mf : mfs) {
final String fieldName = mf.getName();
if (!fieldName.equals(DataType.TYPE_THIS)) {
final ConstraintNode cn = new ConstraintNodeImpl(className, fieldName, "", new StringValue(""));
paletteGroup.addStencil(constraintNodeFactory, stencilBuilder, new ConstraintFactoryHelper(cn, isReadOnly), isReadOnly);
}
}
}
});
return paletteGroup;
}
Aggregations