package org.jbpm.ui.jpdl3.part.tree;
import java.util.ArrayList;
import java.util.List;
import org.jbpm.ui.common.model.Decision;
import org.jbpm.ui.common.model.EndState;
import org.jbpm.ui.common.model.Fork;
import org.jbpm.ui.common.model.GraphElement;
import org.jbpm.ui.common.model.GroupElement;
import org.jbpm.ui.common.model.Join;
import org.jbpm.ui.common.model.StartState;
import org.jbpm.ui.common.model.State;
import org.jbpm.ui.common.model.Subprocess;
import org.jbpm.ui.common.model.Swimlane;
import org.jbpm.ui.common.model.Variable;
import org.jbpm.ui.common.model.WaitState;
import org.jbpm.ui.common.part.tree.ProcessDefinitionTreeEditPart;
import org.jbpm.ui.jpdl3.model.ActionNode;
import org.jbpm.ui.resource.Messages;
public class ProcessDefinitionTreeEditPart3 extends ProcessDefinitionTreeEditPart {
@Override
protected List<GraphElement> getModelChildren() {
List<GraphElement> result = new ArrayList<GraphElement>();
result.addAll(getModel().getChildren(StartState.class));
result.add(new GroupElement(getModel(), State.class, getDesc("task-node")));
result.add(new GroupElement(getModel(), WaitState.class, getDesc("waitState")));
result.add(new GroupElement(getModel(), Fork.class, getDesc("fork")));
result.add(new GroupElement(getModel(), Join.class, getDesc("join")));
result.add(new GroupElement(getModel(), Subprocess.class, getDesc("process-state")));
result.add(new GroupElement(getModel(), Decision.class, getDesc("decision")));
result.add(new GroupElement(getModel(), ActionNode.class, getDesc("node")));
result.add(new GroupElement(getModel(), Variable.class, Messages.getString("default.variable.name")));
result.add(new GroupElement(getModel(), Swimlane.class, Messages.getString("default.swimlane.name")));
result.addAll(getModel().getChildren(EndState.class));
return result;
}
}