X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Frangecondition.h;h=a72fec2d1475db075baffbeb1ed46aa3979480e0;hb=746329e61ac9f26962af740b0b11dee716803521;hp=c107287919f2fe69140fe347ed54c659ad74bdc6;hpb=715006802a65fa5f541f2f4bcccbdd62d1a2bf09;p=poefilter.git diff --git a/source/rangecondition.h b/source/rangecondition.h index c107287..a72fec2 100644 --- a/source/rangecondition.h +++ b/source/rangecondition.h @@ -19,7 +19,10 @@ public: RangeCondition(Type); RangeCondition(Type, Type); - virtual Condition *clone() const; + virtual RangeCondition *clone() const; + virtual bool can_merge(const Condition &, const CompoundCondition &) const; + virtual RangeCondition *merge(const std::vector &, const CompoundCondition &) const; + virtual bool is_viable() const { return min<=max; } virtual void add_lines(std::list &) const; }; @@ -49,7 +52,7 @@ typedef RangeCondition QualityCondition; struct ItemLevelTraits { typedef unsigned Type; - static unsigned get_min() { return 0; } + static unsigned get_min() { return 1; } static unsigned get_max() { return 100; } static const char *get_keyword() { return "ItemLevel"; } }; @@ -60,7 +63,7 @@ typedef RangeCondition ItemLevelCondition; struct DropLevelTraits { typedef unsigned Type; - static unsigned get_min() { return 0; } + static unsigned get_min() { return 1; } static unsigned get_max() { return 100; } static const char *get_keyword() { return "DropLevel"; } }; @@ -125,11 +128,55 @@ RangeCondition::RangeCondition(Type n, Type x): { } template -Condition *RangeCondition::clone() const +RangeCondition *RangeCondition::clone() const { return new RangeCondition(min, max); } +template +bool RangeCondition::can_merge(const Condition &other, const CompoundCondition &parent) const +{ + const RangeCondition *other_range = dynamic_cast *>(&other); + if(!other_range) + return false; + + if(dynamic_cast(&parent)) + return true; + else if(dynamic_cast(&parent)) + return min<=other_range->max+1 && max+1>=other_range->min; + else + return false; +} + +template +RangeCondition *RangeCondition::merge(const std::vector &conditions, const CompoundCondition &parent) const +{ + if(conditions.empty()) + return 0; + + bool intersect = dynamic_cast(&parent); + RangeCondition *result; + if(intersect) + result = new RangeCondition(Traits::get_min(), Traits::get_max()); + else + result = new RangeCondition(Traits::get_max(), Traits::get_min()); + for(std::vector::const_iterator i=conditions.begin(); i!=conditions.end(); ++i) + { + const RangeCondition *c = static_cast *>(*i); + if(intersect) + { + result->min = std::max(result->min, c->min); + result->max = std::min(result->max, c->max); + } + else + { + result->min = std::min(result->min, c->min); + result->max = std::max(result->max, c->max); + } + } + return result; +} + template void RangeCondition::add_lines(std::list &st) const {