use of org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceMarshallingTest method testValue_BigDecimal.
@Test
public void testValue_BigDecimal() throws Exception {
final String expected = "rule \"test_0\"" + "when\n" + " Person( bigDecimalField == 1000000B )\n" + "then\n" + "end";
final GuidedDecisionTree model = new GuidedDecisionTree();
model.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "bigDecimalField", "==", new BigDecimalValue(new BigDecimal(1000000)));
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.TypeNode 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;
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode 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;
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_ActionModifyWithConstraint.
@Test
public void testSingleRule_ActionModifyWithConstraint() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " $p : Person( $n : name )\n" + "then \n" + " modify( $p ) {\n" + " setAge( 25 )\n" + " }\n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
type.setBinding("$p");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name");
c1.setBinding("$n");
expected.setRoot(type);
type.addChild(c1);
final ActionUpdateNode action = new ActionUpdateNodeImpl(type);
action.setModify(true);
action.getFieldValues().add(new ActionFieldValueImpl("age", new IntegerValue(25)));
c1.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 ConstraintNode);
final ConstraintNode _c1 = (ConstraintNode) model.getRoot().getChildren().get(0);
assertEquals(c1.getClassName(), _c1.getClassName());
assertEquals(c1.getFieldName(), _c1.getFieldName());
assertNull(_c1.getOperator());
assertNull(_c1.getValue());
assertTrue(_c1.isBound());
assertEquals(c1.getBinding(), _c1.getBinding());
assertEquals(1, _c1.getChildren().size());
assertNotNull(_c1.getChildren().get(0));
assertTrue(_c1.getChildren().get(0) instanceof ActionUpdateNode);
final ActionUpdateNode _action = (ActionUpdateNode) _c1.getChildren().get(0);
assertEquals(action.getBoundNode().getBinding(), _action.getBoundNode().getBinding());
assertEquals(action.isModify(), _action.isModify());
assertEquals(action.getFieldValues().size(), _action.getFieldValues().size());
assertEquals(1, _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(0, _action.getChildren().size());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testValue_Boolean.
@Test
public void testValue_Boolean() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( booleanField == true )\n" + "then \n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "booleanField", "==", new BooleanValue(true));
expected.setRoot(type);
type.addChild(c1);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "booleanField", Boolean.class.getName(), DataType.TYPE_BOOLEAN);
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());
}
Aggregations