X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fcondition.cpp;h=4d00b6c7a75a3c183aca2367fb474deb04615108;hb=eac6a71e79797e6d37b620a91acb3c0541f1c07e;hp=42e9ee8f3f1360829d2b518d9d22e99a569f0a69;hpb=8b9d1f472e7bfbb1e097bb6d8bcad026e9636822;p=poefilter.git

diff --git a/source/condition.cpp b/source/condition.cpp
index 42e9ee8..4d00b6c 100644
--- a/source/condition.cpp
+++ b/source/condition.cpp
@@ -49,11 +49,15 @@ Condition *CompoundCondition::flatten() const
 			Condition *sub = (*i)->flatten();
 			if(!result)
 				result = sub;
-			else
+			else if(sub)
 				result = dispatch_flatten(result, sub);
 		}
 
-		return result;
+		if(result && result->is_viable())
+			return result;
+
+		delete result;
+		return 0;
 	}
 }
 
@@ -91,6 +95,70 @@ Condition *CompoundCondition::dispatch_flatten(Condition *cond1, Condition *cond
 	throw logic_error("CompoundCondition::dispatch_flatten");
 }
 
+Condition *CompoundCondition::merge_two(Condition *cond1, Condition *cond2, bool del) const
+{
+	vector<Condition *> parts;
+	parts.reserve(2);
+	parts.push_back(cond1);
+	parts.push_back(cond2);
+	Condition *result = cond1->merge(parts, *this);
+
+	if(del)
+	{
+		delete cond1;
+		delete cond2;
+
+		if(!result->is_viable())
+		{
+			delete result;
+			return 0;
+		}
+	}
+
+	return result;
+}
+
+Condition *CompoundCondition::add_merged_to(Condition *cond, CompoundCondition *target, bool del) const
+{
+	bool merged = false;
+	for(vector<Condition *>::iterator i=target->conditions.begin(); i!=target->conditions.end(); ++i)
+		if((*i)->can_merge(*cond, *target))
+		{
+			Condition *m = merge_two(cond, *i, false);
+			delete *i;
+			if(del)
+				delete cond;
+			*i = m;
+			merged = true;
+			break;
+		}
+
+	if(!merged)
+		target->add(del ? cond : cond->clone());
+
+	if(del && !target->is_viable())
+	{
+		delete target;
+		return 0;
+	}
+
+	return target;
+}
+
+Condition *CompoundCondition::merge_contents_to(CompoundCondition *cond, CompoundCondition *target) const
+{
+	for(vector<Condition *>::iterator i=cond->conditions.begin(); i!=cond->conditions.end(); ++i)
+		add_merged_to(*i, target, false);
+
+	delete cond;
+
+	if(target->is_viable())
+		return target;
+
+	delete target;
+	return 0;
+}
+
 
 AndCondition *AndCondition::clone() const
 {
@@ -101,6 +169,9 @@ AndCondition *AndCondition::clone() const
 
 Condition *AndCondition::flatten(Condition *cond1, Condition *cond2) const
 {
+	if(cond1->can_merge(*cond2, *this))
+		return merge_two(cond1, cond2, true);
+
 	AndCondition *result = new AndCondition;
 	result->add(cond1);
 	result->add(cond2);
@@ -109,17 +180,12 @@ Condition *AndCondition::flatten(Condition *cond1, Condition *cond2) const
 
 Condition *AndCondition::flatten(AndCondition *cond1, Condition *cond2) const
 {
-	cond1->add(cond2);
-	return cond1;
+	return add_merged_to(cond2, cond1, true);
 }
 
 Condition *AndCondition::flatten(AndCondition *cond1, AndCondition *cond2) const
 {
-	unsigned count2 = cond2->count();
-	for(unsigned i=0; i<count2; ++i)
-		cond1->add(cond2->get(i)->clone());
-	delete cond2;
-	return cond1;
+	return merge_contents_to(cond2, cond1);
 }
 
 Condition *AndCondition::flatten(OrCondition *cond1, Condition *cond2) const
@@ -127,9 +193,15 @@ Condition *AndCondition::flatten(OrCondition *cond1, Condition *cond2) const
 	OrCondition *result = new OrCondition;
 	unsigned count = cond1->count();
 	for(unsigned i=0; i<count; ++i)
-		result->add(dispatch_flatten(cond1->get(i)->clone(), (i+1<count ? cond2->clone() : cond2)));
+		if(Condition *sub = dispatch_flatten(cond1->get(i)->clone(), (i+1<count ? cond2->clone() : cond2)))
+			result->add(sub);
 	delete cond1;
-	return result;
+
+	if(result->is_viable())
+		return result;
+
+	delete result;
+	return 0;
 }
 
 Condition *AndCondition::flatten(OrCondition *cond1, AndCondition *cond2) const
@@ -144,10 +216,25 @@ Condition *AndCondition::flatten(OrCondition *cond1, OrCondition *cond2) const
 	OrCondition *result = new OrCondition;
 	for(unsigned i=0; i<count1; ++i)
 		for(unsigned j=0; j<count2; ++j)
-			result->add(dispatch_flatten(cond1->get(i)->clone(), cond2->get(j)->clone()));
+			if(Condition *sub = dispatch_flatten(cond1->get(i)->clone(), cond2->get(j)->clone()))
+				result->add(sub);
+
 	delete cond1;
 	delete cond2;
-	return result;
+
+	if(result->is_viable())
+		return result;
+
+	delete result;
+	return 0;
+}
+
+bool AndCondition::is_viable() const
+{
+	for(vector<Condition *>::const_iterator i=conditions.begin(); i!=conditions.end(); ++i)
+		if(!(*i)->is_viable())
+			return false;
+	return !conditions.empty();
 }
 
 void AndCondition::add_lines(list<FilterStatement> &st) const
@@ -166,6 +253,9 @@ OrCondition *OrCondition::clone() const
 
 Condition *OrCondition::flatten(Condition *cond1, Condition *cond2) const
 {
+	if(cond1->can_merge(*cond2, *this))
+		return merge_two(cond1, cond2, true);
+
 	OrCondition *result = new OrCondition;
 	result->add(cond1);
 	result->add(cond2);
@@ -184,8 +274,7 @@ Condition *OrCondition::flatten(AndCondition *cond1, AndCondition *cond2) const
 
 Condition *OrCondition::flatten(OrCondition *cond1, Condition *cond2) const
 {
-	cond1->add(cond2);
-	return cond1;
+	return add_merged_to(cond2, cond1, true);
 }
 
 Condition *OrCondition::flatten(OrCondition *cond1, AndCondition *cond2) const
@@ -195,11 +284,15 @@ Condition *OrCondition::flatten(OrCondition *cond1, AndCondition *cond2) const
 
 Condition *OrCondition::flatten(OrCondition *cond1, OrCondition *cond2) const
 {
-	unsigned count2 = cond2->count();
-	for(unsigned i=0; i<count2; ++i)
-		cond1->add(cond2->get(i)->clone());
-	delete cond2;
-	return cond1;
+	return merge_contents_to(cond2, cond1);
+}
+
+bool OrCondition::is_viable() const
+{
+	for(vector<Condition *>::const_iterator i=conditions.begin(); i!=conditions.end(); ++i)
+		if((*i)->is_viable())
+			return true;
+	return false;
 }
 
 void OrCondition::add_lines(list<FilterStatement> &st) const