From 78fc212489b573f70e9ca515fa57b27ed395dc7f Mon Sep 17 00:00:00 2001
From: DjWarmonger <warmonger@vp.pl>
Date: Sat, 5 Feb 2011 18:24:32 +0000
Subject: [PATCH] A little more work on Expert System.

---
 AI/GeniusAI/ExpertSystem.cpp | 38 ++++++++++++++++++++++--------------
 AI/GeniusAI/ExpertSystem.h   | 37 ++++++++++++++++++++++++-----------
 2 files changed, 49 insertions(+), 26 deletions(-)

diff --git a/AI/GeniusAI/ExpertSystem.cpp b/AI/GeniusAI/ExpertSystem.cpp
index 2bff9171a..7d22d06fb 100644
--- a/AI/GeniusAI/ExpertSystem.cpp
+++ b/AI/GeniusAI/ExpertSystem.cpp
@@ -14,9 +14,9 @@
 
 template <typename ruleType, typename facts> template <typename cond> void ExpertSystemShell<ruleType, facts>::DataDrivenReasoning(runType type)
 {
-	std::set<ruleType>::iterator ir;
+	std::vector<ruleType>::iterator ir;
 	std::list<fact>::iterator iF;
-	std::set<std::pair<cond, input*>>::iterator ic;
+	std::vector<std::pair<cond, input*>>::iterator ic;
 	bool factWasAdded = false; //carry it over inner loop
 	switch (type)
 	{
@@ -72,7 +72,7 @@ template <typename ruleType, typename facts> template <typename cond> void Exper
 }
 void BonusRule::fireRule()
 {
-	for (std::set<std::pair<BonusCondition, BonusHolder*>>::iterator it = cons.begin(); it != cons.end(); it++)
+	for (std::vector<std::pair<BonusCondition, BonusHolder*>>::iterator it = cons.begin(); it != cons.end(); it++)
 	{
 		switch (it->first.parameter)
 		{ //compare fact with condition
@@ -114,18 +114,26 @@ TLogic operator&&(const TLogic &first, const TLogic &second)
 {
 	return LogicConjunction(first, second);
 }
-//TODO: find out why it does not compile
-//template <typename input, typename conType> void Rule<input, conType>::refreshRule(std::set<conType> &conditionSet)
-//{
-//	cons.clear();
-//	for (std::set<conType>::iterator it = conditionSet.begin(); it != conditionSet.end(); it++)
-//		cons.insert(std::make_pair<conType,input*>(*it, NULL)); //pointer to condition and null fact
-//}
-//template <typename input, typename conType> void Rule<input, conType>::refreshRule()
-//{
-//	for (std::set<std::pair<conType, input*>>::iterator it = cons.begin(); it != cons.end(); it++)
-//		*it->second = NULL;
-//}
+TLogic operator||(const TLogic &first, const TLogic &second)
+{
+	return LogicAlternative(first, second);
+}
+template <typename input, typename conType> void Rule<input, conType>::refreshRule(std::vector<conType> &conditionSet)
+{//replace conditions with new vector
+	cons.clear();
+	std::pair <conType,input*> para;
+	para.second = NULL;
+	for (std::vector<conType>::iterator it = conditionSet.begin(); it != conditionSet.end(); it++)
+	{
+		para.first = *it;
+		cons.push_back(para); //pointer to condition and null fact
+	}
+}
+template <typename input, typename conType> void Rule<input, conType>::refreshRule()
+{//clear matching facts
+	for (std::vector<std::pair<conType, input*>>::iterator it = cons.begin(); it != cons.end(); it++)
+		it->second = NULL;
+}
 bool BonusCondition::matchesFact(Bonus &fact)
 {
 	if (object(fact)) //Bonus(fact) matches local Selector(object)
diff --git a/AI/GeniusAI/ExpertSystem.h b/AI/GeniusAI/ExpertSystem.h
index 8a60909a7..4ed7c953d 100644
--- a/AI/GeniusAI/ExpertSystem.h
+++ b/AI/GeniusAI/ExpertSystem.h
@@ -2,7 +2,7 @@
 #include "../vcmi/CCallback.h"
 #include "../vcmi/lib/HeroBonus.h"
 #include <boost/bind.hpp>
-#include <set>
+#include <vector>
 
 /*
  * ExpertSystem.h, part of VCMI engine
@@ -28,11 +28,11 @@ template <typename ruleType, typename fact> class ExpertSystemShell
 private:
 	ICallback* m_cb;
 protected:
-	std::set<ruleType> knowledge;
-	std::set<ruleType*> rulesToErase;
-	std::set<ruleType*> rulesToAdd;
-	std::set<fact*> factsToErase;
-	std::set<fact*> factsToAdd;
+	std::vector<ruleType> knowledge;
+	std::vector<ruleType*> rulesToErase;
+	std::vector<ruleType*> rulesToAdd;
+	std::vector<fact*> factsToErase;
+	std::vector<fact*> factsToAdd;
 	ui16 goalCounter; //count / evaluate achieved goals for runType
 public:
 	ExpertSystemShell(){goalCounter = 0;};
@@ -73,16 +73,16 @@ public:
 	bool fired; //if conditions of rule were met and it produces some output
 	ui8 conditionCounter;
 protected:
-	std::set<std::pair<conType, input*>> cons; //conditions and matching facts
+	std::vector<std::pair<conType, input*>> cons; //conditions and matching facts
 	input decision;
 	virtual void canBeFired(); //if this data makes any sense for rule - type check
 	virtual bool checkCondition(); //if condition is true or false
-	virtual bool checkCondition(std::set<input*> &feed);
+	virtual bool checkCondition(std::vector<input*> &feed);
 	virtual void fireRule(); //use paired conditions and facts by default
 	virtual void fireRule(ExpertSystemShell<input, conType> &system);
-	virtual void fireRule(std::set<input*> &feed);
+	virtual void fireRule(std::vector<input*> &feed);
 	virtual void refreshRule();
-	virtual void refreshRule(std::set<conType> &conditionSet); //in case conditions were erased
+	virtual void refreshRule(std::vector<conType> &conditionSet); //in case conditions were erased
 public:
 	Rule(){fired = false; conditionCounter = 0; decision = NULL;};
 	template <typename givenInput> bool matchesInput() //if condition and data match type
@@ -183,11 +183,26 @@ public:
 	}
 	bool operator()(int prop, si32 val) const
 	{
-		return first(prop,val) && second(prop,val); //TODO: 
+		return first(prop,val) && second(prop,val);
 	}
 };
 TLogic operator&&(const TLogic &first, const TLogic &second);
 
+class LogicAlternative
+{
+	const TLogic first, second;
+public:
+	LogicAlternative(const TLogic First, const TLogic Second)
+		:first(First), second(Second)
+	{
+	}
+	bool operator()(int prop, si32 val) const
+	{
+		return first(prop,val) || second(prop,val); 
+	}
+};
+TLogic operator||(const TLogic &first, const TLogic &second);
+
 class KnowledgeHandler///I'd opt for one omniscent knowledge manager, so no templates here
 {
 public: