osgEarthVegetationLayerSourcePager.h

Go to the documentation of this file.

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


#pragma once

#include <vantageTerrainImplementation/osgEarthBaseGroundCoverSourcePager.h>

#include <osgEarthProcedural/VegetationLayer>
#include <osgEarthProcedural/VegetationFeatureGenerator>

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

#include <set>
#include <string>
#include <map>

class DtOsgEarthVegetationLayerSourceTracker;

class DT_DLL_vantageTerrainImplementation DtOsgEarthVegetationLayerSourcePager :
   public DtOsgEarthBaseGroundCoverSourcePager
{
public:

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

   virtual ~DtOsgEarthVegetationLayerSourcePager() override;

   virtual bool init() override;
   
   virtual DtTerrainSourceTracker* createTerrainSourceTracker(DtVantageTerrainEngine*) override;

   virtual DtString type() const override;
  
   virtual void setVegetationLayer(osg::ref_ptr<osgEarth::Procedural::VegetationLayer> veglayer);

   virtual void setVegetationLayerGroupName(const std::string& veglayerGroup);

   virtual void setSimVegetationConfigFile(const std::string& simVegetationFile);

   virtual unsigned int vegetationLayerGroupLod() const;

   virtual std::string vegetationLayerGroupName() 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(osgEarth::Procedural::VegetationLayer::Placement& treePlacement);

protected:

   virtual void readXmlConfigFile();

   virtual bool reloadTile(const osgEarth::TileKey& tileKey, bool isRemoval = false, unsigned numEvents = 1) override;
   
   virtual std::shared_ptr<DtTileData> createTileDataFromVegetationLayer(
      const osgEarth::TileKey& tilekey,
      DtTerrainIntersectIDType intersectId,
      bool createIfAlreadyLoaded = false) const;
    
   virtual bool pageInTile(const osgEarth::TileKey& tileKey, DtTerrainIntersectIDType intersectId = 0) override;
   
   virtual bool pageOutTile(const osgEarth::TileKey& tileKey) override;
   
   virtual void addTile(const osgEarth::TileKey& tilekey, const std::shared_ptr<DtTileData>& tile) override;
   
   //virtual void addEmptyTile(const osgEarth::TileKey& tilekey, DtTerrainIntersectIDType intersectId = 0);

   class treemodel
   {
   public:
      treemodel(const std::string& name, 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) :
         myName(name),
         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) :
         myName(tree.myName),
         myWidth(tree.myWidth),
         myHeight(tree.myHeight),
         myTrunkHeight(tree.myTrunkHeight),
         myTrunkWidth(tree.myTrunkWidth),
         myCrownOffset(tree.myCrownOffset),
         myCrownHeight(tree.myCrownHeight),
         myCrownWidth(tree.myCrownWidth),
         myGeode(tree.myGeode)
      {
      }

      std::string myName;
      double myWidth;
      double myHeight;
      double myTrunkHeight;
      double myTrunkWidth;
      double myCrownOffset;
      double myCrownHeight;
      double myCrownWidth;
      osg::ref_ptr<osg::Geode> myGeode;
   };

   typedef std::map<std::string, treemodel> DtTreeMap;
   DtTreeMap myTreeMap;
   DtTreeMap myVegetationConfigMap;
   typedef std::map<std::string, bool> DtTreeErrorMap;
   DtTreeErrorMap myErrorMap;

   std::set<std::string> myErrorMessages;
   
   osgEarth::Procedural::VegetationFeatureGenerator myFeatureGenerator;

   osg::ref_ptr<osgEarth::Procedural::VegetationLayer> myVegLayer;

   std::string myVegetationLayerGroupName;

   std::string mySimVegetationFile;
};