ai.libs.jaicore.planning.hierarchical.problems.stn.TaskNetwork Maven / Gradle / Ivy

package ai.libs.jaicore.planning.hierarchical.problems.stn;

import java.util.List;

import ai.libs.jaicore.basic.StringUtil;
import ai.libs.jaicore.graph.Graph;
import ai.libs.jaicore.logic.fol.structure.Literal;

public class TaskNetwork extends Graph {

	public TaskNetwork() {
		super();
	}

	public TaskNetwork(final List chain) {
		int n = chain.size();
		Literal prev = null;
		for (int i = 0; i < n; i++) {
			Literal cur = chain.get(i);
			this.addItem(cur);
			if (prev != null) {
				this.addEdge(prev, cur);
			}
			prev = cur;
		}
	}

	public TaskNetwork(final Graph graph) {
		super(graph);
	}

	public TaskNetwork(final String chain) {
		super();
		Literal current = null;
		int id = 1;
		for (String taskDescription : StringUtil.explode(chain, "->")) {
			if (!taskDescription.trim().isEmpty()) {
				Literal task = new Literal("tn" + "_" + id + "-" + taskDescription.trim());
				this.addItem(task);
				if (current != null) {
					this.addEdge(current, task);
				}
				current = task;
				id++;
			}
		}
	}
}