osgEarthGroundCoverSourcePager.h

Go to the documentation of this file.

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


#pragma once

#include <vantageTerrainImplementation/terrainSourcePager.h>
#include <vantageTerrainImplementation/tileData.h>
#include <vantageTerrainImplementation/tileSpatialIndex.h>
#include <vantageTerrainImplementation/GroundFeatureProcess.h>

#include <osgEarthSplat/GroundCoverLayer>
#include <osgEarthSplat/GroundCoverFeatureGenerator>

#include <osgEarth/Feature>

#include <tbb/spin_rw_mutex.h>
#include <tbb/task_arena.h>

#include <set>

class DtOsgEarthGroundCoverSourceTracker;

class DT_DLL_vantageTerrainImplementation DtOsgEarthGroundCoverSourcePager :
   public DtTerrainSourcePager
{
public:

   DtOsgEarthGroundCoverSourcePager(DtVantageTerrainEngine* terrainEngine, osgEarth::MapNode* mapNode);

   virtual ~DtOsgEarthGroundCoverSourcePager();

   virtual bool init();
   
   virtual void tick();

   unsigned reloadQueuedPages();

   virtual bool pageInTiles(DtTileKeyList& tileIds,
      DtTerrainIntersectIDType intersectId = 0);

   virtual bool pageOutTiles(const DtTileKeyList& tilesToRemove);

   virtual DtTerrainSourceTracker* createTerrainSourceTracker(DtVantageTerrainEngine*);

   virtual void updatePageTracker(DtTerrainIntersectIDType intersectId = 0);

   virtual void setGroundCoverLayer(osg::ref_ptr<osgEarth::Splat::GroundCoverLayer> gclayer);

   virtual unsigned int groundCoverLod() const;

   virtual void findMissingTiles(const DtTerrainPagingGeometryPtr& geometry, DtTileKeyList& tiles) const;

   unsigned reloadPagesWithinExtent(const osgEarth::GeoExtent & extent, bool isRemoval = false);

   virtual unsigned queuePageReloadsWithinExtent(const osgEarth::GeoExtent & extent, const DtUUID& dynamicUuid, bool isRemoval = false);

   virtual osg::ref_ptr<osg::MatrixTransform> createTree(const osg::ref_ptr<osgEarth::Feature>& feature);

protected:

   virtual bool reloadTile(const osgEarth::TileKey & tileKey, bool isRemoval = false, unsigned numEvents = 1);

   virtual boost::shared_ptr<DtTileData> createTileDataFromGroundCover(
      const osgEarth::TileKey& tilekey,
      DtTerrainIntersectIDType intersectId,
      bool createIfAlreadyLoaded = false) const;
      
   virtual bool pageInTile(const osgEarth::TileKey& tileKey,
      DtTerrainIntersectIDType intersectId = 0);

   virtual bool pageOutTile(const osgEarth::TileKey& tileKey);

   virtual void addTile(const osgEarth::TileKey& tilekey, const boost::shared_ptr<DtTileData>& tile);

   virtual void addEmptyTile(const osgEarth::TileKey& tilekey, DtTerrainIntersectIDType intersectId = 0);

   class treemodel
   {
   public:
      treemodel(int id, double width, double height, double sim_trunk_height,
         double sim_trunk_width, double sim_crown_offset, double sim_crown_height, double sim_crown_width, osg::ref_ptr<osg::Geode> geode) :
         myId(id),
         myWidth(width),
         myHeight(height),
         myTrunkHeight(sim_trunk_height),
         myTrunkWidth(sim_trunk_width),
         myCrownOffset(sim_crown_offset),
         myCrownHeight(sim_crown_height),
         myCrownWidth(sim_crown_width),
         myGeode(geode)
      {
      }

      treemodel(const treemodel& tree) :
         myId(tree.myId),
         myWidth(tree.myWidth),
         myHeight(tree.myHeight),
         myTrunkHeight(tree.myTrunkHeight),
         myTrunkWidth(tree.myTrunkWidth),
         myCrownOffset(tree.myCrownOffset),
         myCrownHeight(tree.myCrownHeight),
         myCrownWidth(tree.myCrownWidth),
         myGeode(tree.myGeode)
      {
      }

      int myId;
      double myWidth;
      double myHeight;
      double myTrunkHeight;
      double myTrunkWidth;
      double myCrownOffset;
      double myCrownHeight;
      double myCrownWidth;
      osg::ref_ptr<osg::Geode> myGeode;
   };

   typedef std::map<int, treemodel> DtTreeMap;
   DtTreeMap myTreeMap;

   osgEarth::Splat::GroundCoverFeatureGenerator myFeatureGenerator;

   osg::ref_ptr<osgEarth::Splat::GroundCoverLayer> myGcLayer;

   DtOsgEarthGroundCoverSourceTracker* myOsgTracker;

   mutable tbb::spin_rw_mutex myTileKeysForReloadMutex;

   tbb::spin_rw_mutex myCreateTreeMutex;

   tbb::task_arena myArena;

   std::set<int> myErrorMessages;
};