osgEarthFeatureSet.h

Go to the documentation of this file.

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

#pragma once

#include <features/feature.h>
#include <features/featureSet.h>

#include <osgEarth/Map>
#include <osgEarth/FeatureSource>
#include <osgEarth/Query>
#include <osgEarth/MapNode>
#include <osgEarth/Geometry>

namespace MAKVRinTerra
{
   class DT_DLL_features DtOsgEarthFeature : public DtFeature
   {
   public:
      typedef osg::ref_ptr<osgEarth::Feature> OsgEarthFeaturePtr;

      static DtFeatureGeometry::GeometryVector ConvertGeometry(
         const osgEarth::Geometry* geom, const DtProj& proj);

      DtOsgEarthFeature(OsgEarthFeaturePtr, const DtProj&);

      virtual unsigned numberOfAttributes(const Key::String&) const override;

      virtual boost::optional<Attribute> attribute(const Key&) const override;

      virtual DtFeatureGeometry geometry() const override;

      virtual void forEachAttribute(ForEachFunction) const override;

      virtual DetachCallback addDeleteCallback(const DeleteCallback&) const override { return DetachCallback(); }

      virtual bool isDeleted() const override {
         return false;
      }

   protected:
      OsgEarthFeaturePtr myFeature;
      DtProj::CPtr myProjection;
      
      typedef osgEarth::AttributeTable AttributeTable;
      typedef std::map<Key::String, AttributeTable::value_type> StringAttributeTable;
      StringAttributeTable myAttributes;
   };

   class DT_DLL_features DtOsgEarthFeatureWithGeometry : public DtOsgEarthFeature
   {
   public:
      DtOsgEarthFeatureWithGeometry(OsgEarthFeaturePtr, DtFeatureGeometry);

      virtual DtFeatureGeometry geometry() const override;

   protected:
      DtFeatureGeometry myGeometry;
   };

   class DT_DLL_features DtOsgEarthFeatureSet : public DtFeatureSet
   {
   public:
      typedef std::set<DtFeature::Key> AttributeSet;

      static DtFeatureSet* Make(osg::ref_ptr<osgEarth::FeatureSource> featureSource);

      explicit DtOsgEarthFeatureSet(osg::ref_ptr<osgEarth::FeatureSource> featureSource, const std::string& driver);

      std::string labelWithoutBounds() const;
      virtual std::string label() const override;
      virtual void forEachFeatureParallel(const ForEachFunction&) const override;
      virtual DetachCallback forEachFeatureAsync(const ForEachFunction&) const override;
      virtual DtFeatureSet* filter(
         const DtQuery&, const boost::optional<DtFeatureGeometry>&, DtLoadType) const override;
      virtual DtFeatureSet* clone() const override;
      virtual bool loaded(double) const override;
      virtual DetachCallback addLoadedCallback(const LoadedCallback&) const override;
      virtual bool elided() const override;
      virtual bool empty() const override;
      virtual bool isCacheable(const boost::optional<DtFeatureGeometry>& area) const override;
      virtual void cacheEstimate(
         double& time, double& size, const boost::optional<DtFeatureGeometry>& area) const override;
      virtual void cacheData(
         const boost::optional<DtFeatureGeometry>& area, DtFeatureCacheProgressCallback* progress) const override;

   protected:
      explicit DtOsgEarthFeatureSet(const DtOsgEarthFeatureSet&);
      DtOsgEarthFeatureSet(const DtOsgEarthFeatureSet&, boost::optional<DtFeatureGeometry>);

      virtual void convertBoundsQueryToTileQueries(
         std::list<osgEarth::Query>& output) const;
      virtual void convertBoundsQueryToTileQueries(
         const osgEarth::TileKey& root,
         std::list<osgEarth::Query>& output) const;

      osg::ref_ptr<osgEarth::FeatureSource> myFeatureSource;
      bool myErrorFlag;
      DtProj::CPtr myProjection;
      boost::optional<DtFeatureGeometry> myGeometryFilter;
      osgEarth::Query myQuery;
      AttributeSet myAttributes;
      bool myAlwaysLoadFlag;
      bool myIsTiled;
      int myMaxTileLod;
   };
}