include/vrvOsg/DtOsgReferencedCollection.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2010 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/
/******************************************************************************
** $RCSfile:  $ $Revision:  $ $State:  $
******************************************************************************/


#pragma once
#include <string>
#include <osg/ref_ptr>
#include <osg/Referenced>
#include <osg/Texture2D>
#include <osgDB/Registry>
#include <vrvOsg/DtOsgFileCache.h>
#include <boost/thread/mutex.hpp>
#include <boost/unordered_set.hpp>
#include <boost/unordered_map.hpp>
#include <boost/thread.hpp>
#include <vector>

namespace makVrv
{
   template <typename OsgObject>
   class DtOsgReferencedCollection
   {
   public:

      typedef boost::unordered_map<std::string,osg::ref_ptr<OsgObject> >
         InstanceMap;
      typedef std::vector<std::pair<std::string,OsgObject*> > ReferenceList;

      DtOsgReferencedCollection(unsigned int initialCapacity);

      inline void lock() { myMutex.lock(); }

      inline ReferenceList& getUnsafeList() { return myRefInstanceList; }

      inline InstanceMap& getUnsafeMap() { return myInstanceMap; }

      inline void unlock() { myMutex.unlock(); }

      bool find(const std::string& name, osg::ref_ptr<OsgObject>& object );

      void insert(const std::string& name,const osg::ref_ptr<OsgObject>& obj);

      void releaseGLObjects(osg::State* state);

      void clear();

      osg::ref_ptr<OsgObject> find(const std::string& filename);

      osg::ref_ptr<osg::Texture> findOrAddTexture(const std::string& filename,
         osg::Texture::FilterMode minFilter,osg::Texture::FilterMode magFilter,
         osg::Texture::WrapMode s, osg::Texture::WrapMode t);

      unsigned int flush();

      unsigned int partialFlush(unsigned int maxToCheck,unsigned int maxToFlush);

      unsigned int partialFlushTextures(unsigned int maxToCheck,unsigned int maxToFlush);

   protected:
   
      struct FindReference
      {
         const std::string& myStr;
         FindReference(const std::string& str)  : myStr(str){}
         bool operator()(const std::pair<std::string,OsgObject*>& ref) 
         {
            return ref.first == myStr;
         }
      };

      typedef boost::unordered_set<std::string> PendingSet;
      boost::mutex myMutex;
      InstanceMap myInstanceMap;
      ReferenceList myRefInstanceList;
      PendingSet myPendingInstances;
      unsigned int myPartialFlushIndex;
   };

}