query.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2013 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/


#pragma once

#include <features/feature.h>

#include <boost/logic/tribool.hpp>
#include <boost/cast.hpp>

#include <iosfwd>

namespace MAKVRinTerra
{
   class DtTrueQuery;
   class DtFalseQuery;
   class DtAndQuery;
   class DtOrQuery;

   class DT_DLL_features DtQuery
   {
   public:
      enum Type
      {
         EOther,
         ETrue,
         EFalse,
         EAnd,
         EOr
      };

      typedef DtTrueQuery  True;
      typedef DtFalseQuery False;
      typedef DtAndQuery   And;
      typedef DtOrQuery    Or;

      template <typename Result = void>
      class Visitor;

      class DT_DLL_features Implementation
         : public std::enable_shared_from_this<Implementation>
         , private boost::noncopyable
      {
      public:
         typedef std::shared_ptr<Implementation>       Ptr;
         typedef std::shared_ptr<const Implementation> CPtr;

         typedef DtQuery::Type Type;

         virtual ~Implementation();

         virtual Type type() const;

         virtual bool matches(const DtFeature&) const = 0;

         virtual void print(std::ostream&) const = 0;
         virtual void pprint(std::ostream&, unsigned indent) const;
      };

      DtQuery();
      DtQuery(bool val);
      DtQuery(const Implementation& i) : myImpl(i.shared_from_this()) { }
      DtQuery(Implementation* i) : myImpl(i) { }
      DtQuery(Implementation::CPtr p) : myImpl(p) { }

      const Implementation& impl() const
      {
         return *myImpl;
      }

      template <typename T>
      const T& cast() const
      {
         return *boost::polymorphic_downcast<const T*>(myImpl.get());
      }

      template <typename T>
      const T* trycast() const
      {
         return dynamic_cast<const T*>(myImpl.get());
      }

      Type type() const { return myImpl->type(); }

      bool isTrue() const { return type() == ETrue; }
      bool isFalse() const { return type() == EFalse; }

      bool matches(const DtFeature& feature) const;

      boost::optional<DtQuery> remove(boost::function<bool (const DtQuery&)>) const;

      void print(std::ostream& stream) const;

      void pprint(std::ostream& stream, unsigned indent = 0) const;
      std::string pprint(unsigned indent = 0) const;

   protected:
      Implementation::CPtr myImpl;
   };

   DT_DLL_features DtQuery operator|| (const DtQuery&, const DtQuery&);
   DT_DLL_features DtQuery operator&& (const DtQuery&, const DtQuery&);

   DT_DLL_features std::ostream& operator<< (std::ostream&, const DtQuery&);
   DT_DLL_features std::istream& operator>> (std::istream&, DtQuery&);

   template <typename Result>
   class DT_DLL_features DtQuery::Visitor
   {
   public:
      virtual ~Visitor() { }

      virtual Result apply(const DtQuery& query)
      {
         switch (query.type())
         {
         case ETrue:    return visit(query.cast<True>());
         case EFalse:   return visit(query.cast<False>());
         case EAnd:     return visit(query.cast<And>());
         case EOr:      return visit(query.cast<Or>());
         case EOther:   return visit(query);

         default:
            assert(false);
            DtTHROW_NEW(DtException, "logic error");
         }
      }

      virtual Result visit(const DtQuery&) = 0;
      virtual Result visit(const True&) = 0;
      virtual Result visit(const False&) = 0;
      virtual Result visit(const And&) = 0;
      virtual Result visit(const Or&) = 0;
   };

   template <>
   class DT_DLL_features DtQuery::Visitor<DtQuery>
   {
   public:
      virtual ~Visitor();

      virtual DtQuery apply(const DtQuery&);

      virtual DtQuery visit(const DtQuery&) = 0;
      virtual DtQuery visit(const True&);
      virtual DtQuery visit(const False&);
      virtual DtQuery visit(const And&);
      virtual DtQuery visit(const Or&);
   };

   template <>
   class DT_DLL_features DtQuery::Visitor<void>
   {
   public:
      virtual ~Visitor();

      virtual void apply(const DtQuery&);

      virtual void visit(const DtQuery&) = 0;
      virtual void visit(const True&);
      virtual void visit(const False&);
      virtual void visit(const And&);
      virtual void visit(const Or&);
   };

   class DT_DLL_features DtTrueQuery : public DtQuery::Implementation
   {
   public:
      static DtQuery Create();

      virtual Type type() const override;
      virtual bool matches(const DtFeature& feature) const override;
      virtual void print(std::ostream&) const override;
   };

   class DT_DLL_features DtFalseQuery : public DtQuery::Implementation
   {
   public:
      static DtQuery Create();

      virtual Type type() const override;
      virtual bool matches(const DtFeature& feature) const override;
      virtual void print(std::ostream&) const override;
   };

   class DT_DLL_features DtAndQuery : public DtQuery::Implementation
   {
   public:
      friend class DtQuery;

      static DtQuery Create(const DtQuery&, const DtQuery&);

      virtual Type type() const override;
      virtual bool matches(const DtFeature& feature) const override;
      virtual void print(std::ostream&) const override;
      virtual void pprint(std::ostream&, unsigned indent) const override;

      const DtQuery& left() const;
      const DtQuery& right() const;

   protected:
      DtAndQuery(const DtQuery& left, const DtQuery& right);

   private:
      DtQuery myLeftQuery;
      DtQuery myRightQuery;
   };

   class DT_DLL_features DtOrQuery : public DtQuery::Implementation
   {
   public:
      friend class DtQuery;

      static DtQuery Create(const DtQuery&, const DtQuery&);

      virtual Type type() const override;
      virtual bool matches(const DtFeature& feature) const override;
      virtual void print(std::ostream&) const override;
      virtual void pprint(std::ostream&, unsigned indent) const override;

      const DtQuery& left() const;
      const DtQuery& right() const;

   protected:
      DtOrQuery(const DtQuery& left, const DtQuery& right);

   private:
      DtQuery myLeftQuery;
      DtQuery myRightQuery;
   };

   class DT_DLL_features DtHasGeometry : public DtQuery::Implementation
   {
   public:
      static DtQuery Create(DtFeatureGeometry::Type);

      virtual bool matches(const DtFeature&) const override;
      virtual void print(std::ostream&) const override;

      DtFeatureGeometry::Type geometryType() const;

   protected:
      explicit DtHasGeometry(DtFeatureGeometry::Type);

   private:
      DtFeatureGeometry::Type myType;
   };

   class DT_DLL_features DtHasAttribute : public DtQuery::Implementation
   {
   public:
      static DtQuery Create(DtFeature::Key);
      static DtQuery Create(std::string);
      static DtQuery Create(const char*);

      virtual bool matches(const DtFeature&) const override;
      virtual void print(std::ostream&) const override;

      const DtFeature::Key& key() const;

      DtQuery operator == (const DtFeature::Attribute&) const;
      DtQuery operator < (const DtFeature::Attribute&) const;
      DtQuery operator > (const DtFeature::Attribute&) const;
      DtQuery operator <= (const DtFeature::Attribute&) const;
      DtQuery operator >= (const DtFeature::Attribute&) const;

   protected:
      explicit DtHasAttribute(const DtFeature::Key&);

   private:
      DtFeature::Key myKey;
   };

   class DT_DLL_features DtHasAttributeValue : public DtHasAttribute
   {
      typedef DtHasAttribute Base;

   public:
      enum Comparison
      {
         Equals,
         Less,
         LessEqual,
         Greater,
         GreaterEqual,
         Regex
      };

      static bool Matches(
         Comparison, const DtFeature::Attribute&, const DtFeature::Attribute&);

      static bool Matches(
         Comparison,
         const boost::optional<DtFeature::Attribute>&,
         const boost::optional<DtFeature::Attribute>&);

      static DtQuery Merge(const DtHasAttributeValue&, const DtHasAttributeValue&);

      static DtQuery Create(DtFeature::Key, DtFeature::Attribute, Comparison);
      static DtQuery Create(std::string, DtFeature::Attribute, Comparison);
      static DtQuery Create(const char*, DtFeature::Attribute, Comparison);

      virtual bool matches(const DtFeature& feature) const override;
      virtual void print(std::ostream& stream) const override;

      const DtFeature::Attribute& value() const;
      Comparison comparison() const;

   protected:
      explicit DtHasAttributeValue(DtFeature::Key key, DtFeature::Attribute, Comparison);

   private:
      DtFeature::Attribute myValue;
      Comparison myComparison;
   };

   class DT_DLL_features DtHasAttributeValueAttribute : public DtHasAttribute
   {
      typedef DtHasAttribute Base;

   public:
      typedef DtHasAttributeValue::Comparison Comparison;

      static DtQuery Create(DtFeature::Key, DtFeature::Key, Comparison);
      static DtQuery Create(std::string, DtFeature::Key, Comparison);
      static DtQuery Create(const char*, DtFeature::Key, Comparison);

      virtual bool matches(const DtFeature& feature) const override;
      virtual void print(std::ostream& stream) const override;

      const DtFeature::Key& value() const;
      Comparison comparison() const;

   protected:
      explicit DtHasAttributeValueAttribute(DtFeature::Key key, DtFeature::Key, Comparison);

   private:
      DtFeature::Key myAttributeKey;
      Comparison myComparison;
   };
}