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&);

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

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

      DtFeatureGeometry geometry() const;

      void forEachAttribute(ForEachFunction) const;

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

      bool isDeleted() const
      {
         return false;
      }

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

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

      DtFeatureGeometry geometry() const;

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

   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;
   };
}