DtStreamedSpeedTreeThread.h

Go to the documentation of this file.

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


#pragma once

#include <vrvSt/vrvSt.h>

#include <vrvSt/DtTreeRequestMessage.h>

#include <vrvCore/DtTreeRecord.h>

#include <vrvUtil/DtVirtualBaseClass.h>

#include <osgdb_osgearth_model_feature_custom/CustomFeatureNode.h>

#include <tbb/atomic.h>
#include <tbb/concurrent_queue.h>

#include <unordered_set>
#include <unordered_map>
#include <thread>

#include <deque>

namespace osg
{
   class Node;
   class RenderInfo;
   class EllipsoidModel;
}

namespace osgUtil
{
   class SceneView;
}

namespace makVrv
{
   class DtDe;
   class DtStreamedSpeedTreeForest;

   class DT_DLL_VRVST DtStreamedSpeedTreeThread : public DtVirtualBaseClass
   {
   public:
      static DtStreamedSpeedTreeThread& instance(DtDe& de);

      virtual ~DtStreamedSpeedTreeThread();

   private:
      DtStreamedSpeedTreeThread() = delete;

      DtStreamedSpeedTreeThread(const DtStreamedSpeedTreeThread& rhs) = delete;

      DtStreamedSpeedTreeThread& operator=(const DtStreamedSpeedTreeThread& rhs) = delete;
   public:
      typedef std::unordered_set<const osg::Node*> NodeSet;

      typedef std::unordered_set<DtStreamedSpeedTreeForest*> ForestSet;

      typedef std::unordered_set<std::string> StringSet;

   public:

      virtual void setForestDeleted(DtStreamedSpeedTreeForest* forest);

      virtual void setNodeUnloaded(const osg::Node* node);

      virtual void enqueue(DtStreamedSpeedTreeForest* forest
         , const osgEarth::CustomFeatureNode* featureNode
         , const osg::Node* dataNode
         , bool useScaleRandomization
         , bool useOrientationRandomization
         , float treeMinScale
         , float treeMaxScale);

      virtual bool workToDo(void) const;
   protected:
      DtStreamedSpeedTreeThread(DtDe& de);

      virtual void processMessage(const DtTreeRequestMessage& msg);

      virtual void mergeTreeRequests(DtStreamedSpeedTreeForest* forest
      , const osg::Node* node);

      virtual bool isForestDeleted(DtStreamedSpeedTreeForest* forest) const;

      virtual bool isNodeUnloaded(const osg::Node* node) const;

      virtual double rotationAngleFromGeocentricQuaternion(const osg::Quat& rotation,
         const osg::Vec3d& location) const;

      virtual std::size_t hash_value(const std::string& modelDefName,
         const osgEarth::CustomFeatureNode::OsgEarthMarkerRequest& val) const;

      virtual void run();

      virtual DtTreeRecord* getTreeRecordFromPool(void);

      virtual void returnTreeRecordToPool(DtTreeRecord* treeRecord);

   protected:
      DtDe& myDe;

      tbb::concurrent_bounded_queue<DtTreeRequestMessage*> myMessageQueue;

      tbb::atomic< bool> myStopWork;

      std::thread myWorkerThread;

      StringSet myValidModelDefinitions;

      ForestSet myDeletedForests;

      NodeSet myUnloadedNodes;

      osg::EllipsoidModel* myEllipsoidModel;

      std::unordered_map<std::string, std::string> myModelDefinitionCache;
   
      std::unordered_map<std::string, std::vector<DtTreeRecord*> > myScratchModelDefsToTreesToAdd;

      std::deque<DtTreeRecord*> myTreeRecordPool;
   };
}