X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;f=planner.cpp;h=1210d0479309df2ed3af1a2fe0b8f783fa5d1a33;hb=895d43297511597e9d8734c0fd5ba534f551de5f;hp=c5016a18b4e43b8aa65147e204ed58c6f27e6663;hpb=dc49c0c521090f0eb4b9692b25a129537c07e19e;p=popboot.git

diff --git a/planner.cpp b/planner.cpp
index c5016a1..1210d04 100644
--- a/planner.cpp
+++ b/planner.cpp
@@ -1,20 +1,74 @@
 #include "planner.h"
 #include "node.h"
+#include <iostream>
 using namespace std;
+using namespace __gnu_cxx;
 
 Planner::Planner(std::vector<Action> actions, literals init, literals goal){
 
-	_start = new Node(Action("",init, literals()));
-	Node* finish = new Node(Action("",literals(),goal));
+	_start = new Node(Action("start",literals(), init));
+	Node* finish = new Node(Action("finish",goal,literals()));
 
 	for(vector<Action>::iterator action = actions.begin(); action != actions.end(); ++action){
-		literals preconds = action->preconditions();
-		for (literals::iterator effect = preconds.begin(); effect != preconds.end(); ++effect){
+		const literals& effects = action->effects();
+		for (literals::const_iterator effect = effects.begin(); effect != effects.end(); ++effect){
+			cerr << "Adding effect: " << *effect << endl;
 			_actions[*effect] = *action;
 		}
 	}
+	makePlan(finish);
+}
+
+Planner::~Planner(){
+	for (hash_map<string,Node*>::iterator node = _addedNodes.begin(); node != _addedNodes.end(); ++node){
+		cerr << "Deleting node " << node->second->action().executable() << endl;
+		delete node->second;
+	}
 }
 
 
 void Planner::makePlan(Node* node){
+	cerr << "Fetching preconditions for action: " << node->action().executable() << ".. ";
+	const literals& preconds = node->action().preconditions();
+	cerr << "done" << endl;
+
+	addNode(node);
+
+	if (preconds.size() == 0){
+		cerr << "Found no preconds" << endl;
+		_start->addChild(node);
+	}else{
+		for (literals::const_iterator precond = preconds.begin(); precond != preconds.end(); ++precond){
+			cerr << "Looking for: " << *precond << endl;
+			hash_map<string,Node*>::iterator addedNode = _addedNodes.find(*precond);
+			if(addedNode != _addedNodes.end()){
+				cerr << "Using already added node" << endl;
+				addedNode->second->addChild(node);
+			}else {
+				hash_map<string, Action>::iterator action = _actions.find(*precond);
+				if (action != _actions.end()){
+					cerr << "Adding new node" << endl;
+					Node* newnode = new Node(action->second);
+					newnode->addChild(node);
+					makePlan(newnode);
+				}else{
+					cerr << "Action with effect: " << *precond << " not found!" << endl;
+				}
+			}
+		}
+	}
+}
+
+void Planner::addNode(Node* node){
+	const literals& effects = node->action().effects();
+
+	for (literals::const_iterator effect = effects.begin(); effect != effects.end(); ++effect){
+		cout << "Adding node for effect: " << *effect << endl;
+		_addedNodes[*effect] = node;
+	}
+}
+
+
+void Planner::execute(){
+	_start->execute(literals());
 }