v0.2.52..v0.2.53 changeset AreaCriterion.h - ngageoint/hootenanny GitHub Wiki

diff --git a/hoot-core/src/main/cpp/hoot/core/criterion/AreaCriterion.h b/hoot-core/src/main/cpp/hoot/core/criterion/AreaCriterion.h
index a16ce0d..77f6b97 100644
--- a/hoot-core/src/main/cpp/hoot/core/criterion/AreaCriterion.h
+++ b/hoot-core/src/main/cpp/hoot/core/criterion/AreaCriterion.h
@@ -22,12 +22,14 @@
  * This will properly maintain the copyright information. DigitalGlobe
  * copyrights will be updated automatically.
  *
- * @copyright Copyright (C) 2016, 2017, 2018, 2019 DigitalGlobe (http://www.digitalglobe.com/)
+ * @copyright Copyright (C) 2016, 2017, 2018, 2019, 2020 DigitalGlobe (http://www.digitalglobe.com/)
  */
 #ifndef AREACRITERION_H
 #define AREACRITERION_H
 
 // hoot
+#include <hoot/core/elements/OsmMap.h>
+#include <hoot/core/elements/ConstOsmMapConsumer.h>
 #include <hoot/core/criterion/ConflatableElementCriterion.h>
 
 namespace hoot
@@ -36,13 +38,14 @@ namespace hoot
 /**
  * A criterion that will either keep or remove areas.
  */
-class AreaCriterion : public ConflatableElementCriterion
+class AreaCriterion : public ConflatableElementCriterion, public ConstOsmMapConsumer
 {
 public:
 
   static std::string className() { return "hoot::AreaCriterion"; }
 
   AreaCriterion();
+  AreaCriterion(ConstOsmMapPtr map);
 
   virtual bool isSatisfied(const ConstElementPtr& e) const override;
 
@@ -51,12 +54,18 @@ public:
   virtual GeometryType getGeometryType() const
   { return GeometryType::Polygon; }
 
-  virtual ElementCriterionPtr clone() { return ElementCriterionPtr(new AreaCriterion()); }
+  virtual ElementCriterionPtr clone() { return ElementCriterionPtr(new AreaCriterion(_map)); }
 
   virtual QString getDescription() const { return "Identifies areas"; }
 
   virtual QString toString() const override
   { return QString::fromStdString(className()).remove("hoot::"); }
+
+  virtual void setOsmMap(const OsmMap* map) { _map = map->shared_from_this(); }
+
+private:
+
+  ConstOsmMapPtr _map;
 };
 
 }