DtStreamedSpeedTreeThread.h

Go to the documentation of this file.

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


#pragma once

#include <vrvSt/vrvSt.h>

#include <vrvCore/DtTreeRecord.h>

#include <vrvUtil/DtVirtualBaseClass.h>

#include <osgdb_osgearth_model_feature_custom/CustomFeatureNode.h>

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

#include <boost/thread/mutex.hpp>
#include <boost/unordered_set.hpp>

#include <deque>

namespace osg
{
   class Node;
}

namespace makVrv 
{
   class DtDe;
   class DtStreamedSpeedTreeForest;

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

      virtual ~DtStreamedSpeedTreeThread();

      enum RequestStatus { UNUSED, WORKING, DONE };

      struct TreeRequestOutput
      {
         DtVectorTreeRecords records;

         DtStreamedSpeedTreeForest* forest;

         const osg::Node* dataNode;

         volatile RequestStatus status;
      };

      struct TreeRequestMsg
      {
         osgEarth::Features::CustomFeatureNode::MarkerRequestList requests;

         TreeRequestOutput* out;

         DtStreamedSpeedTreeForest* forest;

         bool useScaleRandomization;
         bool useOrientationRandomization;
         float treeMinScale;
         float treeMaxScale;

         const osg::Node* dataNode;
      };

      typedef std::list<TreeRequestOutput> RequestOutputList;

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

      typedef boost::unordered_set<DtStreamedSpeedTreeForest*> ForestSet;

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

      RequestOutputList& requestOutput();

      NodeSet& unloaded();

      void setForestDeleted(DtStreamedSpeedTreeForest* forest);

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

      void enqueue(
         const osgEarth::Features::CustomFeatureNode::MarkerRequestList& requests,
         TreeRequestOutput* out,
         DtStreamedSpeedTreeForest* forest,
         const osg::Node* dataNode,
         bool useScaleRandomization, 
         bool useOrientationRandomization,
         float treeMinScale,
         float treeMaxScale);

      virtual void run();

      void handleRequest(const TreeRequestMsg& msg);

      bool isForestDeleted(DtStreamedSpeedTreeForest* forest);

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

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

      static void mergeTreeRequests(DtDe& de);

      void block();

      void release();

      virtual void updateBlock();

   protected:

      DtStreamedSpeedTreeThread(DtDe& de);

   protected:

      DtDe& myDe;

      std::deque<TreeRequestMsg> 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;
   };
}