DtOsgModelPreloader.h

Go to the documentation of this file.

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


#pragma once

#include <string>

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

#include <vrvOsg/vrvOsg.h>

#include <osg/Node>

#include <boost/unordered_map.hpp>

#include <boost/thread.hpp>
#include <tbb/atomic.h>
#include <tbb/task.h>
#ifdef WIN32
#include <winsock2.h>
#endif
#include <tbb/mutex.h>

namespace osgUtil {
   class StateToCompile;
}

namespace osgViewer {
   class CompositeViewer;
}

namespace makVrv
{
   class DtDe;
   class DtModelDefinition;
   class DT_DLL_VRVOSG DtOsgModelPreloader : public DtModelPreloader
   {

   public:

      DtOsgModelPreloader(DtDe& de, DtUniqueID id);

      virtual ~DtOsgModelPreloader();

      virtual bool load(const std::string& modelDefName);
      
      virtual void addToIcoList(osgUtil::StateToCompile* state);
 
      virtual std::list<osgUtil::StateToCompile* >& toIcoList();
      
      virtual void finishPreloading(osgViewer::CompositeViewer& viewer);

      static DtOsgModelPreloader* preloader();

   protected:

      friend class DtPreloadAsyncTask;
      virtual void markTaskFinished(DtModelDefinition & def);

      tbb::atomic<int> myNumberOfActiveTasks;
      tbb::mutex myListMutex;
      tbb::mutex myStdOutMutex;
      std::list<osgUtil::StateToCompile*> myToIcoList;

      // renderer needs access to this to finish preloading not sure the best way to do that.
      static DtOsgModelPreloader * thePreloader;
      static int myFramesTillICOCount;
      static int myNumFramesTillICO;
   };


   class DT_DLL_VRVOSG DtPreloadAsyncTask : public tbb::task
   {
   public:
      DtPreloadAsyncTask(DtDe& de, DtOsgModelPreloader& preloader, DtModelDefinition& definition);
      
      virtual ~DtPreloadAsyncTask();

      virtual tbb::task* execute();

   protected:
      DtDe& myDe;
      DtModelDefinition& myDefinition;
      DtOsgModelPreloader& myPreloader;
      static tbb::mutex theModelNewingMutex;
      static tbb::mutex theOptimizeMutex;
   };
}