DtStreamedSpeedTreeForest.h

Go to the documentation of this file.

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


#pragma once

#include <vrvCore/DtUniqueID.h>
#include <vrvSt/vrvSt.h>

#include <vrvCore/DtTreeRecord.h>

#include <unordered_map>
#include <unordered_set>

#include <tbb/mutex.h>

namespace osg
{
   class Node;
   class Observer;
}

namespace osgEarth
{
   class Map;
   class MapNode;
   class Layer;
   class VisibleLayer;
}

namespace osgEarth
{
   class CustomFeatureNode;
}

namespace makVrv 
{
   class DtDe;
   class DtSpeedTreeForest;
   class DtOsgTerrainPatchObject;
   class CellNodeObserver;

   class DtStreamedSpeedTreeForest
   {
   public:
      DtStreamedSpeedTreeForest(DtDe& de, DtUniqueID id);
      virtual ~DtStreamedSpeedTreeForest();

      void onDataLoaded(const DtOsgTerrainPatchObject& patch, 
         const osgEarth::Layer* layer, const osg::Node* data);
      void onDataUnLoaded(const osgEarth::Layer* layer,
         const osg::Node* data);

      void onLayerVisibilityChanged(const osgEarth::VisibleLayer* layer);

      void onLayerLoaded(const osgEarth::MapNode* mapNode,
         osgEarth::Layer* layer);
      void onLayerUnLoaded(const osgEarth::MapNode* mapNode,
         const osgEarth::Layer* layer);
      
      void mapDataToCell(const osg::Node* dataNode, DtUniqueID cellID);

      void removeForestFromMapData(const osg::Node* dataNode);

      DtSpeedTreeForest* forest();
   protected:
      void processNewFeatures(const DtOsgTerrainPatchObject& patch,
         const osg::Node* data);
      void processNewFeature(const DtOsgTerrainPatchObject& patch,
         const osgEarth::CustomFeatureNode* pFeatureNode, const osg::Node* data);

      bool observerExists(const osg::Node* data);
      void createObserver(const osgEarth::Layer* layer,
         const osg::Node* data);
      void destroyObserver(const osg::Node* data, bool bCheck);
      void removeAndDestroyAllObservers(void);

   protected:
      typedef std::map<std::string, std::pair<std::string, bool> > MarkerToModelDefinition;
      static bool verifyExists(const osg::Observer* lhs, const osg::Node* data);
      static bool getValidTreeModelDefinition(DtDe& de, 
         std::string& strCleanModelName, const std::string& strModelName,
         MarkerToModelDefinition& cache);
      typedef std::unordered_set<std::string> ModelDefinitionSet;
   protected:
      ModelDefinitionSet myValidModelDefinitions;
      DtSpeedTreeForest* myForest;
      DtVectorTreeRecords myScratchVecTreeRecords;
      DtDe& myDe;

      typedef std::unordered_map<const osg::Node*, CellNodeObserver*>
         MapCellsToObservers;
      MapCellsToObservers myMapCellsToObservers;

      typedef std::set<DtUniqueID> IdSet;
      typedef std::unordered_map<const osg::Node*, IdSet> MapCellsToSubCells;
      MapCellsToSubCells myMapCellsToSubCells;
      tbb::mutex myMapCellsToSubCellsMutex;
   };
}