Tighten some parameters and return values

[poefilter.git] / source / condition.h

diff --git a/source/condition.h b/source/condition.h
index 53f2a8a8268be8e0248f76d41a1032b8a89516b7..bbabf22e5f27d98c8de47b0b4a58622a910dff63 100644 (file)
--- a/source/condition.h
+++ b/source/condition.h
@@ -18,12 +18,17 @@ public:
        virtual ~Condition() { }
 
        virtual Condition *clone() const = 0;
+       virtual Condition *flatten() const { return clone(); }
        virtual bool can_merge(const Condition &, const CompoundCondition &) const { return false; }
+       virtual Condition *merge(const std::vector<const Condition *> &, const CompoundCondition &) const { return 0; }
+       virtual bool is_viable() const { return true; }
        virtual void add_lines(std::list<FilterStatement> &) const = 0;
-       virtual void add_merged_lines(const std::vector<Condition *> &, const CompoundCondition &, std::list<FilterStatement> &) const { }
 };
 
 
+class AndCondition;
+class OrCondition;
+
 class CompoundCondition: public Condition
 {
 protected:
@@ -37,6 +42,21 @@ protected:
 
 public:
        void add(Condition *);
+       unsigned count() const { return conditions.size(); }
+       const Condition &get(unsigned) const;
+
+       virtual Condition *flatten() const;
+protected:
+       Condition *dispatch_flatten(Condition *, Condition *) const;
+       Condition *merge_two(Condition *, Condition *, bool) const;
+       Condition *add_merged_to(Condition *, CompoundCondition *, bool) const;
+       Condition *merge_contents_to(CompoundCondition *, CompoundCondition *) const;
+       virtual Condition *flatten(Condition *, Condition *) const = 0;
+       virtual Condition *flatten(AndCondition *, Condition *) const = 0;
+       virtual Condition *flatten(AndCondition *, AndCondition *) const = 0;
+       virtual Condition *flatten(OrCondition *, Condition *) const = 0;
+       virtual Condition *flatten(OrCondition *, AndCondition *) const = 0;
+       virtual Condition *flatten(OrCondition *, OrCondition *) const = 0;
 };
 
 
@@ -44,6 +64,15 @@ class AndCondition: public CompoundCondition
 {
 public:
        virtual AndCondition *clone() const;
+protected:
+       virtual Condition *flatten(Condition *, Condition *) const;
+       virtual Condition *flatten(AndCondition *, Condition *) const;
+       virtual Condition *flatten(AndCondition *, AndCondition *) const;
+       virtual Condition *flatten(OrCondition *, Condition *) const;
+       virtual Condition *flatten(OrCondition *, AndCondition *) const;
+       virtual Condition *flatten(OrCondition *, OrCondition *) const;
+public:
+       virtual bool is_viable() const;
        virtual void add_lines(std::list<FilterStatement> &) const;
 };
 
@@ -52,6 +81,15 @@ class OrCondition: public CompoundCondition
 {
 public:
        virtual OrCondition *clone() const;
+protected:
+       virtual Condition *flatten(Condition *, Condition *) const;
+       virtual Condition *flatten(AndCondition *, Condition *) const;
+       virtual Condition *flatten(AndCondition *, AndCondition *) const;
+       virtual Condition *flatten(OrCondition *, Condition *) const;
+       virtual Condition *flatten(OrCondition *, AndCondition *) const;
+       virtual Condition *flatten(OrCondition *, OrCondition *) const;
+public:
+       virtual bool is_viable() const;
        virtual void add_lines(std::list<FilterStatement> &) const;
 };