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 <osg/ref_ptr>
#include <osg/OperationThread>

#include <OpenThreads/Thread>
#include <OpenThreads/Block>

#include <boost/thread/mutex.hpp>
#include <tbb/atomic.h>

#include <unordered_set>
#include <deque>

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

namespace osgUtil
{
   class SceneView;
}

namespace makVrv 
{
   class DtDe;
   class DtStreamedSpeedTreeForest;
   class DtSpeedTreeMergeJob;
   class DtStreamedSpeedTreeThread;


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

      virtual ~DtStreamedSpeedTreeThread();
   public:
      typedef std::list<DtTreeRequestOutput> RequestOutputList;

      typedef std::unordered_set<const osg::Node*> NodeSet;

      typedef std::unordered_set<DtStreamedSpeedTreeForest*> ForestSet;

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

      RequestOutputList& requestOutput();

      NodeSet& unloaded();

      void setForestDeleted(DtStreamedSpeedTreeForest* forest);

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

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

      virtual void run();

      void handleRequest(const DtTreeRequestMessage& msg);

      bool isForestDeleted(DtStreamedSpeedTreeForest* forest);

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

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

      void mergeTreeRequests();

      void block();

      void release();

      virtual void updateBlock();

      virtual void stopMergingWork();

      virtual bool workToDo() const;
      
      virtual bool fetchAndActivateTask();
   protected:

      DtStreamedSpeedTreeThread(DtDe& de);

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

      private:
         DtStreamedSpeedTreeThread();

   protected:

      DtDe& myDe;

      std::deque<DtTreeRequestMessage*> myMessageQueue;

      // Don't keep spinning if we don't have data
      OpenThreads::Block myBlock;

      boost::mutex myMessageQueueMutex;

      StringSet myValidModelDefinitions;

      ForestSet myDeletedForests;

      RequestOutputList myTreeRequestOutput;

      NodeSet myUnloadedNodes;

      bool myTimeToStop;

      osg::EllipsoidModel* myEllipsoidModel;

      tbb::atomic< bool> myStopWork;
      tbb::atomic< bool> myWorkToDo;
      tbb::atomic < bool> myTbbTaskActive;

   };
}